feat: add create/delete/update methods to PolicyService (#13)

* Add policy schema Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com> * Add Create to PolicyService Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com> * Add Delete to PolicyService Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com> * Add Update to PolicyService Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com> * Add AddProject/DeleteProject to PolicyService Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com> * Add AddTag/DeleteTag to PolicyService Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com> * Add PolicyOperator, PolicyViolationState Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com> --------- Signed-off-by: 久米拓馬 <takuma.kume@pepabo.com>
DependencyTrack · Aug 29, 2023 · 9894000 · 9894000
1 parent f3504fe
commit 9894000
Showing 1 changed file with 94 additions and 5 deletions.
diff --git a/policy.go b/policy.go
@@ -9,17 +9,36 @@ import (
 )
 
 type Policy struct {
-	UUID             uuid.UUID
-	Name             string            `json:"name"`
-	Operator         string            `json:"operator"`
-	ViolationState   string            `json:"violationState"`
-	PolicyConditions []PolicyCondition `json:"policyConditions"`
+	UUID             uuid.UUID            `json:"uuid,omitempty"`
+	Name             string               `json:"name"`
+	Operator         PolicyOperator       `json:"operator"`
+	ViolationState   PolicyViolationState `json:"violationState"`
+	PolicyConditions []PolicyCondition    `json:"policyConditions,omitempty"`
+	IncludeChildren  bool                 `json:"includeChildren,omitempty"`
+	Global           bool                 `json:"global,omitempty"`
+	Projects         []Project            `json:"projects,omitempty"`
+	Tags             []Tag                `json:"tags,omitempty"`
 }
 
 type PolicyService struct {
 	client *Client
 }
 
+type PolicyOperator string
+
+const (
+	PolicyOperatorAll PolicyOperator = "ALL"
+	PolicyOperatorAny PolicyOperator = "ANY"
+)
+
+type PolicyViolationState string
+
+const (
+	PolicyViolationStateInfo PolicyViolationState = "INFO"
+	PolicyViolationStateWarn PolicyViolationState = "WARN"
+	PolicyViolationStateFail PolicyViolationState = "FAIL"
+)
+
 func (ps PolicyService) Get(ctx context.Context, policyUUID uuid.UUID) (p Policy, err error) {
 	req, err := ps.client.newRequest(ctx, http.MethodGet, fmt.Sprintf("/api/v1/policy/%s", policyUUID))
 	if err != nil {
@@ -44,3 +63,73 @@ func (ps PolicyService) GetAll(ctx context.Context, po PageOptions) (p Page[Poli
 	p.TotalCount = res.TotalCount
 	return
 }
+
+func (ps PolicyService) Create(ctx context.Context, policy Policy) (p Policy, err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodPut, "/api/v1/policy", withBody(policy))
+	if err != nil {
+		return
+	}
+
+	_, err = ps.client.doRequest(req, &p)
+	return
+}
+
+func (ps PolicyService) Delete(ctx context.Context, policyUUID uuid.UUID) (err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodDelete, fmt.Sprintf("/api/v1/policy/%s", policyUUID))
+	if err != nil {
+		return
+	}
+
+	_, err = ps.client.doRequest(req, nil)
+	return
+}
+
+func (ps PolicyService) Update(ctx context.Context, policy Policy) (p Policy, err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodPost, "/api/v1/policy", withBody(policy))
+	if err != nil {
+		return
+	}
+
+	_, err = ps.client.doRequest(req, &p)
+	return
+}
+
+func (ps PolicyService) AddProject(ctx context.Context, policyUUID, projectUUID uuid.UUID) (p Policy, err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodPost, fmt.Sprintf("/api/v1/policy/%s/project/%s", policyUUID, projectUUID))
+	if err != nil {
+		return
+	}
+
+	_, err = ps.client.doRequest(req, &p)
+	return
+}
+
+func (ps PolicyService) DeleteProject(ctx context.Context, policyUUID, projectUUID uuid.UUID) (p Policy, err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodDelete, fmt.Sprintf("/api/v1/policy/%s/project/%s", policyUUID, projectUUID))
+	if err != nil {
+		return
+	}
+
+	_, err = ps.client.doRequest(req, &p)
+	return
+}
+
+func (ps PolicyService) AddTag(ctx context.Context, policyUUID uuid.UUID, tagName string) (p Policy, err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodPost, fmt.Sprintf("/api/v1/policy/%s/tag/%s", policyUUID, tagName))
+	if err != nil {
+		return
+	}
+
+	_, err = ps.client.doRequest(req, &p)
+	return
+}
+
+func (ps PolicyService) DeleteTag(ctx context.Context, policyUUID uuid.UUID, tagName string) (p Policy, err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodDelete, fmt.Sprintf("/api/v1/policy/%s/tag/%s", policyUUID, tagName))
+	if err != nil {
+		return
+	}
+
+	_, err = ps.client.doRequest(req, &p)
+	return
+}