Add vendor assessment agent

Signed-off-by: Aurélien Sibiril <81782+aureliensibiril@users.noreply.github.com>

pkg/server/api/console/v1/graphql/vendor.graphql

@@ -614,6 +614,13 @@ input CreateVendorRiskAssessmentInput {
 input AssessVendorInput {
     id: ID!
     websiteUrl: String!
+    procedure: String
+}
+
+type VendorSubprocessor {
+    name: String!
+    country: String!
+    purpose: String!
 }
 
 type CreateVendorPayload {
@@ -690,4 +697,6 @@ type CreateVendorRiskAssessmentPayload {
 
 type AssessVendorPayload {
     vendor: Vendor!
+    report: String!
+    subprocessors: [VendorSubprocessor!]!
 }

pkg/server/api/console/v1/types/vendor.go

@@ -18,6 +18,7 @@ import (
 	"go.probo.inc/probo/pkg/coredata"
 	"go.probo.inc/probo/pkg/gid"
 	"go.probo.inc/probo/pkg/page"
+	"go.probo.inc/probo/pkg/probo"
 )
 
 type (
@@ -103,3 +104,15 @@ func NewVendor(v *coredata.Vendor) *Vendor {
 
 	return object
 }
+
+func NewVendorSubprocessors(sps []probo.Subprocessor) []*VendorSubprocessor {
+	result := make([]*VendorSubprocessor, len(sps))
+	for i, sp := range sps {
+		result[i] = &VendorSubprocessor{
+			Name:    sp.Name,
+			Country: sp.Country,
+			Purpose: sp.Purpose,
+		}
+	}
+	return result
+}

pkg/server/api/console/v1/vendor_resolvers.go

@@ -541,20 +541,27 @@ func (r *mutationResolver) AssessVendor(ctx context.Context, input types.AssessV
 
 	prb := r.ProboService(ctx, input.ID.TenantID())
 
-	vendor, err := prb.Vendors.Assess(
+	result, err := prb.Vendors.Assess(
 		ctx,
 		probo.AssessVendorRequest{
 			ID:         input.ID,
 			WebsiteURL: input.WebsiteURL,
+			Procedure:  input.Procedure,
 		},
 	)
 	if err != nil {
+		if errors.Is(err, probo.ErrVendorAssessmentDisabled) {
+			return nil, gqlutils.Unavailable(ctx, probo.ErrVendorAssessmentDisabled)
+		}
+
 		r.logger.ErrorCtx(ctx, "cannot assess vendor", log.Error(err))
 		return nil, gqlutils.Internal(ctx)
 	}
 
 	return &types.AssessVendorPayload{
-		Vendor: types.NewVendor(vendor),
+		Vendor:        types.NewVendor(result.Vendor),
+		Report:        result.Report,
+		Subprocessors: types.NewVendorSubprocessors(result.Subprocessors),
 	}, nil
 }