#!/usr/bin/env python3
"""
MCP Server — Claude Code-equivalent tools for Open WebUI / Claude Code CLI.
Tools: bash, file read/write/list, code search, git ops, Gitea API, repo ingest.
Connects via SSE on port 8002 — add to Open WebUI Tools or ~/.claude/mcp.json
"""
import os, subprocess, textwrap
from pathlib import Path

import httpx
from mcp.server.fastmcp import FastMCP

WORKSPACE   = Path(os.getenv("WORKSPACE_DIR", "/workspace"))
REPOS_DIR   = Path(os.getenv("REPOS_DIR",     "/repos"))
GITEA_URL   = os.getenv("GITEA_URL",   "http://gitea:3000")
GITEA_TOKEN = os.getenv("GITEA_TOKEN", "")
GITHUB_TOKEN= os.getenv("GITHUB_TOKEN","")
RAG_URL     = os.getenv("RAG_URL",     "http://rag-server:8001")
KIWIX_URL   = os.getenv("KIWIX_URL",   "http://kiwix:80")

mcp = FastMCP("local-dev-tools")

# ── bash ──────────────────────────────────────────────────────────────────────
@mcp.tool()
def bash(command: str, cwd: str = "") -> str:
    """Run a shell command. Default cwd is /workspace."""
    work = Path(cwd) if cwd else WORKSPACE
    work.mkdir(parents=True, exist_ok=True)
    try:
        r = subprocess.run(command, shell=True, cwd=work, timeout=120,
                           capture_output=True, text=True)
        out = r.stdout + (f"\n[stderr]\n{r.stderr}" if r.stderr else "")
        if r.returncode != 0:
            out += f"\n[exit {r.returncode}]"
        return out or "(no output)"
    except subprocess.TimeoutExpired:
        return "[timeout after 120s]"
    except Exception as e:
        return f"[error] {e}"

# ── file ops ──────────────────────────────────────────────────────────────────
@mcp.tool()
def read_file(path: str) -> str:
    """Read a file. Use absolute path or relative to /workspace."""
    p = Path(path) if Path(path).is_absolute() else WORKSPACE / path
    if not p.exists():
        return f"[not found] {p}"
    if p.stat().st_size > 500_000:
        return f"[too large — {p.stat().st_size//1024}KB]"
    return p.read_text(encoding="utf-8", errors="replace")

@mcp.tool()
def write_file(path: str, content: str) -> str:
    """Write content to a file (creates parent dirs). Relative to /workspace."""
    p = Path(path) if Path(path).is_absolute() else WORKSPACE / path
    p.parent.mkdir(parents=True, exist_ok=True)
    p.write_text(content, encoding="utf-8")
    return f"Wrote {len(content)} chars to {p}"

@mcp.tool()
def list_files(path: str = "", pattern: str = "**/*") -> str:
    """List files matching a glob pattern."""
    base = Path(path) if path else WORKSPACE
    if not base.exists():
        return f"[not found] {base}"
    files = sorted(str(f.relative_to(base)) for f in base.glob(pattern) if f.is_file())
    return "\n".join(files[:500]) or "(empty)"

@mcp.tool()
def search_code(query: str, path: str = "", glob: str = "",
                case_sensitive: bool = False) -> str:
    """Search file contents with ripgrep. Returns file:line matches."""
    base = path or str(WORKSPACE)
    cmd = ["rg", "--line-number", "--no-heading"]
    if not case_sensitive:
        cmd.append("-i")
    if glob:
        cmd += ["-g", glob]
    cmd += [query, base]
    try:
        r = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
        lines = r.stdout.strip().splitlines()
        if len(lines) > 200:
            lines = lines[:200] + [f"… ({len(r.stdout.splitlines())-200} more)"]
        return "\n".join(lines) or "(no matches)"
    except FileNotFoundError:
        # ripgrep not installed, fall back to grep
        r = subprocess.run(["grep", "-rn", query, base],
                           capture_output=True, text=True, timeout=30)
        return r.stdout[:8000] or "(no matches)"
    except Exception as e:
        return f"[error] {e}"

# ── git ───────────────────────────────────────────────────────────────────────
def _git(args: list[str], repo: str = "") -> str:
    cwd = Path(repo) if repo else WORKSPACE
    r = subprocess.run(["git"] + args, cwd=cwd,
                       capture_output=True, text=True, timeout=60)
    return (r.stdout + r.stderr).strip() or "(no output)"

@mcp.tool()
def git_status(repo: str = "") -> str:
    """Show git status of a repo (default: /workspace)."""
    return _git(["status", "--short"], repo)

@mcp.tool()
def git_diff(repo: str = "", cached: bool = False) -> str:
    """Show git diff (staged if cached=True)."""
    args = ["diff", "--stat", "--cached"] if cached else ["diff", "--stat"]
    return _git(args, repo) + "\n\n" + _git(
        ["diff", "--cached"] if cached else ["diff"], repo)

