#!/usr/bin/env bash
#
# Dotfiles manager and install helper
# (c) Ismo Vuorinen <https://github.com/ivuorinen> 2022
# Licensed under MIT, see LICENSE
#
# shellcheck source-path=$HOME/.dotfiles/local/bin
#
# Helper variables, override with ENVs like `VERBOSE=1 dfm help`
: "${VERBOSE:=0}"
: "${DOTFILES:=$HOME/.dotfiles}"
: "${SHARED_SCRIPT:=$DOTFILES/scripts/shared.sh}"
: "${INSTALL_SCRIPT:=$DOTFILES/scripts/install-dotfiles.sh}"
: "${BREWFILE:=$DOTFILES/Brewfile}"
: "${HOSTFILES:=$DOTFILES/hosts}"

SCRIPT=$(basename "$0")

VERSION_NVM="v0.39.3"

# shellcheck source=./../../scripts/shared.sh
source "$SHARED_SCRIPT"

function section_install
{
  USAGE_PREFIX="$SCRIPT install <command>"

  MENU=(
    "all:Installs everything in the correct order"
    "antigen:Updates the antigen.zsh file"
    "composer:Install composer"
    "dotenv:Install dotenv-linter"
    "gh:Install GitHub CLI Extensions"
    "go:Install Go Packages"
    "imagick:Install ImageMagick CLI"
    "starship:Install starship.rs"
    "macos:Setup nice macOS defaults"
    "neofetch:Install neofetch"
    "nvm:Install Node Version Manager (nvm)"
    "npm:Install NPM Packages"
    "ntfy:Install ntfy"
    "z:Install z"
  )

  case "$1" in
    all)
      $0 install macos
      $0 install antigen
      $0 install starship
      $0 brew install
      $0 install composer
      $0 install dotenv
      $0 install fzf
      $0 install gh
      $0 install go
      $0 install imagick
      $0 install neofetch
      $0 install nvm
      $0 install npm
      $0 install ntfy
      $0 install z
      ;;
    antigen)
      curl -sSfL git.io/antigen -o "$DOTFILES/local/bin/antigen.zsh" \
        && msg_yay "New antigen installed!"
      ;;
    composer)
      bash "$DOTFILES/scripts/install-composer.sh" \
        && msg_yay "composer installed!"
      ;;
    dotenv)
      curl -sSfL \
        https://raw.githubusercontent.com/dotenv-linter/dotenv-linter/master/install.sh \
        | sh -s -- -b "$XDG_BIN_HOME" \
        && msg_yay "dotenv-linter installed!"
      ;;
    fzf)
      bash "$DOTFILES/scripts/install-fzf.sh" \
        && msg_yay "fzf installed!"
      ;;
    gh)
      bash "$DOTFILES/scripts/install-gh-extensions.sh" \
        && msg_yay "github cli extensions installed!"
      ;;
    go)
      bash "$DOTFILES/scripts/install-go-packages.sh" \
        && msg_yay "go packages installed!"
      ;;
    imagick)
      curl -L https://imagemagick.org/archive/binaries/magick > "$XDG_BIN_HOME/magick" \
        && chmod +x "$XDG_BIN_HOME/magick" \
        && msg_yay "imagick installed!"
      ;;
    starship)
      curl -sS https://starship.rs/install.sh | sh -s -- \
        --bin-dir ~/.local/bin \
        --yes \
        && msg_yay "starship installed!"
      ;;
    macos)
      bash "$DOTFILES/scripts/set-macos-defaults.sh" \
        && msg_yay "Brewfile defined apps has been installed!"
      ;;
    neofetch)
      bash "$DOTFILES/scripts/install-neofetch.sh" \
        && msg_yay "neofetch installed!"
      ;;
    nvm)
      curl -o- "https://raw.githubusercontent.com/nvm-sh/nvm/$VERSION_NVM/install.sh" | bash \
        && nvm install --lts --latest-npm --default
      git checkout "$DOTFILES/base/zshrc"
      msg_yay "nvm installed!"
      ;;
    npm)
      bash "$DOTFILES/scripts/install-npm-packages.sh" \
        && msg_yay "NPM Packages have been installed!"
      ;;
    ntfy)
      bash "$DOTFILES/scripts/install-ntfy.sh" \
        && msg_yay "ntfy installed!"
      ;;
    z)
      bash "$DOTFILES/scripts/install-z.sh" \
        && msg_yay "z has been installed!"
      ;;
    *)
      menu_usage "$USAGE_PREFIX" "${MENU[@]}"
      ;;
  esac
}

function section_brew
{
  USAGE_PREFIX="$SCRIPT brew <command>"

  MENU=(
    "install:Installs items defined in Brewfile"
    "update:Updates and upgrades brew packages"
    "updatebundle:Updates Brewfile with descriptions"
    "autoupdate:Setups brew auto-update and runs it immediately"
  )

  have brew && {
    case "$1" in
      install)
        brew bundle install --file="$BREWFILE" && msg_yay "Done!"
        ;;
      update)
        brew update && brew outdated && brew upgrade && brew cleanup
        msg_yay "Done!"
        ;;
      updatebundle)
        # Updates .dotfiles/Brewfile with descriptions
        brew bundle dump \
          --force \
          --file="$BREWFILE" \
          --describe && msg_yay "Done!"
        ;;
      autoupdate)
        brew autoupdate delete
        brew autoupdate start 43200 --upgrade --cleanup --immediate
        ;;
      *)
        menu_usage "$USAGE_PREFIX" "${MENU[@]}"
        ;;
    esac
  }

  ! have brew && menu_section "$USAGE_PREFIX" "brew not available on this system"
}

