@assistant-ui/mcp-docs-server 0.1.17 → 0.1.18

package/dist/utils/logger.js

@@ -0,0 +1,20 @@
+import { IS_PREPARE_MODE } from "../constants.js";
+export const logger = {
+    debug: (message, ...args) => {
+        if (process.env["DEBUG"]) {
+            console.debug(`[DEBUG] ${message}`, ...args);
+        }
+    },
+    info: (message, ...args) => {
+        if (IS_PREPARE_MODE) {
+            console.log(`[INFO] ${message}`, ...args);
+        }
+    },
+    error: (message, ...args) => {
+        console.error(`[ERROR] ${message}`, ...args);
+    },
+    warn: (message, ...args) => {
+        console.warn(`[WARN] ${message}`, ...args);
+    },
+};
+//# sourceMappingURL=logger.js.map

package/dist/utils/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.js","sourceRoot":"","sources":["../../src/utils/logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAElD,MAAM,CAAC,MAAM,MAAM,GAAG;IACpB,KAAK,EAAE,CAAC,OAAe,EAAE,GAAG,IAAW,EAAE,EAAE;QACzC,IAAI,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACzB,OAAO,CAAC,KAAK,CAAC,WAAW,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IACD,IAAI,EAAE,CAAC,OAAe,EAAE,GAAG,IAAW,EAAE,EAAE;QACxC,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,UAAU,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;IACD,KAAK,EAAE,CAAC,OAAe,EAAE,GAAG,IAAW,EAAE,EAAE;QACzC,OAAO,CAAC,KAAK,CAAC,WAAW,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IAC/C,CAAC;IACD,IAAI,EAAE,CAAC,OAAe,EAAE,GAAG,IAAW,EAAE,EAAE;QACxC,OAAO,CAAC,IAAI,CAAC,UAAU,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;IAC7C,CAAC;CACF,CAAC"}

package/dist/utils/mcp-format.d.ts

@@ -0,0 +1,7 @@
+export declare function formatMCPResponse(data: any): {
+    content: Array<{
+        type: "text";
+        text: string;
+    }>;
+};
+//# sourceMappingURL=mcp-format.d.ts.map

package/dist/utils/mcp-format.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-format.d.ts","sourceRoot":"","sources":["../../src/utils/mcp-format.ts"],"names":[],"mappings":"AAAA,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,GAAG,GAAG;IAC5C,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CAChD,CASA"}

package/dist/utils/mcp-format.js

@@ -0,0 +1,11 @@
+export function formatMCPResponse(data) {
+    return {
+        content: [
+            {
+                type: "text",
+                text: typeof data === "string" ? data : JSON.stringify(data, null, 2),
+            },
+        ],
+    };
+}
+//# sourceMappingURL=mcp-format.js.map

package/dist/utils/mcp-format.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-format.js","sourceRoot":"","sources":["../../src/utils/mcp-format.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,iBAAiB,CAAC,IAAS;IAGzC,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;aACtE;SACF;KACF,CAAC;AACJ,CAAC"}

package/dist/utils/mdx.d.ts

@@ -0,0 +1,9 @@
+interface MDXContent {
+    content: string;
+    frontmatter: Record<string, any>;
+    excerpt?: string;
+}
+export declare function readMDXFile(filePath: string): Promise<MDXContent | null>;
+export declare function formatMDXContent(mdxContent: MDXContent): string;
+export {};
+//# sourceMappingURL=mdx.d.ts.map

package/dist/utils/mdx.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mdx.d.ts","sourceRoot":"","sources":["../../src/utils/mdx.ts"],"names":[],"mappings":"AAIA,UAAU,UAAU;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACjC,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,wBAAsB,WAAW,CAC/B,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,CAoB5B;AAED,wBAAgB,gBAAgB,CAAC,UAAU,EAAE,UAAU,GAAG,MAAM,CAI/D"}

package/dist/utils/mdx.js

