#!/bin/sh

ACTION=${1}
shift 1

get_image() {
  IMAGE_NAME="lscr.io/linuxserver/heimdall:latest"
}

do_install() {
  get_image
  echo "docker pull ${IMAGE_NAME}"
  docker pull ${IMAGE_NAME}
  docker rm -f heimdall

  do_install_detail
}

do_install_detail() {
  local config=`uci get heimdall.@heimdall[0].config_path 2>/dev/null`
  local http_port=`uci get heimdall.@heimdall[0].http_port 2>/dev/null`
  local https_port=`uci get heimdall.@heimdall[0].https_port 2>/dev/null`

  if [ -z "$config" ]; then
      echo "config path is empty!"
      exit 1
  fi

  [ -z "$http_port" ] && http_port=8088
  [ -z "$https_port" ] && http_port=8089

  local cmd="docker run --restart=unless-stopped -d \
    -v \"$config:/config\" \
    --dns=172.17.0.1 \
    -p $http_port:80 \
    -p $https_port:443 "

  cmd="$cmd -v /mnt:/mnt"
  mountpoint -q /mnt && cmd="$cmd:rslave"

  local tz="`uci get system.@system[0].zonename`"
  [ -z "$tz" ] || cmd="$cmd -e TZ=$tz"

  cmd="$cmd --name heimdall \"$IMAGE_NAME\""

  echo "$cmd"
  eval "$cmd"

      #Add link to shortcutmenu
  # Get our local LAN IP Address
  LAN_IP=$(uci get network.lan.ipaddr)
  # Strip trailing network mask
  LAN_IP="${LAN_IP%/*}"

  # Add a new list option to the "shortcutmenu" configuration file
  uci add shortcutmenu lists
  uci set shortcutmenu.@lists[-1].webname="$IMAGE_NAME"
  uci set shortcutmenu.@lists[-1].weburl="$LAN_IP:$http_port"
  uci set shortcutmenu.@lists[-1].webpath="/"
  uci commit shortcutmenu

}

usage() {
  echo "usage: $0 sub-command"
  echo "where sub-command is one of:"
  echo "      install                Install the heimdall"
  echo "      upgrade                Upgrade the heimdall"
  echo "      rm/start/stop/restart  Remove/Start/Stop/Restart the heimdall"
  echo "      status                 Heimdall status"
  echo "      port                   Heimdall port"
}

case ${ACTION} in
  "install")
    do_install
  ;;
  "upgrade")
    do_install
  ;;
  "rm")
    docker rm -f heimdall
  ;;
  "start" | "stop" | "restart")
    docker ${ACTION} heimdall
  ;;
  "status")
    docker ps --all -f 'name=heimdall' --format '{{.State}}'
  ;;
  "port")
    docker ps --all -f 'name=heimdall' --format '{{.Ports}}' | grep -om1 '0.0.0.0:[0-9]*' | sed 's/0.0.0.0://'
  ;;
  *)
    usage
    exit 1
  ;;
esac