@pi-harness/pi-harness 0.1.14 → 0.1.16

package/packages/core/src/plugins/yaml-validator.ts

@@ -1,216 +0,0 @@
-import type { Context } from "@deepseek-ai/cordis";
-import { parseAllDocuments, isMap, isSeq, type YAMLParseError, type YAMLWarning } from "yaml";
-import { Type } from "@earendil-works/pi-ai";
-import { defineTool, type AgentToolResult } from "@earendil-works/pi-coding-agent";
-import { readBoundedFile } from "../bounded-file.js";
-import { EmptyConfig } from "../config.js";
-import { resolveExistingWorkspacePath } from "../workspace-path.js";
-
-const maxBytes = 512 * 1024;
-const maxPathLength = 4_096;
-const maxDocuments = 100;
-const maxDiagnostics = 1_000;
-const maxToolDiagnostics = 50;
-const maxPanelDiagnostics = 50;
-const maxDiagnosticMessageLength = 2_000;
-const maxStatusErrorLength = 2_000;
-const pathParameterNames = new Set(["path"]);
-type Diagnostic = { message: string; code?: string; line?: number; column?: number };
-type ValidationStatus = { state: "idle" | "running" | "completed" | "failed" | "cancelled"; at?: string; error?: string };
-type YamlReport = {
-  path: string;
-  valid: boolean;
-  bytes: number;
-  documents: number;
-  rootType: string;
-  errorCount: number;
-  warningCount: number;
-  diagnosticsTruncated: boolean;
-  errors: Diagnostic[];
-  warnings: Diagnostic[];
-};
-
-function diagnostic(issue: YAMLParseError | YAMLWarning): Diagnostic {
-  const position = issue.linePos?.[0];
-  return {
-    message: issue.message.slice(0, maxDiagnosticMessageLength),
-    ...(typeof issue.code === "string" ? { code: issue.code.slice(0, 128) } : {}),
-    ...(position === undefined ? {} : { line: position.line, column: position.col }),
-  };
-}
-
-function rootType(value: unknown): string {
-  if (value === null || value === undefined) return "empty";
-  if (isMap(value)) return "map";
-  if (isSeq(value)) return "seq";
-  return "scalar";
-}
-
-function pathParameter(value: unknown): string {
-  if (value === null || typeof value !== "object" || Array.isArray(value)) throw new Error("YAML validator parameters must be an object");
-  try {
-    const prototype = Object.getPrototypeOf(value) as unknown;
-    if (prototype !== Object.prototype && prototype !== null) throw new Error("YAML validator parameters must be a plain object");
-    const descriptors = Object.getOwnPropertyDescriptors(value);
-    if (Reflect.ownKeys(descriptors).some((key) => typeof key !== "string" || !pathParameterNames.has(key)))
-      throw new Error("YAML validator parameters contain an unknown property");
-    if (Object.values(descriptors).some((descriptor) => !("value" in descriptor))) throw new Error("YAML validator parameters must use data properties");
-    const path: unknown = descriptors.path?.value as unknown;
-    if (typeof path !== "string") throw new Error("YAML validator path must be a string");
-    if (path.length === 0 || path.length > maxPathLength) throw new Error(`YAML validator path must contain 1-${maxPathLength} characters`);
-    if (path.includes("\0")) throw new Error("YAML validator path must not contain NUL characters");
-    return path;
-  } catch (error) {
-    if (
-      error instanceof Error &&
-      /^(?:YAML validator parameters must be a plain object|YAML validator parameters contain an unknown property|YAML validator parameters must use data properties|YAML validator path)/u.test(
-        error.message,
-      )
-    )
-      throw error;
-    throw new Error("YAML validator parameters must be an accessible plain object", { cause: error });
-  }
-}
-
-function throwIfAborted(signal: AbortSignal): void {
-  if (signal.aborted) throw new Error("YAML validation was cancelled", { cause: signal.reason });
-}
-
-function boundedError(error: unknown): string {
-  if (typeof error === "string") return error.slice(0, maxStatusErrorLength);
-  if (typeof error === "object" && error !== null) {
-    const descriptor = Object.getOwnPropertyDescriptor(error, "message");
-    if (descriptor !== undefined && "value" in descriptor && typeof descriptor.value === "string") return descriptor.value.slice(0, maxStatusErrorLength);
-  }
-  return "Unknown YAML validation error";
-}
-
-export default {
-  name: "pi-yaml-validator",
-  inject: ["piHarnessLaunch", "piPluginUi", "piTools"],
-  Config: EmptyConfig,
-  apply(context: Context) {
-    const lifecycle = new AbortController();
-    let latest: YamlReport | undefined;
-    let status: ValidationStatus = { state: "idle" };
-    const validate = async (requested: string, signal: AbortSignal): Promise<YamlReport> => {
-      throwIfAborted(signal);
-      const location = await resolveExistingWorkspacePath(context.piHarnessLaunch.cwd, requested, "YAML path must stay inside the current workspace");
-      throwIfAborted(signal);
-      const bytes = await readBoundedFile(location.target, maxBytes, "YAML file");
-      throwIfAborted(signal);
-      let source: string;
-      try {
-        source = new TextDecoder("utf-8", { fatal: true }).decode(bytes);
-      } catch (error) {
-        throw new Error("YAML file must contain valid UTF-8", { cause: error });
-      }
-      const documents = parseAllDocuments(source);
-      throwIfAborted(signal);
-      if (documents.length > maxDocuments) throw new Error(`YAML streams cannot exceed ${maxDocuments} documents`);
-      const errorCount = documents.reduce((total, document) => total + document.errors.length, 0);
-      const warningCount = documents.reduce((total, document) => total + document.warnings.length, 0);
-      const errors = documents
-        .flatMap((document) => document.errors)
-        .slice(0, maxDiagnostics)
-        .map(diagnostic);
-      const warnings = documents
-        .flatMap((document) => document.warnings)
-        .slice(0, Math.max(0, maxDiagnostics - errors.length))
-        .map(diagnostic);
-      const report: YamlReport = {
-        path: location.relativePath,
-        valid: errorCount === 0,
-        bytes: bytes.byteLength,
-        documents: documents.length,
-        rootType: rootType(documents[0]?.contents),
-        errorCount,
-        warningCount,
-        diagnosticsTruncated: errorCount + warningCount > errors.length + warnings.length,
-        errors,
-        warnings,
-      };
-      latest = structuredClone(report);
-      return report;
-    };
-    const unregisterTool = context.piTools.register(
-      defineTool({
-        name: "yaml_validate",
-        label: "YAML validate",
-        description: "Parse a YAML file in the current workspace and report line-aware errors without modifying it.",
-        promptSnippet: "validate a YAML file and report syntax diagnostics",
-        parameters: Type.Object(
-          { path: Type.String({ description: "YAML path relative to the workspace", minLength: 1, maxLength: maxPathLength }) },
-          { additionalProperties: false },
-        ),
-        executionMode: "sequential",
-        async execute(_toolCallId, params, signal): Promise<AgentToolResult<YamlReport>> {
-          const operationSignal = signal === undefined ? lifecycle.signal : AbortSignal.any([signal, lifecycle.signal]);
-          status = { state: "running" };
-          try {
-            throwIfAborted(operationSignal);
-            const report = await validate(pathParameter(params), operationSignal);
-            status = { state: "completed", at: new Date().toISOString() };
-            const summary = report.valid ? `YAML is valid (${report.documents} document(s)).` : `YAML is invalid with ${report.errorCount} error(s).`;
-            const preview = report.errors.slice(0, maxToolDiagnostics);
-            const omitted = report.errorCount - preview.length;
-            return {
-              content: [
-                {
-                  type: "text",
-                  text: `${summary}\n${preview.map((issue) => `${issue.line ?? "?"}:${issue.column ?? "?"} ${issue.message}`).join("\n")}${omitted > 0 ? `\n… ${omitted} additional error(s) omitted from the text preview.` : ""}`,
-                },
-              ],
-              details: structuredClone(report),
-            };
-          } catch (error) {
-            status = { state: operationSignal.aborted ? "cancelled" : "failed", at: new Date().toISOString(), error: boundedError(error) };
-            throw error;
-          }
-        },
-      }),
-    );
-    let disposePanel: () => void;
-    try {
-      disposePanel = context.piPluginUi.register({
-        id: "yaml-validator-panel",
-        pluginId: "@pi-harness/core/plugins/yaml-validator",
-        title: "YAML Validator",
-        description: "只读解析工作区 YAML，并展示行列级语法诊断。",
-        icon: "⌁",
-        read: () => {
-          const errors = structuredClone(latest?.errors.slice(0, maxPanelDiagnostics) ?? []);
-          const warnings = structuredClone(latest?.warnings.slice(0, Math.max(0, maxPanelDiagnostics - errors.length)) ?? []);
-          return {
-            latest: latest === undefined ? null : { ...structuredClone(latest), errors, warnings },
-            status: structuredClone(status),
-            maxBytes,
-            inventory: {
-              errors: { total: latest?.errorCount ?? 0, shown: errors.length, truncated: (latest?.errorCount ?? 0) > errors.length },
-              warnings: { total: latest?.warningCount ?? 0, shown: warnings.length, truncated: (latest?.warningCount ?? 0) > warnings.length },
-            },
-            limits: {
-              fileBytes: maxBytes,
-              pathCharacters: maxPathLength,
-              documents: maxDocuments,
-              diagnostics: maxDiagnostics,
-              panelDiagnostics: maxPanelDiagnostics,
-              toolDiagnostics: maxToolDiagnostics,
-              diagnosticMessageCharacters: maxDiagnosticMessageLength,
-              statusErrorCharacters: maxStatusErrorLength,
-            },
-          };
-        },
-      });
-    } catch (error) {
-      lifecycle.abort(new Error("YAML validator plugin activation failed", { cause: error }));
-      unregisterTool();
-      throw error;
-    }
-    context.effect(() => () => {
-      lifecycle.abort(new Error("YAML validator plugin disposed"));
-      unregisterTool();
-      disposePanel();
-    });
-  },
-};

