@yeaft/webchat-agent 1.0.533 → 1.0.535

package/local-runtime/server/db/yeaft-project-db.js

@@ -196,6 +196,23 @@ export const yeaftProjectDb = {
     })();
   },
 
+  /**
+   * Assign a newly registered fork to the source's Project. The caller owns the
+   * Session-registration transaction and replay fence; this is one insert, not
+   * moveSession's independent transaction or a user-requested Project move.
+   */
+  inheritSessionProject(userId, agentId, sourceSessionId, targetSessionId) {
+    const ownerId = requireUserId(userId);
+    const targetAgentId = requireId(agentId, 'invalid_agent_id', 'Agent id');
+    const sourceId = requireId(sourceSessionId, 'invalid_session_id', 'Source Session id');
+    const targetId = requireId(targetSessionId, 'invalid_session_id', 'Fork Session id');
+    if (sourceId === targetId) throw new YeaftProjectDbError('invalid_session_id', 'Fork must have a new identity');
+    const project = stmts.getYeaftProjectForSession.get(ownerId, targetAgentId, sourceId);
+    if (!project) return null;
+    stmts.insertYeaftProjectSessionMembership.run(ownerId, project.id, targetAgentId, targetId, Date.now());
+    return project.id;
+  },
+
   reconcileAgentSessions(userId, agentId, sessionIds) {
     const ownerId = requireUserId(userId);
     const targetAgentId = requireId(agentId, 'invalid_agent_id', 'Agent id');

package/local-runtime/server/handlers/agent-output.js

@@ -83,6 +83,30 @@ function syncYeaftSessionMetadata(agentId, agent, event) {
     }
     return { ...event, sessions: decorateYeaftSessionsWithPinned(agentId, event.sessions) };
   }
+  // A fork is a new Session on the same Agent, not a new Project. Resolve
+  // membership here (never from browser/Agent project metadata) before the
+  // successful acknowledgement can focus the new Session in the browser.
+  if (event.ok && ownerId && op === 'copy' && event.sourceSessionId && event.session?.id) {
+    try {
+      if (event.sourceSessionId === event.session.id) throw new Error('Fork must have a new Session identity');
+      transaction(() => {
+        // Replayed acknowledgements must not undo a later user Project move.
+        if (yeaftSessionDb.getForAgent(ownerId, agentId, event.session.id)) return;
+        yeaftSessionDb.upsertFromSnapshot(ownerId, agentId, event.session);
+        yeaftProjectDb.inheritSessionProject(ownerId, agentId, event.sourceSessionId, event.session.id);
+      })();
+    } catch (e) {
+      console.warn('[Server] Fork Project inheritance failed:', e?.message || e);
+      return {
+        ...event,
+        ok: false,
+        error: {
+          code: 'project_inheritance_failed',
+          message: `Fork ${event.session.id} was created, but its Project could not be saved.`,
+        },
+      };
+    }
+  }
   if (!event.ok || !ownerId || !sessionId || (op !== 'archive' && op !== 'delete')) return event;
 
   if (op === 'archive') {

package/local-runtime/version.json

@@ -1 +1 @@
-{"version":"1.0.533"}
+{"version":"1.0.535"}