auto-coder 0.1.398__py3-none-any.whl → 0.1.400__py3-none-any.whl

autocoder/common/pull_requests/manager.py

@@ -0,0 +1,201 @@
+"""
+Pull Request 主管理器
+"""
+from typing import Optional, List, Dict, Any
+from loguru import logger
+
+from .models import PRConfig, PRData, PRResult, PRInfo, RepoInfo, PlatformType, DEFAULT_TEMPLATES
+from .providers import PROVIDERS
+from .utils import detect_platform_from_repo, get_repo_info_from_path, validate_repo_path
+from .exceptions import (
+    PRError, PlatformNotSupportedError, ConfigurationError, ValidationError
+)
+from .config import get_config
+
+
+class PullRequestManager:
+    """Pull Request 主管理器"""
+    
+    def __init__(self, config: Optional[PRConfig] = None):
+        self.config = config
+        self._provider = None
+        self._templates = DEFAULT_TEMPLATES.copy()
+    
+    def _get_provider(self, platform: Optional[str] = None, repo_path: Optional[str] = None):
+        """获取平台提供者"""
+        if self.config and not platform:
+            platform = self.config.platform.value
+        elif not platform and repo_path:
+            # 自动检测平台
+            detected_platform = detect_platform_from_repo(repo_path)
+            if not detected_platform:
+                raise PlatformNotSupportedError("无法从仓库路径检测平台类型")
+            platform = detected_platform.value
+        elif not platform:
+            raise ValidationError("必须指定平台类型或提供仓库路径")
+        
+        if platform not in PROVIDERS:
+            raise PlatformNotSupportedError(f"不支持的平台: {platform}")
+        
+        # 使用现有配置或创建新配置
+        if self.config and self.config.platform.value == platform:
+            config = self.config
+        else:
+            try:
+                config = get_config(platform)
+            except ConfigurationError:
+                raise ConfigurationError(f"平台 {platform} 的配置未找到")
+        
+        provider_class = PROVIDERS[platform]
+        return provider_class(config)
+    
+    def create_pull_request(
+        self,
+        repo_path: str,
+        source_branch: str,
+        target_branch: str,
+        title: str,
+        description: str = "",
+        labels: Optional[List[str]] = None,
+        assignees: Optional[List[str]] = None,
+        reviewers: Optional[List[str]] = None,
+        draft: bool = False,
+        template_type: Optional[str] = None,
+        template_vars: Optional[Dict[str, str]] = None,
+        platform: Optional[str] = None,
+        **kwargs
+    ) -> PRResult:
+        """创建 Pull Request"""
+        try:
+            # 验证仓库路径
+            repo_path = validate_repo_path(repo_path)
+            
+            # 获取仓库信息
+            repo_info = get_repo_info_from_path(repo_path)
+            if not repo_info:
+                raise ValidationError("无法获取仓库信息")
+            
+            # 获取提供者
+            provider = self._get_provider(platform, repo_path)
+            
+            # 创建PR数据
+            pr_data = PRData(
+                title=title,
+                description=description,
+                source_branch=source_branch,
+                target_branch=target_branch,
+                labels=labels or [],
+                assignees=assignees or [],
+                reviewers=reviewers or [],
+                draft=draft,
+                template_type=template_type,
+                template_vars=template_vars or {}
+            )
+            
+            # 创建PR
+            result = provider.create_pr(repo_info, pr_data)
+            
+            return result
+            
+        except Exception as e:
+            logger.error(f"创建PR失败: {e}")
+            return PRResult(
+                success=False,
+                error_message=str(e),
+                platform=PlatformType(platform) if platform else None
+            )
+    
+    def get_pull_request(
+        self,
+        repo_path: str,
+        pr_number: int,
+        platform: Optional[str] = None
+    ) -> PRResult:
+        """获取 Pull Request 信息"""
+        try:
+            repo_path = validate_repo_path(repo_path)
+            repo_info = get_repo_info_from_path(repo_path)
+            if not repo_info:
+                raise ValidationError("无法获取仓库信息")
+            
+            provider = self._get_provider(platform, repo_path)
+            return provider.get_pr(repo_info, pr_number)
+            
+        except Exception as e:
+            logger.error(f"获取PR失败: {e}")
+            return PRResult(
+                success=False,
+                error_message=str(e),
+                platform=PlatformType(platform) if platform else None
+            )
+    
+    def update_pull_request(
+        self,
+        repo_path: str,
+        pr_number: int,
+        platform: Optional[str] = None,
+        **kwargs
+    ) -> PRResult:
+        """更新 Pull Request"""
+        try:
+            repo_path = validate_repo_path(repo_path)
+            repo_info = get_repo_info_from_path(repo_path)
+            if not repo_info:
+                raise ValidationError("无法获取仓库信息")
+            
+            provider = self._get_provider(platform, repo_path)
+            return provider.update_pr(repo_info, pr_number, **kwargs)
+            
+        except Exception as e:
+            logger.error(f"更新PR失败: {e}")
+            return PRResult(
+                success=False,
+                error_message=str(e),
+                platform=PlatformType(platform) if platform else None
+            )
+    
+    def list_pull_requests(
+        self,
+        repo_path: str,
+        state: str = "open",
+        per_page: int = 30,
+        page: int = 1,
+        platform: Optional[str] = None
+    ) -> List[PRInfo]:
+        """列出 Pull Requests"""
+        try:
+            repo_path = validate_repo_path(repo_path)
+            repo_info = get_repo_info_from_path(repo_path)
+            if not repo_info:
+                raise ValidationError("无法获取仓库信息")
+            
+            provider = self._get_provider(platform, repo_path)
+            return provider.list_prs(repo_info, state, per_page, page)
+            
+        except Exception as e:
+            logger.error(f"列出PR失败: {e}")
+            return []
+    
+    def health_check(self, platform: Optional[str] = None, repo_path: Optional[str] = None) -> bool:
+        """健康检查"""
+        try:
+            provider = self._get_provider(platform, repo_path)
+            return provider.health_check()
+        except Exception as e:
+            logger.error(f"健康检查失败: {e}")
+            return False
+
+
+# 全局管理器实例
+_global_manager = PullRequestManager()
+
+
+def set_global_config(config: PRConfig) -> None:
+    """设置全局配置"""
+    global _global_manager
+    _global_manager = PullRequestManager(config)
+
+
+def get_global_manager() -> PullRequestManager:
+    """获取全局管理器"""
+    return _global_manager

