@bytespell/shella 0.1.15 → 0.1.16

package/dist/index.js

@@ -105,6 +105,8 @@ var panes = sqliteTable("panes", {
   eventStream: text("event_stream", { mode: "json" }).$type(),
   /** Renderer ID (e.g., 'builtin:terminal', 'project:custom-viewer'). Null = use default. */
   rendererId: text("renderer_id"),
+  /** Pane state - initialized from initOptions, can be updated by renderers. */
+  state: text("state", { mode: "json" }).$type(),
   createdAt: integer("created_at", { mode: "timestamp_ms" }).notNull().default(sql`(unixepoch() * 1000)`)
 });
 var uiState = sqliteTable("ui_state", {
@@ -159,6 +161,7 @@ sqlite.exec(`
     content TEXT,
     event_stream TEXT,
     renderer_id TEXT,
+    state TEXT,
     created_at INTEGER NOT NULL DEFAULT (unixepoch() * 1000)
   );
 
@@ -201,6 +204,16 @@ try {
   }
 } catch (e) {
 }
+try {
+  const tableInfo = sqlite.prepare(`PRAGMA table_info(panes)`).all();
+  const hasStateColumn = tableInfo.some((col) => col.name === "state");
+  if (!hasStateColumn) {
+    console.log("[db] Adding state column to panes...");
+    sqlite.exec(`ALTER TABLE panes ADD COLUMN state TEXT`);
+    console.log("[db] state column migration complete");
+  }
+} catch (e) {
+}
 
 // src/lib/configDiscovery.ts
 import { execSync } from "child_process";
@@ -671,18 +684,9 @@ async function listFilesFuzzy(workingDir, query, limit) {
 
 // src/amuxBridge.ts
 import { muxManager as muxManager2 } from "@bytespell/amux/streams/manager";
-import { isAcpEvent } from "@bytespell/amux/types";
 function getStreamStorageDir(streamId) {
   return path5.join(getDataDir(), "streams", streamId);
 }
-function generateTitleFromMessage(message) {
-  const cleaned = message.replace(/@[\w./~-]+/g, "").replace(/\s+/g, " ").trim();
-  const firstLine = cleaned.split("\n")[0] || cleaned;
-  if (firstLine.length <= 50) return firstLine;
-  const truncated = firstLine.slice(0, 50);
-  const lastSpace = truncated.lastIndexOf(" ");
-  return lastSpace > 20 ? truncated.slice(0, lastSpace) + "..." : truncated + "...";
-}
 function createAmuxBridge() {
   const bridge = {
     async getStreams(ids) {
@@ -713,15 +717,15 @@ function createAmuxBridge() {
       const id = params.id ?? randomUUID2();
       const now = /* @__PURE__ */ new Date();
       const config = db.select().from(streamConfigs).where(eq2(streamConfigs.id, configId)).get();
-      const streamType = config?.driverType === "pty" ? "pty" : "acp";
-      const initialTitle = streamType === "pty" ? params.directory.split("/").pop() || null : null;
+      const driver = config ? muxManager2.getDriver(config.driverType) : null;
+      const streamType = driver?.streamType ?? config?.driverType ?? "acp";
       db.insert(streams).values({
         id,
         directory: params.directory,
         streamConfigId: configId,
         streamType,
         driverState: null,
-        title: initialTitle,
+        title: null,
         model: null,
         mode: null,
         createdAt: now
@@ -740,7 +744,14 @@ function createAmuxBridge() {
       db.delete(streams).where(eq2(streams.id, id)).run();
     },
     async getStreamConfigs() {
-      return db.select().from(streamConfigs).all();
+      const configs = db.select().from(streamConfigs).all();
+      return configs.map((config) => {
+        const driver = muxManager2.getDriver(config.driverType);
+        return {
+          ...config,
+          streamType: driver?.streamType ?? config.driverType
+        };
+      });
     },
     async startStream(streamId) {
       const dbStream = db.select().from(streams).where(eq2(streams.id, streamId)).get();
@@ -748,13 +759,7 @@ function createAmuxBridge() {
       const dbConfig = db.select().from(streamConfigs).where(eq2(streamConfigs.id, dbStream.streamConfigId)).get();
       if (!dbConfig) throw new Error(`Stream config ${dbStream.streamConfigId} not found`);
       const storageDir = getStreamStorageDir(streamId);
-      const emit = (event) => {
-        if (isAcpEvent(event) && event.sessionUpdate === "session_info_update") {
-          const title = event.title;
-          if (title !== void 0) {
-            db.update(streams).set({ title }).where(eq2(streams.id, streamId)).run();
-          }
-        }
+      const emit = (_event) => {
       };
       const handle = await muxManager2.startStream({
         streamId,
@@ -794,14 +799,6 @@ function createAmuxBridge() {
       await muxManager2.stopForStream(streamId);
     },
     async input(streamId, message) {
-      const stream = db.select().from(streams).where(eq2(streams.id, streamId)).get();
-      if (!stream) throw new Error(`Stream ${streamId} not found`);
-      if (!stream.title) {
-        const title = generateTitleFromMessage(message);
-        if (title) {
-          db.update(streams).set({ title }).where(eq2(streams.id, streamId)).run();
-        }
-      }
       await muxManager2.input(streamId, message);
     },
     async cancel(streamId) {
@@ -974,12 +971,11 @@ var amux;
 function setAmuxBridge(bridge) {
   amux = bridge;
 }
-function generateTitle(existingCount, directory) {
-  const dirName = directory.split("/").pop();
-  if (dirName && dirName !== "~" && dirName !== "") {
-    return dirName;
-  }
-  return `Window ${existingCount + 1}`;
+function getNextWindowNumber() {
+  const row = db.select().from(uiState).where(eq3(uiState.key, "window_counter")).get();
+  const next = (row ? parseInt(row.value, 10) : 0) + 1;
+  db.insert(uiState).values({ key: "window_counter", value: String(next) }).onConflictDoUpdate({ target: uiState.key, set: { value: String(next) } }).run();
+  return next;
 }
 var windowsRouter = router({
   /**
@@ -1053,11 +1049,10 @@ var windowsRouter = router({
     }
     const windowId = input.id ?? randomUUID3();
     const paneId = randomUUID3();
-    const existingCount = db.select().from(windows).all().length;
     const layout = createLeaf(paneId);
     db.insert(windows).values({
       id: windowId,
-      title: input.title || generateTitle(existingCount, directory),
+      title: input.title || String(getNextWindowNumber()),
       hasCustomTitle: !!input.title,
       layout,
       activePaneId: paneId
@@ -1324,6 +1319,8 @@ var layoutRouter = router({
     streamConfigId: z2.string(),
     directory: z2.string().optional(),
     rendererId: z2.string().optional(),
+    /** All initOptions from the form - stored as pane state */
+    initOptions: z2.record(z2.string()).optional(),
     // Legacy alias
     agentConfigId: z2.string().optional()
   })).mutation(async ({ input }) => {
@@ -1343,7 +1340,8 @@ var layoutRouter = router({
     const eventStream = { type: stream.streamType, streamId: stream.id };
     db.update(panes).set({
       eventStream,
-      rendererId: input.rendererId ?? null
+      rendererId: input.rendererId ?? null,
+      state: input.initOptions ?? null
     }).where(eq4(panes.id, input.paneId)).run();
     await amux2.startStream(stream.id);
     return {
@@ -1370,6 +1368,20 @@ var layoutRouter = router({
         createdAt: stream.createdAt.toISOString()
       }
     };
+  }),
+  /**
+   * Update pane state. Renderers can call this to update state during the pane lifecycle.
+   */
+  updatePaneState: publicProcedure.input(z2.object({
+    paneId: z2.string(),
+    state: z2.record(z2.string())
+  })).mutation(async ({ input }) => {
+    const pane = db.select().from(panes).where(eq4(panes.id, input.paneId)).get();
+    if (!pane) {
+      throw new Error(`Pane ${input.paneId} not found`);
+    }
+    db.update(panes).set({ state: input.state }).where(eq4(panes.id, input.paneId)).run();
+    return { ok: true };
   })
 });