flashjdk 1.0.0

package/scripts/flashjdk.sh

@@ -0,0 +1,497 @@
+#!/usr/bin/env bash
+# ============================================================
+# JDK 路径一键切换脚本 (macOS / Linux)
+# 支持：扫描已安装JDK、手动输入路径、自动更新 JAVA_HOME 和 PATH
+# 支持：本地缓存自定义扫描根目录
+# ============================================================
+
+# 自动检测版本号（从 package.json 读取，兼容 npm 全局安装和开发环境）
+_detect_version() {
+    local script_dir
+    script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+    local pkg
+    for pkg in "$script_dir/../package.json" "$script_dir/package.json"; do
+        if [ -f "$pkg" ]; then
+            node -p "require('$pkg').version" 2>/dev/null && return
+            sed -n 's/.*"version"[[:space:]]*:[[:space:]]*"\([^"]*\)".*/\1/p' "$pkg" 2>/dev/null && return
+        fi
+    done
+    echo "unknown"
+}
+VERSION=$(_detect_version)
+
+# 缓存目录：统一存放于 ~/.config/flashjdk/
+CACHE_DIR="$HOME/.config/flashjdk"
+CACHE_FILE="$CACHE_DIR/custom-roots.txt"
+
+# ════ 颜色定义 ════════════════════════════════════════════════
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+CYAN='\033[0;36m'
+MAGENTA='\033[0;35m'
+GRAY='\033[0;90m'
+NC='\033[0m'
+
+log_info()    { echo -e "${CYAN}[$(date +%H:%M:%S)] [INFO]    $1${NC}"; }
+log_success() { echo -e "${GREEN}[$(date +%H:%M:%S)] [SUCCESS] $1${NC}"; }
+log_warn()    { echo -e "${YELLOW}[$(date +%H:%M:%S)] [WARN]    $1${NC}"; }
+log_error()   { echo -e "${RED}[$(date +%H:%M:%S)] [ERROR]   $1${NC}"; }
+log_title()   { echo -e "${MAGENTA}$1${NC}"; }
+separator()   { echo -e "${GRAY}============================================================${NC}"; }
+
+# ════ OS 检测 ════════════════════════════════════════════════
+detect_os() {
+    case "$(uname -s)" in
+        Darwin) echo "mac" ;;
+        Linux)  echo "linux" ;;
+        *)      echo "unknown" ;;
+    esac
+}
+
+OS=$(detect_os)
+
+# ════ 路径工具函数 ══════════════════════════════════════════════
+
+# 规范化路径：去除首尾空白、引号、尾随斜杠
+normalize_path() {
+    local p="$1"
+    # 去除首尾空白
+    p=$(echo "$p" | sed -e 's/^[[:space:]]*//' -e 's/[[:space:]]*$//')
+    # 去除首尾引号
+    p="${p#\"}"; p="${p%\"}"
+    p="${p#\'}"; p="${p%\'}"
+    # 去除尾随斜杠（保留根路径 /）
+    while [[ "$p" != "/" && ( "$p" == */ || "$p" == *\\ ) ]]; do
+        p="${p%/}"
+        p="${p%\\}"
+    done
+    echo "$p"
+}
+
+# 检查路径是否存在（双重检查：-d + ls）
+test_path_safe() {
+    local p="$1"
+    [ -z "$p" ] && return 1
+    [ -d "$p" ] && return 0
+    # Fallback: 尝试 ls（有时能处理 -d 无法处理的情况）
+    ls "$p" >/dev/null 2>&1 && return 0
+    return 1
+}
+
+# ════ 默认扫描根目录（按 OS 区分）════════════════════════════
+get_default_roots() {
+    if [ "$OS" = "mac" ]; then
+        echo "/Library/Java/JavaVirtualMachines"
+        echo "$HOME/Library/Java/JavaVirtualMachines"
+        echo "/usr/local/opt"
+        echo "/opt/homebrew/opt"
+        echo "$HOME/.sdkman/candidates/java"
+        echo "$HOME/.jdks"
+    else
+        echo "/usr/lib/jvm"
+        echo "/usr/local/java"
+        echo "/usr/local/jdk"
+        echo "/opt/java"
+        echo "/opt/jdk"
+        echo "$HOME/.sdkman/candidates/java"
+        echo "$HOME/.jdks"
+    fi
+}
+
+# ════ JDK 发现逻辑 ════════════════════════════════════════════
+
+# 在单个根目录下查找 JDK，输出每个合法的 JDK Home 路径（每行一个）
+_find_jdk_in_root() {
+    local root="$1"
+    [ -d "$root" ] || return
+
+    if [ "$OS" = "mac" ]; then
+        # macOS 标准 .jdk 包结构：xxx.jdk/Contents/Home
+        for bundle in "$root"/*.jdk; do
+            [ -d "$bundle" ] || continue
+            local home="$bundle/Contents/Home"
+            [ -x "$home/bin/java" ] && echo "$home"
+        done
+        # 扁平结构（SDKMAN、Homebrew opt、.jdks 等）：直接包含 bin/java
+        for dir in "$root"/*/; do
+            dir="${dir%/}"
+            [ -d "$dir" ] || continue
+            [[ "$dir" == *.jdk ]] && continue   # 已在上面处理
+            # Homebrew: opt/openjdk@17/libexec/openjdk.jdk/Contents/Home
+            if [ -d "$dir/libexec" ]; then
+                for inner in "$dir/libexec"/*.jdk; do
+                    local h="$inner/Contents/Home"
+                    [ -x "$h/bin/java" ] && echo "$h"
+                done
+            fi
+            [ -x "$dir/bin/java" ] && echo "$dir"
+        done
+    else
+        # Linux：子目录直接含 bin/java
+        for dir in "$root"/*/; do
+            dir="${dir%/}"
+            [ -d "$dir" ] || continue
+            [ -x "$dir/bin/java" ] && echo "$dir"
+        done
+        # 根目录本身也可能是 JDK（手动安装场景）
+        [ -x "$root/bin/java" ] && echo "$root"
+    fi
+}
+
+find_jdk_installations() {
+    local -a roots=("$@")
+    local tmp
+    tmp=$(mktemp)
+    for root in "${roots[@]}"; do
+        _find_jdk_in_root "$root" >> "$tmp"
+    done
+    # 去重排序后输出
+    sort -u "$tmp"
+    rm -f "$tmp"
+}
+
+# macOS 使用系统 java_home 工具补充扫描
+_macos_java_home_list() {
+    command -v /usr/libexec/java_home &>/dev/null || return
+    /usr/libexec/java_home -V 2>&1 | grep -E '^\s+[0-9]' | awk '{print $NF}'
+}
+
+# ════ 缓存读写 ════════════════════════════════════════════════
+read_cached_roots() {
+    [ -f "$CACHE_FILE" ] || return
+    while IFS= read -r line; do
+        [[ -n "$line" && "$line" != \#* ]] && normalize_path "$line"
+    done < "$CACHE_FILE"
+}
+
+save_cached_roots() {
+    mkdir -p "$CACHE_DIR"
+    # 规范化所有路径后再保存
+    local normalized=()
+    local p
+    for p in "$@"; do
+        p=$(normalize_path "$p")
+        [ -n "$p" ] && normalized+=("$p")
+    done
+    # 去重
+    printf '%s\n' "${normalized[@]}" | awk '!seen[$0]++' > "$CACHE_FILE"
+    log_success "已保存到：$CACHE_FILE"
+}
+
+get_all_search_roots() {
+    {
+        get_default_roots
+        read_cached_roots
+    } | awk '!seen[$0]++'
+}
+
+# ════ PATH / JAVA_HOME 更新逻辑 ══════════════════════════════
+
+# 写入系统级配置（需要 sudo）
+_write_system_profile() {
+    local content="$1"
+    local target="/etc/profile.d/flashjdk.sh"
+
+    if sudo -n true 2>/dev/null || sudo true 2>/dev/null; then
+        echo -e "$content" | sudo tee "$target" > /dev/null
+        sudo chmod 644 "$target"
+        log_success "系统配置已写入：$target"
+    else
+        log_warn "sudo 授权失败，跳过系统级配置，仅更新用户配置文件。"
+    fi
+}
+
+# 更新用户 Shell 配置文件，替换 flashjdk 托管块
+_update_user_profile() {
+    local content="$1"
+    local pfile="$2"
+    local begin_mark="# flashjdk: managed block - do not edit"
+    local end_mark="# flashjdk: end"
+
+    # 若文件不存在则创建
+    [ -f "$pfile" ] || touch "$pfile"
+
+    # 删除旧托管块
+    local tmp
+    tmp=$(mktemp)
+    awk "
+        /$begin_mark/ { skip=1 }
+        !skip { print }
+        /$end_mark/ { skip=0 }
+    " "$pfile" > "$tmp" && mv "$tmp" "$pfile"
+
+    # 追加新块
+    {
+        echo ""
+        echo -e "$content"
+    } >> "$pfile"
+
+    log_success "用户配置已更新：$pfile"
+}
+
+update_system_jdk_path() {
+    local new_jdk_root="$1"
+    local new_bin="$new_jdk_root/bin"
+
+    if [ ! -d "$new_bin" ]; then
+        log_error "bin 目录不存在：$new_bin"
+        return 1
+    fi
+
+    local block
+    block="# flashjdk: managed block - do not edit
+export JAVA_HOME=\"$new_jdk_root\"
+export PATH=\"\$JAVA_HOME/bin:\$PATH\"
+# flashjdk: end"
+
+    # 系统级（/etc/profile.d/，对所有用户生效）
+    _write_system_profile "$block"
+
+    # 用户级：根据当前 Shell 选择合适的配置文件
+    local shell_name
+    shell_name=$(basename "${SHELL:-bash}")
+    case "$shell_name" in
+        zsh)
+            _update_user_profile "$block" "$HOME/.zshrc"
+            ;;
+        bash)
+            _update_user_profile "$block" "$HOME/.bashrc"
+            # macOS bash 还需更新 .bash_profile
+            [ "$OS" = "mac" ] && _update_user_profile "$block" "$HOME/.bash_profile"
+            ;;
+        *)
+            _update_user_profile "$block" "$HOME/.profile"
+            ;;
+    esac
+
+    # 立即在当前会话生效
+    export JAVA_HOME="$new_jdk_root"
+    export PATH="$new_bin:$PATH"
+    log_success "当前会话 JAVA_HOME 和 PATH 已立即生效。"
+
+    echo "$new_bin"
+}
+
+# ════ CLI 命令：-list（列出所有 JDK 环境）═════════════════════
+cmd_list() {
+    local -a all_roots=()
+    while IFS= read -r line; do
+        [ -n "$line" ] && all_roots+=("$line")
+    done < <(get_all_search_roots)
+
+    local -a jdk_list=()
+    while IFS= read -r line; do
+        [ -n "$line" ] && jdk_list+=("$line")
+    done < <(find_jdk_installations "${all_roots[@]}")
+
+    if [ "$OS" = "mac" ]; then
+        while IFS= read -r line; do
+            [ -n "$line" ] || continue
+            local dup=0
+            for existing in "${jdk_list[@]}"; do
+                [ "$existing" = "$line" ] && dup=1 && break
+            done
+            [ $dup -eq 0 ] && jdk_list+=("$line")
+        done < <(_macos_java_home_list)
+    fi
+
+    echo ""
+    log_info "当前 JAVA_HOME：${JAVA_HOME:-(未设置)}"
+    echo ""
+
+    if [ ${#jdk_list[@]} -gt 0 ]; then
+        log_success "扫描到 ${#jdk_list[@]} 个 JDK 环境："
+        echo ""
+        for i in "${!jdk_list[@]}"; do
+            local marker=" "
+            [ "${jdk_list[$i]}" = "$JAVA_HOME" ] && marker="*"
+            echo -e "  ${marker} [${CYAN}$((i+1))${NC}] ${jdk_list[$i]}"
+        done
+        echo ""
+        echo -e "  ${GRAY}(* 表示当前使用的 JDK)${NC}"
+    else
+        log_warn "未扫描到 JDK 环境。"
+        echo ""
+        echo -e "  请使用 ${GREEN}flashjdk -set <路径>${NC} 添加扫描根目录。"
+    fi
+    echo ""
+}
+
+# ════ CLI 命令：-set <路径>（添加扫描根目录）═══════════════════
+cmd_set() {
+    local new_path="$1"
+
+    if [ -z "$new_path" ]; then
+        log_error "缺少路径参数。用法：flashjdk -set <路径>"
+        echo ""
+        exit 1
+    fi
+
+    new_path=$(normalize_path "$new_path")
+
+    if ! test_path_safe "$new_path"; then
+        log_error "路径不存在：$new_path"
+        exit 1
+    fi
+
+    local -a cached=()
+    while IFS= read -r line; do
+        [ -n "$line" ] && cached+=("$line")
+    done < <(read_cached_roots)
+
+    if printf '%s\n' "${cached[@]}" | grep -qx "$new_path"; then
+        log_warn "该路径已在缓存中，无需重复添加。"
+        exit 0
+    fi
+
+    cached+=("$new_path")
+    save_cached_roots "${cached[@]}"
+    log_success "已添加扫描根目录：$new_path"
+}
+
+# ════ CLI 命令：-change（选择并切换 JDK）════════════════════════
+cmd_change() {
+    local -a all_roots=()
+    while IFS= read -r line; do
+        [ -n "$line" ] && all_roots+=("$line")
+    done < <(get_all_search_roots)
+
+    echo ""
+    log_info "正在扫描 JDK 环境..."
+
+    local -a jdk_list=()
+    while IFS= read -r line; do
+        [ -n "$line" ] && jdk_list+=("$line")
+    done < <(find_jdk_installations "${all_roots[@]}")
+
+    if [ "$OS" = "mac" ]; then
+        while IFS= read -r line; do
+            [ -n "$line" ] || continue
+            local dup=0
+            for existing in "${jdk_list[@]}"; do
+                [ "$existing" = "$line" ] && dup=1 && break
+            done
+            [ $dup -eq 0 ] && jdk_list+=("$line")
+        done < <(_macos_java_home_list)
+    fi
+
+    if [ ${#jdk_list[@]} -eq 0 ]; then
+        log_warn "未扫描到 JDK 环境。"
+        echo ""
+        echo -e "  请使用 ${GREEN}flashjdk -set <路径>${NC} 添加扫描根目录后重试。"
+        exit 1
+    fi
+
+    echo ""
+    log_info "当前 JAVA_HOME：${JAVA_HOME:-(未设置)}"
+    echo ""
+
+    log_success "扫描到以下 JDK 版本："
+    echo ""
+    for i in "${!jdk_list[@]}"; do
+        local marker=" "
+        [ "${jdk_list[$i]}" = "$JAVA_HOME" ] && marker="*"
+        echo -e "  ${marker} [${CYAN}$((i+1))${NC}] ${jdk_list[$i]}"
+    done
+    echo ""
+    echo -e "  ${GRAY}(* 表示当前使用的 JDK)${NC}"
+    echo ""
+
+    echo -e "${YELLOW}请输入要切换的 JDK 序号（输入 Q 取消）：${NC}"
+
+    while true; do
+        read -rp ">>> " user_input
+        user_input="$(echo "$user_input" | sed -e 's/^[[:space:]]*//' -e 's/[[:space:]]*$//')"
+
+        if [ -z "$user_input" ]; then
+            log_error "输入为空，请重新输入。"
+            continue
+        fi
+        if [ "$(echo "$user_input" | tr '[:lower:]' '[:upper:]')" = "Q" ]; then
+            log_info "已取消。"
+            exit 0
+        fi
+
+        if [[ "$user_input" =~ ^[0-9]+$ ]]; then
+            local idx=$((user_input - 1))
+            if [ "$idx" -ge 0 ] && [ "$idx" -lt ${#jdk_list[@]} ]; then
+                local selected="${jdk_list[$idx]}"
+
+                if [ ! -d "$selected" ]; then
+                    log_error "路径不存在：$selected"
+                    exit 1
+                fi
+                if [ ! -x "$selected/bin/java" ]; then
+                    log_error "bin/java 不存在，请确认选择的是 JDK 根目录。"
+                    exit 1
+                fi
+
+                echo ""
+                separator
+                log_info "正在切换 JDK 到：$selected"
+                update_system_jdk_path "$selected"
+
+                echo ""
+                separator
+                log_info "正在验证 java -version..."
+                echo ""
+                "$selected/bin/java" -version 2>&1
+                echo ""
+                log_success "JDK 切换完成！"
+
+                case "$(basename "${SHELL:-bash}")" in
+                    zsh)  log_info "如需手动刷新：source ~/.zshrc" ;;
+                    bash) log_info "如需手动刷新：source ~/.bashrc" ;;
+                    *)    log_info "如需手动刷新：source ~/.profile" ;;
+                esac
+                exit 0
+            else
+                log_error "序号无效，请重新输入。"
+            fi
+        else
+            log_error "请输入数字序号。"
+        fi
+    done
+}
+
+# ════ 帮助信息 ══════════════════════════════════════════════════
+show_help() {
+    echo ""
+    echo -e "  ${MAGENTA}flashJDK${NC} v${VERSION} — JDK 版本一键切换工具  [${OS}]"
+    echo ""
+    echo -e "  ${GREEN}用法:${NC}"
+    echo ""
+    echo -e "    flashjdk -v                查看版本号"
+    echo -e "    flashjdk -list             列出所有扫描到的 JDK 环境"
+    echo -e "    flashjdk -set <路径>       添加自定义扫描根目录"
+    echo -e "    flashjdk -change           列出所有 JDK 版本并选择切换"
+    echo ""
+    echo -e "  ${GRAY}缓存文件：$CACHE_FILE${NC}"
+    echo ""
+}
+
+# ════ CLI 参数分发 ══════════════════════════════════════════════
+case "${1:-}" in
+    -v|--version)
+        echo "v${VERSION}"
+        ;;
+    -list)
+        cmd_list
+        ;;
+    -set)
+        cmd_set "${2:-}"
+        ;;
+    -change)
+        cmd_change
+        ;;
+    "")
+        show_help
+        ;;
+    *)
+        echo "未知参数: $1"
+        echo ""
+        show_help
+        exit 1
+        ;;
+esac