cicada-mcp 0.1.5__py3-none-any.whl → 0.2.0__py3-none-any.whl

cicada/parser.py

@@ -7,22 +7,22 @@ Author: Cursor(Auto)
 """
 
 import tree_sitter_elixir as ts_elixir
-from tree_sitter import Parser, Language
+from tree_sitter import Language, Parser
 
 from .extractors import (
-    extract_modules,
-    extract_functions,
-    extract_specs,
-    match_specs_to_functions,
-    extract_docs,
-    match_docs_to_functions,
     extract_aliases,
+    extract_behaviours,
+    extract_docs,
+    extract_function_calls,
+    extract_functions,
     extract_imports,
+    extract_modules,
     extract_requires,
+    extract_specs,
     extract_uses,
-    extract_behaviours,
-    extract_function_calls,
     extract_value_mentions,
+    match_docs_to_functions,
+    match_specs_to_functions,
 )
 
 

cicada/pr_finder.py

@@ -8,7 +8,7 @@ import json
 import subprocess
 import sys
 from pathlib import Path
-from typing import Optional, Dict, Any
+from typing import Any
 
 from cicada.utils import load_index
 
@@ -48,7 +48,7 @@ class PRFinder:
             elif not self.index:
                 # Always show warning (even in non-verbose mode) with color
                 print(
-                    f"\033[33m⚠️  No PR index found - using slower network lookups. Create index: python cicada/pr_indexer.py\033[0m",
+                    "\033[33m⚠️  No PR index found - using slower network lookups. Create index: cicada index-pr\033[0m",
                     file=sys.stderr,
                 )
 
@@ -72,9 +72,9 @@ class PRFinder:
             raise RuntimeError(
                 "GitHub CLI (gh) is not installed or not available in PATH. "
                 "Install it from https://cli.github.com/"
-            )
+            ) from None
 
-    def _load_index(self) -> Optional[Dict[str, Any]]:
+    def _load_index(self) -> dict[str, Any] | None:
         """
         Load the PR index from file.
 
@@ -90,7 +90,7 @@ class PRFinder:
 
         return load_index(index_file, verbose=self.verbose, raise_on_error=False)
 
-    def _lookup_pr_in_index(self, commit_sha: str) -> Optional[Dict[str, Any]]:
+    def _lookup_pr_in_index(self, commit_sha: str) -> dict[str, Any] | None:
         """
         Look up PR information from the index.
 
@@ -114,9 +114,7 @@ class PRFinder:
 
         return pr
 
-    def _run_git_blame(
-        self, file_path: str, line_number: int
-    ) -> Optional[Dict[str, str | None]]:
+    def _run_git_blame(self, file_path: str, line_number: int) -> dict[str, str | None] | None:
         """
         Run git blame to find the commit that introduced a specific line.
 
@@ -155,9 +153,7 @@ class PRFinder:
                 if line.startswith("author "):
                     author_name = line[7:]  # Skip 'author '
                 elif line.startswith("author-mail "):
-                    author_email = line[12:].strip(
-                        "<>"
-                    )  # Skip 'author-mail ' and remove < >
+                    author_email = line[12:].strip("<>")  # Skip 'author-mail ' and remove < >
 
             return {
                 "commit": commit_sha,
@@ -166,9 +162,9 @@ class PRFinder:
             }
 
         except subprocess.CalledProcessError as e:
-            raise RuntimeError(f"git blame failed: {e.stderr}")
+            raise RuntimeError(f"git blame failed: {e.stderr}") from e
 
-    def _get_repo_info(self) -> Optional[tuple[str, str]]:
+    def _get_repo_info(self) -> tuple[str, str] | None:
         """
         Get the repository owner and name from git remote.
 
@@ -203,7 +199,7 @@ class PRFinder:
             # Not a GitHub repository or no remote configured
             return None
 
-    def _find_pr_for_commit(self, commit_sha: str) -> Optional[Dict[str, Any]]:
+    def _find_pr_for_commit(self, commit_sha: str) -> dict[str, Any] | None:
         """
         Find the PR that introduced a specific commit.
 
@@ -248,13 +244,13 @@ class PRFinder:
                 "merged_at": pr.get("merged_at"),
             }
 
-        except subprocess.CalledProcessError as e:
+        except subprocess.CalledProcessError:
             # Commit might not be associated with a PR
             return None
         except (json.JSONDecodeError, KeyError) as e:
