reviewdeck 0.5.5 → 0.5.7

package/dist/reviewdeck.mjs

@@ -684,6 +684,7 @@ var init_memory = __esm({
         let all = [...this.sessions.values()];
         if (filter?.userId) all = all.filter((s) => s.userId === filter.userId);
         if (filter?.agentId) all = all.filter((s) => s.agentId === filter.agentId);
+        if (filter?.agentSessionId) all = all.filter((s) => s.agentSessionId === filter.agentSessionId);
         return all;
       }
       // --- Auth: Users ---
@@ -2963,23 +2964,30 @@ var init_postgres = __esm({
       async init() {
         await this.sql`
       CREATE TABLE IF NOT EXISTS sessions (
-        id            TEXT PRIMARY KEY,
-        review_token  TEXT NOT NULL UNIQUE,
-        status        TEXT NOT NULL DEFAULT 'reviewing',
-        user_id       TEXT NOT NULL,
-        agent_id      TEXT,
-        split_meta    JSONB NOT NULL,
-        sub_patches   JSONB NOT NULL,
-        submission    JSONB,
-        created_at    TIMESTAMPTZ NOT NULL DEFAULT now(),
-        updated_at    TIMESTAMPTZ NOT NULL DEFAULT now()
+        id                TEXT PRIMARY KEY,
+        review_token      TEXT NOT NULL UNIQUE,
+        status            TEXT NOT NULL DEFAULT 'reviewing',
+        user_id           TEXT NOT NULL,
+        agent_id          TEXT,
+        agent_session_id  TEXT,
+        split_meta        JSONB NOT NULL,
+        sub_patches       JSONB NOT NULL,
+        submission        JSONB,
+        created_at        TIMESTAMPTZ NOT NULL DEFAULT now(),
+        updated_at        TIMESTAMPTZ NOT NULL DEFAULT now()
       )
+    `;
+        await this.sql`
+      ALTER TABLE sessions ADD COLUMN IF NOT EXISTS agent_session_id TEXT
     `;
         await this.sql`
       CREATE INDEX IF NOT EXISTS idx_sessions_review_token ON sessions (review_token)
     `;
         await this.sql`
       CREATE INDEX IF NOT EXISTS idx_sessions_user_id ON sessions (user_id)
+    `;
+        await this.sql`
+      CREATE INDEX IF NOT EXISTS idx_sessions_agent_session_id ON sessions (agent_session_id)
     `;
         await this.sql`
       CREATE TABLE IF NOT EXISTS uploads (
@@ -3073,13 +3081,14 @@ var init_postgres = __esm({
       }
       async saveSession(session) {
         await this.sql`
-      INSERT INTO sessions (id, review_token, status, user_id, agent_id, split_meta, sub_patches, submission, created_at, updated_at)
+      INSERT INTO sessions (id, review_token, status, user_id, agent_id, agent_session_id, split_meta, sub_patches, submission, created_at, updated_at)
       VALUES (
         ${session.id},
         ${session.reviewToken},
         ${session.status},
         ${session.userId},
         ${session.agentId ?? null},
+        ${session.agentSessionId ?? null},
         ${this.sql.json(session.splitMeta)},
         ${this.sql.json(session.subPatches)},
         ${session.submission ? this.sql.json(session.submission) : null},
@@ -3090,7 +3099,7 @@ var init_postgres = __esm({
       }
       async getSession(id) {
         const rows = await this.sql`
-      SELECT id, review_token, status, user_id, agent_id, split_meta, sub_patches, submission, created_at, updated_at
+      SELECT id, review_token, status, user_id, agent_id, agent_session_id, split_meta, sub_patches, submission, created_at, updated_at
       FROM sessions WHERE id = ${id}
     `;
         if (rows.length === 0) return void 0;
@@ -3098,7 +3107,7 @@ var init_postgres = __esm({
       }
       async getSessionByToken(reviewToken) {
         const rows = await this.sql`
-      SELECT id, review_token, status, user_id, agent_id, split_meta, sub_patches, submission, created_at, updated_at
+      SELECT id, review_token, status, user_id, agent_id, agent_session_id, split_meta, sub_patches, submission, created_at, updated_at
       FROM sessions WHERE review_token = ${reviewToken}
     `;
         if (rows.length === 0) return void 0;
@@ -3116,23 +3125,19 @@ var init_postgres = __esm({
         return this.getSession(id);
       }
       async listSessions(filter) {
-        let rows;
-        if (filter?.userId && filter?.agentId) {
-          rows = await this.sql`
-        SELECT id, review_token, status, user_id, agent_id, split_meta, sub_patches, submission, created_at, updated_at
-        FROM sessions WHERE user_id = ${filter.userId} AND agent_id = ${filter.agentId} ORDER BY created_at DESC
-      `;
-        } else if (filter?.userId) {
-          rows = await this.sql`
-        SELECT id, review_token, status, user_id, agent_id, split_meta, sub_patches, submission, created_at, updated_at
-        FROM sessions WHERE user_id = ${filter.userId} ORDER BY created_at DESC
-      `;
-        } else {
-          rows = await this.sql`
-        SELECT id, review_token, status, user_id, agent_id, split_meta, sub_patches, submission, created_at, updated_at
-        FROM sessions ORDER BY created_at DESC
-      `;
-        }
+        const conds = [];
+        if (filter?.userId) conds.push(this.sql`user_id = ${filter.userId}`);
+        if (filter?.agentId) conds.push(this.sql`agent_id = ${filter.agentId}`);
+        if (filter?.agentSessionId) conds.push(this.sql`agent_session_id = ${filter.agentSessionId}`);
+        const rows = conds.length ? await this.sql`
+          SELECT id, review_token, status, user_id, agent_id, agent_session_id, split_meta, sub_patches, submission, created_at, updated_at
+          FROM sessions
+          WHERE ${conds.reduce((acc, c) => this.sql`${acc} AND ${c}`)}
+          ORDER BY created_at DESC
+        ` : await this.sql`
+          SELECT id, review_token, status, user_id, agent_id, agent_session_id, split_meta, sub_patches, submission, created_at, updated_at
+          FROM sessions ORDER BY created_at DESC
+        `;
         return rows.map((row) => this.rowToSession(row));
       }
       // --- Auth: Users ---
@@ -3314,6 +3319,7 @@ var init_postgres = __esm({
           status: row.status,
           userId: row.user_id,
           agentId: row.agent_id ?? void 0,
+          agentSessionId: row.agent_session_id ?? void 0,
           splitMeta: row.split_meta,
           subPatches: row.sub_patches,
           submission: row.submission ?? null,
@@ -21418,10 +21424,12 @@ function createReviewRoutes(storage, baseUrl) {
     const {
       fileId,
       splitMeta,
-      agentId: bodyAgentId
+      agentId: bodyAgentId,
+      agentSessionId: bodyAgentSessionId
     } = await c.req.json();
     const userId = c.get("userId") ?? "";
     const agentId = bodyAgentId ?? c.req.header("x-agent-id");
+    const agentSessionId = bodyAgentSessionId ?? c.req.header("x-session-id");
     const upload = await storage.getUpload(fileId);
     if (!upload) return c.json({ error: "Upload not found" }, 404);
     const patches = parsePatch(upload.diff);
@@ -21449,6 +21457,7 @@ function createReviewRoutes(storage, baseUrl) {
       status: "reviewing",
       userId,
       agentId,
+      agentSessionId,
       splitMeta,
       subPatches,
       submission: null,
@@ -21476,6 +21485,7 @@ function createReviewRoutes(storage, baseUrl) {
       status: session.status,
       userId: session.userId,
       agentId: session.agentId,
+      agentSessionId: session.agentSessionId,
       patchCount: session.subPatches.length,
       reviewUrl: `${baseUrl}/review/${session.id}?token=${session.reviewToken}`,
       submission: session.submission,
@@ -21486,8 +21496,13 @@ function createReviewRoutes(storage, baseUrl) {
   app.get("/sessions", async (c) => {
     const userId = c.get("userId") ?? "";
     const agentId = c.req.query("agentId");
+    const agentSessionId = c.req.query("agentSessionId");
     const all = c.req.query("all") === "true";
-    const filter = all ? { userId } : { userId, agentId: agentId ?? void 0 };
+    const filter = all ? { userId } : {
+      userId,
+      agentId: agentId ?? void 0,
+      agentSessionId: agentSessionId ?? void 0
+    };
     const sessions = await storage.listSessions(userId ? filter : void 0);
     return c.json(
       sessions.map((s) => ({
@@ -21495,6 +21510,7 @@ function createReviewRoutes(storage, baseUrl) {
         status: s.status,
         userId: s.userId,
         agentId: s.agentId,
+        agentSessionId: s.agentSessionId,
         reviewUrl: `${baseUrl}/review/${s.id}?token=${s.reviewToken}`,
         patchCount: s.subPatches.length,
         createdAt: s.createdAt,
@@ -38949,7 +38965,7 @@ function textResult(data, isError = false) {
   };
   return result;
 }
-function registerTools(server, storage, baseUrl, userId, headerAgentId) {
+function registerTools(server, storage, baseUrl, userId, headerAgentId, headerAgentSessionId) {
   server.tool(
     "create_review",
     "Validate split metadata, generate sub-patches, and create a persistent review session. Returns sessionId and reviewUrl. Upload diffs first via POST /api/uploads.",
@@ -39005,6 +39021,7 @@ function registerTools(server, storage, baseUrl, userId, headerAgentId) {
           status: "reviewing",
           userId,
           agentId: params.agentId ?? headerAgentId,
+          agentSessionId: headerAgentSessionId,
           splitMeta,
           subPatches,
           submission: null,
@@ -39083,11 +39100,12 @@ function createMcpRouter(storage, baseUrl) {
   router.all("/", async (c) => {
     const userId = c.get("userId") ?? "";
     const headerAgentId = c.req.header("x-agent-id");
+    const headerAgentSessionId = c.req.header("x-session-id");
     console.error(
-      `[mcp] ${c.req.method} ${c.req.path} userId=${userId} agentId=${headerAgentId ?? "-"}`
+      `[mcp] ${c.req.method} ${c.req.path} userId=${userId} agentId=${headerAgentId ?? "-"} agentSessionId=${headerAgentSessionId ?? "-"}`
     );
     const server = new McpServer({ name: "reviewdeck", version: "0.5.3" });
-    registerTools(server, storage, baseUrl, userId, headerAgentId);
+    registerTools(server, storage, baseUrl, userId, headerAgentId, headerAgentSessionId);
     const transport = new StreamableHTTPTransport({ sessionIdGenerator: void 0 });
     await server.connect(transport);
     return transport.handleRequest(c);