adding feature flag to cli

formancehq · Nov 26, 2024 · 8f1fd3a · 8f1fd3a
1 parent 46c0983
commit 8f1fd3a
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/internal/cmd/run.go b/internal/cmd/run.go
@@ -26,6 +26,8 @@ var runRawOpt string
 var runStdinFlag bool
 var runOutFormatOpt string
 
+var overdraftFeatureFlag bool
+
 type inputOpts struct {
 	Script    string                       `json:"script"`
 	Variables map[string]string            `json:"variables"`
@@ -115,10 +117,15 @@ func run(path string) {
 		os.Exit(1)
 	}
 
+	featureFlags := map[string]struct{}{}
+	if overdraftFeatureFlag {
+		featureFlags[interpreter.ExperimentalOverdraftFunctionFeatureFlag] = struct{}{}
+	}
+
 	result, err := interpreter.RunProgram(context.Background(), parseResult.Value, opt.Variables, interpreter.StaticStore{
 		Balances: opt.Balances,
 		Meta:     opt.Meta,
-	}, nil)
+	}, featureFlags)
 
 	if err != nil {
 		rng := err.GetRange()
@@ -192,6 +199,9 @@ func getRunCmd() *cobra.Command {
 	cmd.Flags().StringVarP(&runRawOpt, "raw", "r", "", "Raw json input containing script, variables, balances, metadata")
 	cmd.Flags().BoolVar(&runStdinFlag, "stdin", false, "Take input from stdin (same format as the --raw option)")
 
+	// Feature flag
+	cmd.Flags().BoolVar(&overdraftFeatureFlag, interpreter.ExperimentalOverdraftFunctionFeatureFlag, false, "feature flag to enable the overdraft() function")
+
 	// Output options
 	cmd.Flags().StringVar(&runOutFormatOpt, "output-format", OutputFormatPretty, "Set the output format. Available options: pretty, json.")