Can start/connect to ephemeral shell (#277)

Author: Ralph7C2

cmd/ssh.go

@@ -38,7 +38,6 @@ func interactiveSSHView(ctx context.Context, input *views.SSHInput, breadcrumb s
 		input.ServiceIDOrName = validate.ExtractServiceIDFromInstanceID(input.ServiceIDOrName)
 		return InteractiveSSHView(ctx, input, breadcrumb)
 	}
-
 	if input.ServiceIDOrName == "" {
 		// No service specified, show service selection
 		return command.AddToStackFunc(
@@ -48,7 +47,9 @@ func interactiveSSHView(ctx context.Context, input *views.SSHInput, breadcrumb s
 			input,
 			views.NewServiceList(ctx, getServiceListInput(ctx, input), func(ctx context.Context, r resource.Resource) tea.Cmd {
 				input.ServiceIDOrName = r.ID()
-
+				if input.Ephemeral {
+					return InteractiveSSHView(ctx, input, "SSH")
+				}
 				// Show instance selection for the selected service
 				return command.AddToStackFunc(
 					ctx,
@@ -63,6 +64,9 @@ func interactiveSSHView(ctx context.Context, input *views.SSHInput, breadcrumb s
 			}, tui.WithCustomOptions[*service.Model](getSSHTableOptions(ctx, breadcrumb))),
 		)
 	} else if validate.IsServiceID(input.ServiceIDOrName) {
+		if input.Ephemeral {
+			return InteractiveSSHView(ctx, input, "SSH")
+		}
 		// Service ID provided, show instance selection
 		return command.AddToStackFunc(
 			ctx,
@@ -114,6 +118,8 @@ func getSSHTableOptions(ctx context.Context, breadcrumb string) []tui.CustomOpti
 func init() {
 	rootCmd.AddCommand(sshCmd)
 
+	sshCmd.Flags().BoolP("ephemeral", "e", false, "connect to ephemeral instance")
+
 	sshCmd.RunE = func(cmd *cobra.Command, args []string) error {
 		ctx := cmd.Context()
 

pkg/command/wrapper.go

@@ -169,8 +169,13 @@ func AddToStack[T any](stack *tui.StackModel, cmd *cobra.Command, breadcrumb str
 }
 
 func LoadCmd[T any, D any](ctx context.Context, loadData func(context.Context, T) (D, error), in T) tui.TypedCmd[D] {
+	return LoadCmdWithLoadingMsg(ctx, loadData, in, "")
+}
+
+func LoadCmdWithLoadingMsg[T any, D any](ctx context.Context, loadData func(context.Context, T) (D, error), in T, loadingMsg string) tui.TypedCmd[D] {
 	loadDataCmd := func() tea.Msg {
 		return tui.LoadingDataMsg{
+			LoadingMsgTmpl: loadingMsg,
 			Cmd: tea.Sequence(
 				func() tea.Msg {
 					data, err := loadData(ctx, in)

pkg/tui/views/ssh.go

@@ -1,13 +1,18 @@
 package views
 
 import (
+	"bytes"
 	"context"
+	"encoding/json"
 	"fmt"
+	"io"
+	"net/http"
 	"os/exec"
 	"strings"
 
 	"github.com/render-oss/cli/pkg/client"
 	"github.com/render-oss/cli/pkg/command"
+	"github.com/render-oss/cli/pkg/config"
 	"github.com/render-oss/cli/pkg/deploy"
 	"github.com/render-oss/cli/pkg/service"
 	"github.com/render-oss/cli/pkg/tui"
@@ -18,13 +23,18 @@ type SSHInput struct {
 	InstanceID      string // Set when a specific instance is selected or provided
 	Project         *client.Project
 	EnvironmentIDs  []string
+	Ephemeral       bool `cli:"ephemeral"`
 
 	Args []string
 }
 
 // NewSSHView creates an SSH execution view - always returns ExecModel
 func NewSSHView(ctx context.Context, input *SSHInput) *tui.ExecModel {
-	return tui.NewExecModel("ssh", handleSSHError, command.LoadCmd(ctx, loadDataSSH, input))
+	loadingMsg := "%s Preparing SSH connection..."
+	if input.Ephemeral {
+		loadingMsg = "%s Creating ephemeral shell (this may take a moment)..."
+	}
+	return tui.NewExecModel("ssh", handleSSHError, command.LoadCmdWithLoadingMsg(ctx, loadDataSSH, input, loadingMsg))
 }
 
 func handleSSHError(err error) error {
@@ -34,6 +44,46 @@ func handleSSHError(err error) error {
 	}
 }
 
+// createEphemeralShell creates an ephemeral shell pod for the given service.
+// This must be called before SSH'ing into an ephemeral shell.
+func createEphemeralShell(ctx context.Context, serviceID string) error {
+	apiCfg, err := config.DefaultAPIConfig()
+	if err != nil {
+		return fmt.Errorf("failed to get API config: %w", err)
+	}
+
+	url := fmt.Sprintf("%sservices/%s/ephemeral-shell", apiCfg.Host, serviceID)
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, bytes.NewReader(nil))
+	if err != nil {
+		return fmt.Errorf("failed to create request: %w", err)
+	}
+
+	req.Header = client.AddHeaders(req.Header, apiCfg.Key)
+	req.Header.Set("Accept", "application/json")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return fmt.Errorf("failed to create ephemeral shell: %w", err)
+	}
+	defer resp.Body.Close()
+
+	body, _ := io.ReadAll(resp.Body)
+
+	if resp.StatusCode < 200 || resp.StatusCode >= 300 {
+		// Try to extract error message from JSON response
+		var apiErr struct {
+			Message string `json:"message"`
+		}
+		if json.Unmarshal(body, &apiErr) == nil && apiErr.Message != "" {
+			return fmt.Errorf("failed to create ephemeral shell: %s", apiErr.Message)
+		}
+		return fmt.Errorf("failed to create ephemeral shell: received status %d", resp.StatusCode)
+	}
+
+	return nil
+}
+
 func getServiceFromIDOrName(ctx context.Context, c *client.ClientWithResponses, idOrName string) (*client.Service, error) {
 	serviceRepo := service.NewRepo(c)
 
@@ -129,6 +179,22 @@ func loadDataSSH(ctx context.Context, in *SSHInput) (*exec.Cmd, error) {
 		}
 	}
 
+	if in.Ephemeral {
+		// Create the ephemeral shell pod first
+		if err := createEphemeralShell(ctx, serviceInfo.Id); err != nil {
+			return nil, tui.UserFacingError{
+				Title:   "Failed to create ephemeral shell",
+				Message: err.Error(),
+			}
+		}
+
+		// Format: ephemeral.{service-id}@{ssh-host}
+		parts := strings.SplitN(finalSSHAddress, "@", 2)
+		if len(parts) == 2 {
+			finalSSHAddress = "ephemeral." + serviceInfo.Id + "@" + parts[1]
+		}
+	}
+
 	args := []string{finalSSHAddress}
 	args = append(args, in.Args...)