SCALRCORE-28980 API > Policy Engine > Enforce policy group for all cu…

policy_group.go

@@ -78,6 +78,7 @@ type PolicyGroup struct {
 	ErrorMessage string              `jsonapi:"attr,error-message"`
 	OpaVersion   string              `jsonapi:"attr,opa-version"`
 	VCSRepo      *PolicyGroupVCSRepo `jsonapi:"attr,vcs-repo"`
+	IsEnforced   bool                `jsonapi:"attr,is-enforced"`
 
 	// Relations
 	Account      *Account       `jsonapi:"relation,account"`
@@ -112,10 +113,12 @@ type PolicyGroupCreateOptions struct {
 	Name       *string                    `jsonapi:"attr,name"`
 	OpaVersion *string                    `jsonapi:"attr,opa-version,omitempty"`
 	VCSRepo    *PolicyGroupVCSRepoOptions `jsonapi:"attr,vcs-repo"`
+	IsEnforced *bool                      `jsonapi:"attr,is-enforced,omitempty"`
 
 	// Relations
-	Account     *Account     `jsonapi:"relation,account"`
-	VcsProvider *VcsProvider `jsonapi:"relation,vcs-provider"`
+	Account      *Account       `jsonapi:"relation,account"`
+	VcsProvider  *VcsProvider   `jsonapi:"relation,vcs-provider"`
+	Environments []*Environment `jsonapi:"relation,environments"`
 }
 
 func (o PolicyGroupCreateOptions) valid() error {
@@ -146,6 +149,7 @@ type PolicyGroupUpdateOptions struct {
 	Name       *string                    `jsonapi:"attr,name,omitempty"`
 	OpaVersion *string                    `jsonapi:"attr,opa-version,omitempty"`
 	VCSRepo    *PolicyGroupVCSRepoOptions `jsonapi:"attr,vcs-repo,omitempty"`
+	IsEnforced *bool                      `jsonapi:"attr,is-enforced,omitempty"`
 
 	// Relations
 	VcsProvider *VcsProvider `jsonapi:"relation,vcs-provider,omitempty"`

policy_group_environments.go

@@ -16,6 +16,7 @@ var _ PolicyGroupEnvironments = (*policyGroupEnvironment)(nil)
 // Scalr API supports.
 type PolicyGroupEnvironments interface {
 	Create(ctx context.Context, options PolicyGroupEnvironmentsCreateOptions) error
+	Update(ctx context.Context, options PolicyGroupEnvironmentsUpdateOptions) error
 	Delete(ctx context.Context, options PolicyGroupEnvironmentDeleteOptions) error
 }
 
@@ -35,6 +36,12 @@ type PolicyGroupEnvironmentsCreateOptions struct {
 	PolicyGroupEnvironments []*PolicyGroupEnvironment
 }
 
+// PolicyGroupEnvironmentsUpdateOptions represents options for updating policy group environment linkage
+type PolicyGroupEnvironmentsUpdateOptions struct {
+	PolicyGroupID           string
+	PolicyGroupEnvironments []*PolicyGroupEnvironment
+}
+
 type PolicyGroupEnvironmentDeleteOptions struct {
 	PolicyGroupID string
 	EnvironmentID string
@@ -50,6 +57,13 @@ func (o PolicyGroupEnvironmentsCreateOptions) valid() error {
 	return nil
 }
 
+func (o PolicyGroupEnvironmentsUpdateOptions) valid() error {
+	if !validStringID(&o.PolicyGroupID) {
+		return errors.New("invalid value for policy group ID")
+	}
+	return nil
+}
+
 func (o PolicyGroupEnvironmentDeleteOptions) valid() error {
 	if !validStringID(&o.PolicyGroupID) {
 		return errors.New("invalid value for policy group ID")
@@ -62,7 +76,7 @@ func (o PolicyGroupEnvironmentDeleteOptions) valid() error {
 	return nil
 }
 
-// Create a new policy group.
+// Create a new policy group environments linkage.
 func (s *policyGroupEnvironment) Create(ctx context.Context, options PolicyGroupEnvironmentsCreateOptions) error {
 	if err := options.valid(); err != nil {
 		return err
@@ -80,7 +94,25 @@ func (s *policyGroupEnvironment) Create(ctx context.Context, options PolicyGroup
 	return s.client.do(ctx, req, nil)
 }
 
-// Delete policy group by its ID.
+// Update policy group environments linkage
+func (s *policyGroupEnvironment) Update(ctx context.Context, options PolicyGroupEnvironmentsUpdateOptions) error {
+	if err := options.valid(); err != nil {
+		return err
+	}
+	u := fmt.Sprintf("policy-groups/%s/relationships/environments", url.QueryEscape(options.PolicyGroupID))
+	payload, err := jsonapi.Marshal(options.PolicyGroupEnvironments)
+	if err != nil {
+		return err
+	}
+	req, err := s.client.newJsonRequest("PATCH", u, payload)
+	if err != nil {
+		return err
+	}
+
+	return s.client.do(ctx, req, nil)
+}
+
+// Delete policy group environments linkage by its ID.
 func (s *policyGroupEnvironment) Delete(ctx context.Context, options PolicyGroupEnvironmentDeleteOptions) error {
 	if err := options.valid(); err != nil {
 		return err