import { execSync } from "node:child_process";
import fs from "node:fs";
import path from "node:path";
import { fileURLToPath } from "node:url";

const __dirname = path.dirname(fileURLToPath(import.meta.url));
const root = path.resolve(__dirname, "../..");
const distDir = path.join(root, "template", "dist");
const templateDir = path.join(root, "template");

const ICON_PRIORITY = ["logo.png", "logo.jpg", "logo.jpeg", "favicon.ico"];

const BUNDLE_ICONS = [
  "icons/32x32.png",
  "icons/128x128.png",
  "icons/128x128@2x.png",
  "icons/icon.icns",
  "icons/icon.ico",
];

function findIconSource(baseDir) {
  if (!fs.existsSync(baseDir)) return null;

  for (const name of ICON_PRIORITY) {
    const direct = path.join(baseDir, name);
    if (fs.existsSync(direct)) return direct;
  }

  for (const entry of fs.readdirSync(baseDir, { withFileTypes: true })) {
    if (!entry.isDirectory()) continue;
    for (const name of ICON_PRIORITY) {
      const nested = path.join(baseDir, entry.name, name);
      if (fs.existsSync(nested)) return nested;
    }
  }

  return null;
}

export function findIconSourceForJob(root, jobId, distDir) {
  const buildDir = jobId ? path.join(root, "builds", jobId) : null;
  if (buildDir && fs.existsSync(buildDir)) {
    const fromUpload = findIconSource(buildDir);
    if (fromUpload) return fromUpload;
  }
  return findIconSource(distDir);
}

export function generateAppIcons(options = {}) {
  const { jobId } = options;
  if (!fs.existsSync(distDir)) {
    console.log("No dist directory, keeping default icons");
    return false;
  }

  const iconSource = findIconSourceForJob(root, jobId, distDir);
  if (!iconSource) {
    console.log("No logo.png or favicon.ico found, keeping default icons");
    return false;
  }

  console.log(`Generating app icons from: ${path.relative(root, iconSource)}`);
  execSync(`npx tauri icon "${iconSource}"`, {
    cwd: templateDir,
    stdio: "inherit",
  });
  return true;
}

export { BUNDLE_ICONS };

const isCli =
  process.argv[1] &&
  path.resolve(process.argv[1]) === fileURLToPath(import.meta.url);

if (isCli) {
  generateAppIcons({ jobId: process.env.JOB_ID });
}