import { readFileSync, writeFileSync, readdirSync } from 'fs';
import { join, dirname } from 'path';
import { fileURLToPath } from 'url';

const __dirname = dirname(fileURLToPath(import.meta.url));
const root = join(__dirname, '..');

const distIndex = join(root, 'dist', 'index.html');
const rootIndex = join(root, 'index.html');

function getMainJs() {
  const assetsDir = join(root, 'dist', 'assets');
  const files = readdirSync(assetsDir).filter(f => f.startsWith('main-') && f.endsWith('.js'));
  if (files.length === 0) throw new Error('No main-*.js found in dist/assets');
  return `/assets/${files[0]}`;
}

function patchIndexHtml(filePath, mainJs) {
  let content = readFileSync(filePath, 'utf8');
  content = content.replace(
    '<div id="root"></div>',
    `<div id="root"></div>\n  <script type="module" src="${mainJs}"></script>`
  );
  writeFileSync(filePath, content);
  console.log(`Patched: ${filePath} -> script: ${mainJs}`);
}

const mainJs = getMainJs();
patchIndexHtml(distIndex, mainJs);
patchIndexHtml(rootIndex, mainJs);
console.log(`Done. Main JS: ${mainJs}`);