@@ -0,0 +1,27 @@
+import { readFile } from "node:fs/promises";
+import matter from "gray-matter";
+import { logger } from "./logger.js";
+export async function readMDXFile(filePath) {
+    try {
+        const fileContent = await readFile(filePath, "utf-8");
+        const { content, data } = matter(fileContent);
+        const excerptMatch = content.match(/^(.+?)(?:\n\n|$)/);
+        const excerpt = excerptMatch?.[1] !== undefined
+            ? excerptMatch[1].replace(/^#+ /, "")
+            : undefined;
+        return {
+            content,
+            frontmatter: data,
+            ...(excerpt !== undefined && { excerpt }),
+        };
+    }
+    catch (error) {
+        logger.error(`Failed to read MDX file: ${filePath}`, error);
+        return null;
+    }
+}
+export function formatMDXContent(mdxContent) {
+    const { content, frontmatter } = mdxContent;
+    return matter.stringify(content, frontmatter);
+}
+//# sourceMappingURL=mdx.js.map

package/dist/utils/mdx.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mdx.js","sourceRoot":"","sources":["../../src/utils/mdx.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAQrC,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,QAAgB;IAEhB,IAAI,CAAC;QACH,MAAM,WAAW,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACtD,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;QAE9C,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACvD,MAAM,OAAO,GACX,YAAY,EAAE,CAAC,CAAC,CAAC,KAAK,SAAS;YAC7B,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YACrC,CAAC,CAAC,SAAS,CAAC;QAEhB,OAAO;YACL,OAAO;YACP,WAAW,EAAE,IAAI;YACjB,GAAG,CAAC,OAAO,KAAK,SAAS,IAAI,EAAE,OAAO,EAAE,CAAC;SAC1C,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,4BAA4B,QAAQ,EAAE,EAAE,KAAK,CAAC,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,UAAsB;IACrD,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,UAAU,CAAC;IAE5C,OAAO,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;AAChD,CAAC"}

package/dist/utils/paths.d.ts

@@ -0,0 +1,8 @@
+export declare function listDirContents(dirPath: string): Promise<{
+    directories: string[];
+    files: string[];
+}>;
+export declare function getAvailablePaths(): Promise<string[]>;
+export declare function findNearestPaths(requestedPath: string, availablePaths: string[]): string[];
+export declare function pathExists(path: string): Promise<boolean>;
+//# sourceMappingURL=paths.d.ts.map

package/dist/utils/paths.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"paths.d.ts","sourceRoot":"","sources":["../../src/utils/paths.ts"],"names":[],"mappings":"AAYA,wBAAsB,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC;IAC9D,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,KAAK,EAAE,MAAM,EAAE,CAAC;CACjB,CAAC,CAyBD;AAED,wBAAsB,iBAAiB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC,CAwB3D;AAED,wBAAgB,gBAAgB,CAC9B,aAAa,EAAE,MAAM,EACrB,cAAc,EAAE,MAAM,EAAE,GACvB,MAAM,EAAE,CAiCV;AAED,wBAAsB,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAO/D"}

package/dist/utils/paths.js

