#!/bin/bash

set -u

umask 077

data_home="${XDG_DATA_HOME:-${HOME:?HOME is not set}/.local/share}"
data_root="${KOMP_AC_DATA_DIR:-$data_home/komp_ac}"
postgres_root="$data_root/postgresql"
dump_root="$data_root/dumps"
tantivy_root="$data_root/tantivy_indexes"
secret_file="$data_root/jwt.secret"
log_file="$data_root/server.log"
lock_dir="$data_root/flatpak.lock"
server=/app/bin/komp-ac-server
gui=/app/bin/client-gui2

mkdir -p "$data_root" "$postgres_root" "$dump_root" "$tantivy_root"

if [[ ! -s "$secret_file" ]]; then
    secret_tmp="$secret_file.$$"
    if ! od -An -N 48 -tx1 /dev/urandom | tr -d ' \n' > "$secret_tmp"; then
        rm -f -- "$secret_tmp"
        echo "Komp AC: could not generate the local server secret" >&2
        exit 1
    fi
    chmod 600 "$secret_tmp"
    if [[ -e "$secret_file" ]]; then
        rm -f -- "$secret_tmp"
    else
        mv -- "$secret_tmp" "$secret_file"
    fi
fi

if ! mkdir "$lock_dir" 2>/dev/null; then
    lock_pid=""
    if [[ -r "$lock_dir/pid" ]]; then
        read -r lock_pid < "$lock_dir/pid" || true
    fi
    if [[ "$lock_pid" =~ ^[0-9]+$ ]] && kill -0 "$lock_pid" 2>/dev/null; then
        echo "Komp AC is already running (launcher PID $lock_pid)." >&2
        exit 1
    fi
    rm -f -- "$lock_dir/pid"
    if ! rmdir "$lock_dir" 2>/dev/null || ! mkdir "$lock_dir" 2>/dev/null; then
        echo "Komp AC: could not acquire the application lock at $lock_dir" >&2
        exit 1
    fi
fi
printf '%s\n' "$$" > "$lock_dir/pid"

server_pid=""
cleanup() {
    status=$?
    trap - EXIT INT TERM HUP
    if [[ -n "$server_pid" ]] && kill -0 "$server_pid" 2>/dev/null; then
        kill -TERM "$server_pid" 2>/dev/null || true
        wait "$server_pid" 2>/dev/null || true
    fi
    rm -f -- "$lock_dir/pid"
    rmdir "$lock_dir" 2>/dev/null || true
    exit "$status"
}
trap cleanup EXIT
trap 'exit 130' INT
trap 'exit 143' TERM
trap 'exit 129' HUP

export GRPC_ENDPOINT=http://127.0.0.1:50051
export KOMP_AC_TYPST=/app/bin/typst

if (exec 3<>/dev/tcp/127.0.0.1/50051) 2>/dev/null; then
    echo "Komp AC: local port 50051 is already in use" >&2
    exit 1
fi

RUST_DB_MODE=embedded \
KOMP_AC_POSTGRES_DIR="$postgres_root" \
PG_DUMP_DIR="$dump_root" \
TANTIVY_INDEX_DIR="$tantivy_root" \
JWT_SECRET="$(<"$secret_file")" \
GRPC_LISTEN_ADDRESS=127.0.0.1:50051 \
"$server" server >> "$log_file" 2>&1 &
server_pid=$!

ready=false
for ((attempt = 0; attempt < 600; attempt++)); do
    if ! kill -0 "$server_pid" 2>/dev/null; then
        wait "$server_pid" || true
        echo "Komp AC: the bundled server failed to start; see $log_file" >&2
        exit 1
    fi
    if (exec 3<>/dev/tcp/127.0.0.1/50051) 2>/dev/null; then
        ready=true
        break
    fi
    sleep 0.1
done

if [[ "$ready" != true ]]; then
    echo "Komp AC: the bundled server did not become ready; see $log_file" >&2
    exit 1
fi

WEBKIT_DISABLE_COMPOSITING_MODE="${WEBKIT_DISABLE_COMPOSITING_MODE:-1}" \
WEBKIT_DISABLE_DMABUF_RENDERER="${WEBKIT_DISABLE_DMABUF_RENDERER:-1}" \
LIBGL_ALWAYS_SOFTWARE="${LIBGL_ALWAYS_SOFTWARE:-1}" \
"$gui"