feat: add more functions for team management (#3)

* Allow team creation

Signed-off-by: Brian Wilkinson <[email protected]>

* List permissions

Signed-off-by: Brian Wilkinson <[email protected]>

* Add permissions to client

Signed-off-by: Brian Wilkinson <[email protected]>

* Teams use Permissions type

Signed-off-by: Brian Wilkinson <[email protected]>

* Not possible to create team with perms, try adding after

Signed-off-by: Brian Wilkinson <[email protected]>

* Incorrect type returned

Signed-off-by: Brian Wilkinson <[email protected]>

* Team permission has a type

Signed-off-by: Brian Wilkinson <[email protected]>

* Team permission has a type

Signed-off-by: Brian Wilkinson <[email protected]>

* Need to delete team with obj not uuid

Signed-off-by: Brian Wilkinson <[email protected]>

* Path incorrectly specified

Signed-off-by: Brian Wilkinson <[email protected]>

* Revert module name for PR

Signed-off-by: Brian Wilkinson <[email protected]>

---------

Signed-off-by: Brian Wilkinson <[email protected]>

Author: brianwilkinson

client.go

@@ -35,6 +35,7 @@ type Client struct {
 	Finding           FindingService
 	License           LicenseService
 	Metrics           MetricsService
+	Permission        PermissionService
 	Policy            PolicyService
 	PolicyViolation   PolicyViolationService
 	Project           ProjectService
@@ -79,6 +80,7 @@ func NewClient(baseURL string, options ...ClientOption) (*Client, error) {
 	client.Finding = FindingService{client: &client}
 	client.License = LicenseService{client: &client}
 	client.Metrics = MetricsService{client: &client}
+	client.Permission = PermissionService{client: &client}
 	client.Policy = PolicyService{client: &client}
 	client.PolicyViolation = PolicyViolationService{client: &client}
 	client.Project = ProjectService{client: &client}

permission.go

@@ -0,0 +1,42 @@
+package dtrack
+
+import (
+	"context"
+	"fmt"
+	"github.com/google/uuid"
+	"net/http"
+)
+
+type PermissionService struct {
+	client *Client
+}
+
+type Permission struct {
+	Name        string `json:"name"`
+	Description string `json:"description"`
+}
+
+func (ps PermissionService) GetAll(ctx context.Context, po PageOptions) (p Page[Permission], err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodGet, "/api/v1/permission", withPageOptions(po))
+	if err != nil {
+		return
+	}
+
+	res, err := ps.client.doRequest(req, &p.Items)
+	if err != nil {
+		return
+	}
+
+	p.TotalCount = res.TotalCount
+	return
+}
+
+func (ps PermissionService) AddPermissionToTeam(ctx context.Context, permission Permission, team uuid.UUID) (t Team, err error) {
+	req, err := ps.client.newRequest(ctx, http.MethodPost, fmt.Sprintf("/api/v1/permission/%s/team/%s", permission.Name, team.String()))
+	if err != nil {
+		return
+	}
+
+	_, err = ps.client.doRequest(req, &t)
+	return
+}

team.go

@@ -9,9 +9,10 @@ import (
 )
 
 type Team struct {
-	UUID    uuid.UUID `json:"uuid,omitempty"`
-	Name    string    `json:"name,omitempty"`
-	APIKeys []APIKey  `json:"apiKeys,omitempty"`
+	UUID        uuid.UUID    `json:"uuid,omitempty"`
+	Name        string       `json:"name,omitempty"`
+	APIKeys     []APIKey     `json:"apiKeys,omitempty"`
+	Permissions []Permission `json:"permissions"`
 }
 
 type APIKey struct {
@@ -22,13 +23,13 @@ type TeamService struct {
 	client *Client
 }
 
-func (ts TeamService) Get(ctx context.Context, teamUUID uuid.UUID) (p Project, err error) {
+func (ts TeamService) Get(ctx context.Context, teamUUID uuid.UUID) (t Team, err error) {
 	req, err := ts.client.newRequest(ctx, http.MethodGet, fmt.Sprintf("/api/v1/team/%s", teamUUID))
 	if err != nil {
 		return
 	}
 
-	_, err = ts.client.doRequest(req, &p)
+	_, err = ts.client.doRequest(req, &t)
 	return
 }
 
@@ -58,3 +59,23 @@ func (ts TeamService) GenerateAPIKey(ctx context.Context, teamUUID uuid.UUID) (k
 	key = apiKey.Key
 	return
 }
+
+func (ts TeamService) Create(ctx context.Context, team Team) (t Team, err error) {
+	req, err := ts.client.newRequest(ctx, http.MethodPut, "/api/v1/team", withBody(team))
+	if err != nil {
+		return
+	}
+
+	_, err = ts.client.doRequest(req, &t)
+	return
+}
+
+func (ts TeamService) Delete(ctx context.Context, team Team) (err error) {
+	req, err := ts.client.newRequest(ctx, http.MethodDelete, "/api/v1/team", withBody(team))
+	if err != nil {
+		return
+	}
+
+	_, err = ts.client.doRequest(req, nil)
+	return
+}