autocoder/common/pull_requests/models.py

@@ -0,0 +1,164 @@
+"""
+Pull Request 数据模型定义
+"""
+from typing import Optional, List, Dict, Any
+from dataclasses import dataclass, field
+from enum import Enum
+import os
+import json
+from pathlib import Path
+
+
+class PlatformType(str, Enum):
+    """支持的代码托管平台类型"""
+    GITHUB = "github"
+    GITLAB = "gitlab" 
+    GITEE = "gitee"
+    GITCODE = "gitcode"
+
+
+@dataclass
+class PRConfig:
+    """Pull Request 配置类"""
+    platform: PlatformType
+    token: str
+    base_url: Optional[str] = None
+    timeout: int = 30
+    verify_ssl: bool = True
+    retry_count: int = 3
+    default_labels: List[str] = field(default_factory=list)
+    default_assignees: List[str] = field(default_factory=list)
+    
+    # 平台特定配置
+    draft: bool = False
+    maintainer_can_modify: bool = True
+    remove_source_branch: bool = False
+    squash: bool = False
+    
+    def __post_init__(self):
+        if isinstance(self.platform, str):
+            self.platform = PlatformType(self.platform)
+        if self.base_url is None:
+            self.base_url = self._get_default_base_url()
+    
+    def _get_default_base_url(self) -> str:
+        default_urls = {
+            PlatformType.GITHUB: "https://api.github.com",
+            PlatformType.GITLAB: "https://gitlab.com/api/v4",
+            PlatformType.GITEE: "https://gitee.com/api/v5",
+            PlatformType.GITCODE: "https://gitcode.net/api/v4"
+        }
+        return default_urls.get(self.platform, "")
+    
+    @classmethod
+    def from_env(cls, platform: str) -> 'PRConfig':
+        env_mappings = {
+            PlatformType.GITHUB: "GITHUB_TOKEN",
+            PlatformType.GITLAB: "GITLAB_TOKEN", 
+            PlatformType.GITEE: "GITEE_TOKEN",
+            PlatformType.GITCODE: "GITCODE_TOKEN"
+        }
+        platform_type = PlatformType(platform)
+        token_env = env_mappings.get(platform_type)
+        if not token_env:
+            raise ValueError(f"不支持的平台类型: {platform}")
+        token = os.getenv(token_env)
+        if not token:
+            raise ValueError(f"环境变量 {token_env} 未设置")
+        return cls(platform=platform_type, token=token)
+
+
+@dataclass
+class RepoInfo:
+    """仓库信息"""
+    owner: str
+    name: str
+    platform: PlatformType
+    full_name: str = field(init=False)
+    
+    def __post_init__(self):
+        self.full_name = f"{self.owner}/{self.name}"
+
+
+@dataclass 
+class PRData:
+    """Pull Request 数据"""
+    title: str
+    description: str
+    source_branch: str
+    target_branch: str
+    labels: List[str] = field(default_factory=list)
+    assignees: List[str] = field(default_factory=list)
+    reviewers: List[str] = field(default_factory=list)
+    draft: bool = False
+    template_type: Optional[str] = None
+    template_vars: Dict[str, str] = field(default_factory=dict)
+
+
+@dataclass
+class PRResult:
+    """Pull Request 操作结果"""
+    success: bool
+    pr_number: Optional[int] = None
+    pr_url: Optional[str] = None
+    pr_id: Optional[str] = None
+    error_message: Optional[str] = None
+    error_code: Optional[str] = None
+    platform: Optional[PlatformType] = None
+    raw_response: Optional[Dict[str, Any]] = None
+    retry_after: Optional[int] = None
+
+
+@dataclass
+class PRInfo:
+    """Pull Request 详细信息"""
+    number: int
+    title: str
+    description: str
+    state: str
+    source_branch: str
+    target_branch: str
+    author: str
+    created_at: str
+    updated_at: str
+    merged_at: Optional[str] = None
+    pr_url: str = ""
+    labels: List[str] = field(default_factory=list)
+    assignees: List[str] = field(default_factory=list)
+    reviewers: List[str] = field(default_factory=list)
+    mergeable: Optional[bool] = None
+    draft: bool = False
+    raw_data: Optional[Dict[str, Any]] = None
+
+
+# 默认模板配置
+DEFAULT_TEMPLATES = {
+    "bug_fix": {
+        "title_prefix": "🐛 Bug Fix:",
+        "description_template": """
+## 问题描述
+{problem_description}
+
+## 解决方案
+{solution_description}
+
+## 测试
+- [ ] 单元测试通过
+- [ ] 集成测试通过
+- [ ] 手动测试验证
+        """
+    },
+    "feature": {
+        "title_prefix": "✨ Feature:",
+        "description_template": """
+## 新功能说明
+{feature_description}
+
+## 实现细节
+{implementation_details}
+
+## 使用示例
+{usage_examples}
+        """
+    }
+}