-            raise RuntimeError(f"Failed to parse PR information: {e}")
+            raise RuntimeError(f"Failed to parse PR information: {e}") from e
 
-    def find_pr_for_line(self, file_path: str, line_number: int) -> Dict[str, Any]:
+    def find_pr_for_line(self, file_path: str, line_number: int) -> dict[str, Any]:
         """
         Find the PR that introduced a specific line of code.
 
@@ -314,7 +310,7 @@ class PRFinder:
             "pr": pr_info,
         }
 
-    def format_result(self, result: Dict[str, Any], output_format: str = "text") -> str:
+    def format_result(self, result: dict[str, Any], output_format: str = "text") -> str:
         """
         Format the result for display.
 
@@ -352,7 +348,7 @@ class PRFinder:
         if output_format == "markdown":
             output = [
                 f"## Line {result['line_number']} in {result['file_path']}",
-                f"",
+                "",
                 f"**Commit:** `{short_commit}`  ",
                 f"**Author:** {author_str}",
             ]

cicada/pr_indexer/__init__.py

@@ -5,11 +5,11 @@ This package contains classes for indexing GitHub pull requests,
 separated by responsibility for better maintainability.
 """
 
+from .cli import main
 from .github_api_client import GitHubAPIClient
-from .pr_index_builder import PRIndexBuilder
-from .line_mapper import LineMapper
 from .indexer import PRIndexer
-from .cli import main
+from .line_mapper import LineMapper
+from .pr_index_builder import PRIndexBuilder
 
 __all__ = [
     "GitHubAPIClient",

cicada/pr_indexer/cli.py

@@ -8,6 +8,7 @@ from .indexer import PRIndexer
 def main():
     """CLI entry point for pr_indexer."""
     import argparse
+
     from cicada.version_check import check_for_updates
 
     # Check for updates (non-blocking, fails silently)
@@ -38,19 +39,13 @@ def main():
     try:
         indexer = PRIndexer(repo_path=args.repo)
         # Incremental by default, unless --clean is specified
-        _ = indexer.index_repository(
-            output_path=args.output, incremental=not args.clean
-        )
+        _ = indexer.index_repository(output_path=args.output, incremental=not args.clean)
 
-        print(
-            "\n✅ Indexing complete! You can now use the MCP tools for PR history lookups."
-        )
+        print("\nIndexing complete! You can now use the MCP tools for PR history lookups.")
 
     except KeyboardInterrupt:
         print("\n\n⚠️  Indexing interrupted by user.")
-        print(
-            "Partial index may have been saved. Run again to continue (incremental by default)."
-        )
+        print("Partial index may have been saved. Run again to continue (incremental by default).")
         sys.exit(130)  # Standard exit code for SIGINT
 
     except Exception as e:

cicada/pr_indexer/github_api_client.py

@@ -10,7 +10,7 @@ separating API concerns from indexing logic.
 import json
 import subprocess
 from pathlib import Path
-from typing import Dict, List, Any, Tuple
+from typing import Any
 
 from cicada.utils import SubprocessRunner
 
@@ -50,9 +50,9 @@ class GitHubAPIClient:
             raise RuntimeError(
                 "GitHub CLI (gh) is not installed or not available in PATH. "
                 "Install it from https://cli.github.com/"
-            )
+            ) from None
 
-    def get_repo_info(self) -> Tuple[str, str]:
+    def get_repo_info(self) -> tuple[str, str]:
         """
         Get repository owner and name from git remote.
 
@@ -81,9 +81,9 @@ class GitHubAPIClient:
             return owner, repo_name
 
         except subprocess.CalledProcessError:
-            raise RuntimeError("Not a GitHub repository or no remote configured")
+            raise RuntimeError("Not a GitHub repository or no remote configured") from None
 
-    def fetch_pr_list(self, state: str = "all", limit: int = 10000) -> List[int]:
+    def fetch_pr_list(self, state: str = "all", limit: int = 10000) -> list[int]:
         """
         Fetch list of PR numbers.
 
@@ -115,11 +115,11 @@ class GitHubAPIClient:
             return [pr["number"] for pr in pr_list]
 
         except subprocess.CalledProcessError as e:
-            raise RuntimeError(f"Failed to fetch PR list: {e.stderr}")
+            raise RuntimeError(f"Failed to fetch PR list: {e.stderr}") from e
         except json.JSONDecodeError as e:
-            raise RuntimeError(f"Failed to parse PR list: {e}")
+            raise RuntimeError(f"Failed to parse PR list: {e}") from e
 
-    def fetch_prs_batch_graphql(self, pr_numbers: List[int]) -> List[Dict[str, Any]]:
+    def fetch_prs_batch_graphql(self, pr_numbers: list[int]) -> list[dict[str, Any]]:
         """
         Fetch detailed PR information for a batch using GraphQL.
 
