docusaurus-plugin-llms 0.2.2 → 0.3.0

package/lib/generator-current.js

@@ -0,0 +1,398 @@
+"use strict";
+/**
+ * LLM file generation functions for the docusaurus-plugin-llms plugin
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateLLMFile = generateLLMFile;
+exports.generateIndividualMarkdownFiles = generateIndividualMarkdownFiles;
+exports.generateStandardLLMFiles = generateStandardLLMFiles;
+exports.generateCustomLLMFiles = generateCustomLLMFiles;
+exports.collectDocFiles = collectDocFiles;
+const path = __importStar(require("path"));
+const fs = __importStar(require("fs/promises"));
+const utils_1 = require("./utils");
+const processor_1 = require("./processor");
+/**
+ * Clean a description for use in a TOC item
+ * @param description - The original description
+ * @returns Cleaned description suitable for TOC
+ */
+function cleanDescriptionForToc(description) {
+    if (!description)
+        return '';
+    // Get just the first line for TOC display
+    const lines = description.split('\n');
+    const firstLine = lines.length > 0 ? lines[0] : '';
+    // Remove heading markers only at the beginning of the line
+    // Be careful to only remove actual heading markers (# followed by space at beginning)
+    // and not hashtag symbols that are part of the content (inline hashtags)
+    const cleaned = firstLine.replace(/^(#+)\s+/g, '');
+    // Truncate if too long (150 characters max with ellipsis)
+    return cleaned.length > 150 ? cleaned.substring(0, 147) + '...' : cleaned;
+}
+/**
+ * Generate an LLM-friendly file
+ * @param docs - Processed document information
+ * @param outputPath - Path to write the output file
+ * @param fileTitle - Title for the file
+ * @param fileDescription - Description for the file
+ * @param includeFullContent - Whether to include full content or just links
+ * @param version - Version of the file
+ * @param customRootContent - Optional custom content to include at the root level
+ */
+async function generateLLMFile(docs, outputPath, fileTitle, fileDescription, includeFullContent, version, customRootContent) {
+    // Validate path length before proceeding
+    if (!(0, utils_1.validatePathLength)(outputPath)) {
+        throw new Error(`Output path exceeds maximum length: ${outputPath}`);
+    }
+    utils_1.logger.verbose(`Generating file: ${outputPath}, version: ${version || 'undefined'}`);
+    const versionInfo = version ? `\n\nVersion: ${version}` : '';
+    if (includeFullContent) {
+        // Generate full content file with header deduplication
+        const usedHeaders = new Set();
+        const fullContentSections = docs.map(doc => {
+            // Check if content already starts with the same heading to avoid duplication
+            const trimmedContent = doc.content.trim();
+            const contentLines = trimmedContent.split('\n');
+            const firstLine = contentLines.length > 0 ? contentLines[0] : '';
+            // Check if the first line is a heading that matches our title
+            const headingMatch = firstLine.match(/^#+\s+(.+)$/);
+            const firstHeadingText = headingMatch ? headingMatch[1].trim() : null;
+            // Generate unique header using the utility function
+            const uniqueHeader = (0, utils_1.ensureUniqueIdentifier)(doc.title, usedHeaders, (counter, base) => {
+                // Try to make it more descriptive by adding the file path info if available
+                if (doc.path && counter === 2) {
+                    const pathParts = doc.path.split('/');
+                    const folderName = pathParts.length >= 2 ? pathParts[pathParts.length - 2] : '';
+                    if (folderName) {
+                        return `(${folderName.charAt(0).toUpperCase() + folderName.slice(1)})`;
+                    }
+                }
+                return `(${counter})`;
+            });
+            if (firstHeadingText === doc.title) {
+                // Content already has the same heading, replace it with our unique header
+                const restOfContent = trimmedContent.split('\n').slice(1).join('\n');
+                return `## ${uniqueHeader}
+
+${restOfContent}`;
+            }
+            else {
+                // Content doesn't have the same heading, add our unique H2 header
+                return `## ${uniqueHeader}
+
+${doc.content}`;
+            }
+        });
+        // Use custom root content or default message
+        const rootContent = customRootContent || 'This file contains all documentation content in a single document following the llmstxt.org standard.';
+        const llmFileContent = (0, utils_1.createMarkdownContent)(fileTitle, `${fileDescription}${versionInfo}`, `${rootContent}\n\n${fullContentSections.join('\n\n---\n\n')}`, true // include metadata (description)
+        );
+        try {
+            await (0, utils_1.writeFile)(outputPath, llmFileContent);
+        }
+        catch (error) {
+            throw new Error(`Failed to write file ${outputPath}: ${getErrorMessage(error)}`);
+        }
+    }
+    else {
+        // Generate links-only file
+        const tocItems = docs.map(doc => {
+            // Clean and format the description for TOC
+            const cleanedDescription = cleanDescriptionForToc(doc.description);
+            return `- [${doc.title}](${doc.url})${cleanedDescription ? `: ${cleanedDescription}` : ''}`;
+        });
+        // Use custom root content or default message
+        const rootContent = customRootContent || 'This file contains links to documentation sections following the llmstxt.org standard.';
+        const llmFileContent = (0, utils_1.createMarkdownContent)(fileTitle, `${fileDescription}${versionInfo}`, `${rootContent}\n\n## Table of Contents\n\n${tocItems.join('\n')}`, true // include metadata (description)
+        );
+        try {
+            await (0, utils_1.writeFile)(outputPath, llmFileContent);
+        }
+        catch (error) {
+            throw new Error(`Failed to write file ${outputPath}: ${getErrorMessage(error)}`);
+        }
+    }
+    utils_1.logger.info(`Generated: ${outputPath}`);
+}
+/**
+ * Generate individual markdown files for each document
+ * @param docs - Processed document information
+ * @param outputDir - Directory to write the markdown files
+ * @param siteUrl - Base site URL
+ * @param docsDir - The configured docs directory name (e.g., 'docs', 'documentation', etc.)
+ * @param keepFrontMatter - Array of frontmatter keys to preserve in generated files
+ * @param preserveDirectoryStructure - Whether to preserve the full directory structure (default: true)
+ * @returns Updated docs with new URLs pointing to generated markdown files
+ */
+async function generateIndividualMarkdownFiles(docs, outputDir, siteUrl, docsDir = 'docs', keepFrontMatter = [], preserveDirectoryStructure = true) {
+    const updatedDocs = [];
+    const usedPaths = new Set();
+    for (const doc of docs) {
+        // Use the original path structure as default filename.
+        let relativePath = doc.path
+            .replace(/^\/+/, '') // Remove leading slashes
+            .replace(/\.mdx?$/, '.md'); // Ensure .md extension
+        // Strip the docsDir prefix only if preserveDirectoryStructure is false
+        if (!preserveDirectoryStructure) {
+            relativePath = relativePath
+                .replace(new RegExp(`^${docsDir.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')}/`), ''); // Remove configured docs dir prefix
+        }
+        // If frontmatter has slug, use that.
+        if (doc.frontMatter?.slug && typeof doc.frontMatter.slug === 'string') {
+            const slug = doc.frontMatter.slug.trim().replace(/^\/+|\/+$/g, ''); // Trim whitespace and slashes
+            if (slug) { // Only process if slug is not empty after trimming
+                if (slug.includes('/')) {
+                    // Nested slug: create directory structure
+                    relativePath = slug + '.md';
+                }
+                else {
+                    // Simple slug: replace just the filename
+                    const pathParts = relativePath.replace(/\.md$/, '').split('/');
+                    pathParts[pathParts.length - 1] = slug;
+                    relativePath = pathParts.join('/') + '.md';
+                }
+            }
+        }
+        // Otherwise, if frontmatter has id, use that.
+        else if (doc.frontMatter?.id && typeof doc.frontMatter.id === 'string') {
+            const id = doc.frontMatter.id.trim().replace(/^\/+|\/+$/g, ''); // Trim whitespace and slashes
+            if (id) { // Only process if id is not empty after trimming
+                if (id.includes('/')) {
+                    // Nested id: create directory structure
+                    relativePath = id + '.md';
+                }
+                else {
+                    // Simple id: replace just the filename
+                    const pathParts = relativePath.replace(/\.md$/, '').split('/');
+                    pathParts[pathParts.length - 1] = id;
+                    relativePath = pathParts.join('/') + '.md';
+                }
+            }
+        }
+        // Trim any leading/trailing whitespace from the path
+        relativePath = relativePath.trim();
+        // If path is empty or invalid, create a fallback path
+        if (!relativePath || relativePath === '.md' || relativePath === '') {
+            const sanitizedTitle = (0, utils_1.sanitizeForFilename)(doc.title, 'untitled');
+            relativePath = `${sanitizedTitle}.md`;
+        }
+        // Ensure path uniqueness
+        let uniquePath = relativePath;
+        let counter = 1;
+        const MAX_PATH_ITERATIONS = 10000;
+        let pathIterations = 0;
+        while (usedPaths.has(uniquePath.toLowerCase())) {
+            counter++;
+            const pathParts = relativePath.split('.');
+            const extension = pathParts.pop() || 'md';
+            const basePath = pathParts.join('.');
+            uniquePath = `${basePath}-${counter}.${extension}`;
+            pathIterations++;
+            if (pathIterations >= MAX_PATH_ITERATIONS) {
+                // Fallback to timestamp
+                const timestamp = Date.now();
+                uniquePath = `${basePath}-${timestamp}.${extension}`;
+                utils_1.logger.warn(`Maximum iterations reached for unique path. Using timestamp: ${uniquePath}`);
+                break;
+            }
+        }
+        usedPaths.add(uniquePath.toLowerCase());
+        // Create the full file path and validate/shorten if needed
+        let fullPath = path.join(outputDir, uniquePath);
+        fullPath = (0, utils_1.shortenPathIfNeeded)(fullPath, outputDir, uniquePath);
+        // Update uniquePath to reflect the shortened path if it was changed
+        if (fullPath !== path.join(outputDir, uniquePath)) {
+            uniquePath = path.relative(outputDir, fullPath);
+        }
+        const directory = path.dirname(fullPath);
+        // Create directory structure if it doesn't exist
+        try {
+            await fs.mkdir(directory, { recursive: true });
+        }
+        catch (error) {
+            throw new Error(`Failed to create directory ${directory}: ${getErrorMessage(error)}`);
+        }
+        // Extract preserved frontmatter if specified
+        let preservedFrontMatter = {};
+        if (keepFrontMatter.length > 0 && doc.frontMatter) {
+            for (const key of keepFrontMatter) {
+                if (key in doc.frontMatter) {
+                    preservedFrontMatter[key] = doc.frontMatter[key];
+                }
+            }
+        }
+        // Create markdown content using the utility function
+        const markdownContent = (0, utils_1.createMarkdownContent)(doc.title, doc.description, doc.content, true, // includeMetadata
+        Object.keys(preservedFrontMatter).length > 0 ? preservedFrontMatter : undefined);
+        // Write the markdown file
+        try {
+            await (0, utils_1.writeFile)(fullPath, markdownContent);
+        }
+        catch (error) {
+            throw new Error(`Failed to write file ${fullPath}: ${getErrorMessage(error)}`);
+        }
+        // Create updated DocInfo with new URL pointing to the generated markdown file
+        // Convert file path to URL path (use forward slashes)
+        const urlPath = (0, utils_1.normalizePath)(uniquePath);
+        const newUrl = `${siteUrl}/${urlPath}`;
+        updatedDocs.push({
+            ...doc,
+            url: newUrl,
+            path: `/${urlPath}` // Update path to the new markdown file
+        });
+        utils_1.logger.verbose(`Generated markdown file: ${uniquePath}`);
+    }
+    return updatedDocs;
+}
+/**
+ * Generate standard LLM files (llms.txt and llms-full.txt)
+ * @param context - Plugin context
+ * @param allDocFiles - Array of all document files
+ */
+async function generateStandardLLMFiles(context, allDocFiles) {
+    const { outDir, siteUrl, docTitle, docDescription, options } = context;
+    const { generateLLMsTxt, generateLLMsFullTxt, llmsTxtFilename = 'llms.txt', llmsFullTxtFilename = 'llms-full.txt', includeOrder = [], includeUnmatchedLast = true, version, generateMarkdownFiles = false, rootContent, fullRootContent, processingBatchSize = 100 } = options;
+    if (!generateLLMsTxt && !generateLLMsFullTxt) {
+        utils_1.logger.warn('No standard LLM files configured for generation. Skipping.');
+        return;
+    }
+    // Process files for the standard outputs
+    let processedDocs = await (0, processor_1.processFilesWithPatterns)(context, allDocFiles, [], // No specific include patterns - include all
+    [], // No additional ignore patterns beyond global ignoreFiles
+    includeOrder, includeUnmatchedLast);
+    utils_1.logger.verbose(`Processed ${processedDocs.length} documentation files for standard LLM files`);
+    // Check if we have documents to process
+    if (processedDocs.length === 0) {
+        utils_1.logger.warn('No documents found matching patterns for standard LLM files. Skipping.');
+        return;
+    }
+    // Generate individual markdown files if requested
+    if (generateMarkdownFiles) {
+        utils_1.logger.info('Generating individual markdown files...');
+        processedDocs = await generateIndividualMarkdownFiles(processedDocs, outDir, siteUrl, context.docsDir, context.options.keepFrontMatter || [], context.options.preserveDirectoryStructure !== false // Default to true
+        );
+    }
+    // Generate llms.txt
+    if (generateLLMsTxt) {
+        const llmsTxtPath = path.join(outDir, llmsTxtFilename);
+        await generateLLMFile(processedDocs, llmsTxtPath, docTitle, docDescription, false, // links only
+        version, rootContent);
+    }
+    // Generate llms-full.txt
+    if (generateLLMsFullTxt) {
+        const llmsFullTxtPath = path.join(outDir, llmsFullTxtFilename);
+        await generateLLMFile(processedDocs, llmsFullTxtPath, docTitle, docDescription, true, // full content
+        version, fullRootContent);
+    }
+}
+/**
+ * Generate custom LLM files based on configuration
+ * @param context - Plugin context
+ * @param allDocFiles - Array of all document files
+ */
+async function generateCustomLLMFiles(context, allDocFiles) {
+    const { outDir, siteUrl, docTitle, docDescription, options } = context;
+    const { customLLMFiles = [], ignoreFiles = [], generateMarkdownFiles = false } = options;
+    if (customLLMFiles.length === 0) {
+        utils_1.logger.warn('No custom LLM files configured. Skipping.');
+        return;
+    }
+    utils_1.logger.info(`Generating ${customLLMFiles.length} custom LLM files...`);
+    for (const customFile of customLLMFiles) {
+        utils_1.logger.verbose(`Processing custom file: ${customFile.filename}, version: ${customFile.version || 'undefined'}`);
+        // Combine global ignores with custom ignores
+        const combinedIgnores = [...ignoreFiles];
+        if (customFile.ignorePatterns) {
+            combinedIgnores.push(...customFile.ignorePatterns);
+        }
+        // Process files according to the custom configuration
+        let customDocs = await (0, processor_1.processFilesWithPatterns)(context, allDocFiles, customFile.includePatterns, combinedIgnores, customFile.orderPatterns || [], customFile.includeUnmatchedLast ?? false);
+        if (customDocs.length > 0) {
+            // Generate individual markdown files if requested
+            if (generateMarkdownFiles) {
+                utils_1.logger.info(`Generating individual markdown files for custom file: ${customFile.filename}...`);
+                customDocs = await generateIndividualMarkdownFiles(customDocs, outDir, siteUrl, context.docsDir, context.options.keepFrontMatter || [], context.options.preserveDirectoryStructure !== false // Default to true
+                );
+            }
+            // Use custom title/description or fall back to defaults
+            const customTitle = customFile.title || docTitle;
+            const customDescription = customFile.description || docDescription;
+            // Generate the custom LLM file
+            const customFilePath = path.join(outDir, customFile.filename);
+            await generateLLMFile(customDocs, customFilePath, customTitle, customDescription, customFile.fullContent, customFile.version, customFile.rootContent);
+            utils_1.logger.info(`Generated custom LLM file: ${customFile.filename} with ${customDocs.length} documents`);
+        }
+        else {
+            utils_1.logger.warn(`No matching documents found for custom LLM file: ${customFile.filename}`);
+        }
+    }
+}
+/**
+ * Collect all document files from docs directory and optionally blog
+ * @param context - Plugin context
+ * @returns Array of file paths
+ */
+async function collectDocFiles(context) {
+    const { siteDir, docsDir, options } = context;
+    const { ignoreFiles = [], includeBlog = false, warnOnIgnoredFiles = false } = options;
+    const allDocFiles = [];
+    // Process docs directory
+    const fullDocsDir = path.join(siteDir, docsDir);
+    try {
+        await fs.access(fullDocsDir);
+        // Collect all markdown files from docs directory
+        const docFiles = await (0, utils_1.readMarkdownFiles)(fullDocsDir, siteDir, ignoreFiles, docsDir, warnOnIgnoredFiles);
+        allDocFiles.push(...docFiles);
+    }
+    catch (err) {
+        utils_1.logger.warn(`Docs directory not found: ${fullDocsDir}`);
+    }
+    // Process blog if enabled
+    if (includeBlog) {
+        const blogDir = path.join(siteDir, 'blog');
+        try {
+            await fs.access(blogDir);
+            // Collect all markdown files from blog directory
+            const blogFiles = await (0, utils_1.readMarkdownFiles)(blogDir, siteDir, ignoreFiles, docsDir, warnOnIgnoredFiles);
+            allDocFiles.push(...blogFiles);
+        }
+        catch (err) {
+            utils_1.logger.warn(`Blog directory not found: ${blogDir}`);
+        }
+    }
+    return allDocFiles;
+}

package/lib/generator.d.ts

@@ -11,8 +11,9 @@ import { DocInfo, PluginContext } from './types';
  * @param includeFullContent - Whether to include full content or just links
  * @param version - Version of the file
  * @param customRootContent - Optional custom content to include at the root level
+ * @param batchSize - Batch size for processing documents (default: 100)
  */
-export declare function generateLLMFile(docs: DocInfo[], outputPath: string, fileTitle: string, fileDescription: string, includeFullContent: boolean, version?: string, customRootContent?: string): Promise<void>;
+export declare function generateLLMFile(docs: DocInfo[], outputPath: string, fileTitle: string, fileDescription: string, includeFullContent: boolean, version?: string, customRootContent?: string, batchSize?: number): Promise<void>;
 /**
  * Generate individual markdown files for each document
  * @param docs - Processed document information
@@ -20,9 +21,10 @@ export declare function generateLLMFile(docs: DocInfo[], outputPath: string, fil
  * @param siteUrl - Base site URL
  * @param docsDir - The configured docs directory name (e.g., 'docs', 'documentation', etc.)
  * @param keepFrontMatter - Array of frontmatter keys to preserve in generated files
+ * @param preserveDirectoryStructure - Whether to preserve the full directory structure (default: true)
  * @returns Updated docs with new URLs pointing to generated markdown files
  */
-export declare function generateIndividualMarkdownFiles(docs: DocInfo[], outputDir: string, siteUrl: string, docsDir?: string, keepFrontMatter?: string[]): Promise<DocInfo[]>;
+export declare function generateIndividualMarkdownFiles(docs: DocInfo[], outputDir: string, siteUrl: string, docsDir?: string, keepFrontMatter?: string[], preserveDirectoryStructure?: boolean): Promise<DocInfo[]>;
 /**
  * Generate standard LLM files (llms.txt and llms-full.txt)
  * @param context - Plugin context