Add rich error message for sh into off environment

cmoog · cmoog · commit e8f622d2a381 · 2020-10-21T01:36:30.000-05:00
diff --git a/internal/cmd/shell.go b/internal/cmd/shell.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"context"
+	"fmt"
 	"io"
 	"os"
 	"strings"
@@ -15,6 +16,7 @@ import (
 
 	"cdr.dev/coder-cli/coder-sdk"
 	"cdr.dev/coder-cli/internal/activity"
+	"cdr.dev/coder-cli/internal/clog"
 	"cdr.dev/coder-cli/internal/x/xterminal"
 	"cdr.dev/wsep"
 
@@ -153,7 +155,7 @@ func runCommand(ctx context.Context, envName, command string, args []string) err
 	if err != nil {
 		var closeErr websocket.CloseError
 		if xerrors.As(err, &closeErr) {
-			return xerrors.Errorf("network error, is %q online?", envName)
+			return networkErr(client, env)
 		}
 		return xerrors.Errorf("start remote command: %w", err)
 	}
@@ -179,7 +181,6 @@ func runCommand(ctx context.Context, envName, command string, args []string) err
 		}
 	}()
 	go func() {
-
 		if _, err := io.Copy(os.Stderr, process.Stderr()); err != nil {
 			cancel()
 		}
@@ -188,13 +189,26 @@ func runCommand(ctx context.Context, envName, command string, args []string) err
 	if err := process.Wait(); err != nil {
 		var closeErr websocket.CloseError
 		if xerrors.Is(err, ctx.Err()) || xerrors.As(err, &closeErr) {
-			return xerrors.Errorf("network error, is %q online?", envName)
+			return networkErr(client, env)
 		}
 		return err
 	}
 	return nil
 }
 
+func networkErr(client *coder.Client, env *coder.Environment) error {
+	if env.LatestStat.ContainerStatus != coder.EnvironmentOn {
+		return clog.Fatal(
+			"environment is not running", 
+			fmt.Sprintf("environment %q is not running", env.Name),
+			fmt.Sprintf("its current status is %q",env.LatestStat.ContainerStatus),
+			"",
+			clog.Bold("tip: ")+fmt.Sprintf("run \"coder envs rebuild %s --follow\" to start the environment", env.Name),
+		)
+	}
+	return xerrors.Errorf("network error, is %q online?", env.Name)
+}
+
 func heartbeat(ctx context.Context, conn *websocket.Conn, interval time.Duration) {
 	ticker := time.NewTicker(interval)
 	defer ticker.Stop()
