Mohamed BoudraandGitHub
3029b5991a
* feat(app): refine workspace pane actions Prioritize Changes when available, keep Files useful before a selection, and reserve desktop diff clicks for selection so comments require the explicit affordance. Move the Changes shortcut to Shift+G to avoid consuming a likely copy variant. * feat(app): improve access to workspace changes
98 lines
2.8 KiB
TypeScript
98 lines
2.8 KiB
TypeScript
import type { Page } from "@playwright/test";
|
|
import { seedWorkspace, type SeedDaemonClient } from "./seed-client";
|
|
import { getServerId } from "./server-id";
|
|
import { buildHostAgentDetailRoute } from "../../../src/utils/host-routes";
|
|
|
|
export interface MockAgentWorkspace {
|
|
agentId: string;
|
|
workspaceId: string;
|
|
cwd: string;
|
|
client: SeedDaemonClient;
|
|
cleanup(): Promise<void>;
|
|
}
|
|
|
|
export interface MockAgentOptions {
|
|
repoPrefix: string;
|
|
title: string;
|
|
repo?: Parameters<typeof seedWorkspace>[0]["repo"];
|
|
port?: number;
|
|
initialPrompt?: string;
|
|
model?: string;
|
|
modeId?: string;
|
|
featureValues?: Record<string, unknown>;
|
|
}
|
|
|
|
/**
|
|
* Seeds a temp git repo, opens it as a project, and creates a ready mock-provider
|
|
* agent in it via the daemon. Returns the agent id plus a cleanup that closes the
|
|
* client and removes the repo. Pair with {@link openAgentRoute} to drive the UI.
|
|
*/
|
|
export async function seedMockAgentWorkspace(
|
|
options: MockAgentOptions,
|
|
): Promise<MockAgentWorkspace> {
|
|
const workspace = await seedWorkspace({
|
|
repoPrefix: options.repoPrefix,
|
|
repo: options.repo,
|
|
port: options.port,
|
|
});
|
|
try {
|
|
const agent = await workspace.client.createAgent({
|
|
provider: "mock",
|
|
cwd: workspace.repoPath,
|
|
workspaceId: workspace.workspaceId,
|
|
title: options.title,
|
|
modeId: options.modeId ?? "load-test",
|
|
model: options.model ?? "ten-second-stream",
|
|
initialPrompt: options.initialPrompt,
|
|
featureValues: options.featureValues,
|
|
});
|
|
return {
|
|
agentId: agent.id,
|
|
workspaceId: workspace.workspaceId,
|
|
cwd: workspace.repoPath,
|
|
client: workspace.client,
|
|
cleanup: workspace.cleanup,
|
|
};
|
|
} catch (error) {
|
|
await workspace.cleanup();
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
export async function seedRunningMockAgentWorkspace(
|
|
options: MockAgentOptions,
|
|
): Promise<MockAgentWorkspace> {
|
|
const agent = await seedMockAgentWorkspace(options);
|
|
try {
|
|
await agent.client.waitForAgentUpsert(
|
|
agent.agentId,
|
|
(snapshot) => snapshot.status === "running",
|
|
15_000,
|
|
);
|
|
return agent;
|
|
} catch (error) {
|
|
await agent.cleanup();
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
export function buildAgentRoute(
|
|
workspaceId: string,
|
|
agentId: string,
|
|
serverId = getServerId(),
|
|
): string {
|
|
return buildHostAgentDetailRoute(serverId, agentId, workspaceId);
|
|
}
|
|
|
|
/** Boots the app directly at the agent's workspace route and waits for the open intent to settle. */
|
|
export async function openAgentRoute(
|
|
page: Page,
|
|
input: { workspaceId: string; agentId: string },
|
|
): Promise<void> {
|
|
await page.goto(buildAgentRoute(input.workspaceId, input.agentId), { waitUntil: "commit" });
|
|
await page.waitForURL(
|
|
(url) => url.pathname.includes("/workspace/") && !url.searchParams.has("open"),
|
|
{ timeout: 60_000, waitUntil: "commit" },
|
|
);
|
|
}
|