#!/bin/bash
# MeshAI Docker Entrypoint
# Runs ttyd for web config access and the bot

export MESHAI_CONFIG="/data/config.yaml"
export TERM="${TERM:-xterm-256color}"

# First run - no config exists, create defaults
if [ ! -f "$MESHAI_CONFIG" ]; then
    mkdir -p /data
    cat > "$MESHAI_CONFIG" << 'EOF'
# MeshAI Configuration
# Configure via http://localhost:7682

bot:
  name: ai
  owner: ""
  respond_to_mentions: true
  respond_to_dms: true

connection:
  type: tcp
  serial_port: /dev/ttyUSB0
  tcp_host: localhost
  tcp_port: 4403

channels:
  mode: all
  whitelist:
    - 0

response:
  delay_min: 2.2
  delay_max: 3.0
  max_length: 150
  max_messages: 2

rate_limits:
  messages_per_minute: 10
  global_messages_per_minute: 30
  cooldown_seconds: 5.0
  burst_allowance: 3

history:
  database: /data/conversations.db
  max_messages_per_user: 50
  conversation_timeout: 86400
  auto_cleanup: true
  cleanup_interval_hours: 24
  max_age_days: 30

memory:
  enabled: true
  window_size: 4
  summarize_threshold: 8

llm:
  backend: openai
  api_key: ""
  base_url: https://api.openai.com/v1
  model: gpt-4o-mini
  timeout: 30
  system_prompt: >-
    You are a helpful assistant on a Meshtastic mesh network.
    Keep responses VERY brief - under 250 characters total.
    Be concise but friendly. No markdown formatting.

web_status:
  enabled: false
  port: 8080
  show_uptime: true
  show_message_count: true
  show_connected_nodes: true
  show_recent_activity: false
  require_auth: false
  auth_password: ""

announcements:
  enabled: false
  interval_hours: 24
  channel: 0
  messages: []
  random_order: true
EOF
    echo "Default config created. Configure via http://localhost:7682"
fi

# Start ttyd for web-based config access
echo "Starting web config interface on port 7682..."
ttyd -W -p 7682 \
    -t enableClipboard=true \
    -t titleFixed="MeshAI Config" \
    -t 'theme={"background":"#0d1117","foreground":"#c9d1d9","cursor":"#58a6ff","selectionBackground":"#388bfd"}' \
    -t fontSize=14 \
    /bin/bash -c 'while true; do python3 -m meshai --config-file "$MESHAI_CONFIG" --config; sleep 1; done' &

# Keep ttyd running even if bot fails
trap "kill %1 2>/dev/null; kill %2 2>/dev/null" EXIT

# Restart watcher - monitors for restart signal from config tool
BOT_PID_FILE="/tmp/meshai_bot.pid"
(
    while true; do
        if [ -f /tmp/meshai_restart ]; then
            rm -f /tmp/meshai_restart
            echo "Restart signal received, restarting bot..."
            # Kill bot using PID file
            if [ -f "$BOT_PID_FILE" ]; then
                BOT_PID=$(cat "$BOT_PID_FILE")
                if kill -0 "$BOT_PID" 2>/dev/null; then
                    kill "$BOT_PID" 2>/dev/null || true
                    echo "Sent TERM to bot (PID $BOT_PID)"
                fi
            fi
            # Debounce - wait before checking for more signals
            sleep 3
        fi
        sleep 2
    done
) &

# Start the bot in a loop - retry on failure
echo "Starting MeshAI..."
while true; do
    python -m meshai --config-file "$MESHAI_CONFIG" &
    BOT_PID=$!
    echo "$BOT_PID" > "$BOT_PID_FILE"
    wait $BOT_PID || true
    rm -f "$BOT_PID_FILE"
    echo "Bot exited. Check config at http://localhost:7682. Retrying in 5s..."
    sleep 5
done