#!/usr/bin/env python3
"""生成预览环境PR评论内容"""

import json
import os
import sys


def generate_deploy_comment(pr_number, preview_url):
    """生成部署成功的评论内容"""
    return f"""🚀 **预览环境已部署**

| 项目 | 详情 |
|------|------|
| PR号 | #{pr_number} |
| 预览链接 | [{preview_url}]({preview_url}) |
| API环境 | staging |

> 💡 预览环境使用 staging API 数据，请勿在预览环境中操作重要数据。
>
> 🔄 每次提交新代码后预览环境会自动更新。
>
> 🗑️ PR 关闭或合并后，预览环境会自动清理。
"""


def generate_cleanup_comment(pr_number):
    """生成清理完成的评论内容"""
    return f"""🗑️ **预览环境已清理**

PR #{pr_number} 已关闭或合并，对应的预览环境已被清理。

> 如有需要，可以重新打开 PR 来重新生成预览环境。
"""


def main():
    mode = sys.argv[1] if len(sys.argv) > 1 else "deploy"
    pr_number = os.environ.get("PR_NUMBER", "")
    preview_url = os.environ.get("PREVIEW_URL", "")

    if mode == "deploy":
        body = generate_deploy_comment(pr_number, preview_url)
    elif mode == "cleanup":
        body = generate_cleanup_comment(pr_number)
    else:
        print(f"Unknown mode: {mode}", file=sys.stderr)
        sys.exit(1)

    print(json.dumps({"body": body}))


if __name__ == "__main__":
    main()