import ignore from "ignore";
import { isBinaryFile } from "isbinaryfile";
import { countTokens } from "./tokenizer";
import path from "path";

// Multiplier to convert token counts from Claude's tokenizer to OpenAI's tokenizer.
// Derived from external calculations documented in [link-to-docs-or-script].
const CLAUDE_TO_OPENAI_MULTIPLIER = 0.9048;

export const WHITESPACE_DEPENDENT_EXTENSIONS = [
  ".py", // Python
  ".yaml", // YAML
  ".yml", // YAML
  ".jade", // Jade/Pug
  ".haml", // Haml
  ".slim", // Slim
  ".coffee", // CoffeeScript
  ".pug", // Pug
  ".styl", // Stylus
  ".gd", // Godot
];

export const DEFAULT_IGNORES = [
  ".aidigestignore",
  ".aidigestminify",
  // Node.js
  "node_modules",
  "package-lock.json",
  "npm-debug.log",
  // Yarn
  "yarn.lock",
  "yarn-error.log",
  // pnpm
  "pnpm-lock.yaml",
  // Bun
  "bun.lockb",
  "bun.lock",
  // Deno
  "deno.lock",
  // PHP (Composer)
  "vendor",
  "composer.lock",
  // Python
  "__pycache__",
  "*.pyc",
  "*.pyo",
  "*.pyd",
  ".Python",
  "pip-log.txt",
  "pip-delete-this-directory.txt",
  ".venv",
  "venv",
  "ENV",
  "env",
  "uv.lock",
  // Godot
  ".godot",
  "*.import",
  // Ruby
  "Gemfile.lock",
  ".bundle",
  // Java
  "target",
  "*.class",
  // Gradle
  ".gradle",
  "build",
  // Maven
  "pom.xml.tag",
  "pom.xml.releaseBackup",
  "pom.xml.versionsBackup",
  "pom.xml.next",
  // .NET
  "bin",
  "obj",
  "*.suo",
  "*.user",
  // Go
  "go.sum",
  // Rust
  "Cargo.lock",
  "target",
  // General
  ".git",
  ".svn",
  ".hg",
  ".DS_Store",
  "Thumbs.db",
  // Environment variables
  ".env",
  ".env.local",
  ".env.development.local",
  ".env.test.local",
  ".env.production.local",
  "*.env",
  "*.env.*",
  // Common framework directories
  ".svelte-kit",
  ".next",
  ".nuxt",
  ".vuepress",
  ".cache",
  "dist",
  "tmp",
  // Our output file
  "codebase.md",
  // Turborepo cache folder
  ".turbo",
  ".vercel",
  ".netlify",
  "LICENSE",
];

export function removeWhitespace(val: string): string {
  return val.replace(/\s+/g, " ").trim();
}

export function escapeTripleBackticks(content: string): string {
  return content.replace(/\`\`\`/g, "\\`\\`\\`");
}

export function createIgnoreFilter(
  ignorePatterns: string[],
  ignoreFile: string,
  silent: boolean = false,
): ReturnType<typeof ignore> {
  const ig = require("ignore")().add(ignorePatterns);
  if (!silent) {
    if (ignorePatterns.length > 0) {
      console.log(`Ignore patterns from ${ignoreFile}:`);
      ignorePatterns.forEach((pattern) => {
        console.log(`  - ${pattern}`);
      });
    } else {
      console.log("No custom ignore patterns found.");
    }
  }
  return ig;
}

export function estimateTokenCount(text: string): {
  gptTokens: number;
  claudeTokens: number;
} {
  try {
    const claudeTokens = countTokens(text);
    const estimatedGptTokens = Math.round(
      claudeTokens * CLAUDE_TO_OPENAI_MULTIPLIER,
    );

    return {
      gptTokens: estimatedGptTokens,
      claudeTokens,
    };
  } catch (error) {
    console.error("Error estimating token count:", error);
    return { gptTokens: 0, claudeTokens: 0 };
  }
}

export function formatLog(message: string, emoji: string = ""): string {
  return `${emoji ? emoji + " " : ""}${message}`;
}

export async function isTextFile(filePath: string): Promise<boolean> {
  try {
    const isBinary = await isBinaryFile(filePath);
    return !isBinary && !filePath.toLowerCase().endsWith(".svg");
  } catch (error) {
    console.error(`Error checking if file is binary: ${filePath}`, error);
    return false;
  }
}

export function getFileType(filePath: string): string {
  const extension = path.extname(filePath).toLowerCase();
  switch (extension) {
  case ".jpg":
  case ".jpeg":
  case ".png":
  case ".gif":
  case ".bmp":
  case ".webp":
    return "Image";
  case ".svg":
    return "SVG Image";
  case ".wasm":
    return "WebAssembly";
  case ".pdf":
    return "PDF";
  case ".doc":
  case ".docx":
    return "Word Document";
  case ".xls":
  case ".xlsx":
    return "Excel Spreadsheet";
  case ".ppt":
  case ".pptx":
    return "PowerPoint Presentation";
  case ".zip":
  case ".rar":
  case ".7z":
    return "Compressed Archive";
  case ".exe":
    return "Executable";
  case ".dll":
    return "Dynamic-link Library";
  case ".so":
    return "Shared Object";
  case ".dylib":
    return "Dynamic Library";
  default:
    return "Binary";
  }
}

export function shouldTreatAsBinary(filePath: string): boolean {
  return (
    filePath.toLowerCase().endsWith(".svg") ||
    getFileType(filePath) !== "Binary"
  );
}

// Functions moved from index.ts
export function getActualWorkingDirectory(): string {
  // INIT_CWD is set by npm/npx to the directory from which the command was invoked
  // This is more reliable than process.cwd() when running via npx
  if (process.env.INIT_CWD) {
    return process.env.INIT_CWD;
  }

  return process.cwd();
}

// Simple debounce function to avoid multiple rebuilds when many files change at once
export function debounce<F extends (...args: unknown[]) => unknown>(
  func: F,
  wait: number,
): (...args: Parameters<F>) => void {
  let timeout: NodeJS.Timeout | null = null;

  return function (...args: Parameters<F>) {
    if (timeout) {
      clearTimeout(timeout);
    }

    timeout = setTimeout(() => {
      func(...args);
      timeout = null;
    }, wait);
  };
}

export function formatFileSize(bytes: number): string {
  if (bytes < 1024) return `${bytes} B`;
  if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)} KB`;
  return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
}

export function naturalSort(a: string, b: string): number {
  return a.localeCompare(b, undefined, { numeric: true, sensitivity: "base" });
}