package/packages/core/src/profile.ts

@@ -1,37 +0,0 @@
-import { access } from "node:fs/promises";
-import { resolve } from "node:path";
-import { fileURLToPath } from "node:url";
-
-export const BUILTIN_PROFILES = ["default", "development"] as const;
-
-export interface ResolveProfileConfigOptions {
-  profile?: string;
-  configPath?: string;
-  cwd?: string;
-  profilesDir?: string;
-}
-
-export async function resolveProfileConfig(_options: ResolveProfileConfigOptions = {}): Promise<string> {
-  const options = _options;
-  if (options.profile !== undefined && options.configPath !== undefined) throw new Error("profile and configPath cannot be used together");
-  const cwd = options.cwd ?? process.cwd();
-  let configPath: string;
-  if (options.configPath !== undefined) {
-    configPath = resolve(cwd, options.configPath);
-  } else {
-    const profile = options.profile ?? "default";
-    if (!/^[a-z0-9](?:[a-z0-9-]*[a-z0-9])?$/.test(profile)) throw new Error(`Invalid profile name: ${profile}`);
-    const profilesDir = options.profilesDir ?? fileURLToPath(new URL("../profiles", import.meta.url));
-    configPath = resolve(profilesDir, profile, "cordis.yml");
-  }
-  try {
-    await access(configPath);
-  } catch (cause) {
-    if (options.configPath === undefined)
-      throw new Error(`Pi Harness profile config does not exist or is not readable: ${configPath}; built-in profiles are ${BUILTIN_PROFILES.join(", ")}`, {
-        cause,
-      });
-    throw new Error(`Pi Harness profile config does not exist or is not readable: ${configPath}`, { cause });
-  }
-  return configPath;
-}

