feat: Add support for retrieving large file contents in pull requests

Maxim Titovich · Maxim Titovich · commit 11d40083e5cf · 2025-03-11T10:03:32.000+01:00
- Enhance Azure DevOps service to handle large files in pull request changes
- Implement chunked file content retrieval with size and preview metadata
- Add new tool `azure_devops_pull_request_file_content` for fetching file contents
- Update types to include additional file metadata
- Bump version to 1.0.2
diff --git a/README.md b/README.md
@@ -216,6 +216,7 @@ registerTools(server, azureDevOpsService);
 | `azure_devops_pull_request_threads` | Get threads from a pull request     | `repositoryId` (string), `pullRequestId` (number), `project` (string) |
 | `azure_devops_work_item_attachments`| Get attachments for a work item     | `id` (number)                                                         |
 | `azure_devops_pull_request_changes` | Get detailed PR code changes        | `repositoryId` (string), `pullRequestId` (number), `project` (string) |
+| `azure_devops_pull_request_file_content` | Get content of large files in chunks | `repositoryId` (string), `pullRequestId` (number), `filePath` (string), `objectId` (string), `startPosition` (number), `length` (number), `project` (string) |
 | `azure_devops_create_pr_comment`    | Create a comment on a pull request  | `repositoryId` (string), `pullRequestId` (number), `project` (string), `content` (string), and other optional parameters |
 
 ## Development
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "cursor-azure-devops-mcp",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "MCP Server for Cursor IDE-Azure DevOps Integration",
   "main": "build/index.js",
   "type": "module",
diff --git a/src/azure-devops-service.ts b/src/azure-devops-service.ts
@@ -305,13 +305,21 @@ class AzureDevOpsService {
       projectName
     );
 
