#!/bin/bash
# 安装Linux美化版命令行

# 安装依赖和前置工具
if ! command -v zsh &> /dev/null; then
    echo "正在安装 Zsh..."
    sudo apt-get install -y zsh || sudo yum install -y zsh || brew install zsh
fi

if ! command -v git &> /dev/null; then
    echo "正在安装 Git..."
    sudo apt-get install -y git || sudo yum install -y git || brew install git
fi

if ! command -v curl &> /dev/null; then
    echo "正在安装 curl..."
    sudo apt-get install -y curl || sudo yum install -y curl || brew install curl
fi

# 安装 Oh My Zsh（如未安装）
if [ ! -d "${HOME}/.oh-my-zsh" ]; then
    echo "正在安装 Oh My Zsh..."
    sh -c "$(curl -fsSL https://gitee.com/pocmon/ohmyzsh/raw/master/tools/install.sh)" "" --unattended
fi

# 修复路径变量（使用绝对路径）
ZSH_CUSTOM="${ZSH_CUSTOM:-${HOME}/.oh-my-zsh/custom}"

# 单独安装 autojump（目录跳转）二进制插件
if ! command -v autojump &> /dev/null; then
    echo "正在安装 autojump..."
    sudo apt-get install -y autojump || sudo yum install -y autojump || brew install autojump
fi

# 安装常用插件
plugins=(
    mirrors/zsh-syntax-highlighting      # 语法高亮
    mirrors/zsh-autosuggestions          # 自动建议
    yuhldr/zsh-completions               # 自动补全
)

# 安装其他插件
for plugin in "${plugins[@]}"; do
    plugin_name="${plugin##*/}"  # 提取插件名，去掉可能的路径前缀
    plugin_dir="${ZSH_CUSTOM}/plugins/${plugin_name}"
    if [ ! -d "$plugin_dir" ]; then
        echo "正在安装插件: $plugin_name"
        # 将 GitHub 替换为 Gitee 镜像源
        git clone "https://gitee.com/${plugin}.git" "$plugin_dir" --depth 1
        echo "已安装${plugin_name}"
        plugins_list+=("$plugin_name") #储存插件列表
    fi
done

# 安装 Powerlevel10k 主题
if [ ! -d "${ZSH_CUSTOM}/themes/powerlevel10k" ]; then
    echo "正在安装主题: Powerlevel10k"
    # 将 GitHub 替换为 Gitee 镜像源
    git clone --depth=1 "https://gitee.com/romkatv/powerlevel10k.git" "${ZSH_CUSTOM}/themes/powerlevel10k"
fi

# 配置 .zshrc
echo "正在优化 .zshrc 配置..."
cat << EOF > ~/.zshrc #写入配置
export ZSH="\${HOME}/.oh-my-zsh"
ZSH_THEME="powerlevel10k/powerlevel10k"
plugins=($(IFS=' '; echo "${plugins_list[*]}"))

# 加载插件前初始化补全系统
fpath+=(\${ZSH_CUSTOM}/plugins/zsh-completions/src)
autoload -U compinit && compinit

# 加载核心框架
source \$ZSH/oh-my-zsh.sh

# 加载插件
source \${ZSH_CUSTOM}/plugins/zsh-syntax-highlighting/zsh-syntax-highlighting.zsh
source \${ZSH_CUSTOM}/plugins/zsh-autosuggestions/zsh-autosuggestions.zsh

# 加载 autojump
[ -f /usr/share/autojump/autojump.sh ] && source /usr/share/autojump/autojump.sh
[ -f /opt/homebrew/etc/profile.d/autojump.sh ] && source /opt/homebrew/etc/profile.d/autojump.sh

# 自动应用p10k配置
[[ ! -f ~/.p10k.zsh ]] || source ~/.p10k.zsh
EOF

#设置为开机启动
sudo chsh -s /bin/zsh
echo "已设置zsh为默认终端"

echo "-------------------"
echo "安装完成！请运行以下命令："
echo "zsh                     启动zsh(初次进入会设置主题)"
echo "source ~/.zshrc         导入配置"
echo "p10k configure          进入主题设置"