@@ -189,9 +189,7 @@ class GitHubAPIClient:
         """
 
         try:
-            result = self.runner.run_gh_command(
-                ["api", "graphql", "-f", f"query={query}"]
-            )
+            result = self.runner.run_gh_command(["api", "graphql", "-f", f"query={query}"])
 
             data = json.loads(result.stdout)
             repo_data = data.get("data", {}).get("repository", {})
@@ -199,15 +197,13 @@ class GitHubAPIClient:
             return self._parse_graphql_response(repo_data, len(pr_numbers))
 
         except subprocess.CalledProcessError as e:
-            raise RuntimeError(f"GraphQL query failed for PRs {pr_numbers}: {e.stderr}")
+            raise RuntimeError(f"GraphQL query failed for PRs {pr_numbers}: {e.stderr}") from e
         except (json.JSONDecodeError, KeyError) as e:
-            raise RuntimeError(
-                f"Failed to parse GraphQL response for PRs {pr_numbers}: {e}"
-            )
+            raise RuntimeError(f"Failed to parse GraphQL response for PRs {pr_numbers}: {e}") from e
 
     def _parse_graphql_response(
-        self, repo_data: Dict[str, Any], num_prs: int
-    ) -> List[Dict[str, Any]]:
+        self, repo_data: dict[str, Any], num_prs: int
+    ) -> list[dict[str, Any]]:
         """
         Parse GraphQL response into PR dictionaries.
 
@@ -227,8 +223,7 @@ class GitHubAPIClient:
 
             # Extract commits
             commits = [
-                node["commit"]["oid"]
-                for node in pr_data.get("commits", {}).get("nodes", [])
+                node["commit"]["oid"] for node in pr_data.get("commits", {}).get("nodes", [])
             ]
 
             # Extract files
@@ -255,7 +250,7 @@ class GitHubAPIClient:
 
         return detailed_prs
 
-    def _parse_review_comments(self, pr_data: Dict[str, Any]) -> List[Dict[str, Any]]:
+    def _parse_review_comments(self, pr_data: dict[str, Any]) -> list[dict[str, Any]]:
         """
         Parse review thread comments from PR data.
 
@@ -277,9 +272,7 @@ class GitHubAPIClient:
                 comments.append(
                     {
                         "id": comment_node.get("id"),
-                        "author": (comment_node.get("author") or {}).get(
-                            "login", "unknown"
-                        ),
+                        "author": (comment_node.get("author") or {}).get("login", "unknown"),
                         "body": comment_node.get("body", ""),
                         "created_at": comment_node.get("createdAt"),
                         "path": comment_node.get("path"),
@@ -293,7 +286,7 @@ class GitHubAPIClient:
 
         return comments
 
-    def fetch_pr_rest(self, pr_number: int) -> Dict[str, Any]:
+    def fetch_pr_rest(self, pr_number: int) -> dict[str, Any]:
         """
         Fallback method to fetch a single PR using REST API.
 
@@ -351,7 +344,7 @@ class GitHubAPIClient:
                 "comments": [],
             }
 
-    def _fetch_pr_commits_rest(self, pr_number: int) -> List[str]:
+    def _fetch_pr_commits_rest(self, pr_number: int) -> list[str]:
         """
         Fetch commit SHAs for a PR using REST API.
 
@@ -374,17 +367,13 @@ class GitHubAPIClient:
                 ]
             )
 
-            commits = [
-                line.strip()
-                for line in result.stdout.strip().split("\n")
-                if line.strip()
-            ]
+            commits = [line.strip() for line in result.stdout.strip().split("\n") if line.strip()]
             return commits
 
         except subprocess.CalledProcessError:
             return []
 
-    def _fetch_pr_files_rest(self, pr_number: int) -> List[str]:
+    def _fetch_pr_files_rest(self, pr_number: int) -> list[str]:
         """
         Fetch changed files for a PR using REST API.
 
@@ -399,11 +388,7 @@ class GitHubAPIClient:
                 ["pr", "view", str(pr_number), "--json", "files", "-q", ".files[].path"]
             )
 
