Read Github

#!/usr/bin/env python3
"""
CLI wrapper for gitmcp.io MCP servers.
Allows reading GitHub repository documentation via the GitMCP service.
"""

import argparse
import json
import subprocess
import sys
import re
from typing import Optional


def convert_github_to_gitmcp(url_or_path: str) -> str:
    """Convert a GitHub URL or owner/repo path to gitmcp.io URL."""
    # Handle full GitHub URLs
    if "github.com" in url_or_path:
        url = url_or_path.replace("github.com", "gitmcp.io")
        # Strip /tree/..., /blob/..., /commits/..., etc. - gitmcp only needs owner/repo
        url = re.sub(r'/(tree|blob|commits|releases|issues|pull|actions|wiki|settings)/.*$', '', url)
        return url

    # Handle owner/repo format (possibly with extra path segments)
    if "/" in url_or_path and not url_or_path.startswith("http"):
        # Extract just owner/repo, strip any extra paths
        parts = url_or_path.split('/')
        if len(parts) >= 2:
            return f"https://gitmcp.io/{parts[0]}/{parts[1]}"
        return f"https://gitmcp.io/{url_or_path}"

    return url_or_path


def get_repo_name_from_url(url: str) -> str:
    """Extract repo name from URL and convert to underscore format for tool names."""
    # Extract owner/repo from URL pattern like https://gitmcp.io/owner/repo
    match = re.search(r'gitmcp\.io/([^/]+)/([^/]+)', url)
    if match:
        repo_name = match.group(2)
        # Convert dashes to underscores for tool names
        return repo_name.replace('-', '_')

    # Fallback: extract from owner/repo format
    match = re.search(r'^([^/]+)/([^/]+)', url)
    if match:
        repo_name = match.group(2)
        return repo_name.replace('-', '_')

    return ""


def send_jsonrpc(proc: subprocess.Popen, method: str, params: dict = None, msg_id: int = 1) -> dict:
    """Send a JSON-RPC message and get response."""
    request = {
        "jsonrpc": "2.0",
        "id": msg_id,
        "method": method,
    }
    if params:
        request["params"] = params

    message = json.dumps(request) + "\n"
    proc.stdin.write(message)
    proc.stdin.flush()

    # Read response
    response_line = proc.stdout.readline()
    if response_line:
        return json.loads(response_line)
    return {}


def list_tools(mcp_url: str) -> dict:
    """Connect to MCP server and list available tools."""
    proc = subprocess.Popen(
        ["npx", "-y", "mcp-remote", mcp_url],
        stdin=subprocess.PIPE,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        text=True,
        bufsize=1
    )

    try:
        # Initialize
        init_response = send_jsonrpc(proc, "initialize", {
            "protocolVersion": "2024-11-05",
            "capabilities": {},
            "clientInfo": {"name": "gitmcp-cli", "version": "1.0.0"}
        }, 1)

        # Send initialized notification
        proc.stdin.write(json.dumps({"jsonrpc": "2.0", "method": "notifications/initialized"}) + "\n")
        proc.stdin.flush()

        # List tools
        tools_response = send_jsonrpc(proc, "tools/list", {}, 2)

        return tools_response
    finally:
        proc.terminate()


def call_tool(mcp_url: str, tool_name: str, arguments: dict = None) -> dict:
    """Connect to MCP server and call a tool."""
    proc = subprocess.Popen(
        ["npx", "-y", "mcp-remote", mcp_url],
        stdin=subprocess.PIPE,
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE,
        text=True,
        bufsize=1
    )

    try:
        # Initialize
        send_jsonrpc(proc, "initialize", {
            "protocolVersion": "2024-11-05",
            "capabilities": {},
            "clientInfo": {"name": "gitmcp-cli", "version": "1.0.0"}
        }, 1)

        # Send initialized notification
        proc.stdin.write(json.dumps({"jsonrpc": "2.0", "method": "notifications/initialized"}) + "\n")
        proc.stdin.flush()

        # Call tool
        tool_response = send_jsonrpc(proc, "tools/call", {
            "name": tool_name,