Create pipectl init for ECS

cmd/pipectl/main.go

@@ -21,6 +21,7 @@ import (
 	"github.com/pipe-cd/pipecd/pkg/app/pipectl/cmd/deployment"
 	"github.com/pipe-cd/pipecd/pkg/app/pipectl/cmd/encrypt"
 	"github.com/pipe-cd/pipecd/pkg/app/pipectl/cmd/event"
+	"github.com/pipe-cd/pipecd/pkg/app/pipectl/cmd/initialize"
 	"github.com/pipe-cd/pipecd/pkg/app/pipectl/cmd/piped"
 	"github.com/pipe-cd/pipecd/pkg/app/pipectl/cmd/planpreview"
 	"github.com/pipe-cd/pipecd/pkg/app/pipectl/cmd/quickstart"
@@ -41,6 +42,7 @@ func main() {
 		piped.NewCommand(),
 		encrypt.NewCommand(),
 		quickstart.NewCommand(),
+		initialize.NewCommand(),
 	)
 
 	if err := app.Run(); err != nil {

pkg/app/pipectl/cmd/initialize/ecs.go

@@ -0,0 +1,99 @@
+// Copyright 2024 The PipeCD Authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package initialize
+
+import (
+	"github.com/pipe-cd/pipecd/pkg/app/pipectl/prompt"
+	"github.com/pipe-cd/pipecd/pkg/config"
+)
+
+// Use genericConfigs in order to simplify using the GenericApplicationSpec and keep the order as we want.
+type genericECSApplicationSpec struct {
+	Name        string                    `json:"name"`
+	Input       config.ECSDeploymentInput `json:"input"`
+	Description string                    `json:"description,omitempty"`
+}
+
+func generateECSConfig(p prompt.Prompt) (*genericConfig, error) {
+	// inputs
+	var (
+		appName        string
+		serviceDefFile string
+		taskDefFile    string
+		targetGroupArn string
+		containerName  string
+		containerPort  int
+	)
+	inputs := []prompt.Input{
+		{
+			Message:       "Name of the application",
+			TargetPointer: &appName,
+			Required:      true,
+		},
+		{
+			Message:       "Name of the service definition file (e.g. serviceDef.yaml)",
+			TargetPointer: &serviceDefFile,
+			Required:      true,
+		},
+		{
+			Message:       "Name of the task definition file (e.g. taskDef.yaml)",
+			TargetPointer: &taskDefFile,
+			Required:      true,
+		},
+		// target group inputs
+		{
+			Message:       "ARN of the target group to the service",
+			TargetPointer: &targetGroupArn,
+			Required:      false,
+		},
+		{
+			Message:       "Name of the container of the target group",
+			TargetPointer: &containerName,
+			Required:      false,
+		},
+		{
+			Message:       "Port number of the container of the target group",
+			TargetPointer: &containerPort,
+			Required:      false,
+		},
+	}
+
+	err := p.RunSlice(inputs)
+	if err != nil {
+		return nil, err
+	}
+
+	spec := &genericECSApplicationSpec{
+		Name: appName,
+		Input: config.ECSDeploymentInput{
+			ServiceDefinitionFile: serviceDefFile,
+			TaskDefinitionFile:    taskDefFile,
+			TargetGroups: config.ECSTargetGroups{
+				Primary: &config.ECSTargetGroup{
+					TargetGroupArn: targetGroupArn,
+					ContainerName:  containerName,
+					ContainerPort:  containerPort,
+				},
+			},
+		},
+		Description: "Generated by `pipectl init`. See https://pipecd.dev/docs/user-guide/configuration-reference/ for more.",
+	}
+
+	return &genericConfig{
+		Kind:            config.KindECSApp,
+		APIVersion:      config.VersionV1Beta1,
+		ApplicationSpec: spec,
+	}, nil
+}

pkg/app/pipectl/cmd/initialize/ecs_test.go

@@ -0,0 +1,84 @@
+// Copyright 2024 The PipeCD Authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package initialize
+
+import (
+	"os"
+	"strings"
+	"testing"
+
+	"github.com/goccy/go-yaml"
+	"github.com/stretchr/testify/assert"
+
+	"github.com/pipe-cd/pipecd/pkg/app/pipectl/prompt"
+	"github.com/pipe-cd/pipecd/pkg/config"
+)
+
+func TestGenerateECSConfig(t *testing.T) {
+	t.Parallel()
+
+	testcases := []struct {
+		name         string
+		inputs       string // mock for user's input
+		expectedFile string
+		expectedErr  bool
+	}{
+		{
+			name: "valid inputs",
+			inputs: `myApp
+				serviceDef.yaml
+				taskDef.yaml
+				arn:aws:elasticloadbalancing:ap-northeast-1:123456789012:targetgroup/xxx/xxx
+				web
+				80
+				`,
+			expectedFile: "testdata/ecs-app.yaml",
+			expectedErr:  false,
+		},
+		{
+			name: "missing required",
+			inputs: `myApp
+				serviceDef.yaml
+				 `,
+			expectedFile: "",
+			expectedErr:  true,
+		},
+	}
+
+	for _, tc := range testcases {
+		tc := tc
+		t.Run(tc.name, func(t *testing.T) {
+			reader := strings.NewReader(tc.inputs)
+			prompt := prompt.NewPrompt(reader)
+
+			// Generate the config
+			cfg, err := generateECSConfig(prompt)
+			assert.Equal(t, tc.expectedErr, err != nil)
+
+			if err == nil {
+				// Compare the YAML output
+				yml, err := yaml.Marshal(cfg)
+				assert.NoError(t, err)
+				file, err := os.ReadFile(tc.expectedFile)
+				assert.NoError(t, err)
+				assert.Equal(t, string(file), string(yml))
+
+				// Check if the YAML output is compatible with the original Config model
+				_, err = config.DecodeYAML(yml)
+				assert.NoError(t, err)
+			}
+		})
+	}
+}

pkg/app/pipectl/cmd/initialize/initialize.go

@@ -0,0 +1,160 @@
+// Copyright 2024 The PipeCD Authors.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package initialize
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/goccy/go-yaml"
+	"github.com/spf13/cobra"
+
+	"github.com/pipe-cd/pipecd/pkg/app/pipectl/exporter"
+	"github.com/pipe-cd/pipecd/pkg/app/pipectl/prompt"
+	"github.com/pipe-cd/pipecd/pkg/cli"
+	"github.com/pipe-cd/pipecd/pkg/config"
+)
+
+type command struct {
+	// Add flags if needed.
+}
+
+const (
+	platformKubernetes string = "0"
+	platformECS        string = "1"
+)
+
+// Use genericConfigs in order to simplify using the spec.
+type genericConfig struct {
+	APIVersion      string      `json:"apiVersion"`
+	Kind            config.Kind `json:"kind"`
+	ApplicationSpec interface{} `json:"spec"`
+}
+
+func NewCommand() *cobra.Command {
+	c := &command{}
+	cmd := &cobra.Command{
+		Use:     "init",
+		Short:   "Generate a app.pipecd.yaml easily and interactively",
+		Example: `  pipectl init`,
+		Long:    "Generate a app.pipecd.yaml easily, interactively selecting options.",
+		RunE:    cli.WithContext(c.run),
+	}
+
+	return cmd
+}
+
+func (c *command) run(ctx context.Context, input cli.Input) error {
+	// Enable interrupt signal.
+	ctx, cancel := context.WithCancel(ctx)
+	signals := make(chan os.Signal, 1)
+	signal.Notify(signals, os.Interrupt, syscall.SIGHUP, syscall.SIGTERM)
+
+	defer func() {
+		signal.Stop(signals)
+		cancel()
+	}()
+
+	go func() {
+		select {
+		case s := <-signals:
+			fmt.Printf("Interrupted by signal: %v\n", s)
+			cancel()
+			os.Exit(1)
+		case <-ctx.Done():
+		}
+	}()
+
+	p := prompt.NewPrompt(os.Stdin)
+	return generateConfig(ctx, input, p)
+}
+
+func generateConfig(ctx context.Context, input cli.Input, p prompt.Prompt) error {
+	// user's inputs
+	var (
+		platform   string
+		exportPath string
+	)
+
+	platformInput := prompt.Input{
+		Message:       fmt.Sprintf("Which platform? Enter the number [%s]Kubernetes [%s]ECS", platformKubernetes, platformECS),
+		TargetPointer: &platform,
+		Required:      true,
+	}
+	exportPathInput := prompt.Input{
+		Message:       "Path to save the config (if not specified, it goes to stdout)",
+		TargetPointer: &exportPath,
+		Required:      false,
+	}
+
+	err := p.Run(platformInput)
+	if err != nil {
+		return fmt.Errorf("invalid platform number: %v", err)
+	}
+
+	var cfg *genericConfig
+	switch platform {
+	case platformKubernetes:
+		panic("not implemented")
+	case platformECS:
+		cfg, err = generateECSConfig(p)
+	default:
+		return fmt.Errorf("invalid platform number: %s", platform)
+	}
+
+	if err != nil {
+		return err
+	}
+
+	cfgBytes, err := yaml.Marshal(cfg)
+	if err != nil {
+		return err
+	}
+
+	fmt.Println("### The config model was successfully prepared. Move on to exporting. ###")
+	err = p.Run(exportPathInput)
+	if err != nil {
+		printConfig(cfgBytes)
+		return err
+	}
+	err = export(cfgBytes, exportPath)
+	if err != nil {
+		return nil
+	}
+
+	return nil
+}
+
+func export(cfgBytes []byte, exportPath string) error {
+	if len(exportPath) == 0 {
+		// if the path is not specified, print to stdout
+		printConfig(cfgBytes)
+		return nil
+	}
+	err := exporter.Export(cfgBytes, exportPath)
+	if err != nil {
+		printConfig(cfgBytes)
+		return err
+	}
+	return nil
+}
+
+// Print the config to stdout.
+func printConfig(configBytes []byte) {
+	fmt.Printf("\n### Generated Config is below ###\n%s\n", string(configBytes))
+}

pkg/app/pipectl/cmd/initialize/testdata/ecs-app.yaml

@@ -0,0 +1,13 @@
+apiVersion: pipecd.dev/v1beta1
+kind: ECSApp
+spec:
+  name: myApp
+  input:
+    serviceDefinitionFile: serviceDef.yaml
+    taskDefinitionFile: taskDef.yaml
+    targetGroups:
+      primary:
+        targetGroupArn: arn:aws:elasticloadbalancing:ap-northeast-1:123456789012:targetgroup/xxx/xxx
+        containerName: web
+        containerPort: 80
+  description: Generated by `pipectl init`. See https://pipecd.dev/docs/user-guide/configuration-reference/ for more.