@mcp.tool()
def git_log(repo: str = "", n: int = 10) -> str:
    """Show last n git commits."""
    return _git(["log", f"-{n}", "--oneline", "--decorate"], repo)

@mcp.tool()
def git_commit(message: str, repo: str = "", add_all: bool = True) -> str:
    """Stage all changes and create a commit."""
    if add_all:
        _git(["add", "-A"], repo)
    return _git(["commit", "-m", message], repo)

@mcp.tool()
def git_checkout(branch: str, repo: str = "", create: bool = False) -> str:
    """Checkout a branch, optionally creating it."""
    args = ["checkout", "-b", branch] if create else ["checkout", branch]
    return _git(args, repo)

# ── Gitea API ─────────────────────────────────────────────────────────────────
def _gitea(method: str, path: str, body: dict = {}) -> dict:
    if not GITEA_TOKEN:
        return {"error": "GITEA_TOKEN not set in .env"}
    url = f"{GITEA_URL}/api/v1{path}"
    headers = {"Authorization": f"token {GITEA_TOKEN}",
               "Content-Type": "application/json"}
    r = httpx.request(method, url, json=body or None, headers=headers, timeout=30)
    try:
        return r.json()
    except Exception:
        return {"status": r.status_code, "text": r.text}

@mcp.tool()
def gitea_list_repos() -> str:
    """List your Gitea repos."""
    repos = _gitea("GET", "/repos/search?limit=50")
    if "error" in repos:
        return repos["error"]
    return "\n".join(f"{r['full_name']} — {r.get('description','')}"
                     for r in repos.get("data", []))

@mcp.tool()
def gitea_create_repo(name: str, private: bool = True, description: str = "") -> str:
    """Create a new Gitea repository."""
    r = _gitea("POST", "/user/repos",
               {"name": name, "private": private, "description": description,
                "auto_init": True, "default_branch": "main"})
    return r.get("html_url") or str(r)

@mcp.tool()
def gitea_create_issue(repo: str, title: str, body: str = "") -> str:
    """Create an issue on a Gitea repo (format: owner/repo)."""
    r = _gitea("POST", f"/repos/{repo}/issues", {"title": title, "body": body})
    return r.get("html_url") or str(r)

# ── GitHub API ────────────────────────────────────────────────────────────────
@mcp.tool()
def github_api(method: str, endpoint: str, body: str = "") -> str:
    """Call the GitHub REST API. endpoint e.g. /repos/owner/repo/issues"""
    if not GITHUB_TOKEN:
        return "GITHUB_TOKEN not set in .env"
    import json as _json
    headers = {"Authorization": f"Bearer {GITHUB_TOKEN}",
               "Accept": "application/vnd.github+json"}
    r = httpx.request(method.upper(), f"https://api.github.com{endpoint}",
                      json=_json.loads(body) if body else None,
                      headers=headers, timeout=30)
    try:
        return _json.dumps(r.json(), indent=2)
    except Exception:
        return r.text

# ── Kiwix search ──────────────────────────────────────────────────────────────
import re as _re

@mcp.tool()
def kiwix_search(query: str, books: str = "") -> str:
    """Search offline ZIM content via Kiwix (Wikipedia, Stack Overflow, etc).
    books: optional comma-separated book names to filter, e.g. 'stackoverflow.com_en_all'.
    Leave empty to search all ZIMs."""
    params = {"pattern": query, "lang": ""}
    if books:
        params["books"] = books
    try:
        r = httpx.get(f"{KIWIX_URL}/search", params=params, timeout=15)
        # strip HTML tags to get plain text
        text = _re.sub(r"<[^>]+>", " ", r.text)
        text = _re.sub(r"&[a-zA-Z]+;", " ", text)
        text = _re.sub(r"\s{2,}", "\n", text).strip()
        return text[:6000] or "(no results)"
    except Exception as e:
        return f"[kiwix error] {e}"

# ── RAG ingest ────────────────────────────────────────────────────────────────
@mcp.tool()
def ingest_repo(url: str, name: str = "", branch: str = "main") -> str:
    """Clone a git repo and index it in the RAG code collection."""
    r = httpx.post(f"{RAG_URL}/ingest/repo",
                   json={"url": url, "name": name, "branch": branch}, timeout=300)
    return r.text

@mcp.tool()
def rag_health() -> str:
    """Check RAG server status and indexed document counts."""
    try:
        r = httpx.get(f"{RAG_URL}/health", timeout=10)
        return r.text
    except Exception as e:
        return f"RAG server unreachable: {e}"

if __name__ == "__main__":
    import uvicorn
    app = mcp.sse_app()
    uvicorn.run(app, host="0.0.0.0", port=8002)