aigen

Dockerfile

@@ -0,0 +1,111 @@
+# 基础构建阶段：用于构建和安装依赖
+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"]

README.md

@@ -1,10 +0,0 @@
----
-title: N8n Db
-emoji: 🐠
-colorFrom: gray
-colorTo: red
-sdk: docker
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

import-db.sh

@@ -0,0 +1,42 @@
+#!/bin/bash
+
+# 检查是否需要导入数据库
+if [ "$DB_IMPORT" = "yes" ]; then
+    echo "Starting database import..."
+
+    # 读取下载 URL 和解压密码（如果有）
+    DUMP_URL=${DUMP_URL:-""}          # 数据库 dump 文件的下载 URL
+    DUMP_PASSWORD=${DUMP_PASSWORD:-""} # 如果 dump 文件有密码保护
+
+    # 检查 URL 是否为空
+    if [ -z "$DUMP_URL" ]; then
+        echo "DUMP_URL is not set. Skipping database import."
+        # 不退出，仅跳过导入步骤
+        exit 0
+    fi
+
+    # 下载数据库 dump 文件
+    echo "Downloading dump file from $DUMP_URL..."
+    wget -O /tmp/dump.sql "$DUMP_URL" || { echo "Failed to download dump file."; exit 1; }
+
+    # 检查文件是否有密码保护，并使用 pg_restore 或 psql 进行导入
+    if [ -n "$DUMP_PASSWORD" ]; then
+        echo "Dump file is protected with a password."
+
+        # 使用 pg_restore 导入
+        PGPASSWORD="$DUMP_PASSWORD" pg_restore -U $DB_POSTGRESDB_USER -d $DB_POSTGRESDB_DATABASE /tmp/dump.sql || {
+            echo "Failed to import dump file with password."; exit 1;
+        }
+    else
+        echo "Dump file is not password protected."
+
+        # 使用 psql 直接导入
+        psql -U $DB_POSTGRESDB_USER -d $DB_POSTGRESDB_DATABASE -f /tmp/dump.sql || {
+            echo "Failed to import dump file."; exit 1;
+        }
+    fi
+
+    echo "Database import completed successfully."
+else
+    echo "Skipping database import."
+fi

run.sh

@@ -0,0 +1,27 @@
+#!/bin/bash
+
+# 确保以 postgres 用户身份运行
+if [ "$(id -u)" -ne 1000 ]; then
+  echo "切换到 UID 为 1000 的 postgres 用户运行"
+  exec gosu postgres "$0" "$@"
+fi
+
+# 启动 PostgreSQL 服务
+docker-entrypoint.sh postgres &
+
+# 检查 PostgreSQL 服务是否已启动
+echo "等待 PostgreSQL 服务启动..."
+until pg_isready -h localhost; do
+  sleep 5
+done
+echo "PostgreSQL 服务已启动!"
+
+# export N8N_USER_FOLDER=${DATA_DIR}
+export N8N_ENCRYPTION_KEY="n8n8n8n"
+# Allows usage of all builtin modules
+export NODE_FUNCTION_ALLOW_BUILTIN=*
+# Allow usage of external npm modules.
+export NODE_FUNCTION_ALLOW_EXTERNAL=*
+
+# 启动 n8n
+exec n8n