feat: add --existing-version flag to attestation init (#2541)

Signed-off-by: Miguel Martinez <[email protected]>

Author: migmartri

app/cli/cmd/attestation_init.go

@@ -1,5 +1,5 @@
 //
-// Copyright 2024 The Chainloop Authors.
+// Copyright 2024-2025 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.
@@ -33,6 +33,7 @@ func newAttestationInitCmd() *cobra.Command {
 		projectName           string
 		projectVersion        string
 		projectVersionRelease bool
+		existingVersion       bool
 		newWorkflowcontract   string
 	)
 
@@ -67,6 +68,10 @@ func newAttestationInitCmd() *cobra.Command {
 				return errors.New("project version is required when using --release")
 			}
 
+			if existingVersion && projectVersion == "" {
+				return errors.New("project version is required when using --existing-version")
+			}
+
 			return nil
 		},
 		RunE: func(cmd *cobra.Command, _ []string) error {
@@ -94,6 +99,7 @@ func newAttestationInitCmd() *cobra.Command {
 						WorkflowName:                 workflowName,
 						NewWorkflowContractRef:       newWorkflowcontract,
 						ProjectVersionMarkAsReleased: projectVersionRelease,
+						RequireExistingVersion:       existingVersion,
 					})
 
 					return err
@@ -152,6 +158,7 @@ func newAttestationInitCmd() *cobra.Command {
 
 	cmd.Flags().StringVar(&projectVersion, "version", "", "project version, i.e 0.1.0")
 	cmd.Flags().BoolVar(&projectVersionRelease, "release", false, "promote the provided version as a release")
+	cmd.Flags().BoolVar(&existingVersion, "existing-version", false, "return an error if the version doesn't exist in the project")
 
 	return cmd
 }

app/cli/documentation/cli-reference.mdx

@@ -277,6 +277,7 @@ Options
 --contract string         name of an existing contract or the path/URL to a contract file, to attach it to the auto-created workflow (it doesn't update an existing one)
 --contract-revision int   revision of the contract to retrieve, "latest" by default
 --dry-run                 do not record attestation in the control plane, useful for development
+--existing-version        return an error if the version doesn't exist in the project
 -h, --help                    help for init
 --project string          name of the project of this workflow
 --release                 promote the provided version as a release

app/cli/pkg/action/attestation_init.go

@@ -81,6 +81,7 @@ type AttestationInitRunOpts struct {
 	ProjectName                  string
 	ProjectVersion               string
 	ProjectVersionMarkAsReleased bool
+	RequireExistingVersion       bool
 	WorkflowName                 string
 	NewWorkflowContractRef       string
 }
@@ -190,9 +191,10 @@ func (action *AttestationInit) Run(ctx context.Context, opts *AttestationInitRun
 				JobUrl:           discoveredRunner.RunURI(),
 				ContractRevision: int32(opts.ContractRevision),
 				// send the workflow name explicitly provided by the user to detect that functional case
-				WorkflowName:   opts.WorkflowName,
-				ProjectName:    opts.ProjectName,
-				ProjectVersion: opts.ProjectVersion,
+				WorkflowName:           opts.WorkflowName,
+				ProjectName:            opts.ProjectName,
+				ProjectVersion:         opts.ProjectVersion,
+				RequireExistingVersion: opts.RequireExistingVersion,
 			},
 		)
 		if err != nil {

app/controlplane/api/controlplane/v1/workflow_run.pb.go

@@ -122,6 +122,8 @@ message AttestationServiceInitRequest {
 
   // Optional project version
   string project_version = 6;
+  // Optional flag to require that the project version already exists
+  bool require_existing_version = 7;
 }
 
 message AttestationServiceInitResponse {

app/controlplane/api/controlplane/v1/workflow_run.proto

@@ -186,12 +186,13 @@ func (s *AttestationService) Init(ctx context.Context, req *cpAPI.AttestationSer
 
 	// Create workflowRun
 	opts := &biz.WorkflowRunCreateOpts{
-		WorkflowID:       wf.ID.String(),
-		ContractRevision: contractVersion,
-		RunnerRunURL:     req.GetJobUrl(),
-		RunnerType:       req.GetRunner().String(),
-		CASBackendID:     backend.ID,
-		ProjectVersion:   req.GetProjectVersion(),
+		WorkflowID:             wf.ID.String(),
+		ContractRevision:       contractVersion,
+		RunnerRunURL:           req.GetJobUrl(),
+		RunnerType:             req.GetRunner().String(),
+		CASBackendID:           backend.ID,
+		ProjectVersion:         req.GetProjectVersion(),
+		RequireExistingVersion: req.GetRequireExistingVersion(),
 	}
 
 	run, err := s.wrUseCase.Create(ctx, opts)

app/controlplane/api/gen/frontend/controlplane/v1/workflow_run.ts

@@ -190,12 +190,13 @@ func (uc *WorkflowRunExpirerUseCase) ExpirationSweep(ctx context.Context, olderT
 }
 
 type WorkflowRunCreateOpts struct {
-	WorkflowID       string
-	ContractRevision *WorkflowContractWithVersion
-	RunnerRunURL     string
-	RunnerType       string
-	CASBackendID     uuid.UUID
-	ProjectVersion   string
+	WorkflowID             string
+	ContractRevision       *WorkflowContractWithVersion
+	RunnerRunURL           string
+	RunnerType             string
+	CASBackendID           uuid.UUID
+	ProjectVersion         string
+	RequireExistingVersion bool
 }
 
 type WorkflowRunRepoCreateOpts struct {
@@ -204,6 +205,7 @@ type WorkflowRunRepoCreateOpts struct {
 	Backends                     []uuid.UUID
 	LatestRevision, UsedRevision int
 	ProjectVersion               string
+	RequireExistingVersion       bool
 }
 
 // Create will add a new WorkflowRun, associate it to a schemaVersion and increment the counter in the associated workflow
@@ -234,14 +236,15 @@ func (uc *WorkflowRunUseCase) Create(ctx context.Context, opts *WorkflowRunCreat
 	backends := []uuid.UUID{opts.CASBackendID}
 	result, err := uc.wfRunRepo.Create(ctx,
 		&WorkflowRunRepoCreateOpts{
-			WorkflowID:      workflowUUID,
-			SchemaVersionID: contractRevision.Version.ID,
-			RunURL:          opts.RunnerRunURL,
-			RunnerType:      opts.RunnerType,
-			Backends:        backends,
-			LatestRevision:  contractRevision.Contract.LatestRevision,
-			UsedRevision:    contractRevision.Version.Revision,
-			ProjectVersion:  opts.ProjectVersion,
+			WorkflowID:             workflowUUID,
+			SchemaVersionID:        contractRevision.Version.ID,
+			RunURL:                 opts.RunnerRunURL,
+			RunnerType:             opts.RunnerType,
+			Backends:               backends,
+			LatestRevision:         contractRevision.Contract.LatestRevision,
+			UsedRevision:           contractRevision.Version.Revision,
+			ProjectVersion:         opts.ProjectVersion,
+			RequireExistingVersion: opts.RequireExistingVersion,
 		})
 	if err != nil {
 		return nil, err