From 421624d3f30f85216821dd0c9ad5238ea3108d71 Mon Sep 17 00:00:00 2001 From: Maycon Santos Date: Tue, 6 May 2025 22:56:48 +0200 Subject: [PATCH] add account id support for policies methods --- management/client/rest/policies.go | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/management/client/rest/policies.go b/management/client/rest/policies.go index 4ebe73f9f..426a688e2 100644 --- a/management/client/rest/policies.go +++ b/management/client/rest/policies.go @@ -15,8 +15,12 @@ type PoliciesAPI struct { // List list all policies // See more: https://docs.netbird.io/api/resources/policies#list-all-policies -func (a *PoliciesAPI) List(ctx context.Context) ([]api.Policy, error) { - resp, err := a.c.NewRequest(ctx, "GET", "/api/policies", nil) +func (a *PoliciesAPI) List(ctx context.Context, accountID string) ([]api.Policy, error) { + path := "/api/policies" + if accountID != "" { + path += "?account=" + accountID + } + resp, err := a.c.NewRequest(ctx, "GET", path, nil) if err != nil { return nil, err } @@ -55,12 +59,16 @@ func (a *PoliciesAPI) Create(ctx context.Context, request api.PostApiPoliciesJSO // Update update policy info // See more: https://docs.netbird.io/api/resources/policies#update-a-policy -func (a *PoliciesAPI) Update(ctx context.Context, policyID string, request api.PutApiPoliciesPolicyIdJSONRequestBody) (*api.Policy, error) { +func (a *PoliciesAPI) Update(ctx context.Context, policyID string, request api.PutApiPoliciesPolicyIdJSONRequestBody, accountID string) (*api.Policy, error) { + path := "/api/policies/" + policyID + if accountID != "" { + path += "?account=" + accountID + } requestBytes, err := json.Marshal(request) if err != nil { return nil, err } - resp, err := a.c.NewRequest(ctx, "PUT", "/api/policies/"+policyID, bytes.NewReader(requestBytes)) + resp, err := a.c.NewRequest(ctx, "PUT", path, bytes.NewReader(requestBytes)) if err != nil { return nil, err }