@pi-harness/pi-harness 0.1.14 → 0.1.16

package/packages/core/src/plugins/mirage-bridge.ts

@@ -1,191 +0,0 @@
-import { execFile } from "node:child_process";
-import { promisify } from "node:util";
-import type { Context } from "@deepseek-ai/cordis";
-import z from "@deepseek-ai/schemastery";
-import { Type } from "@earendil-works/pi-ai";
-import { defineTool, type AgentToolResult } from "@earendil-works/pi-coding-agent";
-
-const execFileAsync = promisify(execFile);
-const defaultExecutable = "mirage";
-const defaultTimeoutMs = 60_000;
-const maxCommandLength = 4_000;
-const maxOutputBytes = 128 * 1024;
-
-export interface MirageBridgeConfig {
-  executable?: string;
-  workspaceId?: string;
-  timeoutMs?: number;
-}
-
-export type MirageRun = {
-  workspaceId: string;
-  command: string;
-  exitCode: number | null;
-  durationMs: number;
-  output: string;
-};
-
-type MirageBridgeState = {
-  executable: string;
-  workspaceId: string | null;
-  available: boolean | null;
-  version: string | null;
-  lastError: string | null;
-  lastRun: MirageRun | null;
-};
-
-export const Config: z<MirageBridgeConfig> = z.object({
-  executable: z.string().default(defaultExecutable),
-  workspaceId: z.string().default(""),
-  timeoutMs: z.number().default(defaultTimeoutMs),
-});
-
-function normalizeExecutable(input: string | undefined): string {
-  const executable = (input ?? defaultExecutable).trim();
-  if (executable.length < 1 || executable.length > 512 || executable.includes("\0")) throw new Error("Mirage executable must contain 1-512 characters");
-  return executable;
-}
-
-function normalizeWorkspaceId(input: string | undefined): string | null {
-  const workspaceId = input?.trim() ?? "";
-  if (workspaceId === "") return null;
-  if (!/^[a-z0-9][a-z0-9._-]{0,127}$/iu.test(workspaceId))
-    throw new Error("Mirage workspaceId must contain 1-128 letters, numbers, dots, underscores, or hyphens");
-  return workspaceId;
-}
-
-function normalizeCommand(input: string): string {
-  const command = input.trim();
-  if (command.length < 1 || command.length > maxCommandLength) throw new Error(`Mirage command must contain 1-${maxCommandLength} characters`);
-  return command;
-}
-
-function errorMessage(error: unknown): string {
-  if (error instanceof Error) return error.message;
-  return String(error);
-}
-
-export default {
-  name: "pi-mirage-bridge",
-  inject: ["piHarnessLaunch", "piPluginUi", "piTools"],
-  Config,
-  apply(context: Context, config: MirageBridgeConfig) {
-    const executable = normalizeExecutable(config.executable);
-    const workspaceId = normalizeWorkspaceId(config.workspaceId);
-    const timeoutMs = Math.max(1_000, Math.min(120_000, Math.trunc(config.timeoutMs ?? defaultTimeoutMs)));
-    let state: MirageBridgeState = { executable, workspaceId, available: null, version: null, lastError: null, lastRun: null };
-
-    const doctor = async (): Promise<MirageBridgeState> => {
-      try {
-        const result = await execFileAsync(executable, ["--version"], {
-          cwd: context.piHarnessLaunch.cwd,
-          timeout: Math.min(timeoutMs, 10_000),
-          maxBuffer: maxOutputBytes,
-        });
-        const version = `${result.stdout}${result.stderr}`.trim().split(/\r?\n/u)[0]?.slice(0, 256) || "Mirage CLI detected";
-        state = { ...state, available: true, version, lastError: null };
-      } catch (error) {
-        state = { ...state, available: false, version: null, lastError: errorMessage(error).slice(0, 1_000) };
-      }
-      return state;
-    };
-
-    const run = async (commandInput: string): Promise<MirageRun> => {
-      if (workspaceId === null) throw new Error("Mirage workspaceId is not configured");
-      const command = normalizeCommand(commandInput);
-      const startedAt = Date.now();
-      try {
-        const result = await execFileAsync(executable, ["execute", "--workspace_id", workspaceId, "--command", command], {
-          cwd: context.piHarnessLaunch.cwd,
-          timeout: timeoutMs,
-          maxBuffer: maxOutputBytes,
-        });
-        state = {
-          ...state,
-          available: true,
-          lastError: null,
-          lastRun: { workspaceId, command, exitCode: 0, durationMs: Date.now() - startedAt, output: `${result.stdout}${result.stderr}`.slice(-maxOutputBytes) },
-        };
-      } catch (error) {
-        const failure = error as { code?: number | string; stdout?: string; stderr?: string; message?: string };
-        const unavailable = failure.code === "ENOENT";
-        const output = `${failure.stdout ?? ""}${failure.stderr ?? failure.message ?? ""}`.slice(-maxOutputBytes);
-        state = {
-          ...state,
-          available: unavailable ? false : state.available,
-          lastError: unavailable ? output : null,
-          lastRun: {
-            workspaceId,
-            command,
-            exitCode: typeof failure.code === "number" ? failure.code : null,
-            durationMs: Date.now() - startedAt,
-            output,
-          },
-        };
-      }
-      return state.lastRun!;
-    };
-
-    const unregisterDoctor = context.piTools.register(
-      defineTool({
-        name: "mirage_doctor",
-        label: "Check Mirage CLI",
-        description: "Check whether the official Mirage virtual-terminal CLI is available and report its configured workspace.",
-        promptSnippet: "check the official Mirage virtual terminal integration",
-        parameters: Type.Object({}, { additionalProperties: false }),
-        executionMode: "sequential",
-        async execute(): Promise<AgentToolResult<MirageBridgeState>> {
-          const details = await doctor();
-          return {
-            content: [
-              {
-                type: "text",
-                text: details.available
-                  ? `${details.version}; workspace ${details.workspaceId ?? "not configured"}`
-                  : `Mirage unavailable: ${details.lastError}`,
-              },
-            ],
-            details,
-          };
-        },
-      }),
-    );
-    const unregisterExecute = context.piTools.register(
-      defineTool({
-        name: "mirage_execute",
-        label: "Execute in Mirage",
-        description: "Pass one command to the official Mirage virtual terminal in the configured virtual workspace.",
-        promptSnippet: "run a command inside the configured Mirage virtual workspace",
-        parameters: Type.Object(
-          { command: Type.String({ description: "Command interpreted by Mirage, not the host shell" }) },
-          { additionalProperties: false },
-        ),
-        executionMode: "sequential",
-        async execute(_toolCallId, params): Promise<AgentToolResult<MirageRun>> {
-          const details = await run(params.command);
-          return { content: [{ type: "text", text: `Mirage exited with ${details.exitCode ?? "unknown"}.\n${details.output}` }], details };
-        },
-      }),
-    );
-    let disposePanel: () => void;
-    try {
-      disposePanel = context.piPluginUi.register({
-        id: "mirage-bridge-panel",
-        pluginId: "@pi-harness/core/plugins/mirage-bridge",
-        title: "Mirage Bridge",
-        description: "连接官方 Mirage 虚拟终端，在配置的虚拟工作区中执行命令。",
-        icon: "◇",
-        read: () => ({ ...state, timeoutMs }),
-      });
-    } catch (error) {
-      unregisterDoctor();
-      unregisterExecute();
-      throw error;
-    }
-    context.effect(() => () => {
-      unregisterDoctor();
-      unregisterExecute();
-      disposePanel();
-    });
-  },
-};

