mcp-code-indexer 1.1.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

mcp_code_indexer/database/database.py

@@ -234,6 +234,21 @@ class DatabaseManager:
             
             return projects
     
+    async def get_branch_file_counts(self, project_id: str) -> Dict[str, int]:
+        """Get file counts per branch for a project."""
+        async with self.get_connection() as db:
+            cursor = await db.execute(
+                """
+                SELECT branch, COUNT(*) as file_count 
+                FROM file_descriptions 
+                WHERE project_id = ? 
+                GROUP BY branch
+                """,
+                (project_id,)
+            )
+            rows = await cursor.fetchall()
+            return {row[0]: row[1] for row in rows}
+    
     # File description operations
     
     async def create_file_description(self, file_desc: FileDescription) -> None:
@@ -369,13 +384,13 @@ class DatabaseManager:
                     fd.branch,
                     fd.file_path,
                     fd.description,
-                    fts.rank
-                FROM file_descriptions_fts fts
-                JOIN file_descriptions fd ON fd.rowid = fts.rowid
-                WHERE fts MATCH ? 
+                    bm25(file_descriptions_fts) as rank
+                FROM file_descriptions_fts
+                JOIN file_descriptions fd ON fd.rowid = file_descriptions_fts.rowid
+                WHERE file_descriptions_fts MATCH ? 
                   AND fd.project_id = ? 
                   AND fd.branch = ?
