agent-tower 0.4.15 → 0.4.16-beta.0

package/dist/services/__tests__/session-manager.team-run.test.js

@@ -0,0 +1,286 @@
+import { execFileSync } from 'node:child_process';
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { afterAll, beforeAll, beforeEach, describe, expect, it, vi } from 'vitest';
+import { AgentType, SessionStatus } from '../../types/index.js';
+import { EventBus } from '../../core/event-bus.js';
+const testDir = fs.mkdtempSync(path.join(os.tmpdir(), 'agent-tower-session-manager-team-run-'));
+const dbPath = path.join(testDir, 'test.db');
+process.env.AGENT_TOWER_DATABASE_URL = `file:${dbPath}`;
+const { spawnMock, spawnFollowUpMock } = vi.hoisted(() => ({
+    spawnMock: vi.fn(),
+    spawnFollowUpMock: vi.fn(),
+}));
+vi.mock('../../executors/index.js', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        getExecutor: vi.fn(() => ({
+            agentType: 'CODEX',
+            displayName: 'Mock Codex',
+            getAvailabilityInfo: vi.fn(),
+            getCapabilities: vi.fn(() => []),
+            spawn: spawnMock,
+            spawnFollowUp: spawnFollowUpMock,
+        })),
+        getExecutorByProvider: vi.fn(() => ({
+            agentType: 'CODEX',
+            displayName: 'Mock Codex',
+            getAvailabilityInfo: vi.fn(),
+            getCapabilities: vi.fn(() => []),
+            spawn: spawnMock,
+            spawnFollowUp: spawnFollowUpMock,
+        })),
+        getProviderById: vi.fn(() => ({
+            id: 'codex-default',
+            name: 'Codex',
+            agentType: 'CODEX',
+            env: {},
+            config: {},
+            isDefault: true,
+        })),
+    };
+});
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+const serverRoot = path.resolve(__dirname, '../../..');
+const schemaPath = path.join(serverRoot, 'prisma/schema.prisma');
+let prisma;
+let SessionManager;
+function createPty() {
+    return {
+        pid: 12345,
+        onData: vi.fn(() => ({ dispose: vi.fn() })),
+        onExit: vi.fn(() => ({ dispose: vi.fn() })),
+        write: vi.fn(),
+        resize: vi.fn(),
+        kill: vi.fn(),
+    };
+}
+async function createWorkspace() {
+    const project = await prisma.project.create({
+        data: {
+            name: 'Session manager TeamRun project',
+            repoPath: testDir,
+        },
+    });
+    const task = await prisma.task.create({
+        data: {
+            title: 'Session manager TeamRun task',
+            projectId: project.id,
+        },
+    });
+    const workspace = await prisma.workspace.create({
+        data: {
+            taskId: task.id,
+            branchName: 'team-shared',
+            worktreePath: testDir,
+            status: 'ACTIVE',
+        },
+    });
+    return { project, task, workspace };
+}
+describe('SessionManager TeamRun env injection', () => {
+    let manager;
+    beforeAll(async () => {
+        execFileSync('pnpm', ['exec', 'prisma', 'db', 'push', '--skip-generate', `--schema=${schemaPath}`], {
+            cwd: serverRoot,
+            env: { ...process.env, AGENT_TOWER_DATABASE_URL: `file:${dbPath}` },
+            stdio: 'pipe',
+        });
+        const utilsModule = await import('../../utils/index.js');
+        const sessionManagerModule = await import('../session-manager.js');
+        prisma = utilsModule.prisma;
+        SessionManager = sessionManagerModule.SessionManager;
+    });
+    beforeEach(async () => {
+        vi.clearAllMocks();
+        spawnMock.mockImplementation(async () => ({
+            pid: 12345,
+            pty: createPty(),
+        }));
+        spawnFollowUpMock.mockImplementation(async () => ({
+            pid: 12346,
+            pty: createPty(),
+        }));
+        manager = new SessionManager(new EventBus());
+        await prisma.executionProcess.deleteMany();
+        await prisma.agentInvocation.deleteMany();
+        await prisma.workRequest.deleteMany();
+        await prisma.roomMessage.deleteMany();
+        await prisma.teamMember.deleteMany();
+        await prisma.teamRun.deleteMany();
+        await prisma.session.deleteMany();
+        await prisma.workspace.deleteMany();
+        await prisma.task.deleteMany();
+        await prisma.project.deleteMany();
+    });
+    afterAll(async () => {
+        vi.restoreAllMocks();
+        await prisma.$disconnect();
+        fs.rmSync(testDir, { recursive: true, force: true });
+    });
+    it('injects TeamRun identity env when the session is linked to an AgentInvocation', async () => {
+        const { task, workspace } = await createWorkspace();
+        const teamRun = await prisma.teamRun.create({
+            data: {
+                taskId: task.id,
+                mode: 'AUTO',
+            },
+        });
+        const member = await prisma.teamMember.create({
+            data: {
+                teamRunId: teamRun.id,
+                presetId: null,
+                name: 'Member 1',
+                aliases: '["member-1"]',
+                providerId: 'codex-default',
+                rolePrompt: 'Role 1',
+                capabilities: '{}',
+                workspacePolicy: 'shared',
+                triggerPolicy: 'MENTION_ONLY',
+                avatar: null,
+            },
+        });
+        const request = await prisma.workRequest.create({
+            data: {
+                teamRunId: teamRun.id,
+                requesterMemberId: null,
+                requesterType: 'user',
+                targetMemberId: member.id,
+                triggerMessageId: 'message-1',
+                instruction: 'Do the work',
+                status: 'STARTED',
+            },
+        });
+        const session = await prisma.session.create({
+            data: {
+                workspaceId: workspace.id,
+                agentType: AgentType.CODEX,
+                providerId: 'codex-default',
+                prompt: 'prompt',
+                status: SessionStatus.PENDING,
+            },
+        });
+        const invocation = await prisma.agentInvocation.create({
+            data: {
+                teamRunId: teamRun.id,
+                workRequestId: request.id,
+                memberId: member.id,
+                workspaceId: workspace.id,
+                sessionId: session.id,
+                status: 'RUNNING',
+            },
+        });
+        await manager.start(session.id);
+        const spawnConfig = spawnMock.mock.calls[0][0];
+        expect(spawnConfig.env.toObject()).toMatchObject({
+            AGENT_TOWER_SESSION_ID: session.id,
+            AGENT_TOWER_INVOCATION_ID: invocation.id,
+            AGENT_TOWER_TEAM_RUN_ID: teamRun.id,
+            AGENT_TOWER_MEMBER_ID: member.id,
+        });
+        manager.destroyAll();
+    });
+    it('does not inject TeamRun env for a solo session', async () => {
+        const { workspace } = await createWorkspace();
+        const session = await prisma.session.create({
+            data: {
+                workspaceId: workspace.id,
+                agentType: AgentType.CODEX,
+                providerId: 'codex-default',
+                prompt: 'prompt',
+                status: SessionStatus.PENDING,
+            },
+        });
+        await manager.start(session.id);
+        const spawnConfig = spawnMock.mock.calls[0][0];
+        expect(spawnConfig.env.toObject()).not.toHaveProperty('AGENT_TOWER_SESSION_ID');
+        expect(spawnConfig.env.toObject()).not.toHaveProperty('AGENT_TOWER_INVOCATION_ID');
+        expect(spawnConfig.env.toObject()).not.toHaveProperty('AGENT_TOWER_TEAM_RUN_ID');
+        expect(spawnConfig.env.toObject()).not.toHaveProperty('AGENT_TOWER_MEMBER_ID');
+        manager.destroyAll();
+    });
+    it('starts a new Tower session as an executor follow-up while injecting the new invocation env', async () => {
+        const { task, workspace } = await createWorkspace();
+        const teamRun = await prisma.teamRun.create({
+            data: {
+                taskId: task.id,
+                mode: 'AUTO',
+            },
+        });
+        const member = await prisma.teamMember.create({
+            data: {
+                teamRunId: teamRun.id,
+                presetId: null,
+                name: 'Member 1',
+                aliases: '["member-1"]',
+                providerId: 'codex-default',
+                rolePrompt: 'Role 1',
+                capabilities: '{}',
+                workspacePolicy: 'shared',
+                triggerPolicy: 'MENTION_ONLY',
+                avatar: null,
+            },
+        });
+        const previousSession = await prisma.session.create({
+            data: {
+                workspaceId: workspace.id,
+                agentType: AgentType.CODEX,
+                providerId: 'codex-default',
+                prompt: 'previous prompt',
+                status: SessionStatus.COMPLETED,
+                logSnapshot: JSON.stringify({ sessionId: 'agent-native-session-1', entries: [] }),
+            },
+        });
+        const request = await prisma.workRequest.create({
+            data: {
+                teamRunId: teamRun.id,
+                requesterMemberId: null,
+                requesterType: 'user',
+                targetMemberId: member.id,
+                triggerMessageId: 'message-2',
+                instruction: 'Continue the work',
+                status: 'STARTED',
+            },
+        });
+        const nextSession = await prisma.session.create({
+            data: {
+                workspaceId: workspace.id,
+                agentType: AgentType.CODEX,
+                providerId: 'codex-default',
+                prompt: 'next prompt',
+                status: SessionStatus.PENDING,
+            },
+        });
+        const invocation = await prisma.agentInvocation.create({
+            data: {
+                teamRunId: teamRun.id,
+                workRequestId: request.id,
+                memberId: member.id,
+                workspaceId: workspace.id,
+                sessionId: nextSession.id,
+                status: 'RUNNING',
+            },
+        });
+        await manager.startFollowUp(nextSession.id, previousSession.id);
+        expect(spawnFollowUpMock).toHaveBeenCalledTimes(1);
+        expect(spawnFollowUpMock.mock.calls[0][1]).toBe('agent-native-session-1');
+        expect(spawnMock).not.toHaveBeenCalled();
+        const spawnConfig = spawnFollowUpMock.mock.calls[0][0];
+        expect(spawnConfig.prompt).toBe('next prompt');
+        expect(spawnConfig.env.toObject()).toMatchObject({
+            AGENT_TOWER_SESSION_ID: nextSession.id,
+            AGENT_TOWER_INVOCATION_ID: invocation.id,
+            AGENT_TOWER_TEAM_RUN_ID: teamRun.id,
+            AGENT_TOWER_MEMBER_ID: member.id,
+        });
+        await expect(prisma.session.findUnique({ where: { id: nextSession.id } })).resolves.toMatchObject({
+            status: SessionStatus.RUNNING,
+        });
+        manager.destroyAll();
+    });
+});
+//# sourceMappingURL=session-manager.team-run.test.js.map

