#!/usr/bin/env bash

# Copyright (c) 2021-2026 tteck
# Author: tteck (tteckster)
# License: MIT | https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE
# Source: https://www.docker.com/

source /dev/stdin <<<"$FUNCTIONS_FILE_PATH"
color
verb_ip6
catch_errors
setting_up_container
network_check
update_os

setup_deb_based() {
  PORTAINER_AGENT_LATEST_VERSION=$(get_latest_github_release "portainer/agent")

  setup_docker

  if prompt_confirm "${TAB3}Would you like to install Portainer (UI) via the community-scripts addon?" "n" 60; then
    bash -c "$(curl -fsSL https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/tools/addon/portainer.sh)" <<<"y"
  else
    read -r -p "${TAB3}Would you like to install the Portainer Agent (for remote management)? <y/N> " prompt_agent
    if [[ ${prompt_agent,,} =~ ^(y|yes)$ ]]; then
      msg_info "Installing Portainer Agent $PORTAINER_AGENT_LATEST_VERSION"
      $STD docker run -d \
        -p 9001:9001 \
        --name portainer_agent \
        --restart=always \
        -v /var/run/docker.sock:/var/run/docker.sock \
        -v /var/lib/docker/volumes:/var/lib/docker/volumes \
        portainer/agent
      msg_ok "Installed Portainer Agent $PORTAINER_AGENT_LATEST_VERSION"
    fi
  fi

  read -r -p "${TAB3}Expose Docker TCP socket (insecure) ? [n = No, l = Local only (127.0.0.1), a = All interfaces (0.0.0.0)] <n/l/a>: " socket_choice
  case "${socket_choice,,}" in
  l)
    socket="tcp://127.0.0.1:2375"
    ;;
  a)
    socket="tcp://0.0.0.0:2375"
    ;;
  *)
    socket=""
    ;;
  esac

  if [[ -n "$socket" ]]; then
    msg_info "Enabling Docker TCP socket on $socket"
    $STD apt-get install -y jq

    tmpfile=$(mktemp)
    jq --arg sock "$socket" '. + { "hosts": ["unix:///var/run/docker.sock", $sock] }' /etc/docker/daemon.json >"$tmpfile" && mv "$tmpfile" /etc/docker/daemon.json

    mkdir -p /etc/systemd/system/docker.service.d
    cat <<EOF >/etc/systemd/system/docker.service.d/override.conf
[Service]
ExecStart=
ExecStart=$(command -v dockerd || echo /usr/sbin/dockerd)
EOF

    $STD systemctl daemon-reexec
    $STD systemctl daemon-reload

    if systemctl restart docker; then
      msg_ok "Docker TCP socket available on $socket"
    else
      msg_error "Docker failed to restart. Check journalctl -xeu docker.service"
      exit 150
    fi
  fi
}

setup_alpine() {
  msg_info "Installing Dependencies"
  $STD apk add tzdata openssl
  msg_ok "Installed Dependencies"

  msg_info "Installing Docker"
  $STD apk add docker
  $STD rc-service docker start
  $STD rc-update add docker default
  msg_ok "Installed Docker"

  get_latest_release() {
    curl -fsSL https://api.github.com/repos/"$1"/releases/latest | grep '"tag_name":' | cut -d'"' -f4
  }
  DOCKER_COMPOSE_LATEST_VERSION=$(get_latest_release "docker/compose")
  PORTAINER_AGENT_LATEST_VERSION=$(get_latest_release "portainer/agent")

  read -r -p "${TAB3}Would you like to add Docker Compose? <y/N> " prompt
  if [[ "${prompt,,}" =~ ^(y|yes)$ ]]; then
    msg_info "Installing Docker Compose $DOCKER_COMPOSE_LATEST_VERSION"
    DOCKER_CONFIG=${DOCKER_CONFIG:-$HOME/.docker}
    mkdir -p "$DOCKER_CONFIG"/cli-plugins
    curl -fsSL https://github.com/docker/compose/releases/download/"$DOCKER_COMPOSE_LATEST_VERSION"/docker-compose-linux-$(arch_resolve "x86_64" "aarch64") -o ~/.docker/cli-plugins/docker-compose
    chmod +x "$DOCKER_CONFIG"/cli-plugins/docker-compose
    msg_ok "Installed Docker Compose $DOCKER_COMPOSE_LATEST_VERSION"
  fi

  if prompt_confirm "${TAB3}Would you like to install Portainer (UI) via the community-scripts addon?" "n" 60; then
    bash -c "$(curl -fsSL https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main/tools/addon/portainer.sh)" <<<"y"
  else
    read -r -p "${TAB3}Would you like to add the Portainer Agent? <y/N> " prompt
    if [[ "${prompt,,}" =~ ^(y|yes)$ ]]; then
      msg_info "Installing Portainer agent $PORTAINER_AGENT_LATEST_VERSION"
      $STD docker run -d \
        -p 9001:9001 \
        --name portainer_agent \
        --restart=always \
        -v /var/run/docker.sock:/var/run/docker.sock \
        -v /var/lib/docker/volumes:/var/lib/docker/volumes \
        portainer/agent
      msg_ok "Installed Portainer Agent $PORTAINER_AGENT_LATEST_VERSION"
    fi
  fi

  read -r -p "${TAB3}Would you like to expose the Docker TCP socket? <y/N> " prompt
  if [[ "${prompt,,}" =~ ^(y|yes)$ ]]; then
    msg_info "Exposing Docker TCP socket"
    $STD mkdir -p /etc/docker
    $STD echo '{ "hosts": ["unix:///var/run/docker.sock", "tcp://0.0.0.0:2375"] }' >/etc/docker/daemon.json
    $STD rc-service docker restart
    msg_ok "Exposed Docker TCP socket at tcp://+:2375"
  fi
}

run_os_setup

motd_ssh
customize
cleanup_lxc
