ast-mcp-server 0.5.1 → 0.7.0

package/dist/services/outline.js

@@ -1,5 +1,6 @@
 import path from "node:path";
 import { Node, } from "ts-morph";
+import { NO_REQUEST_CONTEXT } from "./request-context.js";
 function joinPresent(parts) {
     return parts.filter((part) => Boolean(part)).join(" ");
 }
@@ -80,20 +81,34 @@ function symbolOf(node, symbolPath, name, signature) {
         endLine: node.getEndLineNumber(),
     };
 }
-function formatClass(statement, prefix = "") {
+function classDeclarationSignature(statement) {
     const name = statement.getName() ?? "<anonymous>";
-    const symbolPath = prefix ? `${prefix}.${name}` : name;
     const heritage = statement.getExtends()?.getText();
     const implementations = statement.getImplements().map((item) => item.getText());
-    const declaration = joinPresent([
+    return joinPresent([
         modifiersOf(statement),
         `class ${name}${typeParametersOf(statement)}`,
         heritage ? `extends ${heritage}` : undefined,
         implementations.length > 0 ? `implements ${implementations.join(", ")}` : undefined,
     ]);
+}
+function interfaceDeclarationSignature(statement) {
+    const name = statement.getName();
+    const extensions = statement.getExtends().map((item) => item.getText());
+    return joinPresent([
+        modifiersOf(statement),
+        `interface ${name}${typeParametersOf(statement)}`,
+        extensions.length > 0 ? `extends ${extensions.join(", ")}` : undefined,
+    ]);
+}
+function formatClass(statement, prefix = "", requestContext = NO_REQUEST_CONTEXT) {
+    const name = statement.getName() ?? "<anonymous>";
+    const symbolPath = prefix ? `${prefix}.${name}` : name;
+    const declaration = classDeclarationSignature(statement);
     const lines = [`${declaration} {`];
     const symbols = [symbolOf(statement, symbolPath, name, declaration)];
     for (const member of statement.getMembers()) {
+        requestContext.checkpoint();
         const signature = memberSignature(member);
         if (!signature)
             continue;
@@ -104,18 +119,14 @@ function formatClass(statement, prefix = "") {
     lines.push("}");
     return { lines, symbols };
 }
-function formatInterface(statement, prefix = "") {
+function formatInterface(statement, prefix = "", requestContext = NO_REQUEST_CONTEXT) {
     const name = statement.getName();
     const symbolPath = prefix ? `${prefix}.${name}` : name;
-    const extensions = statement.getExtends().map((item) => item.getText());
-    const declaration = joinPresent([
-        modifiersOf(statement),
-        `interface ${name}${typeParametersOf(statement)}`,
-        extensions.length > 0 ? `extends ${extensions.join(", ")}` : undefined,
-    ]);
+    const declaration = interfaceDeclarationSignature(statement);
     const lines = [`${declaration} {`];
     const symbols = [symbolOf(statement, symbolPath, name, declaration)];
     for (const member of statement.getMembers()) {
+        requestContext.checkpoint();
         const signature = member.getText().trim();
         lines.push(`  ${signature}`);
         const memberName = "getName" in member ? member.getName() : member.getKindName();
@@ -124,12 +135,13 @@ function formatInterface(statement, prefix = "") {
     lines.push("}");
     return { lines, symbols };
 }
-function formatVariable(statement) {
+function formatVariable(statement, requestContext = NO_REQUEST_CONTEXT) {
     const lines = [];
     const symbols = [];
     const keyword = statement.getDeclarationKind();
     const modifiers = modifiersOf(statement);
     for (const declaration of statement.getDeclarations()) {
+        requestContext.checkpoint();
         const name = declaration.getName();
         const explicitType = declaration.getTypeNode()?.getText();
         const initializer = declaration.getInitializer();
@@ -155,11 +167,13 @@ function moduleStatements(module) {
         return [body];
     return [];
 }
-function formatStatement(statement, prefix = "") {
+function formatStatement(statement, prefix = "", requestContext = NO_REQUEST_CONTEXT) {
+    requestContext.checkpoint();
     if (Node.isClassDeclaration(statement))
-        return formatClass(statement, prefix);
-    if (Node.isInterfaceDeclaration(statement))
-        return formatInterface(statement, prefix);
+        return formatClass(statement, prefix, requestContext);
+    if (Node.isInterfaceDeclaration(statement)) {
+        return formatInterface(statement, prefix, requestContext);
+    }
     if (Node.isFunctionDeclaration(statement)) {
         const name = statement.getName() ?? "<anonymous>";
         const symbolPath = prefix ? `${prefix}.${name}` : name;
@@ -180,13 +194,16 @@ function formatStatement(statement, prefix = "") {
         const symbolPath = prefix ? `${prefix}.${name}` : name;
         const members = statement
             .getMembers()
-            .map((member) => member.getName())
+            .map((member) => {
+            requestContext.checkpoint();
+            return member.getName();
+        })
             .join(", ");
         const signature = joinPresent([modifiersOf(statement), `enum ${name} { ${members} }`]);
         return { lines: [signature], symbols: [symbolOf(statement, symbolPath, name, signature)] };
     }
     if (Node.isVariableStatement(statement))
-        return formatVariable(statement);
+        return formatVariable(statement, requestContext);
     if (Node.isModuleDeclaration(statement)) {
         const name = statement.getName();
         const symbolPath = prefix ? `${prefix}.${name}` : name;
@@ -194,7 +211,8 @@ function formatStatement(statement, prefix = "") {
         const lines = [`${declaration} {`];
         const symbols = [symbolOf(statement, symbolPath, name, declaration)];
         for (const child of moduleStatements(statement)) {
-            const formatted = formatStatement(child, symbolPath);
+            requestContext.checkpoint();
+            const formatted = formatStatement(child, symbolPath, requestContext);
             if (!formatted)
                 continue;
             lines.push(...formatted.lines.map((line) => `  ${line}`));
@@ -205,11 +223,65 @@ function formatStatement(statement, prefix = "") {
     }
     return undefined;
 }
-export function buildFileOutline(sourceFile) {
+export function declarationSignature(node, requestContext = NO_REQUEST_CONTEXT) {
+    requestContext.checkpoint();
+    if (Node.isFunctionDeclaration(node))
+        return functionSignature(node);
+    if (Node.isMethodDeclaration(node))
+        return methodSignature(node);
+    if (Node.isConstructorDeclaration(node) ||
+        Node.isGetAccessorDeclaration(node) ||
+        Node.isSetAccessorDeclaration(node) ||
+        Node.isPropertyDeclaration(node)) {
+        return memberSignature(node) ?? node.getText().trim();
+    }
+    if (Node.isMethodSignature(node) || Node.isPropertySignature(node)) {
+        return node.getText().trim();
+    }
+    if (Node.isClassDeclaration(node))
+        return classDeclarationSignature(node);
+    if (Node.isInterfaceDeclaration(node))
+        return interfaceDeclarationSignature(node);
+    if (Node.isTypeAliasDeclaration(node)) {
+        return joinPresent([
+            modifiersOf(node),
+            `type ${node.getName()}${typeParametersOf(node)} = ${node.getTypeNode()?.getText() ?? "unknown"};`,
+        ]);
+    }
+    if (Node.isEnumDeclaration(node)) {
+        const members = node.getMembers().map((member) => {
+            requestContext.checkpoint();
+            return member.getName();
+        });
+        return joinPresent([modifiersOf(node), `enum ${node.getName()} { ${members.join(", ")} }`]);
+    }
+    if (Node.isModuleDeclaration(node)) {
+        return joinPresent([modifiersOf(node), `namespace ${node.getName()}`]);
+    }
+    if (Node.isVariableDeclaration(node)) {
+        let statement = node.getParent();
+        while (statement && !Node.isVariableStatement(statement))
+            statement = statement.getParent();
+        if (statement && Node.isVariableStatement(statement)) {
+            const explicitType = node.getTypeNode()?.getText();
+            const initializer = node.getInitializer();
+            const callableType = initializer && (Node.isArrowFunction(initializer) || Node.isFunctionExpression(initializer))
+                ? callableExpressionSignature(initializer)
+                : undefined;
+            return joinPresent([
+                modifiersOf(statement),
+                `${statement.getDeclarationKind()} ${node.getName()}${explicitType || callableType ? `: ${explicitType ?? callableType}` : ""};`,
+            ]);
+        }
+    }
+    return node.getText().trim();
+}
+export function buildFileOutline(sourceFile, requestContext = NO_REQUEST_CONTEXT) {
     const lines = [];
     const symbols = [];
     for (const statement of sourceFile.getStatements()) {
-        const formatted = formatStatement(statement);
+        requestContext.checkpoint();
+        const formatted = formatStatement(statement, "", requestContext);
         if (!formatted)
             continue;
         lines.push(...formatted.lines);
@@ -220,7 +292,8 @@ export function buildFileOutline(sourceFile) {
 export function fileOutline(sourceFile) {
     return buildFileOutline(sourceFile).text;
 }
-export function nodeSourceWithLocation(node, projectRoot) {
+export function nodeSourceWithLocation(node, projectRoot, requestContext = NO_REQUEST_CONTEXT) {
+    requestContext.checkpoint();
     const sourceFile = node.getSourceFile();
     return {
         file: projectRoot

package/dist/services/pagination.js

@@ -2,11 +2,17 @@ import { z } from "zod";
 export const DEFAULT_PAGE_LIMIT = 100;
 export const MAX_PAGE_LIMIT = 500;
 export function createPaginationInputSchema(defaultLimit = DEFAULT_PAGE_LIMIT) {
-    if (!Number.isInteger(defaultLimit) || defaultLimit < 1 || defaultLimit > MAX_PAGE_LIMIT) {
+    if (!Number.isSafeInteger(defaultLimit) || defaultLimit < 1 || defaultLimit > MAX_PAGE_LIMIT) {
         throw new Error(`Default page limit must be an integer from 1 to ${MAX_PAGE_LIMIT}.`);
     }
     return {
-        offset: z.number().int().min(0).default(0).describe("Zero-based result offset."),
+        offset: z
+            .number()
+            .int()
+            .min(0)
+            .max(Number.MAX_SAFE_INTEGER)
+            .default(0)
+            .describe("Zero-based result offset."),
         limit: z
             .number()
             .int()
@@ -18,11 +24,11 @@ export function createPaginationInputSchema(defaultLimit = DEFAULT_PAGE_LIMIT) {
 }
 export const PaginationInputSchema = createPaginationInputSchema();
 export const PaginationOutputSchema = {
-    offset: z.number().int().min(0),
+    offset: z.number().int().min(0).max(Number.MAX_SAFE_INTEGER),
     limit: z.number().int().min(1),
     total: z.number().int().min(0),
     has_more: z.boolean(),
-    next_offset: z.number().int().min(0).nullable(),
+    next_offset: z.number().int().min(0).max(Number.MAX_SAFE_INTEGER).nullable(),
 };
 export function paginate(items, offset, limit) {
     const pageItems = items.slice(offset, offset + limit);