fix: handle extra workspace name case

f0ssel · f0ssel · commit a9c112b88299 · 2025-04-23T19:27:03.000Z
diff --git a/cli/ssh.go b/cli/ssh.go
@@ -1453,6 +1453,7 @@ func collectNetworkStats(ctx context.Context, agentConn *workspacesdk.AgentConn,
 // Converts workspace name input to owner/workspace.agent format
 // Possible valid input formats:
 // workspace
+// workspace.agent
 // owner/workspace
 // owner--workspace
 // owner/workspace--agent
@@ -1467,7 +1468,11 @@ func normalizeWorkspaceInput(input string) string {
 	case 1:
 		return input // "workspace"
 	case 2:
-		return fmt.Sprintf("%s/%s", parts[0], parts[1]) // "owner/workspace"
+		// Either "owner/workspace" or "workspace.agent"
+		if strings.Contains(input, "/") || strings.Contains(input, "--") {
+			return fmt.Sprintf("%s/%s", parts[0], parts[1]) // owner/workspace
+		}
+		return fmt.Sprintf("%s.%s", parts[0], parts[1]) // workspace.agent
 	case 3:
 		return fmt.Sprintf("%s/%s.%s", parts[0], parts[1], parts[2]) // "owner/workspace.agent"
 	default:
diff --git a/cli/ssh_test.go b/cli/ssh_test.go
@@ -106,6 +106,7 @@ func TestSSH(t *testing.T) {
 
 		cases := []string{
 			"myworkspace",
+			"myworkspace.dev",
 			"myuser/myworkspace",
 			"myuser--myworkspace",
 			"myuser/myworkspace--dev",
