#!/usr/bin/env bash
set -eu

# Listmonk production setup using `docker compose`.
# See https://listmonk.app/docs/installation/ for detailed installation steps.

printf '\n'

RED="$(tput setaf 1 2>/dev/null || printf '')"
BLUE="$(tput setaf 4 2>/dev/null || printf '')"
GREEN="$(tput setaf 2 2>/dev/null || printf '')"
NO_COLOR="$(tput sgr0 2>/dev/null || printf '')"

info() {
  printf '%s\n' "${BLUE}> ${NO_COLOR} $*"
}

error() {
  printf '%s\n' "${RED}x $*${NO_COLOR}" >&2
}

completed() {
  printf '%s\n' "${GREEN}✓ ${NO_COLOR} $*"
}

exists() {
  command -v "$1" >/dev/null 2>&1
}

sed_inplace() {
    local search_pattern="$1"
    local replacement="$2"
    local file="$3"
    if [[ "$(uname)" == "Darwin" ]]; then
        sed -i '' "s/${search_pattern}/${replacement}/g" "$file"
    else
        sed -i "s/${search_pattern}/${replacement}/g" "$file"
    fi
}

check_dependencies() {
	if ! exists curl; then
		error "curl is not installed."
		exit 1
	fi

	if ! exists docker; then
		error "docker is not installed."
		exit 1
	fi

	# Check for "docker compose" functionality.
	if ! docker compose version >/dev/null 2>&1; then
		echo "'docker compose' functionality is not available. Please update to a newer version of Docker. See https://docs.docker.com/engine/install/ for more details."
		exit 1
	fi

	if ! docker compose ls >/dev/null; then
		echo "no docker access - you might need to run this script with sudo or see https://docs.docker.com/engine/install/linux-postinstall/#manage-docker-as-a-non-root-user"
		exit 1
	fi
}

check_existing_db_volume() {
	info "checking for an existing docker db volume"
	if docker volume inspect listmonk_listmonk-data >/dev/null 2>&1; then
		error "listmonk-data volume already exists. Please use docker compose down -v to remove old volumes for a fresh setup of PostgreSQL."
		exit 1
	fi
}

download() {
	curl --fail --silent --location --output "$2" "$1"
}

is_healthy() {
	info "waiting for db container to be up. retrying in 3s"
	health_status="$(docker inspect -f "{{.State.Health.Status}}" "$1")"
	if [ "$health_status" = "healthy" ]; then
		return 0
	else
		return 1
	fi
}

is_running() {
	info "checking if $1 is running"
	status="$(docker inspect -f "{{.State.Status}}" "$1")"
	if [ "$status" = "running" ]; then
		return 0
	else
		return 1
	fi
}

generate_password(){
	echo "$(LC_ALL=C tr -dc A-Za-z0-9 </dev/urandom | head -c 13 ; echo '')"
}

get_config() {
	info "fetching config.toml from listmonk repo"
	download https://raw.githubusercontent.com/knadh/listmonk/master/config.toml.sample config.toml
}

get_containers() {
	info "fetching docker-compose.yml from listmonk repo"
	download https://raw.githubusercontent.com/knadh/listmonk/master/docker-compose.yml docker-compose.yml
}

modify_config(){
	info "generating a random password"
	db_password=$(generate_password)

	info "modifying config.toml"
	sed_inplace 'host = "localhost"' 'host = "listmonk_db"' config.toml
	sed_inplace 'password = "listmonk"' "password = \"${db_password}\"" config.toml
	sed_inplace 'address = "localhost:9000"' 'address = "0.0.0.0:9000"' config.toml

	info "modifying docker-compose.yml"
	sed_inplace 'POSTGRES_PASSWORD=listmonk' "POSTGRES_PASSWORD=$db_password" docker-compose.yml
}

run_migrations(){
	info "running migrations"
	docker compose up -d db
	while ! is_healthy listmonk_db; do sleep 3; done
	docker compose run --rm app ./listmonk --install
}

start_services(){
	info "starting app"
	docker compose up -d app db
}

show_output(){
	info "finishing setup"
	sleep 3

	if is_running listmonk_db && is_running listmonk_app
	then completed "Listmonk is now up and running. Visit http://localhost:9000 in your browser."
	else
		error "error running containers. something went wrong."
	fi
}

check_dependencies
check_existing_db_volume
get_config
get_containers
modify_config
run_migrations
start_services
show_output