feat: add contract selection on workflow update

app/cli/cmd/workflow_update.go

@@ -23,7 +23,7 @@ import (
 )
 
 func newWorkflowUpdateCmd() *cobra.Command {
-	var workflowID, name, description, project, team string
+	var workflowID, name, description, project, team, contractID string
 	var public bool
 
 	cmd := &cobra.Command{
@@ -48,6 +48,10 @@ func newWorkflowUpdateCmd() *cobra.Command {
 				opts.Description = &description
 			}
 
+			if cmd.Flags().Changed("contract") {
+				opts.ContractID = &contractID
+			}
+
 			res, err := action.NewWorkflowUpdate(actionOpts).Run(context.Background(), workflowID, opts)
 			if err != nil {
 				return err
@@ -67,6 +71,7 @@ func newWorkflowUpdateCmd() *cobra.Command {
 	cmd.Flags().StringVar(&team, "team", "", "team name")
 	cmd.Flags().StringVar(&project, "project", "", "project name")
 	cmd.Flags().BoolVar(&public, "public", false, "is the workflow public")
+	cmd.Flags().StringVar(&contractID, "contract", "", "the ID of an existing contract")
 
 	return cmd
 }

app/cli/internal/action/workflow_update.go

@@ -30,8 +30,8 @@ func NewWorkflowUpdate(cfg *ActionsOpts) *WorkflowUpdate {
 }
 
 type NewWorkflowUpdateOpts struct {
-	Name, Description, Project, Team *string
-	Public                           *bool
+	Name, Description, Project, Team, ContractID *string
+	Public                                       *bool
 }
 
 func (action *WorkflowUpdate) Run(ctx context.Context, id string, opts *NewWorkflowUpdateOpts) (*WorkflowItem, error) {
@@ -40,7 +40,8 @@ func (action *WorkflowUpdate) Run(ctx context.Context, id string, opts *NewWorkf
 		Id:   id,
 		Name: opts.Name, Description: opts.Description,
 		Project: opts.Project, Team: opts.Team,
-		Public: opts.Public,
+		Public:   opts.Public,
+		SchemaId: opts.ContractID,
 	})
 
 	if err != nil {

app/controlplane/api/controlplane/v1/workflow.proto

@@ -49,6 +49,7 @@ message WorkflowServiceUpdateRequest {
   optional string team = 4;
   optional bool public = 5;
   optional string description = 6;
+  optional string schema_id = 7;
 }
 
 message WorkflowServiceUpdateResponse {

app/controlplane/internal/biz/workflow.go

@@ -58,8 +58,8 @@ type WorkflowCreateOpts struct {
 }
 
 type WorkflowUpdateOpts struct {
-	Name, Project, Team, Description *string
-	Public                           *bool
+	Name, Project, Team, Description, ContractID *string
+	Public                                       *bool
 }
 
 type WorkflowUseCase struct {
@@ -144,6 +144,15 @@ func (uc *WorkflowUseCase) Update(ctx context.Context, orgID, workflowID string,
 		return nil, NewErrNotFound("workflow in organization")
 	}
 
+	// Double check that the contract exists
+	if opts.ContractID != nil {
+		if c, err := uc.contractUC.FindByIDInOrg(ctx, orgID, *opts.ContractID); err != nil {
+			return nil, err
+		} else if c == nil {
+			return nil, NewErrNotFound("contract")
+		}
+	}
+
 	wf, err := uc.wfRepo.Update(ctx, workflowUUID, opts)
 	if err != nil {
 		if errors.Is(err, ErrAlreadyExists) {

app/controlplane/internal/biz/workflow_integration_test.go

@@ -157,6 +157,12 @@ func (s *workflowIntegrationTestSuite) TestUpdate() {
 	workflow, err := s.Workflow.Create(ctx, &biz.WorkflowCreateOpts{Name: name, OrgID: s.org.ID})
 	require.NoError(s.T(), err)
 
+	// Create two contracts in two different orgs
+	contract1, err := s.WorkflowContract.Create(ctx, &biz.WorkflowContractCreateOpts{Name: "contract-1", OrgID: s.org.ID})
+	require.NoError(s.T(), err)
+	contract2, err := s.WorkflowContract.Create(ctx, &biz.WorkflowContractCreateOpts{Name: "contract-2", OrgID: org2.ID})
+	require.NoError(s.T(), err)
+
 	s.Run("by default the workflow is private", func() {
 		s.False(workflow.Public)
 	})
@@ -238,6 +244,16 @@ func (s *workflowIntegrationTestSuite) TestUpdate() {
 			updates: &biz.WorkflowUpdateOpts{Name: toPtrS("")},
 			wantErr: true,
 		},
+		{
+			name:    "can update contract",
+			updates: &biz.WorkflowUpdateOpts{ContractID: toPtrS(contract1.ID.String())},
+			want:    &biz.Workflow{Description: description, Team: team, Project: project, ContractID: contract2.ID},
+		},
+		{
+			name:    "can not update contract in another org",
+			updates: &biz.WorkflowUpdateOpts{ContractID: toPtrS(contract2.ID.String())},
+			wantErr: true,
+		},
 		{
 			name:    "but other opts can",
 			updates: &biz.WorkflowUpdateOpts{Team: toPtrS(""), Project: toPtrS(""), Description: toPtrS("")},

app/controlplane/internal/data/workflow.go

@@ -91,6 +91,15 @@ func (r *WorkflowRepo) Update(ctx context.Context, id uuid.UUID, opts *biz.Workf
 		req = req.SetName(*opts.Name)
 	}
 
+	// Update the contract if provided
+	if opts.ContractID != nil {
+		contractUUID, err := uuid.Parse(*opts.ContractID)
+		if err != nil {
+			return nil, err
+		}
+		req = req.SetContractID(contractUUID)
+	}
+
 	wf, err := req.Save(ctx)
 
 	if err != nil {

app/controlplane/internal/service/workflow.go

@@ -1,5 +1,5 @@
 //
-// Copyright 2023 The Chainloop Authors.
+// Copyright 2024 The Chainloop Authors.
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -79,6 +79,7 @@ func (s *WorkflowService) Update(ctx context.Context, req *pb.WorkflowServiceUpd
 		Team:        req.Team,
 		Public:      req.Public,
 		Description: req.Description,
+		ContractID:  req.SchemaId,
 	}
 
 	p, err := s.useCase.Update(ctx, currentOrg.ID, req.Id, updateOpts)