LandPPT/docker-entrypoint.sh

#!/bin/bash

# LandPPT Docker Entrypoint Script
# This script handles initialization and startup of the LandPPT application

set -e

# Activate virtual environment
export VIRTUAL_ENV=/opt/venv
export PATH=/opt/venv/bin:$PATH

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Logging functions
log() {
    echo -e "${GREEN}[$(date +'%Y-%m-%d %H:%M:%S')] $1${NC}"
}

warn() {
    echo -e "${YELLOW}[$(date +'%Y-%m-%d %H:%M:%S')] WARNING: $1${NC}"
}

error() {
    echo -e "${RED}[$(date +'%Y-%m-%d %H:%M:%S')] ERROR: $1${NC}"
}

info() {
    echo -e "${BLUE}[$(date +'%Y-%m-%d %H:%M:%S')] INFO: $1${NC}"
}

# Banner
print_banner() {
    echo -e "${BLUE}"
    echo "╔══════════════════════════════════════════════════════════════════════════════╗"
    echo "║                                   LandPPT                                   ║"
    echo "║                        AI-Powered PPT Generation Platform                   ║"
    echo "║                                Docker Container                             ║"
    echo "╚══════════════════════════════════════════════════════════════════════════════╝"
    echo -e "${NC}"
}

# Check environment variables
check_environment() {
    log "Checking environment configuration..."
    
    # Check if at least one AI provider is configured
    local ai_configured=false
    
    if [ -n "$OPENAI_API_KEY" ] && [ "$OPENAI_API_KEY" != "your_openai_api_key_here" ]; then
        info "✅ OpenAI API key configured"
        ai_configured=true
    fi
    
    if [ -n "$ANTHROPIC_API_KEY" ] && [ "$ANTHROPIC_API_KEY" != "your_anthropic_api_key_here" ]; then
        info "✅ Anthropic API key configured"
        ai_configured=true
    fi
    
    if [ -n "$GOOGLE_API_KEY" ] && [ "$GOOGLE_API_KEY" != "your_google_api_key_here" ]; then
        info "✅ Google API key configured"
        ai_configured=true
    fi
    
    if [ -n "$AZURE_OPENAI_API_KEY" ] && [ "$AZURE_OPENAI_API_KEY" != "your_azure_openai_key_here" ]; then
        info "✅ Azure OpenAI API key configured"
        ai_configured=true
    fi
    
    if [ "$ENABLE_LOCAL_MODELS" = "true" ] && [ -n "$OLLAMA_BASE_URL" ]; then
        info "✅ Ollama configuration detected"
        ai_configured=true
    fi
    
    if [ "$ai_configured" = false ]; then
        warn "⚠️ No AI provider API keys configured. Please set at least one:"
        warn "   - OPENAI_API_KEY"
        warn "   - ANTHROPIC_API_KEY"
        warn "   - GOOGLE_API_KEY"
        warn "   - AZURE_OPENAI_API_KEY"
        warn "   - Or enable ENABLE_LOCAL_MODELS=true with Ollama"
    fi
    
    # Check secret key
    if [ "$SECRET_KEY" = "your-very-secure-secret-key-change-this-in-production" ] || [ "$SECRET_KEY" = "dev-secret-key-not-for-production" ]; then
        warn "⚠️ Using default SECRET_KEY. Please change it for production!"
    fi
}

# Create and verify runtime directories as the application user.
create_directories() {
    log "Checking runtime directories..."

    local dirs=(
        "/app/data"
        "/app/uploads"
        "/app/temp"
        "/app/temp/ai_responses_cache"
        "/app/temp/style_genes_cache"
        "/app/temp/summeryanyfile_cache"
        "/app/temp/templates_cache"
        "/app/research_reports"
        "/app/lib"
        "/app/lib/Linux"
        "/app/lib/MacOS"
        "/app/lib/Windows"
    )

    for dir in "${dirs[@]}"; do
        if ! mkdir -p "$dir" 2>/dev/null; then
            error "Required path cannot be created: $dir"
            return 1
        fi
        if [ ! -w "$dir" ] || [ ! -x "$dir" ]; then
            error "Required path is not writable: $dir"
            return 1
        fi
    done
}


# Import default templates
import_templates() {
    log "Checking template imports..."
    
    if [ -d "/app/template_examples" ] && [ "$(ls -A /app/template_examples/*.json 2>/dev/null)" ]; then
        info "Template examples found, they will be imported on first run"
    else
        warn "No template examples found in /app/template_examples"
    fi
}

# Wait for dependencies
wait_for_dependencies() {
    if [ -n "$OLLAMA_BASE_URL" ] && [ "$ENABLE_LOCAL_MODELS" = "true" ]; then
        log "Waiting for Ollama service..."
        
        local ollama_host=$(echo "$OLLAMA_BASE_URL" | sed 's|http://||' | sed 's|https://||' | cut -d':' -f1)
        local ollama_port=$(echo "$OLLAMA_BASE_URL" | sed 's|http://||' | sed 's|https://||' | cut -d':' -f2)
        
        if [ "$ollama_port" = "$ollama_host" ]; then
            ollama_port="11434"
        fi
        
        local max_attempts=30
        local attempt=1
        
        while [ $attempt -le $max_attempts ]; do
            if nc -z "$ollama_host" "$ollama_port" 2>/dev/null; then
                log "✅ Ollama service is ready"
                break
            fi
            
            if [ $attempt -eq $max_attempts ]; then
                warn "⚠️ Ollama service not available after $max_attempts attempts"
                break
            fi
            
            info "Waiting for Ollama... (attempt $attempt/$max_attempts)"
            sleep 2
            attempt=$((attempt + 1))
        done
    fi
}

# Refuse to start with a privileged identity.
check_runtime_identity() {
    if [ "$(id -u)" -eq 0 ]; then
        error "LandPPT must not run as root"
        return 1
    fi

    info "Running as uid=$(id -u) gid=$(id -g)"
}

# Validate .env access without trying to mutate mounted host files.
check_env_permissions() {
    log "Checking .env file permissions..."

    if [ ! -e "/app/.env" ]; then
        warn ".env file not found, using process environment only"
        return 0
    fi
    if [ ! -f "/app/.env" ]; then
        error "Required path is not a regular file: /app/.env"
        return 1
    fi
    if [ ! -r "/app/.env" ]; then
        error "Required path is not readable: /app/.env"
        return 1
    fi
    if [ ! -w "/app/.env" ]; then
        error "Required path is not writable: /app/.env"
        return 1
    fi

    log ".env file is readable and writable"
}

# Main initialization
main() {
    print_banner

    log "Starting LandPPT initialization..."

    check_runtime_identity
    check_environment
    check_env_permissions
    create_directories
    wait_for_dependencies
    import_templates

    log "🚀 Starting LandPPT application..."
    info "📍 Server will be available at: http://0.0.0.0:${PORT:-8000}"
    info "🧵 Workers: ${WORKERS:-1}"
    info "🌐 Web Interface: http://0.0.0.0:${PORT:-8000}/web"

    # Execute the main command
    exec "$@"
}

# Run main function with all arguments
main "$@"