feat: rename active local workspace.

Author: knightedcodemonkey

playwright/github-byot-ai.spec.ts

@@ -418,6 +418,129 @@ test('Local workspace can be renamed from Workspaces drawer', async ({ page }) =
     .toBe(renamedTitle)
 })
 
+test('Active Local non-PR workspace can be renamed from Workspaces drawer', async ({
+  page,
+}) => {
+  const activeWorkspaceId = 'active_local_workspace_rename_allowed'
+  const originalTitle = 'Active local rename original title'
+  const renamedTitle = 'Active local rename updated title'
+
+  await waitForAppReady(page)
+
+  await seedLocalWorkspaceContexts(page, [
+    {
+      id: activeWorkspaceId,
+      repo: '',
+      workspaceScope: 'local',
+      head: 'feat/active-local-rename-allowed',
+      prTitle: originalTitle,
+      prContextState: 'inactive',
+      prNumber: null,
+      tabs: [
+        {
+          id: 'component',
+          path: 'src/component.tsx',
+          language: 'tsx',
+          role: 'component',
+          content: 'export const App = () => <main>active local rename allowed</main>',
+          order: 0,
+          source: 'workspace',
+          dirty: false,
+        },
+      ],
+      activeTabId: 'component',
+    },
+  ])
+
+  await page.reload()
+  await waitForAppReady(page)
+
+  const workspacesToggle = page.getByRole('button', {
+    name: 'Workspaces',
+    exact: true,
+  })
+  await workspacesToggle.click()
+
+  const workspaceSelect = page.getByLabel('Stored workspace')
+  const renameButton = page.getByRole('button', { name: 'Rename', exact: true })
+  await expect(renameButton).toBeVisible()
+
+  await expect(workspaceSelect).toHaveValue(activeWorkspaceId)
+  await expect(renameButton).toBeEnabled()
+
+  page.once('dialog', async dialog => {
+    expect(dialog.type()).toBe('prompt')
+    expect(dialog.defaultValue()).toBe(originalTitle)
+    await dialog.accept(renamedTitle)
+  })
+
+  await renameButton.click()
+  await expect(page.locator('#workspaces-status')).toContainText('Renamed workspace.')
+
+  const records = await getAllWorkspaceRecords(page)
+  const renamedRecord = records.find(record => record?.id === activeWorkspaceId)
+
+  expect(renamedRecord).toBeTruthy()
+  expect(typeof renamedRecord?.prTitle === 'string' ? renamedRecord.prTitle : '').toBe(
+    renamedTitle,
+  )
+
+  const selectedLabelText = await page
+    .locator('#workspaces-select option:checked')
+    .textContent()
+  expect(String(selectedLabelText ?? '').trim()).toBe(renamedTitle)
+  await expect(page.locator('#workspace-context-status')).toContainText(renamedTitle)
+})
+
+test('Active Local PR-associated workspace cannot be renamed from Workspaces drawer', async ({
+  page,
+}) => {
+  const activeWorkspaceId = 'active_local_workspace_rename_blocked_pr_associated'
+
+  await waitForAppReady(page)
+
+  await seedLocalWorkspaceContexts(page, [
+    {
+      id: activeWorkspaceId,
+      repo: '',
+      workspaceScope: 'local',
+      head: 'feat/active-local-rename-blocked',
+      prTitle: 'PR-associated local workspace',
+      prContextState: 'active',
+      prNumber: 97,
+      tabs: [
+        {
+          id: 'component',
+          path: 'src/component.tsx',
+          language: 'tsx',
+          role: 'component',
+          content: 'export const App = () => <main>active local rename blocked</main>',
+          order: 0,
+          source: 'workspace',
+          dirty: false,
+        },
+      ],
+      activeTabId: 'component',
+    },
+  ])
+
+  await page.reload()
+  await waitForAppReady(page)
+
+  const workspacesToggle = page.getByRole('button', {
+    name: 'Workspaces',
+    exact: true,
+  })
+  await workspacesToggle.click()
+
+  const workspaceSelect = page.getByLabel('Stored workspace')
+  const renameButton = page.getByRole('button', { name: 'Rename', exact: true })
+  await expect(renameButton).toBeVisible()
+
+  await expect(workspaceSelect).toHaveValue(activeWorkspaceId)
+  await expect(renameButton).toBeDisabled()
+})
+
 test('chat stays usable after opening a Local workspace with PAT connected', async ({
   page,
 }) => {

src/app.js

@@ -953,6 +953,7 @@ const { syncActiveWorkspaceRepositoryScope, forkWorkspaceFromCurrentState } =
     setActiveWorkspacePersistedMetadata: ({ prTitle, head } = {}) => {
       activeWorkspacePersistedPrTitle = toNonEmptyWorkspaceText(prTitle)
       activeWorkspacePersistedHeadBranch = toNonEmptyWorkspaceText(head)
+      workspaceContextStatusController.render()
     },
   })
 
