@echo off
setlocal enabledelayedexpansion

:: 定义目标目录
set "TARGET_DIR=%USERPROFILE%\planet-helper"

echo 星球助手的默认工作目录是 "%TARGET_DIR%"

:: 检查目标目录是否存在
if not exist "%TARGET_DIR%" (
    echo 目标目录不存在，正在创建并复制文件...
    mkdir "%TARGET_DIR%"
    if errorlevel 1 (
        echo 错误：无法创建目录 "%TARGET_DIR%"
        pause
        exit /b 1
    )

    xcopy /E /I "chrome-win64" "%TARGET_DIR%\chrome-win64"
    if errorlevel 1 (
        echo 错误：复制 chrome-win64 失败
        pause
        exit /b 1
    )

    xcopy /E /I "data" "%TARGET_DIR%\data"
    if errorlevel 1 (
        echo 错误：复制 data 失败
        pause
        exit /b 1
    )

    copy "星球助手.exe" "%TARGET_DIR%\星球助手.exe"
    if errorlevel 1 (
        echo 错误：复制 星球助手.exe 失败
        pause
        exit /b 1
    )

    echo 安装完成！
    goto OPEN_FOLDER
)

:: 如果目标目录存在，询问用户操作
echo 目标目录已存在。
echo 请选择操作：
echo 1. 强制覆盖安装（会删除所有数据和exe文件，请再三确认）
echo 2. 只更新程序（会保留数据，但会更新exe文件，请再三确认）
set /p choice=请输入选项（1 或 2）：

if "%choice%"=="1" (
    echo 正在强制覆盖安装...
    :: 将目标目录移动到回收站
    powershell -Command "Remove-Item -Recurse -Force -Path '%TARGET_DIR%' -Confirm:$false"
    if errorlevel 1 (
        echo 错误：无法删除目录 "%TARGET_DIR%"
        pause
        exit /b 1
    )

    mkdir "%TARGET_DIR%"
    if errorlevel 1 (
        echo 错误：无法创建目录 "%TARGET_DIR%"
        pause
        exit /b 1
    )

    xcopy /E /I "chrome-win64" "%TARGET_DIR%\chrome-win64"
    if errorlevel 1 (
        echo 错误：复制 chrome-win64 失败
        pause
        exit /b 1
    )

    xcopy /E /I "data" "%TARGET_DIR%\data"
    if errorlevel 1 (
        echo 错误：复制 data 失败
        pause
        exit /b 1
    )

    copy "星球助手.exe" "%TARGET_DIR%\星球助手.exe"
    if errorlevel 1 (
        echo 错误：复制 星球助手.exe 失败
        pause
        exit /b 1
    )

    echo 强制覆盖安装完成！
) else if "%choice%"=="2" (
    echo 正在更新程序...
    :: 将旧的星球助手.exe移动到回收站
    powershell -Command "Remove-Item -Force -Path '%TARGET_DIR%\星球助手.exe' -Confirm:$false"
    if errorlevel 1 (
        echo 错误：无法删除旧的 星球助手.exe
        pause
        exit /b 1
    )

    copy "星球助手.exe" "%TARGET_DIR%\星球助手.exe"
    if errorlevel 1 (
        echo 错误：复制 星球助手.exe 失败
        pause
        exit /b 1
    )

    echo 程序更新完成！
) else (
    echo 无效的选项，退出脚本。
    pause
    exit /b 1
)

:: 完成提示
:OPEN_FOLDER
echo 操作成功完成！
explorer "%TARGET_DIR%"
pause