#!/bin/bash

set -e

function log_generic { # $1: color, $2: prefix, $3: message #
    echo -e "[$(tput setaf $1)$(tput bold)$2$(tput sgr0)] $3"
}

function log_error { # $1: message #
    log_generic 1 ERR "$1"
}

function log_ok { # $1: message #
    log_generic 2 "OK " "$1"
}

function log_info { # $1: message #
    log_generic 4 INF "$1"
}

function print_help {
    echo "
Pleroma Maintenance Script

Usage:
    $0 [action] [flags...]

Actions:
    build      Build the pleroma container and all dependencies
    configure  Runs the interactive configuration script
    run        Start pleroma and sibling services
    stop       Stop pleroma and sibling services
"
}

function run_dockerized {
    log_info "Stopping existing containers (if any)..."
    docker-compose down

    log_info "Rebuilding images..."
    docker-compose build

    log_info "Running action '$1'..."
    docker-compose run server $1

    log_info "Cleaning up.."
    docker-compose down
}

function action__build {
    run_dockerized "build"
    log_ok "Done"
}

function action__configure {
    run_dockerized "configure"
    log_ok "Done"
}

function action__run {
    log_info "Booting pleroma..."
    docker-compose up --remove-orphans -d
    log_ok "Done"
}

function action__stop {
    log_info "Stopping pleroma..."
    docker-compose down
    log_ok "Done"
}

if [[ -z "$1" ]]; then
    log_error "No action provided."
    print_help
    exit 1
fi

case "$1" in
build) action__build;;
configure) action__configure;;
run) action__run;;
stop) action__stop;;
*)
    log_error "The action '$1' is invalid."
    print_help
    exit 1
;;
esac
shift