#!/bin/bash
# Build script for RinHash ROCm GPU support using official ROCm Docker containers
# Based on https://github.com/ROCm/ROCm-docker

set -e

echo "=================================================="
echo "  RinHash ROCm GPU Official Docker Build Script"
echo "=================================================="

# Check if Docker is available
if ! command -v docker &> /dev/null; then
    echo "ERROR: Docker not found in PATH"
    echo "Please install Docker first"
    exit 1
fi

echo "Docker found:"
docker --version
echo ""

# Build directory setup
BUILD_DIR="$(dirname "$0")"
cd "$BUILD_DIR" || exit 1

echo "Using official ROCm Docker containers from:"
echo "https://github.com/ROCm/ROCm-docker"
echo ""
echo "This build includes:"
echo "  - Official ROCm/HIP development environment"
echo "  - RinHash HIP GPU implementation"
echo "  - Shared libraries for GPU acceleration"
echo "  - Integration documentation"
echo ""

# Pull the official ROCm terminal image first
echo "Pulling official ROCm terminal image..."
sudo docker pull rocm/rocm-terminal:latest

if [ $? -ne 0 ]; then
    echo ""
    echo "==============================================="
    echo "   FAILED TO PULL ROCm IMAGE!"
    echo "==============================================="
    echo ""
    echo "Common issues:"
    echo "1. Network connectivity problems"
    echo "2. Docker daemon not running"
    echo "3. Insufficient permissions"
    echo ""
    echo "Try running: sudo systemctl start docker"
    echo ""
    exit 1
fi

echo "ROCm image pulled successfully!"
echo ""

# Build the Docker image
echo "Building Docker image with RinHash ROCm support..."
sudo docker build -f Dockerfile.rocm-official -t rinhash-rocm-official .

if [ $? -ne 0 ]; then
    echo ""
    echo "==============================================="
    echo "   DOCKER BUILD FAILED!"
    echo "==============================================="
    echo ""
    echo "Common issues:"
    echo "1. Source files not found"
    echo "2. Build dependencies missing"
    echo "3. ROCm compilation errors"
    echo ""
    exit 1
fi

echo ""
echo "Docker image built successfully!"
echo ""

# Create output directory
mkdir -p rocm-official-output

echo "Extracting built binaries from container..."
echo ""

# Run the container and extract binaries
sudo docker run --rm \
    -v "$(pwd)/rocm-official-output:/output" \
    rinhash-rocm-official

if [ $? -eq 0 ]; then
    echo ""
    echo "==============================================="
    echo "   BUILD SUCCESSFUL!"
    echo "==============================================="
    echo ""
    echo "ROCm GPU libraries created in rocm-official-output/:"
    ls -la rocm-official-output/
    echo ""
    
    # Test the built libraries
    echo "Testing built ROCm libraries..."
    if [ -f "rocm-official-output/librinhash_hip.so" ]; then
        echo "ROCm GPU library found:"
        file rocm-official-output/librinhash_hip.so
        echo ""
        echo "Library size:"
        du -h rocm-official-output/librinhash_hip.so
    fi
    
    echo ""
    echo "Integration files created:"
    echo "  - librinhash_hip.so (shared library)"
    echo "  - test-rocm.sh (test script)"
    echo "  - INTEGRATION.md (integration guide)"
    echo ""
    
    echo "Next steps:"
    echo "  1. Install ROCm runtime on your system:"
    echo "     sudo apt install rocm-dev hip-runtime-amd"
    echo ""
    echo "  2. Copy the shared library:"
    echo "     sudo cp rocm-official-output/librinhash_hip.so /usr/local/lib/"
    echo "     sudo ldconfig"
    echo ""
    echo "  3. Test ROCm support:"
    echo "     ./rocm-official-output/test-rocm.sh"
    echo ""
    echo "  4. Read integration guide:"
    echo "     cat rocm-official-output/INTEGRATION.md"
    echo ""
else
    echo ""
    echo "==============================================="
    echo "   BUILD FAILED!"
    echo "==============================================="
    echo ""
    echo "Check the error messages above for details."
    echo ""
    exit 1
fi

echo "ROCm GPU build completed successfully!"
echo ""
echo "For more information about ROCm Docker containers, visit:"
echo "https://github.com/ROCm/ROCm-docker"