@lupaflow/tools 0.1.1 → 0.1.2

package/dist/index.js

@@ -0,0 +1,724 @@
+// src/registry.ts
+import { ToolError } from "@lupaflow/core";
+var ToolRegistry = class {
+  tools = /* @__PURE__ */ new Map();
+  register(tool) {
+    this.tools.set(tool.definition.name, tool);
+  }
+  get(name) {
+    const tool = this.tools.get(name);
+    if (!tool) {
+      throw new ToolError(name, `Tool not found: "${name}"`);
+    }
+    return tool;
+  }
+  has(name) {
+    return this.tools.has(name);
+  }
+  list() {
+    return Array.from(this.tools.values());
+  }
+  remove(name) {
+    this.tools.delete(name);
+  }
+  clear() {
+    this.tools.clear();
+  }
+};
+var toolRegistry = new ToolRegistry();
+
+// src/interface.ts
+var BaseTool = class {
+  toJSON() {
+    return this.definition;
+  }
+};
+
+// src/builtins/calculator.ts
+var CalculatorTool = class extends BaseTool {
+  definition = {
+    name: "calculator",
+    description: "Evaluate mathematical expressions safely",
+    parameters: {
+      expression: {
+        type: "string",
+        description: "Mathematical expression to evaluate (e.g., '2 + 2 * 3')"
+      }
+    },
+    required: ["expression"]
+  };
+  async execute(args) {
+    const { expression } = args;
+    const sanitized = expression.replace(/[^0-9+\-*/.()^% ]/g, "");
+    try {
+      const result = Function(`"use strict"; return (${sanitized})`)();
+      return { result, expression };
+    } catch (err) {
+      return { error: err.message, expression };
+    }
+  }
+};
+
+// src/builtins/http.ts
+var HTTPTool = class extends BaseTool {
+  definition = {
+    name: "http",
+    description: "Make HTTP requests (GET, POST, PUT, DELETE)",
+    parameters: {
+      method: {
+        type: "string",
+        description: "HTTP method: GET, POST, PUT, DELETE"
+      },
+      url: {
+        type: "string",
+        description: "Request URL"
+      },
+      headers: {
+        type: "object",
+        description: "Request headers as key-value object"
+      },
+      body: {
+        type: "string",
+        description: "Request body (stringified JSON for POST/PUT)"
+      }
+    },
+    required: ["method", "url"]
+  };
+  async execute(args) {
+    const { method, url, headers, body } = args;
+    const response = await fetch(url, {
+      method: method.toUpperCase(),
+      headers: { "Content-Type": "application/json", ...headers },
+      body: body || void 0
+    });
+    const text = await response.text();
+    let data = text;
+    try {
+      data = JSON.parse(text);
+    } catch {
+    }
+    return {
+      status: response.status,
+      statusText: response.statusText,
+      headers: Object.fromEntries(response.headers.entries()),
+      data
+    };
+  }
+};
+
+// src/builtins/search.ts
+var SearchTool = class extends BaseTool {
+  definition = {
+    name: "search",
+    description: "Search the web for information",
+    parameters: {
+      query: {
+        type: "string",
+        description: "Search query"
+      },
+      maxResults: {
+        type: "number",
+        description: "Maximum number of results (default: 5)"
+      }
+    },
+    required: ["query"]
+  };
+  async execute(args) {
+    const { query, maxResults } = args;
+    const tavilyKey = process.env.TAVILY_API_KEY;
+    if (tavilyKey) {
+      return this.searchTavily(query, maxResults || 5, tavilyKey);
+    }
+    return this.searchDuckDuckGo(query, maxResults || 5);
+  }
+  async searchTavily(query, maxResults, apiKey) {
+    try {
+      const res = await fetch("https://api.tavily.com/search", {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({
+          api_key: apiKey,
+          query,
+          search_depth: "advanced",
+          include_answer: true,
+          max_results: maxResults
+        })
+      });
+      if (!res.ok) {
+        throw new Error(`Tavily API error: ${res.status} ${res.statusText}`);
+      }
+      const data = await res.json();
+      return {
+        query,
+        results: data.results.map((r) => ({
+          title: r.title,
+          url: r.url,
+          snippet: r.content
+        })),
+        answer: data.answer,
+        source: "tavily"
+      };
+    } catch (err) {
+      return { error: err.message, query, source: "tavily" };
+    }
+  }
+  async searchDuckDuckGo(query, maxResults) {
+    try {
+      const url = `https://html.duckduckgo.com/html/?q=${encodeURIComponent(query)}`;
+      const response = await fetch(url);
+      const html = await response.text();
+      const results = [];
+      const linkRegex = /<a[^>]+class="result__a"[^>]*href="([^"]*)"[^>]*>([\s\S]*?)<\/a>/g;
+      const snippetRegex = /<a[^>]+class="result__snippet"[^>]*>([\s\S]*?)<\/a>/g;
+      let count = 0;
+      let linkMatch;
+      while ((linkMatch = linkRegex.exec(html)) !== null && count < maxResults) {
+        const snippetMatch = snippetRegex.exec(html);
+        results.push({
+          link: linkMatch[1],
+          title: linkMatch[2].replace(/<[^>]*>/g, ""),
+          snippet: snippetMatch ? snippetMatch[1].replace(/<[^>]*>/g, "") : ""
+        });
+        count++;
+      }
+      return { query, results, source: "duckduckgo" };
+    } catch (err) {
+      return { error: err.message, query, source: "duckduckgo" };
+    }
+  }
+};
+
+// src/builtins/readFile.ts
+import { readFile } from "fs/promises";
+import { relative, resolve, isAbsolute } from "path";
+var MAX_FILE_SIZE = 1e4;
+var ReadFileTool = class extends BaseTool {
+  definition = {
+    name: "readFile",
+    description: "Read the contents of a file",
+    parameters: {
+      path: {
+        type: "string",
+        description: "Relative path to the file to read"
+      }
+    },
+    required: ["path"]
+  };
+  async execute(args) {
+    const { path } = args;
+    const cwd = process.cwd();
+    const resolved = resolve(cwd, path);
+    const rel = relative(cwd, resolved);
+    if (rel.startsWith("..") || isAbsolute(rel)) {
+      return { error: "Path is outside the project directory" };
+    }
+    try {
+      const content = await readFile(resolved, "utf-8");
+      if (content.length > MAX_FILE_SIZE) {
+        return {
+          content: content.slice(0, MAX_FILE_SIZE),
+          truncated: true,
+          totalLength: content.length
+        };
+      }
+      return { content };
+    } catch (err) {
+      const msg = err.message?.replace(/^ENOENT:\s*/, "") || String(err);
+      return { error: msg };
+    }
+  }
+};
+
+// src/builtins/writeFile.ts
+import { mkdir, readFile as readFile2, writeFile } from "fs/promises";
+import { dirname, relative as relative2, resolve as resolve2, isAbsolute as isAbsolute2 } from "path";
+var WriteFileTool = class extends BaseTool {
+  definition = {
+    name: "writeFile",
+    description: "Create a new file or overwrite an existing one",
+    parameters: {
+      path: {
+        type: "string",
+        description: "Relative path to write"
+      },
+      content: {
+        type: "string",
+        description: "File contents"
+      }
+    },
+    required: ["path", "content"]
+  };
+  async execute(args) {
+    const { path, content } = args;
+    const cwd = process.cwd();
+    const resolved = resolve2(cwd, path);
+    const rel = relative2(cwd, resolved);
+    if (rel.startsWith("..") || isAbsolute2(rel)) {
+      return { error: "Path is outside the project directory" };
+    }
+    try {
+      let oldContent = null;
+      try {
+        oldContent = await readFile2(resolved, "utf-8");
+      } catch {
+      }
+      const backupDir = resolve2(cwd, ".lupaflow/backups");
+      if (oldContent !== null) {
+        const backupPath = resolve2(backupDir, rel);
+        await mkdir(dirname(backupPath), { recursive: true });
+        await writeFile(backupPath, oldContent, "utf-8");
+      }
+      await mkdir(dirname(resolved), { recursive: true });
+      await writeFile(resolved, content, "utf-8");
+      const result = {
+        success: true,
+        path: rel
+      };
+      if (oldContent !== null) {
+        result.diff = computeDiff(oldContent, content);
+      }
+      return result;
+    } catch (err) {
+      return { error: err.message };
+    }
+  }
+};
+function computeDiff(oldStr, newStr) {
+  const oldLines = oldStr.split("\n");
+  const newLines = newStr.split("\n");
+  const lines = [];
+  let i = 0, j = 0;
+  while (i < oldLines.length || j < newLines.length) {
+    if (i < oldLines.length && j < newLines.length && oldLines[i] === newLines[j]) {
+      lines.push(` ${oldLines[i]}`);
+      i++;
+      j++;
+    } else {
+      let found = false;
+      for (let k = 1; k <= 3; k++) {
+        if (i + k < oldLines.length && oldLines[i + k] === newLines[j]) {
+          for (let l = 0; l < k; l++) lines.push(`-${oldLines[i + l]}`);
+          i += k;
+          found = true;
+          break;
+        }
+        if (j + k < newLines.length && oldLines[i] === newLines[j + k]) {
+          for (let l = 0; l < k; l++) lines.push(`+${newLines[j + l]}`);
+          j += k;
+          found = true;
+          break;
+        }
+      }
+      if (!found) {
+        if (i < oldLines.length) lines.push(`-${oldLines[i]}`);
+        if (j < newLines.length) lines.push(`+${newLines[j]}`);
+        i++;
+        j++;
+      }
+    }
+  }
+  return lines.join("\n");
+}
+
+// src/builtins/editFile.ts
+import { mkdir as mkdir2, readFile as readFile3, writeFile as writeFile2 } from "fs/promises";
+import { dirname as dirname2, relative as relative3, resolve as resolve3, isAbsolute as isAbsolute3 } from "path";
+var EditFileTool = class extends BaseTool {
+  definition = {
+    name: "editFile",
+    description: "Replace an exact string match in a file with new content",
+    parameters: {
+      path: {
+        type: "string",
+        description: "Relative path to edit"
+      },
+      oldString: {
+        type: "string",
+        description: "Exact text to replace; must be unique in the file"
+      },
+      newString: {
+        type: "string",
+        description: "Replacement text"
+      }
+    },
+    required: ["path", "oldString", "newString"]
+  };
+  async execute(args) {
+    const { path, oldString, newString } = args;
+    const cwd = process.cwd();
+    const resolved = resolve3(cwd, path);
+    const rel = relative3(cwd, resolved);
+    if (rel.startsWith("..") || isAbsolute3(rel)) {
+      return { error: "Path is outside the project directory" };
+    }
+    try {
+      const content = await readFile3(resolved, "utf-8");
+      const occurrences = content.split(oldString).length - 1;
+      if (occurrences === 0) {
+        return { error: "oldString not found in file" };
+      }
+      if (occurrences > 1) {
+        return { error: `oldString is ambiguous; found ${occurrences} matches` };
+      }
+      const backupDir = resolve3(cwd, ".lupaflow/backups");
+      const backupPath = resolve3(backupDir, rel);
+      await mkdir2(dirname2(backupPath), { recursive: true });
+      await writeFile2(backupPath, content, "utf-8");
+      const newContent = content.replace(oldString, newString);
+      await writeFile2(resolved, newContent, "utf-8");
+      return {
+        success: true,
+        path: rel,
+        diff: computeDiff2(content, newContent)
+      };
+    } catch (err) {
+      return { error: err.message };
+    }
+  }
+};
+function computeDiff2(oldStr, newStr) {
+  const oldLines = oldStr.split("\n");
+  const newLines = newStr.split("\n");
+  const lines = [];
+  let i = 0, j = 0;
+  while (i < oldLines.length || j < newLines.length) {
+    if (i < oldLines.length && j < newLines.length && oldLines[i] === newLines[j]) {
+      lines.push(` ${oldLines[i]}`);
+      i++;
+      j++;
+    } else {
+      let found = false;
+      for (let k = 1; k <= 3; k++) {
+        if (i + k < oldLines.length && oldLines[i + k] === newLines[j]) {
+          for (let l = 0; l < k; l++) lines.push(`-${oldLines[i + l]}`);
+          i += k;
+          found = true;
+          break;
+        }
+        if (j + k < newLines.length && oldLines[i] === newLines[j + k]) {
+          for (let l = 0; l < k; l++) lines.push(`+${newLines[j + l]}`);
+          j += k;
+          found = true;
+          break;
+        }
+      }
+      if (!found) {
+        if (i < oldLines.length) lines.push(`-${oldLines[i]}`);
+        if (j < newLines.length) lines.push(`+${newLines[j]}`);
+        i++;
+        j++;
+      }
+    }
+  }
+  return lines.join("\n");
+}
+
+// src/builtins/deleteFile.ts
+import { unlink } from "fs/promises";
+import { relative as relative4, resolve as resolve4, isAbsolute as isAbsolute4 } from "path";
+var DeleteFileTool = class extends BaseTool {
+  definition = {
+    name: "deleteFile",
+    description: "Delete a file",
+    parameters: {
+      path: {
+        type: "string",
+        description: "Relative path to the file to delete"
+      }
+    },
+    required: ["path"]
+  };
+  async execute(args) {
+    const { path } = args;
+    const cwd = process.cwd();
+    const resolved = resolve4(cwd, path);
+    const rel = relative4(cwd, resolved);
+    if (rel.startsWith("..") || isAbsolute4(rel)) {
+      return { error: "Path is outside the project directory" };
+    }
+    try {
+      await unlink(resolved);
+      return { success: true, path: rel };
+    } catch (err) {
+      return { error: err.message };
+    }
+  }
+};
+
+// src/builtins/listDirectory.ts
+import { readdir, stat } from "fs/promises";
+import { join, relative as relative5, resolve as resolve5, isAbsolute as isAbsolute5 } from "path";
+var ListDirectoryTool = class extends BaseTool {
+  definition = {
+    name: "listDirectory",
+    description: "List entries in a directory",
+    parameters: {
+      path: {
+        type: "string",
+        description: "Relative directory path to list"
+      }
+    },
+    required: ["path"]
+  };
+  async execute(args) {
+    const { path } = args;
+    const cwd = process.cwd();
+    const resolved = resolve5(cwd, path);
+    const rel = relative5(cwd, resolved);
+    if (rel.startsWith("..") || isAbsolute5(rel)) {
+      return { error: "Path is outside the project directory" };
+    }
+    try {
+      const entries = await readdir(resolved);
+      const results = [];
+      for (const entry of entries) {
+        if (entry.startsWith(".") || entry === "node_modules") continue;
+        try {
+          const info = await stat(join(resolved, entry));
+          results.push({
+            name: entry,
+            type: info.isDirectory() ? "directory" : "file"
+          });
+        } catch {
+        }
+      }
+      results.sort(
+        (a, b) => a.type !== b.type ? a.type === "directory" ? -1 : 1 : a.name.localeCompare(b.name)
+      );
+      return { path: rel || ".", entries: results };
+    } catch (err) {
+      return { error: err.message };
+    }
+  }
+};
+
+// src/builtins/glob.ts
+import { relative as relative6, resolve as resolve6, isAbsolute as isAbsolute6 } from "path";
+var MAX_RESULTS = 200;
+var GlobTool = class extends BaseTool {
+  definition = {
+    name: "glob",
+    description: "Find files matching a glob pattern",
+    parameters: {
+      pattern: {
+        type: "string",
+        description: "Glob pattern to match files"
+      },
+      path: {
+        type: "string",
+        description: "Directory to search from (default: .)"
+      }
+    },
+    required: ["pattern"]
+  };
+  async execute(args) {
+    const { pattern, path: dir = "." } = args;
+    const cwd = process.cwd();
+    const resolved = resolve6(cwd, dir);
+    const rel = relative6(cwd, resolved);
+    if (rel.startsWith("..") || isAbsolute6(rel)) {
+      return { error: "Path is outside the project directory" };
+    }
+    try {
+      const glob = new Bun.Glob(pattern);
+      const files = [];
+      let truncated = false;
+      for await (const match of glob.scan({ cwd: resolved, dot: false, onlyFiles: true })) {
+        if (match.includes("node_modules")) continue;
+        if (files.length >= MAX_RESULTS) {
+          truncated = true;
+          break;
+        }
+        files.push(relative6(cwd, resolve6(resolved, match)));
+      }
+      files.sort();
+      const result = { files };
+      if (truncated) result.truncated = true;
+      return result;
+    } catch (err) {
+      return { error: err.message };
+    }
+  }
+};
+
+// src/builtins/grep.ts
+import { relative as relative7, resolve as resolve7, isAbsolute as isAbsolute7 } from "path";
+var MAX_MATCHES = 50;
+var GrepTool = class extends BaseTool {
+  definition = {
+    name: "grep",
+    description: "Search file contents using a regex pattern",
+    parameters: {
+      pattern: {
+        type: "string",
+        description: "Regex pattern to search for"
+      },
+      path: {
+        type: "string",
+        description: "Directory to search from (default: .)"
+      },
+      include: {
+        type: "string",
+        description: "Optional glob for files to include (e.g. *.ts)"
+      }
+    },
+    required: ["pattern"]
+  };
+  async execute(args) {
+    const { pattern, path: dir = ".", include } = args;
+    const cwd = process.cwd();
+    const resolved = resolve7(cwd, dir);
+    const rel = relative7(cwd, resolved);
+    if (rel.startsWith("..") || isAbsolute7(rel)) {
+      return { error: "Path is outside the project directory" };
+    }
+    try {
+      const grepArgs = [
+        "-rn",
+        "--color=never",
+        "--exclude-dir=node_modules",
+        "--exclude-dir=.git",
+        "-E"
+      ];
+      if (include) grepArgs.push(`--include=${include}`);
+      grepArgs.push(pattern, resolved);
+      const proc = Bun.spawn(["grep", ...grepArgs], {
+        cwd,
+        stdout: "pipe",
+        stderr: "pipe"
+      });
+      const [stdout, stderr] = await Promise.all([
+        new Response(proc.stdout).text(),
+        new Response(proc.stderr).text()
+      ]);
+      const exitCode = await proc.exited;
+      if (exitCode !== 0 && exitCode !== 1) {
+        const msg = stderr.trim() || `grep exited with code ${exitCode}`;
+        return { error: msg };
+      }
+      const trimmed = stderr.trim();
+      if (trimmed) {
+        return { error: trimmed };
+      }
+      const lines = stdout.trim().split("\n");
+      if (!lines[0]) {
+        return { matches: [], message: "No matches found" };
+      }
+      const matches = [];
+      let truncated = false;
+      for (const line of lines) {
+        if (matches.length >= MAX_MATCHES) {
+          truncated = true;
+          break;
+        }
+        const match = line.match(/^(.+?):(\d+):(.*)$/);
+        if (match) {
+          matches.push({
+            file: relative7(cwd, match[1]),
+            line: Number(match[2]),
+            content: match[3]
+          });
+        }
+      }
+      const result = { matches };
+      if (truncated) {
+        result.truncated = true;
+        result.totalMatches = lines.length;
+      }
+      return result;
+    } catch (err) {
+      return { error: err.message || String(err) };
+    }
+  }
+};
+
+// src/builtins/shell.ts
+var MAX_OUTPUT = 2e4;
+var DEFAULT_TIMEOUT = 3e4;
+var ShellTool = class extends BaseTool {
+  definition = {
+    name: "shell",
+    description: "Execute a shell command",
+    parameters: {
+      command: {
+        type: "string",
+        description: "Shell command to run"
+      },
+      description: {
+        type: "string",
+        description: "Short description of what the command does"
+      },
+      timeout: {
+        type: "number",
+        description: "Timeout in milliseconds (default: 30000)"
+      }
+    },
+    required: ["command"]
+  };
+  async execute(args) {
+    const { command, timeout = DEFAULT_TIMEOUT } = args;
+    try {
+      const isWindows = process.platform === "win32";
+      const shell = isWindows ? "cmd.exe" : "bash";
+      const shellFlag = isWindows ? "/c" : "-c";
+      const env = isWindows ? { ...process.env } : { ...process.env, TERM: "dumb" };
+      const proc = Bun.spawn([shell, shellFlag, command], {
+        cwd: process.cwd(),
+        stdout: "pipe",
+        stderr: "pipe",
+        env
+      });
+      const timer = setTimeout(() => proc.kill(), timeout);
+      const [stdout, stderr] = await Promise.all([
+        new Response(proc.stdout).text(),
+        new Response(proc.stderr).text()
+      ]);
+      const exitCode = await proc.exited;
+      clearTimeout(timer);
+      return {
+        stdout: truncate(stdout, MAX_OUTPUT),
+        stderr: truncate(stderr, MAX_OUTPUT),
+        exitCode
+      };
+    } catch (err) {
+      return {
+        stdout: "",
+        stderr: err.message || String(err),
+        exitCode: 1
+      };
+    }
+  }
+};
+function truncate(value, limit) {
+  return value.length > limit ? `${value.slice(0, limit)}
+... (truncated, ${value.length} total chars)` : value;
+}
+
+// src/index.ts
+toolRegistry.register(new CalculatorTool());
+toolRegistry.register(new HTTPTool());
+toolRegistry.register(new SearchTool());
+toolRegistry.register(new ReadFileTool());
+toolRegistry.register(new WriteFileTool());
+toolRegistry.register(new EditFileTool());
+toolRegistry.register(new DeleteFileTool());
+toolRegistry.register(new ListDirectoryTool());
+toolRegistry.register(new GlobTool());
+toolRegistry.register(new GrepTool());
+toolRegistry.register(new ShellTool());
+export {
+  BaseTool,
+  CalculatorTool,
+  DeleteFileTool,
+  EditFileTool,
+  GlobTool,
+  GrepTool,
+  HTTPTool,
+  ListDirectoryTool,
+  ReadFileTool,
+  SearchTool,
+  ShellTool,
+  WriteFileTool,
+  toolRegistry
+};
+//# sourceMappingURL=index.js.map