import type { APIRoute } from 'astro';
import {
  getDocsContext,
  getEntrySectionFromContext,
  getEntrySlug,
  getSidebarSectionTitle,
} from '../lib/docs';
import { getLocalizedHref, getLocales } from '../lib/i18n';
import rawThemeConfig from 'virtual:prosefly/lotus/config';
import type { LotusThemeConfig } from '../lib/theme';
import { getOpenApiSearchItems } from '../lib/openapi/content';

const themeConfig = rawThemeConfig as LotusThemeConfig;

function stripMdx(source: string): string {
  return source
    .replace(/```[\s\S]*?```/g, ' ')
    .replace(/~~~[\s\S]*?~~~/g, ' ')
    .replace(/^\s*import\s.+$/gm, ' ')
    .replace(/^\s*export\s.+$/gm, ' ')
    .replace(/\{\/\*[\s\S]*?\*\/\}/g, ' ')
    .replace(/<[^>]+>/g, ' ')
    .replace(/!\[[^\]]*]\([^)]+\)/g, ' ')
    .replace(/\[([^\]]+)]\([^)]+\)/g, '$1')
    .replace(/[`*_~>#|[\]{}]/g, ' ')
    .replace(/^-{3,}$/gm, ' ')
    .replace(/\s+/g, ' ')
    .trim();
}

function createExcerpt(text: string, maxLength = 180): string {
  if (text.length <= maxLength) {
    return text;
  }

  return `${text.slice(0, maxLength).trim()}...`;
}

export async function getStaticPaths() {
  return getLocales(themeConfig)
    .filter((locale) => locale.pathPrefix)
    .map((locale) => ({
      params: { locale: locale.pathPrefix },
      props: { localeKey: locale.key },
    }));
}

export const GET: APIRoute = async ({ props }) => {
  const localeKey = props.localeKey as string | undefined;
  const docsContext = await getDocsContext(undefined, localeKey);
  const entries = docsContext.entries.filter((entry) => entry.data.pagefind !== false);
  const sectionLabels = new Map(
    docsContext.sections.map((section) => [section.slug, section.label]),
  );
  const items = entries.map((entry) => {
    const entryLocaleKey = docsContext.locale.key;
    const section = getEntrySectionFromContext(docsContext, entry);
    const sidebar = section ? docsContext.sidebars[section] : undefined;
    const entrySlug = getEntrySlug(entry);
    const sectionTitle = sidebar
      ? getSidebarSectionTitle(sidebar, entrySlug)
      : undefined;
    const content = stripMdx(entry.body ?? '');

    return {
      title: entry.data.title,
      description: entry.data.description,
      href: getLocalizedHref(themeConfig, entrySlug, entryLocaleKey),
      slug: entrySlug,
      section: section ? sectionLabels.get(section) : undefined,
      group: sectionTitle,
      excerpt: createExcerpt(content),
      content,
    };
  });

  const openApiItems = await getOpenApiSearchItems(docsContext.locale.key);

  return new Response(JSON.stringify({ items: [...items, ...openApiItems] }), {
    headers: {
      'Content-Type': 'application/json; charset=utf-8',
      'Cache-Control': 'public, max-age=0, must-revalidate',
    },
  });
};