function section_dotfiles
{
  USAGE_PREFIX="$SCRIPT dotfiles <command>"

  MENU=(
    "fmt:Run all formatters"
    "yamlfmt:Run yamlfmt to all dotfiles, which are in our control"
    "shfmt:Run shfmt to all dotfiles"
    "reset_all:Reset everything, runs all configured reset actions"
    "reset_nvim:Resets nvim. Deletes caches, removes nvim folders and relinks nvim folders"
  )

  case "$1" in
    fmt)
      msg_ok "Running all formatters"
      $0 dotfiles yamlfmt
      $0 dotfiles shfmt
      msg_done "...done!"
      ;;
    reset_all)
      msg_ok "Running all reset commands"
      $0 dotfiles reset_nvim
      ;;
    reset_nvim)
      rm -rf \
        ~/.local/share/nvim \
        ~/.local/state/nvim \
        ~/.cache/nvim \
        ~/.config/astronvim \
        ~/.config/nvim
      msg_ok "Deleted old nvim files"
      ln -s "$DOTFILES/config/astronvim" ~/.config/astronvim
      ln -s "$DOTFILES/config/nvim" ~/.config/nvim
      msg_ok "Linked nvim and astronvim"
      have npm && $0 install npm
      msg_ok "Installed packages"
      msg_done "nvim reset!"
      ;;
    yamlfmt)
      # format yaml files
      have yamlfmt && yamlfmt -conf "$DOTFILES/.yamlfmt"
      ! have yamlfmt && msg_err "yamlfmt not found"
      ;;
    shfmt)
      # Format shell scripts according to following rules.
      find "$DOTFILES" \
        -type f -perm +111 \
        -not -path '*/.git/*' \
        -not -path '*dotbot*' \
        -not -path '*tmux*' \
        -not -name '*.pl' \
        -not -name '*.py' \
        -not -name '*.php' \
        -not -name '*.zsh' \
        -exec shfmt \
        --language-dialect bash \
        --func-next-line --list --write \
        --indent 2 --case-indent --space-redirects \
        --binary-next-line {} \;
      msg_yay "dotfiles have been shfmt formatted!"
      ;;
    *)
      menu_usage "$USAGE_PREFIX" "${MENU[@]}"
      ;;
  esac
}

function section_check
{
  USAGE_PREFIX="$SCRIPT check <command>"
  X_HOSTNAME=$(hostname)
  X_ARCH=$(uname)

  MENU=(
    "arch <arch>:Empty <arch> returns current. Exit code 0=match to current, 1=no match."
    "host <host>:Empty <host> returns current. Exit code 0=match to current, 1=no match."
  )

  case "$1" in
    a | arch)
      [[ "$2" = "" ]] && echo "$X_ARCH" && exit 0
      [[ $X_ARCH = "$2" ]] && exit 0 || exit 1
      ;;
    h | host | hostname)
      [[ "$2" = "" ]] && echo "$X_HOSTNAME" && exit 0
      [[ $X_HOSTNAME = "$2" ]] && exit 0 || exit 1
      ;;
    *)
      menu_usage "$USAGE_PREFIX" "${MENU[@]}"
      ;;
  esac
}

# Secret menu for visual tests
function section_tests
{
  USAGE_PREFIX="$SCRIPT tests <command>"

  MENU=(
    "msg:List all log functions from shared.sh"
  )

  case "$1" in
    msg)
      msg "msg"
      msg_done "msg_done"
      msg_done_suffix "msg_done_suffix"
      msg_err "msg_err"
      msg_nested "msg_nested"
      msg_nested_done "msg_nested_done"
      msg_ok "msg_ok"
      msg_prompt "msg_prompt"
      msg_prompt_done "msg_prompt_done"
      msg_run "msg_run" "second_param"
      msg_run_done "msg_run_done" "second_param"
      msg_warn "msg_warn"
      msg_yay "msg_yay"
      msg_yay_done "msg_yay_done"
      ;;
    *)
      menu_usage "$USAGE_PREFIX" "${MENU[@]}"
      ;;
  esac
}

function usage
{
  echo ""
  msg_prompt "Usage: $SCRIPT <section> <command>"
  echo $"  Empty <command> prints <section> help."
  echo ""
  section_install
  echo ""
  section_brew
  echo ""
  section_check
  echo ""
  section_dotfiles
}

# The main loop. first keyword after $0 triggers section, or help.
case "$1" in
  install) section_install "$2" ;;
  brew) section_brew "$2" ;;
  check) section_check "$2" ;;
  dotfiles) section_dotfiles "$2" ;;
  tests) section_tests "$2" ;;
  *) usage && exit 0 ;;
esac