#!/bin/bash

# Docker 镜像重新编译脚本
# 用于构建 MaxKB Docker 镜像（包含 MinerU 集成和本地 LibreOffice）

set -e

echo "=========================================="
echo "MaxKB Docker 镜像构建脚本"
echo "=========================================="

# 设置构建参数
DOCKER_IMAGE_TAG=${DOCKER_IMAGE_TAG:-"dev"}
BUILD_AT=$(date -u +"%Y-%m-%d %H:%M:%S UTC")
GITHUB_COMMIT=$(git rev-parse --short HEAD 2>/dev/null || echo "unknown")
IMAGE_NAME="maxkb-local"
IMAGE_TAG="latest"

echo ""
echo "构建参数："
echo "  - 镜像名称: ${IMAGE_NAME}:${IMAGE_TAG}"
echo "  - Docker标签: ${DOCKER_IMAGE_TAG}"
echo "  - 构建时间: ${BUILD_AT}"
echo "  - Git提交: ${GITHUB_COMMIT}"
echo ""

# 检查 LibreOffice 资源文件
echo "检查 LibreOffice 资源文件..."
if [ ! -f "resources/LibreOffice_25.2.3_Linux_x86-64_deb.tar.gz_00" ]; then
    echo "错误: 缺少 LibreOffice 资源文件"
    echo "请确保 resources/ 目录下有以下文件:"
    echo "  - LibreOffice_25.2.3_Linux_x86-64_deb.tar.gz_00"
    echo "  - LibreOffice_25.2.3_Linux_x86-64_deb.tar.gz_01"
    echo "  - LibreOffice_25.2.3_Linux_x86-64_deb.tar.gz_02"
    exit 1
fi
echo "✓ LibreOffice 资源文件存在"

# 停止并删除旧容器（如果存在）
echo ""
echo "清理旧容器..."
if docker ps -a | grep -q maxkb-local; then
    echo "停止并删除旧的 maxkb-local 容器..."
    docker stop maxkb-local 2>/dev/null || true
    docker rm maxkb-local 2>/dev/null || true
fi

# 构建 Docker 镜像
echo ""
echo "开始构建 Docker 镜像..."
echo "这可能需要几分钟时间，请耐心等待..."
echo ""

DOCKER_BUILDKIT=1 docker build \
    --build-arg DOCKER_IMAGE_TAG="${DOCKER_IMAGE_TAG}" \
    --build-arg BUILD_AT="${BUILD_AT}" \
    --build-arg GITHUB_COMMIT="${GITHUB_COMMIT}" \
    --progress=plain \
    --memory=16g \
    -t ${IMAGE_NAME}:${IMAGE_TAG} \
    -f installer/Dockerfile \
    .

# 检查构建结果
if [ $? -eq 0 ]; then
    echo ""
    echo "=========================================="
    echo "✓ Docker 镜像构建成功！"
    echo "=========================================="
    echo ""
    echo "镜像信息："
    docker images | grep ${IMAGE_NAME}
    echo ""
    echo "您可以使用以下命令运行容器："
    echo ""
    echo "  docker run -d \\"
    echo "    --name maxkb-local \\"
    echo "    -p 8080:8080 \\"
    echo "    -v maxkb-data:/opt/maxkb \\"
    echo "    ${IMAGE_NAME}:${IMAGE_TAG}"
    echo ""
    echo "或使用 docker-compose："
    echo "  cd dev && docker-compose up -d"
    echo ""
else
    echo ""
    echo "=========================================="
    echo "✗ Docker 镜像构建失败"
    echo "=========================================="
    echo "请检查错误信息并重试"
    exit 1
fi

# 清理悬空镜像（可选）
echo "是否清理悬空镜像？(y/n)"
read -r CLEANUP
if [ "$CLEANUP" = "y" ] || [ "$CLEANUP" = "Y" ]; then
    echo "清理悬空镜像..."
    docker image prune -f
    echo "✓ 清理完成"
fi

echo ""
echo "构建脚本执行完成！"