refactor(cli): Use attestation state to pass workflow-name (#766)

Signed-off-by: Javier Rodriguez <[email protected]>

Author: javirln

app/cli/cmd/attestation_init.go

@@ -18,17 +18,20 @@ package cmd
 import (
 	"errors"
 	"fmt"
-
-	"github.com/spf13/cobra"
+	"os"
 
 	"github.com/chainloop-dev/chainloop/app/cli/internal/action"
+	"github.com/spf13/cobra"
 )
 
+const workflowNameEnvVarName = "CHAINLOOP_WORKFLOW_NAME"
+
 func newAttestationInitCmd() *cobra.Command {
 	var (
 		force             bool
 		contractRevision  int
 		attestationDryRun bool
+		workflowName      string
 	)
 
 	cmd := &cobra.Command{
@@ -50,7 +53,7 @@ func newAttestationInitCmd() *cobra.Command {
 			}
 
 			// Initialize it
-			attestationID, err := a.Run(cmd.Context(), contractRevision)
+			attestationID, err := a.Run(cmd.Context(), contractRevision, workflowName)
 			if err != nil {
 				if errors.Is(err, action.ErrAttestationAlreadyExist) {
 					return err
@@ -82,6 +85,10 @@ func newAttestationInitCmd() *cobra.Command {
 	cmd.Flags().BoolVarP(&force, "replace", "f", false, "replace any existing in-progress attestation")
 	cmd.Flags().BoolVar(&attestationDryRun, "dry-run", false, "do not record attestation in the control plane, useful for development")
 	cmd.Flags().IntVar(&contractRevision, "contract-revision", 0, "revision of the contract to retrieve, \"latest\" by default")
+	cmd.Flags().StringVar(&workflowName, "workflow-name", "", "name of the workflow to run the attestation. This is ignored when authentication is based on Robot Account")
+	if workflowName == "" {
+		workflowName = os.Getenv(workflowNameEnvVarName)
+	}
 
 	return cmd
 }

app/cli/internal/action/attestation_init.go

@@ -67,7 +67,7 @@ func NewAttestationInit(cfg *AttestationInitOpts) (*AttestationInit, error) {
 }
 
 // returns the attestation ID
-func (action *AttestationInit) Run(ctx context.Context, contractRevision int) (string, error) {
+func (action *AttestationInit) Run(ctx context.Context, contractRevision int, workflowName string) (string, error) {
 	if action.dryRun && action.UseAttestationRemoteState {
 		return "", errors.New("remote state is not compatible with dry-run mode")
 	}
@@ -83,7 +83,10 @@ func (action *AttestationInit) Run(ctx context.Context, contractRevision int) (s
 	action.Logger.Debug().Msg("Retrieving attestation definition")
 	client := pb.NewAttestationServiceClient(action.ActionsOpts.CPConnection)
 	// get information of the workflow
-	contractResp, err := client.GetContract(ctx, &pb.AttestationServiceGetContractRequest{ContractRevision: int32(contractRevision)})
+	contractResp, err := client.GetContract(ctx, &pb.AttestationServiceGetContractRequest{
+		ContractRevision: int32(contractRevision),
+		WorkflowName:     workflowName,
+	})
 	if err != nil {
 		return "", err
 	}