import { cpSync, existsSync, readdirSync } from 'node:fs';
import path from 'node:path';
import { fileURLToPath } from 'node:url';
import { defineConfig } from 'vite';

const root = path.dirname(fileURLToPath(import.meta.url));
const excludedDirs = new Set(['.git', 'dist', 'node_modules']);
const rodaDashboardUrl = 'https://dashboard.informatika.untan.ac.id/index.php?penelitian';

let rodaDashboardHtmlCache;

function collectHtmlInputs(dir) {
  const inputs = {};
  const entries = readdirSync(dir, { withFileTypes: true });

  for (const entry of entries) {
    const fullPath = path.join(dir, entry.name);

    if (entry.isDirectory()) {
      if (!excludedDirs.has(entry.name)) {
        Object.assign(inputs, collectHtmlInputs(fullPath));
      }
      continue;
    }

    if (entry.isFile() && entry.name.endsWith('.html')) {
      const relativePath = path.relative(root, fullPath);
      const inputName = relativePath
        .replace(/\.html$/, '')
        .replace(/[^a-zA-Z0-9_-]+/g, '_');

      inputs[inputName] = fullPath;
    }
  }

  return inputs;
}

function copyStaticAssets() {
  return {
    name: 'copy-static-assets',
    closeBundle() {
      const source = path.join(root, 'assets');
      const destination = path.join(root, 'dist', 'assets');

      if (existsSync(source)) {
        cpSync(source, destination, { recursive: true });
      }
    },
  };
}

async function fetchRodaDashboardHtml() {
  if (rodaDashboardHtmlCache !== undefined) {
    return rodaDashboardHtmlCache;
  }

  const controller = new AbortController();
  const timeoutId = setTimeout(() => controller.abort(), 12000);

  try {
    const response = await fetch(rodaDashboardUrl, {
      signal: controller.signal,
      headers: {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0 Safari/537.36',
      },
    });

    if (!response.ok) {
      throw new Error(`HTTP ${response.status}`);
    }

    rodaDashboardHtmlCache = await response.text();
  } catch (error) {
    const message = error instanceof Error ? error.message : String(error);
    console.warn(`[roda-dashboard] Build-time dashboard fetch skipped: ${message}`);
    rodaDashboardHtmlCache = '';
  } finally {
    clearTimeout(timeoutId);
  }

  return rodaDashboardHtmlCache;
}

function escapeJsonScriptContent(value) {
  return JSON.stringify(value)
    .replaceAll('<', '\\u003c')
    .replaceAll('>', '\\u003e')
    .replaceAll('&', '\\u0026')
    .replaceAll('\u2028', '\\u2028')
    .replaceAll('\u2029', '\\u2029');
}

function embedRodaDashboardData() {
  let command = 'serve';

  return {
    name: 'embed-roda-dashboard-data',
    configResolved(config) {
      command = config.command;
    },
    async transformIndexHtml(html, context) {
      const filename = context?.filename ? path.resolve(context.filename) : '';
      const rodaIndex = path.join(root, 'groups', 'RODA', 'index.html');

      if (filename !== rodaIndex || command !== 'build') {
        return html.replace('<!-- RODA_DASHBOARD_DATA -->', '');
      }

      const dashboardHtml = await fetchRodaDashboardHtml();
      const payload = `<script type="application/json" id="roda-dashboard-html">${escapeJsonScriptContent(dashboardHtml)}</script>`;

      return html.replace('<!-- RODA_DASHBOARD_DATA -->', payload);
    },
  };
}

export default defineConfig({
  plugins: [embedRodaDashboardData(), copyStaticAssets()],
  build: {
    rollupOptions: {
      input: collectHtmlInputs(root),
    },
  },
});
