chore(cli): make attestation cmds context aware (#497)

migmartri · web-flow · commit 8e82f7f5bda7 · 2024-02-11T18:51:35.000+01:00
Signed-off-by: Miguel Martinez Trivino &lt;miguel@chainloop.dev&gt;
diff --git a/app/cli/cmd/attestation_add.go b/app/cli/cmd/attestation_add.go
@@ -55,7 +55,7 @@ func newAttestationAddCmd() *cobra.Command {
 				return err
 			}
 
-			if err := a.Run("", name, value, annotations); err != nil {
+			if err := a.Run(cmd.Context(), "", name, value, annotations); err != nil {
 				if errors.Is(err, action.ErrAttestationNotInitialized) {
 					return err
 				}
diff --git a/app/cli/cmd/attestation_init.go b/app/cli/cmd/attestation_init.go
@@ -48,7 +48,7 @@ func newAttestationInitCmd() *cobra.Command {
 			}
 
 			// Initialize it
-			err = a.Run(contractRevision)
+			err = a.Run(cmd.Context(), contractRevision)
 			if err != nil {
 				if errors.Is(err, action.ErrAttestationAlreadyExist) {
 					return err
@@ -67,7 +67,7 @@ func newAttestationInitCmd() *cobra.Command {
 				return newGracefulError(err)
 			}
 
-			res, err := statusAction.Run("")
+			res, err := statusAction.Run(cmd.Context(), "")
 			if err != nil {
 				return newGracefulError(err)
 			}
diff --git a/app/cli/cmd/attestation_push.go b/app/cli/cmd/attestation_push.go
@@ -73,7 +73,7 @@ func newAttestationPushCmd() *cobra.Command {
 				return err
 			}
 
-			res, err := a.Run("", annotations)
+			res, err := a.Run(cmd.Context(), "", annotations)
 			if err != nil {
 				if errors.Is(err, action.ErrAttestationNotInitialized) {
 					return err
diff --git a/app/cli/cmd/attestation_reset.go b/app/cli/cmd/attestation_reset.go
@@ -46,7 +46,7 @@ func newAttestationResetCmd() *cobra.Command {
 				return fmt.Errorf("failed to load action: %w", err)
 			}
 
-			if err := a.Run("", trigger, reason); err != nil {
+			if err := a.Run(cmd.Context(), "", trigger, reason); err != nil {
 				return newGracefulError(err)
 			}
 
diff --git a/app/cli/cmd/attestation_status.go b/app/cli/cmd/attestation_status.go
@@ -43,7 +43,7 @@ func newAttestationStatusCmd() *cobra.Command {
 				return fmt.Errorf("failed to load action: %w", err)
 			}
 
-			res, err := a.Run("")
+			res, err := a.Run(cmd.Context(), "")
 			if err != nil {
 				return err
 			}
diff --git a/app/cli/internal/action/attestation_add.go b/app/cli/internal/action/attestation_add.go
@@ -57,19 +57,19 @@ func NewAttestationAdd(cfg *AttestationAddOpts) (*AttestationAdd, error) {
 
 var ErrAttestationNotInitialized = errors.New("attestation not yet initialized")
 
-func (action *AttestationAdd) Run(attestationID, materialName, materialValue string, annotations map[string]string) error {
-	if initialized := action.c.AlreadyInitialized(attestationID); !initialized {
+func (action *AttestationAdd) Run(ctx context.Context, attestationID, materialName, materialValue string, annotations map[string]string) error {
+	if initialized := action.c.AlreadyInitialized(ctx, attestationID); !initialized {
 		return ErrAttestationNotInitialized
 	}
 
-	if err := action.c.LoadCraftingState(attestationID); err != nil {
+	if err := action.c.LoadCraftingState(ctx, attestationID); err != nil {
 		action.Logger.Err(err).Msg("loading existing attestation")
 		return err
 	}
 
 	// Get upload creds and CASbackend for the current attestation and set up CAS client
 	client := pb.NewAttestationServiceClient(action.CPConnection)
-	creds, err := client.GetUploadCreds(context.Background(),
+	creds, err := client.GetUploadCreds(ctx,
 		&pb.AttestationServiceGetUploadCredsRequest{
 			WorkflowRunId: action.c.CraftingState.GetAttestation().GetWorkflow().GetWorkflowRunId(),
 		},
@@ -101,7 +101,7 @@ func (action *AttestationAdd) Run(attestationID, materialName, materialValue str
 		casBackend.Uploader = casclient.New(artifactCASConn, casclient.WithLogger(action.Logger))
 	}
 
-	if err := action.c.AddMaterial(attestationID, materialName, materialValue, casBackend, annotations); err != nil {
+	if err := action.c.AddMaterial(ctx, attestationID, materialName, materialValue, casBackend, annotations); err != nil {
 		return fmt.Errorf("adding material: %w", err)
 	}
 
diff --git a/app/cli/internal/action/attestation_init.go b/app/cli/internal/action/attestation_init.go
@@ -61,11 +61,10 @@ func NewAttestationInit(cfg *AttestationInitOpts) (*AttestationInit, error) {
 	}, nil
 }
 
-func (action *AttestationInit) Run(contractRevision int) error {
+func (action *AttestationInit) Run(ctx context.Context, contractRevision int) error {
 	action.Logger.Debug().Msg("Retrieving attestation definition")
 	client := pb.NewAttestationServiceClient(action.ActionsOpts.CPConnection)
 	// get information of the workflow
-	ctx := context.Background()
 	resp, err := client.GetContract(ctx, &pb.AttestationServiceGetContractRequest{ContractRevision: int32(contractRevision)})
 	if err != nil {
 		return err
@@ -123,17 +122,17 @@ func (action *AttestationInit) Run(contractRevision int) error {
 		AttestationID: attestationID,
 	}
 
-	if err := action.c.Init(initOpts); err != nil {
+	if err := action.c.Init(ctx, initOpts); err != nil {
 		return err
 	}
 
 	// Load the env variables both the system populated and the user predefined ones
-	if err := action.c.ResolveEnvVars(attestationID); err != nil {
+	if err := action.c.ResolveEnvVars(ctx, attestationID); err != nil {
 		if action.dryRun {
 			return nil
 		}
 
-		_ = action.c.Reset(attestationID)
+		_ = action.c.Reset(ctx, attestationID)
 		return err
 	}
 
diff --git a/app/cli/internal/action/attestation_push.go b/app/cli/internal/action/attestation_push.go
@@ -60,12 +60,12 @@ func NewAttestationPush(cfg *AttestationPushOpts) (*AttestationPush, error) {
 	}, nil
 }
 
-func (action *AttestationPush) Run(attestationID string, runtimeAnnotations map[string]string) (*AttestationResult, error) {
-	if initialized := action.c.AlreadyInitialized(attestationID); !initialized {
+func (action *AttestationPush) Run(ctx context.Context, attestationID string, runtimeAnnotations map[string]string) (*AttestationResult, error) {
+	if initialized := action.c.AlreadyInitialized(ctx, attestationID); !initialized {
 		return nil, ErrAttestationNotInitialized
 	}
 
-	if err := action.c.LoadCraftingState(attestationID); err != nil {
+	if err := action.c.LoadCraftingState(ctx, attestationID); err != nil {
 		action.Logger.Err(err).Msg("loading existing attestation")
 		return nil, err
 	}
@@ -133,36 +133,36 @@ func (action *AttestationPush) Run(attestationID string, runtimeAnnotations map[
 	if action.c.CraftingState.DryRun {
 		action.Logger.Info().Msg("dry-run completed, push skipped")
 		// We are done, remove the existing att state
-		if err := action.c.Reset(attestationID); err != nil {
+		if err := action.c.Reset(ctx, attestationID); err != nil {
 			return nil, err
 		}
 
 		return attestationResult, nil
 	}
 
-	attestationResult.Digest, err = pushToControlPlane(action.ActionsOpts.CPConnection, envelope, action.c.CraftingState.Attestation.GetWorkflow().GetWorkflowRunId())
+	attestationResult.Digest, err = pushToControlPlane(ctx, action.ActionsOpts.CPConnection, envelope, action.c.CraftingState.Attestation.GetWorkflow().GetWorkflowRunId())
 	if err != nil {
 		return nil, fmt.Errorf("pushing to control plane: %w", err)
 	}
 
 	action.Logger.Info().Msg("push completed")
 
 	// We are done, remove the existing att state
-	if err := action.c.Reset(attestationID); err != nil {
+	if err := action.c.Reset(ctx, attestationID); err != nil {
 		return nil, err
 	}
 
 	return attestationResult, nil
 }
 
-func pushToControlPlane(conn *grpc.ClientConn, envelope *dsse.Envelope, workflowRunID string) (string, error) {
+func pushToControlPlane(ctx context.Context, conn *grpc.ClientConn, envelope *dsse.Envelope, workflowRunID string) (string, error) {
 	encodedAttestation, err := encodeEnvelope(envelope)
 	if err != nil {
 		return "", fmt.Errorf("encoding attestation: %w", err)
 	}
 
 	client := pb.NewAttestationServiceClient(conn)
-	resp, err := client.Store(context.Background(), &pb.AttestationServiceStoreRequest{
+	resp, err := client.Store(ctx, &pb.AttestationServiceStoreRequest{
 		Attestation:   encodedAttestation,
 		WorkflowRunId: workflowRunID,
 	})
diff --git a/app/cli/internal/action/attestation_reset.go b/app/cli/internal/action/attestation_reset.go
@@ -45,12 +45,12 @@ func NewAttestationReset(opts *ActionsOpts) (*AttestationReset, error) {
 	return &AttestationReset{ActionsOpts: opts, c: c}, nil
 }
 
-func (action *AttestationReset) Run(attestationID, trigger, reason string) error {
-	if initialized := action.c.AlreadyInitialized(attestationID); !initialized {
+func (action *AttestationReset) Run(ctx context.Context, attestationID, trigger, reason string) error {
+	if initialized := action.c.AlreadyInitialized(ctx, attestationID); !initialized {
 		return ErrAttestationNotInitialized
 	}
 
-	if err := action.c.LoadCraftingState(attestationID); err != nil {
+	if err := action.c.LoadCraftingState(ctx, attestationID); err != nil {
 		action.Logger.Err(err).Msg("loading existing attestation")
 		return err
 	}
@@ -70,7 +70,7 @@ func (action *AttestationReset) Run(attestationID, trigger, reason string) error
 		}
 	}
 
-	return action.c.Reset(attestationID)
+	return action.c.Reset(ctx, attestationID)
 }
 
 func parseTrigger(in string) pb.AttestationServiceCancelRequest_TriggerType {
diff --git a/app/cli/internal/action/attestation_status.go b/app/cli/internal/action/attestation_status.go
@@ -16,6 +16,7 @@
 package action
 
 import (
+	"context"
 	"fmt"
 	"time"
 
@@ -66,14 +67,14 @@ func NewAttestationStatus(cfg *AttestationStatusOpts) (*AttestationStatus, error
 	return &AttestationStatus{ActionsOpts: cfg.ActionsOpts, c: c}, nil
 }
 
-func (action *AttestationStatus) Run(attestationID string) (*AttestationStatusResult, error) {
+func (action *AttestationStatus) Run(ctx context.Context, attestationID string) (*AttestationStatusResult, error) {
 	c := action.c
 
-	if initialized := c.AlreadyInitialized(attestationID); !initialized {
+	if initialized := c.AlreadyInitialized(ctx, attestationID); !initialized {
 		return nil, ErrAttestationNotInitialized
 	}
 
-	if err := c.LoadCraftingState(attestationID); err != nil {
+	if err := c.LoadCraftingState(ctx, attestationID); err != nil {
 		action.Logger.Err(err).Msg("loading existing attestation")
 		return nil, err
 	}
diff --git a/internal/attestation/crafter/crafter.go b/internal/attestation/crafter/crafter.go
@@ -42,15 +42,15 @@ import (
 // StateManager is an interface for managing the state of the crafting process
 type StateManager interface {
 	// Check if the state is already initialized
-	Initialized(key string) (bool, error)
+	Initialized(ctx context.Context, key string) (bool, error)
 	// Write the state to the manager backend
-	Write(key string, state *api.CraftingState) error
+	Write(ctx context.Context, key string, state *api.CraftingState) error
 	// Read the state from the manager backend
-	Read(key string, state *api.CraftingState) error
+	Read(ctx context.Context, key string, state *api.CraftingState) error
 	// Reset/Delete the state
-	Reset(key string) error
+	Reset(ctx context.Context, key string) error
 	// String returns a string representation of the state manager
-	Info(key string) string
+	Info(ctx context.Context, key string) string
 }
 
 type Crafter struct {
@@ -107,7 +107,7 @@ type InitOpts struct {
 }
 
 // Initialize the crafter with a remote or local schema
-func (c *Crafter) Init(opts *InitOpts) error {
+func (c *Crafter) Init(ctx context.Context, opts *InitOpts) error {
 	if opts.SchemaV1 == nil {
 		return errors.New("schema is nil")
 	} else if opts.WfInfo == nil {
@@ -121,11 +121,11 @@ func (c *Crafter) Init(opts *InitOpts) error {
 		return fmt.Errorf("%w, expected %s", ErrRunnerContextNotFound, runnerType)
 	}
 
-	return c.initCraftingStateFile(opts.AttestationID, opts.SchemaV1, opts.WfInfo, opts.DryRun, runnerType, runnerContext.RunURI())
+	return c.initCraftingStateFile(ctx, opts.AttestationID, opts.SchemaV1, opts.WfInfo, opts.DryRun, runnerType, runnerContext.RunURI())
 }
 
-func (c *Crafter) AlreadyInitialized(stateID string) bool {
-	ok, _ := c.stateManager.Initialized(stateID)
+func (c *Crafter) AlreadyInitialized(ctx context.Context, stateID string) bool {
+	ok, _ := c.stateManager.Initialized(ctx, stateID)
 	return ok
 }
 
@@ -188,6 +188,7 @@ func LoadSchema(pathOrURI string) (*schemaapi.CraftingSchema, error) {
 
 // Initialize the temporary file with the content of the schema
 func (c *Crafter) initCraftingStateFile(
+	ctx context.Context,
 	attestationID string,
 	schema *schemaapi.CraftingSchema,
 	wf *api.WorkflowMetadata,
@@ -201,26 +202,26 @@ func (c *Crafter) initCraftingStateFile(
 		return fmt.Errorf("initializing crafting state: %w", err)
 	}
 
-	if err := c.stateManager.Write(attestationID, state); err != nil {
+	if err := c.stateManager.Write(ctx, attestationID, state); err != nil {
 		return fmt.Errorf("failed to persist crafting state: %w", err)
 	}
 
-	c.logger.Debug().Str("state", c.stateManager.Info(attestationID)).Msg("created state file")
+	c.logger.Debug().Str("state", c.stateManager.Info(ctx, attestationID)).Msg("created state file")
 
-	return c.LoadCraftingState(attestationID)
+	return c.LoadCraftingState(ctx, attestationID)
 }
 
 // Reset removes the current crafting state
-func (c *Crafter) Reset(stateID string) error {
-	return c.stateManager.Reset(stateID)
+func (c *Crafter) Reset(ctx context.Context, stateID string) error {
+	return c.stateManager.Reset(ctx, stateID)
 }
 
-func (c *Crafter) LoadCraftingState(attestationID string) error {
-	c.logger.Debug().Str("state", c.stateManager.Info(attestationID)).Msg("loading state")
+func (c *Crafter) LoadCraftingState(ctx context.Context, attestationID string) error {
+	c.logger.Debug().Str("state", c.stateManager.Info(ctx, attestationID)).Msg("loading state")
 
 	c.CraftingState = &api.CraftingState{}
 
-	if err := c.stateManager.Read(attestationID, c.CraftingState); err != nil {
+	if err := c.stateManager.Read(ctx, attestationID, c.CraftingState); err != nil {
 		return fmt.Errorf("failed to load crafting state: %w", err)
 	}
 
@@ -231,7 +232,7 @@ func (c *Crafter) LoadCraftingState(attestationID string) error {
 	}
 
 	c.Runner = NewRunner(runnerType)
-	c.logger.Debug().Str("state", c.stateManager.Info(attestationID)).Msg("loaded state")
+	c.logger.Debug().Str("state", c.stateManager.Info(ctx, attestationID)).Msg("loaded state")
 
 	return nil
 }
@@ -361,7 +362,7 @@ func initialCraftingState(cwd string, schema *schemaapi.CraftingSchema, wf *api.
 
 // ResolveEnvVars will iterate on the env vars in the allow list and resolve them from the system context
 // strict indicates if it should fail if any env variable can not be found
-func (c *Crafter) ResolveEnvVars(attestationID string) error {
+func (c *Crafter) ResolveEnvVars(ctx context.Context, attestationID string) error {
 	if err := c.requireStateLoaded(); err != nil {
 		return err
 	}
@@ -404,15 +405,15 @@ func (c *Crafter) ResolveEnvVars(attestationID string) error {
 
 	c.CraftingState.Attestation.EnvVars = outputEnvVars
 
-	if err := c.stateManager.Write(attestationID, c.CraftingState); err != nil {
+	if err := c.stateManager.Write(ctx, attestationID, c.CraftingState); err != nil {
 		return fmt.Errorf("failed to persist crafting state: %w", err)
 	}
 
 	return nil
 }
 
 // Inject material to attestation state
-func (c *Crafter) AddMaterial(attestationID, key, value string, casBackend *casclient.CASBackend, runtimeAnnotations map[string]string) error {
+func (c *Crafter) AddMaterial(ctx context.Context, attestationID, key, value string, casBackend *casclient.CASBackend, runtimeAnnotations map[string]string) error {
 	if err := c.requireStateLoaded(); err != nil {
 		return err
 	}
@@ -482,7 +483,7 @@ func (c *Crafter) AddMaterial(attestationID, key, value string, casBackend *casc
 	}
 
 	// 6 - Persist state
-	if err := c.stateManager.Write(attestationID, c.CraftingState); err != nil {
+	if err := c.stateManager.Write(ctx, attestationID, c.CraftingState); err != nil {
 		return fmt.Errorf("failed to persist crafting state: %w", err)
 	}
 
diff --git a/internal/attestation/crafter/crafter_test.go b/internal/attestation/crafter/crafter_test.go
diff --git a/internal/attestation/crafter/statemanager/filesystem/filesystem.go b/internal/attestation/crafter/statemanager/filesystem/filesystem.go
diff --git a/internal/attestation/crafter/statemanager/filesystem/filesystem_test.go b/internal/attestation/crafter/statemanager/filesystem/filesystem_test.go

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,7 @@ func newAttestationAddCmd() *cobra.Command {`
`55`	`55`	`return err`
`56`	`56`	`}`
`57`	`57`
`58`		`- if err := a.Run("", name, value, annotations); err != nil {`
	`58`	`+ if err := a.Run(cmd.Context(), "", name, value, annotations); err != nil {`
`59`	`59`	`if errors.Is(err, action.ErrAttestationNotInitialized) {`
`60`	`60`	`return err`
`61`	`61`	`}`
Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ func newAttestationInitCmd() *cobra.Command {`
`48`	`48`	`}`
`49`	`49`
`50`	`50`	`// Initialize it`
`51`		`- err = a.Run(contractRevision)`
	`51`	`+ err = a.Run(cmd.Context(), contractRevision)`
`52`	`52`	`if err != nil {`
`53`	`53`	`if errors.Is(err, action.ErrAttestationAlreadyExist) {`
`54`	`54`	`return err`
`@@ -67,7 +67,7 @@ func newAttestationInitCmd() *cobra.Command {`
`67`	`67`	`return newGracefulError(err)`
`68`	`68`	`}`
`69`	`69`
`70`		`- res, err := statusAction.Run("")`
	`70`	`+ res, err := statusAction.Run(cmd.Context(), "")`
`71`	`71`	`if err != nil {`
`72`	`72`	`return newGracefulError(err)`
`73`	`73`	`}`
Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ func newAttestationPushCmd() *cobra.Command {`
`73`	`73`	`return err`
`74`	`74`	`}`
`75`	`75`
`76`		`- res, err := a.Run("", annotations)`
	`76`	`+ res, err := a.Run(cmd.Context(), "", annotations)`
`77`	`77`	`if err != nil {`
`78`	`78`	`if errors.Is(err, action.ErrAttestationNotInitialized) {`
`79`	`79`	`return err`
Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,7 @@ func newAttestationResetCmd() *cobra.Command {`
`46`	`46`	`return fmt.Errorf("failed to load action: %w", err)`
`47`	`47`	`}`
`48`	`48`
`49`		`- if err := a.Run("", trigger, reason); err != nil {`
	`49`	`+ if err := a.Run(cmd.Context(), "", trigger, reason); err != nil {`
`50`	`50`	`return newGracefulError(err)`
`51`	`51`	`}`
`52`	`52`
Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ func newAttestationStatusCmd() *cobra.Command {`
`43`	`43`	`return fmt.Errorf("failed to load action: %w", err)`
`44`	`44`	`}`
`45`	`45`
`46`		`- res, err := a.Run("")`
	`46`	`+ res, err := a.Run(cmd.Context(), "")`
`47`	`47`	`if err != nil {`
`48`	`48`	`return err`
`49`	`49`	`}`
Original file line number	Diff line number	Diff line change
`@@ -61,11 +61,10 @@ func NewAttestationInit(cfg AttestationInitOpts) (AttestationInit, error) {`
`61`	`61`	`}, nil`
`62`	`62`	`}`
`63`	`63`
`64`		`-func (action *AttestationInit) Run(contractRevision int) error {`
	`64`	`+func (action *AttestationInit) Run(ctx context.Context, contractRevision int) error {`
`65`	`65`	`action.Logger.Debug().Msg("Retrieving attestation definition")`
`66`	`66`	`client := pb.NewAttestationServiceClient(action.ActionsOpts.CPConnection)`
`67`	`67`	`// get information of the workflow`
`68`		`- ctx := context.Background()`
`69`	`68`	`resp, err := client.GetContract(ctx, &pb.AttestationServiceGetContractRequest{ContractRevision: int32(contractRevision)})`
`70`	`69`	`if err != nil {`
`71`	`70`	`return err`
`@@ -123,17 +122,17 @@ func (action *AttestationInit) Run(contractRevision int) error {`
`123`	`122`	`AttestationID: attestationID,`
`124`	`123`	`}`
`125`	`124`
`126`		`- if err := action.c.Init(initOpts); err != nil {`
	`125`	`+ if err := action.c.Init(ctx, initOpts); err != nil {`
`127`	`126`	`return err`
`128`	`127`	`}`
`129`	`128`
`130`	`129`	`// Load the env variables both the system populated and the user predefined ones`
`131`		`- if err := action.c.ResolveEnvVars(attestationID); err != nil {`
	`130`	`+ if err := action.c.ResolveEnvVars(ctx, attestationID); err != nil {`
`132`	`131`	`if action.dryRun {`
`133`	`132`	`return nil`
`134`	`133`	`}`
`135`	`134`
`136`		`- _ = action.c.Reset(attestationID)`
	`135`	`+ _ = action.c.Reset(ctx, attestationID)`
`137`	`136`	`return err`
`138`	`137`	`}`
`139`	`138`
Original file line number	Diff line number	Diff line change
`@@ -60,12 +60,12 @@ func NewAttestationPush(cfg AttestationPushOpts) (AttestationPush, error) {`
`60`	`60`	`}, nil`
`61`	`61`	`}`
`62`	`62`
`63`		`-func (action AttestationPush) Run(attestationID string, runtimeAnnotations map[string]string) (AttestationResult, error) {`
`64`		`- if initialized := action.c.AlreadyInitialized(attestationID); !initialized {`
	`63`	`+func (action AttestationPush) Run(ctx context.Context, attestationID string, runtimeAnnotations map[string]string) (AttestationResult, error) {`
	`64`	`+ if initialized := action.c.AlreadyInitialized(ctx, attestationID); !initialized {`
`65`	`65`	`return nil, ErrAttestationNotInitialized`
`66`	`66`	`}`
`67`	`67`
`68`		`- if err := action.c.LoadCraftingState(attestationID); err != nil {`
	`68`	`+ if err := action.c.LoadCraftingState(ctx, attestationID); err != nil {`
`69`	`69`	`action.Logger.Err(err).Msg("loading existing attestation")`
`70`	`70`	`return nil, err`
`71`	`71`	`}`
`@@ -133,36 +133,36 @@ func (action *AttestationPush) Run(attestationID string, runtimeAnnotations map[`
`133`	`133`	`if action.c.CraftingState.DryRun {`
`134`	`134`	`action.Logger.Info().Msg("dry-run completed, push skipped")`
`135`	`135`	`// We are done, remove the existing att state`
`136`		`- if err := action.c.Reset(attestationID); err != nil {`
	`136`	`+ if err := action.c.Reset(ctx, attestationID); err != nil {`
`137`	`137`	`return nil, err`
`138`	`138`	`}`
`139`	`139`
`140`	`140`	`return attestationResult, nil`
`141`	`141`	`}`
`142`	`142`
`143`		`- attestationResult.Digest, err = pushToControlPlane(action.ActionsOpts.CPConnection, envelope, action.c.CraftingState.Attestation.GetWorkflow().GetWorkflowRunId())`
	`143`	`+ attestationResult.Digest, err = pushToControlPlane(ctx, action.ActionsOpts.CPConnection, envelope, action.c.CraftingState.Attestation.GetWorkflow().GetWorkflowRunId())`
`144`	`144`	`if err != nil {`
`145`	`145`	`return nil, fmt.Errorf("pushing to control plane: %w", err)`
`146`	`146`	`}`
`147`	`147`
`148`	`148`	`action.Logger.Info().Msg("push completed")`
`149`	`149`
`150`	`150`	`// We are done, remove the existing att state`
`151`		`- if err := action.c.Reset(attestationID); err != nil {`
	`151`	`+ if err := action.c.Reset(ctx, attestationID); err != nil {`
`152`	`152`	`return nil, err`
`153`	`153`	`}`
`154`	`154`
`155`	`155`	`return attestationResult, nil`
`156`	`156`	`}`
`157`	`157`
`158`		`-func pushToControlPlane(conn grpc.ClientConn, envelope dsse.Envelope, workflowRunID string) (string, error) {`
	`158`	`+func pushToControlPlane(ctx context.Context, conn grpc.ClientConn, envelope dsse.Envelope, workflowRunID string) (string, error) {`
`159`	`159`	`encodedAttestation, err := encodeEnvelope(envelope)`
`160`	`160`	`if err != nil {`
`161`	`161`	`return "", fmt.Errorf("encoding attestation: %w", err)`
`162`	`162`	`}`
`163`	`163`
`164`	`164`	`client := pb.NewAttestationServiceClient(conn)`
`165`		`- resp, err := client.Store(context.Background(), &pb.AttestationServiceStoreRequest{`
	`165`	`+ resp, err := client.Store(ctx, &pb.AttestationServiceStoreRequest{`
`166`	`166`	`Attestation: encodedAttestation,`
`167`	`167`	`WorkflowRunId: workflowRunID,`
`168`	`168`	`})`
Original file line number	Diff line number	Diff line change
`@@ -45,12 +45,12 @@ func NewAttestationReset(opts ActionsOpts) (AttestationReset, error) {`
`45`	`45`	`return &AttestationReset{ActionsOpts: opts, c: c}, nil`
`46`	`46`	`}`
`47`	`47`
`48`		`-func (action *AttestationReset) Run(attestationID, trigger, reason string) error {`
`49`		`- if initialized := action.c.AlreadyInitialized(attestationID); !initialized {`
	`48`	`+func (action *AttestationReset) Run(ctx context.Context, attestationID, trigger, reason string) error {`
	`49`	`+ if initialized := action.c.AlreadyInitialized(ctx, attestationID); !initialized {`
`50`	`50`	`return ErrAttestationNotInitialized`
`51`	`51`	`}`
`52`	`52`
`53`		`- if err := action.c.LoadCraftingState(attestationID); err != nil {`
	`53`	`+ if err := action.c.LoadCraftingState(ctx, attestationID); err != nil {`
`54`	`54`	`action.Logger.Err(err).Msg("loading existing attestation")`
`55`	`55`	`return err`
`56`	`56`	`}`
`@@ -70,7 +70,7 @@ func (action *AttestationReset) Run(attestationID, trigger, reason string) error`
`70`	`70`	`}`
`71`	`71`	`}`
`72`	`72`
`73`		`- return action.c.Reset(attestationID)`
	`73`	`+ return action.c.Reset(ctx, attestationID)`
`74`	`74`	`}`
`75`	`75`
`76`	`76`	`func parseTrigger(in string) pb.AttestationServiceCancelRequest_TriggerType {`