MCP OCR Server

"""Handle Tesseract installation during package setup."""

import os
import platform
import subprocess
import sys
from typing import Optional

def get_package_manager() -> Optional[str]:
    """Detect the system's package manager."""
    system = platform.system().lower()
    
    if system == "darwin":
        # Check if Homebrew is installed
        if subprocess.run(["which", "brew"], capture_output=True).returncode == 0:
            return "brew"
    elif system == "linux":
        # Check for apt (Debian/Ubuntu)
        if os.path.exists("/usr/bin/apt"):
            return "apt"
        # Check for dnf (Fedora)
        elif os.path.exists("/usr/bin/dnf"):
            return "dnf"
        # Check for pacman (Arch)
        elif os.path.exists("/usr/bin/pacman"):
            return "pacman"
    
    return None

def install_tesseract():
    """Install Tesseract OCR based on the operating system."""
    system = platform.system().lower()
    pkg_manager = get_package_manager()
    
    try:
        if system == "darwin" and pkg_manager == "brew":
            subprocess.run(["brew", "install", "tesseract", "tesseract-lang", "poppler"], check=True)
            
        elif system == "linux":
            if pkg_manager == "apt":
                subprocess.run(["sudo", "apt-get", "update"], check=True)
                subprocess.run(["sudo", "apt-get", "install", "-y", "tesseract-ocr", "tesseract-ocr-chi-sim", "tesseract-ocr-chi-tra", "poppler-utils"], check=True)
            elif pkg_manager == "dnf":
                subprocess.run(["sudo", "dnf", "install", "-y", "tesseract", "tesseract-langpack-chi_sim", "tesseract-langpack-chi_tra", "poppler-utils"], check=True)
            elif pkg_manager == "pacman":
                subprocess.run(["sudo", "pacman", "-S", "--noconfirm", "tesseract", "tesseract-data-chi_sim", "tesseract-data-chi_tra", "poppler"], check=True)
                
        elif system == "windows":
            print("For Windows users:")
            print("Please download and install Tesseract from: https://github.com/UB-Mannheim/tesseract/wiki")
            print("After installation, ensure the Tesseract installation directory is in your system PATH.")
            return
            
        print("Tesseract OCR installed successfully!")
        
    except subprocess.CalledProcessError as e:
        print(f"Error installing Tesseract: {str(e)}", file=sys.stderr)
        print("Please install Tesseract manually:", file=sys.stderr)
        print("- macOS: brew install tesseract tesseract-lang poppler", file=sys.stderr)
        print("- Ubuntu/Debian: sudo apt-get install tesseract-ocr tesseract-ocr-chi-sim tesseract-ocr-chi-tra poppler-utils", file=sys.stderr)
        print("- Fedora: sudo dnf install tesseract tesseract-langpack-chi_sim tesseract-langpack-chi_tra poppler-utils", file=sys.stderr)
        print("- Arch: sudo pacman -S tesseract tesseract-data-chi_sim tesseract-data-chi_tra poppler", file=sys.stderr)
        print("- Windows: https://github.com/UB-Mannheim/tesseract/wiki and https://github.com/oschwartz10612/poppler-windows/releases", file=sys.stderr)
        sys.exit(1)

if __name__ == "__main__":
    install_tesseract()