setup.sh

#!/usr/bin/env bash

set -euo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$ROOT_DIR"

print_step() {
    echo
    echo "==> $1"
}

copy_env_files() {
    print_step "Creating .env files"

    find . -maxdepth 1 -name "*.example" | while read -r file; do
        target="${file%.example}"

        if [ ! -f "$target" ]; then
            cp "$file" "$target"
            echo "Created $target"
        else
            echo "$target already exists"
        fi
    done
}

wait_ollama() {
    print_step "Waiting for Ollama"

    until docker exec ollama ollama list >/dev/null 2>&1; do
        sleep 2
    done

    echo "Ollama is ready."
}

ensure_model() {
    local model="$1"

    if docker exec ollama ollama list | awk '{print $1}' | grep -Fxq "$model"; then
        echo "$model already installed."
    else
        echo "Downloading $model..."
        docker exec ollama ollama pull "$model"
    fi
}

install_models() {
    print_step "Installing Ollama models"

    while IFS= read -r model || [ -n "$model" ]; do

        model="$(echo "$model" | xargs)"

        [[ -z "$model" ]] && continue
        [[ "$model" =~ ^# ]] && continue

        ensure_model "$model"

    done < models.txt
}

run_alembic() {
    print_step "Running Alembic migrations"

    docker compose exec web alembic upgrade head
}

check_requirements() {

    print_step "Checking Docker"

    docker -v >/dev/null 2>&1 || {
        echo "Docker is not installed."
        exit 1
    }

    docker compose version >/dev/null 2>&1 || {
        echo "Docker Compose is not available."
        exit 1
    }
}

start_containers() {

    print_step "Starting containers"

    docker compose up -d --build
}

finish() {

    echo
    echo "======================================="
    echo "Setup completed successfully!"
    echo "Frontend : http://localhost:8000"
    echo "======================================="
}

check_requirements
copy_env_files
start_containers
wait_ollama
install_models
run_alembic
finish