package/dist/services/__tests__/session-manager.team-run.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-manager.team-run.test.js","sourceRoot":"","sources":["../../../src/services/__tests__/session-manager.team-run.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAEnF,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AAGnD,MAAM,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,uCAAuC,CAAC,CAAC,CAAC;AAChG,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;AAC7C,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,QAAQ,MAAM,EAAE,CAAC;AAExD,MAAM,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IACzD,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;IAClB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC3B,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,0BAA0B,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC3D,MAAM,MAAM,GAAG,MAAM,cAAc,EAA6C,CAAC;IACjF,OAAO;QACL,GAAG,MAAM;QACP,WAAW,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;YACxB,SAAS,EAAE,OAAO;YAClB,WAAW,EAAE,YAAY;YACzB,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC5B,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAChC,KAAK,EAAE,SAAS;YAChB,aAAa,EAAE,iBAAiB;SACjC,CAAC,CAAC;QACH,qBAAqB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;YAClC,SAAS,EAAE,OAAO;YAClB,WAAW,EAAE,YAAY;YACzB,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC5B,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAChC,KAAK,EAAE,SAAS;YAChB,aAAa,EAAE,iBAAiB;SACjC,CAAC,CAAC;QACL,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;YAC5B,EAAE,EAAE,eAAe;YACnB,IAAI,EAAE,OAAO;YACb,SAAS,EAAE,OAAO;YAClB,GAAG,EAAE,EAAE;YACP,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;KACJ,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;AAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;AACvD,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,sBAAsB,CAAC,CAAC;AAEjE,IAAI,MAAoB,CAAC;AACzB,IAAI,cAAqE,CAAC;AAE1E,SAAS,SAAS;IAChB,OAAO;QACL,GAAG,EAAE,KAAK;QACV,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAC3C,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAC3C,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;QACd,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;QACf,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;KACd,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe;IAC5B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;QAC1C,IAAI,EAAE;YACJ,IAAI,EAAE,iCAAiC;YACvC,QAAQ,EAAE,OAAO;SAClB;KACF,CAAC,CAAC;IACH,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;QACpC,IAAI,EAAE;YACJ,KAAK,EAAE,8BAA8B;YACrC,SAAS,EAAE,OAAO,CAAC,EAAE;SACtB;KACF,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC;QAC9C,IAAI,EAAE;YACJ,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,UAAU,EAAE,aAAa;YACzB,YAAY,EAAE,OAAO;YACrB,MAAM,EAAE,QAAQ;SACjB;KACF,CAAC,CAAC;IAEH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;AACtC,CAAC;AAED,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,IAAI,OAA4C,CAAC;IAEjD,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,YAAY,CACV,MAAM,EACN,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,iBAAiB,EAAE,YAAY,UAAU,EAAE,CAAC,EAC7E;YACE,GAAG,EAAE,UAAU;YACf,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,wBAAwB,EAAE,QAAQ,MAAM,EAAE,EAAE;YACnE,KAAK,EAAE,MAAM;SACd,CACF,CAAC;QAEF,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,sBAAsB,CAAC,CAAC;QACzD,MAAM,oBAAoB,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;QACnE,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;QAC5B,cAAc,GAAG,oBAAoB,CAAC,cAAc,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,SAAS,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACxC,GAAG,EAAE,KAAK;YACV,GAAG,EAAE,SAAS,EAAE;SACjB,CAAC,CAAC,CAAC;QACJ,iBAAiB,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YAChD,GAAG,EAAE,KAAK;YACV,GAAG,EAAE,SAAS,EAAE;SACjB,CAAC,CAAC,CAAC;QACJ,OAAO,GAAG,IAAI,cAAc,CAAC,IAAI,QAAQ,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,CAAC,gBAAgB,CAAC,UAAU,EAAE,CAAC;QAC3C,MAAM,MAAM,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;QAC1C,MAAM,MAAM,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;QACtC,MAAM,MAAM,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;QACtC,MAAM,MAAM,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC;QACrC,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAClC,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QAClC,MAAM,MAAM,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;QACpC,MAAM,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;QAC/B,MAAM,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;QAC3B,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;QAC7F,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,MAAM,eAAe,EAAE,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAC1C,IAAI,EAAE;gBACJ,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,IAAI,EAAE,MAAM;aACb;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;YAC5C,IAAI,EAAE;gBACJ,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,QAAQ,EAAE,IAAI;gBACd,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,cAAc;gBACvB,UAAU,EAAE,eAAe;gBAC3B,UAAU,EAAE,QAAQ;gBACpB,YAAY,EAAE,IAAI;gBAClB,eAAe,EAAE,QAAQ;gBACzB,aAAa,EAAE,cAAc;gBAC7B,MAAM,EAAE,IAAI;aACb;SACF,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC;YAC9C,IAAI,EAAE;gBACJ,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,iBAAiB,EAAE,IAAI;gBACvB,aAAa,EAAE,MAAM;gBACrB,cAAc,EAAE,MAAM,CAAC,EAAE;gBACzB,gBAAgB,EAAE,WAAW;gBAC7B,WAAW,EAAE,aAAa;gBAC1B,MAAM,EAAE,SAAS;aAClB;SACF,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAC1C,IAAI,EAAE;gBACJ,WAAW,EAAE,SAAS,CAAC,EAAE;gBACzB,SAAS,EAAE,SAAS,CAAC,KAAK;gBAC1B,UAAU,EAAE,eAAe;gBAC3B,MAAM,EAAE,QAAQ;gBAChB,MAAM,EAAE,aAAa,CAAC,OAAO;aAC9B;SACF,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC;YACrD,IAAI,EAAE;gBACJ,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,aAAa,EAAE,OAAO,CAAC,EAAE;gBACzB,QAAQ,EAAE,MAAM,CAAC,EAAE;gBACnB,WAAW,EAAE,SAAS,CAAC,EAAE;gBACzB,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,MAAM,EAAE,SAAS;aAClB;SACF,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEhC,MAAM,WAAW,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAwB,CAAC;QACvE,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,aAAa,CAAC;YAC/C,sBAAsB,EAAE,OAAO,CAAC,EAAE;YAClC,yBAAyB,EAAE,UAAU,CAAC,EAAE;YACxC,uBAAuB,EAAE,OAAO,CAAC,EAAE;YACnC,qBAAqB,EAAE,MAAM,CAAC,EAAE;SACjC,CAAC,CAAC;QACH,OAAO,CAAC,UAAU,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,eAAe,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAC1C,IAAI,EAAE;gBACJ,WAAW,EAAE,SAAS,CAAC,EAAE;gBACzB,SAAS,EAAE,SAAS,CAAC,KAAK;gBAC1B,UAAU,EAAE,eAAe;gBAC3B,MAAM,EAAE,QAAQ;gBAChB,MAAM,EAAE,aAAa,CAAC,OAAO;aAC9B;SACF,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAEhC,MAAM,WAAW,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAwB,CAAC;QACvE,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,wBAAwB,CAAC,CAAC;QAChF,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,2BAA2B,CAAC,CAAC;QACnF,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,yBAAyB,CAAC,CAAC;QACjF,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,uBAAuB,CAAC,CAAC;QAC/E,OAAO,CAAC,UAAU,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4FAA4F,EAAE,KAAK,IAAI,EAAE;QAC1G,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,GAAG,MAAM,eAAe,EAAE,CAAC;QACpD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAC1C,IAAI,EAAE;gBACJ,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,IAAI,EAAE,MAAM;aACb;SACF,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC;YAC5C,IAAI,EAAE;gBACJ,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,QAAQ,EAAE,IAAI;gBACd,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,cAAc;gBACvB,UAAU,EAAE,eAAe;gBAC3B,UAAU,EAAE,QAAQ;gBACpB,YAAY,EAAE,IAAI;gBAClB,eAAe,EAAE,QAAQ;gBACzB,aAAa,EAAE,cAAc;gBAC7B,MAAM,EAAE,IAAI;aACb;SACF,CAAC,CAAC;QACH,MAAM,eAAe,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAClD,IAAI,EAAE;gBACJ,WAAW,EAAE,SAAS,CAAC,EAAE;gBACzB,SAAS,EAAE,SAAS,CAAC,KAAK;gBAC1B,UAAU,EAAE,eAAe;gBAC3B,MAAM,EAAE,iBAAiB;gBACzB,MAAM,EAAE,aAAa,CAAC,SAAS;gBAC/B,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,wBAAwB,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;aAClF;SACF,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC;YAC9C,IAAI,EAAE;gBACJ,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,iBAAiB,EAAE,IAAI;gBACvB,aAAa,EAAE,MAAM;gBACrB,cAAc,EAAE,MAAM,CAAC,EAAE;gBACzB,gBAAgB,EAAE,WAAW;gBAC7B,WAAW,EAAE,mBAAmB;gBAChC,MAAM,EAAE,SAAS;aAClB;SACF,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAC9C,IAAI,EAAE;gBACJ,WAAW,EAAE,SAAS,CAAC,EAAE;gBACzB,SAAS,EAAE,SAAS,CAAC,KAAK;gBAC1B,UAAU,EAAE,eAAe;gBAC3B,MAAM,EAAE,aAAa;gBACrB,MAAM,EAAE,aAAa,CAAC,OAAO;aAC9B;SACF,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,MAAM,CAAC;YACrD,IAAI,EAAE;gBACJ,SAAS,EAAE,OAAO,CAAC,EAAE;gBACrB,aAAa,EAAE,OAAO,CAAC,EAAE;gBACzB,QAAQ,EAAE,MAAM,CAAC,EAAE;gBACnB,WAAW,EAAE,SAAS,CAAC,EAAE;gBACzB,SAAS,EAAE,WAAW,CAAC,EAAE;gBACzB,MAAM,EAAE,SAAS;aAClB;SACF,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,EAAE,eAAe,CAAC,EAAE,CAAC,CAAC;QAEhE,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAC3E,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzC,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC,CAAwB,CAAC;QAC/E,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC/C,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,aAAa,CAAC;YAC/C,sBAAsB,EAAE,WAAW,CAAC,EAAE;YACtC,yBAAyB,EAAE,UAAU,CAAC,EAAE;YACxC,uBAAuB,EAAE,OAAO,CAAC,EAAE;YACnC,qBAAqB,EAAE,MAAM,CAAC,EAAE;SACjC,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,WAAW,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC;YAChG,MAAM,EAAE,aAAa,CAAC,OAAO;SAC9B,CAAC,CAAC;QACH,OAAO,CAAC,UAAU,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/__tests__/team-lock.service.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=team-lock.service.test.d.ts.map

