Implement repository functions

Signed-off-by: bakito <[email protected]>

Author: bakito

finding_example_test.go

@@ -3,9 +3,8 @@ package dtrack_test
 import (
 	"context"
 
-	"github.com/google/uuid"
-
 	"github.com/DependencyTrack/client-go"
+	"github.com/google/uuid"
 )
 
 // This example demonstrates how to fetch all findings for a given project.

permission.go

@@ -3,8 +3,9 @@ package dtrack
 import (
 	"context"
 	"fmt"
-	"github.com/google/uuid"
 	"net/http"
+
+	"github.com/google/uuid"
 )
 
 type PermissionService struct {

project_property.go

@@ -3,8 +3,9 @@ package dtrack
 import (
 	"context"
 	"fmt"
-	"github.com/google/uuid"
 	"net/http"
+
+	"github.com/google/uuid"
 )
 
 type ProjectProperty struct {

repository.go

@@ -2,9 +2,40 @@ package dtrack
 
 import (
 	"context"
+	"fmt"
 	"net/http"
+
+	"github.com/google/uuid"
+)
+
+const (
+	RepositoryTypeCargo       = "CARGO"
+	RepositoryTypeComposer    = "COMPOSER"
+	RepositoryTypeCpan        = "CPAN"
+	RepositoryTypeGem         = "GEM"
+	RepositoryTypeGoModules   = "GO_MODULES"
+	RepositoryTypeHex         = "HEX"
+	RepositoryTypeMaven       = "MAVEN"
+	RepositoryTypeNpm         = "NPM"
+	RepositoryTypeNuget       = "NUGET"
+	RepositoryTypePypi        = "PYPI"
+	RepositoryTypeUnsupported = "UNSUPPORTED"
 )
 
+type RepositoryType string
+
+type Repository struct {
+	Type            RepositoryType `json:"type"`
+	Identifier      string         `json:"identifier"`
+	Url             string         `json:"url"`
+	ResolutionOrder int            `json:"resolutionOrder"`
+	Enabled         bool           `json:"enabled"`
+	Internal        bool           `json:"internal"`
+	Username        string         `json:"username,omitempty"`
+	Password        string         `json:"password,omitempty"`
+	UUID            uuid.UUID      `json:"uuid,omitempty"`
+}
+
 type RepositoryMetaComponent struct {
 	LatestVersion string `json:"latestVersion"`
 }
@@ -26,3 +57,62 @@ func (rs RepositoryService) GetMetaComponent(ctx context.Context, purl string) (
 	_, err = rs.client.doRequest(req, &r)
 	return
 }
+
+func (rs RepositoryService) GetAll(ctx context.Context, po PageOptions) (p Page[Repository], err error) {
+	req, err := rs.client.newRequest(ctx, http.MethodGet, "/api/v1/repository", withPageOptions(po))
+	if err != nil {
+		return
+	}
+
+	res, err := rs.client.doRequest(req, &p.Items)
+	if err != nil {
+		return
+	}
+
+	p.TotalCount = res.TotalCount
+	return
+}
+
+func (rs RepositoryService) GetByType(ctx context.Context, repoType RepositoryType, po PageOptions) (p Page[Repository], err error) {
+	req, err := rs.client.newRequest(ctx, http.MethodGet, fmt.Sprintf("/api/v1/repository/%s", repoType), withPageOptions(po))
+	if err != nil {
+		return
+	}
+
+	res, err := rs.client.doRequest(req, &p.Items)
+	if err != nil {
+		return
+	}
+
+	p.TotalCount = res.TotalCount
+	return
+}
+
+func (rs RepositoryService) Create(ctx context.Context, repository Repository) (r Repository, err error) {
+	req, err := rs.client.newRequest(ctx, http.MethodPut, "/api/v1/repository", withBody(repository))
+	if err != nil {
+		return
+	}
+
+	_, err = rs.client.doRequest(req, &r)
+	return
+}
+func (rs RepositoryService) Update(ctx context.Context, repository Repository) (r Repository, err error) {
+	req, err := rs.client.newRequest(ctx, http.MethodPost, "/api/v1/repository", withBody(repository))
+	if err != nil {
+		return
+	}
+
+	_, err = rs.client.doRequest(req, &r)
+	return
+}
+
+func (rs RepositoryService) Delete(ctx context.Context, id string) (err error) {
+	req, err := rs.client.newRequest(ctx, http.MethodDelete, fmt.Sprintf("/api/v1/repository/%s", id))
+	if err != nil {
+		return
+	}
+
+	_, err = rs.client.doRequest(req, nil)
+	return
+}

vulnerability.go

@@ -3,9 +3,10 @@ package dtrack
 import (
 	"context"
 	"fmt"
-	"github.com/google/uuid"
 	"net/http"
 	"strconv"
+
+	"github.com/google/uuid"
 )
 
 type Vulnerability struct {