#!/bin/bash

# Function to show usage
show_usage() {
    echo "Usage: $0 [OPTIONS]"
    echo ""
    echo "OPTIONS:"
    echo "  --version VERSION       Set the version tag (default: v0.1.0-alpha.1)"
    echo "  --platforms PLATFORMS   Set target platforms (default: linux/amd64,linux/arm64; local --load uses the first platform)"
    echo "  --skills-ref REF        FalkorDB/skills ref to bake in (default: main for local builds; --push requires full commit SHA)"
    echo "  --image-name NAME       Set image name (default: text-to-cypher)"
    echo "  --registry REGISTRY     Set registry prefix (e.g., ghcr.io/owner/repo)"
    echo "  --push                  Push to registry (default: load locally)"
    echo "  --local                 Load locally only (default behavior)"
    echo "  --help                  Show this help message"
    echo ""
    echo "Examples:"
    echo "  $0 --local --version latest"
    echo "  $0 --platforms linux/amd64 --version v1.0.0"
    echo "  $0 --version v1.0.0 --skills-ref <commit-sha> --push"
    echo "  $0 --registry ghcr.io/owner/repo --push --version v1.0.0 --skills-ref <commit-sha>"
}

# Default values
VERSION="v0.1.0-alpha.1"
PLATFORMS="linux/amd64,linux/arm64"
SKILLS_REF="main"
SKILLS_REF_EXPLICIT=false
IMAGE_NAME="text-to-cypher"
REGISTRY=""
PUSH=false

# Parse command line arguments
while [[ $# -gt 0 ]]; do
    case $1 in
        --version)
            VERSION="$2"
            shift 2
            ;;
        --platforms)
            PLATFORMS="$2"
            shift 2
            ;;
        --skills-ref)
            SKILLS_REF="$2"
            SKILLS_REF_EXPLICIT=true
            shift 2
            ;;
        --image-name)
            IMAGE_NAME="$2"
            shift 2
            ;;
        --registry)
            REGISTRY="$2"
            shift 2
            ;;
        --push)
            PUSH=true
            shift
            ;;
        --local)
            PUSH=false
            shift
            ;;
        --help)
            show_usage
            exit 0
            ;;
        *)
            echo "Unknown option: $1"
            show_usage
            exit 1
            ;;
    esac
done

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

if [ "$PUSH" = "true" ]; then
    if [ "$SKILLS_REF_EXPLICIT" != "true" ]; then
        echo -e "${RED}Release/push builds must set --skills-ref to a full FalkorDB/skills commit SHA.${NC}"
        echo "Use --skills-ref main only for local/dev builds."
        exit 1
    fi

    if [[ ! "$SKILLS_REF" =~ ^[0-9a-fA-F]{40}$ ]]; then
        echo -e "${RED}Release/push builds require an immutable 40-character FalkorDB/skills commit SHA.${NC}"
        echo "Received: '${SKILLS_REF}'"
        exit 1
    fi
elif [[ "$PLATFORMS" == *,* ]]; then
    LOCAL_PLATFORM="${PLATFORMS%%,*}"
    echo -e "${YELLOW}Local Docker loads support one platform; using ${LOCAL_PLATFORM} from '${PLATFORMS}'.${NC}"
    echo "Use --push for multi-platform images."
    PLATFORMS="${LOCAL_PLATFORM}"
fi

echo -e "${YELLOW}Building Docker image with buildx...${NC}"
echo -e "${BLUE}Version: ${VERSION}${NC}"
echo -e "${BLUE}Platforms: ${PLATFORMS}${NC}"
echo -e "${BLUE}Skills ref: ${SKILLS_REF}${NC}"
echo -e "${BLUE}Image: ${IMAGE_NAME}${NC}"
if [ -n "$REGISTRY" ]; then
    echo -e "${BLUE}Registry: ${REGISTRY}${NC}"
    FULL_IMAGE_NAME="${REGISTRY}/${IMAGE_NAME}"
else
    FULL_IMAGE_NAME="${IMAGE_NAME}"
fi

# Ensure buildx is available
if ! docker buildx version >/dev/null 2>&1; then
    echo -e "${RED}Docker buildx is required for multi-architecture builds${NC}"
    echo "Please install Docker Desktop or enable buildx"
    exit 1
fi

# Create a builder instance if it doesn't exist
BUILDER_NAME="text-to-cypher-builder"
if ! docker buildx inspect ${BUILDER_NAME} >/dev/null 2>&1; then
    echo -e "${YELLOW}Creating buildx builder instance...${NC}"
    docker buildx create --name ${BUILDER_NAME} --driver docker-container --bootstrap
fi

echo -e "${YELLOW}Using buildx builder: ${BUILDER_NAME}${NC}"
docker buildx use ${BUILDER_NAME}

# Build the Docker image
if [ "$PUSH" = "true" ]; then
    echo -e "${GREEN}Building and pushing multi-platform Docker image...${NC}"
    docker buildx build \
        --platform "${PLATFORMS}" \
        --build-arg "VERSION=${VERSION}" \
        --build-arg "SKILLS_REF=${SKILLS_REF}" \
        -t "${FULL_IMAGE_NAME}:${VERSION}" \
        -t "${FULL_IMAGE_NAME}:latest" \
        --push \
        .
else
    echo -e "${GREEN}Building Docker image for local load (${PLATFORMS})...${NC}"
    docker buildx build \
        --platform "${PLATFORMS}" \
        --build-arg "VERSION=${VERSION}" \
        --build-arg "SKILLS_REF=${SKILLS_REF}" \
        -t "${FULL_IMAGE_NAME}:${VERSION}" \
        -t "${FULL_IMAGE_NAME}:latest" \
        --load \
        .
fi

# Check if build was successful
if [ $? -eq 0 ]; then
    echo -e "${GREEN}✅ Docker image built successfully!${NC}"
    echo ""
    echo -e "${YELLOW}To run the container:${NC}"
    echo "  docker run -p 6379:6379 -p 3000:3000 -p 8080:8080 -p 3001:3001 ${FULL_IMAGE_NAME}:${VERSION}"
    echo ""
    echo -e "${YELLOW}Available ports:${NC}"
    echo "  - 6379: FalkorDB (Redis protocol)"
    echo "  - 3000: FalkorDB web interface"
    echo "  - 8080: text-to-cypher HTTP API"
    echo "  - 3001: text-to-cypher additional port"
    echo ""
    echo -e "${YELLOW}To run with environment variables:${NC}"
    echo "  docker run -p 6379:6379 -p 3000:3000 -p 8080:8080 -p 3001:3001 \\"
    echo "    -e DEFAULT_MODEL=gpt-4o-mini -e DEFAULT_KEY=your-key ${FULL_IMAGE_NAME}:${VERSION}"
    echo ""
    echo -e "${BLUE}Built for platforms: ${PLATFORMS}${NC}"
    
    if [ "$PUSH" = "true" ]; then
        echo -e "${BLUE}Image pushed to registry${NC}"
        if [ -n "$REGISTRY" ]; then
            echo -e "${BLUE}Registry: ${REGISTRY}${NC}"
        fi
    else
        echo -e "${BLUE}Image available locally (use --push to push to registry)${NC}"
    fi
else
    echo -e "${RED}❌ Docker build failed!${NC}"
    exit 1
fi