Function createOPFSTools

src/app/service/agent/core/tools/opfs_tools.ts:133–247

()

// ---- Factory ----

export function createOPFSTools(): {
  tools: Array<{ definition: ToolDefinition; executor: ToolExecutor }>;
} {
  const writeExecutor: ToolExecutor = {
    execute: async (args: Record<string, unknown>) => {
      const path = requireString(args, "path");
      const result = await writeWorkspaceFile(path, args.content as string | Blob);
      return JSON.stringify(result);
    },
  };

  const readExecutor: ToolExecutor = {
    execute: async (args: Record<string, unknown>) => {
      const safePath = sanitizePath(requireString(args, "path"));
      if (!safePath) throw new Error("path is required");

      const workspace = await getWorkspaceRoot();
      const { dirPath, fileName } = splitPath(safePath);
      const dir = dirPath ? await getDirectory(workspace, dirPath) : workspace;
      const fileHandle = await dir.getFileHandle(fileName);
      const file = await fileHandle.getFile();
      const mimeType = guessMimeType(safePath);
      const arrayBuffer = await file.arrayBuffer();

      // 确定返回模式：auto 通过内容字节检测文本/二进制
      const mode = (args.mode as string) || "auto";
      const useText = mode === "text" || (mode === "auto" && isText(new Uint8Array(arrayBuffer)));

      // blob 模式：返回 blob URL
      if (!useText) {
        if (!createBlobUrlFn) {
          throw new Error("Blob URL creation not available (Offscreen not initialized)");
        }
        const blobUrl = await createBlobUrlFn(arrayBuffer, mimeType);
        return JSON.stringify({ path: safePath, blobUrl, size: file.size, mimeType, type: "binary" });
      }

      // text 模式：返回文本内容
      const text = new TextDecoder().decode(arrayBuffer);
      const lines = text.split("\n");
      const totalLines = lines.length;

      const offset = typeof args.offset === "number" ? args.offset : undefined;
      const limit = typeof args.limit === "number" ? args.limit : undefined;

      // 超过行数限制且未指定分页参数，报错要求分段读取
      if (offset == null && limit == null && totalLines > MAX_TEXT_LINES) {
        throw new Error(
          `文件共 ${totalLines} 行，超过单次读取上限（${MAX_TEXT_LINES} 行）。` +
            `请使用 offset 和 limit 参数分段读取，例如：offset=1, limit=${MAX_TEXT_LINES}`
        );
      }

      const startLine = offset != null ? Math.max(1, offset) : 1;
      const endLine = limit != null ? Math.min(totalLines, startLine + limit - 1) : totalLines;
      const selectedLines = lines.slice(startLine - 1, endLine);
      const content = selectedLines.join("\n");