cli/cmd/devtool/dotenv.go - main - dimamir/rill

cli/cmd/devtool/dotenv.go
121 строка3 KB
Himadri Singh
Rill E2E setup (#6082)
15 ноя 2024, 20:24
15 ноя 2024, 20:243c0e9b1
package devtool

import (
	"context"
	"fmt"
	"os"
	"os/exec"

	"github.com/rilldata/rill/cli/pkg/cmdutil"
	"github.com/spf13/cobra"
)

func DotenvCmd(ch *cmdutil.Helper) *cobra.Command {
	cmd := &cobra.Command{
		Use:   "dotenv",
		Short: "Utilities for managing .env files",
	}

	cmd.AddCommand(DotenvRefreshCmd(ch))
	cmd.AddCommand(DotenvUploadCmd(ch))

	return cmd
}

func DotenvRefreshCmd(ch *cmdutil.Helper) *cobra.Command {
	cmd := &cobra.Command{
		Use:   "refresh {cloud}",
		Short: "Refresh .env file from shared storage",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			preset := args[0]
			if preset != "cloud" {
				return fmt.Errorf(".env not used for preset %q", preset)
			}

			err := checkRillRepo()
			if err != nil {
				return err
			}

			err = downloadDotenv(cmd.Context(), preset)
			if err != nil {
				return err
			}

			return nil
		},
	}

	return cmd
}

func DotenvUploadCmd(ch *cmdutil.Helper) *cobra.Command {
	cmd := &cobra.Command{
		Use:   "upload {cloud}",
		Short: "Distribute your current .env file to the team",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			preset := args[0]
			if preset != "cloud" {
				return fmt.Errorf(".env not used for preset %q", preset)
			}

			err := checkRillRepo()
			if err != nil {
				return err
			}

			err = checkDotenv()
			if err != nil {
				return err
			}

			ch.PrintfWarn("This will overwrite the .env file in shared storage with the contents of your local .env file.\n")
			ch.PrintfWarn("The updated .env will automatically be used by other users of the devtool.\n")
			ok, err := cmdutil.ConfirmPrompt("Do you want to continue?", "", false)
			if err != nil {
				return err
			}
			if !ok {
				return nil
			}

			err = uploadDotenv(cmd.Context(), preset)
			if err != nil {
				return err
			}

			return nil
		},
	}

	return cmd
}

var dotenvURLs = map[string]string{
	"cloud": "gs://rill-devtool/dotenv/cloud-dev.env",
	"e2e":   "gs://rill-devtool/dotenv/cloud-e2e.env",
}

func checkDotenv() error {
	_, err := os.Stat(".env")
	if err != nil {
		return fmt.Errorf(".env file not found at the root of the rill repository")
	}
	return nil
}

func downloadDotenv(ctx context.Context, preset string) error {
	logInfo.Printf("Downloading .env file from %s\n", dotenvURLs[preset])

	err := exec.CommandContext(ctx, "gcloud", "storage", "cp", dotenvURLs[preset], ".env").Run()
	if err != nil {
		return fmt.Errorf("error syncing '.env' file from GCS (you must be a Rill team member and have authenticated `gcloud`): %w", err)
	}
	return nil
}

func uploadDotenv(ctx context.Context, preset string) error {
	return exec.CommandContext(ctx, "gcloud", "storage", "cp", ".env", dotenvURLs[preset]).Run()
}