#!/usr/bin/env bash
set -euo pipefail
IFS=$'\n\t'

# ===========================================
# Fast-Note Sync Service 管理脚本 (Premium)
# ===========================================

REPO="haierkeys/fast-note-sync-service"
BIN_BASE="fast-note-sync-service"
INSTALL_DIR="/opt/fast-note"
BIN_PATH="$INSTALL_DIR/$BIN_BASE"
LINK_BIN="/usr/local/bin/fns-bin"
INSTALLER_LINK="/usr/local/bin/fns"
INSTALLER_SELF_PATH="/opt/fast-note/fast-note-installer.sh"
SERVICE_NAME="fast-note.service"
LOG_FILE="/var/log/fast-note.log"
TMPDIR="${TMPDIR:-/tmp}"
GITHUB_RAW="https://github.com/$REPO/releases/download"
GITHUB_API="https://api.github.com/repos/$REPO"
CNB_API_BASE="https://api.cnb.cool/$REPO/-/releases"
CNB_TOKEN="58tjez3744HL9Z10cRaCHdeEPhK"
GITHUB_SCRIPT_URL="https://raw.githubusercontent.com/haierkeys/fast-note-sync-service/master/scripts/quest_install.sh"
CNB_SCRIPT_URL="https://cnb.cool/haierkeys/fast-note-sync-service/-/git/raw/master/scripts/quest_install.sh?cnb"
CNB_MIRROR_CONF="$HOME/.fast-note-mirror"
USE_CNB=false
SUDO=""

# --- Color system // 颜色系统 ---
_RED=$(tput setaf 1)
_GREEN=$(tput setaf 2)
_YELLOW=$(tput setaf 3)
_BLUE=$(tput setaf 4)
_MAGENTA=$(tput setaf 5)
_CYAN=$(tput setaf 6)
_BOLD=$(tput bold)
_ITALIC=$(tput sitm)
_DIM=$(tput dim)
_RESET=$(tput sgr0)

# --- Visual decorations // 视觉装饰 ---
_INFO="  [i] "
_SUCCESS="  [+] "
_WARN="  [!] "
_ERROR="  [-] "
_STEP="  >> "

draw_banner() {
    clear
    load_version
    local ver_display="${_YELLOW}$L_NOT_INSTALLED${_RESET}"
    if [ -n "$INSTALLED_VER" ]; then
        local clean_v="${INSTALLED_VER#v}"
        ver_display="${_GREEN}$BIN_BASE v${clean_v}${_RESET}"
    fi
    
    local source_display="${_BLUE}GitHub${_RESET}"
    if [ "$USE_CNB" = "true" ]; then
        source_display="${_MAGENTA}CNB.cool${_RESET}"
    fi
    
    local svc_file="N/A"
    local os_type
    os_type=$(detect_os)
    if [ "$os_type" = "linux" ]; then
        svc_file="/etc/systemd/system/fast-note.service"
        elif [ "$os_type" = "darwin" ]; then
        svc_file="/Library/LaunchDaemons/com.haierkeys.fast-note.plist"
    fi
    
    local latest_v
    latest_v=$(get_latest_tag)
    local clean_lv="${latest_v#v}"
    local latest_display="${_GREEN}v${clean_lv}${_RESET}"
    
    # Highlight latest version if update available
    # 如果有更新，高亮显示最新版本
    if [ -n "$INSTALLED_VER" ] && [ "${INSTALLED_VER#v}" != "$clean_lv" ]; then
        latest_display="${_YELLOW}v${clean_lv} (Update Available)${_RESET}"
        [ "$CURRENT_LANG" = "zh" ] && latest_display="${_YELLOW}v${clean_lv} (有新版本)${_RESET}"
    fi
    
    cat <<EOF
${_CYAN}${_BOLD}
    ______           __     _   __      __          _____
   / ____/___ ______/ /_   / | / /___  / /____     / ___/__  ______  _____
  / /_  / __  / ___/ __/  /  |/ / __ \/ __/ _ \    \__ \/ / / / __ \/ ___/
 / __/ / /_/ (__  ) /_   / /|  / /_/ / /_/  __/   ___/ / /_/ / / / / /__
/_/    \__,_/____/\__/  /_/ |_/\____/\__/\___/   /____/\__, /_/ /_/\___/
                                                      /____/

       Fast Note Sync Service Manager Script
   ================================================
   $L_CUR_VER: $ver_display
   $L_LATEST_VER: $latest_display
   $L_SOURCE  : $source_display
   ------------------------------------------------
   ${_DIM}$L_PATH_INFO:${_RESET}
   ${_BLUE}$L_MAIN_DIR :${_RESET} ${_BOLD}$INSTALL_DIR${_RESET}
   ${_BLUE}$L_DATA_DIR :${_RESET} ${_BOLD}$INSTALL_DIR/storage${_RESET}
   ${_BLUE}$L_CONF_DIR :${_RESET} ${_BOLD}$INSTALL_DIR/config${_RESET}
   ${_BLUE}$L_LOG_FILE_PATH :${_RESET} ${_BOLD}$LOG_FILE${_RESET}
   ${_BLUE}$L_SVC_FILE_PATH :${_RESET} ${_BOLD}$svc_file${_RESET}
   ================================================
EOF
    echo -e "\n"
}

