multi-lang-build 0.2.0__py3-none-any.whl

multi_lang_build/mirror/config.py

@@ -0,0 +1,251 @@
+"""Mirror configuration for domestic acceleration."""
+
+from typing import TypedDict, Literal
+from pathlib import Path
+
+
+# ============================================================================
+# Constants for mainstream domestic mirrors
+# ============================================================================
+
+# NPM/PNPM mirrors
+NPM_REGISTRY_NPMMIRROR: Literal["https://registry.npmmirror.com"] = "https://registry.npmmirror.com"
+NPM_REGISTRY_TAOBAO: Literal["https://registry.npmmirror.com"] = "https://registry.npmmirror.com"  # Taobao mirrors npmmirror
+
+# Go mirrors (including Qiniu cloud)
+GO_PROXY_GOPROXY_CN: Literal["https://goproxy.cn"] = "https://goproxy.cn"
+GO_PROXY_GOPROXY_IO: Literal["https://goproxy.io"] = "https://goproxy.io"
+GO_PROXY_GOVIP_CN: Literal["https://goproxy.vip.cn"] = "https://goproxy.vip.cn"
+GO_PROXY_FASTGOPROXY: Literal["https://goproxy.cn,direct"] = "https://goproxy.cn,direct"
+
+# Go sumdb mirrors
+GO_SUMDB_GOLANG: Literal["sum.golang.org"] = "sum.golang.org"
+GO_SUMDB_GOLANG_GOOGLE_CN: Literal["sum.golang.google.cn"] = "sum.golang.google.cn"
+
+# PyPI/Pip mirrors
+PYPI_MIRROR_TSINGHUA: Literal["https://pypi.tuna.tsinghua.edu.cn"] = "https://pypi.tuna.tsinghua.edu.cn"
+PYPI_MIRROR_ALIYUN: Literal["https://mirrors.aliyun.com/pypi/simple"] = "https://mirrors.aliyun.com/pypi/simple"
+PYPI_MIRROR_DOUBAN: Literal["https://pypi.doubanio.com/simple"] = "https://pypi.doubanio.com/simple"
+PYPI_MIRROR_HUAWEI: Literal["https://repo.huaweicloud.com/repository/pypi/simple"] = "https://repo.huaweicloud.com/repository/pypi/simple"
+PYPI_MIRROR_NETEASE: Literal["https://mirrors.163.com/pypi/simple"] = "https://mirrors.163.com/pypi/simple"
+PYPI_MIRROR_SAU: Literal["https://pypi.sau.edu.cn/simple"] = "https://pypi.sau.edu.cn/simple"
+
+# Poetry mirrors (uses same PyPI mirrors)
+POETRY_REPO_TSINGHUA: Literal["https://pypi.tuna.tsinghua.edu.cn/simple"] = "https://pypi.tuna.tsinghua.edu.cn/simple"
+POETRY_REPO_ALIYUN: Literal["https://mirrors.aliyun.com/pypi/simple"] = "https://mirrors.aliyun.com/pypi/simple"
+
+# Default mirror selection
+DEFAULT_GO_MIRROR: str = GO_PROXY_GOPROXY_CN
+DEFAULT_PIP_MIRROR: str = PYPI_MIRROR_TSINGHUA
+
+
+class MirrorConfig(TypedDict):
+    """Mirror configuration for package managers."""
+    name: str
+    url: str
+    environment_prefix: str
+    environment_variables: dict[str, str]
+
+
+# Predefined mirror configurations for different package managers
+MIRROR_CONFIGS: dict[str, MirrorConfig] = {
+    "npm": {
+        "name": "NPM China Mirror (npmmirror)",
+        "url": NPM_REGISTRY_NPMMIRROR,
+        "environment_prefix": "NPM",
+        "environment_variables": {
+            "NPM_CONFIG_REGISTRY": NPM_REGISTRY_NPMMIRROR,
+        },
+    },
+    "pnpm": {
+        "name": "PNPM China Mirror (npmmirror)",
+        "url": NPM_REGISTRY_NPMMIRROR,
+        "environment_prefix": "PNPM",
+        "environment_variables": {
+            "PNPM_CONFIG_REGISTRY": NPM_REGISTRY_NPMMIRROR,
+        },
+    },
+    "yarn": {
+        "name": "Yarn China Mirror (npmmirror)",
+        "url": NPM_REGISTRY_NPMMIRROR,
+        "environment_prefix": "YARN",
+        "environment_variables": {
+            "YARN_NPM_MIRROR": NPM_REGISTRY_NPMMIRROR,
+        },
+    },
+    "go": {
+        "name": "Go China Mirror (goproxy.cn)",
+        "url": GO_PROXY_GOPROXY_CN,
+        "environment_prefix": "GOPROXY",
+        "environment_variables": {
+            "GOPROXY": f"{GO_PROXY_GOPROXY_CN},direct",
+            "GOSUMDB": GO_SUMDB_GOLANG_GOOGLE_CN,
+        },
+    },
+    "go_qiniu": {
+        "name": "Go China Mirror (goproxy.io)",
+        "url": GO_PROXY_GOPROXY_IO,
+        "environment_prefix": "GOPROXY",
+        "environment_variables": {
+            "GOPROXY": f"{GO_PROXY_GOPROXY_IO},direct",
+            "GOSUMDB": GO_SUMDB_GOLANG_GOOGLE_CN,
+        },
+    },
+    "go_vip": {
+        "name": "Go China Mirror (goproxy.vip.cn)",
+        "url": GO_PROXY_GOVIP_CN,
+        "environment_prefix": "GOPROXY",
+        "environment_variables": {
+            "GOPROXY": f"{GO_PROXY_GOVIP_CN},direct",
+            "GOSUMDB": GO_SUMDB_GOLANG_GOOGLE_CN,
+        },
+    },
+    "pip": {
+        "name": "Pip China Mirror (Tsinghua)",
+        "url": PYPI_MIRROR_TSINGHUA,
+        "environment_prefix": "PIP",
+        "environment_variables": {
+            "PIP_INDEX_URL": f"{PYPI_MIRROR_TSINGHUA}/simple",
+            "PIP_TRUSTED_HOST": "pypi.tuna.tsinghua.edu.cn",
+        },
+    },
+    "pip_aliyun": {
+        "name": "Pip China Mirror (Aliyun)",
+        "url": PYPI_MIRROR_ALIYUN,
+        "environment_prefix": "PIP",
+        "environment_variables": {
+            "PIP_INDEX_URL": f"{PYPI_MIRROR_ALIYUN}/simple",
+            "PIP_TRUSTED_HOST": "mirrors.aliyun.com",
+        },
+    },
+    "pip_douban": {
+        "name": "Pip China Mirror (Douban)",
+        "url": PYPI_MIRROR_DOUBAN,
+        "environment_prefix": "PIP",
+        "environment_variables": {
+            "PIP_INDEX_URL": f"{PYPI_MIRROR_DOUBAN}/simple",
+            "PIP_TRUSTED_HOST": "pypi.doubanio.com",
+        },
+    },
+    "pip_huawei": {
+        "name": "Pip China Mirror (Huawei)",
+        "url": PYPI_MIRROR_HUAWEI,
+        "environment_prefix": "PIP",
+        "environment_variables": {
+            "PIP_INDEX_URL": f"{PYPI_MIRROR_HUAWEI}/simple",
+            "PIP_TRUSTED_HOST": "repo.huaweicloud.com",
+        },
+    },
+    "python": {
+        "name": "PyPI China Mirror (Tsinghua)",
+        "url": PYPI_MIRROR_TSINGHUA,
+        "environment_prefix": "PIP",
+        "environment_variables": {
+            "PIP_INDEX_URL": f"{PYPI_MIRROR_TSINGHUA}/simple",
+            "PIP_TRUSTED_HOST": "pypi.tuna.tsinghua.edu.cn",
+        },
+    },
+    "poetry": {
+        "name": "Poetry China Mirror (Tsinghua)",
+        "url": PYPI_MIRROR_TSINGHUA,
+        "environment_prefix": "POETRY",
+        "environment_variables": {
+            "POETRY_REPOSITORIES_PYPI_URL": f"{POETRY_REPO_TSINGHUA}/simple",
+        },
+    },
+}
+
+
+def get_mirror_config(package_manager: str) -> MirrorConfig | None:
+    """Get mirror configuration for a specific package manager.
+    
+    Args:
+        package_manager: Name of the package manager (e.g., "npm", "go", "pip")
+        
+    Returns:
+        MirrorConfig if found, None otherwise.
+    """
+    normalized_name = package_manager.lower().strip()
+    return MIRROR_CONFIGS.get(normalized_name)
+
+
+def apply_mirror_environment(
+    package_manager: str, 
+    environment: dict[str, str]
+) -> dict[str, str]:
+    """Apply mirror environment variables to the environment dict.
+    
+    Args:
+        package_manager: Name of the package manager
+        environment: Existing environment variables
+        
+    Returns:
+        Updated environment with mirror variables added.
+    """
+    mirror_config = get_mirror_config(package_manager)
+    if mirror_config is None:
+        return environment
+    
+    mirror_vars = mirror_config.get("environment_variables", {})
+    updated_env = environment.copy()
+    updated_env.update(mirror_vars)
+    
+    return updated_env
+
+
+def reset_mirror_environment(
+    package_manager: str, 
+    environment: dict[str, str]
+) -> dict[str, str]:
+    """Remove mirror environment variables from the environment dict.
+    
+    Args:
+        package_manager: Name of the package manager
+        environment: Existing environment variables
+        
+    Returns:
+        Environment with mirror variables removed.
+    """
+    mirror_config = get_mirror_config(package_manager)
+    if mirror_config is None:
+        return environment
+    
+    mirror_vars = mirror_config.get("environment_variables", {})
+    updated_env = environment.copy()
+    
+    for key in mirror_vars:
+        updated_env.pop(key, None)
+    
+    return updated_env
+
+
+def get_all_mirror_names() -> list[str]:
+    """Get list of all available mirror names.
+    
+    Returns:
+        List of package manager names with mirror support.
+    """
+    return list(MIRROR_CONFIGS.keys())
+
+
+def validate_mirror_config(mirror_config: MirrorConfig) -> bool:
+    """Validate a mirror configuration.
+    
+    Args:
+        mirror_config: Mirror configuration to validate
+        
+    Returns:
+        True if valid, False otherwise.
+    """
+    required_fields = ["name", "url", "environment_prefix", "environment_variables"]
+    
+    for field in required_fields:
+        if field not in mirror_config or not mirror_config[field]:
+            return False
+    
+    # Validate URL format
+    url = mirror_config["url"]
+    if not (url.startswith("http://") or url.startswith("https://")):
+        return False
+    
+    return True

