#!/bin/bash
# ============================================================
#  YuKnow 扩展 —— macOS/Linux 一键安装
#
#  作用：把扩展加入 Chrome 强制安装列表，规避
#        CRX_REQUIRED_PROOF_MISSING 错误，且保留自动更新。
#
#  用法：
#    macOS : bash install_mac.sh
#    Linux : sudo bash install_mac.sh
# ============================================================
set -u

# Chrome 策略格式：扩展ID;更新地址（注意是分号，ID 在前）
EXT_ID="killfmgkglebjdgmkkicbmldadeejdao"
UPDATE_URL="https://chuangrui.online/update.xml"
POLICY_VALUE="${EXT_ID};${UPDATE_URL}"

OS="$(uname -s)"
echo "======================================"
echo " YuKnow 扩展 安装程序"
echo " 扩展 ID : ${EXT_ID}"
echo " 更新地址: ${UPDATE_URL}"
echo " 系统    : ${OS}"
echo "======================================"
echo ""

if [ "$OS" = "Darwin" ]; then
    echo "→ 写入 Chrome 策略（用户级）"
    defaults write com.google.Chrome ExtensionInstallForcelist -array-add "$POLICY_VALUE"

    if [ $? -eq 0 ]; then
        echo "✓ 策略写入成功"
        # 验证
        echo ""
        echo "当前策略值："
        defaults read com.google.Chrome ExtensionInstallForcelist 2>/dev/null || echo "  (读取失败，可忽略)"
    else
        echo "✗ 写入失败。请尝试手动执行："
        echo "  defaults write com.google.Chrome ExtensionInstallForcelist -array-add '$POLICY_VALUE'"
        exit 1
    fi

elif [ "$OS" = "Linux" ]; then
    DIR="/etc/opt/chrome/policies/managed"
    sudo mkdir -p "$DIR"
    FILE="$DIR/yuknow-extension.json"
    echo "→ 写入策略文件: $FILE"
    sudo tee "$FILE" > /dev/null <<EOF
{
  "ExtensionInstallForcelist": [
    "${EXT_ID};${UPDATE_URL}"
  ]
}
EOF
    if [ $? -eq 0 ]; then
        echo "✓ 策略写入成功"
    else
        echo "✗ 写入失败，请检查 sudo 权限"
        exit 1
    fi
else
    echo "✗ 暂不支持的系统: $OS"
    echo "  Windows 用户请改用 install_windows.reg（双击导入）"
    exit 1
fi

echo ""
echo "======================================"
echo " ✓ 完成"
echo "======================================"
echo ""
echo "下一步（重要）："
echo "  1. 完全退出 Chrome（Mac 按 Cmd+Q，不是关窗口）"
echo "  2. 重新打开 Chrome"
echo "  3. 等待 30~60 秒，扩展会自动安装完成"
echo "     （右上角拼图图标里能看到「投诉单流程整理」）"
echo ""
echo "以后每次发新版，Chrome 会自动更新，你无需任何操作。"
echo ""
echo "如需卸载：chrome://extensions 里点「移除」即可。"