msg() { echo -e "${_BOLD}$1${_RESET}"; }
info() { echo -e "${_INFO}${_CYAN}$1${_RESET}"; }
success() { echo -e "${_SUCCESS}${_GREEN}$1${_RESET}"; }
warn() { echo -e "${_WARN}${_YELLOW}$1${_RESET}"; }
error() { echo -e "${_ERROR}${_RED}$1${_RESET}"; }
step() { echo -e "${_STEP}${_BLUE}$1${_RESET}"; }

# --- Language Support ---
LANG_CONF="$HOME/.fast-note-sync.lang"
VERSION_CONF="$INSTALL_DIR/.version"
VERSION_CONF_GO="$INSTALL_DIR/config/lastVersion"
CURRENT_LANG="en" # Default to English
INSTALLED_VER=""

save_lang() {
    echo "$CURRENT_LANG" > "$LANG_CONF" 2>/dev/null || true
}

load_lang() {
    local force_load="${1:-}"
    # Read from file only if forced or if it's the first time
    if [ "$force_load" = "init" ] && [ -f "$LANG_CONF" ]; then
        CURRENT_LANG=$(cat "$LANG_CONF" 2>/dev/null | tr -d '[:space:]' || echo "en")
    fi
    
    if [ "$CURRENT_LANG" = "zh" ]; then
        L_MENU_1="安装 / 升级服务"
        L_MENU_1_D="下载服务程序并自动安装管理工具至系统"
        L_MENU_2="启动服务"
        L_MENU_2_D="在后台启动同步服务"
        L_MENU_3="停止服务"
        L_MENU_3_D="终止正在运行的服务进程"
        L_MENU_4="服务状态"
        L_MENU_4_D="检查运行状态并预览最新日志"
        L_MENU_5="全部卸载"
        L_MENU_5_D="彻底移除程序、配置及所有日志文件"
        L_MENU_6="安装脚本到系统"
        L_MENU_6_D="将管理工具添加到全局快捷命令 fns"
        L_MENU_7="设置开机启动"
        L_MENU_7_D="配置 Systemd (Linux) 或 Launchd (macOS) 开机自启"
        L_MENU_8="切换下载镜像"
        L_MENU_8_D="在 GitHub 与 CNB 镜像之间切换"
        L_MENU_0="退出"
        L_MENU_L="Switch to English (切换至英文)"
        L_SWITCH_TO_CNB="已切换至 CNB 镜像"
        L_SWITCH_TO_GITHUB="已切换至 GitHub 镜像"
        L_SELECT="请选择"
        L_INPUT_VER="输入版本 (留空使用 latest)"
        L_INPUT_URL="输入脚本 URL (留空复制本地"
        L_ENTER_URL="输入脚本 URL [默认:"
        L_PATH_WARN="警告: 安装目录 %s 不在您的 PATH 环境变量中。"
        L_PATH_FIX="请手动添加以在任何地方使用快捷命令: export PATH=\$PATH:%s"
        L_SOURCE="下载源"
        
        L_ERR_ROOT="需要 root 权限或安装 sudo 后重试"
        L_TRY_DL="尝试下载"
        L_DL_FAIL_API="直接下载失败，尝试通过 API 查找..."
        L_ERR_NO_REL="无法获取 release 信息"
        L_FOUND_ASSET="找到资产"
        L_ERR_NO_ASSET="未能找到合适的资产"
        L_EXTRACTING="正在解压资产到"
        L_ERR_EXTRACT="解压失败"
        L_ERR_NO_EXE="未在压缩包中找到可执行文件"
        L_LINK_CREATED="已创建快捷命令"
        
        L_SVC_RUNNING="服务已经在运行中"
        L_STARTING="正在启动服务..."
        L_START_SUCCESS="服务已成功启动"
        L_LOG_PREVIEW="实时日志预览"
        L_START_FAIL="启动失败！请检查日志详情"
        L_STOPPING="正在发送停止信号..."
        L_STOP_SUCCESS="服务已停止"
        L_STATUS="状态"
        L_STATUS_RUN="运行中"
        L_STATUS_STOP="已停止"
        L_LOG_RECENT="最近 20 行日志预览"
        
        L_UN_WARN="准备执行全部卸载！将删除目录、日志及所有配置。"
        L_UN_CONFIRM="确认执行全部卸载吗？"
        L_UN_CANCEL="已取消卸载"
        L_CLEAN_PROC="清理残留进程..."
        L_CLEAN_FILES="移除安装目录与文件..."
        L_UN_DONE="全部卸载完成"
        
        L_DL_SCRIPT="从指定 URL 下载安装脚本..."
        L_ERR_DL_SCRIPT="下载安装脚本失败"
        L_CP_SCRIPT="复制当前脚本到系统目录..."
        L_ST_DL_SCRIPT="脚本通过 stdin 执行，正在尝试自动获取..."
        L_INST_DONE="安装脚本已就绪"
        
        L_PRE_DL="准备下载 fast-note"
        L_INST_ALL_DONE="安装/升级流程已完成"
        L_INST_TIP="提示: 输入 fns 或选择菜单启动服务。"
        L_INVALID="无效选项，请重新选择"
        L_USAGE="用法"
        L_CUR_VER="当前版本"
        L_LATEST_VER="最新版本"
        L_NOT_INSTALLED="未安装"
        L_PATH_INFO="路径信息"
        L_MAIN_DIR="程序目录"
        L_DATA_DIR="数据目录"
        L_CONF_DIR="配置目录"
        L_LOG_FILE_PATH="日志文件"
        L_SVC_FILE_PATH="系统服务"
        
        L_AUTO_LINUX="正在配置 Systemd 开机自启..."
        L_AUTO_MAC="正在配置 Launchd 开机自启..."
        L_AUTO_WIN="Windows 暂不支持自动配置服务，请手动添加计划任务。"
        L_AUTO_DONE="开机自启配置完成"
        L_AUTO_FAIL="开机自启配置失败"
    else
        L_MENU_1="Install / Update Service"
        L_MENU_1_D="Download service and install manager to system"
        L_MENU_2="Start Service"
        L_MENU_2_D="Start the sync service in background"
        L_MENU_3="Stop Service"
        L_MENU_3_D="Terminate the running service process"
        L_MENU_4="Service Status"
        L_MENU_4_D="Check status and preview recent logs"
        L_MENU_5="Uninstall All"
        L_MENU_5_D="Remove program, config, and all logs"
        L_MENU_6="Install Self to System"
        L_MENU_6_D="Add this tool to global commands (fns)"
        L_MENU_7="Set Auto-Start"
        L_MENU_7_D="Configure Systemd (Linux) or Launchd (macOS) auto-start"
        L_MENU_8="Switch Download Mirror"
        L_MENU_8_D="Switch between GitHub and CNB mirror"
        L_MENU_0="Quit"
        L_MENU_L="切换至中文 (Switch to Chinese)"
        L_SWITCH_TO_CNB="Switched to CNB mirror"
        L_SWITCH_TO_GITHUB="Switched to GitHub mirror"
        L_SELECT="Please select"
        L_INPUT_VER="Enter version (leave blank for latest)"
        L_INPUT_URL="Enter script URL (leave blank to copy local"
        L_ENTER_URL="Enter script URL [Default:"
        L_PATH_WARN="WARNING: Install directory %s is not in your PATH."
        L_PATH_FIX="Add it manually to use commands everywhere: export PATH=\$PATH:%s"
        L_SOURCE="Download Source"
        
        L_ERR_ROOT="Root privileges or sudo required"
        L_TRY_DL="Trying to download"
        L_DL_FAIL_API="Direct download failed, trying via API..."
        L_ERR_NO_REL="Failed to get release info"
        L_FOUND_ASSET="Asset found"
        L_ERR_NO_ASSET="No suitable asset found"
        L_EXTRACTING="Extracting asset to"
        L_ERR_EXTRACT="Extraction failed"
        L_ERR_NO_EXE="Executable not found in package"
        L_LINK_CREATED="Symbolic link created"
        
        L_SVC_RUNNING="Service is already running"
        L_STARTING="Starting service..."
        L_START_SUCCESS="Service started successfully"
        L_LOG_PREVIEW="Real-time log preview"
        L_START_FAIL="Start failed! Please check logs"
        L_STOPPING="Sending stop signal..."
        L_STOP_SUCCESS="Service stopped"
        L_STATUS="Status"
        L_STATUS_RUN="Running"
        L_STATUS_STOP="Stopped"
        L_LOG_RECENT="Recent 20 lines of log"
        
        L_UN_WARN="Preparing full uninstall! All data will be deleted."
        L_UN_CONFIRM="Confirm full uninstall?"
        L_UN_CANCEL="Uninstall cancelled"
        L_CLEAN_PROC="Cleaning up processes..."
        L_CLEAN_FILES="Removing directories and files..."
        L_UN_DONE="Full uninstall completed"
        
        L_DL_SCRIPT="Downloading script from URL..."
        L_ERR_DL_SCRIPT="Failed to download script"
        L_CP_SCRIPT="Copying current script to system..."
        L_ST_DL_SCRIPT="Running via stdin, trying to fetch automatically..."
        L_INST_DONE="Installer is ready"
        
        L_PRE_DL="Preparing to download fast-note"
        L_INST_ALL_DONE="Install/Update process completed"
        L_INST_TIP="Tip: Type fns or use menu to start service."
        L_INVALID="Invalid option, please try again"
        L_USAGE="Usage"
        L_CUR_VER="Installed Version"
        L_LATEST_VER="Latest Version"
        L_NOT_INSTALLED="Not installed"
        L_PATH_INFO="Path Info"
        L_MAIN_DIR="Main Dir"
        L_DATA_DIR="Data Dir"
        L_CONF_DIR="Conf Dir"
        L_LOG_FILE_PATH="Log File"
        L_SVC_FILE_PATH="Svc File"
        
        L_AUTO_LINUX="Configuring Systemd auto-start..."
        L_AUTO_MAC="Configuring Launchd auto-start..."
        L_AUTO_WIN="Windows does not support auto-service config yet. Please add manually."
        L_AUTO_DONE="Auto-start configured successfully"
        L_AUTO_FAIL="Auto-start configuration failed"
    fi
}
# Initial load // 初始加载
load_lang "init"