package/dist/services/__tests__/team-lock.service.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"team-lock.service.test.d.ts","sourceRoot":"","sources":["../../../src/services/__tests__/team-lock.service.test.ts"],"names":[],"mappings":""}

package/dist/services/__tests__/team-lock.service.test.js

@@ -0,0 +1,81 @@
+import { describe, expect, it } from 'vitest';
+import { TeamLockService } from '../team-lock.service.js';
+const readOnlyCapabilities = {
+    readRoom: true,
+    postRoomMessage: true,
+    mentionMembers: true,
+    stopMemberWork: false,
+    markReadyForReview: false,
+    readFiles: true,
+    writeFiles: false,
+    runCommands: false,
+    readDiff: true,
+    mergeWorkspace: false,
+};
+function lockRequest(capabilities, workspacePolicy = 'shared') {
+    return {
+        teamRunId: 'team-run-1',
+        memberId: 'member-1',
+        workspaceId: 'workspace-1',
+        projectId: 'project-1',
+        workspacePolicy,
+        capabilities: { ...readOnlyCapabilities, ...capabilities },
+    };
+}
+describe('TeamLockService', () => {
+    it('does not require locks for read-only capabilities', () => {
+        const service = new TeamLockService();
+        expect(service.getRequiredLocks(lockRequest({}))).toEqual([]);
+    });
+    it('requires a workspace write lock for shared writeFiles members', () => {
+        const service = new TeamLockService();
+        expect(service.getRequiredLocks(lockRequest({ writeFiles: true }))).toEqual([
+            'workspace:workspace-1:write',
+        ]);
+    });
+    it('requires a workspace command lock for shared runCommands members', () => {
+        const service = new TeamLockService();
+        expect(service.getRequiredLocks(lockRequest({ runCommands: true }))).toEqual([
+            'workspace:workspace-1:command',
+        ]);
+    });
+    it('requires a project merge lock for mergeWorkspace members', () => {
+        const service = new TeamLockService();
+        expect(service.getRequiredLocks(lockRequest({ mergeWorkspace: true }))).toEqual([
+            'project:project-1:merge',
+        ]);
+    });
+    it('keeps project merge locks when workspacePolicy is none', () => {
+        const service = new TeamLockService();
+        expect(service.getRequiredLocks(lockRequest({ mergeWorkspace: true }, 'none'))).toEqual([
+            'project:project-1:merge',
+        ]);
+    });
+    it('does not require workspace write or command locks when workspacePolicy is none', () => {
+        const service = new TeamLockService();
+        expect(service.getRequiredLocks(lockRequest({
+            writeFiles: true,
+            runCommands: true,
+        }, 'none'))).toEqual([]);
+    });
+    it('prevents different owners from acquiring the same lock until release', () => {
+        const service = new TeamLockService();
+        expect(service.acquire('owner-1', ['workspace:workspace-1:write'])).toBe(true);
+        expect(service.acquire('owner-2', ['workspace:workspace-1:write'])).toBe(false);
+        service.releaseByOwner('owner-1');
+        expect(service.acquire('owner-2', ['workspace:workspace-1:write'])).toBe(true);
+    });
+    it('does not partially acquire multiple locks when any requested lock is held', () => {
+        const service = new TeamLockService();
+        expect(service.acquire('owner-1', ['workspace:workspace-1:write'])).toBe(true);
+        expect(service.acquire('owner-2', [
+            'workspace:workspace-1:write',
+            'workspace:workspace-1:command',
+        ])).toBe(false);
+        expect(service.listLocks()).toEqual([
+            { key: 'workspace:workspace-1:write', ownerId: 'owner-1' },
+        ]);
+        expect(service.canAcquire(['workspace:workspace-1:command'])).toBe(true);
+    });
+});
+//# sourceMappingURL=team-lock.service.test.js.map

