#!/usr/bin/env bash
_CS_DEFAULT_URL="https://raw.githubusercontent.com/community-scripts/ProxmoxVE/main"
_cs_boot="${COMMUNITY_SCRIPTS_CORE_DIR:-$(dirname "${BASH_SOURCE[0]}")/../../core}/core/build.func"
source "$_cs_boot" 2>/dev/null || source <(curl -fsSL "${COMMUNITY_SCRIPTS_CORE_URL:-https://raw.githubusercontent.com/community-scripts/core/main}/core/build.func")
# Copyright (c) 2021-2026 community-scripts ORG
# Author: hoholms
# License: MIT | https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE
# Source: https://github.com/grafana/loki

APP="Loki"
var_tags="${var_tags:-monitoring;logs}"
var_cpu="${var_cpu:-1}"
var_arm64="${var_arm64:-yes}"
var_unprivileged="${var_unprivileged:-1}"
if [[ -z "${var_os:-}" ]] && command -v pveversion >/dev/null 2>&1; then
  var_os=$(msg_menu "Choose the container OS" \
    "debian" "Debian 13" \
    "alpine" "Alpine (smaller footprint)")
fi

if [[ "${var_os:-}" == "alpine" ]]; then
  var_ram="${var_ram:-256}"
  var_disk="${var_disk:-1}"
  var_version="${var_version:-3.24}"
else
  var_ram="${var_ram:-512}"
  var_disk="${var_disk:-2}"
  var_version="${var_version:-13}"
fi

header_info "$APP"
variables
color
catch_errors

update_deb_based() {
  if ! dpkg -s loki >/dev/null 2>&1; then
    msg_error "No ${APP} Installation Found!"
    exit 233
  fi

  CHOICE=$(msg_menu "Loki Update Options" \
    "1" "Update Loki" \
    "2" "Allow 0.0.0.0 for listening" \
    "3" "Allow only ${LOCAL_IP} for listening")

  case $CHOICE in
  1)
    msg_info "Stopping Loki"
    systemctl stop loki
    msg_ok "Stopped Loki"

    msg_info "Updating Loki"
    $STD apt update
    $STD apt install -y --only-upgrade loki
    msg_ok "Updated Loki"

    msg_info "Starting Loki"
    systemctl start loki
    msg_ok "Started Loki"
    msg_ok "Updated successfully!"
    exit
    ;;
  2)
    msg_info "Configuring Loki to listen on 0.0.0.0"
    sed -i 's/http_listen_address:.*/http_listen_address: 0.0.0.0/' /etc/loki/config.yml
    sed -i 's/http_listen_port:.*/http_listen_port: 3100/' /etc/loki/config.yml
    systemctl restart loki
    msg_ok "Configured Loki to listen on 0.0.0.0"
    exit
    ;;
  3)
    msg_info "Configuring Loki to listen on ${LOCAL_IP}"
    sed -i "s/http_listen_address:.*/http_listen_address: $LOCAL_IP/" /etc/loki/config.yml
    sed -i 's/http_listen_port:.*/http_listen_port: 3100/' /etc/loki/config.yml
    systemctl restart loki
    msg_ok "Configured Loki to listen on ${LOCAL_IP}"
    exit
    ;;
  esac
}

update_alpine() {
  LXCIP=$(ip a s dev eth0 | awk '/inet / {print $2}' | cut -d/ -f1)

  CHOICE=$(msg_menu "Loki Update Options" \
    "1" "Check for Loki Updates" \
    "2" "Allow 0.0.0.0 for listening" \
    "3" "Allow only ${LXCIP} for listening")

  case $CHOICE in
  1)
    $STD apk -U upgrade
    msg_ok "Updated successfully!"
    exit
    ;;
  2)
    sed -i -e "s/cfg:server.http_addr=.*/cfg:server.http_addr=0.0.0.0/g" /etc/conf.d/loki
    service loki restart
    msg_ok "Allowed listening on all interfaces!"
    exit
    ;;
  3)
    sed -i -e "s/cfg:server.http_addr=.*/cfg:server.http_addr=$LXCIP/g" /etc/conf.d/loki
    service loki restart
    msg_ok "Allowed listening only on ${LXCIP}!"
    exit
    ;;
  esac
}

function update_script() {
  header_info
  check_container_storage
  check_container_resources
  run_os_update
}

start
build_container
description

msg_ok "Completed successfully!\n"
echo -e "${CREATING}${GN}${APP} setup has been successfully initialized!${CL}"
echo -e "${INFO}${YW}Access loki using the following URL:${CL}"
echo -e "${GATEWAY}${BGN}http://${IP}:3100${CL}\n"