const { mkdir, rmdir, readdir, stat, readFile, writeFile } = require('fs').promises;
const { join, dirname } = require('path');
const marked = require('./');
const { highlight, highlightAuto } = require('highlight.js');
const cwd = process.cwd();
const inputDir = join(cwd, 'docs');
const outputDir = join(cwd, 'public');
const templateFile = join(inputDir, '_document.html');

async function init() {
  console.log('Cleaning up output directory ' + outputDir);
  await rmdir(outputDir, { recursive: true });
  await mkdir(outputDir);
  const tmpl = await readFile(templateFile, 'utf8');
  console.log('Building markdown...');
  await build(inputDir, tmpl);
  console.log('Build complete!');
}

async function build(currentDir, tmpl) {
  const files = await readdir(currentDir);
  for (const file of files) {
    const filename = join(currentDir, file);
    const stats = await stat(filename);
    const { mode } = stats;
    if (stats.isDirectory()) {
      //console.log('Found directory ' + filename);
      await build(filename, tmpl);
    } else {
      //console.log('Reading file ' + filename);
      let contents = await readFile(filename, 'utf8');
      if (filename.endsWith('.md')) {
        const html = marked(contents, {
          highlight: (code, lang) => {
            if (!lang) {
              return highlightAuto(code).value;
            }
            return highlight(lang, code).value;
          },
        });
        contents = tmpl.replace('<!--{{content}}-->', html);
      }
      const outfile = filename.replace(inputDir, outputDir).replace('.md', '.html').toLowerCase();
      //console.log('Ensure directory ' + dirname(outfile));
      await mkdir(dirname(outfile), { recursive: true });
      //console.log('Writing file ' + outfile);
      await writeFile(outfile, contents, { mode });
    }
  }
}

init().catch(console.error);