@echo off
chcp 936 >nul
title 向日葵/AweSun 一键优化配置

:: 检查管理员权限
net session >nul 2>&1
if %errorlevel% neq 0 (
    color 0c
    echo ========================================
    echo    错误：需要管理员权限！
    echo ========================================
    echo.
    echo 请右键此文件，选择 "以管理员身份运行"
    echo.
    pause
    exit /b
)

cd /d "%~dp0"

echo ========================================
echo     正在配置向日葵/AweSun 远程控制软件
echo ========================================
echo.

:: ========== 1. 导入注册表配置 ==========
echo [1/3] 正在导入注册表配置...

set "reg_success=0"

:: 尝试新版 AweSun 注册表路径
reg add "HKEY_USERS\.DEFAULT\Software\AweSun" /v "base_encry_pwd" /t REG_SZ /d "gFjC7qBWrv0=" /f >nul 2>&1
if %errorlevel% equ 0 (
    echo [OK] 注册表配置导入成功！
    set "reg_success=1"
    goto :reg_done
)

:: 尝试旧版向日葵注册表路径（兜底）
reg add "HKEY_USERS\.DEFAULT\Software\Oray\SunLogin\SunloginClient\SunloginInfo" /v "base_encry_pwd" /t REG_SZ /d "gFjC7qBWrv0=" /f >nul 2>&1
if %errorlevel% equ 0 (
    echo [OK] 注册表配置导入成功！
    set "reg_success=1"
    goto :reg_done
)

:: 如果都失败
echo [FAIL] 注册表配置导入失败，请检查权限

:reg_done

:: ========== 2. 从卸载路径获取安装位置 ==========
echo.
echo [2/3] 正在获取安装路径...

set "install_path="

:: 从卸载路径读取 AweSun（新版）
for /f "skip=2 tokens=2*" %%a in ('reg query "HKLM\SOFTWARE\WOW6432Node\Microsoft\Windows\CurrentVersion\Uninstall\Oray AweSun RemoteClient" /v "InstallLocation" 2^>nul') do (
    set "install_path=%%b"
    goto :path_found
)

:: 从卸载路径读取旧版向日葵
for /f "skip=2 tokens=2*" %%a in ('reg query "HKLM\SOFTWARE\WOW6432Node\Microsoft\Windows\CurrentVersion\Uninstall\Oray SunLogin RemoteClient" /v "InstallLocation" 2^>nul') do (
    set "install_path=%%b"
    goto :path_found
)

:: 如果注册表都找不到，尝试常见目录兜底
if exist "C:\Program Files\Aweray\AweSun\" (
    set "install_path=C:\Program Files\Aweray\AweSun"
    goto :path_found
)
if exist "C:\Program Files (x86)\Aweray\AweSun\" (
    set "install_path=C:\Program Files (x86)\Aweray\AweSun"
    goto :path_found
)
if exist "C:\Program Files\SunloginClient\" (
    set "install_path=C:\Program Files\SunloginClient"
    goto :path_found
)
if exist "C:\Program Files\Oray\SunLogin\SunloginClient\" (
    set "install_path=C:\Program Files\Oray\SunLogin\SunloginClient"
    goto :path_found
)
if exist "C:\Program Files (x86)\Oray\SunLogin\SunloginClient\" (
    set "install_path=C:\Program Files (x86)\Oray\SunLogin\SunloginClient"
    goto :path_found
)

:: 手动输入
echo [WARN] 未找到卸载路径！
echo.
echo 请打开软件桌面图标，右键-属性，查看"目标"中的路径
echo 例如: C:\Program Files\Aweray\AweSun
echo.
set /p "install_path=请输入安装目录: "

if not defined install_path (
    echo [ERROR] 未输入路径
    pause
    exit /b
)

set "install_path=%install_path:"=%"
if "%install_path:~-1%"=="\" set "install_path=%install_path:~0,-1%"

:: 检查用户输入的目录
if not exist "%install_path%\AweSun.exe" (
    if not exist "%install_path%\SunloginClient.exe" (
        echo [WARN] 该目录下未找到主程序，请确认路径正确！
        pause
        exit /b
    )
)

:path_found
:: 去除路径末尾的反斜杠
if "%install_path:~-1%"=="\" set "install_path=%install_path:~0,-1%"
echo [OK] 安装路径: %install_path%

:: ========== 3. 写入配置文件 ==========
echo.
echo [3/3] 正在写入配置文件...

if not exist "%install_path%" (
    echo [ERROR] 目录不存在: %install_path%
    pause
    exit /b
)
:: ========== 写入配置内容 ==========
(
echo [base]
echo language=0
echo closeremind=1
echo remind_onoff=0
echo warnning_tips=0
echo auto_full_screen=1
echo full_screen_move_up=1
echo.
echo [security]
echo host_id_freq=0
echo show_wrong_process=1
echo autolockdesktop=0
echo.
echo [local_security]
echo autolockdesktop_new=1
) > "%install_path%\config.ini"

if exist "%install_path%\config.ini" (
    echo [OK] 配置文件写入成功！
) else (
    echo [FAIL] 配置文件写入失败，请检查权限
)

echo.
echo ========================================
echo         配置完成！请重启软件生效。
echo.
echo         已设置长期验证码: 123456
echo ========================================
echo.
pause