#!/usr/bin/env bashio
# shellcheck shell=bash
# shellcheck disable=SC2155,SC2016

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

bashio::log.info "Defining database"
bashio::log.info "-----------------"

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

    "external_postgresql")

        bashio::log.info "Using external postgresql"
        bashio::log.info ""

        # Check if values exist
        if ! bashio::config.has_value 'DB_USERNAME' && \
            ! bashio::config.has_value 'DB_HOSTNAME' && \
            ! bashio::config.has_value 'DB_PASSWORD' && \
            ! bashio::config.has_value 'DB_DATABASE_NAME' && \
            ! bashio::config.has_value 'JWT_SECRET' && \
            ! bashio::config.has_value 'DB_PORT'
        then
            ! bashio::exit.nok "Please make sure that the following options are set : DB_USERNAME, DB_HOSTNAME, DB_PASSWORD, DB_DATABASE_NAME, DB_PORT"
        fi

        # Settings parameters
        export DB_USERNAME=$(bashio::config 'DB_USERNAME')
        export DB_HOSTNAME=$(bashio::config 'DB_HOSTNAME')
        export DB_PASSWORD=$(bashio::config 'DB_PASSWORD')
        export DB_DATABASE_NAME=$(bashio::config 'DB_DATABASE_NAME')
        export DB_PORT=$(bashio::config 'DB_PORT')
        export JWT_SECRET=$(bashio::config 'JWT_SECRET')
        ;;

    **)

        bashio::log.info "Using internal postgresql"
        bashio::log.info ""

        # Settings files & permissions
        ln -s /usr/lib/postgresql/14/bin/postgres /usr/bin || true
        ln -s /usr/lib/postgresql/14/bin/psql /usr/psql || true
        mkdir -p /data/postgresql
        cp -rnf /var/lib/postgresql/14/main/* /data/postgresql/
        chown -R postgres /data/postgresql
        chmod -R 700 /data/postgresql

        # Start postgresql
        /etc/init.d/postgresql start

        # Create database
        echo "CREATE ROLE root WITH LOGIN SUPERUSER CREATEDB CREATEROLE PASSWORD 'securepassword';
             CREATE DATABASE immich; CREATE USER immich WITH ENCRYPTED PASSWORD 'immich';
             GRANT ALL PRIVILEGES ON DATABASE immich to immich;
        \q"> setup_postgres.sql
        chown postgres setup_postgres.sql
        # shellcheck disable=SC2024
        sudo -iu postgres psql < setup_postgres.sql
        rm setup_postgres.sql

        # Settings parameters
        export DB_USERNAME=immich
        export DB_HOSTNAME=localhost
        export DB_PASSWORD=immich
        export DB_DATABASE_NAME=immich
        export DB_PORT=5432
        export JWT_SECRET=$(bashio::config 'JWT_SECRET')
        ;;

esac

# Export variables
if [ -d /var/run/s6/container_environment ]; then
    printf "%s" "$DB_USERNAME" > /var/run/s6/container_environment/DB_USERNAME
    printf "%s" "$DB_PASSWORD" > /var/run/s6/container_environment/DB_PASSWORD
    printf "%s" "$DB_DATABASE_NAME" > /var/run/s6/container_environment/DB_DATABASE_NAME
    printf "%s" "$DB_PORT" > /var/run/s6/container_environment/DB_PORT
    printf "%s" "$DB_HOSTNAME" > /var/run/s6/container_environment/DB_HOSTNAME
    printf "%s" "$JWT_SECRET" > /var/run/s6/container_environment/JWT_SECRET
fi

{
    printf "%s" "DB_USERNAME=\"$DB_USERNAME\""
    printf "%s" "DB_PASSWORD=\"$DB_PASSWORD\""
    printf "%s" "DB_DATABASE_NAME=\"$DB_DATABASE_NAME\""
    printf "%s" "DB_PORT=\"$DB_PORT\""
    printf "%s" "DB_HOSTNAME=\"$DB_HOSTNAME\""
    printf "%s" "JWT_SECRET=\"$JWT_SECRET\""
} >> ~/.bashrc