cli/cmd/env/push.go - main - dimamir/rill

cli/cmd/env/push.go
121 строка3 KB
Anshul Khandelwal
nudge user to pass flag when unable to infer project name (#6053)
13 ноя 2024, 18:32
13 ноя 2024, 18:3273f7cb7
package env

import (
	"fmt"
	"path/filepath"

	"github.com/rilldata/rill/cli/pkg/cmdutil"
	adminv1 "github.com/rilldata/rill/proto/gen/rill/admin/v1"
	"github.com/rilldata/rill/runtime/compilers/rillv1"
	"github.com/spf13/cobra"
)

func PushCmd(ch *cmdutil.Helper) *cobra.Command {
	var projectPath, projectName, environment string

	pushCmd := &cobra.Command{
		Use:   "push",
		Short: "Push local .env contents to cloud",
		RunE: func(cmd *cobra.Command, args []string) error {
			// If projectPath is provided, normalize it
			if projectPath != "" {
				var err error
				projectPath, err = normalizeProjectPath(projectPath)
				if err != nil {
					return err
				}
			}

			// Parse and verify the project directory
			repo, instanceID, err := cmdutil.RepoForProjectPath(projectPath)
			if err != nil {
				return err
			}
			parser, err := rillv1.Parse(cmd.Context(), repo, instanceID, "prod", "duckdb")
			if err != nil {
				return fmt.Errorf("failed to parse project: %w", err)
			}
			if parser.RillYAML == nil {
				return fmt.Errorf("not a valid Rill project (missing a rill.yaml file)")
			}

			// Find the cloud project name
			if projectName == "" {
				projectName, err = ch.InferProjectName(cmd.Context(), ch.Org, projectPath)
				if err != nil {
					return fmt.Errorf("unable to infer project name (use `--project` to explicitly specify the name): %w", err)
				}
			}

			// Fetch the project variables from the cloud
			client, err := ch.Client()
			if err != nil {
				return err
			}
			res, err := client.GetProjectVariables(cmd.Context(), &adminv1.GetProjectVariablesRequest{
				Organization: ch.Org,
				Project:      projectName,
				Environment:  environment,
			})
			if err != nil {
				return err
			}

			// Merge the current .env file with the cloud variables
			vars := make(map[string]string)
			for _, v := range res.Variables {
				vars[v.Name] = v.Value
			}
			added := 0
			changed := 0
			for k, v := range parser.DotEnv {
				if _, ok := vars[k]; !ok {
					added++
				} else if vars[k] != v {
					changed++
				}
				vars[k] = v
			}

			// If there were no changes, exit early
			if added+changed == 0 {
				ch.Print("There are no new or changed variables in your local .env file.\n")
				return nil
			}

			// Prompt for confirmation if any existing variables have changed
			if changed != 0 {
				ch.Printf("Found %d variable(s) in your local .env file that will overwrite existing variables in the cloud env for project %q.\n", changed, projectName)
				ok, err := cmdutil.ConfirmPrompt("Do you want to continue?", "", true)
				if err != nil {
					return err
				}
				if !ok {
					return nil
				}
			}

			// Write the merged variables back to the cloud project
			if added+changed != 0 {
				_, err = client.UpdateProjectVariables(cmd.Context(), &adminv1.UpdateProjectVariablesRequest{
					Organization: ch.Org,
					Project:      projectName,
					Environment:  environment,
					Variables:    vars,
				})
				if err != nil {
					return err
				}
			}

			ch.Printf("Updated cloud env for project %q with variables from %q.\n", projectName, filepath.Join(projectPath, ".env"))
			return nil
		},
	}

	pushCmd.Flags().StringVar(&projectPath, "path", ".", "Project directory")
	pushCmd.Flags().StringVar(&projectName, "project", "", "Cloud project name (will attempt to infer from Git remote if not provided)")
	pushCmd.Flags().StringVar(&environment, "environment", "", "Optional environment to resolve for (options: dev, prod)")

	return pushCmd
}