# Detect mirror source from script arguments or saved config
# 从脚本参数或已保存配置中检测镜像来源
save_mirror() {
    echo "$USE_CNB" > "$CNB_MIRROR_CONF" 2>/dev/null || true
}

load_mirror() {
    if [ -f "$CNB_MIRROR_CONF" ]; then
        local saved
        saved=$(cat "$CNB_MIRROR_CONF" 2>/dev/null | tr -d '[:space:]' || echo "false")
        if [ "$saved" = "true" ]; then
            USE_CNB=true
        else
            USE_CNB=false
        fi
    fi
}

parse_mirror_from_args() {
    # Check if any argument contains '?cnb' or '--cnb' flag
    # 检查是否含有 ?cnb 或 --cnb 参数
    for arg in "$@"; do
        if [[ "$arg" == *"?cnb"* ]] || [[ "$arg" == "--cnb" ]]; then
            USE_CNB=true
            return
        fi
    done
    # No cnb flag found; load from saved config
    # 未找到 cnb 标志，从已保存配置加载
    load_mirror
}
parse_mirror_from_args "$@"

# --- Version Tracking // 版本追踪 ---
load_version() {
    INSTALLED_VER=""
    local go_ver="" shell_ver=""
    if [ -f "$VERSION_CONF_GO" ]; then
        go_ver=$(cat "$VERSION_CONF_GO" 2>/dev/null | tr -d '[:space:]' || echo "")
    fi
    if [ -f "$VERSION_CONF" ]; then
        shell_ver=$(cat "$VERSION_CONF" 2>/dev/null | tr -d '[:space:]' || echo "")
    fi
    if [ -n "$go_ver" ]; then
        INSTALLED_VER="$go_ver"
        # Sync .version to match config/lastVersion when they differ
        if [ "$go_ver" != "$shell_ver" ] && [ -d "$INSTALL_DIR" ]; then
            echo "$go_ver" | tee "$VERSION_CONF" >/dev/null 2>&1 || true
        fi
    elif [ -n "$shell_ver" ]; then
        INSTALLED_VER="$shell_ver"
    fi
}

