@ragrabbit/mcp

Verified
MIT License
OverviewInspectSchema Related Servers Reviews Score
apps/saas/app/(llms.txt)/llms.txt/route.ts
import { NextResponse } from "next/server";
import type { IndexedPage } from "../utils";
import { buildPageTree, getFirstOrganization, getIndexedContent } from "../utils";

export const revalidate = 3600; // 1 hour

export async function generateStaticParams() {
  const org = await getFirstOrganization();
  return [{ organizationId: org?.id }];
}

export async function generateMetadata({ params }: { params: { organizationId?: string } }) {
  return {
    title: "Knowledge Base",
    description: "Knowledge base TOC",
  };
}

function generateTocAndContent(pages: IndexedPage[], level: number = 0) {
  const toc: string[] = [];
  const indent = "  ".repeat(level);

  pages.forEach((page) => {
    if (page.title) {
      toc.push(`${indent}- [${page.title}](${page.url})`);
    }

    // Process children recursively
    if (page.children && page.children.length > 0) {
      const childResults = generateTocAndContent(page.children, level + 1);
      toc.push(...childResults.toc);
    }
  });

  return { toc };
}

export async function GET(request: Request) {
  const { searchParams } = new URL(request.url);
  let organizationId = searchParams.get("organizationId");
  const title = "Knowledge Base";

  if (!organizationId) {
    const org = await getFirstOrganization();
    if (!org) {
      return new NextResponse("No organizations found", { status: 404 });
    }
    organizationId = org.id;
  }

  const pages = await getIndexedContent(organizationId);
  const pageTree = buildPageTree(pages);
  const { toc } = generateTocAndContent(pageTree);

  const markdown = `# ${title}\n\n## Table of Contents\n\n${toc.join("\n")}}`;

  // Set content type header
  const headers = new Headers();
  headers.set("Content-Type", "text/markdown");

  return new NextResponse(markdown, {
    headers,
    status: 200,
  });
}