feat: Add "coder" CLI (#221)

* feat: Add "coder" CLI

* Add CLI test for login

* Add "bin/coder" target to Makefile

* Update promptui to fix race

* Fix error scope

* Don't run CLI tests on Windows

* Fix requested changes

coderd/cmd/root.go

@@ -1,9 +1,14 @@
 package cmd
 
 import (
+	"context"
+	"io"
+	"io/ioutil"
 	"net"
 	"net/http"
+	"net/url"
 	"os"
+	"time"
 
 	"github.com/spf13/cobra"
 	"golang.org/x/xerrors"
@@ -11,8 +16,13 @@ import (
 	"cdr.dev/slog"
 	"cdr.dev/slog/sloggers/sloghuman"
 	"github.com/coder/coder/coderd"
+	"github.com/coder/coder/codersdk"
 	"github.com/coder/coder/database"
 	"github.com/coder/coder/database/databasefake"
+	"github.com/coder/coder/provisioner/terraform"
+	"github.com/coder/coder/provisionerd"
+	"github.com/coder/coder/provisionersdk"
+	"github.com/coder/coder/provisionersdk/proto"
 )
 
 func Root() *cobra.Command {
@@ -22,8 +32,9 @@ func Root() *cobra.Command {
 	root := &cobra.Command{
 		Use: "coderd",
 		RunE: func(cmd *cobra.Command, args []string) error {
+			logger := slog.Make(sloghuman.Sink(os.Stderr))
 			handler := coderd.New(&coderd.Options{
-				Logger:   slog.Make(sloghuman.Sink(os.Stderr)),
+				Logger:   logger,
 				Database: databasefake.New(),
 				Pubsub:   database.NewPubsubInMemory(),
 			})
@@ -34,6 +45,16 @@ func Root() *cobra.Command {
 			}
 			defer listener.Close()
 
+			client := codersdk.New(&url.URL{
+				Scheme: "http",
+				Host:   address,
+			})
+			closer, err := newProvisionerDaemon(cmd.Context(), client, logger)
+			if err != nil {
+				return xerrors.Errorf("create provisioner daemon: %w", err)
+			}
+			defer closer.Close()
+
 			errCh := make(chan error)
 			go func() {
 				defer close(errCh)
@@ -56,3 +77,31 @@ func Root() *cobra.Command {
 
 	return root
 }
+
+func newProvisionerDaemon(ctx context.Context, client *codersdk.Client, logger slog.Logger) (io.Closer, error) {
+	terraformClient, terraformServer := provisionersdk.TransportPipe()
+	go func() {
+		err := terraform.Serve(ctx, &terraform.ServeOptions{
+			ServeOptions: &provisionersdk.ServeOptions{
+				Listener: terraformServer,
+			},
+			Logger: logger,
+		})
+		if err != nil {
+			panic(err)
+		}
+	}()
+	tempDir, err := ioutil.TempDir("", "provisionerd")
+	if err != nil {
+		return nil, err
+	}
+	return provisionerd.New(client.ProvisionerDaemonClient, &provisionerd.Options{
+		Logger:         logger,
+		PollInterval:   50 * time.Millisecond,
+		UpdateInterval: 50 * time.Millisecond,
+		Provisioners: provisionerd.Provisioners{
+			string(database.ProvisionerTypeTerraform): proto.NewDRPCProvisionerClient(provisionersdk.Conn(terraformClient)),
+		},
+		WorkDirectory: tempDir,
+	}), nil
+}

coderd/coderd.go

@@ -37,6 +37,7 @@ func New(options *Options) http.Handler {
 		r.Post("/login", api.postLogin)
 		r.Post("/logout", api.postLogout)
 		// Used for setup.
+		r.Get("/user", api.user)
 		r.Post("/user", api.postUser)
 		r.Route("/users", func(r chi.Router) {
 			r.Use(

coderd/users.go

@@ -55,6 +55,26 @@ type LoginWithPasswordResponse struct {
 	SessionToken string `json:"session_token" validate:"required"`
 }
 
+// Returns whether the initial user has been created or not.
+func (api *api) user(rw http.ResponseWriter, r *http.Request) {
+	userCount, err := api.Database.GetUserCount(r.Context())
+	if err != nil {
+		httpapi.Write(rw, http.StatusInternalServerError, httpapi.Response{
+			Message: fmt.Sprintf("get user count: %s", err.Error()),
+		})
+		return
+	}
+	if userCount == 0 {
+		httpapi.Write(rw, http.StatusNotFound, httpapi.Response{
+			Message: "The initial user has not been created!",
+		})
+		return
+	}
+	httpapi.Write(rw, http.StatusOK, httpapi.Response{
+		Message: "The initial user has already been created!",
+	})
+}
+
 // Creates the initial user for a Coder deployment.
 func (api *api) postUser(rw http.ResponseWriter, r *http.Request) {
 	var createUser CreateInitialUserRequest

coderd/users_test.go

@@ -13,6 +13,26 @@ import (
 	"github.com/coder/coder/httpmw"
 )
 
+func TestUser(t *testing.T) {
+	t.Parallel()
+	t.Run("NotFound", func(t *testing.T) {
+		t.Parallel()
+		client := coderdtest.New(t)
+		has, err := client.HasInitialUser(context.Background())
+		require.NoError(t, err)
+		require.False(t, has)
+	})
+
+	t.Run("Found", func(t *testing.T) {
+		t.Parallel()
+		client := coderdtest.New(t)
+		_ = coderdtest.CreateInitialUser(t, client)
+		has, err := client.HasInitialUser(context.Background())
+		require.NoError(t, err)
+		require.True(t, has)
+	})
+}
+
 func TestPostUser(t *testing.T) {
 	t.Parallel()
 	t.Run("BadRequest", func(t *testing.T) {