fixed pr changes for host

brevdev · Feb 25, 2024 · fa32414 · fa32414
1 parent be0bd1c
commit fa32414
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/pkg/cmd/open/open.go b/pkg/cmd/open/open.go
@@ -115,7 +115,7 @@ func runOpenCommand(t *terminal.Terminal, tstore OpenStore, wsIDOrName string, s
 
 	localIdentifier := workspace.GetLocalIdentifier()
 	if host {
-		localIdentifier += "-host"
+		localIdentifier = workspace.GetHostIdentifier()
 	}
 
 	err = res.Await()

diff --git a/pkg/cmd/shell/shell.go b/pkg/cmd/shell/shell.go
@@ -97,7 +97,7 @@ func runShellCommand(t *terminal.Terminal, sstore ShellStore, workspaceNameOrID,
 
 	localIdentifier := workspace.GetLocalIdentifier()
 	if host {
-		localIdentifier += "-host"
+		localIdentifier = workspace.GetHostIdentifier()
 	}
 
 	sshName := string(localIdentifier)

diff --git a/pkg/entity/entity.go b/pkg/entity/entity.go
@@ -450,6 +450,10 @@ func (w Workspace) GetLocalIdentifier() WorkspaceLocalID {
 	return w.createSimpleName()
 }
 
+func (w Workspace) GetHostIdentifier() WorkspaceLocalID {
+	return w.createSimpleName() + "-host"
+}
+
 func MakeIDSuffix(id string) string {
 	return id[len(id)-4:]
 }