save_version() {
    local v="$1"
    if [ -d "$INSTALL_DIR" ]; then
        echo "$v" | $SUDO tee "$VERSION_CONF" >/dev/null 2>&1 || true
    fi
}

ensure_root() {
    if [ "$EUID" -ne 0 ]; then
        if command -v sudo >/dev/null 2>&1; then
            SUDO="sudo"
        else
            error "$L_ERR_ROOT"
            exit 1
        fi
    else
        SUDO=""
    fi
}

detect_os() {
    local os="$(uname -s | tr '[:upper:]' '[:lower:]')"
    case "$os" in
        linux) echo "linux" ;;
        darwin) echo "darwin" ;;
        mingw*|msys*|cygwin*) echo "windows" ;;
        *) echo "$os" ;;
    esac
}

_arch_map() {
    local a="$(uname -m)"
    case "$a" in
        x86_64|amd64) echo "amd64" ;;
        aarch64|arm64) echo "arm64" ;;
        armv7*|armv6*) echo "armv7" ;;
        *) echo "$a" ;;
    esac
}

# try get latest tag from GitHub API; fallback to "latest" string
# 尝试从 GitHub API 获取最新 tag；失败则回退到 "latest" 字符串
get_latest_tag() {
    if [ "$USE_CNB" = "true" ]; then
        local latest
        # CNB releases API returns a list; we only need the first object's tag_name
        # CNB releases API 返回一个列表；我们只需要第一个对象的 tag_name
        latest=$(curl -fsSL -H "Accept: application/vnd.cnb.api+json" -H "Authorization: Bearer $CNB_TOKEN" "$CNB_API_BASE" | \
        grep -oE '"tag_name"[[:space:]]*:[[:space:]]*"[^"]+"' | head -n1 | sed -E 's/.*"([^"]+)"$/\1/' || true)
        if [ -n "$latest" ]; then echo "$latest"; return 0; fi
    fi
    
    if command -v curl >/dev/null 2>&1; then
        local latest
        latest="$(curl -fsSL "$GITHUB_API/releases/latest" 2>/dev/null | sed -nE 's/.*"tag_name"[[:space:]]*:[[:space:]]*"([^"]+)".*/\1/p' || true)"
        if [ -n "$latest" ]; then
            echo "$latest"
            return 0
        fi
    fi
    echo "latest"
}

# construct expected asset file name: fast-note-sync-service-<ver>-<os>-<arch>.tar.gz
# 构建预期的资产文件名：fast-note-sync-service-<ver>-<os>-<arch>.tar.gz
asset_name_for() {
    local ver="$1" os="$2" arch="$3"
    # strip leading v if present in tag
    local clean_ver="${ver#v}"
    echo "${BIN_BASE}-${clean_ver}-${os}-${arch}.tar.gz"
}

# download using direct URL based on naming convention; if fails, try API lookup
# 基于命名规范使用直接 URL 下载；如果失败，尝试 API 查找
download_release_asset() {
    local ver="$1" os="$2" arch="$3"
    local clean_ver="${ver#v}"
    local asset_name
    asset_name="$(asset_name_for "$ver" "$os" "$arch")"
    local out="$TMPDIR/$asset_name"
    
    if [ "$USE_CNB" = "true" ]; then
        # Resolve "latest" tag via CNB API if needed
        # 如需要，通过 CNB API 解析 "latest" tag
        local cnb_tag="$ver"
        if [ "$cnb_tag" = "latest" ]; then
            local api_tag
            # Get the first occurrence of tag_name from the top of the list
            # 从列表顶部获取第一个 tag_name 的出现
            api_tag=$(curl -fsSL -H "Accept: application/vnd.cnb.api+json" -H "Authorization: Bearer $CNB_TOKEN" "$CNB_API_BASE" | \
            grep -oE '"tag_name"[[:space:]]*:[[:space:]]*"[^"]+"' | head -n1 | sed -E 's/.*"([^"]+)"$/\1/' || true)
            [ -n "$api_tag" ] && cnb_tag="$api_tag"
        fi
        
        # Construct CNB download URL directly: https://cnb.cool/{repo}/-/releases/download/{tag}/{filename}
        # 直接构造 CNB 下载 URL，无需解析 API JSON
        local cnb_url="https://cnb.cool/$REPO/-/releases/download/${cnb_tag}/${asset_name}"
        info "$L_TRY_DL (CNB): ${_BOLD}$cnb_url${_RESET}" >&2
        if curl -fSL -o "$out" "$cnb_url"; then
            echo "$out"
            return 0
        fi
        warn "$L_DL_FAIL_API" >&2
    fi
    
    local url="$GITHUB_RAW/${clean_ver}/${asset_name}"
    
    info "$L_TRY_DL: ${_BOLD}$url${_RESET}" >&2
    if curl -fSL -o "$out" "$url"; then
        echo "$out"
        return 0
    fi
    
    warn "$L_DL_FAIL_API" >&2
    
    # try API: find release by tag or latest
    local release_json
    if [ "$ver" = "latest" ] || [ -z "$ver" ]; then
        release_json="$(curl -fsSL "$GITHUB_API/releases/latest" 2>/dev/null || true)"
    else
        # find release by tag
        release_json="$(curl -fsSL "$GITHUB_API/releases/tags/$ver" 2>/dev/null || true)"
        if [ -z "$release_json" ]; then
            release_json="$(curl -fsSL "$GITHUB_API/releases" 2>/dev/null | grep -A20 "\"tag_name\": \"$ver\"" -n || true)"
        fi
    fi
    
    if [ -z "$release_json" ]; then
        echo "${_RED}$L_ERR_NO_REL${_RESET}" >&2
        return 2
    fi
    
    # If jq exists use it
    if command -v jq >/dev/null 2>&1; then
        local asset_url
        asset_url="$(echo "$release_json" | jq -r --arg name "$asset_name" '.assets[] | select(.name==$name) | .browser_download_url' 2>/dev/null || true)"
        if [ -z "$asset_url" ]; then
            asset_url="$(echo "$release_json" | jq -r --arg os "$os" --arg arch "$arch" '.assets[] | select(.name|test($os) and .name|test($arch)) | .browser_download_url' 2>/dev/null | head -n1 || true)"
        fi
        if [ -n "$asset_url" ]; then
            info "$L_FOUND_ASSET: ${_BOLD}$asset_url${_RESET}" >&2
            curl -L --fail -o "$out" "$asset_url"
            echo "$out"
            return 0
        fi
    else
        # fallback to grep/sed extraction
        local asset_url
        asset_url="$(echo "$release_json" | grep -oE '"browser_download_url"[[:space:]]*:[[:space:]]*"[^"]+' | sed -E 's/.*:"([^"]+)$/\1/' | grep "$os" | grep "$arch" | head -n1 || true)"
        if [ -n "$asset_url" ]; then
            info "$L_FOUND_ASSET: ${_BOLD}$asset_url${_RESET}" >&2
            curl -L --fail -o "$out" "$asset_url"
            echo "$out"
            return 0
        fi
    fi
    
    error "$L_ERR_NO_ASSET (os:$os arch:$arch)" >&2
    return 3
}

install_binary_from_tar() {
    local tarball="$1"
    ensure_root
    
    # 准备临时解压目录
    local extract_tmp
    extract_tmp="$(mktemp -d)"
    trap "$SUDO rm -rf '$extract_tmp'" EXIT
    
    step "$L_EXTRACTING $INSTALL_DIR ..."
    $SUDO tar -xzf "$tarball" -C "$extract_tmp" || { error "$L_ERR_EXTRACT"; return 1; }
    
    # 1. 强制更新二进制程序
    local exe
    if [ -f "$extract_tmp/$BIN_BASE" ]; then
        exe="$extract_tmp/$BIN_BASE"
    else
        exe="$(find "$extract_tmp" -maxdepth 2 -type f -perm -111 | head -n1 || true)"
    fi
    
    if [ -n "$exe" ]; then
        $SUDO mkdir -p "$INSTALL_DIR"
        $SUDO mv -f "$exe" "$BIN_PATH"
        $SUDO chmod +x "$BIN_PATH"
    else
        error "$L_ERR_NO_EXE"
        return 2
    fi
    
    # 2. 保护性移动配置文件
    if [ -d "$extract_tmp/config" ]; then
        $SUDO mkdir -p "$INSTALL_DIR/config"
        # 仅拷贝目标位置不存在的文件，防止覆盖用户修改过的 config.yaml 等
        $SUDO cp -rn "$extract_tmp/config/"* "$INSTALL_DIR/config/" 2>/dev/null || true
    fi
    
    # 3. 创建软链接
    $SUDO mkdir -p "$(dirname "$LINK_BIN")"
    $SUDO ln -sf "$BIN_PATH" "$LINK_BIN"
    success "$L_LINK_CREATED: ${_BOLD}$LINK_BIN${_RESET}"
}


# Auto-Start config // 开机自启配置
enable_autostart() {
    local os="$(detect_os)"
    
    if [ "$os" = "linux" ]; then
        if [ -d "/etc/systemd/system" ] && command -v systemctl >/dev/null 2>&1; then
            step "$L_AUTO_LINUX"
            # create service file
            cat <<EOF | $SUDO tee /etc/systemd/system/fast-note.service >/dev/null
[Unit]
Description=Fast Note Sync Service
After=network.target

[Service]
Type=simple
WorkingDirectory=$INSTALL_DIR
ExecStart=$BIN_PATH run
Restart=on-failure
User=root
StandardOutput=append:$LOG_FILE
StandardError=append:$LOG_FILE

[Install]
WantedBy=multi-user.target
EOF
            $SUDO systemctl daemon-reload || true
            $SUDO systemctl enable fast-note.service || true
            # 如果没有运行，顺便启动
            if ! systemctl is-active --quiet fast-note.service; then
                $SUDO systemctl start fast-note.service || true
            fi
            success "$L_AUTO_DONE"
            return 0
        fi
        elif [ "$os" = "darwin" ]; then
        # macOS launchd
        step "$L_AUTO_MAC"
        ensure_root
        local plist_path="/Library/LaunchDaemons/com.haierkeys.fast-note.plist"
        
        cat <<EOF | $SUDO tee "$plist_path" >/dev/null
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
    <key>Label</key>
    <string>com.haierkeys.fast-note</string>
    <key>WorkingDirectory</key>
    <string>$INSTALL_DIR</string>
    <key>ProgramArguments</key>
    <array>
        <string>$BIN_PATH</string>
        <string>run</string>
    </array>
    <key>RunAtLoad</key>
    <true/>
    <key>KeepAlive</key>
    <true/>
    <key>StandardOutPath</key>
    <string>$LOG_FILE</string>
    <key>StandardErrorPath</key>
    <string>$LOG_FILE</string>
</dict>
</plist>
EOF
        $SUDO chmod 644 "$plist_path"
        # unload if exists
        $SUDO launchctl unload -w "$plist_path" 2>/dev/null || true
        $SUDO launchctl load -w "$plist_path" 2>/dev/null || true
        success "$L_AUTO_DONE"
        return 0
        elif [ "$os" = "windows" ]; then
        warn "$L_AUTO_WIN"
        return 0
    fi
    
    warn "$L_AUTO_FAIL: No supported service manager found."
}


# Data Migration // 数据迁移
# Check if /storage exists (caused by WorkingDir bug) and migrate it
auto_migrate_data() {
    if [ -d "/storage" ]; then
        ensure_root
        warn "Found potential data in root directory /storage, migrating to $INSTALL_DIR/storage ..."
        $SUDO mkdir -p "$INSTALL_DIR"
        # Combine directories
        $SUDO cp -af /storage/* "$INSTALL_DIR/storage/" 2>/dev/null || true
        # Backup then remove
        local backup_tag
        backup_tag=$(date +%Y%m%d%H%M%S)
        $SUDO mv /storage "/storage.bak.$backup_tag"
        success "Data migrated to $INSTALL_DIR/storage. Original moved to /storage.bak.$backup_tag"
    fi
}


# Service control functions // 服务控制函数
start_service() {
    auto_migrate_data
    ensure_root
    local os="$(detect_os)"
    
    # 优先尝试 Systemd
    if [ "$os" = "linux" ] && [ -f "/etc/systemd/system/fast-note.service" ]; then
        step "Systemd: $L_STARTING"
        $SUDO systemctl start fast-note.service
        sleep 2
        if systemctl is-active --quiet fast-note.service; then
            success "$L_START_SUCCESS"
            return 0
        fi
        # 优先尝试 Launchd
        elif [ "$os" = "darwin" ] && [ -f "/Library/LaunchDaemons/com.haierkeys.fast-note.plist" ]; then
        step "Launchd: $L_STARTING"
        ensure_root
        $SUDO launchctl load -w "/Library/LaunchDaemons/com.haierkeys.fast-note.plist" 2>/dev/null || true
        success "$L_START_SUCCESS"
        return 0
    fi
    
    # Fallback to manual
    if pgrep -f "$BIN_PATH" >/dev/null 2>&1; then
        warn "$L_SVC_RUNNING (PID: $(pgrep -f "$BIN_PATH"))"
        return
    fi
    step "$L_STARTING (nohup)"
    # 保持 bash -c 包装以解决重定向权限问题
    $SUDO bash -c "set -m; cd $INSTALL_DIR && nohup $BIN_PATH run >> $LOG_FILE 2>&1 &"
    
    sleep 2
    if pgrep -f "$BIN_PATH" >/dev/null 2>&1; then
        success "$L_START_SUCCESS"
        info "$L_LOG_PREVIEW: ${_BOLD}tail -f $LOG_FILE${_RESET}"
    else
        error "$L_START_FAIL: ${_BOLD}sudo tail -n 20 $LOG_FILE${_RESET}"
    fi
}

stop_service() {
    ensure_root
    local os="$(detect_os)"
    
    # 优先尝试 Systemd
    if [ "$os" = "linux" ] && [ -f "/etc/systemd/system/fast-note.service" ]; then
        step "Systemd: $L_STOPPING"
        $SUDO systemctl stop fast-note.service
        success "$L_STOP_SUCCESS"
        return 0
        # 优先尝试 Launchd
        elif [ "$os" = "darwin" ] && [ -f "/Library/LaunchDaemons/com.haierkeys.fast-note.plist" ]; then
        step "Launchd: $L_STOPPING"
        ensure_root
        $SUDO launchctl unload -w "/Library/LaunchDaemons/com.haierkeys.fast-note.plist" 2>/dev/null || true
        success "$L_STOP_SUCCESS"
        return 0
    fi
    
    if ! pgrep -f "$BIN_PATH" >/dev/null 2>&1; then
        return 0
    fi
    step "$L_STOPPING"
    $SUDO pkill -f "$BIN_PATH" || true
    success "$L_STOP_SUCCESS"
}

status_service() {
    local pids
    pids="$(pgrep -f "$BIN_PATH" || true)"
    if [ -n "$pids" ]; then
        success "$L_STATUS: ${_BOLD}$L_STATUS_RUN${_RESET} (PID: $pids)"
        echo -e "\n${_BLUE}${_BOLD}$L_LOG_RECENT ($LOG_FILE):${_RESET}"
        echo "${_CYAN}------------------------------------------------------------${_RESET}"
        $SUDO tail -n 20 "$LOG_FILE" 2>/dev/null || true
        echo "${_CYAN}------------------------------------------------------------${_RESET}"
    else
        warn "$L_STATUS: ${_BOLD}$L_STATUS_STOP${_RESET}"
    fi
}


full_uninstall() {
    ensure_root
    warn "$L_UN_WARN"
    read -rp "  $(echo -e "${_BOLD}$L_UN_CONFIRM [y/N]: ${_RESET}")" yn
    yn="${yn:-N}"
    if [[ ! "$yn" =~ ^[Yy]$ ]]; then
        info "$L_UN_CANCEL"
        return 0
    fi
    
    # Stop service first
    stop_service 2>/dev/null || true
    
    step "$L_CLEAN_PROC"
    $SUDO pkill -f "$BIN_PATH" || true
    
    # Cleanup auto-start
    if [ -f "/etc/systemd/system/fast-note.service" ]; then
        $SUDO systemctl disable fast-note.service 2>/dev/null || true
        $SUDO rm -f "/etc/systemd/system/fast-note.service"
        $SUDO systemctl daemon-reload 2>/dev/null || true
    fi
    if [ -f "/Library/LaunchDaemons/com.haierkeys.fast-note.plist" ]; then
        ensure_root
        $SUDO launchctl unload -w "/Library/LaunchDaemons/com.haierkeys.fast-note.plist" 2>/dev/null || true
        $SUDO rm -f "/Library/LaunchDaemons/com.haierkeys.fast-note.plist"
    fi
    
    step "$L_CLEAN_FILES"
    $SUDO rm -rf "$INSTALL_DIR" || true
    $SUDO rm -f "$LINK_BIN" || true
    $SUDO rm -f "$INSTALLER_SELF_PATH" || true
    $SUDO rm -f "$INSTALLER_LINK" || true
    # 清理旧命令 (兼容性)
    $SUDO rm -f "/usr/local/bin/fast-note" || true
    $SUDO rm -f "/usr/local/bin/fast-note-installer" || true
    $SUDO rm -f "$LOG_FILE" || true
    # 清理用户级配置文件 / Clean up per-user config files
    rm -f "$LANG_CONF" || true
    rm -f "$CNB_MIRROR_CONF" || true
    
    success "$L_UN_DONE"
}

install_self() {
    ensure_root
    local src_url="${1:-}"
    
    # Auto-select script URL based on current mirror setting
    # 根据当前镜像设置自动选择脚本 URL
    if [ -z "$src_url" ]; then
        if [ "$USE_CNB" = "true" ]; then
            src_url="$CNB_SCRIPT_URL"
        else
            src_url="$GITHUB_SCRIPT_URL"
        fi
    fi
    
    # 如果没有指定 URL 且当前不是通过本地文件运行（如 curl|bash 或 stdin）
    if [ -z "$src_url" ] && [ ! -f "$0" ]; then
        warn "$L_ST_DL_SCRIPT"
        src_url="$GITHUB_SCRIPT_URL"
    fi
    
    if [ -n "$src_url" ]; then
        step "$L_DL_SCRIPT"
        $SUDO mkdir -p "$(dirname "$INSTALLER_SELF_PATH")"
        $SUDO curl -fsSL "$src_url" -o "$INSTALLER_SELF_PATH" || { error "$L_ERR_DL_SCRIPT"; return 1; }
    else
        # 检查是否已经是同一个文件（例如通过 fns 链接运行时）
        if [ ! "$0" -ef "$INSTALLER_SELF_PATH" ]; then
            step "$L_CP_SCRIPT"
            $SUDO mkdir -p "$(dirname "$INSTALLER_SELF_PATH")"
            $SUDO cp -f "$0" "$INSTALLER_SELF_PATH"
        fi
    fi
    
    $SUDO chmod +x "$INSTALLER_SELF_PATH"
    $SUDO mkdir -p "$(dirname "$INSTALLER_LINK")"
    
    # Create fns wrapper that passes mirror flag when USE_CNB=true
    # 创建 fns 包装脚本，在 USE_CNB=true 时传递镜像参数
    if [ "$USE_CNB" = "true" ]; then
        cat <<'WRAPPER' | $SUDO tee "$INSTALLER_LINK" >/dev/null
#!/usr/bin/env bash
exec /opt/fast-note/fast-note-installer.sh --cnb "$@"
WRAPPER
    else
        cat <<'WRAPPER' | $SUDO tee "$INSTALLER_LINK" >/dev/null
#!/usr/bin/env bash
exec /opt/fast-note/fast-note-installer.sh "$@"
WRAPPER
    fi
    $SUDO chmod +x "$INSTALLER_LINK"
    success "$L_INST_DONE: ${_BOLD}$INSTALLER_LINK${_RESET}"
}

check_path() {
    local target_dir
    target_dir="$(dirname "$1")"
    if [[ ":$PATH:" != *":$target_dir:"* ]]; then
        echo -e "\n"
        warn "$(printf "$L_PATH_WARN" "$target_dir")"
        info "$(printf "$L_PATH_FIX" "$target_dir")"
    fi
}

switch_mirror() {
    if [ "$USE_CNB" = "true" ]; then
        USE_CNB=false
        save_mirror
        success "$L_SWITCH_TO_GITHUB"
    else
        USE_CNB=true
        save_mirror
        success "$L_SWITCH_TO_CNB"
    fi
    # Re-install fns wrapper to reflect new mirror setting
    # 重新安装 fns 包装脚本以反映新的镜像设置
    install_self >/dev/null 2>&1 || true
}

show_menu() {
    draw_banner
    echo -e "  [1] ${_BOLD}$L_MENU_1${_RESET}"
    echo -e "      ${_CYAN}${_ITALIC}${_DIM}$L_MENU_1_D${_RESET}"
    echo -e "  [2] ${_BOLD}$L_MENU_2${_RESET}"
    echo -e "      ${_CYAN}${_ITALIC}${_DIM}$L_MENU_2_D${_RESET}"
    echo -e "  [3] ${_BOLD}$L_MENU_3${_RESET}"
    echo -e "      ${_CYAN}${_ITALIC}${_DIM}$L_MENU_3_D${_RESET}"
    echo -e "  [4] ${_BOLD}$L_MENU_4${_RESET}"
    echo -e "      ${_CYAN}${_ITALIC}${_DIM}$L_MENU_4_D${_RESET}"
    echo -e "  [5] ${_BOLD}$L_MENU_5${_RESET}"
    echo -e "      ${_CYAN}${_ITALIC}${_DIM}$L_MENU_5_D${_RESET}"
    echo -e "  [6] ${_BOLD}$L_MENU_6${_RESET}"
    echo -e "      ${_CYAN}${_ITALIC}${_DIM}$L_MENU_6_D${_RESET}"
    echo -e "  [7] ${_BOLD}$L_MENU_7${_RESET}"
    echo -e "      ${_CYAN}${_ITALIC}${_DIM}$L_MENU_7_D${_RESET}"
    echo -e "  [8] ${_BOLD}$L_MENU_8${_RESET}"
    echo -e "      ${_CYAN}${_ITALIC}${_DIM}$L_MENU_8_D${_RESET}"
    echo -e "  [L] ${_BOLD}$L_MENU_L${_RESET}"
    echo -e "  [0] ${_BOLD}$L_MENU_0${_RESET}"
    echo -e "\n${_BLUE} ================================================ ${_RESET}"
    
    while true; do
        read -rp "  $(echo -e "${_BOLD}$L_SELECT [0-8, L]: ${_RESET}")" opt
        case "$opt" in
            1) read -rp "  $(echo -e "${_BOLD}$L_INPUT_VER: ${_RESET}")" v; v="${v:-latest}"; install_cmd "$v";;
            2) start_service;;
            3) stop_service;;
            4) status_service;;
            5) full_uninstall;;
            6) install_self "";;
            7) enable_autostart;;
            8) switch_mirror; load_lang; show_menu; return;;
            L|l)
                if [ "$CURRENT_LANG" = "en" ]; then CURRENT_LANG="zh"; else CURRENT_LANG="en"; fi
                save_lang
                load_lang
                draw_banner
                show_menu
                return
            ;;
            0|q) exit 0;;
            *) warn "$L_INVALID";;
        esac
    done
}

install_cmd() {
    local ver="${1:-latest}"
    local os arch tarball
    os="$(detect_os)"
    arch="$(_arch_map)"
    
    stop_service
    
    step "$L_PRE_DL ${_BOLD}$ver${_RESET} ($os/$arch)..."
    if [ "$ver" = "latest" ]; then
        ver="$(get_latest_tag || echo latest)"
    fi
    tarball="$(download_release_asset "$ver" "$os" "$arch")" || { error "$L_ERR_NO_REL"; return 1; }
    install_binary_from_tar "$tarball"
    save_version "$ver"
    install_self >&2
    
    # Enable auto-start by default on install/update
    enable_autostart
    
    success "$L_INST_ALL_DONE"
    info "$L_INST_TIP"
    check_path "$LINK_BIN"
    check_path "$INSTALLER_LINK"
    
    start_service
}

# main dispatcher // 主调度器
# Filter out mirror flags before dispatching // 过滤镜像标志后再调度
_dispatch_args=()
for _arg in "$@"; do
    case "$_arg" in
        --cnb|--github) ;;  # already handled by parse_mirror_from_args
        *) _dispatch_args+=("$_arg") ;;
    esac
done
cmd="${_dispatch_args[0]:-menu}"
case "$cmd" in
    install)
        ensure_root
        install_cmd "${_dispatch_args[1]:-latest}"
    ;;
    uninstall|full-uninstall|full_uninstall)
        full_uninstall
    ;;
    start)
        start_service
    ;;
    stop)
        stop_service
    ;;
    status)
        status_service
    ;;
    update)
        ensure_root
        install_cmd "latest"
    ;;
    install-self)
        install_self "${_dispatch_args[1]:-}"
    ;;
    enable-autostart)
        enable_autostart
    ;;
    menu)
        show_menu
    ;;
    *)
        draw_banner
        echo -e "$L_USAGE: $0 {install|uninstall|full-uninstall|start|stop|status|update|install-self|enable-autostart|menu}"
        exit 1
    ;;
esac