generate.mjs

'use strict';

import { join } from 'node:path';

import { create, save, insertMultiple } from '@orama/orama';

import { SCHEMA } from './constants.mjs';
import { buildHierarchicalTitle } from './utils/title.mjs';
import getConfig from '../../utils/configuration/index.mjs';
import { writeFile } from '../../utils/file.mjs';
import { groupNodesByModule } from '../../utils/generators.mjs';
import { transformNodeToString } from '../../utils/unist.mjs';

/**
 * Generates the Orama database.
 *
 * @type {import('./types').Generator['generate']}
 */
export async function generate(input) {
  const config = getConfig('orama-db');

  const db = create({ schema: SCHEMA });

  const apiGroups = groupNodesByModule(input);

  // Process all API groups and flatten into a single document array
  const documents = Array.from(apiGroups.values()).flatMap(headings =>
    headings.map((entry, index) => {
      const hierarchicalTitle = buildHierarchicalTitle(headings, index);

      const paragraph = entry.content.children.find(
        child => child.type === 'paragraph'
      );

      return {
        title: hierarchicalTitle,
        description: paragraph
          ? transformNodeToString(paragraph, true)
          : undefined,
        href: `${entry.path}.html#${entry.heading.data.slug}`,
        siteSection: headings[0].heading.data.name,
      };
    })
  );

  // Insert all documents
  await insertMultiple(db, documents);

  const result = save(db);

  // Persist
  if (config.output) {
    await writeFile(
      join(config.output, 'orama-db.json'),
      config.minify ? JSON.stringify(result) : JSON.stringify(result, null, 2)
    );
  }

  return result;
}