-    // Get detailed content for each change (with size limits for safety)
-    const MAX_FILE_SIZE = 100000; // Limit file size to 100KB for performance
+    // File size and content handling constants
+    const MAX_INLINE_FILE_SIZE = 500000; // Increased to 500KB for inline content
+    const MAX_CHUNK_SIZE = 100000; // 100KB chunks for larger files
+    const PREVIEW_SIZE = 10000; // 10KB preview for very large files
+
+    // Get detailed content for each change
     const enhancedChanges = await Promise.all(
       (changes.changeEntries || []).map(async (change: any) => {
         const filePath = change.item?.path || '';
         let originalContent = null;
         let modifiedContent = null;
+        let originalContentSize = 0;
+        let modifiedContentSize = 0;
+        let originalContentPreview = null;
+        let modifiedContentPreview = null;
 
         // Skip folders or binary files
         const isBinary = this.isBinaryFile(filePath);
@@ -322,21 +330,44 @@ class AzureDevOpsService {
             // Get original content if the file wasn't newly added
             if (change.changeType !== 'add' && change.originalObjectId) {
               try {
-                const originalItemContent = await this.gitClient.getItemContent(
+                // First get the item metadata to check file size
+                const originalItem = await this.gitClient.getItem(
                   repositoryId,
                   filePath,
                   projectName,
-                  change.originalObjectId,
-                  undefined,
-                  true,
-                  true
+                  change.originalObjectId
                 );
 
-                // Check if the content is too large
-                if (originalItemContent && originalItemContent.length < MAX_FILE_SIZE) {
+                originalContentSize = originalItem?.contentMetadata?.contentLength || 0;
+
+                // For files within the inline limit, get full content
+                if (originalContentSize <= MAX_INLINE_FILE_SIZE) {
+                  const originalItemContent = await this.gitClient.getItemContent(
+                    repositoryId,
+                    filePath,
+                    projectName,
+                    change.originalObjectId,
+                    undefined,
+                    true,
+                    true
+                  );
+
                   originalContent = originalItemContent.toString('utf8');
-                } else {
-                  originalContent = '(File too large to display)';
+                }
+                // For large files, get a preview
+                else {
+                  // Get just the beginning of the file for preview
+                  const previewContent = await this.gitClient.getItemText(
+                    repositoryId,
+                    filePath,
+                    projectName,
+                    change.originalObjectId,
+                    0, // Start at beginning
+                    PREVIEW_SIZE // Get preview bytes
+                  );
+
+                  originalContentPreview = previewContent;
+                  originalContent = `(File too large to display inline - ${Math.round(originalContentSize / 1024)}KB. Preview shown.)`;
                 }
               } catch (error) {
                 console.error(`Error getting original content for ${filePath}:`, error);
@@ -347,21 +378,44 @@ class AzureDevOpsService {
             // Get modified content if the file wasn't deleted
             if (change.changeType !== 'delete' && change.item.objectId) {
               try {
-                const modifiedItemContent = await this.gitClient.getItemContent(
+                // First get the item metadata to check file size
+                const modifiedItem = await this.gitClient.getItem(
                   repositoryId,
                   filePath,
                   projectName,
-                  change.item.objectId,
-                  undefined,
-                  true,
-                  true
+                  change.item.objectId
                 );
 
-                // Check if the content is too large
-                if (modifiedItemContent && modifiedItemContent.length < MAX_FILE_SIZE) {
+                modifiedContentSize = modifiedItem?.contentMetadata?.contentLength || 0;
+
+                // For files within the inline limit, get full content
+                if (modifiedContentSize <= MAX_INLINE_FILE_SIZE) {
+                  const modifiedItemContent = await this.gitClient.getItemContent(
+                    repositoryId,
+                    filePath,
+                    projectName,
+                    change.item.objectId,
+                    undefined,
+                    true,
+                    true
+                  );
+
                   modifiedContent = modifiedItemContent.toString('utf8');
-                } else {
-                  modifiedContent = '(File too large to display)';
+                }
+                // For large files, get a preview
+                else {
+                  // Get just the beginning of the file for preview
+                  const previewContent = await this.gitClient.getItemText(
+                    repositoryId,
+                    filePath,
+                    projectName,
+                    change.item.objectId,
+                    0, // Start at beginning
+                    PREVIEW_SIZE // Get preview bytes
+                  );
+
+                  modifiedContentPreview = previewContent;
+                  modifiedContent = `(File too large to display inline - ${Math.round(modifiedContentSize / 1024)}KB. Preview shown.)`;
                 }
               } catch (error) {
                 console.error(`Error getting modified content for ${filePath}:`, error);
@@ -378,6 +432,12 @@ class AzureDevOpsService {
           ...change,
           originalContent,
           modifiedContent,
+          originalContentSize,
+          modifiedContentSize,
+          originalContentPreview,
+          modifiedContentPreview,
+          isBinary,
+          isFolder,
         } as PullRequestChange;
 
         return enhancedChange;
@@ -390,6 +450,60 @@ class AzureDevOpsService {
     };
   }
 
+  /**
+   * Get content for a specific file in a pull request by chunks
+   * This allows retrieving parts of large files that can't be displayed inline
+   */
+  async getPullRequestFileContent(
+    repositoryId: string,
+    pullRequestId: number,
+    filePath: string,
+    objectId: string,
+    startPosition: number,
+    length: number,
+    project?: string
+  ): Promise<{ content: string; size: number; position: number; length: number }> {
+    await this.initialize();
+
+    if (!this.gitClient) {
+      throw new Error('Git client not initialized');
+    }
+
+    // Use the provided project or fall back to the default project
+    const projectName = project || this.defaultProject;
+
+    if (!projectName) {
+      throw new Error('Project name is required');
+    }
+
+    try {
+      // Get metadata about the file to know its full size
+      const item = await this.gitClient.getItem(repositoryId, filePath, projectName, objectId);
+
+      const fileSize = item?.contentMetadata?.contentLength || 0;
+
+      // Get the specified chunk of content
+      const content = await this.gitClient.getItemText(
+        repositoryId,
+        filePath,
+        projectName,
+        objectId,
+        startPosition,
+        length
+      );
+
+      return {
+        content,
+        size: fileSize,
+        position: startPosition,
+        length: content.length,
+      };
+    } catch (error) {
+      console.error(`Error getting file content for ${filePath}:`, error);
+      throw new Error(`Failed to retrieve content for file: ${filePath}`);
+    }
+  }
+
   /**
    * Helper function to determine if a file is likely binary based on extension
    */
diff --git a/src/index.ts b/src/index.ts
@@ -269,6 +269,42 @@ server.tool(
       pullRequestId,
       project
     );
+
+    return {
+      content: [
+        {
+          type: 'text',
+          text: JSON.stringify(result, null, 2),
+        },
+      ],
+    };
+  }
+);
+
+// New tool for getting content of large files in pull requests by chunks
+server.tool(
+  'azure_devops_pull_request_file_content',
+  'Get content of a specific file in a pull request by chunks (for large files)',
+  {
+    repositoryId: z.string().describe('Repository ID'),
+    pullRequestId: z.number().describe('Pull request ID'),
+    filePath: z.string().describe('File path'),
+    objectId: z.string().describe('Object ID of the file version'),
+    startPosition: z.number().describe('Starting position in the file (bytes)'),
+    length: z.number().describe('Length to read (bytes)'),
+    project: z.string().describe('Project name'),
+  },
+  async ({ repositoryId, pullRequestId, filePath, objectId, startPosition, length, project }) => {
+    const result = await azureDevOpsService.getPullRequestFileContent(
+      repositoryId,
+      pullRequestId,
+      filePath,
+      objectId,
+      startPosition,
+      length,
+      project
+    );
+
     return {
       content: [
         {
@@ -354,6 +390,9 @@ async function main() {
     console.error(
       '- azure_devops_pull_request_changes: Get detailed code changes for a pull request'
     );
+    console.error(
+      '- azure_devops_pull_request_file_content: Get content of a specific file in chunks (for large files)'
+    );
     console.error('- azure_devops_create_pr_comment: Create a comment on a pull request');
   } catch (error) {
     console.error('Error starting server:', error);
diff --git a/src/sse-server.ts b/src/sse-server.ts
@@ -231,7 +231,8 @@ app.get('/sse', async (req, res) => {
         pullRequestId: z.number().describe('Pull request ID'),
         project: z.string().describe('Project name'),
       },
-      async ({ repositoryId, pullRequestId, project }) => {
+      async ({ repositoryId, pullRequestId, project }, { signal }) => {
+        signal.throwIfAborted();
         const result = await azureDevOpsService.getPullRequestChanges(
           repositoryId,
           pullRequestId,
@@ -248,6 +249,44 @@ app.get('/sse', async (req, res) => {
       }
     );
 
+    // New tool for getting content of large files in pull requests by chunks
+    server.tool(
+      'azure_devops_pull_request_file_content',
+      'Get content of a specific file in a pull request by chunks (for large files)',
+      {
+        repositoryId: z.string().describe('Repository ID'),
+        pullRequestId: z.number().describe('Pull request ID'),
+        filePath: z.string().describe('File path'),
+        objectId: z.string().describe('Object ID of the file version'),
+        startPosition: z.number().describe('Starting position in the file (bytes)'),
+        length: z.number().describe('Length to read (bytes)'),
+        project: z.string().describe('Project name'),
+      },
+      async (
+        { repositoryId, pullRequestId, filePath, objectId, startPosition, length, project },
+        { signal }
+      ) => {
+        signal.throwIfAborted();
+        const result = await azureDevOpsService.getPullRequestFileContent(
+          repositoryId,
+          pullRequestId,
+          filePath,
+          objectId,
+          startPosition,
+          length,
+          project
+        );
+        return {
+          content: [
+            {
+              type: 'text',
+              text: JSON.stringify(result, null, 2),
+            },
+          ],
+        };
+      }
+    );
+
     // New tool for creating pull request comments
     server.tool(
       'azure_devops_create_pr_comment',
diff --git a/src/types.ts b/src/types.ts
@@ -140,6 +140,12 @@ export interface PullRequestChange {
   changeType?: string; // Add, Edit, Delete
   originalContent?: string; // Content before change
   modifiedContent?: string; // Content after change
+  originalContentSize?: number; // Size of original file in bytes
+  modifiedContentSize?: number; // Size of modified file in bytes
+  originalContentPreview?: string; // Preview of content for large files
+  modifiedContentPreview?: string; // Preview of content for large files
+  isBinary?: boolean; // Whether the file is binary
+  isFolder?: boolean; // Whether the item is a folder
 }
 
 export interface PullRequestChanges {

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`{`
`2`	`2`	`"name": "cursor-azure-devops-mcp",`
`3`		`- "version": "1.0.1",`
	`3`	`+ "version": "1.0.2",`
`4`	`4`	`"description": "MCP Server for Cursor IDE-Azure DevOps Integration",`
`5`	`5`	`"main": "build/index.js",`
`6`	`6`	`"type": "module",`