# 基础构建阶段：用于构建和安装依赖
FROM python:3.10-slim AS builder

# 设置构建时变量
ARG requirements
ARG NODEJS_VER=20
ARG PACKAGES=n8n
ARG WORKDIR=/app
ARG DUMP_URL
ARG DUMP_PASSWORD

# 安装 Node.js、n8n 以及必要的系统工具
RUN apt-get update && apt-get install -y curl gnupg build-essential && \
    curl -fsSL https://deb.nodesource.com/setup_${NODEJS_VER}.x | bash - && \
    apt-get install -y nodejs && \
    npm install -g ${PACKAGES} && \
    apt-get clean && rm -rf /var/lib/apt/lists/*

# 设置 Python 虚拟环境
ENV VIRTUAL_ENV=/opt/venv
RUN python3 -m venv $VIRTUAL_ENV && \
    $VIRTUAL_ENV/bin/pip install --upgrade pip && \
    $VIRTUAL_ENV/bin/pip install ${requirements:-requests}

# 主运行阶段：基于较小的 PostgreSQL 镜像
FROM postgres:latest

# 设置环境变量
ARG POSTGRES_USER=n8n
ARG POSTGRES_PASSWORD=n8n
ARG POSTGRES_DB=n8n
ARG WEBHOOK_URL=https://aigenai-db.hf.space/
ARG WORKDIR=/app
ARG DB_IMPORT=no  # 新增 ARG 用于控制数据库导入

# 设置 n8n 环境变量
ENV N8N_HOST=0.0.0.0 \
    N8N_PORT=7860 \
    N8N_PROTOCOL=https \
    WEBHOOK_URL=${WEBHOOK_URL} \
    GENERIC_TIMEZONE=Asia/Shanghai \
    N8N_METRICS=true \
    QUEUE_HEALTH_CHECK_ACTIVE=true \
    N8N_PAYLOAD_SIZE_MAX=256

# 设置数据库连接相关的环境变量
ENV DB_TYPE=postgresdb \
    DB_POSTGRESDB_HOST=localhost \
    DB_POSTGRESDB_PORT=5432 \
    DB_POSTGRESDB_USER=${POSTGRES_USER} \
    DB_POSTGRESDB_PASSWORD=${POSTGRES_PASSWORD} \
    DB_POSTGRESDB_DATABASE=${POSTGRES_DB}

# 设置 Python 环境变量
ENV VIRTUAL_ENV=/opt/venv \
    PATH="$VIRTUAL_ENV/bin:$PATH"

# 复制构建阶段的 n8n 和 Python 运行环境
COPY --from=builder /usr/local/bin/node /usr/local/bin/node
COPY --from=builder /usr/local/lib/node_modules /usr/local/lib/node_modules
COPY --from=builder $VIRTUAL_ENV $VIRTUAL_ENV

# 安装必要的软件包
RUN apt-get update && apt-get install -y \
    curl unzip gnupg build-essential sudo vim git procps lsof net-tools \
    ca-certificates openssl tzdata python3-venv gosu \
    htop jq wget && \
    ln -fs /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && \
    dpkg-reconfigure --frontend noninteractive tzdata

# 设置工作目录并复制启动脚本
WORKDIR ${WORKDIR}
COPY run.sh ${WORKDIR}/run.sh
RUN chmod +x ${WORKDIR}/run.sh

# 更改现有的 postgres 用户 UID 和 GID 为 1000
USER root
RUN usermod -u 1000 postgres && groupmod -g 1000 postgres && \
    chown -R postgres:postgres /var/lib/postgresql && \
    chown -R postgres:postgres /var/run/postgresql

# 切换到 postgres 用户，初始化数据库并创建角色
USER postgres
RUN initdb -D /var/lib/postgresql/data && \
    pg_ctl start -D /var/lib/postgresql/data && \
    psql --command "CREATE ROLE $POSTGRES_USER WITH LOGIN SUPERUSER PASSWORD '$POSTGRES_PASSWORD';" && \
    createdb -O $POSTGRES_USER $POSTGRES_DB && \
    pg_ctl stop -D /var/lib/postgresql/data

# 复制数据库导入脚本和数据文件
COPY dump.sql /docker-entrypoint-initdb.d/

# 添加条件导入的 shell 脚本
COPY import-db.sh /docker-entrypoint-initdb.d/
RUN chmod +x /docker-entrypoint-initdb.d/import-db.sh

# 脚本 import-db.sh 的内容：
# #!/bin/bash
# if [ "$DB_IMPORT" = "yes" ]; then
#     echo "Importing database dump..."
#     psql -U $POSTGRES_USER -d $POSTGRES_DB -f /docker-entrypoint-initdb.d/dump.sql
# else
#     echo "Skipping database import."
# fi

# 健康检查配置（可选）
HEALTHCHECK --interval=120s --timeout=10s --start-period=10s --retries=3 \
    CMD curl -f http://localhost:7860/HEALTHZ || exit 1

# 启动容器时执行run.sh脚本
CMD ["./run.sh"]