-            files = [
-                line.strip()
-                for line in result.stdout.strip().split("\n")
-                if line.strip()
-            ]
+            files = [line.strip() for line in result.stdout.strip().split("\n") if line.strip()]
             return files
 
         except subprocess.CalledProcessError:
@@ -427,5 +412,5 @@ class GitHubAPIClient:
             if pr_list:
                 return pr_list[0]["number"]
             return 0
-        except:
+        except (subprocess.CalledProcessError, json.JSONDecodeError, KeyError, IndexError):
             return 0

cicada/pr_indexer/indexer.py

@@ -5,11 +5,11 @@ Fetches all PRs from a GitHub repository and builds an index mapping commits to
 """
 
 from pathlib import Path
-from typing import Optional, Dict, List, Any
+from typing import Any
 
 from .github_api_client import GitHubAPIClient
-from .pr_index_builder import PRIndexBuilder
 from .line_mapper import LineMapper
+from .pr_index_builder import PRIndexBuilder
 
 
 class PRIndexer:
@@ -36,9 +36,7 @@ class PRIndexer:
         self.repo_owner, self.repo_name = temp_client.get_repo_info()
 
         # Initialize components
-        self.api_client = GitHubAPIClient(
-            self.repo_path, self.repo_owner, self.repo_name
-        )
+        self.api_client = GitHubAPIClient(self.repo_path, self.repo_owner, self.repo_name)
         self.index_builder = PRIndexBuilder(self.repo_owner, self.repo_name)
         self.line_mapper = LineMapper(self.repo_path)
 
@@ -48,7 +46,7 @@ class PRIndexer:
         if not git_dir.exists():
             raise ValueError(f"Not a git repository: {self.repo_path}")
 
-    def fetch_all_prs(self, state: str = "all") -> List[Dict[str, Any]]:
+    def fetch_all_prs(self, state: str = "all") -> list[dict[str, Any]]:
         """
         Fetch all pull requests from GitHub using GraphQL for efficiency.
 
@@ -93,11 +91,9 @@ class PRIndexer:
             return detailed_prs
 
         except RuntimeError as e:
-            raise RuntimeError(f"Failed to fetch PRs: {e}")
+            raise RuntimeError(f"Failed to fetch PRs: {e}") from e
 
-    def incremental_update(
-        self, existing_index: Dict[str, Any]
-    ) -> List[Dict[str, Any]]:
+    def incremental_update(self, existing_index: dict[str, Any]) -> list[dict[str, Any]]:
         """
         Fetch PRs bidirectionally: newer (above max) and older (below min).
 
@@ -108,7 +104,7 @@ class PRIndexer:
             List of new PRs
         """
         # Get min and max PR numbers currently in the index
-        existing_pr_numbers = [int(num) for num in existing_index.get("prs", {}).keys()]
+        existing_pr_numbers = [int(num) for num in existing_index.get("prs", {})]
 
         if not existing_pr_numbers:
             print("Empty index, performing full fetch...")
@@ -145,13 +141,11 @@ class PRIndexer:
             print()
 
         # Fetch detailed info for all PRs
-        detailed_prs = self._fetch_prs_in_batches(
-            newer_pr_numbers, older_pr_numbers, min_pr
-        )
+        detailed_prs = self._fetch_prs_in_batches(newer_pr_numbers, older_pr_numbers, min_pr)
 
         return detailed_prs
 
-    def _fetch_newer_prs(self, max_pr: int) -> List[int]:
+    def _fetch_newer_prs(self, max_pr: int) -> list[int]:
         """Fetch PR numbers newer than max_pr."""
         pr_numbers = self.api_client.fetch_pr_list(state="all", limit=1000)
 
@@ -164,7 +158,7 @@ class PRIndexer:
 
         return newer
 
-    def _fetch_older_prs(self, min_pr: int) -> List[int]:
+    def _fetch_older_prs(self, min_pr: int) -> list[int]:
         """Fetch PR numbers older than min_pr."""
         if min_pr <= 1:
             return []
@@ -184,8 +178,8 @@ class PRIndexer:
             return []
 
     def _fetch_prs_in_batches(
-        self, newer_pr_numbers: List[int], older_pr_numbers: List[int], min_pr: int
-    ) -> List[Dict[str, Any]]:
+        self, newer_pr_numbers: list[int], older_pr_numbers: list[int], min_pr: int
+    ) -> list[dict[str, Any]]:
         """Fetch PRs in batches, showing progress."""
         detailed_prs = []
         batch_size = 10
