#!/usr/bin/with-contenv bashio
# shellcheck shell=bash
# shellcheck disable=SC2155

####################
# Define variables #
####################

bashio::log.info "Defining variables"

# Define variables
export PAPERLESS_ADMIN_PASSWORD="admin"
export PAPERLESS_ADMIN_USER="admin"
if bashio::config.has_value "PUID"; then export USERMAP_UID="$(bashio::config "PUID")"; fi
if bashio::config.has_value "PGID"; then export USERMAP_GID="$(bashio::config "PGID")"; fi
if bashio::config.has_value "TZ"; then export PAPERLESS_TIME_ZONE="$(bashio::config "TZ")"; fi
if bashio::config.has_value "PAPERLESS_URL"; then export PAPERLESS_URL="$(bashio::config "PAPERLESS_URL")"; fi
if bashio::config.has_value "OCRLANG"; then
    PAPERLESS_OCR_LANGUAGES="$(bashio::config "OCRLANG")"
    export PAPERLESS_OCR_LANGUAGES="${PAPERLESS_OCR_LANGUAGES,,}"
fi
if bashio::config.has_value "PAPERLESS_OCR_MODE"; then export PAPERLESS_OCR_MODE="$(bashio::config "PAPERLESS_OCR_MODE")"; fi
if bashio::config.has_value "PAPERLESS_DATA_DIR"; then export PAPERLESS_DATA_DIR="$(bashio::config "PAPERLESS_DATA_DIR")"; fi
if bashio::config.has_value "PAPERLESS_MEDIA_ROOT"; then export PAPERLESS_MEDIA_ROOT="$(bashio::config "PAPERLESS_MEDIA_ROOT")"; fi
if bashio::config.has_value "PAPERLESS_CONSUMPTION_DIR"; then export PAPERLESS_CONSUMPTION_DIR="$(bashio::config "PAPERLESS_CONSUMPTION_DIR")"; fi
if bashio::config.has_value "PAPERLESS_EXPORT_DIR"; then export PAPERLESS_EXPORT_DIR="$(bashio::config "PAPERLESS_EXPORT_DIR")"; fi

# Redis is provided by this add-on (paperless-ngx expects an external broker)
export PAPERLESS_REDIS="redis://localhost:6379"

# Create folder and permissions if needed
chown -R paperless:paperless /config || true
for variable in "$PAPERLESS_DATA_DIR" "$PAPERLESS_MEDIA_ROOT" "$PAPERLESS_CONSUMPTION_DIR" "$PAPERLESS_EXPORT_DIR"; do
    [ -z "$variable" ] && continue
    echo "Creating directory \"$variable\""
    mkdir -p "$variable"
    chmod -R 755 "$variable"
    chown -R paperless:paperless "$variable"
done

###################
# Define database #
###################

bashio::log.info "Defining database"

case "$(bashio::config 'database')" in

    # Use mariadb
    mariadb_addon)
        bashio::log.info "Using MariaDB addon. Requirements : running MariaDB addon. Discovering values..."
        if ! bashio::services.available 'mysql'; then
            bashio::log.fatal \
                "Local database access should be provided by the MariaDB addon"
            bashio::exit.nok \
                "Please ensure it is installed and started"
        fi

        # Use values
        export PAPERLESS_DBENGINE=mariadb
        export PAPERLESS_DBHOST="$(bashio::services 'mysql' 'host')"
        export PAPERLESS_DBPORT="$(bashio::services 'mysql' 'port')"
        export PAPERLESS_DBNAME=paperless
        export PAPERLESS_DBUSER="$(bashio::services "mysql" "username")"
        export PAPERLESS_DBPASS="$(bashio::services "mysql" "password")"

        # Create database
        mysql --host="$PAPERLESS_DBHOST" --port="$PAPERLESS_DBPORT" --user="$PAPERLESS_DBUSER" --password="$PAPERLESS_DBPASS" -e"CREATE DATABASE IF NOT EXISTS $PAPERLESS_DBNAME;"

        # Informations
        bashio::log.warning "This addon is using the Maria DB addon"
        bashio::log.warning "Please ensure this is included in your backups"
        bashio::log.warning "Uninstalling the MariaDB addon will remove any data"
        ;;

        # Use sqlite
    *)
        bashio::log.info "Using sqlite as database driver"
        ;;
esac

set +u

#############################
# Export to s6 environment  #
#############################

# Upstream now runs on s6-overlay v3: variables are shared with the supervised
# services (svc-webserver, svc-worker, ...) by writing them to the s6
# container_environment, which is read by `with-contenv` when each service starts.
S6_ENV_DIR=""
for candidate in /var/run/s6/container_environment /run/s6/container_environment; do
    if [ -d "$candidate" ]; then S6_ENV_DIR="$candidate"; break; fi
done

# For all relevant variables
for variable in PAPERLESS_DATA_DIR PAPERLESS_MEDIA_ROOT PAPERLESS_CONSUMPTION_DIR PAPERLESS_EXPORT_DIR USERMAP_UID USERMAP_GID PAPERLESS_TIME_ZONE PAPERLESS_URL PAPERLESS_OCR_LANGUAGES PAPERLESS_OCR_MODE PAPERLESS_ADMIN_PASSWORD PAPERLESS_ADMIN_USER PAPERLESS_REDIS PAPERLESS_DBENGINE PAPERLESS_DBHOST PAPERLESS_DBPORT PAPERLESS_DBNAME PAPERLESS_DBUSER PAPERLESS_DBPASS; do

    # Skip if not defined
    if [[ -z "$(eval echo "\$$variable")" ]]; then continue; fi

    # Variable content
    variablecontent="$(eval echo "\$$variable")"
    # Sanitize " ' ` in current variable
    variablecontent="${variablecontent//[\"\'\`]/}"
    bashio::log.blue "$variable=\"$variablecontent\""
    # Export
    export "$variable"="$variablecontent"
    # Add to bashrc
    eval echo "$variable=\"$variablecontent\"" >> ~/.bashrc
    # set .env
    echo "$variable=\"$variablecontent\"" >> /.env || true
    # set /etc/environment
    mkdir -p /etc
    echo "$variable=\"$variablecontent\"" >> /etc/environment
    # For s6 (read by the upstream supervised services via with-contenv)
    if [ -n "$S6_ENV_DIR" ]; then printf "%s" "${variablecontent}" > "$S6_ENV_DIR/${variable}"; fi
done

#################
# Staring redis #
#################
if command -v redis-server > /dev/null 2>&1; then
    bashio::log.info "Starting redis"
    redis-server &
else
    bashio::log.fatal "redis-server is not installed; paperless-ngx requires it as broker"
    bashio::exit.nok "redis-server is required (install it in PACKAGES or configure an external broker)"
fi

#################
# Staring nginx #
#################
# nginx is only used for the optional SSL endpoint (port 8443). The web UI is
# served directly by the upstream svc-webserver on port 8000 regardless.
if command -v nginx > /dev/null 2>&1; then
    bashio::log.info "Starting nginx"
    nginx &
else
    bashio::log.warning "nginx not installed; skipping the optional SSL proxy (the web UI stays available on port 8000)"
fi

################
# Starting app #
################
# The upstream s6-overlay init (svc-webserver, svc-worker, svc-scheduler,
# svc-consumer, svc-flower) is started by s6 once this stage-2 hook returns.
bashio::log.info "Initial username and password are admin. Please change in the administration panel of the webUI after login."