Merge pull request #52 from zeabur/xiayuhang1106/zea-1524-template-queries-and-mutations

Xiayuhang1106/zea 1524 template queries and mutations

Author: MichaelYuhe

.golangci.yml

@@ -16,5 +16,6 @@ linters-settings:
   govet:
     check-shadowing: true
     enable-all: true
+    disable: ["fieldalignment"]
   errcheck:
     exclude-functions:

internal/cmd/project/create/create.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/briandowns/spinner"
 	"github.com/spf13/cobra"
+
 	"github.com/zeabur/cli/internal/cmdutil"
 )
 
@@ -41,7 +42,6 @@ func runCreate(f *cmdutil.Factory, opts *Options) error {
 	}
 
 	return runCreateNonInteractive(f, opts)
-
 }
 
 func runCreateInteractive(f *cmdutil.Factory, opts *Options) error {

internal/cmd/root/root.go

@@ -20,6 +20,7 @@ import (
 	deploymentCmd "github.com/zeabur/cli/internal/cmd/deployment"
 	projectCmd "github.com/zeabur/cli/internal/cmd/project"
 	serviceCmd "github.com/zeabur/cli/internal/cmd/service"
+	templateCmd "github.com/zeabur/cli/internal/cmd/template"
 	versionCmd "github.com/zeabur/cli/internal/cmd/version"
 	"github.com/zeabur/cli/internal/cmdutil"
 	"github.com/zeabur/cli/pkg/api"
@@ -108,6 +109,7 @@ func NewCmdRoot(f *cmdutil.Factory, version, commit, date string) (*cobra.Comman
 	cmd.AddCommand(projectCmd.NewCmdProject(f))
 	cmd.AddCommand(serviceCmd.NewCmdService(f))
 	cmd.AddCommand(deploymentCmd.NewCmdDeployment(f))
+	cmd.AddCommand(templateCmd.NewCmdTemplate(f))
 
 	cmd.AddCommand(contextCmd.NewCmdContext(f))
 

internal/cmd/template/delete/delete.go

@@ -0,0 +1,94 @@
+package delete
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/briandowns/spinner"
+	"github.com/spf13/cobra"
+
+	"github.com/zeabur/cli/internal/cmdutil"
+)
+
+type Options struct {
+	code string
+}
+
+func NewCmdDelete(f *cmdutil.Factory) *cobra.Command {
+	opts := Options{}
+
+	cmd := &cobra.Command{
+		Use:   "delete",
+		Short: "Delete template by code",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return runDelete(f, opts)
+		},
+	}
+
+	cmd.Flags().StringVarP(&opts.code, "code", "c", "", "Template code")
+
+	return cmd
+}
+
+func runDelete(f *cmdutil.Factory, opts Options) error {
+	if f.Interactive {
+		return runDeleteInteractive(f, opts)
+	}
+	return runDeleteNonInteractive(f, opts)
+
+}
+
+func runDeleteInteractive(f *cmdutil.Factory, opts Options) error {
+	code, err := f.Prompter.Input("Template Code: ", "")
+	if err != nil {
+		return err
+	}
+
+	opts.code = code
+
+	err = deleteTemplate(f, opts)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func runDeleteNonInteractive(f *cmdutil.Factory, opts Options) error {
+	err := paramCheck(opts)
+	if err != nil {
+		return err
+	}
+
+	err = deleteTemplate(f, opts)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func deleteTemplate(f *cmdutil.Factory, opts Options) error {
+	s := spinner.New(cmdutil.SpinnerCharSet, cmdutil.SpinnerInterval,
+		spinner.WithColor(cmdutil.SpinnerColor),
+		spinner.WithSuffix(" Deleting template..."),
+	)
+	s.Start()
+	err := f.ApiClient.DeleteTemplate(context.Background(), opts.code)
+	if err != nil {
+		return err
+	}
+	s.Stop()
+
+	f.Log.Info("Delete template successfully")
+
+	return nil
+}
+
+func paramCheck(opts Options) error {
+	if opts.code == "" {
+		return fmt.Errorf("template code is required")
+	}
+
+	return nil
+}

internal/cmd/template/deploy/deploy.go

@@ -0,0 +1,64 @@
+package deploy
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/briandowns/spinner"
+	"github.com/spf13/cobra"
+	"github.com/zeabur/cli/internal/cmdutil"
+)
+
+type Options struct {
+	code string
+}
+
+func NewCmdDeploy(f *cmdutil.Factory) *cobra.Command {
+	opts := &Options{}
+
+	cmd := &cobra.Command{
+		Use:   "deploy",
+		Short: "Deploy a template",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return runDeploy(f, opts)
+		},
+	}
+
+	cmd.Flags().StringVarP(&opts.code, "code", "c", "", "Template code")
+
+	return cmd
+}
+
+func runDeploy(f *cmdutil.Factory, opts *Options) error {
+	var err error
+
+	err = paramCheck(opts)
+	if err != nil {
+		return err
+	}
+
+	s := spinner.New(cmdutil.SpinnerCharSet, cmdutil.SpinnerInterval,
+		spinner.WithColor(cmdutil.SpinnerColor),
+		spinner.WithSuffix(" Fetching template..."),
+	)
+	s.Start()
+
+	template, err := f.ApiClient.GetTemplate(context.Background(), opts.code)
+	if err != nil {
+		return err
+	}
+
+	fmt.Printf("Template: %s\n", template.Name)
+
+	s.Stop()
+
+	return nil
+}
+
+func paramCheck(opts *Options) error {
+	if opts.code == "" {
+		return fmt.Errorf("code is required")
+	}
+
+	return nil
+}

internal/cmd/template/get/get.go

@@ -0,0 +1,98 @@
+package get
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/briandowns/spinner"
+	"github.com/spf13/cobra"
+
+	"github.com/zeabur/cli/internal/cmdutil"
+)
+
+type Options struct {
+	code string
+}
+
+func NewCmdGet(f *cmdutil.Factory) *cobra.Command {
+	opts := Options{}
+
+	cmd := &cobra.Command{
+		Use:   "get",
+		Short: "Get template by code",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return runGet(f, opts)
+		},
+	}
+
+	cmd.Flags().StringVarP(&opts.code, "code", "c", "", "Template code")
+
+	return cmd
+}
+
+func runGet(f *cmdutil.Factory, opts Options) error {
+	if f.Interactive {
+		return runGetInteractive(f, opts)
+	}
+	return runGetNonInteractive(f, opts)
+
+}
+
+func runGetInteractive(f *cmdutil.Factory, opts Options) error {
+	code, err := f.Prompter.Input("Template Code: ", "")
+	if err != nil {
+		return err
+	}
+
+	opts.code = code
+
+	err = getTemplate(f, opts)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func runGetNonInteractive(f *cmdutil.Factory, opts Options) error {
+	err := paramCheck(opts)
+	if err != nil {
+		return err
+	}
+
+	err = getTemplate(f, opts)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func getTemplate(f *cmdutil.Factory, opts Options) error {
+	s := spinner.New(cmdutil.SpinnerCharSet, cmdutil.SpinnerInterval,
+		spinner.WithColor(cmdutil.SpinnerColor),
+		spinner.WithSuffix(" Fetching template..."),
+	)
+	s.Start()
+	template, err := f.ApiClient.GetTemplate(context.Background(), opts.code)
+	if err != nil {
+		return err
+	}
+	s.Stop()
+
+	if template == nil || template.Code == "" {
+		fmt.Println("Template not found")
+	} else {
+		f.Printer.Table([]string{"Code", "Name", "Description"}, [][]string{{template.Code, template.Name, template.Description}})
+	}
+
+	return nil
+}
+
+func paramCheck(opts Options) error {
+	if opts.code == "" {
+		return fmt.Errorf("template code is required")
+	}
+
+	return nil
+}

internal/cmd/template/list/list.go

@@ -0,0 +1,37 @@
+package list
+
+import (
+	"context"
+
+	"github.com/spf13/cobra"
+	"github.com/zeabur/cli/internal/cmdutil"
+)
+
+type Options struct {
+}
+
+func NewCmdList(f *cmdutil.Factory) *cobra.Command {
+	opts := Options{}
+
+	cmd := &cobra.Command{
+		Use:     "list",
+		Short:   "List templates",
+		Aliases: []string{"ls"},
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return runList(f, opts)
+		},
+	}
+
+	return cmd
+}
+
+func runList(f *cmdutil.Factory, opts Options) error {
+	templates, err := f.ApiClient.ListAllTemplates(context.Background())
+	if err != nil {
+		return err
+	}
+
+	f.Printer.Table(templates.Header(), templates.Rows())
+
+	return nil
+}

internal/cmd/template/template.go

@@ -0,0 +1,26 @@
+package template
+
+import (
+	"github.com/spf13/cobra"
+
+	"github.com/zeabur/cli/internal/cmdutil"
+
+	templateDeleteCmd "github.com/zeabur/cli/internal/cmd/template/delete"
+	templateDeployCmd "github.com/zeabur/cli/internal/cmd/template/deploy"
+	templateGetCmd "github.com/zeabur/cli/internal/cmd/template/get"
+	templateListCmd "github.com/zeabur/cli/internal/cmd/template/list"
+)
+
+func NewCmdTemplate(f *cmdutil.Factory) *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "template",
+		Short: "Manage templates",
+	}
+
+	cmd.AddCommand(templateListCmd.NewCmdList(f))
+	cmd.AddCommand(templateDeployCmd.NewCmdDeploy(f))
+	cmd.AddCommand(templateGetCmd.NewCmdGet(f))
+	cmd.AddCommand(templateDeleteCmd.NewCmdDelete(f))
+
+	return cmd
+}

pkg/api/interface.go

@@ -16,6 +16,7 @@ type Client interface {
 	DeploymentAPI
 	LogAPI
 	GitAPI
+	TemplateAPI
 }
 
 type (
@@ -81,4 +82,13 @@ type (
 		GetRepoInfo() (string, string, error)
 		GetRepoBranchesByRepoID(repoID int) ([]string, error)
 	}
+
+	TemplateAPI interface {
+		ListTemplates(ctx context.Context, skip, limit int) (*model.TemplateConnection, error)
+		ListAllTemplates(ctx context.Context) (model.Templates, error)
+		GetTemplate(ctx context.Context, code string) (*model.Template, error)
+
+		DeployTemplate(ctx context.Context, code string, variables map[string]interface{}, repoConfigs model.RepoConfigs) (*model.Project, error)
+		DeleteTemplate(ctx context.Context, code string) error
+	}
 )