@@ -197,9 +191,7 @@ class PRIndexer:
                 print(f"\n⬆️  Fetching {len(newer_pr_numbers)} newer PRs...")
                 for i in range(0, len(newer_pr_numbers), batch_size):
                     batch = newer_pr_numbers[i : i + batch_size]
-                    print(
-                        f"  Batch {i//batch_size + 1}/{newer_batches} ({len(batch)} PRs)..."
-                    )
+                    print(f"  Batch {i//batch_size + 1}/{newer_batches} ({len(batch)} PRs)...")
                     batch_prs = self.api_client.fetch_prs_batch_graphql(batch)
                     detailed_prs.extend(batch_prs)
 
@@ -212,9 +204,7 @@ class PRIndexer:
                 )
                 for i in range(0, len(older_pr_numbers), batch_size):
                     batch = older_pr_numbers[i : i + batch_size]
-                    print(
-                        f"  Batch {i//batch_size + 1}/{older_batches} ({len(batch)} PRs)..."
-                    )
+                    print(f"  Batch {i//batch_size + 1}/{older_batches} ({len(batch)} PRs)...")
                     batch_prs = self.api_client.fetch_prs_batch_graphql(batch)
                     detailed_prs.extend(batch_prs)
 
@@ -227,9 +217,7 @@ class PRIndexer:
 
         return detailed_prs
 
-    def index_repository(
-        self, output_path: str = ".cicada/pr_index.json", incremental: bool = False
-    ):
+    def index_repository(self, output_path: str, incremental: bool = False):
         """
         Index the repository's PRs and save to file.
 
@@ -266,8 +254,8 @@ class PRIndexer:
         return index
 
     def _perform_full_index(
-        self, _output_path: str, existing_index: Optional[Dict[str, Any]]
-    ) -> Dict[str, Any]:
+        self, _output_path: str, existing_index: dict[str, Any] | None
+    ) -> dict[str, Any]:
         """Perform a full index build."""
         total_prs_in_repo = self.api_client.get_total_pr_count()
         print(f"Starting clean rebuild ({total_prs_in_repo} PRs in repository)...")

cicada/pr_indexer/line_mapper.py

@@ -7,7 +7,7 @@ allowing comments to track code changes over time.
 
 import subprocess
 from pathlib import Path
-from typing import Dict, List, Any, Optional
+from typing import Any
 
 from cicada.utils import SubprocessRunner
 
@@ -31,7 +31,7 @@ class LineMapper:
         self.repo_path = repo_path
         self.runner = SubprocessRunner(cwd=repo_path)
 
-    def map_all_comment_lines(self, prs: List[Dict[str, Any]]) -> None:
+    def map_all_comment_lines(self, prs: list[dict[str, Any]]) -> None:
         """
         Map all comment lines in PRs to current line numbers.
 
@@ -83,7 +83,7 @@ class LineMapper:
 
     def map_line_to_current(
         self, file_path: str, original_line: int, commit_sha: str
-    ) -> Optional[int]:
+    ) -> int | None:
         """
         Map a line number from a PR commit to the current HEAD.
 
@@ -127,9 +127,7 @@ class LineMapper:
 
             # Search for the same content in current file
             # Look for exact match near the original line number
-            current_line = self._find_matching_line(
-                current_lines, original_content, original_line
-            )
+            current_line = self._find_matching_line(current_lines, original_content, original_line)
 
             return current_line
 
@@ -152,7 +150,7 @@ class LineMapper:
         except subprocess.CalledProcessError:
             return False
 
-    def _get_file_lines(self, ref: str, file_path: str) -> Optional[List[str]]:
+    def _get_file_lines(self, ref: str, file_path: str) -> list[str] | None:
         """
         Get file lines at a specific git ref.
 
@@ -164,9 +162,7 @@ class LineMapper:
             List of file lines, or None if file doesn't exist at that ref
         """
         try:
-            result = self.runner.run_git_command(
-                ["show", f"{ref}:{file_path}"], check=False
-            )
+            result = self.runner.run_git_command(["show", f"{ref}:{file_path}"], check=False)
 
             if result.returncode != 0:
                 return None
@@ -178,11 +174,11 @@ class LineMapper:
 
     def _find_matching_line(
         self,
-        current_lines: List[str],
+        current_lines: list[str],
         original_content: str,
         original_line: int,
         search_range: int = 20,
-    ) -> Optional[int]:
+    ) -> int | None:
         """
         Find a matching line in the current file.