package/packages/core/src/plugins/mock-server.ts

@@ -1,189 +0,0 @@
-import { createServer, type IncomingMessage, type ServerResponse, type Server } from "node:http";
-import type { Context } from "@deepseek-ai/cordis";
-import z from "@deepseek-ai/schemastery";
-import { Type } from "@earendil-works/pi-ai";
-import { defineTool, type AgentToolResult } from "@earendil-works/pi-coding-agent";
-
-type MockRoute = { path: string; method?: string; status?: number; headers?: Record<string, string>; body?: string };
-type MockServerState = { running: boolean; url: string | null; routes: number; lastRequest: string | null };
-
-const maxRoutes = 64;
-const maxBodyBytes = 128 * 1024;
-const allowedMethods = new Set(["GET", "POST", "PUT", "PATCH", "DELETE", "HEAD", "OPTIONS"]);
-
-export interface MockServerPluginConfig {
-  port?: number;
-  routes?: MockRoute[];
-}
-
-const routeConfig = z.object({
-  path: z.string(),
-  method: z.string().default("GET"),
-  status: z.number().default(200),
-  headers: z.dict(z.string()).default({}),
-  body: z.string().default(""),
-});
-
-export const Config: z<MockServerPluginConfig> = z.object({ port: z.number().default(0), routes: z.array(routeConfig).default([]) });
-
-function normalizeRoutes(routes: MockRoute[]): MockRoute[] {
-  if (routes.length > maxRoutes) throw new Error(`Mock server cannot exceed ${maxRoutes} routes`);
-  return routes.map((route) => {
-    const path = route.path.trim();
-    if (!path.startsWith("/") || path.length > 2048 || path.includes("#")) throw new Error("Mock route path must start with / and be at most 2048 characters");
-    const method = (route.method ?? "GET").trim().toUpperCase();
-    if (!allowedMethods.has(method)) throw new Error(`Mock route method is not allowed: ${method}`);
-    const status = route.status ?? 200;
-    if (!Number.isInteger(status) || status < 100 || status > 599) throw new Error("Mock route status must be an integer between 100 and 599");
-    const body = route.body ?? "";
-    if (Buffer.byteLength(body, "utf8") > maxBodyBytes) throw new Error(`Mock route body cannot exceed ${maxBodyBytes} bytes`);
-    const headers = Object.fromEntries(Object.entries(route.headers ?? {}).map(([name, value]) => [name, String(value)]));
-    return { path, method, status, headers, body };
-  });
-}
-
-function routeKey(method: string, path: string): string {
-  return `${method} ${path}`;
-}
-
-export default {
-  name: "pi-mock-server",
-  inject: ["piPluginUi", "piTools"],
-  Config,
-  apply(context: Context, config: MockServerPluginConfig) {
-    const routes = normalizeRoutes(config.routes ?? []);
-    let server: Server | undefined;
-    let state: MockServerState = { running: false, url: null, routes: routes.length, lastRequest: null };
-    const start = async (requestedPort?: number): Promise<MockServerState> => {
-      if (server !== undefined) throw new Error("Mock server is already running");
-      const port = requestedPort ?? config.port ?? 0;
-      if (!Number.isInteger(port) || port < 0 || port > 65535) throw new Error("Mock server port must be an integer between 0 and 65535");
-      const routeMap = new Map(routes.map((route) => [routeKey(route.method ?? "GET", route.path), route]));
-      server = createServer((request: IncomingMessage, response: ServerResponse) => {
-        const method = (request.method ?? "GET").toUpperCase();
-        const pathname = new URL(request.url ?? "/", "http://127.0.0.1").pathname;
-        state = { ...state, lastRequest: routeKey(method, pathname) };
-        const route = routeMap.get(routeKey(method, pathname));
-        if (route === undefined) {
-          response.statusCode = 404;
-          response.setHeader("content-type", "text/plain; charset=utf-8");
-          response.end("Not found");
-          return;
-        }
-        response.statusCode = route.status ?? 200;
-        for (const [name, value] of Object.entries(route.headers ?? {})) response.setHeader(name, value);
-        if (!response.hasHeader("content-type")) response.setHeader("content-type", "text/plain; charset=utf-8");
-        response.end(route.body ?? "");
-      });
-      try {
-        await new Promise<void>((resolve, reject) => {
-          const onError = (error: Error): void => {
-            server?.off("listening", onListening);
-            reject(error);
-          };
-          const onListening = (): void => {
-            server?.off("error", onError);
-            resolve();
-          };
-          server!.once("error", onError);
-          server!.once("listening", onListening);
-          server!.listen(port, "127.0.0.1");
-        });
-      } catch (error) {
-        const failed = server;
-        server = undefined;
-        failed?.close();
-        throw error;
-      }
-      const address = server.address();
-      if (address === null || typeof address === "string") throw new Error("Mock server did not expose a TCP address");
-      state = { ...state, running: true, url: `http://127.0.0.1:${address.port}` };
-      return state;
-    };
-    const stop = async (): Promise<boolean> => {
-      if (server === undefined) return false;
-      const current = server;
-      server = undefined;
-      await new Promise<void>((resolve, reject) => current.close((error) => (error ? reject(error) : resolve())));
-      state = { ...state, running: false, url: null };
-      return true;
-    };
-    let unregisterStart: () => void = () => {};
-    let unregisterStop: () => void = () => {};
-    let unregisterStatus: () => void = () => {};
-    try {
-      unregisterStart = context.piTools.register(
-        defineTool({
-          name: "mock_server_start",
-          label: "Mock server start",
-          description: "Start a local deterministic HTTP mock server from configured routes.",
-          promptSnippet: "start the local mock HTTP server",
-          parameters: Type.Object(
-            { port: Type.Optional(Type.Number({ description: "Bind port; 0 selects a free local port" })) },
-            { additionalProperties: false },
-          ),
-          executionMode: "sequential",
-          async execute(_toolCallId, params): Promise<AgentToolResult<MockServerState>> {
-            const result = await start(params.port);
-            return { content: [{ type: "text", text: `Mock server listening at ${result.url}` }], details: result };
-          },
-        }),
-      );
-      unregisterStop = context.piTools.register(
-        defineTool({
-          name: "mock_server_stop",
-          label: "Mock server stop",
-          description: "Stop the local HTTP mock server.",
-          promptSnippet: "stop the local mock HTTP server",
-          parameters: Type.Object({}, { additionalProperties: false }),
-          executionMode: "sequential",
-          async execute(): Promise<AgentToolResult<{ stopped: boolean }>> {
-            const stopped = await stop();
-            return { content: [{ type: "text", text: stopped ? "Mock server stopped." : "Mock server was not running." }], details: { stopped } };
-          },
-        }),
-      );
-      unregisterStatus = context.piTools.register(
-        defineTool({
-          name: "mock_server_status",
-          label: "Mock server status",
-          description: "Show local HTTP mock server status and route count.",
-          promptSnippet: "check the local mock server status",
-          parameters: Type.Object({}, { additionalProperties: false }),
-          executionMode: "sequential",
-          execute(): Promise<AgentToolResult<MockServerState>> {
-            return Promise.resolve({ content: [{ type: "text", text: `${state.running ? "running" : "stopped"} ${state.url ?? ""}`.trim() }], details: state });
-          },
-        }),
-      );
-    } catch (error) {
-      unregisterStart();
-      unregisterStop();
-      unregisterStatus();
-      throw error;
-    }
-    let disposePanel: () => void;
-    try {
-      disposePanel = context.piPluginUi.register({
-        id: "mock-server-panel",
-        pluginId: "@pi-harness/core/plugins/mock-server",
-        title: "Mock Server",
-        description: "在本机回环地址提供可控的 HTTP mock 路由。",
-        icon: "⇄",
-        read: () => state,
-      });
-    } catch (error) {
-      unregisterStart();
-      unregisterStop();
-      unregisterStatus();
-      throw error;
-    }
-    context.effect(() => () => {
-      unregisterStart();
-      unregisterStop();
-      unregisterStatus();
-      disposePanel();
-      void stop().catch(() => undefined);
-    });
-  },
-};

