Gemini赋能代码评审：GitLab与AI的协同，提效代码质量保障

graph TD
    A[GitLab: MR创建/更新] --> B{GitLab Webhook};
    B --> C[中间服务/CI];
    C -- 获取Diff --> D[调用Gemini API];
    D -- 代码分析, 生成建议 --> C;
    C -- 格式化与推送 --> E[飞书等协作平台];
    A -.-> F[人工评审 (AI辅助)];
    E -.-> F;

import google.generativeai as genai
import os
import requests # For GitLab API & Feishu Webhook

# --- Configuration ---
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
# ... (other configurations like GitLab token, project ID, Feishu webhook URL)

genai.configure(api_key=GEMINI_API_KEY)
model = genai.GenerativeModel('gemini-pro')

def get_mr_code_diff(mr_iid: int, project_id: str, gitlab_token: str) -> str:
    """Fetches code diff for a given Merge Request from GitLab."""
    # Simplified: In reality, use GitLab API to fetch diffs for the MR
    print(f"Fetching diff for MR: {mr_iid} in project {project_id}")
    # Placeholder for actual GitLab API call
    return """
diff --git a/example.py b/example.py
--- a/example.py
+++ b/example.py
@@ -1,3 +1,5 @@
 def process_data(data):
- if data is None: return [] # Inconsistent return type
+    if data is None:
+        return [] # Still potentially inconsistent based on expected type
     # ... more complex logic ...
+    return [item * 2 for item in data if item > 0] # Example change
"""

def analyze_with_gemini(code_diff: str) -> str:
    """Sends code diff to Gemini for analysis based on a structured prompt."""
    prompt = f"""
    As an expert Senior Software Engineer, review the following Python code changes (git diff format).
    Focus on:
    1. **Potential Bugs & Logical Errors**: Identify critical flaws.
    2. **Performance Issues**: Suggest optimizations.
    3. **Code Style & Best Practices**: Adherence to PEP8, DRY, SOLID.
    4. **Maintainability & Readability**: Clarity, complexity, comments.
    5. **Security (Basic)**: Obvious vulnerabilities.

    Provide concise, actionable feedback in Markdown.

    Code Diff:
    ```diff
    {code_diff}
    ```
    """
    try:
        response = model.generate_content(prompt)
        return response.text
    except Exception as e:
        # Log error appropriately
        return f"Error during Gemini analysis: {str(e)}"

def send_feedback_to_platform(mr_iid: int, feedback: str, webhook_url: str):
    """Sends formatted feedback to a collaboration platform like Feishu."""
    # Simplified: Construct payload for Feishu card message or similar
    payload = {
        "msg_type": "interactive",
        "card": {
            "header": {"title": {"tag": "plain_text", "content": f"AI Code Review (MR !{mr_iid})"}},
            "elements": [{"tag": "div", "text": {"tag": "lark_md", "content": feedback}}]
        }
    }
    try:
        response = requests.post(webhook_url, json=payload)
        response.raise_for_status()
    except Exception as e:
        # Log error
        print(f"Failed to send feedback: {str(e)}")

# --- Example Orchestration (triggered by webhook) ---
def process_merge_request_event(event_data):
    mr_iid = event_data.get("object_attributes", {}).get("iid")
    project_id = event_data.get("project", {}).get("id")
    # ... (retrieve GITLAB_TOKEN, FEISHU_WEBHOOK_URL from config)

    if not mr_iid or not project_id:
        print("Missing MR IID or Project ID in event data.")
        return

    code_diff = get_mr_code_diff(mr_iid, str(project_id), GITLAB_TOKEN)
    if code_diff:
        analysis_result = analyze_with_gemini(code_diff)
        send_feedback_to_platform(mr_iid, analysis_result, FEISHU_WEBHOOK_URL)

# Note: Production code requires robust error handling, async processing,
# security for tokens, and proper GitLab API client usage.