#!/bin/bash

# ChromaDB Startup Script
# This script starts a ChromaDB instance using Docker

echo "Starting ChromaDB instance..."

# Check if Docker is running
if ! docker info > /dev/null 2>&1; then
    echo "Error: Docker is not running. Please start Docker first."
    exit 1
fi

# ChromaDB configuration
CHROMA_PORT=8000
CHROMA_CONTAINER_NAME="chromadb-olympus"
CHROMA_IMAGE="ghcr.io/chroma-core/chroma:1.3.5"
CHROMA_DATA_DIR="./chroma-data"

# Create data directory if it doesn't exist
mkdir -p "$CHROMA_DATA_DIR"

# Check if container already exists
if [ "$(docker ps -aq -f name=$CHROMA_CONTAINER_NAME)" ]; then
    echo "ChromaDB container already exists."
    
    # Check if it's running
    if [ "$(docker ps -q -f name=$CHROMA_CONTAINER_NAME)" ]; then
        echo "ChromaDB is already running on port $CHROMA_PORT"
        echo "Access it at: http://localhost:$CHROMA_PORT"
        exit 0
    else
        echo "Starting existing ChromaDB container..."
        docker start $CHROMA_CONTAINER_NAME
    fi
else
    echo "Creating and starting new ChromaDB container..."
    docker run -d \
        --name $CHROMA_CONTAINER_NAME \
        -p $CHROMA_PORT:8000 \
        -v "$(pwd)/$CHROMA_DATA_DIR:/chroma/chroma" \
        $CHROMA_IMAGE
fi

# Wait for ChromaDB to be ready
echo "Waiting for ChromaDB to be ready..."
sleep 3

# Check if ChromaDB is responding (with timeout)
if curl -s --max-time 5 http://localhost:$CHROMA_PORT/api/v1/heartbeat > /dev/null 2>&1; then
    echo "✅ ChromaDB is up and running!"
    echo "   - URL: http://localhost:$CHROMA_PORT"
    echo "   - Data directory: $CHROMA_DATA_DIR"
    echo "   - Container name: $CHROMA_CONTAINER_NAME"
    echo ""
    echo "To stop ChromaDB, run: docker stop $CHROMA_CONTAINER_NAME"
    echo "To view logs, run: docker logs -f $CHROMA_CONTAINER_NAME"
else
    echo "⚠️  ChromaDB container started. Checking if it's ready..."
    # Try without API endpoint - just check if container is running
    if docker ps | grep -q $CHROMA_CONTAINER_NAME; then
        echo "✅ ChromaDB container is running!"
        echo "   - URL: http://localhost:$CHROMA_PORT"
        echo "   - Data directory: $CHROMA_DATA_DIR"
        echo "   - Container name: $CHROMA_CONTAINER_NAME"
        echo "   - Note: API may take a few more seconds to be ready"
        echo ""
        echo "To stop ChromaDB, run: docker stop $CHROMA_CONTAINER_NAME"
        echo "To view logs, run: docker logs -f $CHROMA_CONTAINER_NAME"
    else
        echo "❌ ChromaDB container failed to start"
        echo "   Check logs with: docker logs $CHROMA_CONTAINER_NAME"
        exit 1
    fi
fi

echo ""
echo "Update your application.yml with:"
echo "  spring:"
echo "    ai:"
echo "      vectorstore:"
echo "        chroma:"
echo "          client:"
echo "            host: \"http://localhost\""
echo "            port: \"$CHROMA_PORT\""