#!/usr/bin/env bash

# Copyright (c) 2021-2025 community-scripts ORG
# Author: MickLesk
# License: MIT | https://github.com/community-scripts/ProxmoxVE/raw/main/LICENSE

function header_info() {
  clear
  cat <<"EOF"
   ______                  ____             __
  / ____/___  ____  __  __/ __ \____ ______/ /___  __
 / /   / __ \/ __ \/ / / / /_/ / __ `/ ___/ __/ / / /
/ /___/ /_/ / /_/ / /_/ / ____/ /_/ / /  / /_/ /_/ /
\____/\____/ .___/\__, /_/    \__,_/_/   \__/\__, /
          /_/    /____/                     /____/
EOF
}

YW=$(echo "\033[33m")
GN=$(echo "\033[1;92m")
RD=$(echo "\033[01;31m")
BL=$(echo "\033[36m")
CL=$(echo "\033[m")
CM="${GN}✔️${CL}"
CROSS="${RD}✖️${CL}"
INFO="${BL}ℹ️${CL}"

APP="CopyParty"
BIN_PATH="/usr/local/bin/copyparty-sfx.py"
CONF_PATH="/etc/copyparty.conf"
LOG_PATH="/var/log/copyparty"
DATA_PATH="/var/lib/copyparty"
SERVICE_PATH_DEB="/etc/systemd/system/copyparty.service"
SERVICE_PATH_ALP="/etc/init.d/copyparty"
SVC_USER="copyparty"
SVC_GROUP="copyparty"
SRC_URL="https://github.com/9001/copyparty/releases/latest/download/copyparty-sfx.py"
DEFAULT_PORT=3923

# OS Detection
if [[ -f "/etc/alpine-release" ]]; then
  OS="Alpine"
  PKG_MANAGER="apk add --no-cache"
  SERVICE_PATH="$SERVICE_PATH_ALP"
elif [[ -f "/etc/debian_version" ]]; then
  OS="Debian"
  PKG_MANAGER="apt-get install -y"
  SERVICE_PATH="$SERVICE_PATH_DEB"
else
  echo -e "${CROSS} Unsupported OS detected. Exiting."
  exit 1
fi

header_info

function msg_info() { echo -e "${INFO} ${YW}$1...${CL}"; }
function msg_ok() { echo -e "${CM} ${GN}$1${CL}"; }
function msg_error() { echo -e "${CROSS} ${RD}$1${CL}"; }

# User/Group/Dirs
function setup_user_and_dirs() {
  msg_info "Creating $SVC_USER user and directories"
  if ! id "$SVC_USER" &>/dev/null; then
    if [[ "$OS" == "Debian" ]]; then
      useradd -r -s /sbin/nologin -d "$DATA_PATH" "$SVC_USER"
    else
      adduser -D -H -h "$DATA_PATH" -s /sbin/nologin "$SVC_USER"
    fi
  fi
  mkdir -p "$DATA_PATH" "$LOG_PATH"
  chown -R "$SVC_USER:$SVC_GROUP" "$DATA_PATH" "$LOG_PATH"
  chmod 755 "$DATA_PATH" "$LOG_PATH"
  msg_ok "User/Group/Dirs ready"
}

function uninstall_copyparty() {
  msg_info "Uninstalling $APP"
  if [[ "$OS" == "Debian" ]]; then
    systemctl disable --now copyparty &>/dev/null
    rm -f "$SERVICE_PATH_DEB"
  else
    rc-service copyparty stop &>/dev/null
    rc-update del copyparty &>/dev/null
    rm -f "$SERVICE_PATH_ALP"
  fi
  rm -f "$BIN_PATH" "$CONF_PATH"
  msg_ok "$APP has been uninstalled."
  exit 0
}

function update_copyparty() {
  msg_info "Updating $APP"
  curl -fsSL "$SRC_URL" -o "$BIN_PATH"
  chmod +x "$BIN_PATH"
  msg_ok "Updated $APP"
  exit 0
}

# --- Existing Install/Update/Uninstall Check ---
if [[ -f "$BIN_PATH" ]]; then
  echo -e "${YW}⚠️ $APP is already installed.${CL}"
  echo -n "Uninstall $APP? (y/N): "
  read -r uninstall_prompt
  if [[ "${uninstall_prompt,,}" =~ ^(y|yes)$ ]]; then
    uninstall_copyparty
  fi

  echo -n "Update $APP? (y/N): "
  read -r update_prompt
  if [[ "${update_prompt,,}" =~ ^(y|yes)$ ]]; then
    update_copyparty
  else
    echo -e "${YW}⚠️ Update skipped. Exiting.${CL}"
    exit 0
  fi
fi

# --- Deps ---
msg_info "Installing dependencies"
if [[ "$OS" == "Debian" ]]; then
  $PKG_MANAGER python3 curl &>/dev/null
else
  $PKG_MANAGER python3 curl &>/dev/null
fi
msg_ok "Dependencies installed"

# --- User/Dirs ---
setup_user_and_dirs

# --- Download Binary ---
msg_info "Downloading $APP"
curl -fsSL "$SRC_URL" -o "$BIN_PATH"
chmod +x "$BIN_PATH"
chown "$SVC_USER:$SVC_GROUP" "$BIN_PATH"
msg_ok "Downloaded to $BIN_PATH"

# --- Config: Interaktiv, Auth, Rootdir, Port ---
echo -n "Enter port for $APP (default: $DEFAULT_PORT): "
read -r PORT
PORT=${PORT:-$DEFAULT_PORT}

echo -n "Set data directory (default: $DATA_PATH): "
read -r USER_DATA_PATH
USER_DATA_PATH=${USER_DATA_PATH:-$DATA_PATH}
mkdir -p "$USER_DATA_PATH"
chown "$SVC_USER:$SVC_GROUP" "$USER_DATA_PATH"

echo -n "Enable authentication? (Y/n): "
read -r auth_enable
if [[ "${auth_enable,,}" =~ ^(n|no)$ ]]; then
  AUTH_LINE=""
  msg_ok "Configured without authentication"
else
  echo -n "Set admin username [default: admin]: "
  read -r ADMIN_USER
  ADMIN_USER=${ADMIN_USER:-admin}
  echo -n "Set admin password [default: helper-scripts.com]: "
  read -rs ADMIN_PASS
  ADMIN_PASS=${ADMIN_PASS:-helper-scripts.com}
  echo
  AUTH_LINE="auth vhost=/:$ADMIN_USER:$ADMIN_PASS:admin,,"
  msg_ok "Configured with admin user: $ADMIN_USER"
fi

# --- Generate /etc/copyparty.conf ---
msg_info "Writing config to $CONF_PATH"
cat <<EOF >"$CONF_PATH"
# CopyParty main config
# More options: https://github.com/9001/copyparty/blob/hovudstraum/contrib/systemd/copyparty.conf
#
# storage
root = $USER_DATA_PATH

# HTTP port
port = $PORT

# logs
logs = $LOG_PATH

# authentication
$AUTH_LINE

# allow upload
ul = *

# Example: allow delete
#dl = *

# basic web-ui
w = *

EOF
chmod 640 "$CONF_PATH"
chown "$SVC_USER:$SVC_GROUP" "$CONF_PATH"
msg_ok "Config written"

# --- Systemd/OpenRC Service ---
msg_info "Creating service"
if [[ "$OS" == "Debian" ]]; then
  cat <<EOF >"$SERVICE_PATH_DEB"
[Unit]
Description=CopyParty file server

[Service]
Type=simple
User=$SVC_USER
Group=$SVC_GROUP
WorkingDirectory=$USER_DATA_PATH
Environment=PYTHONUNBUFFERED=x
LogsDirectory=copyparty
ExecStart=/usr/bin/python3 $BIN_PATH -c $CONF_PATH
Restart=always

[Install]
WantedBy=multi-user.target
EOF
  systemctl daemon-reload
  systemctl enable --now copyparty &>/dev/null
else
  cat <<EOF >"$SERVICE_PATH_ALP"
#!/sbin/openrc-run

command="/usr/bin/python3"
command_args="$BIN_PATH -c $CONF_PATH"
command_background=true
directory="$USER_DATA_PATH"
pidfile="$USER_DATA_PATH/copyparty.pid"

depend() {
    need net
}
EOF
  chmod +x "$SERVICE_PATH_ALP"
  rc-update add copyparty default &>/dev/null
  rc-service copyparty start &>/dev/null
fi
msg_ok "Service created and started"

# IP detection (as root, maybe interface up/loopback fallback)
IFACE=$(ip -4 route | awk '/default/ {print $5; exit}')
IP=$(ip -4 addr show "$IFACE" | awk '/inet / {print $2}' | cut -d/ -f1 | head -n 1)
[[ -z "$IP" ]] && IP=$(hostname -I | awk '{print $1}')
[[ -z "$IP" ]] && IP="127.0.0.1"

echo -e "${CM} ${GN}$APP is running at: ${BL}http://$IP:$PORT${CL}"
echo -e "${INFO} Storage directory: ${YW}$USER_DATA_PATH${CL}"
if [[ -n "$AUTH_LINE" ]]; then
  echo -e "${INFO} Login: ${GN}${ADMIN_USER}${CL} / ${GN}${ADMIN_PASS}${CL}"
fi