#! /bin/sh

set -eu

check_git_repository() {
    # Temporary function because function-lib isn't sourced yet
    error() {
        echo "$(tput bold)$(tput setaf 1)Error: ${*}$(tput sgr0)" >&2
    }

    if ! git --version; then
        error "Git is not installed. Please install it: https://git-scm.com/downloads/"
        exit 1
    fi
    if ! git rev-parse --is-inside-work-tree >/dev/null 2>&1; then
        error "This script must be run from inside the pretendo-docker Git repository."
        echo "Make sure that you cloned the repository with Git, not downloaded it as a ZIP file from GitHub."
        exit 1
    fi
    if [ "$(basename -s .git "$(git config --get remote.origin.url)")" != "pretendo-docker" ]; then
        error "This script must be run from inside the pretendo-docker Git repository."
        echo "It looks like you are running it from a different Git repository at $(pwd) from $(git config --get remote.origin.url)."
        exit 1
    fi
}

check_prerequisites() {
    prerequisites_failed=false
    prerequisites_warning=false
    if ! docker version; then
        error "Docker is not installed. Please install it: https://docs.docker.com/get-docker/"
        info "If you see a \"Permission denied while trying to connect to the Docker daemon\" error, you need to run this script with sudo."
        prerequisites_failed=true
    fi
    if ! docker compose version; then
        error "Docker Compose is not installed. Please install it: https://docs.docker.com/compose/install/"
        prerequisites_failed=true
    fi
    if ! ftp -? >/dev/null; then
        warning "FTP is not installed. You will not be able to upload files to your consoles automatically."
        prerequisites_warning=true
    fi

    if [ "$prerequisites_failed" = true ]; then
        error "Prerequisites check failed. Please install the missing prerequisites and try again."
        exit 1
    elif [ "$prerequisites_warning" = true ]; then
        warning "Prerequisites check completed with warnings."
        printf "Do you want to continue anyway (y/N)? "
        read -r continue_anyway
        if [ "$continue_anyway" != "Y" ] && [ "$continue_anyway" != "y" ]; then
            exit 1
        fi
    else
        success "All prerequisites are installed."
    fi

}

ci_setup() {
    warning "It looks like the script is running in CI. Only setup tasks required for building the Docker images will be performed."
    stage "Setting up submodules and applying patches."
    ./scripts/setup-submodule-patches.sh
    stage "Setting up environment variables."
    ./scripts/setup-environment.sh "1.1.1.1" "2.2.2.2" "3.3.3.3"
    success "CI setup completed."
}

setup_environment_variables() {
    echo "Enter the IP address of your Pretendo Network server. It must be accessible to your console."
    read -r server_ip
    echo "Enter the IP address of your Wii U (optional). It is only used for automatic FTP uploads of modified Inkay patches."
    read -r wiiu_ip
    echo "Enter the IP address of your 3DS (optional). It is only used for automatic FTP uploads."
    read -r ds_ip
    ./scripts/setup-environment.sh "$server_ip" "$wiiu_ip" "$ds_ip"
}

setup_containers() {
    info "Setting up MongoDB container..."
    ./scripts/firstrun-mongodb-container.sh
    info "Setting up MinIO container..."
    ./scripts/firstrun-minio-container.sh
    info "Setting up Pretendo account servers database..."
    ./scripts/update-account-servers-database.sh
    info "Setting up Pretendo Miiverse endpoints database..."
    ./scripts/update-miiverse-endpoints.sh
    info "Updating Postgres password..."
    ./scripts/update-postgres-password.sh
    info "Stopping containers after initial setup..."
    docker compose down
}

export PRETENDO_SETUP_IN_PROGRESS=true

check_git_repository

git_base=$(git rev-parse --show-toplevel)
cd "$git_base"
. "$git_base/scripts/.function-lib.sh"

title "Unofficial Pretendo Network setup script"
header "Pretendo setup script started at $(date)."

if [ -n "${CI+x}" ]; then
    ci_setup
    exit 0
fi

stage "Checking prerequisites."
check_prerequisites

stage "Setting up submodules and applying patches."
./scripts/setup-submodule-patches.sh

stage "Setting up environment variables."
setup_environment_variables

stage "Pulling Docker images."
docker compose pull

stage "Building Docker images."
docker compose build

stage "Setting up containers with first-run scripts."
setup_containers

success "Setup completed! You can now start your Pretendo Network server with \"docker compose up -d\"."
header "Pretendo setup script finished at $(date)."