package/dist/services/__tests__/team-lock.service.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"team-lock.service.test.js","sourceRoot":"","sources":["../../../src/services/__tests__/team-lock.service.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAE9C,OAAO,EAAE,eAAe,EAAoB,MAAM,yBAAyB,CAAC;AAE5E,MAAM,oBAAoB,GAA2B;IACnD,QAAQ,EAAE,IAAI;IACd,eAAe,EAAE,IAAI;IACrB,cAAc,EAAE,IAAI;IACpB,cAAc,EAAE,KAAK;IACrB,kBAAkB,EAAE,KAAK;IACzB,SAAS,EAAE,IAAI;IACf,UAAU,EAAE,KAAK;IACjB,WAAW,EAAE,KAAK;IAClB,QAAQ,EAAE,IAAI;IACd,cAAc,EAAE,KAAK;CACtB,CAAC;AAEF,SAAS,WAAW,CAClB,YAA6C,EAC7C,kBAAmC,QAAQ;IAE3C,OAAO;QACL,SAAS,EAAE,YAAY;QACvB,QAAQ,EAAE,UAAU;QACpB,WAAW,EAAE,aAAa;QAC1B,SAAS,EAAE,WAAW;QACtB,eAAe;QACf,YAAY,EAAE,EAAE,GAAG,oBAAoB,EAAE,GAAG,YAAY,EAAE;KAC3D,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1E,6BAA6B;SAC9B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3E,+BAA+B;SAChC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAClE,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC9E,yBAAyB;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACtF,yBAAyB;SAC1B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACxF,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,WAAW,CAAC;YAC1C,UAAU,EAAE,IAAI;YAChB,WAAW,EAAE,IAAI;SAClB,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC9E,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/E,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEhF,OAAO,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAElC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;QACnF,MAAM,OAAO,GAAG,IAAI,eAAe,EAAE,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/E,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE;YAChC,6BAA6B;YAC7B,+BAA+B;SAChC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEhB,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,OAAO,CAAC;YAClC,EAAE,GAAG,EAAE,6BAA6B,EAAE,OAAO,EAAE,SAAS,EAAE;SAC3D,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,+BAA+B,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/services/__tests__/team-reconciler.service.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=team-reconciler.service.test.d.ts.map

package/dist/services/__tests__/team-reconciler.service.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"team-reconciler.service.test.d.ts","sourceRoot":"","sources":["../../../src/services/__tests__/team-reconciler.service.test.ts"],"names":[],"mappings":""}