@@ -1168,6 +1169,11 @@ const githubWorkflows = createGitHubWorkflowsSetup({
     workspaceStorage,
     getActiveWorkspaceRecordId: () => activeWorkspaceRecordId,
     getActiveWorkspaceDisplayLabel,
+    setActiveWorkspacePersistedMetadata: ({ prTitle, head } = {}) => {
+      activeWorkspacePersistedPrTitle = toNonEmptyWorkspaceText(prTitle)
+      activeWorkspacePersistedHeadBranch = toNonEmptyWorkspaceText(head)
+      workspaceContextStatusController.render()
+    },
     setActiveWorkspaceRecordId,
     setActiveWorkspaceCreatedAt: value => (activeWorkspaceCreatedAt = value),
     buildWorkspaceRecordSnapshot,

src/modules/app-core/github-workflows-setup.js

@@ -24,6 +24,7 @@ const createGitHubWorkflowsSetup = ({
     setActiveWorkspaceCreatedAt: workspace.setActiveWorkspaceCreatedAt,
     buildWorkspaceRecordSnapshot: workspace.buildWorkspaceRecordSnapshot,
     getActiveWorkspaceDisplayLabel: workspace.getActiveWorkspaceDisplayLabel,
+    setActiveWorkspacePersistedMetadata: workspace.setActiveWorkspacePersistedMetadata,
     listLocalContextRecords: workspace.listLocalContextRecords,
     refreshLocalContextOptions: workspace.refreshLocalContextOptions,
     applyWorkspaceRecord: workspace.applyWorkspaceRecord,

src/modules/app-core/github-workflows.js

@@ -51,6 +51,7 @@ const initializeGitHubWorkflows = ({
   workspaceStorage,
   getActiveWorkspaceRecordId,
   getActiveWorkspaceDisplayLabel,
+  setActiveWorkspacePersistedMetadata,
   setActiveWorkspaceRecordId,
   setActiveWorkspaceCreatedAt,
   buildWorkspaceRecordSnapshot,
@@ -560,14 +561,19 @@ const initializeGitHubWorkflows = ({
           return false
         }
 
-        if (getActiveWorkspaceRecordId() === workspaceId) {
+        const hasWorkspacePrNumber =
+          typeof record.prNumber === 'number' && Number.isFinite(record.prNumber)
+        const isPrAssociatedWorkspace = hasWorkspacePrNumber
+        if (isPrAssociatedWorkspace) {
           workspacesDrawerController?.setStatus(
-            'Open a different workspace before renaming this one.',
+            'Use Pull Request controls to rename PR-associated workspaces.',
             'error',
           )
           return false
         }
 
+        const isActiveWorkspace = getActiveWorkspaceRecordId() === workspaceId
+
         const currentWorkspaceName =
           toSafeWorkspaceText(record.prTitle) ||
           toSafeWorkspaceText(record.head) ||
@@ -601,6 +607,16 @@ const initializeGitHubWorkflows = ({
           lastModified: Date.now(),
         })
 
+        if (
+          isActiveWorkspace &&
+          typeof setActiveWorkspacePersistedMetadata === 'function'
+        ) {
+          setActiveWorkspacePersistedMetadata({
+            prTitle: nextWorkspaceName,
+            head: toSafeWorkspaceText(record.head),
+          })
+        }
+
         await refreshLocalContextOptions()
         return true
       } catch {

src/modules/workspace/workspaces-drawer/drawer.js

@@ -78,12 +78,15 @@ export const createWorkspacesDrawer = ({
   }
 
   const isInactiveWithoutPrNumber = workspace => {
-    const state = toSafeText(workspace?.prContextState).toLowerCase()
+    const state = toSafeText(workspace?.prContextState).toLowerCase() || 'inactive'
     const hasPrNumber =
       typeof workspace?.prNumber === 'number' && Number.isFinite(workspace.prNumber)
     return state === 'inactive' && !hasPrNumber
   }
 
+  const hasFinitePrNumber = workspace =>
+    typeof workspace?.prNumber === 'number' && Number.isFinite(workspace.prNumber)
+
   const shouldRenderAsLocalEntry = workspace => {
     if (toSafeWorkspaceScope(workspace) === localWorkspaceScopeValue) {
       return true
@@ -157,14 +160,15 @@ export const createWorkspacesDrawer = ({
     const selectedWorkspaceScope = toSafeWorkspaceScope(selectedEntry)
     const isSelectedLocalWorkspace =
       hasSelection && selectedWorkspaceScope === localWorkspaceScopeValue
+    const isSelectedWorkspaceNonPr = hasSelection && !hasFinitePrNumber(selectedEntry)
     const isSelectedWorkspaceActive =
       hasSelection &&
       Boolean(activeWorkspaceId) &&
       normalizedSelectedId === activeWorkspaceId
     const canRenameWorkspace =
       typeof onRenameSelected === 'function' &&
       isSelectedLocalWorkspace &&
-      !isSelectedWorkspaceActive
+      isSelectedWorkspaceNonPr
     const canCreateWorkspace = typeof onCreateWorkspace === 'function'
     const canInitializeWorkspace = typeof onInitializeWorkspace === 'function'
     const hasStoredWorkspaces =