Add behavior to aws byoc client (#627)

* Add behavior to aws byoc client

* Use unspecified as 0 for behavior

* Rename to UNSPECIFIED_BEHAVIOR and create behavior flag type

* Hide unspecifed_behavior from help message

---------

Co-authored-by: Edward J <[email protected]>

Author: edwardrf

src/cmd/cli/command/behavior.go

@@ -0,0 +1,40 @@
+package command
+
+import (
+	"fmt"
+	"strings"
+
+	defangv1 "github.com/DefangLabs/defang/src/protos/io/defang/v1"
+)
+
+type Behavior defangv1.Behavior
+
+func (b Behavior) String() string {
+	return strings.ToLower(defangv1.Behavior_name[int32(b)])
+}
+func (b *Behavior) Set(s string) error {
+	behavior, ok := defangv1.Behavior_value[strings.ToUpper(s)]
+	if !ok {
+		return fmt.Errorf("invalid behavior: %s, valid values are: %v", s, strings.Join(allBehaviors(), ", "))
+	}
+	*b = Behavior(behavior)
+	return nil
+}
+func (b Behavior) Type() string {
+	return "behavior"
+}
+
+func (b Behavior) Value() defangv1.Behavior {
+	return defangv1.Behavior(b)
+}
+
+func allBehaviors() []string {
+	behaviors := make([]string, 0, len(defangv1.Behavior_name)-1)
+	for i, behavior := range defangv1.Behavior_name {
+		if i == 0 {
+			continue
+		}
+		behaviors = append(behaviors, strings.ToLower(behavior))
+	}
+	return behaviors
+}

src/cmd/cli/command/compose.go

@@ -19,6 +19,7 @@ import (
 )
 
 func makeComposeUpCmd() *cobra.Command {
+	behavior := Behavior(defangv1.Behavior_DEVELOPMENT)
 	composeUpCmd := &cobra.Command{
 		Use:         "up",
 		Annotations: authNeededAnnotation,
@@ -29,7 +30,7 @@ func makeComposeUpCmd() *cobra.Command {
 			var detach, _ = cmd.Flags().GetBool("detach")
 
 			since := time.Now()
-			deploy, project, err := cli.ComposeUp(cmd.Context(), client, force)
+			deploy, project, err := cli.ComposeUp(cmd.Context(), client, force, behavior.Value())
 			if err != nil {
 				if !errors.Is(err, types.ErrComposeFileNotFound) {
 					return err
@@ -151,6 +152,7 @@ func makeComposeUpCmd() *cobra.Command {
 	composeUpCmd.Flags().BoolP("detach", "d", false, "run in detached mode")
 	composeUpCmd.Flags().Bool("force", false, "force a build of the image even if nothing has changed")
 	composeUpCmd.Flags().Bool("tail", false, "tail the service logs after updating") // obsolete, but keep for backwards compatibility
+	composeUpCmd.Flags().VarP(&behavior, "behavior", "b", "behavior for the deployment, possible values: "+strings.Join(allBehaviors(), ", "))
 	_ = composeUpCmd.Flags().MarkHidden("tail")
 	return composeUpCmd
 }
@@ -165,7 +167,7 @@ func makeComposeStartCmd() *cobra.Command {
 		RunE: func(cmd *cobra.Command, args []string) error {
 			var force, _ = cmd.Flags().GetBool("force")
 
-			deploy, _, err := cli.ComposeUp(cmd.Context(), client, force)
+			deploy, _, err := cli.ComposeUp(cmd.Context(), client, force, defangv1.Behavior_UNSPECIFIED_BEHAVIOR)
 			if err != nil {
 				return err
 			}
@@ -283,7 +285,7 @@ func makeComposeConfigCmd() *cobra.Command {
 		RunE: func(cmd *cobra.Command, args []string) error {
 			cli.DoDryRun = true // config is like start in a dry run
 			// force=false to calculate the digest
-			if _, _, err := cli.ComposeUp(cmd.Context(), client, false); !errors.Is(err, cli.ErrDryRun) {
+			if _, _, err := cli.ComposeUp(cmd.Context(), client, false, defangv1.Behavior_UNSPECIFIED_BEHAVIOR); !errors.Is(err, cli.ErrDryRun) {
 				return err
 			}
 			return nil

src/pkg/cli/client/byoc/aws/byoc.go

@@ -178,7 +178,7 @@ func (b *ByocAws) Deploy(ctx context.Context, req *defangv1.DeployRequest) (*def
 			return nil, err
 		}
 	}
-	taskArn, err := b.runCdCommand(ctx, "up", payloadString)
+	taskArn, err := b.runCdCommand(ctx, req.Behavior, "up", payloadString)
 	if err != nil {
 		return nil, err
 	}
@@ -326,8 +326,9 @@ func (b *ByocAws) environment() map[string]string {
 	}
 }
 
-func (b *ByocAws) runCdCommand(ctx context.Context, cmd ...string) (ecs.TaskArn, error) {
+func (b *ByocAws) runCdCommand(ctx context.Context, behavior defangv1.Behavior, cmd ...string) (ecs.TaskArn, error) {
 	env := b.environment()
+	env["DEFANG_BEHAVIOR"] = strings.ToLower(behavior.String())
 	if term.DoDebug() {
 		debugEnv := fmt.Sprintf("AWS_REGION=%q", b.driver.Region)
 		if awsProfile := os.Getenv("AWS_PROFILE"); awsProfile != "" {
@@ -346,7 +347,7 @@ func (b *ByocAws) Delete(ctx context.Context, req *defangv1.DeleteRequest) (*def
 		return nil, err
 	}
 	// FIXME: this should only delete the services that are specified in the request, not all
-	taskArn, err := b.runCdCommand(ctx, "up", "")
+	taskArn, err := b.runCdCommand(ctx, defangv1.Behavior_UNSPECIFIED_BEHAVIOR, "up", "")
 	if err != nil {
 		return nil, annotateAwsError(err)
 	}
@@ -655,7 +656,7 @@ func (b *ByocAws) BootstrapCommand(ctx context.Context, command string) (string,
 	if err := b.setUp(ctx); err != nil {
 		return "", err
 	}
-	cdTaskArn, err := b.runCdCommand(ctx, command)
+	cdTaskArn, err := b.runCdCommand(ctx, defangv1.Behavior_UNSPECIFIED_BEHAVIOR, command)
 	if err != nil || cdTaskArn == nil {
 		return "", annotateAwsError(err)
 	}

src/pkg/cli/composeUp.go

@@ -30,7 +30,7 @@ func buildContext(force bool) compose.BuildContext {
 }
 
 // ComposeUp validates a compose project and uploads the services using the client
-func ComposeUp(ctx context.Context, c client.Client, force bool) (*defangv1.DeployResponse, *types.Project, error) {
+func ComposeUp(ctx context.Context, c client.Client, force bool, behavior defangv1.Behavior) (*defangv1.DeployResponse, *types.Project, error) {
 	project, err := c.LoadProject(ctx)
 	if err != nil {
 		return nil, project, err
@@ -61,6 +61,7 @@ func ComposeUp(ctx context.Context, c client.Client, force bool) (*defangv1.Depl
 	}
 
 	resp, err := c.Deploy(ctx, &defangv1.DeployRequest{
+		Behavior: behavior,
 		Project:  project.Name,
 		Services: services,
 	})

src/pkg/cli/composeUp_test.go

@@ -9,6 +9,7 @@ import (
 
 	"github.com/DefangLabs/defang/src/pkg/cli/client"
 	"github.com/DefangLabs/defang/src/pkg/cli/compose"
+	defangv1 "github.com/DefangLabs/defang/src/protos/io/defang/v1"
 )
 
 func TestComposeUp(t *testing.T) {
@@ -26,7 +27,7 @@ func TestComposeUp(t *testing.T) {
 	}))
 	defer server.Close()
 
-	_, project, err := ComposeUp(context.Background(), client.MockClient{UploadUrl: server.URL + "/", Project: proj}, false)
+	_, project, err := ComposeUp(context.Background(), client.MockClient{UploadUrl: server.URL + "/", Project: proj}, false, defangv1.Behavior_DEVELOPMENT)
 	if !errors.Is(err, ErrDryRun) {
 		t.Fatalf("ComposeUp() failed: %v", err)
 	}