import { existsSync, readFileSync } from "fs";
import { join, dirname, basename } from "path";
import { fileURLToPath } from "url";
import { loadR2Config, uploadFile } from "../src/index.js";

function loadEnvLocal(): void {
  const scriptDir = dirname(fileURLToPath(import.meta.url));
  const envPath = join(scriptDir, "../.env.local");
  if (!existsSync(envPath)) return;
  for (const line of readFileSync(envPath, "utf-8").split("\n")) {
    const trimmed = line.trim();
    if (!trimmed || trimmed.startsWith("#")) continue;
    const eq = trimmed.indexOf("=");
    if (eq <= 0) continue;
    const key = trimmed.slice(0, eq).trim();
    let val = trimmed.slice(eq + 1).trim();
    if ((val.startsWith('"') && val.endsWith('"')) || (val.startsWith("'") && val.endsWith("'"))) {
      val = val.slice(1, -1);
    }
    if (!process.env[key]) process.env[key] = val;
  }
}

function parseArgs(args: string[]): { file: string; key: string; contentType: string; dryRun: boolean } {
  let file = "";
  let key = "";
  let contentType = "";
  let dryRun = false;

  for (const arg of args) {
    if (arg === "--dry-run") { dryRun = true; continue; }
    if (arg === "-h" || arg === "--help") {
      console.error("Usage: bun run upload -- --file=<path> [--key=<r2-key>] [--content-type=<mime>] [--dry-run]");
      process.exit(0);
    }
    if (arg.startsWith("--file=")) { file = arg.slice("--file=".length).trim(); continue; }
    if (arg.startsWith("--key=")) { key = arg.slice("--key=".length).trim(); continue; }
    if (arg.startsWith("--content-type=")) { contentType = arg.slice("--content-type=".length).trim(); continue; }
  }
  return { file, key, contentType, dryRun };
}

async function main(): Promise<void> {
  loadEnvLocal();
  const args = parseArgs(process.argv.slice(2));

  if (!args.file) {
    console.error(JSON.stringify({ status: "failed", error: "missing --file" }));
    process.exit(1);
  }

  const key = args.key || `uploads/${Date.now()}/${basename(args.file)}`;

  if (args.dryRun) {
    console.log(JSON.stringify({ status: "success", error: null, key, url: `(dry-run) ${key}`, size: 0, contentType: args.contentType || "auto" }, null, 2));
    return;
  }

  const config = loadR2Config();
  const result = await uploadFile(config, args.file, key, args.contentType || undefined);
  console.log(JSON.stringify(result, null, 2));
  if (result.status === "failed") process.exit(1);
}

main().catch((err) => {
  console.error(JSON.stringify({ status: "failed", error: err instanceof Error ? err.message : String(err) }));
  process.exit(1);
});