-                ORDER BY fts.rank
+                ORDER BY bm25(file_descriptions_fts)
                 LIMIT ?
                 """,
                 (query, project_id, branch, max_results)

mcp_code_indexer/server/mcp_server.py

@@ -473,6 +473,41 @@ class MCPCodeIndexServer:
             await self.db_manager.update_project(project)
             logger.debug(f"Updated project metadata for {project.name}")
     
+    async def _find_best_branch(self, project_id: str, requested_branch: str) -> Optional[str]:
+        """
+        Find the best available branch for a project when the requested branch has no files.
+        Returns the branch with the most files, or None if no branches have files.
+        """
+        try:
+            # Get all branches and their file counts for this project
+            branch_counts = await self.db_manager.get_branch_file_counts(project_id)
+            
+            if not branch_counts:
+                return None
+            
+            # First try common branch name variations
+            common_variations = {
+                'main': ['master', 'develop', 'development', 'dev'],
+                'master': ['main', 'develop', 'development', 'dev'], 
+                'develop': ['development', 'main', 'master', 'dev'],
+                'development': ['develop', 'main', 'master', 'dev'],
+                'dev': ['develop', 'development', 'main', 'master']
+            }
+            
+            # Try variations of the requested branch
+            if requested_branch.lower() in common_variations:
+                for variation in common_variations[requested_branch.lower()]:
+                    if variation in branch_counts and branch_counts[variation] > 0:
+                        return variation
+            
+            # Fall back to the branch with the most files
+            best_branch = max(branch_counts.items(), key=lambda x: x[1])
+            return best_branch[0] if best_branch[1] > 0 else None
+            
+        except Exception as e:
+            logger.warning(f"Error finding best branch: {e}")
+            return None
+    
     async def _handle_get_file_description(self, arguments: Dict[str, Any]) -> Dict[str, Any]:
         """Handle get_file_description tool calls."""
         project_id = await self._get_or_create_project_id(arguments)
@@ -523,13 +558,24 @@ class MCPCodeIndexServer:
     async def _handle_check_codebase_size(self, arguments: Dict[str, Any]) -> Dict[str, Any]:
         """Handle check_codebase_size tool calls."""
         project_id = await self._get_or_create_project_id(arguments)
+        requested_branch = arguments["branch"]
         
-        # Get all file descriptions for this project/branch
+        # Get file descriptions for this project/branch
         file_descriptions = await self.db_manager.get_all_file_descriptions(
             project_id=project_id,
-            branch=arguments["branch"]
+            branch=requested_branch
         )
         
+        # If no files found for requested branch, try to find the best available branch
+        if not file_descriptions:
+            available_branch = await self._find_best_branch(project_id, requested_branch)
+            if available_branch and available_branch != requested_branch:
+                file_descriptions = await self.db_manager.get_all_file_descriptions(
+                    project_id=project_id,
+                    branch=available_branch
+                )
+                logger.info(f"No files found for branch '{requested_branch}', using '{available_branch}' instead")
+        
         # Calculate total tokens
         total_tokens = self.token_counter.calculate_codebase_tokens(file_descriptions)
         is_large = self.token_counter.is_large_codebase(total_tokens)

{mcp_code_indexer-1.1.0.dist-info → mcp_code_indexer-1.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-code-indexer
-Version: 1.1.0
+Version: 1.1.1
 Summary: MCP server that tracks file descriptions across codebases, enabling AI agents to efficiently navigate and understand code through searchable summaries and token-aware overviews.
 Author: MCP Code Indexer Contributors
 Maintainer: MCP Code Indexer Contributors

{mcp_code_indexer-1.1.0.dist-info → mcp_code_indexer-1.1.1.dist-info}/RECORD

@@ -6,17 +6,17 @@ mcp_code_indexer/main.py,sha256=Rou-mAN9-12PPP8jC7dIs2_UNambJuC2F8BF--j-0m8,3715
 mcp_code_indexer/merge_handler.py,sha256=lJR8eVq2qSrF6MW9mR3Fy8UzrNAaQ7RsI2FMNXne3vQ,14692
 mcp_code_indexer/token_counter.py,sha256=WrifOkbF99nWWHlRlhCHAB2KN7qr83GOHl7apE-hJcE,8460
 mcp_code_indexer/database/__init__.py,sha256=aPq_aaRp0aSwOBIq9GkuMNjmLxA411zg2vhdrAuHm-w,38
-mcp_code_indexer/database/database.py,sha256=CqkGXkJLANr2qfxA2FLX1oYiNj0lTm1LyxD97dzJHSs,19650
+mcp_code_indexer/database/database.py,sha256=eG2xY5cd-oxRZ6mgGkqqBiJJfGCPqJgzoFq6kR99WfA,20300
 mcp_code_indexer/database/models.py,sha256=3wOxHKb6j3zKPWFSwB5g1TLpI507vLNZcqsxZR4VuRs,5528
 mcp_code_indexer/middleware/__init__.py,sha256=p-mP0pMsfiU2yajCPvokCUxUEkh_lu4XJP1LyyMW2ug,220
 mcp_code_indexer/middleware/error_middleware.py,sha256=v6jaHmPxf3qerYdb85X1tHIXLxgcbybpitKVakFLQTA,10109
 mcp_code_indexer/server/__init__.py,sha256=16xMcuriUOBlawRqWNBk6niwrvtv_JD5xvI36X1Vsmk,41
-mcp_code_indexer/server/mcp_server.py,sha256=RNM-0tB_1Gm5P5Vz4rNTXQKcjOyIWa4s6nirZ1BNi-g,43208
+mcp_code_indexer/server/mcp_server.py,sha256=LxYt6AQ2hifAZIrduyGGBz22kxfcMnCAsHPjih37X5k,45523
 mcp_code_indexer/tiktoken_cache/9b5ad71b2ce5302211f9c61530b329a4922fc6a4,sha256=Ijkht27pm96ZW3_3OFE-7xAPtR0YyTWXoRO8_-hlsqc,1681126
 mcp_code_indexer/tools/__init__.py,sha256=m01mxML2UdD7y5rih_XNhNSCMzQTz7WQ_T1TeOcYlnE,49
-mcp_code_indexer-1.1.0.dist-info/licenses/LICENSE,sha256=JN9dyPPgYwH9C-UjYM7FLNZjQ6BF7kAzpF3_4PwY4rY,1086
-mcp_code_indexer-1.1.0.dist-info/METADATA,sha256=n6fIk9JrhEmPG2LVpMCFI7XkP3Aey4dM1V_zKuhAkoA,11930
-mcp_code_indexer-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mcp_code_indexer-1.1.0.dist-info/entry_points.txt,sha256=8HqWOw1Is7jOP1bvIgaSwouvT9z_Boe-9hd4NzyJOhY,68
-mcp_code_indexer-1.1.0.dist-info/top_level.txt,sha256=yKYCM-gMGt-cnupGfAhnZaoEsROLB6DQ1KFUuyKx4rw,17
-mcp_code_indexer-1.1.0.dist-info/RECORD,,
+mcp_code_indexer-1.1.1.dist-info/licenses/LICENSE,sha256=JN9dyPPgYwH9C-UjYM7FLNZjQ6BF7kAzpF3_4PwY4rY,1086
+mcp_code_indexer-1.1.1.dist-info/METADATA,sha256=h8Kqpz8nH14e73F1AoBwXAy3BgnBYKh04igxTq2euKw,11930
+mcp_code_indexer-1.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mcp_code_indexer-1.1.1.dist-info/entry_points.txt,sha256=8HqWOw1Is7jOP1bvIgaSwouvT9z_Boe-9hd4NzyJOhY,68
+mcp_code_indexer-1.1.1.dist-info/top_level.txt,sha256=yKYCM-gMGt-cnupGfAhnZaoEsROLB6DQ1KFUuyKx4rw,17
+mcp_code_indexer-1.1.1.dist-info/RECORD,,