autocoder/common/pull_requests/providers/__init__.py

@@ -0,0 +1,23 @@
+"""
+Pull Request 平台提供者模块
+"""
+from .github_provider import GitHubProvider
+from .gitlab_provider import GitLabProvider
+from .gitee_provider import GiteeProvider
+from .gitcode_provider import GitCodeProvider
+
+# 提供者映射
+PROVIDERS = {
+    'github': GitHubProvider,
+    'gitlab': GitLabProvider,
+    'gitee': GiteeProvider,
+    'gitcode': GitCodeProvider
+}
+
+__all__ = [
+    'GitHubProvider',
+    'GitLabProvider', 
+    'GiteeProvider',
+    'GitCodeProvider',
+    'PROVIDERS'
+]

autocoder/common/pull_requests/providers/gitcode_provider.py

@@ -0,0 +1,19 @@
+"""
+GitCode API 提供者实现
+"""
+from typing import List
+from .github_provider import GitHubProvider
+from ..models import RepoInfo, PRData, PRResult, PRInfo
+
+
+class GitCodeProvider(GitHubProvider):
+    """GitCode API 提供者（基于GitHub提供者）"""
+    
+    def _get_auth_header(self) -> str:
+        """获取认证头"""
+        return f"Bearer {self.config.token}"
+    
+    def create_pr(self, repo_info: RepoInfo, pr_data: PRData) -> PRResult:
+        """创建 Merge Request"""
+        # 简化实现，实际应该调用GitCode API
+        return super().create_pr(repo_info, pr_data)

autocoder/common/pull_requests/providers/gitee_provider.py

