@bramblex/codex-workbench 0.1.14 → 0.1.15

package/src/ui/workbench.js

@@ -10,6 +10,7 @@ const {
   LOCAL_SOURCE,
   createSourceDirectory,
   listSourceDirectories,
+  listSourceBackends,
   loadLocalWorkbenchSessions,
   loadRemoteSourceSessions,
   loadWorkbenchSessions,
@@ -149,6 +150,27 @@ async function runWorkbench() {
     style: { border: { fg: 'red' }, fg: 'white', bg: 'black' },
   });
 
+  const backendPicker = blessed.list({
+    parent: screen,
+    label: ' Backend ',
+    top: 'center',
+    left: 'center',
+    width: 42,
+    height: 8,
+    border: 'line',
+    hidden: true,
+    mouse: true,
+    keys: true,
+    vi: false,
+    style: {
+      border: { fg: 'yellow' },
+      selected: { fg: 'black', bg: 'yellow', bold: true },
+      item: { fg: 'white' },
+    },
+  });
+
+  let backendPickerState = null;
+
   const sessionsForSource = (sourceId) => sessions.filter((session) => session.sourceId === sourceId);
 
   const buildGroups = () => {
@@ -223,12 +245,13 @@ async function runWorkbench() {
 
   const sessionLabel = (session) => {
     const flags = [
+      session.backend || '',
       session.name ? 'renamed' : '',
       session.note ? 'note' : '',
     ].filter(Boolean).join(',');
     const title = session.name || session.first || session.last || '(no prompt)';
     const flagText = flags ? `[${flags}]` : '';
-    return `${shortId(session.id)}  ${String(session.turns).padStart(2)}t  ${truncate(localTime(session.updatedAt), 18)}  ${flagText} ${truncate(title, 90)}`;
+    return `${shortId(session.id)}  ${String(session.turns).padStart(2)}t  ${truncate(localTime(session.updatedAt), 18)}  ${flagText} ${truncate(title, 88)}`;
   };
 
   const detailContent = (session) => {
@@ -238,6 +261,7 @@ async function runWorkbench() {
       title,
       '',
       `id:       ${session.id}`,
+      `backend:  ${session.backend || 'unknown'}`,
       `source:   ${session.sourceLabel || 'Local'}`,
       `cwd:      ${session.cwd}`,
       `started:  ${localTime(session.startedAt)}`,
@@ -256,7 +280,7 @@ async function runWorkbench() {
     status.style.fg = isError ? 'red' : 'white';
   };
 
-  const visibleSession = (session) => !session.archived && !session.hidden;
+  const visibleSession = (session) => !session.archived;
 
   const sortSessionList = (list) => {
     list.sort((a, b) => new Date(b.updatedAt) - new Date(a.updatedAt));
@@ -463,6 +487,31 @@ async function runWorkbench() {
     question.ask(label, (err, answer) => resolve(!err && Boolean(answer)));
   });
 
+  const askBackend = (source) => new Promise((resolve) => {
+    let backends = [];
+    try {
+      backends = listSourceBackends(source);
+    } catch (err) {
+      setMessage(`error: ${err.message}`, true);
+      render();
+      resolve(null);
+      return;
+    }
+    if (backends.length <= 1) {
+      resolve(backends[0] ? backends[0].id : null);
+      return;
+    }
+
+    backendPickerState = { backends, resolve };
+    backendPicker.clearItems();
+    backendPicker.setItems(backends.map((backend) => `${backend.id}  ${backend.label || backend.id}`));
+    backendPicker.select(0);
+    backendPicker.show();
+    backendPicker.setFront();
+    backendPicker.focus();
+    screen.render();
+  });
+
   const directoryPicker = createDirectoryPicker({
     askInput,
     focusOnClose: () => focusPanel(projectsList, 'projects'),
@@ -470,7 +519,16 @@ async function runWorkbench() {
     truncate,
   });
 
-  const promptOpen = () => prompt.visible || question.visible || directoryPicker.isOpen();
+  const closeBackendPicker = (backend = null) => {
+    if (!backendPickerState) return;
+    const { resolve } = backendPickerState;
+    backendPickerState = null;
+    backendPicker.hide();
+    focusPanel(sessionsList, 'sessions');
+    resolve(backend);
+  };
+
+  const promptOpen = () => prompt.visible || question.visible || directoryPicker.isOpen() || Boolean(backendPickerState);
 
   const leaveScreen = () => {
     closed = true;
@@ -573,17 +631,23 @@ async function runWorkbench() {
     createDirectory: (parent, name) => createSourceDirectory(source, parent, name),
   });
 
-  const runNewCodexAndReturn = (cwd, args = []) => {
+  const runNewCodexAndReturn = async (cwd, args = []) => {
     const source = currentSource();
     const resolvedCwd = source.remote ? cwd : usableCwd(cwd);
+    const backend = await askBackend(source);
+    if (!backend) {
+      setMessage('New session cancelled.');
+      render();
+      return null;
+    }
     screen.leave();
     let status = 0;
     try {
-      status = runSourceNewSession(source, resolvedCwd, args);
+      status = runSourceNewSession(source, resolvedCwd, args, backend);
     } finally {
       screen.enter();
     }
-    const label = source.remote ? `${source.label}: ${resolvedCwd}` : resolvedCwd;
+    const label = source.remote ? `${source.label}: ${resolvedCwd}` : `${resolvedCwd} (${backend})`;
     if (status === 0) refreshAfterAction(`New session finished in ${label}.`, false, resolvedCwd, source.id);
     else refreshAfterAction(`new session exited with code ${status}.`, true, resolvedCwd, source.id);
     return status;
@@ -695,6 +759,16 @@ async function runWorkbench() {
     focusPanel(projectsList, 'projects');
   });
 
+  backendPicker.key(['enter'], () => {
+    if (!backendPickerState) return;
+    const backend = backendPickerState.backends[backendPicker.selected];
+    closeBackendPicker(backend ? backend.id : null);
+  });
+
+  backendPicker.key(['escape', 'q'], () => {
+    closeBackendPicker(null);
+  });
+
   screen.on('resize', () => {
     applyLayout();
     if (directoryPicker.isOpen()) directoryPicker.applyLayout();
@@ -766,10 +840,10 @@ async function runWorkbench() {
         setMessage('New project cancelled.');
         return render();
       }
-      runNewCodexAndReturn(dir);
+      await runNewCodexAndReturn(dir);
       return;
     }
-    runNewCodexAndReturn(currentProjectCwd());
+    await runNewCodexAndReturn(currentProjectCwd());
   });
 
   screen.key(['r'], () => runAction(async (session) => {
@@ -799,11 +873,8 @@ async function runWorkbench() {
     const status = runCodexAndReturn('delete', session, ['--force'], `Deleted ${shortId(session.id)}.`);
     if (status !== 0) {
       if (session.sourceRemote) {
-        const hideSession = await askConfirm(`Remote delete failed for ${shortId(session.id)}. Hide from remote workbench instead?`);
-        if (hideSession) {
-          updateSourceMetadata(session, { hidden: true });
-          refreshAfterAction(`Hidden ${shortId(session.id)}.`);
-        }
+        setMessage(`Remote delete failed for ${shortId(session.id)}.`, true);
+        render();
         return;
       }
       const removeFile = await askConfirm(`Codex could not delete ${shortId(session.id)}. Delete its session file?`);
@@ -812,11 +883,6 @@ async function runWorkbench() {
         refreshAfterAction(`Deleted file for ${shortId(session.id)}.`);
         return;
       }
-      const hideSession = await askConfirm(`Hide ${shortId(session.id)} from workbench instead?`);
-      if (hideSession) {
-        updateSourceMetadata(session, { hidden: true });
-        refreshAfterAction(`Hidden ${shortId(session.id)}.`);
-      }
     }
   }));