#!/bin/bash


install_homebrew() {
    # Check if Homebrew is already installed
    if command -v brew &> /dev/null; then
        echo "Homebrew is already installed."
    else
        # Check if the system is macOS (Darwin)
        if [ "$(uname -s)" == "Darwin" ]; then
            echo "Installing Homebrew..."
            /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
        else
            echo "This script is intended for macOS only. Continue..."
        fi
    fi
}

install_zsh() {
    # Check if zsh is already installed
    if command -v zsh &> /dev/null; then
        echo "zsh is already installed."
    else
        # Check the operating system
        if [ "$(uname)" == "Darwin" ]; then
            # macOS specific installation
            brew install zsh
        elif [ "$(expr substr $(uname -s) 1 5)" == "Linux" ]; then
            # Linux specific installation
            sudo apt-get install zsh  # 适用于 Debian/Ubuntu 等
            # 或者使用其他 Linux 发行版的安装命令
        else
            echo "Unsupported operating system. Cannot install zsh."
            exit 1
        fi
    fi

    # Set zsh as the default shell
    if [ "$(uname)" == "Darwin" ]; then
        chsh -s $(which zsh)
    elif [ "$(expr substr $(uname -s) 1 5)" == "Linux" ]; then
        chsh -s $(which zsh)
    else
        echo "Unsupported operating system. Cannot set zsh as default shell."
        exit 1
    fi

    echo "Zsh installation completed and set as the default shell. Please restart your terminal."
}

reinstall_oh_my_zsh() {
    # Remove existing ~/.oh-my-zsh folder if it exists
    if [ -d ~/.oh-my-zsh ]; then
        echo "Removing existing Oh My Zsh folder..."
        rm -rf ~/.oh-my-zsh
    fi

    # Install Oh My Zsh
    git clone https://git.mingzailao.live/mingzailao/ohmyzsh.git ~/.oh-my-zsh
    cp ~/.oh-my-zsh/templates/zshrc.zsh-template ~/.zshrc

    echo "Oh My Zsh reinstallation completed."
}

# install home-brew
install_homebrew


# install zsh
install_zsh

# install oh-my-zsh
reinstall_oh_my_zsh

# install powerlevel10k
#git clone https://git.mingzailao.live/mingzailao/powerlevel10k.git ~/powerlevel10k
#cp ./.p10k.zsh ~/
#echo 'source ~/powerlevel10k/powerlevel10k.zsh-theme' >>~/.zshrc