mirror of
https://github.com/davidkaya/aryx.git
synced 2026-08-06 19:58:43 +02:00
fix: isolate scratchpad session working directories
Give each scratchpad session its own working directory, migrate existing scratchpad sessions on workspace load, and route sidecar turns through the session-specific cwd. Add regression coverage for create, duplicate, delete, and migration flows, and document the new persistence model. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
@@ -0,0 +1,196 @@
|
||||
import { afterEach, beforeEach, describe, expect, mock, test } from 'bun:test';
|
||||
import { access, mkdir, rm } from 'node:fs/promises';
|
||||
import { join } from 'node:path';
|
||||
|
||||
import type { PatternDefinition } from '@shared/domain/pattern';
|
||||
import { createScratchpadProject } from '@shared/domain/project';
|
||||
import type { SessionRecord } from '@shared/domain/session';
|
||||
import { createWorkspaceSeed, type WorkspaceState } from '@shared/domain/workspace';
|
||||
|
||||
const TIMESTAMP = '2026-03-28T00:00:00.000Z';
|
||||
const USER_DATA_PATH = 'C:\\workspace\\personal\\repositories\\aryx\\tests\\fixtures';
|
||||
|
||||
mock.module('electron', () => {
|
||||
const electronMock = {
|
||||
app: {
|
||||
isPackaged: false,
|
||||
getAppPath: () => 'C:\\workspace\\personal\\repositories\\aryx',
|
||||
getPath: () => USER_DATA_PATH,
|
||||
},
|
||||
dialog: {
|
||||
showOpenDialog: async () => ({ canceled: true, filePaths: [] }),
|
||||
},
|
||||
shell: {
|
||||
openPath: async () => '',
|
||||
},
|
||||
};
|
||||
|
||||
return {
|
||||
...electronMock,
|
||||
default: electronMock,
|
||||
};
|
||||
});
|
||||
|
||||
mock.module('keytar', () => ({
|
||||
default: {
|
||||
getPassword: async () => null,
|
||||
setPassword: async () => undefined,
|
||||
deletePassword: async () => false,
|
||||
},
|
||||
}));
|
||||
|
||||
const { AryxAppService } = await import('@main/AryxAppService');
|
||||
|
||||
async function pathExists(path: string): Promise<boolean> {
|
||||
try {
|
||||
await access(path);
|
||||
return true;
|
||||
} catch {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
function requireSinglePattern(workspace: WorkspaceState): PatternDefinition {
|
||||
const pattern = workspace.patterns.find((candidate) => candidate.mode === 'single');
|
||||
if (!pattern) {
|
||||
throw new Error('Expected the workspace seed to include a single-agent pattern.');
|
||||
}
|
||||
|
||||
return pattern;
|
||||
}
|
||||
|
||||
function createScratchpadSession(patternId: string, overrides?: Partial<SessionRecord>): SessionRecord {
|
||||
return {
|
||||
id: 'session-scratchpad',
|
||||
projectId: 'project-scratchpad',
|
||||
patternId,
|
||||
title: 'Scratchpad',
|
||||
createdAt: TIMESTAMP,
|
||||
updatedAt: TIMESTAMP,
|
||||
status: 'idle',
|
||||
messages: [],
|
||||
runs: [],
|
||||
...overrides,
|
||||
};
|
||||
}
|
||||
|
||||
function createService(
|
||||
workspace: WorkspaceState,
|
||||
options?: {
|
||||
onDeleteSession?: (sessionId: string) => Promise<void>;
|
||||
},
|
||||
): InstanceType<typeof AryxAppService> {
|
||||
const service = new AryxAppService();
|
||||
const internals = service as unknown as Record<string, unknown>;
|
||||
internals.loadWorkspace = async () => workspace;
|
||||
internals.persistAndBroadcast = async (nextWorkspace: WorkspaceState) => nextWorkspace;
|
||||
internals.loadAvailableModelCatalog = async () => [];
|
||||
|
||||
(
|
||||
service as unknown as {
|
||||
sidecar: {
|
||||
deleteSession: (sessionId: string) => Promise<void>;
|
||||
};
|
||||
}
|
||||
).sidecar = {
|
||||
deleteSession: async (sessionId: string) => {
|
||||
await options?.onDeleteSession?.(sessionId);
|
||||
},
|
||||
};
|
||||
|
||||
return service;
|
||||
}
|
||||
|
||||
beforeEach(async () => {
|
||||
await rm(join(USER_DATA_PATH, 'scratchpad'), { recursive: true, force: true });
|
||||
});
|
||||
|
||||
afterEach(async () => {
|
||||
await rm(join(USER_DATA_PATH, 'scratchpad'), { recursive: true, force: true });
|
||||
});
|
||||
|
||||
describe('AryxAppService scratchpad directories', () => {
|
||||
test('creates a dedicated directory for new scratchpad sessions', async () => {
|
||||
const workspace = createWorkspaceSeed();
|
||||
const pattern = requireSinglePattern(workspace);
|
||||
const scratchpadProject = createScratchpadProject(join(USER_DATA_PATH, 'scratchpad'), TIMESTAMP);
|
||||
workspace.projects = [scratchpadProject];
|
||||
workspace.selectedProjectId = scratchpadProject.id;
|
||||
workspace.selectedPatternId = pattern.id;
|
||||
|
||||
const service = createService(workspace);
|
||||
|
||||
const result = await service.createSession(scratchpadProject.id, pattern.id);
|
||||
const session = result.sessions[0];
|
||||
if (!session) {
|
||||
throw new Error('Expected createSession to prepend a scratchpad session.');
|
||||
}
|
||||
|
||||
expect(session.cwd).toBe(join(USER_DATA_PATH, 'scratchpad', session.id));
|
||||
expect(await pathExists(session.cwd!)).toBe(true);
|
||||
});
|
||||
|
||||
test('creates a new directory when duplicating a scratchpad session', async () => {
|
||||
const workspace = createWorkspaceSeed();
|
||||
const pattern = requireSinglePattern(workspace);
|
||||
const scratchpadProject = createScratchpadProject(join(USER_DATA_PATH, 'scratchpad'), TIMESTAMP);
|
||||
const originalDirectory = join(USER_DATA_PATH, 'scratchpad', 'session-original');
|
||||
const originalSession = createScratchpadSession(pattern.id, {
|
||||
id: 'session-original',
|
||||
cwd: originalDirectory,
|
||||
messages: [
|
||||
{
|
||||
id: 'msg-1',
|
||||
role: 'user',
|
||||
authorName: 'You',
|
||||
content: 'Draft a plan.',
|
||||
createdAt: TIMESTAMP,
|
||||
},
|
||||
],
|
||||
});
|
||||
|
||||
workspace.projects = [scratchpadProject];
|
||||
workspace.sessions = [originalSession];
|
||||
workspace.selectedProjectId = scratchpadProject.id;
|
||||
workspace.selectedPatternId = pattern.id;
|
||||
workspace.selectedSessionId = originalSession.id;
|
||||
|
||||
const service = createService(workspace);
|
||||
|
||||
const result = await service.duplicateSession(originalSession.id);
|
||||
const duplicate = result.sessions[0];
|
||||
if (!duplicate) {
|
||||
throw new Error('Expected duplicateSession to prepend the duplicate.');
|
||||
}
|
||||
|
||||
expect(duplicate.id).not.toBe(originalSession.id);
|
||||
expect(duplicate.cwd).toBe(join(USER_DATA_PATH, 'scratchpad', duplicate.id));
|
||||
expect(duplicate.cwd).not.toBe(originalSession.cwd);
|
||||
expect(await pathExists(duplicate.cwd!)).toBe(true);
|
||||
});
|
||||
|
||||
test('removes the scratchpad directory when deleting a session', async () => {
|
||||
const workspace = createWorkspaceSeed();
|
||||
const pattern = requireSinglePattern(workspace);
|
||||
const scratchpadProject = createScratchpadProject(join(USER_DATA_PATH, 'scratchpad'), TIMESTAMP);
|
||||
const sessionDirectory = join(USER_DATA_PATH, 'scratchpad', 'session-scratchpad');
|
||||
const session = createScratchpadSession(pattern.id, {
|
||||
cwd: sessionDirectory,
|
||||
});
|
||||
|
||||
workspace.projects = [scratchpadProject];
|
||||
workspace.sessions = [session];
|
||||
workspace.selectedProjectId = scratchpadProject.id;
|
||||
workspace.selectedPatternId = pattern.id;
|
||||
workspace.selectedSessionId = session.id;
|
||||
|
||||
const service = createService(workspace);
|
||||
await mkdir(sessionDirectory, { recursive: true });
|
||||
expect(await pathExists(sessionDirectory)).toBe(true);
|
||||
|
||||
const result = await service.deleteSession(session.id);
|
||||
|
||||
expect(result.sessions).toHaveLength(0);
|
||||
expect(await pathExists(sessionDirectory)).toBe(false);
|
||||
});
|
||||
});
|
||||
@@ -8,6 +8,7 @@ import { createWorkspaceSeed, type WorkspaceState } from '@shared/domain/workspa
|
||||
|
||||
const TIMESTAMP = '2026-03-25T00:00:00.000Z';
|
||||
const SCRATCHPAD_PATH = 'C:\\workspace\\personal\\repositories\\aryx\\scratchpad';
|
||||
const SCRATCHPAD_SESSION_PATH = `${SCRATCHPAD_PATH}\\session-scratchpad`;
|
||||
|
||||
mock.module('electron', () => {
|
||||
const electronMock = {
|
||||
@@ -60,6 +61,7 @@ function createWorkspaceFixture(): {
|
||||
createdAt: TIMESTAMP,
|
||||
updatedAt: TIMESTAMP,
|
||||
status: 'idle',
|
||||
cwd: SCRATCHPAD_SESSION_PATH,
|
||||
messages: [],
|
||||
runs: [],
|
||||
};
|
||||
@@ -186,7 +188,7 @@ describe('AryxAppService scratchpad tooling support', () => {
|
||||
|
||||
expect(command).toMatchObject({
|
||||
sessionId: session.id,
|
||||
projectPath: SCRATCHPAD_PATH,
|
||||
projectPath: SCRATCHPAD_SESSION_PATH,
|
||||
workspaceKind: 'scratchpad',
|
||||
tooling: {
|
||||
mcpServers: [
|
||||
|
||||
@@ -0,0 +1,90 @@
|
||||
import { afterEach, beforeEach, describe, expect, mock, test } from 'bun:test';
|
||||
import { access, mkdir, readFile, rm, writeFile } from 'node:fs/promises';
|
||||
import { join } from 'node:path';
|
||||
|
||||
import { createScratchpadProject } from '@shared/domain/project';
|
||||
import type { SessionRecord } from '@shared/domain/session';
|
||||
import { createWorkspaceSeed, type WorkspaceState } from '@shared/domain/workspace';
|
||||
|
||||
const TIMESTAMP = '2026-03-28T00:00:00.000Z';
|
||||
const USER_DATA_PATH = 'C:\\workspace\\personal\\repositories\\aryx\\tests\\fixtures';
|
||||
|
||||
mock.module('electron', () => {
|
||||
const electronMock = {
|
||||
app: {
|
||||
getPath: () => USER_DATA_PATH,
|
||||
},
|
||||
};
|
||||
|
||||
return {
|
||||
...electronMock,
|
||||
default: electronMock,
|
||||
};
|
||||
});
|
||||
|
||||
const { WorkspaceRepository } = await import('@main/persistence/workspaceRepository');
|
||||
|
||||
function createStoredWorkspace(): WorkspaceState {
|
||||
const workspace = createWorkspaceSeed();
|
||||
const scratchpadProject = createScratchpadProject('C:\\legacy\\scratchpad', TIMESTAMP);
|
||||
const patternId = workspace.patterns[0]?.id;
|
||||
if (!patternId) {
|
||||
throw new Error('Expected workspace seed to include at least one pattern.');
|
||||
}
|
||||
|
||||
const session: SessionRecord = {
|
||||
id: 'session-scratchpad',
|
||||
projectId: scratchpadProject.id,
|
||||
patternId,
|
||||
title: 'Scratchpad',
|
||||
createdAt: TIMESTAMP,
|
||||
updatedAt: TIMESTAMP,
|
||||
status: 'idle',
|
||||
messages: [],
|
||||
runs: [],
|
||||
};
|
||||
|
||||
return {
|
||||
...workspace,
|
||||
projects: [scratchpadProject],
|
||||
sessions: [session],
|
||||
selectedProjectId: scratchpadProject.id,
|
||||
selectedPatternId: patternId,
|
||||
selectedSessionId: session.id,
|
||||
};
|
||||
}
|
||||
|
||||
beforeEach(async () => {
|
||||
await rm(join(USER_DATA_PATH, 'workspace.json'), { force: true });
|
||||
await rm(join(USER_DATA_PATH, 'scratchpad'), { recursive: true, force: true });
|
||||
});
|
||||
|
||||
afterEach(async () => {
|
||||
await rm(join(USER_DATA_PATH, 'workspace.json'), { force: true });
|
||||
await rm(join(USER_DATA_PATH, 'scratchpad'), { recursive: true, force: true });
|
||||
});
|
||||
|
||||
describe('WorkspaceRepository scratchpad migration', () => {
|
||||
test('assigns per-session scratchpad directories while loading existing workspace state', async () => {
|
||||
const workspaceFilePath = join(USER_DATA_PATH, 'workspace.json');
|
||||
await mkdir(USER_DATA_PATH, { recursive: true });
|
||||
await writeFile(workspaceFilePath, JSON.stringify(createStoredWorkspace(), null, 2), 'utf8');
|
||||
|
||||
const repository = new WorkspaceRepository();
|
||||
const loaded = await repository.load();
|
||||
const loadedSession = loaded.sessions[0];
|
||||
if (!loadedSession) {
|
||||
throw new Error('Expected load() to return the migrated scratchpad session.');
|
||||
}
|
||||
|
||||
const expectedScratchpadPath = join(USER_DATA_PATH, 'scratchpad');
|
||||
const expectedSessionPath = join(expectedScratchpadPath, loadedSession.id);
|
||||
|
||||
expect(loaded.projects[0]?.path).toBe(expectedScratchpadPath);
|
||||
expect(loadedSession.cwd).toBe(expectedSessionPath);
|
||||
await access(expectedSessionPath);
|
||||
|
||||
const persisted = JSON.parse(await readFile(workspaceFilePath, 'utf8')) as WorkspaceState;
|
||||
expect(persisted.sessions[0]?.cwd).toBe(expectedSessionPath);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user