multi_lang_build/py.typed

@@ -0,0 +1,2 @@
+# Marker file for PEP 561 type information
+# This package uses static type checking

multi_lang_build/register.py

@@ -0,0 +1,383 @@
+"""Register multi-lang-build as a skill for various IDEs."""
+
+from __future__ import annotations
+
+import json
+import os
+import subprocess
+import sys
+from pathlib import Path
+from typing import Literal
+
+
+def get_cli_path() -> str:
+    """Get the path to the multi-lang-build CLI."""
+    # Try to find the CLI in PATH
+    result = subprocess.run(
+        ["which", "multi-lang-build"],
+        capture_output=True,
+        text=True,
+    )
+    if result.returncode == 0:
+        return result.stdout.strip()
+    
+    # Fallback: use Python module
+    return f"{sys.executable} -m multi_lang_build"
+
+
+def register_claude_code() -> bool:
+    """Register as Claude Code skill.
+    
+    Claude Code uses ~/.claude/CLAU.md for global instructions
+    and project-specific skills via the API.
+    """
+    try:
+        # Check if claude CLI is available
+        result = subprocess.run(
+            ["which", "claude"],
+            capture_output=True,
+            text=True,
+        )
+        
+        if result.returncode != 0:
+            print("❌ Claude Code CLI not found. Please install it first:")
+            print("   npm install -g @anthropic-ai/claude-code")
+            return False
+        
+        # Create global instructions directory
+        claude_dir = Path.home() / ".claude"
+        claude_dir.mkdir(exist_ok=True)
+        
+        # Read existing CLAU.md or create new
+        claude_md = claude_dir / "CLAUDE.md"
+        existing_content = ""
+        if claude_md.exists():
+            existing_content = claude_md.read_text()
+        
+        # Add multi-lang-build skill section
+        skill_section = """
+
+## Multi-Lang Build Tool
+
+You can use the `multi-lang-build` tool to compile projects:
+
+### Available Commands:
+- `multi-lang-build pnpm <source> --output <dir>` - Build pnpm projects
+- `multi-lang-build go <source> --output <bin>` - Build Go projects
+- `multi-lang-build python <source> --output <dir>` - Build Python projects
+
+### Examples:
+```bash
+# Build Go project
+multi-lang-build go ./src --output ./bin/app --mirror
+
+# Build with specific target (multi-entry support)
+multi-lang-build go ./src --output ./bin/server --target cmd/server/main.go
+
+# Build Python project
+multi-lang-build python ./src --output ./dist --mirror
+
+# Build pnpm project
+multi-lang-build pnpm ./src --output ./dist --mirror
+```
+
+When working with Go projects, check if there are multiple main packages and use the `--target` flag to specify which one to build.
+"""
+        
+        if "## Multi-Lang Build Tool" not in existing_content:
+            with open(claude_md, "a") as f:
+                f.write(skill_section)
+            print(f"✅ Registered with Claude Code (added to {claude_md})")
+        else:
+            print(f"ℹ️ Already registered with Claude Code ({claude_md})")
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ Failed to register with Claude Code: {e}")
+        return False
+
+
+def register_opencode() -> bool:
+    """Register as OpenCode skill.
+    
+    OpenCode typically uses ~/.config/opencode/ for configuration.
+    """
+    try:
+        # Check if opencode CLI is available
+        result = subprocess.run(
+            ["which", "opencode"],
+            capture_output=True,
+            text=True,
+        )
+        
+        if result.returncode != 0:
+            print("❌ OpenCode CLI not found. Please install it first.")
+            return False
+        
+        # Create OpenCode config directory
+        config_dir = Path.home() / ".config" / "opencode"
+        config_dir.mkdir(parents=True, exist_ok=True)
+        
+        # Create or update skills.json
+        skills_file = config_dir / "skills.json"
+        skills: dict = {}
+        
+        if skills_file.exists():
+            with open(skills_file) as f:
+                skills = json.load(f)
+        
+        # Add multi-lang-build skill
+        skills["multi-lang-build"] = {
+            "name": "multi-lang-build",
+            "description": "Multi-language automated build tool with mirror acceleration",
+            "commands": {
+                "build-go": "multi-lang-build go <source> --output <output> [--mirror] [--target <target>]",
+                "build-python": "multi-lang-build python <source> --output <output> [--mirror]",
+                "build-pnpm": "multi-lang-build pnpm <source> --output <output> [--mirror]",
+            },
+            "examples": [
+                "multi-lang-build go ./src --output ./bin/app --mirror",
+                "multi-lang-build go ./src --output ./bin/server --target cmd/server/main.go",
+                "multi-lang-build python ./src --output ./dist --mirror",
+                "multi-lang-build pnpm ./src --output ./dist --mirror",
+            ]
+        }
+        
+        with open(skills_file, "w") as f:
+            json.dump(skills, f, indent=2)
+        
+        print(f"✅ Registered with OpenCode (config: {skills_file})")
+        return True
+        
+    except Exception as e:
+        print(f"❌ Failed to register with OpenCode: {e}")
+        return False
+
+
+def register_trae() -> bool:
+    """Register as Trae skill.
+    
+    Trae typically uses ~/.trae/ for configuration.
+    """
+    try:
+        # Check if trae CLI is available
+        result = subprocess.run(
+            ["which", "trae"],
+            capture_output=True,
+            text=True,
+        )
+        
+        if result.returncode != 0:
+            print("❌ Trae CLI not found. Please install it first.")
+            return False
+        
+        # Create Trae config directory
+        config_dir = Path.home() / ".trae"
+        config_dir.mkdir(exist_ok=True)
+        
+        # Create or update skills.json
+        skills_file = config_dir / "skills.json"
+        skills: dict = {}
+        
+        if skills_file.exists():
+            with open(skills_file) as f:
+                skills = json.load(f)
+        
+        # Add multi-lang-build skill
+        skills["multi-lang-build"] = {
+            "name": "Multi-Lang Build",
+            "description": "Automated build tool for Go, Python, and pnpm projects",
+            "category": "build",
+            "commands": [
+                {
+                    "name": "build-go",
+                    "command": "multi-lang-build go {source} --output {output}",
+                    "description": "Build Go project",
+                    "args": ["source", "output", "target", "mirror"]
+                },
+                {
+                    "name": "build-python",
+                    "command": "multi-lang-build python {source} --output {output}",
+                    "description": "Build Python project",
+                    "args": ["source", "output", "mirror"]
+                },
+                {
+                    "name": "build-pnpm",
+                    "command": "multi-lang-build pnpm {source} --output {output}",
+                    "description": "Build pnpm project",
+                    "args": ["source", "output", "mirror"]
+                }
+            ]
+        }
+        
+        with open(skills_file, "w") as f:
+            json.dump(skills, f, indent=2)
+        
+        print(f"✅ Registered with Trae (config: {skills_file})")
+        return True
+        
+    except Exception as e:
+        print(f"❌ Failed to register with Trae: {e}")
+        return False
+
+
+def register_codebuddy() -> bool:
+    """Register as CodeBuddy skill.
+    
+    CodeBuddy typically uses ~/.codebuddy/ for configuration.
+    """
+    try:
+        # Check if codebuddy CLI is available
+        result = subprocess.run(
+            ["which", "codebuddy"],
+            capture_output=True,
+            text=True,
+        )
+        
+        if result.returncode != 0:
+            print("❌ CodeBuddy CLI not found. Please install it first.")
+            return False
+        
+        # Create CodeBuddy config directory
+        config_dir = Path.home() / ".codebuddy"
+        config_dir.mkdir(exist_ok=True)
+        
+        # Create or update skills.yaml (CodeBuddy may use YAML)
+        skills_file = config_dir / "skills.yaml"
+        
+        skill_config = """# Multi-Lang Build Skill
+skills:
+  multi-lang-build:
+    name: "Multi-Lang Build"
+    description: "Multi-language automated build tool with mirror acceleration"
+    version: "0.2.0"
+    
+    commands:
+      build-go:
+        description: "Build Go project"
+        command: "multi-lang-build go {source} --output {output}"
+        args:
+          - name: source
+            description: "Source directory"
+            required: true
+          - name: output
+            description: "Output path"
+            required: true
+          - name: target
+            description: "Build target (file or directory)"
+            required: false
+          - name: mirror
+            description: "Enable mirror acceleration"
+            type: flag
+            
+      build-python:
+        description: "Build Python project"
+        command: "multi-lang-build python {source} --output {output}"
+        args:
+          - name: source
+            description: "Source directory"
+            required: true
+          - name: output
+            description: "Output directory"
+            required: true
+          - name: mirror
+            description: "Enable mirror acceleration"
+            type: flag
+            
+      build-pnpm:
+        description: "Build pnpm project"
+        command: "multi-lang-build pnpm {source} --output {output}"
+        args:
+          - name: source
+            description: "Source directory"
+            required: true
+          - name: output
+            description: "Output directory"
+            required: true
+          - name: mirror
+            description: "Enable mirror acceleration"
+            type: flag
+"""
+        
+        # Append or create
+        if skills_file.exists():
+            content = skills_file.read_text()
+            if "multi-lang-build" not in content:
+                with open(skills_file, "a") as f:
+                    f.write("\n" + skill_config)
+                print(f"✅ Registered with CodeBuddy (updated {skills_file})")
+            else:
+                print(f"ℹ️ Already registered with CodeBuddy ({skills_file})")
+        else:
+            skills_file.write_text(skill_config)
+            print(f"✅ Registered with CodeBuddy (created {skills_file})")
+        
+        return True
+        
+    except Exception as e:
+        print(f"❌ Failed to register with CodeBuddy: {e}")
+        return False
+
+
+def register_skill(
+    ide: Literal["claude", "opencode", "trae", "codebuddy", "all"] = "claude"
+) -> bool:
+    """Register multi-lang-build as a skill for the specified IDE.
+    
+    Args:
+        ide: The IDE to register with. Options: "claude", "opencode", "trae", "codebuddy", "all"
+             Default is "claude".
+    
+    Returns:
+        True if registration succeeded, False otherwise.
+    """
+    print(f"📝 Registering multi-lang-build for {ide}...\n")
+    
+    if ide == "all":
+        results = [
+            ("Claude Code", register_claude_code()),
+            ("OpenCode", register_opencode()),
+            ("Trae", register_trae()),
+            ("CodeBuddy", register_codebuddy()),
+        ]
+        
+        print("\n" + "=" * 50)
+        print("Registration Summary:")
+        for name, success in results:
+            status = "✅" if success else "❌"
+            print(f"  {status} {name}")
+        
+        return all(success for _, success in results)
+    
+    elif ide == "claude":
+        return register_claude_code()
+    elif ide == "opencode":
+        return register_opencode()
+    elif ide == "trae":
+        return register_trae()
+    elif ide == "codebuddy":
+        return register_codebuddy()
+    else:
+        print(f"❌ Unknown IDE: {ide}")
+        print("Supported IDEs: claude, opencode, trae, codebuddy, all")
+        return False
+
+
+if __name__ == "__main__":
+    import argparse
+    
+    parser = argparse.ArgumentParser(
+        description="Register multi-lang-build as an IDE skill"
+    )
+    parser.add_argument(
+        "ide",
+        nargs="?",
+        default="claude",
+        choices=["claude", "opencode", "trae", "codebuddy", "all"],
+        help="IDE to register with (default: claude)",
+    )
+    
+    args = parser.parse_args()
+    success = register_skill(args.ide)
+    sys.exit(0 if success else 1)

multi_lang_build/types.py

@@ -0,0 +1,43 @@
+"""Type definitions for auto-build project."""
+
+from typing import TypeAlias, TypedDict, NotRequired, Literal
+from pathlib import Path
+
+
+class BuildConfig(TypedDict, total=False):
+    """Build configuration for compiler."""
+    source_dir: Path
+    output_dir: Path
+    environment: dict[str, str]
+    mirror_enabled: bool
+    mirror_region: Literal["china", "global"]
+    extra_args: list[str]
+
+
+class BuildResult(TypedDict):
+    """Result of a build operation."""
+    success: bool
+    return_code: int
+    stdout: str
+    stderr: str
+    output_path: Path | None
+    duration_seconds: float
+
+
+class MirrorConfig(TypedDict, total=False):
+    """Mirror configuration for package managers."""
+    name: str
+    url: str
+    environment_prefix: str
+    environment_variables: dict[str, str]
+
+
+LanguageType: TypeAlias = Literal["pnpm", "npm", "yarn", "go", "python", "pip", "poetry"]
+
+
+class CompilerInfo(TypedDict):
+    """Information about a compiler."""
+    name: str
+    version: str
+    supported_mirrors: list[str]
+    default_mirror: str