package/packages/core/src/runtime.ts

@@ -1,52 +0,0 @@
-import type { AgentSession, AgentSessionRuntime } from "@earendil-works/pi-coding-agent";
-import type { PiRuntimeService } from "./services.js";
-
-export class PiRuntimeDisposedError extends Error {
-  override readonly name = "PiRuntimeDisposedError";
-
-  constructor() {
-    super("Pi runtime is disposed");
-  }
-}
-
-export class PiRuntime implements PiRuntimeService {
-  readonly sessionRuntime: AgentSessionRuntime;
-  #disposed = false;
-  #sessionDisposed = false;
-  #disposePromise: Promise<void> | undefined;
-
-  constructor(sessionRuntime: AgentSessionRuntime) {
-    this.sessionRuntime = sessionRuntime;
-  }
-
-  get session(): AgentSession {
-    return this.sessionRuntime.session;
-  }
-
-  async prompt(text: string): Promise<void> {
-    if (this.#disposed) throw new PiRuntimeDisposedError();
-    await this.session.prompt(text);
-  }
-
-  async abort(): Promise<void> {
-    if (this.#sessionDisposed) return;
-    await this.session.abort();
-  }
-
-  dispose(): Promise<void> {
-    if (this.#disposePromise !== undefined) return this.#disposePromise;
-    this.#disposed = true;
-    this.#disposePromise = this.#disposeSession();
-    return this.#disposePromise;
-  }
-
-  async #disposeSession(): Promise<void> {
-    try {
-      // Settle an in-flight turn first so its tool results are persisted before session_shutdown runs.
-      if (!this.session.isIdle) await this.session.abort();
-    } finally {
-      this.#sessionDisposed = true;
-      await this.sessionRuntime.dispose();
-    }
-  }
-}

package/packages/core/src/services.ts

@@ -1,232 +0,0 @@
-import { isAbsolute } from "node:path";
-import { pathToFileURL } from "node:url";
-import type { Context } from "@deepseek-ai/cordis";
-import type {
-  AgentSession,
-  AgentSessionEvent,
-  AgentSessionRuntime,
-  AgentSessionServices,
-  ExtensionError,
-  SessionManager,
-  ToolDefinition,
-} from "@earendil-works/pi-coding-agent";
-import type { Api, Model } from "@earendil-works/pi-ai";
-import type { ModelRuntime } from "@earendil-works/pi-coding-agent";
-
-export interface PiHarnessLaunch {
-  readonly cwd: string;
-  readonly cwdUrl?: string;
-  readonly agentDir: string;
-  readonly configPath?: string;
-  readonly args: readonly string[];
-  requestExit(code: number): void;
-}
-
-export interface PiModelsService {
-  readonly runtime: ModelRuntime;
-  readonly model: Model<Api>;
-}
-
-export interface PiModelRuntimeService {
-  readonly runtime: ModelRuntime;
-  readonly provider: string;
-  readonly model: string;
-}
-
-export type PiResourcesService = AgentSessionServices & {
-  createForCwd(cwd: string): Promise<AgentSessionServices>;
-};
-
-export interface PiSessionService {
-  readonly manager: SessionManager;
-}
-
-export interface PiMcpServerSnapshot {
-  readonly id: string;
-  readonly command: readonly string[];
-  readonly status: string;
-  readonly startedAt: number;
-}
-
-export interface PiMcpService {
-  snapshot(): { readonly servers: readonly PiMcpServerSnapshot[] };
-}
-
-export interface PiRuntimeService {
-  readonly session: AgentSession;
-  readonly sessionRuntime: AgentSessionRuntime;
-  prompt(text: string): Promise<void>;
-  abort(): Promise<void>;
-  dispose(): Promise<void>;
-}
-
-export interface PiTelemetryEvent {
-  readonly name: string;
-  readonly properties?: Readonly<Record<string, unknown>>;
-}
-
-export interface PiTelemetryService {
-  readonly enabled: false;
-  send(event: PiTelemetryEvent): { blocked: true; name: string };
-  snapshot(): { blocked: number; names: readonly string[] };
-}
-
-export interface PiToolsSnapshot {
-  readonly names: string[];
-  readonly customTools: ToolDefinition[];
-}
-
-export interface PiToolsLease extends PiToolsSnapshot {
-  release(): void;
-}
-
-const PI_BUILTIN_TOOL_NAMES = ["read", "bash", "powershell", "edit", "write", "grep", "find", "ls"];
-const MAX_PI_TOOL_NAMES = 256;
-const MAX_PI_TOOL_NAME_LENGTH = 128;
-const PI_TOOL_NAME_PATTERN = /^[^\s\p{Cc}]+$/u;
-
-function assertPiToolName(name: unknown): asserts name is string {
-  if (typeof name !== "string" || name.length === 0 || name.length > MAX_PI_TOOL_NAME_LENGTH || !PI_TOOL_NAME_PATTERN.test(name))
-    throw new Error(`Pi tool name must contain 1-${MAX_PI_TOOL_NAME_LENGTH} non-whitespace, non-control characters`);
-}
-
-export class PiToolRegistry {
-  readonly #names: string[];
-  readonly #customTools = new Map<string, ToolDefinition>();
-  #leases = 0;
-
-  constructor(names: readonly string[] = []) {
-    const uncheckedNames: unknown = names;
-    if (!Array.isArray(uncheckedNames) || uncheckedNames.length > MAX_PI_TOOL_NAMES)
-      throw new Error(`Pi tool names must contain at most ${MAX_PI_TOOL_NAMES} entries`);
-    const validatedNames: string[] = [];
-    for (const name of uncheckedNames) {
-      const uncheckedName: unknown = name;
-      assertPiToolName(uncheckedName);
-      validatedNames.push(uncheckedName);
-    }
-    if (new Set(validatedNames).size !== validatedNames.length) throw new Error("Pi tool names must be unique");
-    this.#names = validatedNames;
-  }
-
-  register(tool: ToolDefinition): () => void {
-    assertPiToolName(tool.name);
-    if (this.#leases > 0) throw new Error(`Pi tool registry is leased by pi-runtime; declare a Cordis injection that activates ${tool.name} before pi-runtime`);
-    if (this.#names.includes(tool.name) || this.#customTools.has(tool.name)) throw new Error(`Pi tool is already registered: ${tool.name}`);
-    if (PI_BUILTIN_TOOL_NAMES.includes(tool.name)) throw new Error(`Pi tool name is reserved by a built-in tool: ${tool.name}`);
-    this.#customTools.set(tool.name, tool);
-    return () => {
-      if (this.#customTools.get(tool.name) === tool) this.#customTools.delete(tool.name);
-    };
-  }
-
-  snapshot(): PiToolsSnapshot {
-    return { names: [...this.#names], customTools: [...this.#customTools.values()] };
-  }
-
-  acquire(): PiToolsLease {
-    this.#leases += 1;
-    let released = false;
-    return {
-      ...this.snapshot(),
-      release: () => {
-        if (released) return;
-        released = true;
-        this.#leases -= 1;
-      },
-    };
-  }
-}
-
-export interface PiPluginPanel {
-  readonly id: string;
-  readonly pluginId: string;
-  readonly title: string;
-  readonly description?: string;
-  readonly icon?: string;
-  readonly visible?: () => boolean | Promise<boolean>;
-  readonly read: () => unknown;
-}
-
-export interface PiPluginPanelSnapshot {
-  readonly id: string;
-  readonly pluginId: string;
-  readonly title: string;
-  readonly description?: string;
-  readonly icon?: string;
-  readonly data?: unknown;
-  readonly error?: string;
-}
-
-export class PiPluginUiRegistry {
-  readonly #panels = new Map<string, PiPluginPanel>();
-
-  register(panel: PiPluginPanel): () => void {
-    if (panel.id.trim() === "" || panel.pluginId.trim() === "" || panel.title.trim() === "")
-      throw new Error("Plugin UI panel id, pluginId, and title are required");
-    if (this.#panels.has(panel.id)) throw new Error(`Plugin UI panel is already registered: ${panel.id}`);
-    this.#panels.set(panel.id, panel);
-    let active = true;
-    return () => {
-      if (!active) return;
-      active = false;
-      if (this.#panels.get(panel.id) === panel) this.#panels.delete(panel.id);
-    };
-  }
-
-  async snapshot(): Promise<readonly PiPluginPanelSnapshot[]> {
-    const snapshots: PiPluginPanelSnapshot[] = [];
-    for (const panel of this.#panels.values()) {
-      if (panel.visible !== undefined && !(await panel.visible())) continue;
-      try {
-        const data = await panel.read();
-        snapshots.push({
-          id: panel.id,
-          pluginId: panel.pluginId,
-          title: panel.title,
-          ...(panel.description === undefined ? {} : { description: panel.description }),
-          ...(panel.icon === undefined ? {} : { icon: panel.icon }),
-          data: structuredClone(data),
-        });
-      } catch (error) {
-        snapshots.push({
-          id: panel.id,
-          pluginId: panel.pluginId,
-          title: panel.title,
-          ...(panel.description === undefined ? {} : { description: panel.description }),
-          ...(panel.icon === undefined ? {} : { icon: panel.icon }),
-          error: error instanceof Error ? error.message : String(error),
-        });
-      }
-    }
-    return snapshots;
-  }
-}
-
-declare module "@deepseek-ai/cordis" {
-  interface Context {
-    piHarnessLaunch: PiHarnessLaunch;
-    piModelRuntime: PiModelRuntimeService;
-    piModels: PiModelsService;
-    piResources: PiResourcesService;
-    piSession: PiSessionService;
-    piMcp: PiMcpService;
-    piTools: PiToolRegistry;
-    piPluginUi: PiPluginUiRegistry;
-    piRuntime: PiRuntimeService;
-    piTelemetry: PiTelemetryService;
-  }
-
-  interface Events {
-    "pi/session-event"(event: AgentSessionEvent): void;
-    "pi/extension-error"(error: ExtensionError): void;
-    "pi/telemetry"(event: PiTelemetryEvent): void;
-  }
-}
-
-export function provideLaunchContext(context: Context, launch: PiHarnessLaunch): () => void {
-  if (!isAbsolute(launch.cwd)) throw new Error(`Pi Harness launch cwd must be an absolute path: ${launch.cwd}`);
-  if (!isAbsolute(launch.agentDir)) throw new Error(`Pi Harness agent directory must be an absolute path: ${launch.agentDir}`);
-  const value: PiHarnessLaunch = Object.freeze({ ...launch, cwdUrl: pathToFileURL(launch.cwd).href, args: Object.freeze([...launch.args]) });
-  return context.provide("piHarnessLaunch", value);
-}