import { tool } from "@opencode-ai/plugin";
import { $ } from "bun";

async function run(args: string[]): Promise<string> {
  const result = await $`npx pdf-brain ${args}`.text();
  return result.trim();
}

export const add = tool({
  description: "Add a PDF to the library",
  args: {
    path: tool.schema.string().describe("Path to PDF file or URL"),
    tags: tool.schema.string().optional().describe("Comma-separated tags"),
  },
  async execute({ path, tags }) {
    const args = ["add", path];
    if (tags) args.push("--tags", tags);
    return run(args);
  },
});

export const search = tool({
  description: "Semantic search across all PDFs",
  args: {
    query: tool.schema.string().describe("Natural language search query"),
    limit: tool.schema.number().optional().describe("Max results"),
    fts: tool.schema.boolean().optional().describe("Full-text search only"),
  },
  async execute({ query, limit, fts }) {
    const args = ["search", query];
    if (limit) args.push("--limit", String(limit));
    if (fts) args.push("--fts");
    return run(args);
  },
});

export const list = tool({
  description: "List all PDFs in the library",
  args: { tag: tool.schema.string().optional() },
  async execute({ tag }) {
    const args = ["list"];
    if (tag) args.push("--tag", tag);
    return run(args);
  },
});

export const stats = tool({
  description: "Show library statistics",
  args: {},
  async execute() {
    return run(["stats"]);
  },
});