From 9deb80c4879ca824473b6a827ca4371449efa782 Mon Sep 17 00:00:00 2001
From: Andrea Luzzardi <aluzzardi@gmail.com>
Date: Fri, 28 May 2021 16:54:37 -0700
Subject: [PATCH] dagger input edit

See and edit all the inputs in bulk using $EDITOR

Signed-off-by: Andrea Luzzardi <aluzzardi@gmail.com>
---
 cmd/dagger/cmd/input/edit.go | 113 +++++++++++++++++++++++++++++++++++
 cmd/dagger/cmd/input/root.go |   1 +
 go.mod                       |   1 +
 3 files changed, 115 insertions(+)
 create mode 100644 cmd/dagger/cmd/input/edit.go

diff --git a/cmd/dagger/cmd/input/edit.go b/cmd/dagger/cmd/input/edit.go
new file mode 100644
index 00000000..92f20502
--- /dev/null
+++ b/cmd/dagger/cmd/input/edit.go
@@ -0,0 +1,113 @@
+package input
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"os/exec"
+	"strings"
+
+	"go.dagger.io/dagger/cmd/dagger/cmd/common"
+	"go.dagger.io/dagger/cmd/dagger/logger"
+	"go.dagger.io/dagger/state"
+
+	"github.com/google/shlex"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+	"gopkg.in/yaml.v3"
+)
+
+var editCmd = &cobra.Command{
+	Use:   "edit",
+	Short: "Interactively edit the inputs of an environment",
+	Args:  cobra.MaximumNArgs(1),
+	PreRun: func(cmd *cobra.Command, args []string) {
+		// Fix Viper bug for duplicate flags:
+		// https://github.com/spf13/viper/issues/233
+		if err := viper.BindPFlags(cmd.Flags()); err != nil {
+			panic(err)
+		}
+	},
+	Run: func(cmd *cobra.Command, args []string) {
+		lg := logger.New()
+		ctx := lg.WithContext(cmd.Context())
+
+		workspace := common.CurrentWorkspace(ctx)
+		st := common.CurrentEnvironmentState(ctx, workspace)
+
+		data, err := yaml.Marshal(st)
+		if err != nil {
+			lg.Fatal().Err(err).Msg("unable to marshal state")
+		}
+
+		f, err := os.CreateTemp("", fmt.Sprintf("%s-*.yaml", st.Name))
+		if err != nil {
+			lg.Fatal().Err(err).Msg("failed to create temporary file")
+		}
+		tmpPath := f.Name()
+		defer os.Remove(tmpPath)
+
+		if _, err := f.Write(data); err != nil {
+			lg.Fatal().Err(err).Msg("unable to write file")
+		}
+		f.Close()
+
+		if err := runEditor(ctx, tmpPath); err != nil {
+			lg.Fatal().Err(err).Msg("failed to start editor")
+		}
+
+		data, err = os.ReadFile(tmpPath)
+		if err != nil {
+			lg.Fatal().Err(err).Msg("failed to read temporary file")
+		}
+		var newState state.State
+		if err := yaml.Unmarshal(data, &newState); err != nil {
+			lg.Fatal().Err(err).Msg("failed to decode file")
+		}
+		st.Inputs = newState.Inputs
+		if err := workspace.Save(ctx, st); err != nil {
+			lg.Fatal().Err(err).Msg("failed to save state")
+		}
+	},
+}
+
+func runEditor(ctx context.Context, path string) error {
+	editor := os.Getenv("EDITOR")
+	var cmd *exec.Cmd
+	if editor == "" {
+		editor, err := lookupAnyEditor("vim", "nano", "vi")
+		if err != nil {
+			return err
+		}
+		cmd = exec.CommandContext(ctx, editor, path)
+	} else {
+		parts, err := shlex.Split(editor)
+		if err != nil {
+			return fmt.Errorf("invalid $EDITOR: %s", editor)
+		}
+		parts = append(parts, path)
+		cmd = exec.CommandContext(ctx, parts[0], parts[1:]...) // #nosec
+	}
+
+	cmd.Env = os.Environ()
+	cmd.Stdin = os.Stdin
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	return cmd.Run()
+}
+
+func lookupAnyEditor(editorNames ...string) (editorPath string, err error) {
+	for _, editorName := range editorNames {
+		editorPath, err = exec.LookPath(editorName)
+		if err == nil {
+			return editorPath, nil
+		}
+	}
+	return "", fmt.Errorf("no editor available: dagger attempts to use the editor defined in the EDITOR environment variable, and if that's not set defaults to any of %s, but none of them could be found", strings.Join(editorNames, ", "))
+}
+
+func init() {
+	if err := viper.BindPFlags(editCmd.Flags()); err != nil {
+		panic(err)
+	}
+}
diff --git a/cmd/dagger/cmd/input/root.go b/cmd/dagger/cmd/input/root.go
index ef1ae3cf..febb0690 100644
--- a/cmd/dagger/cmd/input/root.go
+++ b/cmd/dagger/cmd/input/root.go
@@ -20,6 +20,7 @@ var Cmd = &cobra.Command{
 
 func init() {
 	Cmd.AddCommand(
+		editCmd,
 		dirCmd,
 		gitCmd,
 		containerCmd,
diff --git a/go.mod b/go.mod
index 68685ce2..0ed3cba4 100644
--- a/go.mod
+++ b/go.mod
@@ -10,6 +10,7 @@ require (
 	github.com/containerd/console v1.0.2
 	github.com/docker/distribution v2.7.1+incompatible
 	github.com/emicklei/proto v1.9.0 // indirect
+	github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510
 	github.com/hashicorp/go-version v1.3.0
 	github.com/jaguilar/vt100 v0.0.0-20150826170717-2703a27b14ea
 	github.com/mattn/go-colorable v0.1.8 // indirect