@@ -0,0 +1,84 @@
+import { readdir, stat } from "node:fs/promises";
+import { join, extname } from "node:path";
+import { DOCS_PATH, MDX_EXTENSION, MD_EXTENSION } from "../constants.js";
+import { logger } from "./logger.js";
+const SIMILARITY_THRESHOLDS = {
+    EXACT_MATCH: 1,
+    CONTAINS_MATCH: 0.8,
+    PARTIAL_MATCH: 0.5,
+    MIN_SUGGESTION: 0.3,
+};
+export async function listDirContents(dirPath) {
+    try {
+        const items = await readdir(dirPath, { withFileTypes: true });
+        const directories = items
+            .filter((item) => item.isDirectory())
+            .map((item) => item.name)
+            .filter((name) => !name.startsWith("."))
+            .sort();
+        const files = items
+            .filter((item) => item.isFile() &&
+            (extname(item.name) === MDX_EXTENSION ||
+                extname(item.name) === MD_EXTENSION))
+            .map((item) => item.name)
+            .sort();
+        return { directories, files };
+    }
+    catch (error) {
+        logger.error(`Failed to list directory contents: ${dirPath}`, error);
+        return { directories: [], files: [] };
+    }
+}
+export async function getAvailablePaths() {
+    const paths = [];
+    async function scanDirectory(dir, prefix = "") {
+        const { directories, files } = await listDirContents(dir);
+        for (const file of files) {
+            const name = file.replace(MDX_EXTENSION, "").replace(MD_EXTENSION, "");
+            paths.push(prefix ? `${prefix}/${name}` : name);
+        }
+        for (const subdir of directories) {
+            const subdirPath = join(dir, subdir);
+            const newPrefix = prefix ? `${prefix}/${subdir}` : subdir;
+            paths.push(newPrefix);
+            await scanDirectory(subdirPath, newPrefix);
+        }
+    }
+    await scanDirectory(DOCS_PATH);
+    return paths.sort();
+}
+export function findNearestPaths(requestedPath, availablePaths) {
+    const normalizedRequest = requestedPath
+        .toLowerCase()
+        .replace(/[^a-z0-9]/g, "");
+    const scored = availablePaths.map((path) => {
+        const normalizedPath = path.toLowerCase().replace(/[^a-z0-9]/g, "");
+        if (normalizedPath.includes(normalizedRequest)) {
+            return { path, score: SIMILARITY_THRESHOLDS.EXACT_MATCH };
+        }
+        if (normalizedRequest.includes(normalizedPath)) {
+            return { path, score: SIMILARITY_THRESHOLDS.CONTAINS_MATCH };
+        }
+        const overlap = [...normalizedRequest].filter((char) => normalizedPath.includes(char)).length;
+        return {
+            path,
+            score: (overlap / normalizedRequest.length) *
+                SIMILARITY_THRESHOLDS.PARTIAL_MATCH,
+        };
+    });
+    return scored
+        .filter((item) => item.score > SIMILARITY_THRESHOLDS.MIN_SUGGESTION)
+        .sort((a, b) => b.score - a.score)
+        .slice(0, 3)
+        .map((item) => item.path);
+}
+export async function pathExists(path) {
+    try {
+        await stat(path);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+//# sourceMappingURL=paths.js.map

package/dist/utils/paths.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"paths.js","sourceRoot":"","sources":["../../src/utils/paths.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACzE,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,MAAM,qBAAqB,GAAG;IAC5B,WAAW,EAAE,CAAC;IACd,cAAc,EAAE,GAAG;IACnB,aAAa,EAAE,GAAG;IAClB,cAAc,EAAE,GAAG;CACX,CAAC;AAEX,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,OAAe;IAInD,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;QAE9D,MAAM,WAAW,GAAG,KAAK;aACtB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;aACpC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACxB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;aACvC,IAAI,EAAE,CAAC;QAEV,MAAM,KAAK,GAAG,KAAK;aAChB,MAAM,CACL,CAAC,IAAI,EAAE,EAAE,CACP,IAAI,CAAC,MAAM,EAAE;YACb,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,aAAa;gBACnC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,YAAY,CAAC,CACzC;aACA,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACxB,IAAI,EAAE,CAAC;QAEV,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;IAChC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,sCAAsC,OAAO,EAAE,EAAE,KAAK,CAAC,CAAC;QACrE,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;IACxC,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB;IACrC,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,KAAK,UAAU,aAAa,CAC1B,GAAW,EACX,SAAiB,EAAE;QAEnB,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,MAAM,eAAe,CAAC,GAAG,CAAC,CAAC;QAE1D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;YACvE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC;QAED,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;YACjC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YACrC,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;YAC1D,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtB,MAAM,aAAa,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,MAAM,aAAa,CAAC,SAAS,CAAC,CAAC;IAC/B,OAAO,KAAK,CAAC,IAAI,EAAE,CAAC;AACtB,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,aAAqB,EACrB,cAAwB;IAExB,MAAM,iBAAiB,GAAG,aAAa;SACpC,WAAW,EAAE;SACb,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;IAE7B,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACzC,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;QAEpE,IAAI,cAAc,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAC/C,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,qBAAqB,CAAC,WAAW,EAAE,CAAC;QAC5D,CAAC;QAED,IAAI,iBAAiB,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YAC/C,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,qBAAqB,CAAC,cAAc,EAAE,CAAC;QAC/D,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACrD,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,CAC9B,CAAC,MAAM,CAAC;QAET,OAAO;YACL,IAAI;YACJ,KAAK,EACH,CAAC,OAAO,GAAG,iBAAiB,CAAC,MAAM,CAAC;gBACpC,qBAAqB,CAAC,aAAa;SACtC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM;SACV,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,GAAG,qBAAqB,CAAC,cAAc,CAAC;SACnE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;SACjC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAAC,IAAY;IAC3C,IAAI,CAAC;QACH,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC;QACjB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}

package/dist/utils/security.d.ts

@@ -0,0 +1,2 @@
+export declare function sanitizePath(userPath: string): string;
+//# sourceMappingURL=security.d.ts.map

package/dist/utils/security.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"security.d.ts","sourceRoot":"","sources":["../../src/utils/security.ts"],"names":[],"mappings":"AAEA,wBAAgB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAiDrD"}

package/dist/utils/security.js

@@ -0,0 +1,43 @@
+import path from "node:path";
+export function sanitizePath(userPath) {
+    if (!userPath || typeof userPath !== "string") {
+        throw new Error("Invalid path: Path must be a non-empty string");
+    }
+    // Check for traversal patterns before normalization - defense in depth
+    if (userPath.includes("../") || userPath.includes("..\\")) {
+        throw new Error("Invalid path: Directory traversal attempt detected");
+    }
+    const normalized = path.normalize(userPath);
+    if (path.isAbsolute(normalized)) {
+        throw new Error("Invalid path: Absolute paths are not allowed");
+    }
+    const relativePath = path.relative("", normalized);
+    if (relativePath.startsWith("..")) {
+        throw new Error("Invalid path: Directory traversal attempt detected");
+    }
+    if (relativePath.includes("..")) {
+        throw new Error("Invalid path: Path contains invalid traversal sequences");
+    }
+    if (relativePath.includes("\0")) {
+        throw new Error("Invalid path: Path contains null bytes");
+    }
+    if (process.platform !== "win32") {
+        if (normalized.includes("\\")) {
+            throw new Error("Invalid path: Backslashes not allowed");
+        }
+    }
+    else {
+        if (normalized.includes(":") || normalized.startsWith("\\\\")) {
+            throw new Error("Invalid path: Path contains invalid Windows characters");
+        }
+    }
+    const segments = relativePath.split(path.sep);
+    for (const segment of segments) {
+        if (segment.startsWith(".") && segment !== ".") {
+            throw new Error("Invalid path: Hidden files are not allowed");
+        }
+    }
+    // Always return Unix-style paths for consistency across platforms
+    return relativePath.replace(/\\/g, "/");
+}
+//# sourceMappingURL=security.js.map

package/dist/utils/security.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"security.js","sourceRoot":"","sources":["../../src/utils/security.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAE7B,MAAM,UAAU,YAAY,CAAC,QAAgB;IAC3C,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC9C,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,uEAAuE;IACvE,IAAI,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1D,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;IACxE,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAE5C,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAEnD,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;IACxE,CAAC;IAED,IAAI,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;IAC7E,CAAC;IAED,IAAI,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACjC,IAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;SAAM,CAAC;QACN,IAAI,UAAU,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC9D,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;IAED,MAAM,QAAQ,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,OAAO,KAAK,GAAG,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED,kEAAkE;IAClE,OAAO,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;AAC1C,CAAC"}

package/package.json

@@ -1,46 +1,64 @@
 {
   "name": "@assistant-ui/mcp-docs-server",
-  "version": "0.1.17",
+  "version": "0.1.18",
   "description": "MCP server for assistant-ui documentation and examples",
+  "keywords": [
+    "mcp",
+    "model-context-protocol",
+    "assistant-ui",
+    "documentation",
+    "ai",
+    "claude"
+  ],
+  "author": "AgentbaseAI Inc.",
+  "license": "MIT",
   "type": "module",
-  "main": "dist/index.js",
+  "exports": {
+    ".": {
+      "aui-source": "./src/index.ts",
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
   "bin": {
     "assistant-ui-mcp": "./dist/stdio.js"
   },
+  "files": [
+    "dist",
+    "src",
+    ".docs",
+    "README.md"
+  ],
+  "sideEffects": false,
   "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.24.3",
-    "zod": "^4.1.13",
+    "@modelcontextprotocol/sdk": "^1.25.1",
     "gray-matter": "^4.0.3",
-    "cross-env": "^10.1.0"
+    "zod": "^4.2.1"
   },
   "devDependencies": {
-    "@types/node": "^25.0.0",
-    "tsup": "^8.5.1",
+    "@types/node": "^25.0.3",
     "tsx": "^4.21.0",
-    "typescript": "^5.9.3",
-    "vitest": "^4.0.15"
+    "vitest": "^4.0.16",
+    "@assistant-ui/x-buildutils": "0.0.1"
   },
-  "files": [
-    "dist",
-    ".docs",
-    "README.md"
-  ],
   "publishConfig": {
     "access": "public",
     "provenance": true
   },
+  "homepage": "https://www.assistant-ui.com/",
   "repository": {
     "type": "git",
-    "url": "https://github.com/assistant-ui/assistant-ui/tree/main/packages/mcp-docs-server"
+    "url": "git+https://github.com/assistant-ui/assistant-ui.git",
+    "directory": "packages/mcp-docs-server"
   },
   "bugs": {
     "url": "https://github.com/assistant-ui/assistant-ui/issues"
   },
   "scripts": {
-    "clean": "tsx scripts/clean.mts",
-    "prepare-docs": "cross-env PREPARE=true node ./dist/prepare-docs/prepare.js",
-    "build:cli": "tsup src/stdio.ts src/prepare-docs/prepare.ts src/index.ts --format esm --treeshake=smallest --splitting",
-    "build": "pnpm clean && pnpm build:cli && pnpm prepare-docs",
+    "prepare-docs": "node ./dist/prepare-docs/prepare.js",
+    "build": "aui-build && pnpm prepare-docs",
     "dev": "tsx src/stdio.ts",
     "pretest": "pnpm build",
     "test": "vitest run",

package/src/constants.ts

@@ -0,0 +1,22 @@
+import { fileURLToPath } from "node:url";
+import { dirname, join } from "node:path";
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+
+export const ROOT_DIR = join(__dirname, "../../../");
+export const PACKAGE_DIR = join(__dirname, "../");
+
+export const EXAMPLES_PATH = join(ROOT_DIR, "examples");
+
+const DOCS_BASE = join(PACKAGE_DIR, ".docs");
+export const DOCS_PATH = join(DOCS_BASE, "raw/docs");
+export const CODE_EXAMPLES_PATH = join(DOCS_BASE, "organized/code-examples");
+
+export const MDX_EXTENSION = ".mdx";
+export const MD_EXTENSION = ".md";
+
+export const MAX_FILE_SIZE = 10 * 1024 * 1024;
+
+export const IS_PREPARE_MODE = process.argv[1]?.includes(
+  "prepare-docs/prepare",
+);

package/src/index.ts

@@ -0,0 +1,51 @@
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { docsTools } from "./tools/docs.js";
+import { examplesTools } from "./tools/examples.js";
+import { logger } from "./utils/logger.js";
+import { PACKAGE_DIR } from "./constants.js";
+
+import { readFileSync } from "node:fs";
+import { join } from "node:path";
+
+const packageJson = JSON.parse(
+  readFileSync(join(PACKAGE_DIR, "package.json"), "utf-8"),
+);
+
+export const server = new McpServer({
+  name: "assistant-ui-docs",
+  version: packageJson.version,
+});
+
+server.tool(
+  docsTools.name,
+  docsTools.description,
+  docsTools.parameters,
+  docsTools.execute,
+);
+server.tool(
+  examplesTools.name,
+  examplesTools.description,
+  examplesTools.parameters,
+  examplesTools.execute,
+);
+
+export async function runServer() {
+  try {
+    logger.info(
+      `Starting assistant-ui MCP docs server v${packageJson.version}`,
+    );
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+  } catch (error) {
+    logger.error("Failed to start MCP server", error);
+    process.exit(1);
+  }
+}
+
+if (import.meta.url === `file://${process.argv[1]}`) {
+  void runServer().catch((error) => {
+    console.error("Failed to start server:", error);
+    process.exit(1);
+  });
+}

package/src/prepare-docs/code-examples.ts

@@ -0,0 +1,158 @@
+import { rm, mkdir, readdir, readFile, writeFile } from "node:fs/promises";
+import { join, relative, extname } from "node:path";
+import { logger } from "../utils/logger.js";
+import { ROOT_DIR, EXAMPLES_PATH } from "../constants.js";
+
+const OUTPUT_DIR = join(
+  ROOT_DIR,
+  "packages/mcp-docs-server/.docs/organized/code-examples",
+);
+const MAX_LINES = 10000;
+
+interface FileContent {
+  path: string;
+  content: string;
+}
+
+async function scanDirectory(
+  dir: string,
+  baseDir: string,
+): Promise<FileContent[]> {
+  const files: FileContent[] = [];
+
+  try {
+    const entries = await readdir(dir, { withFileTypes: true });
+
+    for (const entry of entries) {
+      const fullPath = join(dir, entry.name);
+
+      if (entry.isDirectory()) {
+        const skipDirs = [
+          "node_modules",
+          "dist",
+          "build",
+          ".next",
+          ".git",
+          ".turbo",
+        ];
+        if (!skipDirs.includes(entry.name)) {
+          const subFiles = await scanDirectory(fullPath, baseDir);
+          files.push(...subFiles);
+        }
+      } else if (entry.isFile()) {
+        const includeExts = [
+          ".ts",
+          ".tsx",
+          ".js",
+          ".jsx",
+          ".json",
+          ".css",
+          ".md",
+          ".mdx",
+        ];
+        const ext = extname(entry.name).toLowerCase();
+
+        if (
+          includeExts.includes(ext) ||
+          entry.name === "package.json" ||
+          entry.name === "tsconfig.json"
+        ) {
+          try {
+            const content = await readFile(fullPath, "utf-8");
+            const relativePath = relative(baseDir, fullPath);
+            files.push({ path: relativePath, content });
+          } catch (error) {
+            logger.warn(`Failed to read file: ${fullPath}`, error);
+          }
+        }
+      }
+    }
+  } catch (error) {
+    logger.error(`Failed to scan directory: ${dir}`, error);
+  }
+
+  return files;
+}
+
+function getFileType(filename: string): string {
+  const ext = extname(filename).toLowerCase();
+  const extMap: Record<string, string> = {
+    ".ts": "typescript",
+    ".tsx": "tsx",
+    ".js": "javascript",
+    ".jsx": "jsx",
+    ".json": "json",
+    ".css": "css",
+    ".md": "markdown",
+    ".mdx": "mdx",
+  };
+  return extMap[ext] || "text";
+}
+
+export async function prepareCodeExamples(): Promise<void> {
+  logger.info("Preparing code examples...");
+
+  try {
+    await rm(OUTPUT_DIR, { recursive: true, force: true });
+    await mkdir(OUTPUT_DIR, { recursive: true });
+
+    const exampleDirs = await readdir(EXAMPLES_PATH, { withFileTypes: true });
+
+    for (const dir of exampleDirs) {
+      if (dir.isDirectory() && !dir.name.startsWith(".")) {
+        const examplePath = join(EXAMPLES_PATH, dir.name);
+        logger.info(`Processing example: ${dir.name}`);
+
+        let description = "";
+        try {
+          const packageJsonPath = join(examplePath, "package.json");
+          const packageJson = JSON.parse(
+            await readFile(packageJsonPath, "utf-8"),
+          );
+          description = packageJson.description || "";
+        } catch (error: any) {
+          if (error?.code !== "ENOENT") {
+            logger.warn(`Failed to read package.json for ${dir.name}:`, error);
+          } else {
+            logger.debug(`No package.json found for example: ${dir.name}`);
+          }
+        }
+
+        const files = await scanDirectory(examplePath, examplePath);
+
+        files.sort((a, b) => a.path.localeCompare(b.path));
+
+        let markdown = `# Example: ${dir.name}\n\n`;
+        if (description) {
+          markdown += `${description}\n\n`;
+        }
+
+        let totalLines = 0;
+        for (const file of files) {
+          const lines = file.content.split("\n").length;
+          if (totalLines + lines > MAX_LINES) {
+            markdown += `\n_Note: Additional files truncated due to size limits_\n`;
+            break;
+          }
+
+          // Normalize Windows backslashes to forward slashes for consistent markdown output
+          markdown += `## ${file.path.replace(/\\/g, "/")}\n\n`;
+          markdown += `\`\`\`${getFileType(file.path)}\n`;
+          markdown += file.content;
+          markdown += `\n\`\`\`\n\n`;
+
+          totalLines += lines;
+        }
+
+        const outputPath = join(OUTPUT_DIR, `${dir.name}.md`);
+        await writeFile(outputPath, markdown, "utf-8");
+        logger.debug(`Created example: ${outputPath}`);
+      }
+    }
+
+    logger.info("Code examples preparation complete");
+  } catch (error) {
+    logger.error("Failed to prepare code examples", error);
+    throw error;
+  }
+}

package/src/prepare-docs/copy-raw.ts

@@ -0,0 +1,55 @@
+import { rm, mkdir, readdir, copyFile } from "node:fs/promises";
+import { join, extname } from "node:path";
+import { logger } from "../utils/logger.js";
+import { ROOT_DIR } from "../constants.js";
+
+const DOCS_SOURCE = join(ROOT_DIR, "apps/docs/content/docs");
+const BLOG_SOURCE = join(ROOT_DIR, "apps/docs/content/blog");
+const DOCS_DEST = join(ROOT_DIR, "packages/mcp-docs-server/.docs/raw");
+
+async function copyDir(src: string, dest: string): Promise<void> {
+  try {
+    await mkdir(dest, { recursive: true });
+    const entries = await readdir(src, { withFileTypes: true });
+
+    for (const entry of entries) {
+      const srcPath = join(src, entry.name);
+      const destPath = join(dest, entry.name);
+
+      if (entry.isDirectory()) {
+        await copyDir(srcPath, destPath);
+      } else if (entry.isFile()) {
+        const ext = extname(entry.name).toLowerCase();
+        if (ext === ".mdx" || ext === ".md") {
+          await copyFile(srcPath, destPath);
+          logger.debug(`Copied: ${srcPath} -> ${destPath}`);
+        }
+      }
+    }
+  } catch (error) {
+    logger.error(`Failed to copy directory: ${src}`, error);
+    throw error;
+  }
+}
+
+export async function copyRaw(): Promise<void> {
+  logger.info("Copying raw documentation files...");
+
+  try {
+    await rm(DOCS_DEST, { recursive: true, force: true });
+    await mkdir(DOCS_DEST, { recursive: true });
+
+    const docsPath = join(DOCS_DEST, "docs");
+    await copyDir(DOCS_SOURCE, docsPath);
+    logger.info(`Copied documentation to ${docsPath}`);
+
+    const blogPath = join(DOCS_DEST, "blog");
+    await copyDir(BLOG_SOURCE, blogPath);
+    logger.info(`Copied blog posts to ${blogPath}`);
+
+    logger.info("Raw documentation copy complete");
+  } catch (error) {
+    logger.error("Failed to copy raw documentation", error);
+    throw error;
+  }
+}

package/src/prepare-docs/prepare.ts

@@ -0,0 +1,24 @@
+import { logger } from "../utils/logger.js";
+import { copyRaw } from "./copy-raw.js";
+import { prepareCodeExamples } from "./code-examples.js";
+
+async function prepare(): Promise<void> {
+  logger.info("Starting documentation preparation...");
+
+  try {
+    await copyRaw();
+    await prepareCodeExamples();
+
+    logger.info("Documentation preparation complete");
+  } catch (error) {
+    logger.error("Documentation preparation failed", error);
+    throw error;
+  }
+}
+
+if (import.meta.url === `file://${process.argv[1]}`) {
+  prepare().catch((error) => {
+    logger.error("Preparation failed", error);
+    process.exit(1);
+  });
+}