feat: Add "coder" CLI

coder · kylecarbs · Feb 10, 2022 · Feb 9, 2022 · Feb 9, 2022 · Feb 9, 2022
commit d6a1eb825d3d02695f30eb367e3059e78c8b28d5
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -31,16 +31,19 @@
     "drpcconn",
     "drpcmux",
     "drpcserver",
+    "fatih",
     "goleak",
     "hashicorp",
     "httpmw",
     "Jobf",
+    "manifoldco",
     "moby",
     "nhooyr",
     "nolint",
     "nosec",
     "oneof",
     "parameterscopeid",
+    "promptui",
     "protobuf",
     "provisionerd",
     "provisionersdk",

diff --git a/cli/config/file.go b/cli/config/file.go
@@ -0,0 +1,71 @@
+package config
+
+import (
+	"io/ioutil"
+	"os"
+	"path/filepath"
+)
+
+// Root represents the configuration directory.
+type Root string
+
+func (r Root) Session() File {
+	return File(filepath.Join(string(r), "session"))
+}
+
+func (r Root) URL() File {
+	return File(filepath.Join(string(r), "url"))
+}
+
+func (r Root) Organization() File {
+	return File(filepath.Join(string(r), "organization"))
+}
+
+// File provides convenience methods for interacting with *os.File.
+type File string
+
+// Delete deletes the file.
+func (f File) Delete() error {
+	return os.Remove(string(f))
+}
+
+// Write writes the string to the file.
+func (f File) Write(s string) error {
+	return write(string(f), 0600, []byte(s))
+}
+
+// Read reads the file to a string.
+func (f File) Read() (string, error) {
+	byt, err := read(string(f))
+	return string(byt), err
+}
+
+// open opens a file in the configuration directory,
+// creating all intermediate directories.
+func open(path string, flag int, mode os.FileMode) (*os.File, error) {
+	err := os.MkdirAll(filepath.Dir(path), 0750)
+	if err != nil {
+		return nil, err
+	}
+
+	return os.OpenFile(path, flag, mode)
+}
+
+func write(path string, mode os.FileMode, dat []byte) error {
+	fi, err := open(path, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, mode)
+	if err != nil {
+		return err
+	}
+	defer fi.Close()
+	_, err = fi.Write(dat)
+	return err
+}
+
+func read(path string) ([]byte, error) {
+	fi, err := open(path, os.O_RDONLY, 0)
+	if err != nil {
+		return nil, err
+	}
+	defer fi.Close()
+	return ioutil.ReadAll(fi)
+}
diff --git a/cli/config/file_test.go b/cli/config/file_test.go
@@ -0,0 +1,38 @@
+package config_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/cli/config"
+)
+
+func TestFile(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Write", func(t *testing.T) {
+		t.Parallel()
+		err := config.Root(t.TempDir()).Session().Write("test")
+		require.NoError(t, err)
+	})
+
+	t.Run("Read", func(t *testing.T) {
+		t.Parallel()
+		root := config.Root(t.TempDir())
+		err := root.Session().Write("test")
+		require.NoError(t, err)
+		data, err := root.Session().Read()
+		require.NoError(t, err)
+		require.Equal(t, "test", string(data))
+	})
+
+	t.Run("Delete", func(t *testing.T) {
+		t.Parallel()
+		root := config.Root(t.TempDir())
+		err := root.Session().Write("test")
+		require.NoError(t, err)
+		err = root.Session().Delete()
+		require.NoError(t, err)
+	})
+}
diff --git a/cli/login.go b/cli/login.go
@@ -0,0 +1,131 @@
+package cli
+
+import (
+	"errors"
+	"fmt"
+	"net/url"
+	"strings"
+
+	"github.com/coder/coder/coderd"
+	"github.com/coder/coder/codersdk"
+	"github.com/fatih/color"
+	"github.com/go-playground/validator/v10"
+	"github.com/manifoldco/promptui"
+	"github.com/spf13/cobra"
+	"golang.org/x/xerrors"
+)
+
+func login() *cobra.Command {
+	return &cobra.Command{
+		Use:  "login <url>",
+		Args: cobra.ExactArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			rawURL := args[0]
+			if !strings.HasPrefix(rawURL, "http://") && !strings.HasPrefix(rawURL, "https://") {
+				scheme := "https"
+				if strings.HasPrefix(rawURL, "localhost") {
+					scheme = "http"
+				}
+				rawURL = fmt.Sprintf("%s://%s", scheme, rawURL)
+			}
+			serverURL, err := url.Parse(rawURL)
+			if err != nil {
+				return err
+			}
+			// Default to HTTPs. Enables simple URLs like: master.cdr.dev
+			if serverURL.Scheme == "" {
+				serverURL.Scheme = "https"
+			}
+
+			client := codersdk.New(serverURL)
+			hasInitialUser, err := client.HasInitialUser(cmd.Context())
+			if err != nil {
+				return err
+			}
+			if !hasInitialUser {
+				if !isTTY(cmd.InOrStdin()) {
+					return xerrors.New("the initial user cannot be created in non-interactive mode. use the API")
+				}
+				fmt.Fprintf(cmd.OutOrStdout(), "%s Your Coder deployment hasn't been setup!\n", color.HiBlackString(">"))
+
+				_, err := runPrompt(cmd, &promptui.Prompt{
+					Label:     "Would you like to create the first user?",
+					IsConfirm: true,
+					Default:   "y",
+				})
+				if err != nil {
+					return err
+				}
+
+				username, err := runPrompt(cmd, &promptui.Prompt{
+					Label:   "What username would you like?",
+					Default: "kyle",
+				})
+				if err != nil {
+					return err
+				}
+
+				organization, err := runPrompt(cmd, &promptui.Prompt{
+					Label:   "What is the name of your organization?",
+					Default: "acme-corp",
+				})
+				if err != nil {
+					return err
+				}
+
+				email, err := runPrompt(cmd, &promptui.Prompt{
+					Label: "What's your email?",
+					Validate: func(s string) error {
+						err := validator.New().Var(s, "email")
+						if err != nil {
+							return errors.New("That's not a valid email address!")
+						}
+						return err
+					},
+				})
+				if err != nil {
+					return err
+				}
+
+				password, err := runPrompt(cmd, &promptui.Prompt{
+					Label: "Enter a password:",
+					Mask:  '*',
+				})
+				if err != nil {
+					return err
+				}
+
+				_, err = client.CreateInitialUser(cmd.Context(), coderd.CreateInitialUserRequest{
+					Email:        email,
+					Username:     username,
+					Password:     password,
+					Organization: organization,
+				})
+				if err != nil {
+					return xerrors.Errorf("create initial user: %w", err)
+				}
+				resp, err := client.LoginWithPassword(cmd.Context(), coderd.LoginWithPasswordRequest{
+					Email:    email,
+					Password: password,
+				})
+				if err != nil {
+					return xerrors.Errorf("login with password: %w", err)
+				}
+				config := createConfig(cmd)
+				err = config.Session().Write(resp.SessionToken)
+				if err != nil {
+					return xerrors.Errorf("write session token: %w", err)
+				}
+				err = config.URL().Write(serverURL.String())
+				if err != nil {
+					return xerrors.Errorf("write server url: %w", err)
+				}
+
+				fmt.Fprintf(cmd.OutOrStdout(), "%s Welcome to Coder, %s! You're logged in.\n", color.HiBlackString(">"), color.HiCyanString(username))
+				return nil
+			}
+
+			return nil
+		},
+	}
+}