add func to pass ctx to Main

Author: ibuildthecloud

builder.go

@@ -73,8 +73,7 @@ func Name(obj any) string {
 	return commandName
 }
 
-func Main(cmd *cobra.Command) {
-	ctx := SetupSignalContext()
+func MainCtx(ctx context.Context, cmd *cobra.Command) {
 	if err := cmd.ExecuteContext(ctx); err != nil {
 		if strings.EqualFold("interrupt", err.Error()) || errors.Is(err, context.Canceled) {
 			os.Exit(1)
@@ -83,6 +82,10 @@ func Main(cmd *cobra.Command) {
 	}
 }
 
+func Main(cmd *cobra.Command) {
+	MainCtx(SetupSignalContext(), cmd)
+}
+
 // Command populates a cobra.Command object by extracting args from struct tags of the
 // Runnable obj passed. The Run method is assigned to the RunE of the command.
 // children should be either Runnable or *cobra.Command