// import type { RsbuildPlugin } from '@rsbuild/core';
import glob from 'fast-glob';
import path from 'node:path';
import type { Group } from './components/Overview';

const camelCase = (input: string): string =>
  input.replace(/[-_](\w)/g, (_, c) => c.toUpperCase());

export const moduleFederationPluginOverview: any = {
  name: 'module-federation-doc:overview',

  async setup(api) {
    const root = path.join(__dirname, '../docs/zh/configure/');
    const globPath = path.join(root, '**/*.{mdx,md}');

    const files = await glob(globPath);
    const groups: Group[] = [];

    for (const file of files) {
      const filename = file.replace(root, '').replace(/\.mdx?/, '');
      const pair = filename.split('/');

      if (pair.length < 2) {
        continue;
      }

      const group = groups.find((group) => group.name === pair[0]);
      const item = {
        text: `${camelCase(pair[0])}.${camelCase(pair[1])}`,
        link: `/configure/${filename}`,
      };
      if (group) {
        group.items.push(item);
      } else {
        groups.push({
          name: camelCase(pair[0]),
          items: [item],
        });
      }
    }

    const order = ['name'];

    groups.sort((a, b) => order.indexOf(a.name) - order.indexOf(b.name));

    api.modifyRsbuildConfig((config, { mergeRsbuildConfig }) => {
      return mergeRsbuildConfig(config, {
        source: {
          define: {
            OVERVIEW_GROUPS: JSON.stringify(groups),
          },
        },
      });
    });
  },
};