diff --git a/Dockerfile b/Dockerfile index 63c60a39..c636ed37 100644 --- a/Dockerfile +++ b/Dockerfile @@ -10,6 +10,7 @@ LABEL version="1.0.0" RUN apk add --no-cache \ curl \ dumb-init \ + sed \ && rm -rf /var/cache/apk/* # 👤 创建应用用户 diff --git a/docker-entrypoint.sh b/docker-entrypoint.sh index 96748011..b7749b65 100644 --- a/docker-entrypoint.sh +++ b/docker-entrypoint.sh @@ -3,6 +3,84 @@ set -e echo "🚀 Claude Relay Service 启动中..." +# 生成随机字符串的函数 +generate_random_string() { + length=$1 + # 使用 /dev/urandom 生成随机字符串 + tr -dc 'a-zA-Z0-9' < /dev/urandom | head -c $length +} + +# 检查并复制配置文件 +if [ ! -f "/app/config/config.js" ]; then + echo "📋 检测到 config.js 不存在,从模板创建..." + if [ -f "/app/config/config.example.js" ]; then + cp /app/config/config.example.js /app/config/config.js + echo "✅ config.js 已创建" + else + echo "❌ 错误: config.example.js 不存在" + exit 1 + fi +fi + +# 检查并创建 .env 文件 +if [ ! -f "/app/.env" ]; then + echo "📋 检测到 .env 不存在,从模板创建..." + if [ -f "/app/.env.example" ]; then + cp /app/.env.example /app/.env + + # 生成随机的 JWT_SECRET (64字符) + if [ -z "$JWT_SECRET" ]; then + JWT_SECRET=$(generate_random_string 64) + echo "🔑 生成 JWT_SECRET" + fi + + # 生成随机的 ENCRYPTION_KEY (32字符) + if [ -z "$ENCRYPTION_KEY" ]; then + ENCRYPTION_KEY=$(generate_random_string 32) + echo "🔑 生成 ENCRYPTION_KEY" + fi + + # 更新 .env 文件中的密钥 + sed -i "s/JWT_SECRET=.*/JWT_SECRET=${JWT_SECRET}/" /app/.env + sed -i "s/ENCRYPTION_KEY=.*/ENCRYPTION_KEY=${ENCRYPTION_KEY}/" /app/.env + + # 设置 Redis 配置以连接到容器内的 Redis + sed -i "s/REDIS_HOST=.*/REDIS_HOST=redis/" /app/.env + + echo "✅ .env 已创建并配置" + else + echo "❌ 错误: .env.example 不存在" + exit 1 + fi +else + echo "✅ 检测到已有 .env 文件" + + # 确保环境变量中有必要的密钥 + if [ -z "$JWT_SECRET" ]; then + # 从 .env 文件读取 + JWT_SECRET=$(grep "^JWT_SECRET=" /app/.env | cut -d'=' -f2) + if [ -z "$JWT_SECRET" ] || [ "$JWT_SECRET" = "your-jwt-secret-here" ]; then + JWT_SECRET=$(generate_random_string 64) + sed -i "s/JWT_SECRET=.*/JWT_SECRET=${JWT_SECRET}/" /app/.env + echo "🔑 更新 JWT_SECRET" + fi + fi + + if [ -z "$ENCRYPTION_KEY" ]; then + # 从 .env 文件读取 + ENCRYPTION_KEY=$(grep "^ENCRYPTION_KEY=" /app/.env | cut -d'=' -f2) + if [ -z "$ENCRYPTION_KEY" ] || [ "$ENCRYPTION_KEY" = "your-encryption-key-here" ]; then + ENCRYPTION_KEY=$(generate_random_string 32) + sed -i "s/ENCRYPTION_KEY=.*/ENCRYPTION_KEY=${ENCRYPTION_KEY}/" /app/.env + echo "🔑 更新 ENCRYPTION_KEY" + fi + fi +fi + +# 导出环境变量 +export JWT_SECRET +export ENCRYPTION_KEY + # 检查是否需要初始化 if [ ! -f "/app/data/init.json" ]; then echo "📋 首次启动,执行初始化设置..."