#!/bin/bash
source "/variables.sh"
##############################################################################################################################
# Funktionen
##############################################################################################################################
function print_container_info {
  sepurator
  echo "* BorgServer powered by $BORG_VERSION"
  echo "* Image Hostname: $HOSTNAME"
  echo "* Image Version: $DOCKER_IMAGE_VERSION"
}

function print_user_info {
  sepurator
  echo "* USER:  $USER - ID:  $UID"
  echo "* GROUP: $USER - GID: $GID"
}

function add_borg_user {
  if ! id "borg" &>/dev/null; then
    sh -c "echo '$USER ALL=(ALL) NOPASSWD: ALL' >> /etc/sudoers"
    adduser \
    -s /bin/bash \
    --disabled-password \
    --gecos "" \
    --home "/" \
    --uid "$UID" \
    "$USER"
    echo "$USER:*" | chpasswd 2>> /dev/null
    addgroup -g "$GID" "$USER"  2>> /dev/null
  fi
}

function make_and_import_ssh_keys {
  local create_folders="0"

  if [ ! -f "/.ssh/authorized_keys" ]; then
    touch "/.ssh/authorized_keys"
  else
    rm "/.ssh/authorized_keys"
    touch "/.ssh/authorized_keys"
  fi

  for key in ${SSH_FOLDERS[@]}; do
    if [ ! -d "${key}" ]; then
      mkdir -p "${key}"
      echo "Created ${key}"
      create_folders="1"
    fi
  done

  chown -R "$USER":"$USER" "/sshkeys"

  if [ $create_folders == "1" ]; then
    sepurator
  fi

  echo "* IMPORT SSH KEYS"
  echo ""

  FILES=$(ls -1 /sshkeys/clients)
  for key in $FILES; do
    echo "-  Adding SSH-Key $key"
    cat "/sshkeys/clients/$key" >> "/.ssh/authorized_keys"
    echo "" >> "/.ssh/authorized_keys"
  done

  chown -R "$USER":"$USER" "/.ssh"
  chmod 700 "/.ssh"
  chmod 600 "/.ssh/authorized_keys"
}

function print_message {
  echo ""
  echo "- $1"
  echo ""
}

function generate_host_sshkey {
  local generated_keys="0"
  echo "* GENERATE HOST SSH-KEYs"
  # Generate SSH-Keys
  if [ ! -f "/sshkeys/host/ssh_host_rsa_key" ]; then
    print_message "HOST SSH-KEY RSA not found, generating..."
    ssh-keygen -t rsa -b 4096 -f "/sshkeys/host/ssh_host_rsa_key" -N ""
    print_message "HOST SSH-KEY RSA Generated"
    generated_keys="1"
  fi
  if [ ! -f "/sshkeys/host/ssh_host_ecdsa_key" ]; then
    print_message "HOST SSH-KEY ECDSA not found, generating..."
    ssh-keygen -t ecdsa -b 521 -f "/sshkeys/host/ssh_host_ecdsa_key" -N ""
    print_message "HOST SSH-KEY ECDSA Generated"
    generated_keys="1"
  fi
  if [ ! -f "/sshkeys/host/ssh_host_ed25519_key" ]; then
    print_message "HOST SSH-KEY ED25519 not found, generating..."
    ssh-keygen -t ed25519 -b 521 -f "/sshkeys/host/ssh_host_ed25519_key" -N ""
    print_message "HOST SSH-KEY ED25519 Generated"
    generated_keys="1"
  fi

  if [ "$generated_keys" == "0" ]; then
    echo ""
    echo "- HOST SSH-KEYs already exist"
  fi

  chown -R "$USER":"$USER" "/sshkeys/host"
}

function maintenance_enable {
  if [ "$MAINTENANCE_ENABLE" != "false" ]; then
    echo "* MAINTENANCE MODE - ENABLED"
    echo ""
    if [ -f "/crontab.txt" ]; then
      /usr/bin/crontab "/crontab.txt"
      /usr/sbin/crond -b 2> /dev/null
      echo "- Crontab loaded successfully"
    else
      echo "- Can not find /crontab.txt"
    fi
    sepurator
  fi
}

function set_timezone {
  if [ "$TZ" != "" ]; then
    echo "* Setting Timezone to $TZ"
    echo "TZ=$TZ" > /etc/environment
  else
    echo "* Timezone not set - Use UTC Time"
  fi
  sepurator
}

function run_install_script {
  if [ "$RUN_INSTALL_SCRIPT" != "false" ]; then
    if [ ! -f "/.runnedInstall" ]; then
      echo "* RUNNING INSTALL SCRIPT"
      sepurator
      sh "$RUN_INSTALL_SCRIPT"
      echo ""
      sepurator
      touch "/.runnedInstall"
    fi
  fi
}

function run_prometheus_exporter() {
  if [ "$RUN_PROMETHEUS_EXPORTER" != "false" ]; then
    echo "* STARTING Prometheus Exporter for Borg Backup"

    crontab -l > /tmp/cron_bkp
    echo "" >> /tmp/cron_bkp

    echo "* Add Cronjob to Crontab"
    echo "$RUN_PROMETHEUS_EXPORTER /usr/local/bin/borg_exporter.sh 2>&1" >> /tmp/cron_bkp
    crontab /tmp/cron_bkp
    rm /tmp/cron_bkp

    echo "* STARTING Node Exporter"
    node_exporter --collector.textfile.directory="$NODE_EXPORTER_DIR" &
    sepurator
  fi
}
##############################################################################################################################
# Main Code
##############################################################################################################################
add_borg_user

print_container_info
print_user_info
sepurator
make_and_import_ssh_keys
sepurator
generate_host_sshkey
sepurator

maintenance_enable
set_timezone
run_prometheus_exporter
run_install_script

echo "* Init done! - Starting SSH-Daemon..."
sepurator
echo ""
exec /usr/sbin/sshd -D -e "$@" 2> /logs/sshd.log