Remove Zinit folders

zinit_remove.zsh

# ZINIT_HOME="/home/username/.local/share/zinit/zinit.git"

zinit_remove() {
    local target_dir

    case "$1" in
        -s|--snippets) 
            target_dir="${ZINIT_HOME:s/zinit\.git/snippets}"
            ;;
        -p|--plugins) 
            target_dir="${ZINIT_HOME:s/zinit\.git/plugins}"
            ;;
        -a|--all)
            zinit_remove -s
            zinit_remove -p
            return
            ;;
        *)
            echo "Invalid option. Select from -s, -p, or -a only."
            return 1
    esac

    if [[ -d ${target_dir} ]]; then
        echo "Are you sure you want to delete ${target_dir}? [y/n]"
        read -k1 confirm
        if [[ ${confirm} != "y" ]]; then
            echo "Abort operation."
            return 1
        fi

        echo "Deleting ${target_dir} ..."
        rm -rf ${target_dir}
    else
        echo "${target_dir} does not exist."
    fi
}