@@ -0,0 +1,20 @@
+"""
+Gitee API 提供者实现
+"""
+from typing import List
+from .github_provider import GitHubProvider
+from ..models import RepoInfo, PRData, PRResult, PRInfo
+
+
+class GiteeProvider(GitHubProvider):
+    """Gitee API 提供者（基于GitHub提供者）"""
+    
+    def _get_auth_header(self) -> str:
+        """获取认证头"""
+        # Gitee 使用 token 参数而不是 Authorization 头
+        return ""
+    
+    def create_pr(self, repo_info: RepoInfo, pr_data: PRData) -> PRResult:
+        """创建 Pull Request"""
+        # 简化实现，实际应该调用Gitee API
+        return super().create_pr(repo_info, pr_data)

autocoder/common/pull_requests/providers/github_provider.py

@@ -0,0 +1,214 @@
+"""
+GitHub API 提供者实现
+"""
+from typing import List, Dict, Any
+from loguru import logger
+
+from ..base_provider import BasePlatformProvider
+from ..models import RepoInfo, PRData, PRResult, PRInfo
+from ..utils import build_pr_url
+
+
+class GitHubProvider(BasePlatformProvider):
+    """GitHub API 提供者"""
+    
+    def _get_auth_header(self) -> str:
+        """获取认证头"""
+        return f"token {self.config.token}"
+    
+    def create_pr(self, repo_info: RepoInfo, pr_data: PRData) -> PRResult:
+        """创建 Pull Request"""
+        self._validate_repo_info(repo_info)
+        self._validate_pr_data(pr_data)
+        
+        url = f"{self.config.base_url}/repos/{repo_info.full_name}/pulls"
+        
+        payload = {
+            "title": pr_data.title,
+            "body": pr_data.description,
+            "head": pr_data.source_branch,
+            "base": pr_data.target_branch,
+            "draft": pr_data.draft or self.config.draft,
+            "maintainer_can_modify": self.config.maintainer_can_modify
+        }
+        
+        try:
+            response = self._make_request('POST', url, data=payload)
+            data = response.json()
+            
+            pr_url = build_pr_url(self.config.platform, repo_info, data['number'])
+            
+            result = PRResult(
+                success=True,
+                pr_number=data['number'],
+                pr_url=pr_url,
+                pr_id=str(data['id']),
+                platform=self.config.platform,
+                raw_response=data
+            )
+            
+            logger.info(f"GitHub PR 创建成功: {pr_url}")
+            return result
+            
+        except Exception as e:
+            logger.error(f"创建 GitHub PR 失败: {e}")
+            return PRResult(
+                success=False,
+                error_message=str(e),
+                platform=self.config.platform
+            )
+    
+    def get_pr(self, repo_info: RepoInfo, pr_number: int) -> PRResult:
+        """获取 PR 信息"""
+        url = f"{self.config.base_url}/repos/{repo_info.full_name}/pulls/{pr_number}"
+        
+        try:
+            response = self._make_request('GET', url)
+            data = response.json()
+            
+            pr_info = PRInfo(
+                number=data['number'],
+                title=data['title'],
+                description=data.get('body', ''),
+                state=data['state'],
+                source_branch=data['head']['ref'],
+                target_branch=data['base']['ref'],
+                author=data['user']['login'],
+                created_at=data['created_at'],
+                updated_at=data['updated_at'],
+                merged_at=data.get('merged_at'),
+                pr_url=data['html_url'],
+                labels=[label['name'] for label in data.get('labels', [])],
+                assignees=[assignee['login'] for assignee in data.get('assignees', [])],
+                mergeable=data.get('mergeable'),
+                draft=data.get('draft', False),
+                raw_data=data
+            )
+            
+            return PRResult(
+                success=True,
+                pr_number=pr_info.number,
+                pr_url=pr_info.pr_url,
+                platform=self.config.platform,
+                raw_response=data
+            )
+            
+        except Exception as e:
+            logger.error(f"获取 GitHub PR 失败: {e}")
+            return PRResult(
+                success=False,
+                error_message=str(e),
+                platform=self.config.platform
+            )
+    
+    def update_pr(self, repo_info: RepoInfo, pr_number: int, **kwargs) -> PRResult:
+        """更新 PR"""
+        url = f"{self.config.base_url}/repos/{repo_info.full_name}/pulls/{pr_number}"
+        
+        payload = {}
+        if 'title' in kwargs:
+            payload['title'] = kwargs['title']
+        if 'description' in kwargs:
+            payload['body'] = kwargs['description']
+        if 'state' in kwargs:
+            payload['state'] = kwargs['state']
+        
+        try:
+            response = self._make_request('PATCH', url, data=payload)
+            data = response.json()
+            
+            return PRResult(
+                success=True,
+                pr_number=data['number'],
+                pr_url=data['html_url'],
+                platform=self.config.platform,
+                raw_response=data
+            )
+            
+        except Exception as e:
+            logger.error(f"更新 GitHub PR 失败: {e}")
+            return PRResult(
+                success=False,
+                error_message=str(e),
+                platform=self.config.platform
+            )
+    
+    def close_pr(self, repo_info: RepoInfo, pr_number: int) -> PRResult:
+        """关闭 PR"""
+        return self.update_pr(repo_info, pr_number, state='closed')
+    
+    def merge_pr(self, repo_info: RepoInfo, pr_number: int, **kwargs) -> PRResult:
+        """合并 PR"""
+        url = f"{self.config.base_url}/repos/{repo_info.full_name}/pulls/{pr_number}/merge"
+        
+        payload = {
+            'commit_title': kwargs.get('commit_title', ''),
+            'commit_message': kwargs.get('commit_message', ''),
+            'merge_method': kwargs.get('merge_method', 'merge')
+        }
+        
+        try:
+            response = self._make_request('PUT', url, data=payload)
+            data = response.json()
+            
+            return PRResult(
+                success=True,
+                pr_number=pr_number,
+                platform=self.config.platform,
+                raw_response=data
+            )
+            
+        except Exception as e:
+            logger.error(f"合并 GitHub PR 失败: {e}")
+            return PRResult(
+                success=False,
+                error_message=str(e),
+                platform=self.config.platform
+            )
+    
+    def list_prs(
+        self, 
+        repo_info: RepoInfo, 
+        state: str = "open",
+        per_page: int = 30,
+        page: int = 1
+    ) -> List[PRInfo]:
+        """列出仓库的PR"""
+        url = f"{self.config.base_url}/repos/{repo_info.full_name}/pulls"
+        params = {
+            'state': state,
+            'per_page': str(per_page),
+            'page': str(page)
+        }
+        
+        try:
+            response = self._make_request('GET', url, params=params)
+            data = response.json()
+            
+            prs = []
+            for pr_data in data:
+                pr_info = PRInfo(
+                    number=pr_data['number'],
+                    title=pr_data['title'],
+                    description=pr_data.get('body', ''),
+                    state=pr_data['state'],
+                    source_branch=pr_data['head']['ref'],
+                    target_branch=pr_data['base']['ref'],
+                    author=pr_data['user']['login'],
+                    created_at=pr_data['created_at'],
+                    updated_at=pr_data['updated_at'],
+                    merged_at=pr_data.get('merged_at'),
+                    pr_url=pr_data['html_url'],
+                    labels=[label['name'] for label in pr_data.get('labels', [])],
+                    assignees=[assignee['login'] for assignee in pr_data.get('assignees', [])],
+                    mergeable=pr_data.get('mergeable'),
+                    draft=pr_data.get('draft', False),
+                    raw_data=pr_data
+                )
+                prs.append(pr_info)
+            
+            return prs
+            
+        except Exception as e:
+            logger.error(f"列出 GitHub PR 失败: {e}")
+            return []

autocoder/common/pull_requests/providers/gitlab_provider.py

@@ -0,0 +1,29 @@
+"""
+GitLab API 提供者实现
+"""
+from typing import List
+from .github_provider import GitHubProvider
+from ..models import RepoInfo, PRData, PRResult, PRInfo
+
+
+class GitLabProvider(GitHubProvider):
+    """GitLab API 提供者（基于GitHub提供者）"""
+    
+    def _get_auth_header(self) -> str:
+        """获取认证头"""
+        return f"Bearer {self.config.token}"
+    
+    def create_pr(self, repo_info: RepoInfo, pr_data: PRData) -> PRResult:
+        """创建 Merge Request (GitLab的PR称为MR)"""
+        # 简化实现，实际应该调用GitLab API
+        return super().create_pr(repo_info, pr_data)
+    
+    def list_prs(
+        self, 
+        repo_info: RepoInfo, 
+        state: str = "opened",  # GitLab使用"opened"而不是"open"
+        per_page: int = 30,
+        page: int = 1
+    ) -> List[PRInfo]:
+        """列出仓库的MR"""
+        return super().list_prs(repo_info, state, per_page, page)

autocoder/common/pull_requests/test_module.py

@@ -0,0 +1 @@
+