import { type } from "arktype";
import { $ } from "../utils/shell.ts";
import { contextualize, tool } from "./shared.ts";

export const ListFiles = type({
  path: type.string
    .describe("The path to list files from (defaults to current directory)")
    .default("."),
});

export const ListFilesTool = tool({
  name: "list_files",
  description:
    "List files in the repository using git ls-files. Useful for discovering the file structure and locating files.",
  parameters: ListFiles,
  execute: contextualize(async ({ path }) => {
    try {
      // Use git ls-files to list tracked files
      // This respects .gitignore and gives a clean list of source files
      const pathStr = path ?? ".";
      const output = $("git", pathStr === "." ? ["ls-files"] : ["ls-files", pathStr], {
        log: false,
      });
      const files = output.split("\n").filter((f) => f.trim() !== "");

      if (files.length === 0) {
        // Fallback for non-git environments or untracked files
        const findOutput = $(
          "find",
          [pathStr, "-maxdepth", "3", "-not", "-path", "*/.*", "-type", "f"],
          { log: false }
        );
        return {
          files: findOutput.split("\n").filter((f) => f.trim() !== ""),
          method: "find",
        };
      }

      return { files, method: "git" };
    } catch (error) {
      const errorMessage = error instanceof Error ? error.message : String(error);
      return {
        error: `Failed to list files: ${errorMessage}`,
        hint: "Try using a specific path if the repository root is not the current directory.",
      };
    }
  }),
});