CDI-468: add origin shielding

Author: anton-sharonov

client.go

@@ -3,6 +3,7 @@ package gcdn
 import (
 	"github.com/G-Core/gcorelabscdn-go/gcore"
 	"github.com/G-Core/gcorelabscdn-go/origingroups"
+	"github.com/G-Core/gcorelabscdn-go/originshielding"
 	"github.com/G-Core/gcorelabscdn-go/resources"
 	"github.com/G-Core/gcorelabscdn-go/rules"
 	"github.com/G-Core/gcorelabscdn-go/sslcerts"
@@ -18,20 +19,22 @@ type ClientService interface {
 var _ ClientService = (*Service)(nil)
 
 type Service struct {
-	r                   gcore.Requester
-	resourcesService    resources.ResourceService
-	rulesService        rules.RulesService
-	originGroupsService origingroups.OriginGroupService
-	sslCertsService     sslcerts.SSLCertService
+	r                      gcore.Requester
+	resourcesService       resources.ResourceService
+	rulesService           rules.RulesService
+	originGroupsService    origingroups.OriginGroupService
+	originShieldingService originshielding.OriginShieldingService
+	sslCertsService        sslcerts.SSLCertService
 }
 
 func NewService(r gcore.Requester) *Service {
 	return &Service{
-		r:                   r,
-		resourcesService:    resources.NewService(r),
-		rulesService:        rules.NewService(r),
-		originGroupsService: origingroups.NewService(r),
-		sslCertsService:     sslcerts.NewService(r),
+		r:                      r,
+		resourcesService:       resources.NewService(r),
+		rulesService:           rules.NewService(r),
+		originGroupsService:    origingroups.NewService(r),
+		originShieldingService: originshielding.NewService(r),
+		sslCertsService:        sslcerts.NewService(r),
 	}
 }
 
@@ -47,6 +50,10 @@ func (s *Service) OriginGroups() origingroups.OriginGroupService {
 	return s.originGroupsService
 }
 
+func (s *Service) OriginShielding() originshielding.OriginShieldingService {
+	return s.originShieldingService
+}
+
 func (s *Service) SSLCerts() sslcerts.SSLCertService {
 	return s.sslCertsService
 }

originshielding/originshielding.go

@@ -0,0 +1,26 @@
+package originshielding
+
+import (
+	"context"
+)
+
+type OriginShieldingService interface {
+	Update(ctx context.Context, id int64, req *UpdateRequest) (*OriginShieldingData, error)
+	Get(ctx context.Context, id int64) (*OriginShieldingData, error)
+	GetLocations(ctx context.Context) (*OriginShieldingLocations, error)
+}
+
+type OriginShieldingData struct {
+	ShieldingPop int `json:"shielding_pop"`
+}
+
+type UpdateRequest struct {
+	ShieldingPop int `json:"shielding_pop"`
+}
+
+type OriginShieldingLocations struct {
+	ID         int    `json:"id"`
+	Datacenter string `json:"datacenter"`
+	Country    string `json:"country"`
+	City       string `json:"city"`
+}

originshielding/service.go

@@ -0,0 +1,45 @@
+package originshielding
+
+import (
+	"context"
+	"fmt"
+	"github.com/G-Core/gcorelabscdn-go/gcore"
+	"net/http"
+)
+
+var _ OriginShieldingService = (*Service)(nil)
+
+type Service struct {
+	r gcore.Requester
+}
+
+func NewService(r gcore.Requester) *Service {
+	return &Service{r: r}
+}
+
+func (s *Service) Update(ctx context.Context, id int64, req *UpdateRequest) (*OriginShieldingData, error) {
+	var origin_shielding OriginShieldingData
+	if err := s.r.Request(ctx, http.MethodPut, fmt.Sprintf("/cdn/resources/%d/shielding_v2", id), req, &origin_shielding); err != nil {
+		return nil, fmt.Errorf("request: %w", err)
+	}
+	return &origin_shielding, nil
+}
+
+func (s *Service) Get(ctx context.Context, id int64) (*OriginShieldingData, error) {
+	var origin_shielding OriginShieldingData
+	if err := s.r.Request(ctx, http.MethodGet, fmt.Sprintf("/cdn/resources/%d/shielding_v2", id), nil, &origin_shielding); err != nil {
+		return nil, fmt.Errorf("request: %w", err)
+	}
+
+	return &origin_shielding, nil
+
+}
+
+func (s *Service) GetLocations(ctx context.Context) (*OriginShieldingLocations, error) {
+	var origin_shielding_locations OriginShieldingLocations
+	if err := s.r.Request(ctx, http.MethodGet, fmt.Sprintf("/cdn/shieldingpop_v2"), nil, &origin_shielding_locations); err != nil {
+		return nil, fmt.Errorf("request: %w", err)
+	}
+
+	return &origin_shielding_locations, nil
+}