import argparse
import os
import sys
from pathlib import Path

# 直接复用你在 test.py 里已经配置好的 client（里面含有 API Key 和 base_url）
try:
    from test import client  # type: ignore
except Exception as import_error:  # noqa: PIE786
    raise SystemExit(
        "无法从 test.py 导入 client。请先确保 test.py 能正常运行，或在此脚本中自行创建 client。"
    ) from import_error


def detect_language_by_suffix(file_path: Path) -> str:
    suffix = file_path.suffix.lower()
    mapping = {
        ".py": "python",
        ".js": "javascript",
        ".ts": "typescript",
        ".tsx": "tsx",
        ".jsx": "jsx",
        ".java": "java",
        ".go": "go",
        ".rs": "rust",
        ".rb": "ruby",
        ".php": "php",
        ".cs": "csharp",
        ".cpp": "cpp",
        ".cc": "cpp",
        ".cxx": "cpp",
        ".c": "c",
        ".h": "c",
        ".json": "json",
        ".yaml": "yaml",
        ".yml": "yaml",
        ".md": "markdown",
        ".sql": "sql",
        ".sh": "bash",
        ".ps1": "powershell",
    }
    return mapping.get(suffix, "")


def read_text_file(file_path: Path) -> str:
    try:
        return file_path.read_text(encoding="utf-8")
    except UnicodeDecodeError:
        # 回退到系统默认编码
        return file_path.read_text(errors="replace")


def build_messages(code_path: Path, code_content: str, instruction: str):
    language = detect_language_by_suffix(code_path)
    system_prompt = (
        "你是严谨的代码审查与重构助手，请用中文回答。"
        "输出请使用结构化 Markdown，包含：\n"
        "1) 概览\n2) 问题清单（按严重程度排序，标注位置/片段）\n"
        "3) 可执行的修复建议\n4) 示例修复代码（只给关键片段）\n"
        "5) 边界与测试要点\n"
        "除非必要，不要重复粘贴整份源码。"
    )
    user_prompt = (
        f"任务：{instruction.strip()}\n"
        f"文件：{code_path.name}\n"
        "代码如下：\n"
        f"```{language}\n{code_content}\n```"
    )
    return [
        {"role": "system", "content": system_prompt},
        {"role": "user", "content": user_prompt},
    ]


def analyze_file(input_path: Path, output_path: Path, instruction: str, model: str = "deepseek-chat") -> None:
    code = read_text_file(input_path)
    messages = build_messages(input_path, code, instruction)

    resp = client.chat.completions.create(
        model=model,
        messages=messages,
        stream=False,
        temperature=0.2,
    )

    content = resp.choices[0].message.content if resp.choices else ""
    output_path.write_text(content, encoding="utf-8")


def main(argv: list[str]) -> int:
    parser = argparse.ArgumentParser(
        description="分析指定代码文件，输出结构化反馈到文件（复用 test.py 的 client）",
    )
    parser.add_argument("input", help="要分析的源代码文件路径")
    parser.add_argument("output", help="把反馈写入到的目标文件路径，例如 review.md")
    parser.add_argument(
        "--instruction",
        default="请找出代码问题、潜在缺陷、可读性/性能/安全改进，并给出修复建议",
        help="自定义任务说明（可选）",
    )
    parser.add_argument(
        "--model",
        default="deepseek-chat",
        help="模型名称（默认 deepseek-chat）",
    )

    args = parser.parse_args(argv)

    input_path = Path(args.input).expanduser().resolve()
    output_path = Path(args.output).expanduser().resolve()

    if not input_path.exists():
        raise SystemExit(f"输入文件不存在: {input_path}")

    output_path.parent.mkdir(parents=True, exist_ok=True)

    analyze_file(input_path, output_path, args.instruction, model=args.model)
    print(f"分析完成，结果已写入: {output_path}")
    return 0


if __name__ == "__main__":
    raise SystemExit(main(sys.argv[1:]))