package/packages/core/src/plugins/model.ts

@@ -1,18 +0,0 @@
-import type { Context } from "@deepseek-ai/cordis";
-import z from "@deepseek-ai/schemastery";
-import { assertKnownConfigKeys } from "../config.js";
-
-export const Config = z.object({});
-
-export default {
-  name: "pi-model",
-  inject: ["piModelRuntime", "piResources"],
-  Config,
-  apply(context: Context, config: unknown) {
-    assertKnownConfigKeys("pi-model", config, []);
-    const { runtime, provider, model: modelId } = context.piModelRuntime;
-    const model = runtime.getModel(provider, modelId);
-    if (model === undefined) throw new Error(`Pi model is not registered: ${provider}/${modelId}`);
-    context.provide("piModels", { runtime, model });
-  },
-};

package/packages/core/src/plugins/models.ts

@@ -1,43 +0,0 @@
-import { join } from "node:path";
-import type { Context } from "@deepseek-ai/cordis";
-import z from "@deepseek-ai/schemastery";
-import { ModelRuntime } from "@earendil-works/pi-coding-agent";
-import { assertKnownConfigKeys } from "../config.js";
-
-export interface ModelsPluginConfig {
-  provider: string;
-  model: string;
-  refreshOnCreate?: boolean;
-}
-
-const maxProviderIdLength = 128;
-const maxModelIdLength = 512;
-const modelSelectionIdPattern = /^[^\s\p{Cc}]+$/u;
-const modelRefreshTimeoutMs = 15_000;
-
-export const Config: z<ModelsPluginConfig> = z.object({
-  provider: z.string().required().min(1).max(maxProviderIdLength).pattern(modelSelectionIdPattern),
-  model: z.string().required().min(1).max(maxModelIdLength).pattern(modelSelectionIdPattern),
-  refreshOnCreate: z.boolean().default(false),
-});
-
-export default {
-  name: "pi-models",
-  inject: ["piHarnessLaunch"],
-  Config,
-  async apply(context: Context, config: ModelsPluginConfig) {
-    assertKnownConfigKeys("pi-models", config, ["provider", "model", "refreshOnCreate"]);
-    const agentDir = context.piHarnessLaunch.agentDir;
-    const runtime = await ModelRuntime.create({
-      authPath: join(agentDir, "auth.json"),
-      modelsPath: join(agentDir, "models.json"),
-      modelsStorePath: join(agentDir, "models-store.json"),
-      refreshOnCreate: config.refreshOnCreate ?? false,
-      allowModelNetwork: config.refreshOnCreate ?? false,
-      modelRefreshTimeoutMs,
-    });
-    const runtimeError = runtime.getError();
-    if (runtimeError !== undefined) throw new Error(runtimeError);
-    context.provide("piModelRuntime", { runtime, provider: config.provider, model: config.model });
-  },
-};