generate.mjs

'use strict';

import { readFile, cp, mkdir } from 'node:fs/promises';
import { join } from 'node:path';

import { processJSXEntries } from './utils/processing.mjs';
import getConfig from '../../utils/configuration/index.mjs';
import { writeFile } from '../../utils/file.mjs';

/**
 * Aggregates and copies local assets referenced in the markdown ASTs to the output directory.
 *
 * @param {Array<import('./types').JSXContent>} input - The processed entries containing asset maps
 * @param {string} outputDir - The absolute path to the generation output directory
 * @returns {Promise<void>}
 */
async function copyProjectAssets(input, outputDir) {
  const allAssets = new Map();

  for (const entry of input) {
    if (entry.assetsMap) {
      for (const [source, name] of Object.entries(entry.assetsMap)) {
        allAssets.set(source, join(outputDir, 'assets', name));
      }
    }
  }

  if (allAssets.size === 0) {
    return;
  }

  const assetsOutputDir = join(outputDir, 'assets');
  await mkdir(assetsOutputDir, { recursive: true });

  for (const [source, dest] of allAssets.entries()) {
    try {
      await cp(source, dest, { force: true });
    } catch (err) {
      console.error(`[doc-kit] Error copying asset: ${source}`, err);
    }
  }
}

/**
 * Main generation function that processes JSX AST entries into web bundles.
 *
 * @type {import('./types').Generator['generate']}
 */
export async function generate(input) {
  const config = getConfig('web');

  const template = await readFile(config.templatePath, 'utf-8');

  // Process all entries: convert JSX to HTML/CSS/JS
  const { results, css, chunks } = await processJSXEntries(input, template);

  // Process all entries together (required for code-split bundles)
  if (config.output) {
    // Write HTML files
    for (const { html, path } of results) {
      await writeFile(join(config.output, `${path}.html`), html, 'utf-8');
    }

    // Write code-split JavaScript chunks
    for (const chunk of chunks) {
      await writeFile(join(config.output, chunk.fileName), chunk.code, 'utf-8');
    }

    // Write CSS bundle
    await writeFile(join(config.output, 'styles.css'), css, 'utf-8');

    await copyProjectAssets(input, config.output);
  }

  return results.map(({ html }) => ({ html: html.toString(), css }));
}