import os
import random

from subprocess import run
from pathlib import Path

ON_CF = os.getenv("CF_PAGES") == "1"

if ON_CF:
    print("Running on Cloudflare Pages, trying to git fetch --all")
    result = run(["git", "fetch", "--all"], check=False)
    print(f"git fetch --all: {result}")
    # 提前尝试输出一遍下面这堆信息
    run(["git", "branch", "--show-current"], check=False)
    run(["git", "rev-parse", "HEAD"], check=False)
    run(["git", "describe", "--tags"], check=False)
    run(["git", "log", "-1", "--pretty=%B"], check=False)


def get_env_info() -> dict[str, str]:
    # 读取环境变量
    env_info = {}
    # git branch
    if ON_CF:
        branch = os.getenv("CF_PAGES_BRANCH") or "unknown"
    else:
        branch = run(
            ["git", "branch", "--show-current"],
            capture_output=True,
            text=True,
            encoding="utf-8",
        ).stdout
    env_info["branch"] = branch.strip()
    # git commit hash
    if ON_CF:
        commit = os.getenv("CF_PAGES_COMMIT_SHA") or "unknown"
    else:
        commit = run(
            ["git", "rev-parse", "HEAD"],
            capture_output=True,
            text=True,
            encoding="utf-8",
        ).stdout
    env_info["commit"] = commit.strip()
    # git commit message
    message = run(
        ["git", "log", "-1", "--pretty=%B"],
        capture_output=True,
        text=True,
        encoding="utf-8",
    )
    env_info["message"] = message.stdout.strip()
    # git tag
    if ON_CF:
        tag = run(
            ["git", "describe", "--tags"], capture_output=True, text=True, encoding="utf-8"
        ).stdout.split("-")[0] or "cf_pages"
    else:
        tag = run(
            ["git", "describe", "--tags"], capture_output=True, text=True, encoding="utf-8"
        ).stdout.split("-")[0]
    env_info["tag"] = tag.strip()
    return env_info


if __name__ == "__main__":
    # 虽然但是, 我还是决定用 python 写这个脚本

    border_raw = "/* border: 2px solid marker_color */"
    border_template = "border: 2px solid {};"
    marker_raw = '<div id="version-marker" style="display: none;"></div>'
    marker_template = '<div id="version-marker">{}</div>'

    # 读取环境变量
    env_info = get_env_info()
    tag = env_info["tag"]
    branch = env_info["branch"]
    commit = env_info["commit"]
    message = env_info["message"]

    for file in Path.cwd().rglob("index.html"):
        try:
            with open(file, "r", encoding="utf-8") as f:
                raw_content = f.read()
        except Exception as e:
            print(f"Error: {e}")
            continue
        print(f"Reading: {file}")

        # 替换内容
        # 首先判断是否是 /branch 目录下的 index.html
        if "branch" in str(file):
            # 如果是, 则将颜色替换为 random(这里是为了区分不同的分支, 并且颜色相对固定)
            file_branch_name = file.parent.name
            randomer = random.Random(file_branch_name)
            hash_color = randomer.randint(0, 0xFFFFFF)
            border = border_template.format(f"#{hash_color:06x}")

            # git 信息:
            version_info = f"{file_branch_name}/{branch}:{tag}-{commit[:6]}<br/>{message}"
            marker = marker_template.format(version_info)

            print(f"Branch: {file_branch_name}\n{border}\n{marker}\n")

        else:
            # 淡绿色!
            border = border_template.format("greenyellow")

            # git 信息:
            version_info = f"{branch}:{tag}-{commit[:6]}"
            marker = marker_template.format(version_info)
            print(f"Master: {border}\n{marker}\n")

        raw_content = raw_content.replace(border_raw, border).replace(
            marker_raw, marker
        )

        # 写入文件
        try:
            with open(file, "w", encoding="utf-8") as f:
                f.write(raw_content)
        except Exception as e:
            print(f"Error writing file: {e}")
            continue