docker-entrypoint.sh

#!/bin/sh
set -e

# Function to log messages
log() {
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] $1"
}

# Function to handle signals
cleanup() {
    log "Received termination signal, shutting down gracefully..."
    if [ ! -z "$APP_PID" ]; then
        kill -TERM "$APP_PID" 2>/dev/null || true
        wait "$APP_PID" 2>/dev/null || true
    fi
    exit 0
}

# Set up signal handlers
trap cleanup TERM INT

# Ensure data directory exists
mkdir -p /app/data

# Check if database exists
DB_PATH="${CHATVAULT_DB_PATH:-/app/data/chat.db}"

# Function to run migrations
run_migrations() {
    log "Running database migrations..."
    ./chatvault migrate || {
        log "Failed to run migrations"
        exit 1
    }
    log "Migrations completed successfully"
}

# Function to seed database
seed_database() {
    log "Seeding database with sample data..."
    ./chatvault seed || {
        log "Failed to seed database"
        exit 1
    }
    log "Database seeded successfully"
}

# Handle different startup modes
case "$1" in
    "migrate")
        log "Migration mode: Running migrations only"
        run_migrations
        exit 0
        ;;
    "seed")
        log "Seed mode: Running migrations and seeding database"
        run_migrations
        seed_database
        exit 0
        ;;
    "server"|"")
        log "Server mode: Starting ChatVault server"

        # Check if database exists, if not run migrations
        if [ ! -f "$DB_PATH" ]; then
            log "Database not found, running initial setup..."
            run_migrations

            # Optionally seed with sample data if CHATVAULT_SEED is set
            if [ "$CHATVAULT_SEED" = "true" ]; then
                seed_database
            fi
        else
            log "Database found at $DB_PATH"

            # Check if we should always run migrations on startup
            if [ "$CHATVAULT_AUTO_MIGRATE" = "true" ]; then
                log "Auto-migration enabled, running migrations..."
                run_migrations
            fi
        fi

        log "Starting ChatVault server..."

        # Start the server in the background so we can handle signals
        ./chatvault &
        APP_PID=$!

        # Wait for the server process
        wait "$APP_PID"
        ;;
    *)
        log "Unknown command: $1"
        log "Available commands: migrate, seed, server (default)"
        log "Environment variables:"
        log "  CHATVAULT_SEED=true        - Seed database on first run"
        log "  CHATVAULT_AUTO_MIGRATE=true - Run migrations on every startup"
        exit 1
        ;;
esac