Initial commit: opencode_bridge JupyterLab extension (Slices 1-3.5)

A JupyterLab extension that bridges the cell UI to a local OpenCode Serve process. The extension is a dual package: a Python server extension exposed under /opencode-bridge/*, plus a TypeScript frontend that registers per-cell toolbars. Backend (Python, tornado) - Slice 1: config + auth + OpenCode HTTP client (tornado.httpclient, no aiohttp). 4 settings in schema/plugin.json (url, user, password, request timeout). - Slice 2: handlers for /hello, /health, /providers, /edit. - Slice 2.1 (correction): SessionManager with 1 notebook = 1 session mapping, async-safe via per-path locks, 404 recovery via invalidate(). Two new endpoints: GET /sessions, DELETE /session?notebook=<path>. - 32 pytest tests pass. Frontend (TypeScript, JupyterLab 4.6) - src/types.ts: CellContext, OpenCodeRequest/Response, OpenCodeSettings. - src/context/cell_context.ts: extract CellContext from a CodeCell + its parent NotebookPanel, structured error collection. - src/api/opencode_client.ts: callOpenCodeEdit, callOpenCodeProviders. - src/components/opencode_cell_footer.ts: OpenCodeCellFooter Widget implementing ICellFooter with 3 buttons (optimize / fix / edit), resolved via this.parent instanceof CodeCell. NOT cellToolbar (does not exist in JL 4.6) and NOT Widget.findParent (removed in @lumino/widgets 2.x). - src/components/opencode_cell_factory.ts: Cell.ContentFactory subclass returning the OpenCodeCellFooter. - src/components/opencode_installer.ts: installOpenCodeEverywhere patches every notebook (existing + new) to use the custom factory. - src/index.ts: registers the factory, loads settings, fetches /providers on activation and logs the list to the console. - 23 jest tests pass (mocked JupyterLab boundary, pnpm path safe). Settings - 6 fields: 3 auth (url/user/password) + 1 timeout + 2 model selection (provider/model). Provider list is fetched at startup from /opencode-bridge/providers and printed to the browser console so users can copy values into Settings Editor. Docs - design.md: 6 sections covering architecture, UI flow, API contract, TS skeletons, session management (v0.2.1 correction), and provider/model selection (v0.2.2 addition). - CLAUDE.md: agent guidance for working in this repo. - TODO.md: remaining work for Slices 4-7 + v0.4+ backlog. CI - Gitea release workflow at .github/workflows/build.yml. - Bark notification helper (non-fatal on failure). Generated artefacts ignored: opencode_bridge/labextension/, _version.py, *.tsbuildinfo, junit.xml, test.ipynb scratch notebook.
2026-07-22 19:07:58 +08:00
commit c919c95842
61 changed files with 26642 additions and 0 deletions
@@ -0,0 +1,81 @@
+import {
+  JupyterFrontEnd,
+  JupyterFrontEndPlugin
+} from '@jupyterlab/application';
+
+import { INotebookTracker } from '@jupyterlab/notebook';
+import { ISettingRegistry } from '@jupyterlab/settingregistry';
+
+import { installOpenCodeEverywhere } from './components/opencode_installer';
+import { setOpenCodeRuntime } from './components/opencode_cell_footer';
+import { requestAPI } from './request';
+import { readOpenCodeSettings } from './types';
+import { callOpenCodeProviders } from './api/opencode_client';
+
+/**
+ * Initialization data for the opencode_bridge extension.
+ */
+const plugin: JupyterFrontEndPlugin<void> = {
+  id: 'opencode_bridge:plugin',
+  description: 'A JupyterLab extension bridging the UI to OpenCode Serve.',
+  autoStart: true,
+  optional: [ISettingRegistry],
+  requires: [INotebookTracker],
+  activate: (
+    app: JupyterFrontEnd,
+    tracker: INotebookTracker,
+    settingRegistry: ISettingRegistry | null
+  ) => {
+    console.log('JupyterLab extension opencode_bridge is activated!');
+
+    // Install the per-cell footer factory on every notebook.
+    installOpenCodeEverywhere(tracker);
+
+    // Load settings + push into toolbar module.
+    if (settingRegistry) {
+      void settingRegistry
+        .load(plugin.id)
+        .then(settings => {
+          const bridge = readOpenCodeSettings(settings.composite);
+          setOpenCodeRuntime({
+            settings: bridge,
+            serverSettings: app.serviceManager.serverSettings
+          });
+          console.log('opencode_bridge settings loaded:', bridge);
+
+          void callOpenCodeProviders(app.serviceManager.serverSettings)
+            .then(data => {
+              const lines: string[] = ['[opencode_bridge] Available OpenCode providers:'];
+              for (const p of data.providers) {
+                const models = p.models.map(m => m.id).join(', ');
+                lines.push(`  - ${p.id}: ${models || '(no models)'}`);
+              }
+              // eslint-disable-next-line no-console
+              console.log(lines.join('\n'));
+            })
+            .catch(reason => {
+              // eslint-disable-next-line no-console
+              console.warn(
+                '[opencode_bridge] Could not fetch providers (is the opencode-bridge server extension enabled and opencode serve running?):',
+                reason
+              );
+            });
+        })
+        .catch(reason => {
+          console.error('Failed to load settings for opencode_bridge.', reason);
+        });
+    }
+
+    requestAPI<unknown>('hello', app.serviceManager.serverSettings)
+      .then(data => {
+        console.log('hello endpoint:', data);
+      })
+      .catch(reason => {
+        console.error(
+          `The opencode_bridge server extension appears to be missing.\n${reason}`
+        );
+      });
+  }
+};
+
+export default plugin;