feat: add cobra cmd to context

decorators.go

@@ -621,6 +621,8 @@ type CommandWithExecute interface {
 // CommandWithExecuteDecorator is a decorator that sets the command execution.
 type CommandWithExecuteDecorator struct{}
 
+type CobraCtxCmd struct{}
+
 // Decorate sets the command execution.
 func (CommandWithExecuteDecorator) Decorate(_ *Ecdysis, cmd *cobra.Command, c Command) error {
 	v, ok := c.(CommandWithExecute)
@@ -636,7 +638,11 @@ func (CommandWithExecuteDecorator) Decorate(_ *Ecdysis, cmd *cobra.Command, c Co
 				return err
 			}
 		}
-		return v.Execute(cmd.Context())
+
+		// Provide the cobra command to the context.
+		// This is useful for situations such as wanting to execute cmd.Help() directly from Execute().
+		ctx := context.WithValue(cmd.Context(), CobraCtxCmd{}, cmd)
+		return v.Execute(ctx)
 	}
 
 	return nil

ecdysis_test.go

@@ -112,6 +112,7 @@ func TestBuildCobraCommand_Behavioral(t *testing.T) {
 	ctx := context.Background()
 	ctrl := gomock.NewController(t)
 	cmd := NewMockBehavioralTestCommand(ctrl)
+	ctx = context.WithValue(ctx, CobraCtxCmd{}, cmd)
 
 	wantLogger := slog.New(slog.NewTextHandler(nil, nil))
 	ecdysis := New(WithDecorators(CommandWithLoggerDecorator{Logger: wantLogger}))
@@ -124,7 +125,7 @@ func TestBuildCobraCommand_Behavioral(t *testing.T) {
 
 	// Set up the remaining expectations before executing the command.
 	call = cmd.EXPECT().Args(gomock.Any()).Return(nil).After(call)
-	cmd.EXPECT().Execute(ctx).Return(nil).After(call)
+	cmd.EXPECT().Execute(gomock.AssignableToTypeOf(ctx)).Return(nil).After(call)
 
 	err := got.ExecuteContext(ctx)
 	if err != nil {