Migrate to mcpgen

Signed-off-by: Bryan Frimin <bryan@getprobo.com>

pkg/server/api/mcp/v1/schema.resolvers.go

@@ -0,0 +1,110 @@
+package mcp_v1
+
+// This file will be automatically regenerated based on the schema, any resolver implementations
+// will be copied through when generating and any unknown code will be moved to the end.
+// Code generated by mcpgen. DO NOT EDIT.
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/modelcontextprotocol/go-sdk/mcp"
+	"go.probo.inc/probo/pkg/coredata"
+	"go.probo.inc/probo/pkg/page"
+	"go.probo.inc/probo/pkg/probo"
+	"go.probo.inc/probo/pkg/server/api/mcp/v1/types"
+)
+
+// ListOrganizationsTool handles the listOrganizations tool
+// List all organizations the user has access to
+func (r *Resolver) ListOrganizationsTool(ctx context.Context, req *mcp.CallToolRequest, input *types.ListOrganizationsInput) (*mcp.CallToolResult, types.ListOrganizationsOutput, error) {
+	mcpCtx := MCPContextFromContext(ctx)
+	organizations, err := r.authzSvc.GetAllUserOrganizations(ctx, mcpCtx.UserID)
+	if err != nil {
+		return nil, types.ListOrganizationsOutput{}, fmt.Errorf("failed to list organizations: %w", err)
+	}
+
+	result := types.ListOrganizationsOutput{
+		Organizations: make([]*types.Organization, 0, len(organizations)),
+	}
+
+	for _, org := range organizations {
+		result.Organizations = append(result.Organizations, types.NewOrganization(org))
+	}
+
+	return nil, result, nil
+}
+
+// ListVendorsTool handles the listVendors tool
+// List all vendors for the organization
+func (r *Resolver) ListVendorsTool(ctx context.Context, req *mcp.CallToolRequest, input *types.ListVendorsInput) (*mcp.CallToolResult, types.ListVendorsOutput, error) {
+	prb := r.ProboService(ctx, input.OrganizationID.TenantID())
+
+	pageOrderBy := page.OrderBy[coredata.VendorOrderField]{
+		Field:     coredata.VendorOrderFieldCreatedAt,
+		Direction: page.OrderDirectionDesc,
+	}
+	if input.OrderBy != nil {
+		pageOrderBy = page.OrderBy[coredata.VendorOrderField]{
+			Field:     input.OrderBy.Field,
+			Direction: input.OrderBy.Direction,
+		}
+	}
+
+	cursor := types.NewCursor(input.Size, input.Cursor, pageOrderBy)
+
+	var vendorFilter = coredata.NewVendorFilter(nil, nil)
+	if input.Filter != nil {
+		vendorFilter = coredata.NewVendorFilter(&input.Filter.SnapshotID, nil)
+	}
+
+	page, err := prb.Vendors.ListForOrganizationID(ctx, input.OrganizationID, cursor, vendorFilter)
+	if err != nil {
+		panic(fmt.Errorf("cannot list organization vendors: %w", err))
+	}
+
+	return nil, types.NewListVendorsOutput(page), nil
+}
+
+// AddVendorTool handles the addVendor tool
+// Add a new vendor to the organization
+func (r *Resolver) AddVendorTool(ctx context.Context, req *mcp.CallToolRequest, input *types.AddVendorInput) (*mcp.CallToolResult, types.AddVendorOutput, error) {
+	svc := r.ProboService(ctx, input.OrganizationID.TenantID())
+
+	vendor, err := svc.Vendors.Create(
+		ctx,
+		probo.CreateVendorRequest{
+			OrganizationID: input.OrganizationID,
+			Name:           input.Name,
+			Description:    input.Description,
+			// HeadquarterAddress:            input.HeadquarterAddress,
+			// LegalName:                     input.LegalName,
+			// WebsiteURL:                    input.WebsiteURL,
+			// Category:                      input.Category,
+			// PrivacyPolicyURL:              input.PrivacyPolicyURL,
+			// ServiceLevelAgreementURL:      input.ServiceLevelAgreementURL,
+			// DataProcessingAgreementURL:    input.DataProcessingAgreementURL,
+			// BusinessAssociateAgreementURL: input.BusinessAssociateAgreementURL,
+			// SubprocessorsListURL:          input.SubprocessorsListURL,
+			// Certifications:                input.Certifications,
+			// Countries:                     input.Countries,
+			// SecurityPageURL:               input.SecurityPageURL,
+			// TrustPageURL:                  input.TrustPageURL,
+			// TermsOfServiceURL:             input.TermsOfServiceURL,
+			// StatusPageURL:                 input.StatusPageURL,
+			// BusinessOwnerID:               input.BusinessOwnerID,
+			// SecurityOwnerID:               input.SecurityOwnerID,
+		},
+	)
+	if err != nil {
+		return nil, types.AddVendorOutput{}, fmt.Errorf("failed to create vendor: %w", err)
+	}
+
+	return nil, types.NewAddVendorOutput(vendor), nil
+}
+
+// UpdateVendorTool handles the updateVendor tool
+// Update an existing vendor
+func (r *Resolver) UpdateVendorTool(ctx context.Context, req *mcp.CallToolRequest, input *types.UpdateVendorInput) (*mcp.CallToolResult, types.UpdateVendorOutput, error) {
+	return nil, types.UpdateVendorOutput{}, fmt.Errorf("updateVendor not implemented")
+}