mirror of
https://github.com/Wei-Shaw/claude-relay-service.git
synced 2026-01-22 16:43:35 +00:00
first commit
This commit is contained in:
shaw
55
.env.example
Normal file
55
.env.example
Normal file
@@ -0,0 +1,55 @@
|
|||||||
|
# 🚀 Claude Relay Service Configuration
|
||||||
|
|
||||||
|
# 🌐 服务器配置
|
||||||
|
PORT=3000
|
||||||
|
HOST=0.0.0.0
|
||||||
|
NODE_ENV=production
|
||||||
|
|
||||||
|
# 🔐 安全配置
|
||||||
|
JWT_SECRET=your-jwt-secret-here
|
||||||
|
ADMIN_SESSION_TIMEOUT=86400000
|
||||||
|
API_KEY_PREFIX=cr_
|
||||||
|
ENCRYPTION_KEY=your-encryption-key-here
|
||||||
|
|
||||||
|
# 📊 Redis 配置
|
||||||
|
REDIS_HOST=localhost
|
||||||
|
REDIS_PORT=6379
|
||||||
|
REDIS_PASSWORD=
|
||||||
|
REDIS_DB=0
|
||||||
|
|
||||||
|
# 🎯 Claude API 配置
|
||||||
|
CLAUDE_API_URL=https://api.anthropic.com/v1/messages
|
||||||
|
CLAUDE_API_VERSION=2023-06-01
|
||||||
|
CLAUDE_BETA_HEADER=claude-code-20250219,oauth-2025-04-20,interleaved-thinking-2025-05-14,fine-grained-tool-streaming-2025-05-14
|
||||||
|
|
||||||
|
# 🌐 代理配置
|
||||||
|
DEFAULT_PROXY_TIMEOUT=30000
|
||||||
|
MAX_PROXY_RETRIES=3
|
||||||
|
|
||||||
|
# 📈 使用限制
|
||||||
|
DEFAULT_TOKEN_LIMIT=1000000
|
||||||
|
DEFAULT_REQUEST_LIMIT=1000
|
||||||
|
|
||||||
|
# 🚦 速率限制
|
||||||
|
RATE_LIMIT_WINDOW=60000
|
||||||
|
RATE_LIMIT_MAX_REQUESTS=100
|
||||||
|
|
||||||
|
# 📝 日志配置
|
||||||
|
LOG_LEVEL=info
|
||||||
|
LOG_MAX_SIZE=10m
|
||||||
|
LOG_MAX_FILES=5
|
||||||
|
|
||||||
|
# 🔧 系统配置
|
||||||
|
CLEANUP_INTERVAL=3600000
|
||||||
|
TOKEN_USAGE_RETENTION=2592000000
|
||||||
|
HEALTH_CHECK_INTERVAL=60000
|
||||||
|
|
||||||
|
# 🎨 Web 界面配置
|
||||||
|
WEB_TITLE=Claude Relay Service
|
||||||
|
WEB_DESCRIPTION=Multi-account Claude API relay service with beautiful management interface
|
||||||
|
WEB_LOGO_URL=/assets/logo.png
|
||||||
|
|
||||||
|
# 🛠️ 开发配置
|
||||||
|
DEBUG=false
|
||||||
|
ENABLE_CORS=true
|
||||||
|
TRUST_PROXY=true
|
||||||
19
.eslintrc.js
Normal file
19
.eslintrc.js
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
module.exports = {
|
||||||
|
env: {
|
||||||
|
node: true,
|
||||||
|
es2022: true,
|
||||||
|
},
|
||||||
|
extends: [
|
||||||
|
'eslint:recommended',
|
||||||
|
],
|
||||||
|
parserOptions: {
|
||||||
|
ecmaVersion: 'latest',
|
||||||
|
sourceType: 'module',
|
||||||
|
},
|
||||||
|
rules: {
|
||||||
|
'no-unused-vars': ['error', { 'argsIgnorePattern': '^_' }],
|
||||||
|
'no-console': 'off',
|
||||||
|
'quotes': ['error', 'single'],
|
||||||
|
'semi': ['error', 'always'],
|
||||||
|
},
|
||||||
|
};
|
||||||
229
.gitignore
vendored
Normal file
229
.gitignore
vendored
Normal file
@@ -0,0 +1,229 @@
|
|||||||
|
# Dependencies
|
||||||
|
node_modules/
|
||||||
|
npm-debug.log*
|
||||||
|
yarn-debug.log*
|
||||||
|
yarn-error.log*
|
||||||
|
pnpm-debug.log*
|
||||||
|
|
||||||
|
# Environment variables
|
||||||
|
.env
|
||||||
|
.env.*
|
||||||
|
!.env.example
|
||||||
|
|
||||||
|
# Claude specific directories
|
||||||
|
.claude/
|
||||||
|
|
||||||
|
# Data directory (contains sensitive information)
|
||||||
|
data/
|
||||||
|
!data/.gitkeep
|
||||||
|
|
||||||
|
# Logs directory
|
||||||
|
logs/
|
||||||
|
*.log
|
||||||
|
startup.log
|
||||||
|
app.log
|
||||||
|
|
||||||
|
# Configuration files (may contain sensitive data)
|
||||||
|
config/config.js
|
||||||
|
!config/config.example.js
|
||||||
|
|
||||||
|
# Runtime data
|
||||||
|
pids/
|
||||||
|
*.pid
|
||||||
|
*.seed
|
||||||
|
*.pid.lock
|
||||||
|
|
||||||
|
# Coverage directory used by tools like istanbul
|
||||||
|
coverage/
|
||||||
|
*.lcov
|
||||||
|
|
||||||
|
# nyc test coverage
|
||||||
|
.nyc_output
|
||||||
|
|
||||||
|
# Grunt intermediate storage
|
||||||
|
.grunt
|
||||||
|
|
||||||
|
# Bower dependency directory
|
||||||
|
bower_components
|
||||||
|
|
||||||
|
# node-waf configuration
|
||||||
|
.lock-wscript
|
||||||
|
|
||||||
|
# Compiled binary addons
|
||||||
|
build/Release
|
||||||
|
|
||||||
|
# Dependency directories
|
||||||
|
jspm_packages/
|
||||||
|
|
||||||
|
# TypeScript cache
|
||||||
|
*.tsbuildinfo
|
||||||
|
|
||||||
|
# Optional npm cache directory
|
||||||
|
.npm
|
||||||
|
|
||||||
|
# Optional eslint cache
|
||||||
|
.eslintcache
|
||||||
|
|
||||||
|
# Optional stylelint cache
|
||||||
|
.stylelintcache
|
||||||
|
|
||||||
|
# Microbundle cache
|
||||||
|
.rpt2_cache/
|
||||||
|
.rts2_cache_cjs/
|
||||||
|
.rts2_cache_es/
|
||||||
|
.rts2_cache_umd/
|
||||||
|
|
||||||
|
# Optional REPL history
|
||||||
|
.node_repl_history
|
||||||
|
|
||||||
|
# Output of 'npm pack'
|
||||||
|
*.tgz
|
||||||
|
|
||||||
|
# Yarn Integrity file
|
||||||
|
.yarn-integrity
|
||||||
|
|
||||||
|
# parcel-bundler cache
|
||||||
|
.cache
|
||||||
|
.parcel-cache
|
||||||
|
|
||||||
|
# Next.js build output
|
||||||
|
.next
|
||||||
|
|
||||||
|
# Nuxt.js build / generate output
|
||||||
|
.nuxt
|
||||||
|
dist
|
||||||
|
|
||||||
|
# Gatsby files
|
||||||
|
.cache/
|
||||||
|
public
|
||||||
|
|
||||||
|
# Vuepress build output
|
||||||
|
.vuepress/dist
|
||||||
|
|
||||||
|
# Serverless directories
|
||||||
|
.serverless/
|
||||||
|
|
||||||
|
# FuseBox cache
|
||||||
|
.fusebox/
|
||||||
|
|
||||||
|
# DynamoDB Local files
|
||||||
|
.dynamodb/
|
||||||
|
|
||||||
|
# TernJS port file
|
||||||
|
.tern-port
|
||||||
|
|
||||||
|
# Stores VSCode versions used for testing VSCode extensions
|
||||||
|
.vscode-test
|
||||||
|
|
||||||
|
# Temporary folders
|
||||||
|
tmp/
|
||||||
|
temp/
|
||||||
|
.tmp/
|
||||||
|
.temp/
|
||||||
|
|
||||||
|
# OS generated files
|
||||||
|
.DS_Store
|
||||||
|
.DS_Store?
|
||||||
|
._*
|
||||||
|
.Spotlight-V100
|
||||||
|
.Trashes
|
||||||
|
ehthumbs.db
|
||||||
|
Thumbs.db
|
||||||
|
desktop.ini
|
||||||
|
|
||||||
|
# IDE files
|
||||||
|
.vscode/
|
||||||
|
.idea/
|
||||||
|
*.swp
|
||||||
|
*.swo
|
||||||
|
*~
|
||||||
|
|
||||||
|
# Backup files
|
||||||
|
*.bak
|
||||||
|
*.backup
|
||||||
|
*~
|
||||||
|
|
||||||
|
# Archive files (unless specifically needed)
|
||||||
|
*.7z
|
||||||
|
*.dmg
|
||||||
|
*.gz
|
||||||
|
*.iso
|
||||||
|
*.jar
|
||||||
|
*.rar
|
||||||
|
*.tar
|
||||||
|
*.zip
|
||||||
|
|
||||||
|
# Application specific files
|
||||||
|
# JWT secrets and encryption keys
|
||||||
|
secrets/
|
||||||
|
keys/
|
||||||
|
certs/
|
||||||
|
|
||||||
|
# Database dumps
|
||||||
|
*.sql
|
||||||
|
*.db
|
||||||
|
*.sqlite
|
||||||
|
*.sqlite3
|
||||||
|
|
||||||
|
# Redis dumps
|
||||||
|
dump.rdb
|
||||||
|
appendonly.aof
|
||||||
|
|
||||||
|
# PM2 files
|
||||||
|
ecosystem.config.js
|
||||||
|
.pm2/
|
||||||
|
|
||||||
|
# Docker files (keep main ones, ignore volumes)
|
||||||
|
.docker/
|
||||||
|
docker-volumes/
|
||||||
|
|
||||||
|
# Monitoring data
|
||||||
|
prometheus/
|
||||||
|
grafana/
|
||||||
|
|
||||||
|
# Test files and coverage
|
||||||
|
test-results/
|
||||||
|
coverage/
|
||||||
|
.nyc_output/
|
||||||
|
|
||||||
|
# Documentation build
|
||||||
|
docs/build/
|
||||||
|
docs/dist/
|
||||||
|
|
||||||
|
# Deployment files
|
||||||
|
deploy/
|
||||||
|
.deploy/
|
||||||
|
|
||||||
|
# Package lock files (choose one)
|
||||||
|
# Uncomment the one you DON'T want to track
|
||||||
|
# package-lock.json
|
||||||
|
# yarn.lock
|
||||||
|
# pnpm-lock.yaml
|
||||||
|
|
||||||
|
# Local development files
|
||||||
|
.local/
|
||||||
|
local/
|
||||||
|
|
||||||
|
# Debug files
|
||||||
|
debug.log
|
||||||
|
error.log
|
||||||
|
access.log
|
||||||
|
|
||||||
|
# Session files
|
||||||
|
sessions/
|
||||||
|
|
||||||
|
# Upload directories
|
||||||
|
uploads/
|
||||||
|
files/
|
||||||
|
|
||||||
|
# Cache directories
|
||||||
|
.cache/
|
||||||
|
cache/
|
||||||
|
|
||||||
|
# Build artifacts
|
||||||
|
build/
|
||||||
|
dist/
|
||||||
|
out/
|
||||||
|
|
||||||
|
# Runtime files
|
||||||
|
*.sock
|
||||||
181
CLAUDE.md
Normal file
181
CLAUDE.md
Normal file
@@ -0,0 +1,181 @@
|
|||||||
|
# CLAUDE.md
|
||||||
|
|
||||||
|
This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
|
||||||
|
|
||||||
|
这个文件为 Claude Code (claude.ai/code) 提供在此代码库中工作的指导。
|
||||||
|
|
||||||
|
## 项目概述
|
||||||
|
|
||||||
|
Claude Relay Service 是一个功能完整的 Claude API 中转服务，支持多账户管理、API Key 认证、代理配置和现代化 Web 管理界面。该服务作为客户端（如 SillyTavern）与 Anthropic API 之间的中间件，提供认证、限流、监控等功能。
|
||||||
|
|
||||||
|
## 核心架构
|
||||||
|
|
||||||
|
### 关键架构概念
|
||||||
|
- **代理认证流**: 客户端用自建API Key → 验证 → 获取Claude账户OAuth token → 转发到Anthropic
|
||||||
|
- **Token管理**: 自动监控OAuth token过期并刷新，支持10秒提前刷新策略
|
||||||
|
- **代理支持**: 每个Claude账户支持独立代理配置，OAuth token交换也通过代理进行
|
||||||
|
- **数据加密**: 敏感数据（refreshToken, accessToken）使用AES加密存储在Redis
|
||||||
|
|
||||||
|
### 主要服务组件
|
||||||
|
- **claudeRelayService.js**: 核心代理服务，处理请求转发和流式响应
|
||||||
|
- **claudeAccountService.js**: Claude账户管理，OAuth token刷新和账户选择
|
||||||
|
- **apiKeyService.js**: API Key管理，验证、限流和使用统计
|
||||||
|
- **oauthHelper.js**: OAuth工具，PKCE流程实现和代理支持
|
||||||
|
|
||||||
|
### 认证和代理流程
|
||||||
|
1. 客户端使用自建API Key（cr_前缀格式）发送请求
|
||||||
|
2. authenticateApiKey中间件验证API Key有效性和速率限制
|
||||||
|
3. claudeAccountService自动选择可用Claude账户
|
||||||
|
4. 检查OAuth access token有效性，过期则自动刷新（使用代理）
|
||||||
|
5. 移除客户端API Key，使用OAuth Bearer token转发请求
|
||||||
|
6. 通过账户配置的代理发送到Anthropic API
|
||||||
|
7. 流式或非流式返回响应，记录使用统计
|
||||||
|
|
||||||
|
### OAuth集成
|
||||||
|
- **PKCE流程**: 完整的OAuth 2.0 PKCE实现，支持代理
|
||||||
|
- **自动刷新**: 智能token过期检测和自动刷新机制
|
||||||
|
- **代理支持**: OAuth授权和token交换全程支持代理配置
|
||||||
|
- **安全存储**: claudeAiOauth数据加密存储，包含accessToken、refreshToken、scopes
|
||||||
|
|
||||||
|
## 常用命令
|
||||||
|
|
||||||
|
### 基本开发命令
|
||||||
|
```bash
|
||||||
|
# 安装依赖和初始化
|
||||||
|
npm install
|
||||||
|
npm run setup # 生成配置和管理员凭据
|
||||||
|
npm run install:web # 安装Web界面依赖
|
||||||
|
|
||||||
|
# 开发和运行
|
||||||
|
npm run dev # 开发模式（热重载）
|
||||||
|
npm start # 生产模式
|
||||||
|
npm test # 运行测试
|
||||||
|
npm run lint # 代码检查
|
||||||
|
|
||||||
|
# Docker部署
|
||||||
|
docker-compose up -d # 推荐方式
|
||||||
|
docker-compose --profile monitoring up -d # 包含监控
|
||||||
|
|
||||||
|
# 服务管理
|
||||||
|
npm run service:start:daemon # 后台启动（推荐）
|
||||||
|
npm run service:status # 查看服务状态
|
||||||
|
npm run service:logs # 查看日志
|
||||||
|
npm run service:stop # 停止服务
|
||||||
|
|
||||||
|
# CLI管理工具
|
||||||
|
npm run cli admin # 管理员操作
|
||||||
|
npm run cli keys # API Key管理
|
||||||
|
npm run cli accounts # Claude账户管理
|
||||||
|
npm run cli status # 系统状态
|
||||||
|
```
|
||||||
|
|
||||||
|
### 开发环境配置
|
||||||
|
必须配置的环境变量：
|
||||||
|
- `JWT_SECRET`: JWT密钥（32字符以上随机字符串）
|
||||||
|
- `ENCRYPTION_KEY`: 数据加密密钥（32字符固定长度）
|
||||||
|
- `REDIS_HOST`: Redis主机地址（默认localhost）
|
||||||
|
- `REDIS_PORT`: Redis端口（默认6379）
|
||||||
|
- `REDIS_PASSWORD`: Redis密码（可选）
|
||||||
|
|
||||||
|
初始化命令：
|
||||||
|
```bash
|
||||||
|
cp config/config.example.js config/config.js
|
||||||
|
cp .env.example .env
|
||||||
|
npm run setup # 自动生成密钥并创建管理员账户
|
||||||
|
```
|
||||||
|
|
||||||
|
## Web界面功能
|
||||||
|
|
||||||
|
### OAuth账户添加流程
|
||||||
|
1. **基本信息和代理设置**: 配置账户名称、描述和代理参数
|
||||||
|
2. **OAuth授权**:
|
||||||
|
- 生成授权URL → 用户打开链接并登录Claude Code账号
|
||||||
|
- 授权后会显示Authorization Code → 复制并粘贴到输入框
|
||||||
|
- 系统自动交换token并创建账户
|
||||||
|
|
||||||
|
### 核心管理功能
|
||||||
|
- **实时仪表板**: 系统统计、账户状态、使用量监控
|
||||||
|
- **API Key管理**: 创建、配额设置、使用统计查看
|
||||||
|
- **Claude账户管理**: OAuth账户添加、代理配置、状态监控
|
||||||
|
- **系统日志**: 实时日志查看，多级别过滤
|
||||||
|
|
||||||
|
## 重要端点
|
||||||
|
|
||||||
|
### API转发端点
|
||||||
|
- `POST /api/v1/messages` - 主要消息处理端点（支持流式）
|
||||||
|
- `GET /api/v1/models` - 模型列表（兼容性）
|
||||||
|
- `GET /api/v1/usage` - 使用统计查询
|
||||||
|
- `GET /api/v1/key-info` - API Key信息
|
||||||
|
|
||||||
|
### OAuth管理端点
|
||||||
|
- `POST /admin/claude-accounts/generate-auth-url` - 生成OAuth授权URL（含代理）
|
||||||
|
- `POST /admin/claude-accounts/exchange-code` - 交换authorization code
|
||||||
|
- `POST /admin/claude-accounts` - 创建OAuth账户
|
||||||
|
|
||||||
|
### 系统端点
|
||||||
|
- `GET /health` - 健康检查
|
||||||
|
- `GET /web` - Web管理界面
|
||||||
|
- `GET /admin/dashboard` - 系统概览数据
|
||||||
|
|
||||||
|
## 故障排除
|
||||||
|
|
||||||
|
### OAuth相关问题
|
||||||
|
1. **代理配置错误**: 检查代理设置是否正确，OAuth token交换也需要代理
|
||||||
|
2. **授权码无效**: 确保复制了完整的Authorization Code，没有遗漏字符
|
||||||
|
3. **Token刷新失败**: 检查refreshToken有效性和代理配置
|
||||||
|
|
||||||
|
### 常见开发问题
|
||||||
|
1. **Redis连接失败**: 确认Redis服务运行，检查连接配置
|
||||||
|
2. **管理员登录失败**: 检查init.json同步到Redis，运行npm run setup
|
||||||
|
3. **API Key格式错误**: 确保使用cr_前缀格式
|
||||||
|
4. **代理连接问题**: 验证SOCKS5/HTTP代理配置和认证信息
|
||||||
|
|
||||||
|
### 调试工具
|
||||||
|
- **日志系统**: Winston结构化日志，支持不同级别
|
||||||
|
- **CLI工具**: 命令行状态查看和管理
|
||||||
|
- **Web界面**: 实时日志查看和系统监控
|
||||||
|
- **健康检查**: /health端点提供系统状态
|
||||||
|
|
||||||
|
## 开发最佳实践
|
||||||
|
|
||||||
|
### 代码修改原则
|
||||||
|
- 对现有文件进行修改时，首先检查代码库的现有模式和风格
|
||||||
|
- 尽可能重用现有的服务和工具函数，避免重复代码
|
||||||
|
- 遵循项目现有的错误处理和日志记录模式
|
||||||
|
- 敏感数据必须使用加密存储（参考 claudeAccountService.js 中的加密实现）
|
||||||
|
|
||||||
|
### 测试和质量保证
|
||||||
|
- 运行 `npm run lint` 进行代码风格检查（使用 ESLint）
|
||||||
|
- 运行 `npm test` 执行测试套件（Jest + SuperTest 配置）
|
||||||
|
- 在修改核心服务后，使用 CLI 工具验证功能：`npm run cli status`
|
||||||
|
- 检查日志文件 `logs/claude-relay-*.log` 确认服务正常运行
|
||||||
|
- 注意：当前项目缺少实际测试文件，建议补充单元测试和集成测试
|
||||||
|
|
||||||
|
### 常见文件位置
|
||||||
|
- 核心服务逻辑：`src/services/` 目录
|
||||||
|
- 路由处理：`src/routes/` 目录
|
||||||
|
- 中间件：`src/middleware/` 目录
|
||||||
|
- 配置管理：`config/config.js`
|
||||||
|
- Redis 模型：`src/models/redis.js`
|
||||||
|
- 工具函数：`src/utils/` 目录
|
||||||
|
|
||||||
|
### 重要架构决策
|
||||||
|
- 所有敏感数据（OAuth token、refreshToken）都使用 AES 加密存储在 Redis
|
||||||
|
- 每个 Claude 账户支持独立的代理配置，包括 SOCKS5 和 HTTP 代理
|
||||||
|
- API Key 使用哈希存储，支持 `cr_` 前缀格式
|
||||||
|
- 请求流程：API Key 验证 → 账户选择 → Token 刷新（如需）→ 请求转发
|
||||||
|
- 支持流式和非流式响应，客户端断开时自动清理资源
|
||||||
|
|
||||||
|
### 核心数据流和性能优化
|
||||||
|
- **哈希映射优化**: API Key 验证从 O(n) 优化到 O(1) 查找
|
||||||
|
- **智能 Usage 捕获**: 从 SSE 流中解析真实的 token 使用数据
|
||||||
|
- **多维度统计**: 支持按时间、模型、用户的实时使用统计
|
||||||
|
- **异步处理**: 非阻塞的统计记录和日志写入
|
||||||
|
- **原子操作**: Redis 管道操作确保数据一致性
|
||||||
|
|
||||||
|
### 安全和容错机制
|
||||||
|
- **多层加密**: API Key 哈希 + OAuth Token AES 加密
|
||||||
|
- **零信任验证**: 每个请求都需要完整的认证链
|
||||||
|
- **优雅降级**: Redis 连接失败时的回退机制
|
||||||
|
- **自动重试**: 指数退避重试策略和错误隔离
|
||||||
|
- **资源清理**: 客户端断开时的自动清理机制
|
||||||
48
Dockerfile
Normal file
48
Dockerfile
Normal file
@@ -0,0 +1,48 @@
|
|||||||
|
# 🐳 使用官方 Node.js 18 Alpine 镜像
|
||||||
|
FROM node:18-alpine
|
||||||
|
|
||||||
|
# 📋 设置标签
|
||||||
|
LABEL maintainer="claude-relay-service@example.com"
|
||||||
|
LABEL description="Claude Code API Relay Service"
|
||||||
|
LABEL version="1.0.0"
|
||||||
|
|
||||||
|
# 🔧 安装系统依赖
|
||||||
|
RUN apk add --no-cache \
|
||||||
|
curl \
|
||||||
|
dumb-init \
|
||||||
|
&& rm -rf /var/cache/apk/*
|
||||||
|
|
||||||
|
# 👤 创建应用用户
|
||||||
|
RUN addgroup -g 1001 -S nodejs && \
|
||||||
|
adduser -S claude -u 1001 -G nodejs
|
||||||
|
|
||||||
|
# 📁 设置工作目录
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
# 📦 复制 package 文件
|
||||||
|
COPY package*.json ./
|
||||||
|
|
||||||
|
# 🔽 安装依赖 (生产环境)
|
||||||
|
RUN npm ci --only=production && \
|
||||||
|
npm cache clean --force
|
||||||
|
|
||||||
|
# 📋 复制应用代码
|
||||||
|
COPY --chown=claude:nodejs . .
|
||||||
|
|
||||||
|
# 📁 创建必要目录
|
||||||
|
RUN mkdir -p logs data temp && \
|
||||||
|
chown -R claude:nodejs logs data temp
|
||||||
|
|
||||||
|
# 🔐 切换到非 root 用户
|
||||||
|
USER claude
|
||||||
|
|
||||||
|
# 🌐 暴露端口
|
||||||
|
EXPOSE 3000
|
||||||
|
|
||||||
|
# 🏥 健康检查
|
||||||
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
|
||||||
|
CMD curl -f http://localhost:3000/health || exit 1
|
||||||
|
|
||||||
|
# 🚀 启动应用
|
||||||
|
ENTRYPOINT ["dumb-init", "--"]
|
||||||
|
CMD ["node", "src/app.js"]
|
||||||
21
LICENSE
21
LICENSE
@@ -1,21 +0,0 @@
|
|||||||
MIT License
|
|
||||||
|
|
||||||
Copyright (c) 2025 Wesley Liddick
|
|
||||||
|
|
||||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
||||||
of this software and associated documentation files (the "Software"), to deal
|
|
||||||
in the Software without restriction, including without limitation the rights
|
|
||||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
||||||
copies of the Software, and to permit persons to whom the Software is
|
|
||||||
furnished to do so, subject to the following conditions:
|
|
||||||
|
|
||||||
The above copyright notice and this permission notice shall be included in all
|
|
||||||
copies or substantial portions of the Software.
|
|
||||||
|
|
||||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
||||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
||||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
||||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
||||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
||||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
||||||
SOFTWARE.
|
|
||||||
407
README.md
Normal file
407
README.md
Normal file
@@ -0,0 +1,407 @@
|
|||||||
|
# Claude Relay Service
|
||||||
|
|
||||||
|
<div align="center">
|
||||||
|
|
||||||
|
[](https://opensource.org/licenses/MIT)
|
||||||
|
[](https://nodejs.org/)
|
||||||
|
[](https://redis.io/)
|
||||||
|
[](https://www.docker.com/)
|
||||||
|
|
||||||
|
**🔐 自己搭建的Claude API中转服务，支持多账户管理**
|
||||||
|
|
||||||
|
[English](#english) • [中文文档](#中文文档)
|
||||||
|
|
||||||
|
</div>
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## ⚠️ 重要提醒
|
||||||
|
|
||||||
|
**使用本项目前请仔细阅读：**
|
||||||
|
|
||||||
|
🚨 **服务条款风险**: 使用本项目可能违反Anthropic的服务条款。请在使用前仔细阅读Anthropic的用户协议，使用本项目的一切风险由用户自行承担。
|
||||||
|
|
||||||
|
📖 **免责声明**: 本项目仅供技术学习和研究使用，作者不对因使用本项目导致的账户封禁、服务中断或其他损失承担任何责任。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🤔 这个项目适合你吗？
|
||||||
|
|
||||||
|
- 🌍 **地区限制**: 所在地区无法直接访问Claude Code服务？
|
||||||
|
- 🔒 **隐私担忧**: 担心第三方镜像服务会记录或泄露你的对话内容？
|
||||||
|
- 👥 **成本分摊**: 想和朋友一起分摊Claude Code Max订阅费用？
|
||||||
|
- ⚡ **稳定性**: 第三方镜像站经常故障不稳定，影响效率 ？
|
||||||
|
|
||||||
|
如果有以上困惑，那这个项目可能适合你。
|
||||||
|
|
||||||
|
### 适合的场景
|
||||||
|
|
||||||
|
✅ **找朋友拼车**: 三五好友一起分摊Claude Code Max订阅，Opus爽用
|
||||||
|
✅ **隐私敏感**: 不想让第三方镜像看到你的对话内容
|
||||||
|
✅ **技术折腾**: 有基本的技术基础，愿意自己搭建和维护
|
||||||
|
✅ **稳定需求**: 需要长期稳定的Claude访问，不想受制于镜像站
|
||||||
|
✅ **地区受限**: 无法直接访问Claude官方服务
|
||||||
|
|
||||||
|
### 不适合的场景
|
||||||
|
|
||||||
|
❌ **纯小白**: 完全不懂技术，连服务器都不会买
|
||||||
|
❌ **偶尔使用**: 一个月用不了几次，没必要折腾
|
||||||
|
❌ **注册问题**: 无法自行注册Claude账号
|
||||||
|
❌ **支付问题**: 没有支付渠道订阅Claude Code
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 💭 为什么要自己搭？
|
||||||
|
|
||||||
|
|
||||||
|
### 现有镜像站可能的问题
|
||||||
|
|
||||||
|
- 🕵️ **隐私风险**: 你的对话内容都被人家看得一清二楚，商业机密什么的就别想了
|
||||||
|
- 🐌 **性能不稳**: 用的人多了就慢，高峰期经常卡死
|
||||||
|
- 💰 **价格不透明**: 不知道实际成本
|
||||||
|
|
||||||
|
### 自建的好处
|
||||||
|
|
||||||
|
- 🔐 **数据安全**: 所有接口请求都只经过你自己的服务器，直连Anthropic API
|
||||||
|
- ⚡ **性能可控**: 就你们几个人用，Max 200刀套餐基本上可以爽用Opus
|
||||||
|
- 💰 **成本透明**: 用了多少token一目了然，按官方价格换算了具体费用
|
||||||
|
- 📊 **监控完整**: 使用情况、成本分析、性能监控全都有
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🚀 核心功能
|
||||||
|
|
||||||
|
### 基础功能
|
||||||
|
- ✅ **多账户管理**: 可以添加多个Claude账户自动轮换
|
||||||
|
- ✅ **自定义API Key**: 给每个人分配独立的Key
|
||||||
|
- ✅ **使用统计**: 详细记录每个人用了多少token
|
||||||
|
|
||||||
|
### 高级功能
|
||||||
|
- 🔄 **智能切换**: 账户出问题自动换下一个
|
||||||
|
- 🚀 **性能优化**: 连接池、缓存，减少延迟
|
||||||
|
- 📊 **监控面板**: Web界面查看所有数据
|
||||||
|
- 🛡️ **安全控制**: 访问限制、速率控制
|
||||||
|
- 🌐 **代理支持**: 支持HTTP/SOCKS5代理
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📋 部署要求
|
||||||
|
|
||||||
|
### 硬件要求（最低配置）
|
||||||
|
- **CPU**: 1核心就够了
|
||||||
|
- **内存**: 512MB（建议1GB）
|
||||||
|
- **硬盘**: 30GB可用空间
|
||||||
|
- **网络**: 能访问到Anthropic API（建议使用US地区的机器）
|
||||||
|
- **建议**: 2核4G的基本够了，网络尽量选回国线路快一点的（为了提高速度，建议不要开代理或者设置服务器的IP直连）
|
||||||
|
|
||||||
|
### 软件要求
|
||||||
|
- **Node.js** 18或更高版本
|
||||||
|
- **Redis** 6或更高版本
|
||||||
|
- **操作系统**: 建议Linux
|
||||||
|
|
||||||
|
### 费用估算
|
||||||
|
- **服务器**: 轻量云服务器，一个月10-30块
|
||||||
|
- **Claude订阅**: 看你怎么分摊了
|
||||||
|
- **其他**: 基本没有了
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🐳 最简单的部署方式（Docker）
|
||||||
|
|
||||||
|
如果你懒得折腾环境，直接用Docker：
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 1. 下载项目
|
||||||
|
git clone https://github.com/yourusername/claude-relay-service.git
|
||||||
|
cd claude-relay-service
|
||||||
|
|
||||||
|
# 2. 一键启动
|
||||||
|
docker-compose up -d
|
||||||
|
|
||||||
|
# 3. 查看是否启动成功
|
||||||
|
docker-compose ps
|
||||||
|
```
|
||||||
|
|
||||||
|
就这么简单，服务就跑起来了。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📦 手动部署（适合折腾党）
|
||||||
|
|
||||||
|
### 第一步：环境准备
|
||||||
|
|
||||||
|
**Ubuntu/Debian用户：**
|
||||||
|
```bash
|
||||||
|
# 安装Node.js
|
||||||
|
curl -fsSL https://deb.nodesource.com/setup_18.x | sudo -E bash -
|
||||||
|
sudo apt-get install -y nodejs
|
||||||
|
|
||||||
|
# 安装Redis
|
||||||
|
sudo apt update
|
||||||
|
sudo apt install redis-server
|
||||||
|
sudo systemctl start redis-server
|
||||||
|
```
|
||||||
|
|
||||||
|
**CentOS/RHEL用户：**
|
||||||
|
```bash
|
||||||
|
# 安装Node.js
|
||||||
|
curl -fsSL https://rpm.nodesource.com/setup_18.x | sudo bash -
|
||||||
|
sudo yum install -y nodejs
|
||||||
|
|
||||||
|
# 安装Redis
|
||||||
|
sudo yum install redis
|
||||||
|
sudo systemctl start redis
|
||||||
|
```
|
||||||
|
|
||||||
|
### 第二步：下载和配置
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 下载项目
|
||||||
|
git clone https://github.com/yourusername/claude-relay-service.git
|
||||||
|
cd claude-relay-service
|
||||||
|
|
||||||
|
# 安装依赖
|
||||||
|
npm install
|
||||||
|
|
||||||
|
# 复制配置文件（重要！）
|
||||||
|
cp config/config.example.js config/config.js
|
||||||
|
cp .env.example .env
|
||||||
|
```
|
||||||
|
|
||||||
|
### 第三步：配置文件设置
|
||||||
|
|
||||||
|
**编辑 `.env` 文件：**
|
||||||
|
```bash
|
||||||
|
# 这两个密钥随便生成，但要记住
|
||||||
|
JWT_SECRET=你的超级秘密密钥
|
||||||
|
ENCRYPTION_KEY=32位的加密密钥随便写
|
||||||
|
|
||||||
|
# Redis配置
|
||||||
|
REDIS_HOST=localhost
|
||||||
|
REDIS_PORT=6379
|
||||||
|
REDIS_PASSWORD=
|
||||||
|
```
|
||||||
|
|
||||||
|
**编辑 `config/config.js` 文件：**
|
||||||
|
```javascript
|
||||||
|
module.exports = {
|
||||||
|
server: {
|
||||||
|
port: 3000, // 服务端口，可以改
|
||||||
|
host: '0.0.0.0' // 不用改
|
||||||
|
},
|
||||||
|
redis: {
|
||||||
|
host: '127.0.0.1', // Redis地址
|
||||||
|
port: 6379 // Redis端口
|
||||||
|
},
|
||||||
|
// 其他配置保持默认就行
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 第四步：启动服务
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 初始化
|
||||||
|
npm run setup
|
||||||
|
|
||||||
|
# 启动服务
|
||||||
|
npm run service:start:daemon # 后台运行（推荐）
|
||||||
|
|
||||||
|
# 查看状态
|
||||||
|
npm run service:status
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🎮 开始使用
|
||||||
|
|
||||||
|
### 1. 打开管理界面
|
||||||
|
|
||||||
|
浏览器访问：`http://你的服务器IP:3000/web`
|
||||||
|
|
||||||
|
默认管理员账号：admin / admin123
|
||||||
|
|
||||||
|
### 2. 添加Claude账户
|
||||||
|
|
||||||
|
这一步比较关键，需要OAuth授权：
|
||||||
|
|
||||||
|
1. 点击「Claude账户」标签
|
||||||
|
2. 如果你在国内，先配置代理（重要！）
|
||||||
|
3. 点击「添加账户」
|
||||||
|
4. 点击「生成授权链接」，会打开一个新页面
|
||||||
|
5. 在新页面完成Claude登录和授权
|
||||||
|
6. 复制返回的Authorization Code
|
||||||
|
7. 粘贴到页面完成添加
|
||||||
|
|
||||||
|
**注意**: 如果你在国内，这一步可能需要科学上网。
|
||||||
|
|
||||||
|
### 3. 创建API Key
|
||||||
|
|
||||||
|
给每个使用者分配一个Key：
|
||||||
|
|
||||||
|
1. 点击「API Keys」标签
|
||||||
|
2. 点击「创建新Key」
|
||||||
|
3. 给Key起个名字，比如「张三的Key」
|
||||||
|
4. 设置使用限制（可选）
|
||||||
|
5. 保存，记下生成的Key
|
||||||
|
|
||||||
|
### 4. 开始使用API
|
||||||
|
|
||||||
|
现在你可以用自己的服务替换官方API了：
|
||||||
|
|
||||||
|
**原来的请求：**
|
||||||
|
```bash
|
||||||
|
curl https://api.anthropic.com/v1/messages \
|
||||||
|
-H "x-api-key: 官方的key" \
|
||||||
|
-H "content-type: application/json" \
|
||||||
|
-d '{"model":"claude-3-sonnet-20240229","messages":[{"role":"user","content":"你好"}]}'
|
||||||
|
```
|
||||||
|
|
||||||
|
**现在的请求：**
|
||||||
|
```bash
|
||||||
|
curl http://你的域名:3000/api/v1/messages \
|
||||||
|
-H "x-api-key: cr_你创建的key" \
|
||||||
|
-H "content-type: application/json" \
|
||||||
|
-d '{"model":"claude-3-sonnet-20240229","messages":[{"role":"user","content":"你好"}]}'
|
||||||
|
```
|
||||||
|
|
||||||
|
就是把域名换一下，API Key换成你自己生成的，其他都一样。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🔧 日常维护
|
||||||
|
|
||||||
|
### 服务管理
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 查看服务状态
|
||||||
|
npm run service:status
|
||||||
|
|
||||||
|
# 查看日志
|
||||||
|
npm run service:logs
|
||||||
|
|
||||||
|
# 重启服务
|
||||||
|
npm run service:restart:daemon
|
||||||
|
|
||||||
|
# 停止服务
|
||||||
|
npm run service:stop
|
||||||
|
```
|
||||||
|
|
||||||
|
### 监控使用情况
|
||||||
|
|
||||||
|
- **Web界面**: `http://你的域名:3000/web` - 查看使用统计
|
||||||
|
- **健康检查**: `http://你的域名:3000/health` - 确认服务正常
|
||||||
|
- **日志文件**: `logs/` 目录下的各种日志文件
|
||||||
|
|
||||||
|
### 常见问题处理
|
||||||
|
|
||||||
|
**Redis连不上？**
|
||||||
|
```bash
|
||||||
|
# 检查Redis是否启动
|
||||||
|
redis-cli ping
|
||||||
|
|
||||||
|
# 应该返回 PONG
|
||||||
|
```
|
||||||
|
|
||||||
|
**OAuth授权失败？**
|
||||||
|
- 检查代理设置是否正确
|
||||||
|
- 确保能正常访问 claude.ai
|
||||||
|
- 清除浏览器缓存重试
|
||||||
|
|
||||||
|
**API请求失败？**
|
||||||
|
- 检查API Key是否正确
|
||||||
|
- 查看日志文件找错误信息
|
||||||
|
- 确认Claude账户状态正常
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🛠️ 高级玩法
|
||||||
|
|
||||||
|
### 设置代理（国内用户必看）
|
||||||
|
|
||||||
|
如果你在国内，需要配置代理才能正常使用：
|
||||||
|
|
||||||
|
```javascript
|
||||||
|
// 在账户配置中添加
|
||||||
|
{
|
||||||
|
"proxy": {
|
||||||
|
"type": "socks5", // 或者 "http"
|
||||||
|
"host": "127.0.0.1",
|
||||||
|
"port": 1080,
|
||||||
|
"username": "用户名", // 如果代理需要认证
|
||||||
|
"password": "密码" // 如果代理需要认证
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 命令行管理工具
|
||||||
|
|
||||||
|
懒得打开网页？用命令行：
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 查看所有API Key
|
||||||
|
npm run cli keys list
|
||||||
|
|
||||||
|
# 创建新Key
|
||||||
|
npm run cli keys create --name "测试Key" --limit 1000
|
||||||
|
|
||||||
|
# 查看账户状态
|
||||||
|
npm run cli accounts list
|
||||||
|
|
||||||
|
# 测试账户连接
|
||||||
|
npm run cli accounts test --id 账户ID
|
||||||
|
```
|
||||||
|
|
||||||
|
### 监控集成
|
||||||
|
|
||||||
|
如果你想要更专业的监控，可以接入Prometheus：
|
||||||
|
|
||||||
|
访问 `http://你的域名(或IP):3000/metrics` 获取指标数据。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 💡 使用建议
|
||||||
|
|
||||||
|
### 账户管理
|
||||||
|
- **多账户**: 建议添加2-3个Claude账户，防止单点故障
|
||||||
|
- **定期检查**: 每周看看账户状态，及时处理异常
|
||||||
|
- **备用方案**: 准备几个备用账户，关键时刻能顶上
|
||||||
|
|
||||||
|
### 成本控制
|
||||||
|
- **设置限额**: 给每个API Key设置合理的使用限制
|
||||||
|
- **监控支出**: 定期查看成本统计，控制预算
|
||||||
|
- **合理分配**: 根据使用频率分配配额
|
||||||
|
|
||||||
|
### 安全建议
|
||||||
|
- **定期备份**: 重要配置和数据要备份
|
||||||
|
- **监控日志**: 定期查看异常日志
|
||||||
|
- **更新密钥**: 定期更换JWT和加密密钥
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🆘 遇到问题怎么办？
|
||||||
|
|
||||||
|
### 自助排查
|
||||||
|
1. **查看日志**: `logs/` 目录下的日志文件
|
||||||
|
2. **检查配置**: 确认配置文件设置正确
|
||||||
|
3. **测试连通性**: 用 curl 测试API是否正常
|
||||||
|
4. **重启服务**: 有时候重启一下就好了
|
||||||
|
|
||||||
|
### 寻求帮助
|
||||||
|
- **GitHub Issues**: 提交详细的错误信息
|
||||||
|
- **查看文档**: 仔细阅读错误信息和文档
|
||||||
|
- **社区讨论**: 看看其他人是否遇到类似问题
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📄 许可证
|
||||||
|
本项目采用 [MIT许可证](LICENSE)。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
<div align="center">
|
||||||
|
|
||||||
|
**⭐ 觉得有用的话给个Star呗，这是对作者最大的鼓励！**
|
||||||
|
|
||||||
|
**🤝 有问题欢迎提Issue，有改进建议欢迎PR**
|
||||||
|
|
||||||
|
</div>
|
||||||
408
README_EN.md
Normal file
408
README_EN.md
Normal file
@@ -0,0 +1,408 @@
|
|||||||
|
# Claude Relay Service
|
||||||
|
|
||||||
|
<div align="center">
|
||||||
|
|
||||||
|
[](https://opensource.org/licenses/MIT)
|
||||||
|
[](https://nodejs.org/)
|
||||||
|
[](https://redis.io/)
|
||||||
|
[](https://www.docker.com/)
|
||||||
|
|
||||||
|
**🔐 Self-hosted Claude API relay service with multi-account management**
|
||||||
|
|
||||||
|
[English](#english) • [中文文档](#中文文档)
|
||||||
|
|
||||||
|
</div>
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## ⚠️ Important Notice
|
||||||
|
|
||||||
|
**Please read carefully before using this project:**
|
||||||
|
|
||||||
|
🚨 **Terms of Service Risk**: Using this project may violate Anthropic's terms of service. Please carefully read Anthropic's user agreement before use. All risks from using this project are borne by the user.
|
||||||
|
|
||||||
|
📖 **Disclaimer**: This project is for technical learning and research purposes only. The author is not responsible for any account bans, service interruptions, or other losses caused by using this project.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🤔 Is This Project Right for You?
|
||||||
|
|
||||||
|
- 🌍 **Regional Restrictions**: Can't directly access Claude Code service in your region?
|
||||||
|
- 🔒 **Privacy Concerns**: Worried about third-party mirror services logging or leaking your conversation content?
|
||||||
|
- 👥 **Cost Sharing**: Want to share Claude Code Max subscription costs with friends?
|
||||||
|
- ⚡ **Stability**: Third-party mirror sites often have outages and instability, affecting efficiency?
|
||||||
|
|
||||||
|
If you nodded yes, this project might be for you.
|
||||||
|
|
||||||
|
### Suitable Scenarios
|
||||||
|
|
||||||
|
✅ **Cost Sharing with Friends**: 3-5 friends sharing Claude Code Max subscription, enjoying Opus freely
|
||||||
|
✅ **Privacy Sensitive**: Don't want third parties to see your conversation content
|
||||||
|
✅ **Technical Tinkering**: Have basic technical skills, willing to build and maintain yourself
|
||||||
|
✅ **Stability Needs**: Need long-term stable Claude access, don't want to be restricted by mirror sites
|
||||||
|
✅ **Regional Restrictions**: Cannot directly access Claude official service
|
||||||
|
|
||||||
|
### Unsuitable Scenarios
|
||||||
|
|
||||||
|
❌ **Complete Beginner**: Don't understand technology at all, don't even know how to buy a server
|
||||||
|
❌ **Occasional Use**: Use it only a few times a month, not worth the hassle
|
||||||
|
❌ **Registration Issues**: Cannot register Claude account yourself
|
||||||
|
❌ **Payment Issues**: No payment method to subscribe to Claude Code
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 💭 Why Build Your Own?
|
||||||
|
|
||||||
|
Honestly, there are quite a few Claude proxy services online now, but there are also many issues:
|
||||||
|
|
||||||
|
### Problems with Existing Proxies
|
||||||
|
|
||||||
|
- 🕵️ **Privacy Risk**: Your conversation content is seen clearly by others, forget about business secrets
|
||||||
|
- 🐌 **Performance Instability**: Slow when many people use it, often crashes during peak hours
|
||||||
|
- 💰 **Price Opacity**: Don't know the actual costs
|
||||||
|
|
||||||
|
### Benefits of Self-hosting
|
||||||
|
|
||||||
|
- 🔐 **Data Security**: All API requests only go through your own server, direct connection to Anthropic API
|
||||||
|
- ⚡ **Controllable Performance**: Only a few of you using it, as fast as you want
|
||||||
|
- 💰 **Cost Transparency**: Clear view of how many tokens used, specific costs calculated at official prices
|
||||||
|
- 📊 **Complete Monitoring**: Usage statistics, cost analysis, performance monitoring all available
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🚀 Core Features
|
||||||
|
|
||||||
|
### Basic Features
|
||||||
|
- ✅ **Multi-account Management**: Add multiple Claude accounts for automatic rotation
|
||||||
|
- ✅ **Custom API Keys**: Assign independent keys to each person
|
||||||
|
- ✅ **Usage Statistics**: Detailed records of how many tokens each person used
|
||||||
|
|
||||||
|
### Advanced Features
|
||||||
|
- 🔄 **Smart Switching**: Automatically switch to next account when one has issues
|
||||||
|
- 🚀 **Performance Optimization**: Connection pooling, caching to reduce latency
|
||||||
|
- 📊 **Monitoring Dashboard**: Web interface to view all data
|
||||||
|
- 🛡️ **Security Control**: Access restrictions, rate limiting
|
||||||
|
- 🌐 **Proxy Support**: Support for HTTP/SOCKS5 proxies
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📋 Deployment Requirements
|
||||||
|
|
||||||
|
### Hardware Requirements (Minimum Configuration)
|
||||||
|
- **CPU**: 1 core is enough
|
||||||
|
- **Memory**: 512MB (1GB recommended)
|
||||||
|
- **Storage**: 30GB available space
|
||||||
|
- **Network**: Access to Anthropic API (recommend US region servers)
|
||||||
|
- **Suggestion**: 2 cores 4GB is basically enough, choose network with good return routes to your country (to improve speed, recommend not using proxy or setting server IP for direct connection)
|
||||||
|
|
||||||
|
### Software Requirements
|
||||||
|
- **Node.js** 18 or higher
|
||||||
|
- **Redis** 6 or higher
|
||||||
|
- **Operating System**: Linux recommended
|
||||||
|
|
||||||
|
### Cost Estimation
|
||||||
|
- **Server**: Light cloud server, 10-30 RMB per month
|
||||||
|
- **Claude Subscription**: Depends on how you share costs
|
||||||
|
- **Others**: Basically none
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🐳 Simplest Deployment Method (Docker)
|
||||||
|
|
||||||
|
If you're too lazy to set up the environment, use Docker directly:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 1. Download project
|
||||||
|
git clone https://github.com/yourusername/claude-relay-service.git
|
||||||
|
cd claude-relay-service
|
||||||
|
|
||||||
|
# 2. One-click start
|
||||||
|
docker-compose up -d
|
||||||
|
|
||||||
|
# 3. Check if started successfully
|
||||||
|
docker-compose ps
|
||||||
|
```
|
||||||
|
|
||||||
|
That simple, the service is running.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📦 Manual Deployment (For Tinkerers)
|
||||||
|
|
||||||
|
### Step 1: Environment Setup
|
||||||
|
|
||||||
|
**Ubuntu/Debian users:**
|
||||||
|
```bash
|
||||||
|
# Install Node.js
|
||||||
|
curl -fsSL https://deb.nodesource.com/setup_18.x | sudo -E bash -
|
||||||
|
sudo apt-get install -y nodejs
|
||||||
|
|
||||||
|
# Install Redis
|
||||||
|
sudo apt update
|
||||||
|
sudo apt install redis-server
|
||||||
|
sudo systemctl start redis-server
|
||||||
|
```
|
||||||
|
|
||||||
|
**CentOS/RHEL users:**
|
||||||
|
```bash
|
||||||
|
# Install Node.js
|
||||||
|
curl -fsSL https://rpm.nodesource.com/setup_18.x | sudo bash -
|
||||||
|
sudo yum install -y nodejs
|
||||||
|
|
||||||
|
# Install Redis
|
||||||
|
sudo yum install redis
|
||||||
|
sudo systemctl start redis
|
||||||
|
```
|
||||||
|
|
||||||
|
### Step 2: Download and Configure
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Download project
|
||||||
|
git clone https://github.com/yourusername/claude-relay-service.git
|
||||||
|
cd claude-relay-service
|
||||||
|
|
||||||
|
# Install dependencies
|
||||||
|
npm install
|
||||||
|
|
||||||
|
# Copy configuration files (Important!)
|
||||||
|
cp config/config.example.js config/config.js
|
||||||
|
cp .env.example .env
|
||||||
|
```
|
||||||
|
|
||||||
|
### Step 3: Configuration File Setup
|
||||||
|
|
||||||
|
**Edit `.env` file:**
|
||||||
|
```bash
|
||||||
|
# Generate these two keys randomly, but remember them
|
||||||
|
JWT_SECRET=your-super-secret-key
|
||||||
|
ENCRYPTION_KEY=32-character-encryption-key-write-randomly
|
||||||
|
|
||||||
|
# Redis configuration
|
||||||
|
REDIS_HOST=localhost
|
||||||
|
REDIS_PORT=6379
|
||||||
|
REDIS_PASSWORD=
|
||||||
|
```
|
||||||
|
|
||||||
|
**Edit `config/config.js` file:**
|
||||||
|
```javascript
|
||||||
|
module.exports = {
|
||||||
|
server: {
|
||||||
|
port: 3000, // Service port, can be changed
|
||||||
|
host: '0.0.0.0' // Don't change
|
||||||
|
},
|
||||||
|
redis: {
|
||||||
|
host: '127.0.0.1', // Redis address
|
||||||
|
port: 6379 // Redis port
|
||||||
|
},
|
||||||
|
// Keep other configurations as default
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Step 4: Start Service
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Initialize
|
||||||
|
npm run setup
|
||||||
|
|
||||||
|
# Start service
|
||||||
|
npm run service:start:daemon # Run in background (recommended)
|
||||||
|
|
||||||
|
# Check status
|
||||||
|
npm run service:status
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🎮 Getting Started
|
||||||
|
|
||||||
|
### 1. Open Management Interface
|
||||||
|
|
||||||
|
Browser visit: `http://your-server-IP:3000/web`
|
||||||
|
|
||||||
|
Default admin account: admin / admin123
|
||||||
|
|
||||||
|
### 2. Add Claude Account
|
||||||
|
|
||||||
|
This step is quite important, requires OAuth authorization:
|
||||||
|
|
||||||
|
1. Click "Claude Accounts" tab
|
||||||
|
2. If you're in China, configure proxy first (Important!)
|
||||||
|
3. Click "Add Account"
|
||||||
|
4. Click "Generate Authorization Link", will open a new page
|
||||||
|
5. Complete Claude login and authorization in the new page
|
||||||
|
6. Copy the returned Authorization Code
|
||||||
|
7. Paste to page to complete addition
|
||||||
|
|
||||||
|
**Note**: If you're in China, this step may require VPN.
|
||||||
|
|
||||||
|
### 3. Create API Key
|
||||||
|
|
||||||
|
Assign a key to each user:
|
||||||
|
|
||||||
|
1. Click "API Keys" tab
|
||||||
|
2. Click "Create New Key"
|
||||||
|
3. Give the key a name, like "Zhang San's Key"
|
||||||
|
4. Set usage limits (optional)
|
||||||
|
5. Save, note down the generated key
|
||||||
|
|
||||||
|
### 4. Start Using API
|
||||||
|
|
||||||
|
Now you can replace the official API with your own service:
|
||||||
|
|
||||||
|
**Original request:**
|
||||||
|
```bash
|
||||||
|
curl https://api.anthropic.com/v1/messages \
|
||||||
|
-H "x-api-key: official-key" \
|
||||||
|
-H "content-type: application/json" \
|
||||||
|
-d '{"model":"claude-3-sonnet-20240229","messages":[{"role":"user","content":"Hello"}]}'
|
||||||
|
```
|
||||||
|
|
||||||
|
**Current request:**
|
||||||
|
```bash
|
||||||
|
curl http://your-domain:3000/api/v1/messages \
|
||||||
|
-H "x-api-key: cr_your-created-key" \
|
||||||
|
-H "content-type: application/json" \
|
||||||
|
-d '{"model":"claude-3-sonnet-20240229","messages":[{"role":"user","content":"Hello"}]}'
|
||||||
|
```
|
||||||
|
|
||||||
|
Just change the domain and API Key to your own generated one, everything else is the same.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🔧 Daily Maintenance
|
||||||
|
|
||||||
|
### Service Management
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Check service status
|
||||||
|
npm run service:status
|
||||||
|
|
||||||
|
# View logs
|
||||||
|
npm run service:logs
|
||||||
|
|
||||||
|
# Restart service
|
||||||
|
npm run service:restart:daemon
|
||||||
|
|
||||||
|
# Stop service
|
||||||
|
npm run service:stop
|
||||||
|
```
|
||||||
|
|
||||||
|
### Monitor Usage
|
||||||
|
|
||||||
|
- **Web Interface**: `http://your-domain:3000/web` - View usage statistics
|
||||||
|
- **Health Check**: `http://your-domain:3000/health` - Confirm service is normal
|
||||||
|
- **Log Files**: Various log files in `logs/` directory
|
||||||
|
|
||||||
|
### Common Issue Resolution
|
||||||
|
|
||||||
|
**Can't connect to Redis?**
|
||||||
|
```bash
|
||||||
|
# Check if Redis is running
|
||||||
|
redis-cli ping
|
||||||
|
|
||||||
|
# Should return PONG
|
||||||
|
```
|
||||||
|
|
||||||
|
**OAuth authorization failed?**
|
||||||
|
- Check if proxy settings are correct
|
||||||
|
- Ensure normal access to claude.ai
|
||||||
|
- Clear browser cache and retry
|
||||||
|
|
||||||
|
**API request failed?**
|
||||||
|
- Check if API Key is correct
|
||||||
|
- View log files for error information
|
||||||
|
- Confirm Claude account status is normal
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🛠️ Advanced Usage
|
||||||
|
|
||||||
|
### Setting Up Proxy (Must-read for Chinese Users)
|
||||||
|
|
||||||
|
If you're in China, you need to configure proxy to use normally:
|
||||||
|
|
||||||
|
```javascript
|
||||||
|
// Add in account configuration
|
||||||
|
{
|
||||||
|
"proxy": {
|
||||||
|
"type": "socks5", // or "http"
|
||||||
|
"host": "127.0.0.1",
|
||||||
|
"port": 1080,
|
||||||
|
"username": "username", // if proxy requires authentication
|
||||||
|
"password": "password" // if proxy requires authentication
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Command Line Management Tool
|
||||||
|
|
||||||
|
Too lazy to open webpage? Use command line:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# View all API Keys
|
||||||
|
npm run cli keys list
|
||||||
|
|
||||||
|
# Create new Key
|
||||||
|
npm run cli keys create --name "Test Key" --limit 1000
|
||||||
|
|
||||||
|
# View account status
|
||||||
|
npm run cli accounts list
|
||||||
|
|
||||||
|
# Test account connection
|
||||||
|
npm run cli accounts test --id account-ID
|
||||||
|
```
|
||||||
|
|
||||||
|
### Monitoring Integration
|
||||||
|
|
||||||
|
If you want more professional monitoring, you can integrate Prometheus:
|
||||||
|
|
||||||
|
Visit `http://your-domain(or-IP):3000/metrics` to get metrics data.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 💡 Usage Recommendations
|
||||||
|
|
||||||
|
### Account Management
|
||||||
|
- **Multiple Accounts**: Recommend adding 2-3 Claude accounts to prevent single point of failure
|
||||||
|
- **Regular Checks**: Check account status weekly, handle exceptions promptly
|
||||||
|
- **Backup Plan**: Prepare several backup accounts that can step in during critical moments
|
||||||
|
|
||||||
|
### Cost Control
|
||||||
|
- **Set Limits**: Set reasonable usage limits for each API Key
|
||||||
|
- **Monitor Spending**: Regularly check cost statistics, control budget
|
||||||
|
- **Reasonable Allocation**: Allocate quotas based on usage frequency
|
||||||
|
|
||||||
|
### Security Recommendations
|
||||||
|
- **Regular Backups**: Back up important configurations and data
|
||||||
|
- **Monitor Logs**: Regularly check exception logs
|
||||||
|
- **Update Keys**: Regularly change JWT and encryption keys
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🆘 What to Do When You Encounter Problems?
|
||||||
|
|
||||||
|
### Self-troubleshooting
|
||||||
|
1. **Check Logs**: Log files in `logs/` directory
|
||||||
|
2. **Check Configuration**: Confirm configuration files are set correctly
|
||||||
|
3. **Test Connectivity**: Use curl to test if API is normal
|
||||||
|
4. **Restart Service**: Sometimes restarting fixes it
|
||||||
|
|
||||||
|
### Seeking Help
|
||||||
|
- **GitHub Issues**: Submit detailed error information
|
||||||
|
- **Read Documentation**: Carefully read error messages and documentation
|
||||||
|
- **Community Discussion**: See if others have encountered similar problems
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📄 License
|
||||||
|
This project uses the [MIT License](LICENSE).
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
<div align="center">
|
||||||
|
|
||||||
|
**⭐ If you find it useful, please give it a Star, this is the greatest encouragement to the author!**
|
||||||
|
|
||||||
|
**🤝 Feel free to submit Issues for problems, welcome PRs for improvement suggestions**
|
||||||
|
|
||||||
|
</div>
|
||||||
772
cli/index.js
Normal file
772
cli/index.js
Normal file
@@ -0,0 +1,772 @@
|
|||||||
|
#!/usr/bin/env node
|
||||||
|
|
||||||
|
const { Command } = require('commander');
|
||||||
|
const inquirer = require('inquirer');
|
||||||
|
const chalk = require('chalk');
|
||||||
|
const ora = require('ora');
|
||||||
|
const Table = require('table').table;
|
||||||
|
const bcrypt = require('bcryptjs');
|
||||||
|
const crypto = require('crypto');
|
||||||
|
|
||||||
|
const config = require('../config/config');
|
||||||
|
const redis = require('../src/models/redis');
|
||||||
|
const apiKeyService = require('../src/services/apiKeyService');
|
||||||
|
const claudeAccountService = require('../src/services/claudeAccountService');
|
||||||
|
|
||||||
|
const program = new Command();
|
||||||
|
|
||||||
|
// 🎨 样式
|
||||||
|
const styles = {
|
||||||
|
title: chalk.bold.blue,
|
||||||
|
success: chalk.green,
|
||||||
|
error: chalk.red,
|
||||||
|
warning: chalk.yellow,
|
||||||
|
info: chalk.cyan,
|
||||||
|
dim: chalk.dim
|
||||||
|
};
|
||||||
|
|
||||||
|
// 🔧 初始化
|
||||||
|
async function initialize() {
|
||||||
|
const spinner = ora('正在连接 Redis...').start();
|
||||||
|
try {
|
||||||
|
await redis.connect();
|
||||||
|
spinner.succeed('Redis 连接成功');
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('Redis 连接失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔐 管理员账户管理
|
||||||
|
program
|
||||||
|
.command('admin')
|
||||||
|
.description('管理员账户操作')
|
||||||
|
.action(async () => {
|
||||||
|
await initialize();
|
||||||
|
|
||||||
|
const { action } = await inquirer.prompt({
|
||||||
|
type: 'list',
|
||||||
|
name: 'action',
|
||||||
|
message: '选择操作:',
|
||||||
|
choices: [
|
||||||
|
{ name: '🔑 设置管理员密码', value: 'set-password' },
|
||||||
|
{ name: '👤 创建初始管理员', value: 'create-admin' },
|
||||||
|
{ name: '🔄 重置管理员密码', value: 'reset-password' },
|
||||||
|
{ name: '📊 查看管理员信息', value: 'view-admin' }
|
||||||
|
]
|
||||||
|
});
|
||||||
|
|
||||||
|
switch (action) {
|
||||||
|
case 'set-password':
|
||||||
|
await setAdminPassword();
|
||||||
|
break;
|
||||||
|
case 'create-admin':
|
||||||
|
await createInitialAdmin();
|
||||||
|
break;
|
||||||
|
case 'reset-password':
|
||||||
|
await resetAdminPassword();
|
||||||
|
break;
|
||||||
|
case 'view-admin':
|
||||||
|
await viewAdminInfo();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
await redis.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🔑 API Key 管理
|
||||||
|
program
|
||||||
|
.command('keys')
|
||||||
|
.description('API Key 管理')
|
||||||
|
.action(async () => {
|
||||||
|
await initialize();
|
||||||
|
|
||||||
|
// 尝试兼容不同版本的inquirer
|
||||||
|
let prompt = inquirer.prompt || inquirer.default?.prompt || inquirer;
|
||||||
|
if (typeof prompt !== 'function') {
|
||||||
|
prompt = (await import('inquirer')).default;
|
||||||
|
}
|
||||||
|
|
||||||
|
const { action } = await prompt({
|
||||||
|
type: 'list',
|
||||||
|
name: 'action',
|
||||||
|
message: '选择操作:',
|
||||||
|
choices: [
|
||||||
|
{ name: '📋 列出所有 API Keys', value: 'list' },
|
||||||
|
{ name: '🔑 创建新的 API Key', value: 'create' },
|
||||||
|
{ name: '📝 更新 API Key', value: 'update' },
|
||||||
|
{ name: '🗑️ 删除 API Key', value: 'delete' },
|
||||||
|
{ name: '📊 查看使用统计', value: 'stats' },
|
||||||
|
{ name: '🧹 重置所有统计数据', value: 'reset-stats' }
|
||||||
|
]
|
||||||
|
});
|
||||||
|
|
||||||
|
switch (action) {
|
||||||
|
case 'list':
|
||||||
|
await listApiKeys();
|
||||||
|
break;
|
||||||
|
case 'create':
|
||||||
|
await createApiKey();
|
||||||
|
break;
|
||||||
|
case 'update':
|
||||||
|
await updateApiKey();
|
||||||
|
break;
|
||||||
|
case 'delete':
|
||||||
|
await deleteApiKey();
|
||||||
|
break;
|
||||||
|
case 'stats':
|
||||||
|
await viewApiKeyStats();
|
||||||
|
break;
|
||||||
|
case 'reset-stats':
|
||||||
|
await resetAllApiKeyStats();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
await redis.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🏢 Claude 账户管理
|
||||||
|
program
|
||||||
|
.command('accounts')
|
||||||
|
.description('Claude 账户管理')
|
||||||
|
.action(async () => {
|
||||||
|
await initialize();
|
||||||
|
|
||||||
|
const { action } = await inquirer.prompt({
|
||||||
|
type: 'list',
|
||||||
|
name: 'action',
|
||||||
|
message: '选择操作:',
|
||||||
|
choices: [
|
||||||
|
{ name: '📋 列出所有账户', value: 'list' },
|
||||||
|
{ name: '🏢 创建新账户', value: 'create' },
|
||||||
|
{ name: '📝 更新账户', value: 'update' },
|
||||||
|
{ name: '🗑️ 删除账户', value: 'delete' },
|
||||||
|
{ name: '🔄 刷新 Token', value: 'refresh' },
|
||||||
|
{ name: '🧪 测试账户', value: 'test' }
|
||||||
|
]
|
||||||
|
});
|
||||||
|
|
||||||
|
switch (action) {
|
||||||
|
case 'list':
|
||||||
|
await listClaudeAccounts();
|
||||||
|
break;
|
||||||
|
case 'create':
|
||||||
|
await createClaudeAccount();
|
||||||
|
break;
|
||||||
|
case 'update':
|
||||||
|
await updateClaudeAccount();
|
||||||
|
break;
|
||||||
|
case 'delete':
|
||||||
|
await deleteClaudeAccount();
|
||||||
|
break;
|
||||||
|
case 'refresh':
|
||||||
|
await refreshAccountToken();
|
||||||
|
break;
|
||||||
|
case 'test':
|
||||||
|
await testClaudeAccount();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
await redis.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🧹 重置统计数据命令
|
||||||
|
program
|
||||||
|
.command('reset-stats')
|
||||||
|
.description('重置所有API Key的统计数据')
|
||||||
|
.option('--force', '跳过确认直接重置')
|
||||||
|
.option('--debug', '显示详细的Redis键调试信息')
|
||||||
|
.action(async (options) => {
|
||||||
|
await initialize();
|
||||||
|
|
||||||
|
console.log(styles.title('\n🧹 重置所有API Key统计数据\n'));
|
||||||
|
|
||||||
|
// 如果启用调试，显示当前Redis键
|
||||||
|
if (options.debug) {
|
||||||
|
console.log(styles.info('🔍 调试模式: 检查Redis中的实际键...\n'));
|
||||||
|
try {
|
||||||
|
const usageKeys = await redis.getClient().keys('usage:*');
|
||||||
|
const apiKeyKeys = await redis.getClient().keys('apikey:*');
|
||||||
|
|
||||||
|
console.log(styles.dim('API Key 键:'));
|
||||||
|
apiKeyKeys.forEach(key => console.log(` ${key}`));
|
||||||
|
|
||||||
|
console.log(styles.dim('\nUsage 键:'));
|
||||||
|
usageKeys.forEach(key => console.log(` ${key}`));
|
||||||
|
|
||||||
|
// 检查今日统计键
|
||||||
|
const today = new Date().toISOString().split('T')[0];
|
||||||
|
const dailyKeys = await redis.getClient().keys(`usage:daily:*:${today}`);
|
||||||
|
console.log(styles.dim(`\n今日统计键 (${today}):`));
|
||||||
|
dailyKeys.forEach(key => console.log(` ${key}`));
|
||||||
|
|
||||||
|
console.log('');
|
||||||
|
} catch (error) {
|
||||||
|
console.error(styles.error('调试信息获取失败:', error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 显示警告信息
|
||||||
|
console.log(styles.warning('⚠️ 警告: 此操作将删除所有API Key的使用统计数据!'));
|
||||||
|
console.log(styles.dim(' 包括: Token使用量、请求数量、每日/每月统计、最后使用时间等'));
|
||||||
|
console.log(styles.dim(' 此操作不可逆，请谨慎操作!\n'));
|
||||||
|
|
||||||
|
if (!options.force) {
|
||||||
|
console.log(styles.info('如需强制执行，请使用: npm run cli reset-stats -- --force\n'));
|
||||||
|
console.log(styles.error('操作已取消 - 请添加 --force 参数确认重置'));
|
||||||
|
await redis.disconnect();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取当前统计概览
|
||||||
|
const spinner = ora('正在获取当前统计数据...').start();
|
||||||
|
try {
|
||||||
|
const apiKeys = await apiKeyService.getAllApiKeys();
|
||||||
|
const totalTokens = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.tokens || 0), 0);
|
||||||
|
const totalRequests = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.requests || 0), 0);
|
||||||
|
|
||||||
|
spinner.succeed('统计数据获取完成');
|
||||||
|
|
||||||
|
console.log(styles.info('\n📊 当前统计概览:'));
|
||||||
|
console.log(` API Keys 数量: ${apiKeys.length}`);
|
||||||
|
console.log(` 总 Token 使用量: ${totalTokens.toLocaleString()}`);
|
||||||
|
console.log(` 总请求数量: ${totalRequests.toLocaleString()}\n`);
|
||||||
|
|
||||||
|
// 执行重置操作
|
||||||
|
const resetSpinner = ora('正在重置所有API Key统计数据...').start();
|
||||||
|
|
||||||
|
const stats = await redis.resetAllUsageStats();
|
||||||
|
|
||||||
|
resetSpinner.succeed('所有统计数据重置完成');
|
||||||
|
|
||||||
|
// 显示重置结果
|
||||||
|
console.log(styles.success('\n✅ 重置操作完成!\n'));
|
||||||
|
console.log(styles.info('📊 重置详情:'));
|
||||||
|
console.log(` 重置的API Key数量: ${stats.resetApiKeys}`);
|
||||||
|
console.log(` 删除的总体统计: ${stats.deletedKeys} 个`);
|
||||||
|
console.log(` 删除的每日统计: ${stats.deletedDailyKeys} 个`);
|
||||||
|
console.log(` 删除的每月统计: ${stats.deletedMonthlyKeys} 个`);
|
||||||
|
|
||||||
|
console.log(styles.warning('\n💡 提示: API Key本身未被删除，只是清空了使用统计数据'));
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('重置操作失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
|
||||||
|
await redis.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
// 📊 系统状态
|
||||||
|
program
|
||||||
|
.command('status')
|
||||||
|
.description('查看系统状态')
|
||||||
|
.action(async () => {
|
||||||
|
await initialize();
|
||||||
|
|
||||||
|
const spinner = ora('正在获取系统状态...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const [systemStats, apiKeys, accounts] = await Promise.all([
|
||||||
|
redis.getSystemStats(),
|
||||||
|
apiKeyService.getAllApiKeys(),
|
||||||
|
claudeAccountService.getAllAccounts()
|
||||||
|
]);
|
||||||
|
|
||||||
|
spinner.succeed('系统状态获取成功');
|
||||||
|
|
||||||
|
console.log(styles.title('\n📊 系统状态概览\n'));
|
||||||
|
|
||||||
|
const statusData = [
|
||||||
|
['项目', '数量', '状态'],
|
||||||
|
['API Keys', apiKeys.length, `${apiKeys.filter(k => k.isActive).length} 活跃`],
|
||||||
|
['Claude 账户', accounts.length, `${accounts.filter(a => a.isActive).length} 活跃`],
|
||||||
|
['Redis 连接', redis.isConnected ? '已连接' : '未连接', redis.isConnected ? '🟢' : '🔴'],
|
||||||
|
['运行时间', `${Math.floor(process.uptime() / 60)} 分钟`, '🕐']
|
||||||
|
];
|
||||||
|
|
||||||
|
console.log(table(statusData));
|
||||||
|
|
||||||
|
// 使用统计
|
||||||
|
const totalTokens = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.tokens || 0), 0);
|
||||||
|
const totalRequests = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.requests || 0), 0);
|
||||||
|
|
||||||
|
console.log(styles.title('\n📈 使用统计\n'));
|
||||||
|
console.log(`总 Token 使用量: ${styles.success(totalTokens.toLocaleString())}`);
|
||||||
|
console.log(`总请求数: ${styles.success(totalRequests.toLocaleString())}`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('获取系统状态失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
|
||||||
|
await redis.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🧹 清理命令
|
||||||
|
program
|
||||||
|
.command('cleanup')
|
||||||
|
.description('清理过期数据')
|
||||||
|
.action(async () => {
|
||||||
|
await initialize();
|
||||||
|
|
||||||
|
const { confirm } = await inquirer.prompt({
|
||||||
|
type: 'confirm',
|
||||||
|
name: 'confirm',
|
||||||
|
message: '确定要清理过期数据吗？',
|
||||||
|
default: false
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!confirm) {
|
||||||
|
console.log(styles.warning('操作已取消'));
|
||||||
|
await redis.disconnect();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const spinner = ora('正在清理过期数据...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const [expiredKeys, errorAccounts] = await Promise.all([
|
||||||
|
apiKeyService.cleanupExpiredKeys(),
|
||||||
|
claudeAccountService.cleanupErrorAccounts()
|
||||||
|
]);
|
||||||
|
|
||||||
|
await redis.cleanup();
|
||||||
|
|
||||||
|
spinner.succeed('清理完成');
|
||||||
|
console.log(`${styles.success('✅')} 清理了 ${expiredKeys} 个过期 API Key`);
|
||||||
|
console.log(`${styles.success('✅')} 重置了 ${errorAccounts} 个错误账户`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('清理失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
|
||||||
|
await redis.disconnect();
|
||||||
|
});
|
||||||
|
|
||||||
|
// 实现具体功能函数
|
||||||
|
|
||||||
|
async function createInitialAdmin() {
|
||||||
|
console.log(styles.title('\n🔐 创建初始管理员账户\n'));
|
||||||
|
|
||||||
|
const adminData = await inquirer.prompt([
|
||||||
|
{
|
||||||
|
type: 'input',
|
||||||
|
name: 'username',
|
||||||
|
message: '用户名:',
|
||||||
|
default: 'admin',
|
||||||
|
validate: input => input.length >= 3 || '用户名至少3个字符'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
type: 'password',
|
||||||
|
name: 'password',
|
||||||
|
message: '密码:',
|
||||||
|
validate: input => input.length >= 8 || '密码至少8个字符'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
type: 'password',
|
||||||
|
name: 'confirmPassword',
|
||||||
|
message: '确认密码:',
|
||||||
|
validate: (input, answers) => input === answers.password || '密码不匹配'
|
||||||
|
}
|
||||||
|
]);
|
||||||
|
|
||||||
|
const spinner = ora('正在创建管理员账户...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const passwordHash = await bcrypt.hash(adminData.password, 12);
|
||||||
|
|
||||||
|
const credentials = {
|
||||||
|
username: adminData.username,
|
||||||
|
passwordHash,
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
id: crypto.randomBytes(16).toString('hex')
|
||||||
|
};
|
||||||
|
|
||||||
|
await redis.setSession('admin_credentials', credentials, 0); // 永不过期
|
||||||
|
|
||||||
|
spinner.succeed('管理员账户创建成功');
|
||||||
|
console.log(`${styles.success('✅')} 用户名: ${adminData.username}`);
|
||||||
|
console.log(`${styles.info('ℹ️')} 请妥善保管登录凭据`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('创建管理员账户失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function setAdminPassword() {
|
||||||
|
console.log(styles.title('\n🔑 设置管理员密码\n'));
|
||||||
|
|
||||||
|
const passwordData = await inquirer.prompt([
|
||||||
|
{
|
||||||
|
type: 'password',
|
||||||
|
name: 'newPassword',
|
||||||
|
message: '新密码:',
|
||||||
|
validate: input => input.length >= 8 || '密码至少8个字符'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
type: 'password',
|
||||||
|
name: 'confirmPassword',
|
||||||
|
message: '确认密码:',
|
||||||
|
validate: (input, answers) => input === answers.newPassword || '密码不匹配'
|
||||||
|
}
|
||||||
|
]);
|
||||||
|
|
||||||
|
const spinner = ora('正在更新密码...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const adminData = await redis.getSession('admin_credentials');
|
||||||
|
|
||||||
|
if (!adminData || Object.keys(adminData).length === 0) {
|
||||||
|
spinner.fail('未找到管理员账户');
|
||||||
|
console.log(styles.warning('请先创建初始管理员账户'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const passwordHash = await bcrypt.hash(passwordData.newPassword, 12);
|
||||||
|
adminData.passwordHash = passwordHash;
|
||||||
|
adminData.updatedAt = new Date().toISOString();
|
||||||
|
|
||||||
|
await redis.setSession('admin_credentials', adminData, 0);
|
||||||
|
|
||||||
|
spinner.succeed('密码更新成功');
|
||||||
|
console.log(`${styles.success('✅')} 管理员密码已更新`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('密码更新失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function listApiKeys() {
|
||||||
|
const spinner = ora('正在获取 API Keys...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const apiKeys = await apiKeyService.getAllApiKeys();
|
||||||
|
spinner.succeed(`找到 ${apiKeys.length} 个 API Key`);
|
||||||
|
|
||||||
|
if (apiKeys.length === 0) {
|
||||||
|
console.log(styles.warning('没有找到任何 API Key'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const tableData = [
|
||||||
|
['ID', '名称', '状态', 'Token使用', '请求数', '创建时间']
|
||||||
|
];
|
||||||
|
|
||||||
|
apiKeys.forEach(key => {
|
||||||
|
tableData.push([
|
||||||
|
key.id.substring(0, 8) + '...',
|
||||||
|
key.name,
|
||||||
|
key.isActive ? '🟢 活跃' : '🔴 停用',
|
||||||
|
key.usage?.total?.tokens?.toLocaleString() || '0',
|
||||||
|
key.usage?.total?.requests?.toLocaleString() || '0',
|
||||||
|
new Date(key.createdAt).toLocaleDateString()
|
||||||
|
]);
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log('\n📋 API Keys 列表:\n');
|
||||||
|
console.log(table(tableData));
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('获取 API Keys 失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createApiKey() {
|
||||||
|
console.log(styles.title('\n🔑 创建新的 API Key\n'));
|
||||||
|
|
||||||
|
const keyData = await inquirer.prompt([
|
||||||
|
{
|
||||||
|
type: 'input',
|
||||||
|
name: 'name',
|
||||||
|
message: 'API Key 名称:',
|
||||||
|
validate: input => input.length > 0 || '名称不能为空'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
type: 'input',
|
||||||
|
name: 'description',
|
||||||
|
message: '描述 (可选):'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
type: 'number',
|
||||||
|
name: 'tokenLimit',
|
||||||
|
message: 'Token 限制 (0=无限制):',
|
||||||
|
default: 1000000
|
||||||
|
},
|
||||||
|
{
|
||||||
|
type: 'number',
|
||||||
|
name: 'requestLimit',
|
||||||
|
message: '请求限制 (0=无限制):',
|
||||||
|
default: 1000
|
||||||
|
}
|
||||||
|
]);
|
||||||
|
|
||||||
|
const spinner = ora('正在创建 API Key...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const newKey = await apiKeyService.generateApiKey(keyData);
|
||||||
|
|
||||||
|
spinner.succeed('API Key 创建成功');
|
||||||
|
console.log(`${styles.success('✅')} API Key: ${styles.warning(newKey.apiKey)}`);
|
||||||
|
console.log(`${styles.info('ℹ️')} 请妥善保管此 API Key，它只会显示一次`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('创建 API Key 失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function resetAllApiKeyStats() {
|
||||||
|
console.log(styles.title('\n🧹 重置所有API Key统计数据\n'));
|
||||||
|
|
||||||
|
// 显示警告信息
|
||||||
|
console.log(styles.warning('⚠️ 警告: 此操作将删除所有API Key的使用统计数据!'));
|
||||||
|
console.log(styles.dim(' 包括: Token使用量、请求数量、每日/每月统计、最后使用时间等'));
|
||||||
|
console.log(styles.dim(' 此操作不可逆，请谨慎操作!\n'));
|
||||||
|
|
||||||
|
// 第一次确认
|
||||||
|
const { firstConfirm } = await inquirer.prompt({
|
||||||
|
type: 'confirm',
|
||||||
|
name: 'firstConfirm',
|
||||||
|
message: '您确定要重置所有API Key的统计数据吗？',
|
||||||
|
default: false
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!firstConfirm) {
|
||||||
|
console.log(styles.info('操作已取消'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取当前统计概览
|
||||||
|
const spinner = ora('正在获取当前统计数据...').start();
|
||||||
|
try {
|
||||||
|
const apiKeys = await apiKeyService.getAllApiKeys();
|
||||||
|
const totalTokens = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.tokens || 0), 0);
|
||||||
|
const totalRequests = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.requests || 0), 0);
|
||||||
|
|
||||||
|
spinner.succeed('统计数据获取完成');
|
||||||
|
|
||||||
|
console.log(styles.info('\n📊 当前统计概览:'));
|
||||||
|
console.log(` API Keys 数量: ${apiKeys.length}`);
|
||||||
|
console.log(` 总 Token 使用量: ${totalTokens.toLocaleString()}`);
|
||||||
|
console.log(` 总请求数量: ${totalRequests.toLocaleString()}\n`);
|
||||||
|
|
||||||
|
// 第二次确认（需要输入"RESET"）
|
||||||
|
const { confirmation } = await inquirer.prompt({
|
||||||
|
type: 'input',
|
||||||
|
name: 'confirmation',
|
||||||
|
message: '请输入 "RESET" 来确认重置操作:',
|
||||||
|
validate: input => input === 'RESET' || '请输入正确的确认文本 "RESET"'
|
||||||
|
});
|
||||||
|
|
||||||
|
if (confirmation !== 'RESET') {
|
||||||
|
console.log(styles.info('操作已取消'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 执行重置操作
|
||||||
|
const resetSpinner = ora('正在重置所有API Key统计数据...').start();
|
||||||
|
|
||||||
|
const stats = await redis.resetAllUsageStats();
|
||||||
|
|
||||||
|
resetSpinner.succeed('所有统计数据重置完成');
|
||||||
|
|
||||||
|
// 显示重置结果
|
||||||
|
console.log(styles.success('\n✅ 重置操作完成!\n'));
|
||||||
|
console.log(styles.info('📊 重置详情:'));
|
||||||
|
console.log(` 重置的API Key数量: ${stats.resetApiKeys}`);
|
||||||
|
console.log(` 删除的总体统计: ${stats.deletedKeys} 个`);
|
||||||
|
console.log(` 删除的每日统计: ${stats.deletedDailyKeys} 个`);
|
||||||
|
console.log(` 删除的每月统计: ${stats.deletedMonthlyKeys} 个`);
|
||||||
|
|
||||||
|
console.log(styles.warning('\n💡 提示: API Key本身未被删除，只是清空了使用统计数据'));
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('重置操作失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function viewApiKeyStats() {
|
||||||
|
console.log(styles.title('\n📊 API Key 使用统计\n'));
|
||||||
|
|
||||||
|
const spinner = ora('正在获取统计数据...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const apiKeys = await apiKeyService.getAllApiKeys();
|
||||||
|
|
||||||
|
if (apiKeys.length === 0) {
|
||||||
|
spinner.succeed('获取完成');
|
||||||
|
console.log(styles.warning('没有找到任何 API Key'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
spinner.succeed(`找到 ${apiKeys.length} 个 API Key 的统计数据`);
|
||||||
|
|
||||||
|
const tableData = [
|
||||||
|
['名称', 'Token总量', '输入Token', '输出Token', '请求数', '最后使用']
|
||||||
|
];
|
||||||
|
|
||||||
|
let totalTokens = 0;
|
||||||
|
let totalRequests = 0;
|
||||||
|
|
||||||
|
apiKeys.forEach(key => {
|
||||||
|
const usage = key.usage?.total || {};
|
||||||
|
const tokens = usage.tokens || 0;
|
||||||
|
const inputTokens = usage.inputTokens || 0;
|
||||||
|
const outputTokens = usage.outputTokens || 0;
|
||||||
|
const requests = usage.requests || 0;
|
||||||
|
|
||||||
|
totalTokens += tokens;
|
||||||
|
totalRequests += requests;
|
||||||
|
|
||||||
|
tableData.push([
|
||||||
|
key.name,
|
||||||
|
tokens.toLocaleString(),
|
||||||
|
inputTokens.toLocaleString(),
|
||||||
|
outputTokens.toLocaleString(),
|
||||||
|
requests.toLocaleString(),
|
||||||
|
key.lastUsedAt ? new Date(key.lastUsedAt).toLocaleDateString() : '从未使用'
|
||||||
|
]);
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log(table(tableData));
|
||||||
|
|
||||||
|
console.log(styles.info('\n📈 总计统计:'));
|
||||||
|
console.log(`总 Token 使用量: ${styles.success(totalTokens.toLocaleString())}`);
|
||||||
|
console.log(`总请求数量: ${styles.success(totalRequests.toLocaleString())}`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('获取统计数据失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function updateApiKey() {
|
||||||
|
console.log(styles.title('\n📝 更新 API Key\n'));
|
||||||
|
console.log(styles.warning('功能开发中...'));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function deleteApiKey() {
|
||||||
|
console.log(styles.title('\n🗑️ 删除 API Key\n'));
|
||||||
|
console.log(styles.warning('功能开发中...'));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function resetAdminPassword() {
|
||||||
|
console.log(styles.title('\n🔄 重置管理员密码\n'));
|
||||||
|
console.log(styles.warning('功能开发中...'));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function viewAdminInfo() {
|
||||||
|
console.log(styles.title('\n👤 管理员信息\n'));
|
||||||
|
|
||||||
|
const spinner = ora('正在获取管理员信息...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const adminData = await redis.getSession('admin_credentials');
|
||||||
|
|
||||||
|
if (!adminData || Object.keys(adminData).length === 0) {
|
||||||
|
spinner.fail('未找到管理员账户');
|
||||||
|
console.log(styles.warning('请先创建初始管理员账户'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
spinner.succeed('管理员信息获取成功');
|
||||||
|
|
||||||
|
console.log(`用户名: ${styles.info(adminData.username)}`);
|
||||||
|
console.log(`创建时间: ${styles.dim(new Date(adminData.createdAt).toLocaleString())}`);
|
||||||
|
console.log(`最后登录: ${adminData.lastLogin ? styles.dim(new Date(adminData.lastLogin).toLocaleString()) : '从未登录'}`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('获取管理员信息失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function createClaudeAccount() {
|
||||||
|
console.log(styles.title('\n🏢 创建 Claude 账户\n'));
|
||||||
|
console.log(styles.warning('功能开发中... 请使用Web界面创建OAuth账户'));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function updateClaudeAccount() {
|
||||||
|
console.log(styles.title('\n📝 更新 Claude 账户\n'));
|
||||||
|
console.log(styles.warning('功能开发中...'));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function deleteClaudeAccount() {
|
||||||
|
console.log(styles.title('\n🗑️ 删除 Claude 账户\n'));
|
||||||
|
console.log(styles.warning('功能开发中...'));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function refreshAccountToken() {
|
||||||
|
console.log(styles.title('\n🔄 刷新账户 Token\n'));
|
||||||
|
console.log(styles.warning('功能开发中...'));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function testClaudeAccount() {
|
||||||
|
console.log(styles.title('\n🧪 测试 Claude 账户\n'));
|
||||||
|
console.log(styles.warning('功能开发中...'));
|
||||||
|
}
|
||||||
|
|
||||||
|
async function listClaudeAccounts() {
|
||||||
|
const spinner = ora('正在获取 Claude 账户...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const accounts = await claudeAccountService.getAllAccounts();
|
||||||
|
spinner.succeed(`找到 ${accounts.length} 个 Claude 账户`);
|
||||||
|
|
||||||
|
if (accounts.length === 0) {
|
||||||
|
console.log(styles.warning('没有找到任何 Claude 账户'));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const tableData = [
|
||||||
|
['ID', '名称', '邮箱', '状态', '代理', '最后使用']
|
||||||
|
];
|
||||||
|
|
||||||
|
accounts.forEach(account => {
|
||||||
|
tableData.push([
|
||||||
|
account.id.substring(0, 8) + '...',
|
||||||
|
account.name,
|
||||||
|
account.email || '-',
|
||||||
|
account.isActive ? (account.status === 'active' ? '🟢 活跃' : '🟡 待激活') : '🔴 停用',
|
||||||
|
account.proxy ? '🌐 是' : '-',
|
||||||
|
account.lastUsedAt ? new Date(account.lastUsedAt).toLocaleDateString() : '-'
|
||||||
|
]);
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log('\n🏢 Claude 账户列表:\n');
|
||||||
|
console.log(table(tableData));
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('获取 Claude 账户失败');
|
||||||
|
console.error(styles.error(error.message));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 程序信息
|
||||||
|
program
|
||||||
|
.name('claude-relay-cli')
|
||||||
|
.description('Claude Relay Service 命令行管理工具')
|
||||||
|
.version('1.0.0');
|
||||||
|
|
||||||
|
// 解析命令行参数
|
||||||
|
program.parse();
|
||||||
|
|
||||||
|
// 如果没有提供命令，显示帮助
|
||||||
|
if (!process.argv.slice(2).length) {
|
||||||
|
console.log(styles.title('🚀 Claude Relay Service CLI\n'));
|
||||||
|
console.log('使用以下命令管理服务:\n');
|
||||||
|
console.log(' claude-relay-cli admin - 管理员账户操作');
|
||||||
|
console.log(' claude-relay-cli keys - API Key 管理 (包含重置统计数据)');
|
||||||
|
console.log(' claude-relay-cli accounts - Claude 账户管理');
|
||||||
|
console.log(' claude-relay-cli status - 查看系统状态');
|
||||||
|
console.log(' claude-relay-cli cleanup - 清理过期数据');
|
||||||
|
console.log(' claude-relay-cli reset-stats - 重置所有API Key统计数据');
|
||||||
|
console.log('\n使用 --help 查看详细帮助信息');
|
||||||
|
}
|
||||||
90
config/config.example.js
Normal file
90
config/config.example.js
Normal file
@@ -0,0 +1,90 @@
|
|||||||
|
const path = require('path');
|
||||||
|
require('dotenv').config();
|
||||||
|
|
||||||
|
const config = {
|
||||||
|
// 🌐 服务器配置
|
||||||
|
server: {
|
||||||
|
port: parseInt(process.env.PORT) || 3000,
|
||||||
|
host: process.env.HOST || '0.0.0.0',
|
||||||
|
nodeEnv: process.env.NODE_ENV || 'development',
|
||||||
|
trustProxy: process.env.TRUST_PROXY === 'true'
|
||||||
|
},
|
||||||
|
|
||||||
|
// 🔐 安全配置
|
||||||
|
security: {
|
||||||
|
jwtSecret: process.env.JWT_SECRET || 'CHANGE-THIS-JWT-SECRET-IN-PRODUCTION',
|
||||||
|
adminSessionTimeout: parseInt(process.env.ADMIN_SESSION_TIMEOUT) || 86400000, // 24小时
|
||||||
|
apiKeyPrefix: process.env.API_KEY_PREFIX || 'cr_',
|
||||||
|
encryptionKey: process.env.ENCRYPTION_KEY || 'CHANGE-THIS-32-CHARACTER-KEY-NOW'
|
||||||
|
},
|
||||||
|
|
||||||
|
// 📊 Redis配置
|
||||||
|
redis: {
|
||||||
|
host: process.env.REDIS_HOST || '127.0.0.1',
|
||||||
|
port: parseInt(process.env.REDIS_PORT) || 6379,
|
||||||
|
password: process.env.REDIS_PASSWORD || '',
|
||||||
|
db: parseInt(process.env.REDIS_DB) || 0,
|
||||||
|
connectTimeout: 10000,
|
||||||
|
commandTimeout: 5000,
|
||||||
|
retryDelayOnFailover: 100,
|
||||||
|
maxRetriesPerRequest: 3,
|
||||||
|
lazyConnect: true
|
||||||
|
},
|
||||||
|
|
||||||
|
// 🎯 Claude API配置
|
||||||
|
claude: {
|
||||||
|
apiUrl: process.env.CLAUDE_API_URL || 'https://api.anthropic.com/v1/messages',
|
||||||
|
apiVersion: process.env.CLAUDE_API_VERSION || '2023-06-01',
|
||||||
|
betaHeader: process.env.CLAUDE_BETA_HEADER || 'claude-code-20250219,oauth-2025-04-20,interleaved-thinking-2025-05-14,fine-grained-tool-streaming-2025-05-14'
|
||||||
|
},
|
||||||
|
|
||||||
|
// 🌐 代理配置
|
||||||
|
proxy: {
|
||||||
|
timeout: parseInt(process.env.DEFAULT_PROXY_TIMEOUT) || 30000,
|
||||||
|
maxRetries: parseInt(process.env.MAX_PROXY_RETRIES) || 3
|
||||||
|
},
|
||||||
|
|
||||||
|
// 📈 使用限制
|
||||||
|
limits: {
|
||||||
|
defaultTokenLimit: parseInt(process.env.DEFAULT_TOKEN_LIMIT) || 1000000,
|
||||||
|
defaultRequestLimit: parseInt(process.env.DEFAULT_REQUEST_LIMIT) || 1000
|
||||||
|
},
|
||||||
|
|
||||||
|
// 🚦 速率限制
|
||||||
|
rateLimit: {
|
||||||
|
windowMs: parseInt(process.env.RATE_LIMIT_WINDOW) || 60000,
|
||||||
|
maxRequests: parseInt(process.env.RATE_LIMIT_MAX_REQUESTS) || 100
|
||||||
|
},
|
||||||
|
|
||||||
|
// 📝 日志配置
|
||||||
|
logging: {
|
||||||
|
level: process.env.LOG_LEVEL || 'info',
|
||||||
|
dirname: path.join(__dirname, '..', 'logs'),
|
||||||
|
maxSize: process.env.LOG_MAX_SIZE || '10m',
|
||||||
|
maxFiles: parseInt(process.env.LOG_MAX_FILES) || 5
|
||||||
|
},
|
||||||
|
|
||||||
|
// 🔧 系统配置
|
||||||
|
system: {
|
||||||
|
cleanupInterval: parseInt(process.env.CLEANUP_INTERVAL) || 3600000, // 1小时
|
||||||
|
tokenUsageRetention: parseInt(process.env.TOKEN_USAGE_RETENTION) || 2592000000, // 30天
|
||||||
|
healthCheckInterval: parseInt(process.env.HEALTH_CHECK_INTERVAL) || 60000 // 1分钟
|
||||||
|
},
|
||||||
|
|
||||||
|
// 🎨 Web界面配置
|
||||||
|
web: {
|
||||||
|
title: process.env.WEB_TITLE || 'Claude Relay Service',
|
||||||
|
description: process.env.WEB_DESCRIPTION || 'Multi-account Claude API relay service with beautiful management interface',
|
||||||
|
logoUrl: process.env.WEB_LOGO_URL || '/assets/logo.png',
|
||||||
|
enableCors: process.env.ENABLE_CORS === 'true',
|
||||||
|
sessionSecret: process.env.WEB_SESSION_SECRET || 'CHANGE-THIS-SESSION-SECRET'
|
||||||
|
},
|
||||||
|
|
||||||
|
// 🛠️ 开发配置
|
||||||
|
development: {
|
||||||
|
debug: process.env.DEBUG === 'true',
|
||||||
|
hotReload: process.env.HOT_RELOAD === 'true'
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
module.exports = config;
|
||||||
117
docker-compose.yml
Normal file
117
docker-compose.yml
Normal file
@@ -0,0 +1,117 @@
|
|||||||
|
version: '3.8'
|
||||||
|
|
||||||
|
services:
|
||||||
|
# 🚀 Claude Relay Service
|
||||||
|
claude-relay:
|
||||||
|
build: .
|
||||||
|
container_name: claude-relay-service
|
||||||
|
restart: unless-stopped
|
||||||
|
ports:
|
||||||
|
- "${PORT:-3000}:3000"
|
||||||
|
environment:
|
||||||
|
- NODE_ENV=production
|
||||||
|
- PORT=3000
|
||||||
|
- REDIS_HOST=redis
|
||||||
|
- REDIS_PORT=6379
|
||||||
|
volumes:
|
||||||
|
- ./logs:/app/logs
|
||||||
|
- ./data:/app/data
|
||||||
|
depends_on:
|
||||||
|
- redis
|
||||||
|
networks:
|
||||||
|
- claude-relay-network
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD", "curl", "-f", "http://localhost:3000/health"]
|
||||||
|
interval: 30s
|
||||||
|
timeout: 10s
|
||||||
|
retries: 3
|
||||||
|
|
||||||
|
# 📊 Redis Database
|
||||||
|
redis:
|
||||||
|
image: redis:7-alpine
|
||||||
|
container_name: claude-relay-redis
|
||||||
|
restart: unless-stopped
|
||||||
|
ports:
|
||||||
|
- "${REDIS_PORT:-6379}:6379"
|
||||||
|
volumes:
|
||||||
|
- redis_data:/data
|
||||||
|
- ./config/redis.conf:/usr/local/etc/redis/redis.conf:ro
|
||||||
|
command: redis-server /usr/local/etc/redis/redis.conf
|
||||||
|
networks:
|
||||||
|
- claude-relay-network
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD", "redis-cli", "ping"]
|
||||||
|
interval: 30s
|
||||||
|
timeout: 10s
|
||||||
|
retries: 3
|
||||||
|
|
||||||
|
# 📈 Redis Monitoring (Optional)
|
||||||
|
redis-commander:
|
||||||
|
image: rediscommander/redis-commander:latest
|
||||||
|
container_name: claude-relay-redis-web
|
||||||
|
restart: unless-stopped
|
||||||
|
ports:
|
||||||
|
- "${REDIS_WEB_PORT:-8081}:8081"
|
||||||
|
environment:
|
||||||
|
- REDIS_HOSTS=local:redis:6379
|
||||||
|
depends_on:
|
||||||
|
- redis
|
||||||
|
networks:
|
||||||
|
- claude-relay-network
|
||||||
|
profiles:
|
||||||
|
- monitoring
|
||||||
|
|
||||||
|
# 📊 Application Monitoring (Optional)
|
||||||
|
prometheus:
|
||||||
|
image: prom/prometheus:latest
|
||||||
|
container_name: claude-relay-prometheus
|
||||||
|
restart: unless-stopped
|
||||||
|
ports:
|
||||||
|
- "${PROMETHEUS_PORT:-9090}:9090"
|
||||||
|
volumes:
|
||||||
|
- ./config/prometheus.yml:/etc/prometheus/prometheus.yml:ro
|
||||||
|
- prometheus_data:/prometheus
|
||||||
|
command:
|
||||||
|
- '--config.file=/etc/prometheus/prometheus.yml'
|
||||||
|
- '--storage.tsdb.path=/prometheus'
|
||||||
|
- '--web.console.libraries=/etc/prometheus/console_libraries'
|
||||||
|
- '--web.console.templates=/etc/prometheus/consoles'
|
||||||
|
- '--web.enable-lifecycle'
|
||||||
|
networks:
|
||||||
|
- claude-relay-network
|
||||||
|
profiles:
|
||||||
|
- monitoring
|
||||||
|
|
||||||
|
# 📈 Grafana Dashboard (Optional)
|
||||||
|
grafana:
|
||||||
|
image: grafana/grafana:latest
|
||||||
|
container_name: claude-relay-grafana
|
||||||
|
restart: unless-stopped
|
||||||
|
ports:
|
||||||
|
- "${GRAFANA_PORT:-3001}:3000"
|
||||||
|
environment:
|
||||||
|
- GF_SECURITY_ADMIN_PASSWORD=${GRAFANA_ADMIN_PASSWORD:-admin123}
|
||||||
|
volumes:
|
||||||
|
- grafana_data:/var/lib/grafana
|
||||||
|
- ./config/grafana:/etc/grafana/provisioning
|
||||||
|
depends_on:
|
||||||
|
- prometheus
|
||||||
|
networks:
|
||||||
|
- claude-relay-network
|
||||||
|
profiles:
|
||||||
|
- monitoring
|
||||||
|
|
||||||
|
volumes:
|
||||||
|
redis_data:
|
||||||
|
driver: local
|
||||||
|
prometheus_data:
|
||||||
|
driver: local
|
||||||
|
grafana_data:
|
||||||
|
driver: local
|
||||||
|
|
||||||
|
networks:
|
||||||
|
claude-relay-network:
|
||||||
|
driver: bridge
|
||||||
|
ipam:
|
||||||
|
config:
|
||||||
|
- subnet: 172.20.0.0/16
|
||||||
7555
package-lock.json
generated
Normal file
7555
package-lock.json
generated
Normal file
File diff suppressed because it is too large
Load Diff
72
package.json
Normal file
72
package.json
Normal file
@@ -0,0 +1,72 @@
|
|||||||
|
{
|
||||||
|
"name": "claude-relay-service",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"description": "Claude Code API relay service with multi-account management and API key authentication",
|
||||||
|
"main": "src/app.js",
|
||||||
|
"scripts": {
|
||||||
|
"start": "node src/app.js",
|
||||||
|
"dev": "nodemon src/app.js",
|
||||||
|
"build:web": "cd web && npm run build",
|
||||||
|
"install:web": "cd web && npm install",
|
||||||
|
"setup": "node scripts/setup.js",
|
||||||
|
"cli": "node cli/index.js",
|
||||||
|
"service": "node scripts/manage.js",
|
||||||
|
"service:start": "node scripts/manage.js start",
|
||||||
|
"service:start:daemon": "node scripts/manage.js start -d",
|
||||||
|
"service:start:d": "node scripts/manage.js start -d",
|
||||||
|
"service:daemon": "node scripts/manage.js start -d",
|
||||||
|
"service:stop": "node scripts/manage.js stop",
|
||||||
|
"service:restart": "node scripts/manage.js restart",
|
||||||
|
"service:restart:daemon": "node scripts/manage.js restart -d",
|
||||||
|
"service:restart:d": "node scripts/manage.js restart -d",
|
||||||
|
"service:status": "node scripts/manage.js status",
|
||||||
|
"service:logs": "node scripts/manage.js logs",
|
||||||
|
"test": "jest",
|
||||||
|
"lint": "eslint src/**/*.js",
|
||||||
|
"docker:build": "docker build -t claude-relay-service .",
|
||||||
|
"docker:up": "docker-compose up -d",
|
||||||
|
"docker:down": "docker-compose down"
|
||||||
|
},
|
||||||
|
"dependencies": {
|
||||||
|
"axios": "^1.6.0",
|
||||||
|
"bcryptjs": "^2.4.3",
|
||||||
|
"chalk": "^4.1.2",
|
||||||
|
"commander": "^11.1.0",
|
||||||
|
"compression": "^1.7.4",
|
||||||
|
"cors": "^2.8.5",
|
||||||
|
"dotenv": "^16.3.1",
|
||||||
|
"express": "^4.18.2",
|
||||||
|
"helmet": "^7.1.0",
|
||||||
|
"https-proxy-agent": "^7.0.2",
|
||||||
|
"inquirer": "^9.2.15",
|
||||||
|
"ioredis": "^5.3.2",
|
||||||
|
"morgan": "^1.10.0",
|
||||||
|
"ora": "^5.4.1",
|
||||||
|
"rate-limiter-flexible": "^5.0.5",
|
||||||
|
"socks-proxy-agent": "^8.0.2",
|
||||||
|
"table": "^6.8.1",
|
||||||
|
"uuid": "^9.0.1",
|
||||||
|
"winston": "^3.11.0",
|
||||||
|
"winston-daily-rotate-file": "^4.7.1"
|
||||||
|
},
|
||||||
|
"devDependencies": {
|
||||||
|
"@types/node": "^20.8.9",
|
||||||
|
"eslint": "^8.53.0",
|
||||||
|
"jest": "^29.7.0",
|
||||||
|
"nodemon": "^3.0.1",
|
||||||
|
"supertest": "^6.3.3"
|
||||||
|
},
|
||||||
|
"engines": {
|
||||||
|
"node": ">=18.0.0"
|
||||||
|
},
|
||||||
|
"keywords": [
|
||||||
|
"claude",
|
||||||
|
"api",
|
||||||
|
"proxy",
|
||||||
|
"relay",
|
||||||
|
"claude-code",
|
||||||
|
"anthropic"
|
||||||
|
],
|
||||||
|
"author": "Claude Relay Service",
|
||||||
|
"license": "MIT"
|
||||||
|
}
|
||||||
335
scripts/manage.js
Normal file
335
scripts/manage.js
Normal file
@@ -0,0 +1,335 @@
|
|||||||
|
#!/usr/bin/env node
|
||||||
|
|
||||||
|
const { spawn, exec } = require('child_process');
|
||||||
|
const fs = require('fs');
|
||||||
|
const path = require('path');
|
||||||
|
const process = require('process');
|
||||||
|
|
||||||
|
const PID_FILE = path.join(__dirname, '..', 'claude-relay-service.pid');
|
||||||
|
const LOG_FILE = path.join(__dirname, '..', 'logs', 'service.log');
|
||||||
|
const ERROR_LOG_FILE = path.join(__dirname, '..', 'logs', 'service-error.log');
|
||||||
|
const APP_FILE = path.join(__dirname, '..', 'src', 'app.js');
|
||||||
|
|
||||||
|
class ServiceManager {
|
||||||
|
constructor() {
|
||||||
|
this.ensureLogDir();
|
||||||
|
}
|
||||||
|
|
||||||
|
ensureLogDir() {
|
||||||
|
const logDir = path.dirname(LOG_FILE);
|
||||||
|
if (!fs.existsSync(logDir)) {
|
||||||
|
fs.mkdirSync(logDir, { recursive: true });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
getPid() {
|
||||||
|
try {
|
||||||
|
if (fs.existsSync(PID_FILE)) {
|
||||||
|
const pid = parseInt(fs.readFileSync(PID_FILE, 'utf8').trim());
|
||||||
|
return pid;
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('读取PID文件失败:', error.message);
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
isProcessRunning(pid) {
|
||||||
|
try {
|
||||||
|
process.kill(pid, 0);
|
||||||
|
return true;
|
||||||
|
} catch (error) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
writePid(pid) {
|
||||||
|
try {
|
||||||
|
fs.writeFileSync(PID_FILE, pid.toString());
|
||||||
|
console.log(`✅ PID ${pid} 已保存到 ${PID_FILE}`);
|
||||||
|
} catch (error) {
|
||||||
|
console.error('写入PID文件失败:', error.message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
removePidFile() {
|
||||||
|
try {
|
||||||
|
if (fs.existsSync(PID_FILE)) {
|
||||||
|
fs.unlinkSync(PID_FILE);
|
||||||
|
console.log('🗑️ 已清理PID文件');
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('清理PID文件失败:', error.message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
getStatus() {
|
||||||
|
const pid = this.getPid();
|
||||||
|
if (pid && this.isProcessRunning(pid)) {
|
||||||
|
return { running: true, pid };
|
||||||
|
}
|
||||||
|
return { running: false, pid: null };
|
||||||
|
}
|
||||||
|
|
||||||
|
start(daemon = false) {
|
||||||
|
const status = this.getStatus();
|
||||||
|
if (status.running) {
|
||||||
|
console.log(`⚠️ 服务已在运行中 (PID: ${status.pid})`);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('🚀 启动 Claude Relay Service...');
|
||||||
|
|
||||||
|
if (daemon) {
|
||||||
|
// 后台运行模式 - 使用nohup实现真正的后台运行
|
||||||
|
const { exec } = require('child_process');
|
||||||
|
|
||||||
|
const command = `nohup node "${APP_FILE}" > "${LOG_FILE}" 2> "${ERROR_LOG_FILE}" & echo $!`;
|
||||||
|
|
||||||
|
exec(command, (error, stdout, stderr) => {
|
||||||
|
if (error) {
|
||||||
|
console.error('❌ 后台启动失败:', error.message);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const pid = parseInt(stdout.trim());
|
||||||
|
if (pid && !isNaN(pid)) {
|
||||||
|
this.writePid(pid);
|
||||||
|
console.log(`🔄 服务已在后台启动 (PID: ${pid})`);
|
||||||
|
console.log(`📝 日志文件: ${LOG_FILE}`);
|
||||||
|
console.log(`❌ 错误日志: ${ERROR_LOG_FILE}`);
|
||||||
|
console.log('✅ 终端现在可以安全关闭');
|
||||||
|
} else {
|
||||||
|
console.error('❌ 无法获取进程ID');
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 给exec一点时间执行
|
||||||
|
setTimeout(() => {
|
||||||
|
process.exit(0);
|
||||||
|
}, 1000);
|
||||||
|
|
||||||
|
} else {
|
||||||
|
// 前台运行模式
|
||||||
|
const child = spawn('node', [APP_FILE], {
|
||||||
|
stdio: 'inherit'
|
||||||
|
});
|
||||||
|
|
||||||
|
console.log(`🔄 服务已启动 (PID: ${child.pid})`);
|
||||||
|
|
||||||
|
this.writePid(child.pid);
|
||||||
|
|
||||||
|
// 监听进程退出
|
||||||
|
child.on('exit', (code, signal) => {
|
||||||
|
this.removePidFile();
|
||||||
|
if (code !== 0) {
|
||||||
|
console.log(`💥 进程退出 (代码: ${code}, 信号: ${signal})`);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
child.on('error', (error) => {
|
||||||
|
console.error('❌ 启动失败:', error.message);
|
||||||
|
this.removePidFile();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
stop() {
|
||||||
|
const status = this.getStatus();
|
||||||
|
if (!status.running) {
|
||||||
|
console.log('⚠️ 服务未在运行');
|
||||||
|
this.removePidFile(); // 清理可能存在的过期PID文件
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`🛑 停止服务 (PID: ${status.pid})...`);
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 优雅关闭：先发送SIGTERM
|
||||||
|
process.kill(status.pid, 'SIGTERM');
|
||||||
|
|
||||||
|
// 等待进程退出
|
||||||
|
let attempts = 0;
|
||||||
|
const maxAttempts = 30; // 30秒超时
|
||||||
|
|
||||||
|
const checkExit = setInterval(() => {
|
||||||
|
attempts++;
|
||||||
|
if (!this.isProcessRunning(status.pid)) {
|
||||||
|
clearInterval(checkExit);
|
||||||
|
console.log('✅ 服务已停止');
|
||||||
|
this.removePidFile();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (attempts >= maxAttempts) {
|
||||||
|
clearInterval(checkExit);
|
||||||
|
console.log('⚠️ 优雅关闭超时，强制终止进程...');
|
||||||
|
try {
|
||||||
|
process.kill(status.pid, 'SIGKILL');
|
||||||
|
console.log('✅ 服务已强制停止');
|
||||||
|
} catch (error) {
|
||||||
|
console.error('❌ 强制停止失败:', error.message);
|
||||||
|
}
|
||||||
|
this.removePidFile();
|
||||||
|
}
|
||||||
|
}, 1000);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
console.error('❌ 停止服务失败:', error.message);
|
||||||
|
this.removePidFile();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
restart(daemon = false) {
|
||||||
|
console.log('🔄 重启服务...');
|
||||||
|
const stopResult = this.stop();
|
||||||
|
|
||||||
|
// 等待停止完成
|
||||||
|
setTimeout(() => {
|
||||||
|
this.start(daemon);
|
||||||
|
}, 2000);
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
status() {
|
||||||
|
const status = this.getStatus();
|
||||||
|
if (status.running) {
|
||||||
|
console.log(`✅ 服务正在运行 (PID: ${status.pid})`);
|
||||||
|
|
||||||
|
// 显示进程信息
|
||||||
|
exec(`ps -p ${status.pid} -o pid,ppid,pcpu,pmem,etime,cmd --no-headers`, (error, stdout) => {
|
||||||
|
if (!error && stdout.trim()) {
|
||||||
|
console.log('\n📊 进程信息:');
|
||||||
|
console.log('PID\tPPID\tCPU%\tMEM%\tTIME\t\tCOMMAND');
|
||||||
|
console.log(stdout.trim());
|
||||||
|
}
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
console.log('❌ 服务未运行');
|
||||||
|
}
|
||||||
|
return status.running;
|
||||||
|
}
|
||||||
|
|
||||||
|
logs(lines = 50) {
|
||||||
|
console.log(`📖 最近 ${lines} 行日志:\n`);
|
||||||
|
|
||||||
|
exec(`tail -n ${lines} ${LOG_FILE}`, (error, stdout) => {
|
||||||
|
if (error) {
|
||||||
|
console.error('读取日志失败:', error.message);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
console.log(stdout);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
help() {
|
||||||
|
console.log(`
|
||||||
|
🔧 Claude Relay Service 进程管理器
|
||||||
|
|
||||||
|
用法: npm run service <command> [options]
|
||||||
|
|
||||||
|
重要提示：
|
||||||
|
如果要传递参数，请在npm run命令中使用 -- 分隔符
|
||||||
|
npm run service <command> -- [options]
|
||||||
|
|
||||||
|
命令:
|
||||||
|
start [-d|--daemon] 启动服务 (-d: 后台运行)
|
||||||
|
stop 停止服务
|
||||||
|
restart [-d|--daemon] 重启服务 (-d: 后台运行)
|
||||||
|
status 查看服务状态
|
||||||
|
logs [lines] 查看日志 (默认50行)
|
||||||
|
help 显示帮助信息
|
||||||
|
|
||||||
|
命令缩写:
|
||||||
|
s, start 启动服务
|
||||||
|
r, restart 重启服务
|
||||||
|
st, status 查看状态
|
||||||
|
l, log, logs 查看日志
|
||||||
|
halt, stop 停止服务
|
||||||
|
h, help 显示帮助
|
||||||
|
|
||||||
|
示例:
|
||||||
|
npm run service start # 前台启动
|
||||||
|
npm run service -- start -d # 后台启动（正确方式）
|
||||||
|
npm run service:start:d # 后台启动（推荐快捷方式）
|
||||||
|
npm run service:daemon # 后台启动（推荐快捷方式）
|
||||||
|
npm run service stop # 停止服务
|
||||||
|
npm run service -- restart -d # 后台重启（正确方式）
|
||||||
|
npm run service:restart:d # 后台重启（推荐快捷方式）
|
||||||
|
npm run service status # 查看状态
|
||||||
|
npm run service logs # 查看日志
|
||||||
|
npm run service -- logs 100 # 查看最近100行日志
|
||||||
|
|
||||||
|
推荐的快捷方式（无需 -- 分隔符）:
|
||||||
|
npm run service:start:d # 等同于 npm run service -- start -d
|
||||||
|
npm run service:restart:d # 等同于 npm run service -- restart -d
|
||||||
|
npm run service:daemon # 等同于 npm run service -- start -d
|
||||||
|
|
||||||
|
直接使用脚本（推荐）:
|
||||||
|
node scripts/manage.js start -d # 后台启动
|
||||||
|
node scripts/manage.js restart -d # 后台重启
|
||||||
|
node scripts/manage.js status # 查看状态
|
||||||
|
node scripts/manage.js logs 100 # 查看最近100行日志
|
||||||
|
|
||||||
|
文件位置:
|
||||||
|
PID文件: ${PID_FILE}
|
||||||
|
日志文件: ${LOG_FILE}
|
||||||
|
错误日志: ${ERROR_LOG_FILE}
|
||||||
|
`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 主程序
|
||||||
|
function main() {
|
||||||
|
const manager = new ServiceManager();
|
||||||
|
const args = process.argv.slice(2);
|
||||||
|
const command = args[0];
|
||||||
|
const isDaemon = args.includes('-d') || args.includes('--daemon');
|
||||||
|
|
||||||
|
switch (command) {
|
||||||
|
case 'start':
|
||||||
|
case 's':
|
||||||
|
manager.start(isDaemon);
|
||||||
|
break;
|
||||||
|
case 'stop':
|
||||||
|
case 'halt':
|
||||||
|
manager.stop();
|
||||||
|
break;
|
||||||
|
case 'restart':
|
||||||
|
case 'r':
|
||||||
|
manager.restart(isDaemon);
|
||||||
|
break;
|
||||||
|
case 'status':
|
||||||
|
case 'st':
|
||||||
|
manager.status();
|
||||||
|
break;
|
||||||
|
case 'logs':
|
||||||
|
case 'log':
|
||||||
|
case 'l':
|
||||||
|
const lines = parseInt(args[1]) || 50;
|
||||||
|
manager.logs(lines);
|
||||||
|
break;
|
||||||
|
case 'help':
|
||||||
|
case '--help':
|
||||||
|
case '-h':
|
||||||
|
case 'h':
|
||||||
|
manager.help();
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
console.log('❌ 未知命令:', command);
|
||||||
|
manager.help();
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (require.main === module) {
|
||||||
|
main();
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = ServiceManager;
|
||||||
107
scripts/setup.js
Normal file
107
scripts/setup.js
Normal file
@@ -0,0 +1,107 @@
|
|||||||
|
const fs = require('fs');
|
||||||
|
const path = require('path');
|
||||||
|
const crypto = require('crypto');
|
||||||
|
const bcrypt = require('bcryptjs');
|
||||||
|
const chalk = require('chalk');
|
||||||
|
const ora = require('ora');
|
||||||
|
|
||||||
|
const config = require('../config/config');
|
||||||
|
|
||||||
|
async function setup() {
|
||||||
|
console.log(chalk.blue.bold('\n🚀 Claude Relay Service 初始化设置\n'));
|
||||||
|
|
||||||
|
const spinner = ora('正在进行初始化设置...').start();
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 1. 创建必要目录
|
||||||
|
const directories = [
|
||||||
|
'logs',
|
||||||
|
'data',
|
||||||
|
'temp'
|
||||||
|
];
|
||||||
|
|
||||||
|
directories.forEach(dir => {
|
||||||
|
const dirPath = path.join(__dirname, '..', dir);
|
||||||
|
if (!fs.existsSync(dirPath)) {
|
||||||
|
fs.mkdirSync(dirPath, { recursive: true });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 2. 生成环境配置文件
|
||||||
|
if (!fs.existsSync(path.join(__dirname, '..', '.env'))) {
|
||||||
|
const envTemplate = fs.readFileSync(path.join(__dirname, '..', '.env.example'), 'utf8');
|
||||||
|
|
||||||
|
// 生成随机密钥
|
||||||
|
const jwtSecret = crypto.randomBytes(64).toString('hex');
|
||||||
|
const encryptionKey = crypto.randomBytes(32).toString('hex');
|
||||||
|
|
||||||
|
const envContent = envTemplate
|
||||||
|
.replace('your-jwt-secret-here', jwtSecret)
|
||||||
|
.replace('your-encryption-key-here', encryptionKey);
|
||||||
|
|
||||||
|
fs.writeFileSync(path.join(__dirname, '..', '.env'), envContent);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. 生成随机管理员凭据
|
||||||
|
const adminUsername = `cr_admin_${crypto.randomBytes(4).toString('hex')}`;
|
||||||
|
const adminPassword = crypto.randomBytes(16).toString('base64').replace(/[^a-zA-Z0-9]/g, '').substring(0, 16);
|
||||||
|
|
||||||
|
// 4. 创建初始化完成标记文件
|
||||||
|
const initData = {
|
||||||
|
initializedAt: new Date().toISOString(),
|
||||||
|
adminUsername,
|
||||||
|
adminPassword,
|
||||||
|
version: '1.0.0'
|
||||||
|
};
|
||||||
|
|
||||||
|
fs.writeFileSync(
|
||||||
|
path.join(__dirname, '..', 'data', 'init.json'),
|
||||||
|
JSON.stringify(initData, null, 2)
|
||||||
|
);
|
||||||
|
|
||||||
|
spinner.succeed('初始化设置完成');
|
||||||
|
|
||||||
|
console.log(chalk.green('\n✅ 设置完成！\n'));
|
||||||
|
console.log(chalk.yellow('📋 重要信息：\n'));
|
||||||
|
console.log(` 管理员用户名: ${chalk.cyan(adminUsername)}`);
|
||||||
|
console.log(` 管理员密码: ${chalk.cyan(adminPassword)}`);
|
||||||
|
console.log(chalk.red('\n⚠️ 请立即保存这些凭据！首次登录后建议修改密码。\n'));
|
||||||
|
|
||||||
|
console.log(chalk.blue('🚀 启动服务：\n'));
|
||||||
|
console.log(' npm start - 启动生产服务');
|
||||||
|
console.log(' npm run dev - 启动开发服务');
|
||||||
|
console.log(' npm run cli admin - 管理员CLI工具\n');
|
||||||
|
|
||||||
|
console.log(chalk.blue('🌐 访问地址：\n'));
|
||||||
|
console.log(` Web管理界面: http://localhost:${config.server.port}/web`);
|
||||||
|
console.log(` API端点: http://localhost:${config.server.port}/api/v1/messages`);
|
||||||
|
console.log(` 健康检查: http://localhost:${config.server.port}/health\n`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
spinner.fail('初始化设置失败');
|
||||||
|
console.error(chalk.red('❌ 错误:'), error.message);
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否已初始化
|
||||||
|
function checkInitialized() {
|
||||||
|
const initFile = path.join(__dirname, '..', 'data', 'init.json');
|
||||||
|
if (fs.existsSync(initFile)) {
|
||||||
|
const initData = JSON.parse(fs.readFileSync(initFile, 'utf8'));
|
||||||
|
console.log(chalk.yellow('⚠️ 服务已经初始化过了！'));
|
||||||
|
console.log(` 初始化时间: ${new Date(initData.initializedAt).toLocaleString()}`);
|
||||||
|
console.log(` 管理员用户名: ${initData.adminUsername}`);
|
||||||
|
console.log('\n如需重新初始化，请删除 data/init.json 文件。');
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (require.main === module) {
|
||||||
|
if (!checkInitialized()) {
|
||||||
|
setup();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = { setup, checkInitialized };
|
||||||
367
src/app.js
Normal file
367
src/app.js
Normal file
@@ -0,0 +1,367 @@
|
|||||||
|
const express = require('express');
|
||||||
|
const cors = require('cors');
|
||||||
|
const helmet = require('helmet');
|
||||||
|
const morgan = require('morgan');
|
||||||
|
const compression = require('compression');
|
||||||
|
const path = require('path');
|
||||||
|
const fs = require('fs');
|
||||||
|
const bcrypt = require('bcryptjs');
|
||||||
|
|
||||||
|
const config = require('../config/config');
|
||||||
|
const logger = require('./utils/logger');
|
||||||
|
const redis = require('./models/redis');
|
||||||
|
const pricingService = require('./services/pricingService');
|
||||||
|
|
||||||
|
// Import routes
|
||||||
|
const apiRoutes = require('./routes/api');
|
||||||
|
const adminRoutes = require('./routes/admin');
|
||||||
|
const webRoutes = require('./routes/web');
|
||||||
|
|
||||||
|
// Import middleware
|
||||||
|
const {
|
||||||
|
corsMiddleware,
|
||||||
|
requestLogger,
|
||||||
|
securityMiddleware,
|
||||||
|
errorHandler,
|
||||||
|
globalRateLimit,
|
||||||
|
requestSizeLimit
|
||||||
|
} = require('./middleware/auth');
|
||||||
|
|
||||||
|
class Application {
|
||||||
|
constructor() {
|
||||||
|
this.app = express();
|
||||||
|
this.server = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
async initialize() {
|
||||||
|
try {
|
||||||
|
// 🔗 连接Redis
|
||||||
|
logger.info('🔄 Connecting to Redis...');
|
||||||
|
await redis.connect();
|
||||||
|
logger.success('✅ Redis connected successfully');
|
||||||
|
|
||||||
|
// 💰 初始化价格服务
|
||||||
|
logger.info('🔄 Initializing pricing service...');
|
||||||
|
await pricingService.initialize();
|
||||||
|
|
||||||
|
// 🔧 初始化管理员凭据
|
||||||
|
logger.info('🔄 Initializing admin credentials...');
|
||||||
|
await this.initializeAdmin();
|
||||||
|
|
||||||
|
// 🛡️ 安全中间件
|
||||||
|
this.app.use(helmet({
|
||||||
|
contentSecurityPolicy: false, // 允许内联样式和脚本
|
||||||
|
crossOriginEmbedderPolicy: false
|
||||||
|
}));
|
||||||
|
|
||||||
|
// 🌐 CORS
|
||||||
|
if (config.web.enableCors) {
|
||||||
|
this.app.use(cors());
|
||||||
|
} else {
|
||||||
|
this.app.use(corsMiddleware);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📦 压缩
|
||||||
|
this.app.use(compression());
|
||||||
|
|
||||||
|
// 🚦 全局速率限制（仅在生产环境启用）
|
||||||
|
if (process.env.NODE_ENV === 'production') {
|
||||||
|
this.app.use(globalRateLimit);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📏 请求大小限制
|
||||||
|
this.app.use(requestSizeLimit);
|
||||||
|
|
||||||
|
// 📝 请求日志（使用自定义logger而不是morgan）
|
||||||
|
this.app.use(requestLogger);
|
||||||
|
|
||||||
|
// 🔧 基础中间件
|
||||||
|
this.app.use(express.json({
|
||||||
|
limit: '10mb',
|
||||||
|
verify: (req, res, buf, encoding) => {
|
||||||
|
// 验证JSON格式
|
||||||
|
if (buf && buf.length && !buf.toString(encoding || 'utf8').trim()) {
|
||||||
|
throw new Error('Invalid JSON: empty body');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}));
|
||||||
|
this.app.use(express.urlencoded({ extended: true, limit: '10mb' }));
|
||||||
|
this.app.use(securityMiddleware);
|
||||||
|
|
||||||
|
// 🎯 信任代理
|
||||||
|
if (config.server.trustProxy) {
|
||||||
|
this.app.set('trust proxy', 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🛣️ 路由
|
||||||
|
this.app.use('/api', apiRoutes);
|
||||||
|
this.app.use('/admin', adminRoutes);
|
||||||
|
this.app.use('/web', webRoutes);
|
||||||
|
|
||||||
|
// 🏠 根路径重定向到管理界面
|
||||||
|
this.app.get('/', (req, res) => {
|
||||||
|
res.redirect('/web');
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🏥 增强的健康检查端点
|
||||||
|
this.app.get('/health', async (req, res) => {
|
||||||
|
try {
|
||||||
|
const timer = logger.timer('health-check');
|
||||||
|
|
||||||
|
// 检查各个组件健康状态
|
||||||
|
const [redisHealth, loggerHealth] = await Promise.all([
|
||||||
|
this.checkRedisHealth(),
|
||||||
|
this.checkLoggerHealth()
|
||||||
|
]);
|
||||||
|
|
||||||
|
const memory = process.memoryUsage();
|
||||||
|
const health = {
|
||||||
|
status: 'healthy',
|
||||||
|
service: 'claude-relay-service',
|
||||||
|
version: '1.0.0',
|
||||||
|
timestamp: new Date().toISOString(),
|
||||||
|
uptime: process.uptime(),
|
||||||
|
memory: {
|
||||||
|
used: Math.round(memory.heapUsed / 1024 / 1024) + 'MB',
|
||||||
|
total: Math.round(memory.heapTotal / 1024 / 1024) + 'MB',
|
||||||
|
external: Math.round(memory.external / 1024 / 1024) + 'MB'
|
||||||
|
},
|
||||||
|
components: {
|
||||||
|
redis: redisHealth,
|
||||||
|
logger: loggerHealth
|
||||||
|
},
|
||||||
|
stats: logger.getStats()
|
||||||
|
};
|
||||||
|
|
||||||
|
timer.end('completed');
|
||||||
|
res.json(health);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Health check failed:', { error: error.message, stack: error.stack });
|
||||||
|
res.status(503).json({
|
||||||
|
status: 'unhealthy',
|
||||||
|
error: error.message,
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 📊 指标端点
|
||||||
|
this.app.get('/metrics', async (req, res) => {
|
||||||
|
try {
|
||||||
|
const stats = await redis.getSystemStats();
|
||||||
|
const metrics = {
|
||||||
|
...stats,
|
||||||
|
uptime: process.uptime(),
|
||||||
|
memory: process.memoryUsage(),
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
};
|
||||||
|
|
||||||
|
res.json(metrics);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Metrics collection failed:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to collect metrics' });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🚫 404 处理
|
||||||
|
this.app.use('*', (req, res) => {
|
||||||
|
res.status(404).json({
|
||||||
|
error: 'Not Found',
|
||||||
|
message: `Route ${req.originalUrl} not found`,
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🚨 错误处理
|
||||||
|
this.app.use(errorHandler);
|
||||||
|
|
||||||
|
logger.success('✅ Application initialized successfully');
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('💥 Application initialization failed:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔧 初始化管理员凭据
|
||||||
|
async initializeAdmin() {
|
||||||
|
try {
|
||||||
|
// 检查Redis中是否已存在管理员凭据
|
||||||
|
const existingAdmin = await redis.getSession('admin_credentials');
|
||||||
|
|
||||||
|
if (!existingAdmin || Object.keys(existingAdmin).length === 0) {
|
||||||
|
// 尝试从初始化文件读取
|
||||||
|
const initFilePath = path.join(__dirname, '..', 'data', 'init.json');
|
||||||
|
|
||||||
|
if (fs.existsSync(initFilePath)) {
|
||||||
|
const initData = JSON.parse(fs.readFileSync(initFilePath, 'utf8'));
|
||||||
|
|
||||||
|
// 将明文密码哈希化
|
||||||
|
const saltRounds = 10;
|
||||||
|
const passwordHash = await bcrypt.hash(initData.adminPassword, saltRounds);
|
||||||
|
|
||||||
|
// 存储到Redis
|
||||||
|
const adminCredentials = {
|
||||||
|
username: initData.adminUsername,
|
||||||
|
passwordHash: passwordHash,
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
lastLogin: null
|
||||||
|
};
|
||||||
|
|
||||||
|
await redis.setSession('admin_credentials', adminCredentials);
|
||||||
|
|
||||||
|
logger.success('✅ Admin credentials initialized from setup data');
|
||||||
|
} else {
|
||||||
|
logger.warn('⚠️ No admin credentials found. Please run npm run setup first.');
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
logger.info('ℹ️ Admin credentials already exist in Redis');
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to initialize admin credentials:', { error: error.message, stack: error.stack });
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔍 Redis健康检查
|
||||||
|
async checkRedisHealth() {
|
||||||
|
try {
|
||||||
|
const start = Date.now();
|
||||||
|
await redis.getClient().ping();
|
||||||
|
const latency = Date.now() - start;
|
||||||
|
|
||||||
|
return {
|
||||||
|
status: 'healthy',
|
||||||
|
connected: redis.isConnected,
|
||||||
|
latency: `${latency}ms`
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
status: 'unhealthy',
|
||||||
|
connected: false,
|
||||||
|
error: error.message
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📝 Logger健康检查
|
||||||
|
async checkLoggerHealth() {
|
||||||
|
try {
|
||||||
|
const health = logger.healthCheck();
|
||||||
|
return {
|
||||||
|
status: health.healthy ? 'healthy' : 'unhealthy',
|
||||||
|
...health
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
return {
|
||||||
|
status: 'unhealthy',
|
||||||
|
error: error.message
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async start() {
|
||||||
|
try {
|
||||||
|
await this.initialize();
|
||||||
|
|
||||||
|
this.server = this.app.listen(config.server.port, config.server.host, () => {
|
||||||
|
logger.start(`🚀 Claude Relay Service started on ${config.server.host}:${config.server.port}`);
|
||||||
|
logger.info(`🌐 Web interface: http://${config.server.host}:${config.server.port}/web`);
|
||||||
|
logger.info(`🔗 API endpoint: http://${config.server.host}:${config.server.port}/api/v1/messages`);
|
||||||
|
logger.info(`⚙️ Admin API: http://${config.server.host}:${config.server.port}/admin`);
|
||||||
|
logger.info(`🏥 Health check: http://${config.server.host}:${config.server.port}/health`);
|
||||||
|
logger.info(`📊 Metrics: http://${config.server.host}:${config.server.port}/metrics`);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🔄 定期清理任务
|
||||||
|
this.startCleanupTasks();
|
||||||
|
|
||||||
|
// 🛑 优雅关闭
|
||||||
|
this.setupGracefulShutdown();
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('💥 Failed to start server:', error);
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
startCleanupTasks() {
|
||||||
|
// 🧹 每小时清理一次过期数据
|
||||||
|
setInterval(async () => {
|
||||||
|
try {
|
||||||
|
logger.info('🧹 Starting scheduled cleanup...');
|
||||||
|
|
||||||
|
const apiKeyService = require('./services/apiKeyService');
|
||||||
|
const claudeAccountService = require('./services/claudeAccountService');
|
||||||
|
|
||||||
|
const [expiredKeys, errorAccounts] = await Promise.all([
|
||||||
|
apiKeyService.cleanupExpiredKeys(),
|
||||||
|
claudeAccountService.cleanupErrorAccounts()
|
||||||
|
]);
|
||||||
|
|
||||||
|
await redis.cleanup();
|
||||||
|
|
||||||
|
logger.success(`🧹 Cleanup completed: ${expiredKeys} expired keys, ${errorAccounts} error accounts reset`);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Cleanup task failed:', error);
|
||||||
|
}
|
||||||
|
}, config.system.cleanupInterval);
|
||||||
|
|
||||||
|
logger.info(`🔄 Cleanup tasks scheduled every ${config.system.cleanupInterval / 1000 / 60} minutes`);
|
||||||
|
}
|
||||||
|
|
||||||
|
setupGracefulShutdown() {
|
||||||
|
const shutdown = async (signal) => {
|
||||||
|
logger.info(`🛑 Received ${signal}, starting graceful shutdown...`);
|
||||||
|
|
||||||
|
if (this.server) {
|
||||||
|
this.server.close(async () => {
|
||||||
|
logger.info('🚪 HTTP server closed');
|
||||||
|
|
||||||
|
try {
|
||||||
|
await redis.disconnect();
|
||||||
|
logger.info('👋 Redis disconnected');
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Error disconnecting Redis:', error);
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.success('✅ Graceful shutdown completed');
|
||||||
|
process.exit(0);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 强制关闭超时
|
||||||
|
setTimeout(() => {
|
||||||
|
logger.warn('⚠️ Forced shutdown due to timeout');
|
||||||
|
process.exit(1);
|
||||||
|
}, 10000);
|
||||||
|
} else {
|
||||||
|
process.exit(0);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
process.on('SIGTERM', () => shutdown('SIGTERM'));
|
||||||
|
process.on('SIGINT', () => shutdown('SIGINT'));
|
||||||
|
|
||||||
|
// 处理未捕获异常
|
||||||
|
process.on('uncaughtException', (error) => {
|
||||||
|
logger.error('💥 Uncaught exception:', error);
|
||||||
|
shutdown('uncaughtException');
|
||||||
|
});
|
||||||
|
|
||||||
|
process.on('unhandledRejection', (reason, promise) => {
|
||||||
|
logger.error('💥 Unhandled rejection at:', promise, 'reason:', reason);
|
||||||
|
shutdown('unhandledRejection');
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 启动应用
|
||||||
|
if (require.main === module) {
|
||||||
|
const app = new Application();
|
||||||
|
app.start().catch((error) => {
|
||||||
|
logger.error('💥 Application startup failed:', error);
|
||||||
|
process.exit(1);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = Application;
|
||||||
532
src/middleware/auth.js
Normal file
532
src/middleware/auth.js
Normal file
@@ -0,0 +1,532 @@
|
|||||||
|
const apiKeyService = require('../services/apiKeyService');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
const redis = require('../models/redis');
|
||||||
|
const { RateLimiterRedis } = require('rate-limiter-flexible');
|
||||||
|
|
||||||
|
// 🔑 API Key验证中间件（优化版）
|
||||||
|
const authenticateApiKey = async (req, res, next) => {
|
||||||
|
const startTime = Date.now();
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 安全提取API Key，支持多种格式
|
||||||
|
const apiKey = req.headers['x-api-key'] ||
|
||||||
|
req.headers['authorization']?.replace(/^Bearer\s+/i, '') ||
|
||||||
|
req.headers['api-key'];
|
||||||
|
|
||||||
|
if (!apiKey) {
|
||||||
|
logger.security(`🔒 Missing API key attempt from ${req.ip || 'unknown'}`);
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Missing API key',
|
||||||
|
message: 'Please provide an API key in the x-api-key header or Authorization header'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 基本API Key格式验证
|
||||||
|
if (typeof apiKey !== 'string' || apiKey.length < 10 || apiKey.length > 512) {
|
||||||
|
logger.security(`🔒 Invalid API key format from ${req.ip || 'unknown'}`);
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Invalid API key format',
|
||||||
|
message: 'API key format is invalid'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证API Key（带缓存优化）
|
||||||
|
const validation = await apiKeyService.validateApiKey(apiKey);
|
||||||
|
|
||||||
|
if (!validation.valid) {
|
||||||
|
const clientIP = req.ip || req.connection?.remoteAddress || 'unknown';
|
||||||
|
logger.security(`🔒 Invalid API key attempt: ${validation.error} from ${clientIP}`);
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Invalid API key',
|
||||||
|
message: validation.error
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查速率限制（优化：只在验证成功后检查）
|
||||||
|
const rateLimitResult = await apiKeyService.checkRateLimit(validation.keyData.id);
|
||||||
|
|
||||||
|
if (!rateLimitResult.allowed) {
|
||||||
|
logger.security(`🚦 Rate limit exceeded for key: ${validation.keyData.id} (${validation.keyData.name})`);
|
||||||
|
return res.status(429).json({
|
||||||
|
error: 'Rate limit exceeded',
|
||||||
|
message: `Too many requests. Limit: ${rateLimitResult.limit} requests per minute`,
|
||||||
|
resetTime: rateLimitResult.resetTime,
|
||||||
|
retryAfter: rateLimitResult.resetTime
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置标准速率限制响应头
|
||||||
|
res.setHeader('X-RateLimit-Limit', rateLimitResult.limit);
|
||||||
|
res.setHeader('X-RateLimit-Remaining', Math.max(0, rateLimitResult.limit - rateLimitResult.current));
|
||||||
|
res.setHeader('X-RateLimit-Reset', rateLimitResult.resetTime);
|
||||||
|
res.setHeader('X-RateLimit-Policy', `${rateLimitResult.limit};w=60`);
|
||||||
|
|
||||||
|
// 将验证信息添加到请求对象（只包含必要信息）
|
||||||
|
req.apiKey = {
|
||||||
|
id: validation.keyData.id,
|
||||||
|
name: validation.keyData.name,
|
||||||
|
tokenLimit: validation.keyData.tokenLimit,
|
||||||
|
requestLimit: validation.keyData.requestLimit,
|
||||||
|
claudeAccountId: validation.keyData.claudeAccountId
|
||||||
|
};
|
||||||
|
req.usage = validation.keyData.usage;
|
||||||
|
|
||||||
|
const authDuration = Date.now() - startTime;
|
||||||
|
logger.api(`🔓 Authenticated request from key: ${validation.keyData.name} (${validation.keyData.id}) in ${authDuration}ms`);
|
||||||
|
|
||||||
|
next();
|
||||||
|
} catch (error) {
|
||||||
|
const authDuration = Date.now() - startTime;
|
||||||
|
logger.error(`❌ Authentication middleware error (${authDuration}ms):`, {
|
||||||
|
error: error.message,
|
||||||
|
stack: error.stack,
|
||||||
|
ip: req.ip,
|
||||||
|
userAgent: req.get('User-Agent'),
|
||||||
|
url: req.originalUrl
|
||||||
|
});
|
||||||
|
|
||||||
|
res.status(500).json({
|
||||||
|
error: 'Authentication error',
|
||||||
|
message: 'Internal server error during authentication'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// 🛡️ 管理员验证中间件（优化版）
|
||||||
|
const authenticateAdmin = async (req, res, next) => {
|
||||||
|
const startTime = Date.now();
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 安全提取token，支持多种方式
|
||||||
|
const token = req.headers['authorization']?.replace(/^Bearer\s+/i, '') ||
|
||||||
|
req.cookies?.adminToken ||
|
||||||
|
req.headers['x-admin-token'];
|
||||||
|
|
||||||
|
if (!token) {
|
||||||
|
logger.security(`🔒 Missing admin token attempt from ${req.ip || 'unknown'}`);
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Missing admin token',
|
||||||
|
message: 'Please provide an admin token'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 基本token格式验证
|
||||||
|
if (typeof token !== 'string' || token.length < 32 || token.length > 512) {
|
||||||
|
logger.security(`🔒 Invalid admin token format from ${req.ip || 'unknown'}`);
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Invalid admin token format',
|
||||||
|
message: 'Admin token format is invalid'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取管理员会话（带超时处理）
|
||||||
|
const adminSession = await Promise.race([
|
||||||
|
redis.getSession(token),
|
||||||
|
new Promise((_, reject) =>
|
||||||
|
setTimeout(() => reject(new Error('Session lookup timeout')), 5000)
|
||||||
|
)
|
||||||
|
]);
|
||||||
|
|
||||||
|
if (!adminSession || Object.keys(adminSession).length === 0) {
|
||||||
|
logger.security(`🔒 Invalid admin token attempt from ${req.ip || 'unknown'}`);
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Invalid admin token',
|
||||||
|
message: 'Invalid or expired admin session'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查会话活跃性（可选：检查最后活动时间）
|
||||||
|
const now = new Date();
|
||||||
|
const lastActivity = new Date(adminSession.lastActivity || adminSession.loginTime);
|
||||||
|
const inactiveDuration = now - lastActivity;
|
||||||
|
const maxInactivity = 24 * 60 * 60 * 1000; // 24小时
|
||||||
|
|
||||||
|
if (inactiveDuration > maxInactivity) {
|
||||||
|
logger.security(`🔒 Expired admin session for ${adminSession.username} from ${req.ip || 'unknown'}`);
|
||||||
|
await redis.deleteSession(token); // 清理过期会话
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Session expired',
|
||||||
|
message: 'Admin session has expired due to inactivity'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新最后活动时间（异步，不阻塞请求）
|
||||||
|
redis.setSession(token, {
|
||||||
|
...adminSession,
|
||||||
|
lastActivity: now.toISOString()
|
||||||
|
}, 86400).catch(error => {
|
||||||
|
logger.error('Failed to update admin session activity:', error);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 设置管理员信息（只包含必要信息）
|
||||||
|
req.admin = {
|
||||||
|
id: adminSession.adminId || 'admin',
|
||||||
|
username: adminSession.username,
|
||||||
|
sessionId: token,
|
||||||
|
loginTime: adminSession.loginTime
|
||||||
|
};
|
||||||
|
|
||||||
|
const authDuration = Date.now() - startTime;
|
||||||
|
logger.security(`🔐 Admin authenticated: ${adminSession.username} in ${authDuration}ms`);
|
||||||
|
|
||||||
|
next();
|
||||||
|
} catch (error) {
|
||||||
|
const authDuration = Date.now() - startTime;
|
||||||
|
logger.error(`❌ Admin authentication error (${authDuration}ms):`, {
|
||||||
|
error: error.message,
|
||||||
|
ip: req.ip,
|
||||||
|
userAgent: req.get('User-Agent'),
|
||||||
|
url: req.originalUrl
|
||||||
|
});
|
||||||
|
|
||||||
|
res.status(500).json({
|
||||||
|
error: 'Authentication error',
|
||||||
|
message: 'Internal server error during admin authentication'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// 注意：使用统计现在直接在/api/v1/messages路由中处理，
|
||||||
|
// 以便从Claude API响应中提取真实的usage数据
|
||||||
|
|
||||||
|
// 🚦 CORS中间件（优化版）
|
||||||
|
const corsMiddleware = (req, res, next) => {
|
||||||
|
const origin = req.headers.origin;
|
||||||
|
|
||||||
|
// 允许的源（可以从配置文件读取）
|
||||||
|
const allowedOrigins = [
|
||||||
|
'http://localhost:3000',
|
||||||
|
'https://localhost:3000',
|
||||||
|
'http://127.0.0.1:3000',
|
||||||
|
'https://127.0.0.1:3000'
|
||||||
|
];
|
||||||
|
|
||||||
|
// 设置CORS头
|
||||||
|
if (allowedOrigins.includes(origin) || !origin) {
|
||||||
|
res.header('Access-Control-Allow-Origin', origin || '*');
|
||||||
|
}
|
||||||
|
|
||||||
|
res.header('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, OPTIONS');
|
||||||
|
res.header('Access-Control-Allow-Headers', [
|
||||||
|
'Origin',
|
||||||
|
'X-Requested-With',
|
||||||
|
'Content-Type',
|
||||||
|
'Accept',
|
||||||
|
'Authorization',
|
||||||
|
'x-api-key',
|
||||||
|
'api-key',
|
||||||
|
'x-admin-token'
|
||||||
|
].join(', '));
|
||||||
|
|
||||||
|
res.header('Access-Control-Expose-Headers', [
|
||||||
|
'X-RateLimit-Limit',
|
||||||
|
'X-RateLimit-Remaining',
|
||||||
|
'X-RateLimit-Reset',
|
||||||
|
'X-RateLimit-Policy'
|
||||||
|
].join(', '));
|
||||||
|
|
||||||
|
res.header('Access-Control-Max-Age', '86400'); // 24小时预检缓存
|
||||||
|
res.header('Access-Control-Allow-Credentials', 'true');
|
||||||
|
|
||||||
|
if (req.method === 'OPTIONS') {
|
||||||
|
res.status(204).end();
|
||||||
|
} else {
|
||||||
|
next();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// 📝 请求日志中间件（优化版）
|
||||||
|
const requestLogger = (req, res, next) => {
|
||||||
|
const start = Date.now();
|
||||||
|
const requestId = Math.random().toString(36).substring(2, 15);
|
||||||
|
|
||||||
|
// 添加请求ID到请求对象
|
||||||
|
req.requestId = requestId;
|
||||||
|
res.setHeader('X-Request-ID', requestId);
|
||||||
|
|
||||||
|
// 获取客户端信息
|
||||||
|
const clientIP = req.ip || req.connection?.remoteAddress || req.socket?.remoteAddress || 'unknown';
|
||||||
|
const userAgent = req.get('User-Agent') || 'unknown';
|
||||||
|
const referer = req.get('Referer') || 'none';
|
||||||
|
|
||||||
|
// 记录请求开始
|
||||||
|
if (req.originalUrl !== '/health') { // 避免健康检查日志过多
|
||||||
|
logger.request(`▶️ [${requestId}] ${req.method} ${req.originalUrl} | IP: ${clientIP}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
res.on('finish', () => {
|
||||||
|
const duration = Date.now() - start;
|
||||||
|
const contentLength = res.get('Content-Length') || '0';
|
||||||
|
|
||||||
|
// 构建日志元数据
|
||||||
|
const logMetadata = {
|
||||||
|
requestId,
|
||||||
|
method: req.method,
|
||||||
|
url: req.originalUrl,
|
||||||
|
status: res.statusCode,
|
||||||
|
duration,
|
||||||
|
contentLength,
|
||||||
|
ip: clientIP,
|
||||||
|
userAgent,
|
||||||
|
referer
|
||||||
|
};
|
||||||
|
|
||||||
|
// 根据状态码选择日志级别
|
||||||
|
if (res.statusCode >= 500) {
|
||||||
|
logger.error(`◀️ [${requestId}] ${req.method} ${req.originalUrl} | ${res.statusCode} | ${duration}ms | ${contentLength}B`, logMetadata);
|
||||||
|
} else if (res.statusCode >= 400) {
|
||||||
|
logger.warn(`◀️ [${requestId}] ${req.method} ${req.originalUrl} | ${res.statusCode} | ${duration}ms | ${contentLength}B`, logMetadata);
|
||||||
|
} else if (req.originalUrl !== '/health') {
|
||||||
|
logger.request(`◀️ [${requestId}] ${req.method} ${req.originalUrl} | ${res.statusCode} | ${duration}ms | ${contentLength}B`, logMetadata);
|
||||||
|
}
|
||||||
|
|
||||||
|
// API Key相关日志
|
||||||
|
if (req.apiKey) {
|
||||||
|
logger.api(`📱 [${requestId}] Request from ${req.apiKey.name} (${req.apiKey.id}) | ${duration}ms`);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 慢请求警告
|
||||||
|
if (duration > 5000) {
|
||||||
|
logger.warn(`🐌 [${requestId}] Slow request detected: ${duration}ms for ${req.method} ${req.originalUrl}`);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
res.on('error', (error) => {
|
||||||
|
const duration = Date.now() - start;
|
||||||
|
logger.error(`💥 [${requestId}] Response error after ${duration}ms:`, error);
|
||||||
|
});
|
||||||
|
|
||||||
|
next();
|
||||||
|
};
|
||||||
|
|
||||||
|
// 🛡️ 安全中间件（增强版）
|
||||||
|
const securityMiddleware = (req, res, next) => {
|
||||||
|
// 设置基础安全头
|
||||||
|
res.setHeader('X-Content-Type-Options', 'nosniff');
|
||||||
|
res.setHeader('X-Frame-Options', 'DENY');
|
||||||
|
res.setHeader('X-XSS-Protection', '1; mode=block');
|
||||||
|
res.setHeader('Referrer-Policy', 'strict-origin-when-cross-origin');
|
||||||
|
|
||||||
|
// 添加更多安全头
|
||||||
|
res.setHeader('X-DNS-Prefetch-Control', 'off');
|
||||||
|
res.setHeader('X-Download-Options', 'noopen');
|
||||||
|
res.setHeader('X-Permitted-Cross-Domain-Policies', 'none');
|
||||||
|
|
||||||
|
// Cross-Origin-Opener-Policy (仅对可信来源设置)
|
||||||
|
const host = req.get('host') || '';
|
||||||
|
const isLocalhost = host.includes('localhost') || host.includes('127.0.0.1') || host.includes('0.0.0.0');
|
||||||
|
const isHttps = req.secure || req.headers['x-forwarded-proto'] === 'https';
|
||||||
|
|
||||||
|
if (isLocalhost || isHttps) {
|
||||||
|
res.setHeader('Cross-Origin-Opener-Policy', 'same-origin');
|
||||||
|
res.setHeader('Cross-Origin-Resource-Policy', 'same-origin');
|
||||||
|
res.setHeader('Origin-Agent-Cluster', '?1');
|
||||||
|
}
|
||||||
|
|
||||||
|
// Content Security Policy (适用于web界面)
|
||||||
|
if (req.path.startsWith('/web') || req.path === '/') {
|
||||||
|
res.setHeader('Content-Security-Policy', [
|
||||||
|
'default-src \'self\'',
|
||||||
|
'script-src \'self\' \'unsafe-inline\' \'unsafe-eval\' https://unpkg.com https://cdn.tailwindcss.com https://cdnjs.cloudflare.com https://cdn.jsdelivr.net',
|
||||||
|
'style-src \'self\' \'unsafe-inline\' https://cdn.tailwindcss.com https://cdnjs.cloudflare.com',
|
||||||
|
'font-src \'self\' https://cdnjs.cloudflare.com',
|
||||||
|
'img-src \'self\' data:',
|
||||||
|
'connect-src \'self\'',
|
||||||
|
'frame-ancestors \'none\'',
|
||||||
|
'base-uri \'self\'',
|
||||||
|
'form-action \'self\''
|
||||||
|
].join('; '));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Strict Transport Security (HTTPS)
|
||||||
|
if (req.secure || req.headers['x-forwarded-proto'] === 'https') {
|
||||||
|
res.setHeader('Strict-Transport-Security', 'max-age=15552000; includeSubDomains');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 移除泄露服务器信息的头
|
||||||
|
res.removeHeader('X-Powered-By');
|
||||||
|
res.removeHeader('Server');
|
||||||
|
|
||||||
|
// 防止信息泄露
|
||||||
|
res.setHeader('Cache-Control', 'no-cache, no-store, must-revalidate');
|
||||||
|
res.setHeader('Pragma', 'no-cache');
|
||||||
|
res.setHeader('Expires', '0');
|
||||||
|
|
||||||
|
next();
|
||||||
|
};
|
||||||
|
|
||||||
|
// 🚨 错误处理中间件（增强版）
|
||||||
|
const errorHandler = (error, req, res, _next) => {
|
||||||
|
const requestId = req.requestId || 'unknown';
|
||||||
|
const isDevelopment = process.env.NODE_ENV === 'development';
|
||||||
|
|
||||||
|
// 记录详细错误信息
|
||||||
|
logger.error(`💥 [${requestId}] Unhandled error:`, {
|
||||||
|
error: error.message,
|
||||||
|
stack: error.stack,
|
||||||
|
url: req.originalUrl,
|
||||||
|
method: req.method,
|
||||||
|
ip: req.ip || 'unknown',
|
||||||
|
userAgent: req.get('User-Agent') || 'unknown',
|
||||||
|
apiKey: req.apiKey ? req.apiKey.id : 'none',
|
||||||
|
admin: req.admin ? req.admin.username : 'none'
|
||||||
|
});
|
||||||
|
|
||||||
|
// 确定HTTP状态码
|
||||||
|
let statusCode = 500;
|
||||||
|
let errorMessage = 'Internal Server Error';
|
||||||
|
let userMessage = 'Something went wrong';
|
||||||
|
|
||||||
|
if (error.status && error.status >= 400 && error.status < 600) {
|
||||||
|
statusCode = error.status;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 根据错误类型提供友好的错误消息
|
||||||
|
switch (error.name) {
|
||||||
|
case 'ValidationError':
|
||||||
|
statusCode = 400;
|
||||||
|
errorMessage = 'Validation Error';
|
||||||
|
userMessage = 'Invalid input data';
|
||||||
|
break;
|
||||||
|
case 'CastError':
|
||||||
|
statusCode = 400;
|
||||||
|
errorMessage = 'Cast Error';
|
||||||
|
userMessage = 'Invalid data format';
|
||||||
|
break;
|
||||||
|
case 'MongoError':
|
||||||
|
case 'RedisError':
|
||||||
|
statusCode = 503;
|
||||||
|
errorMessage = 'Database Error';
|
||||||
|
userMessage = 'Database temporarily unavailable';
|
||||||
|
break;
|
||||||
|
case 'TimeoutError':
|
||||||
|
statusCode = 408;
|
||||||
|
errorMessage = 'Request Timeout';
|
||||||
|
userMessage = 'Request took too long to process';
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
if (error.message && !isDevelopment) {
|
||||||
|
// 在生产环境中，只显示安全的错误消息
|
||||||
|
if (error.message.includes('ECONNREFUSED')) {
|
||||||
|
userMessage = 'Service temporarily unavailable';
|
||||||
|
} else if (error.message.includes('timeout')) {
|
||||||
|
userMessage = 'Request timeout';
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置响应头
|
||||||
|
res.setHeader('X-Request-ID', requestId);
|
||||||
|
|
||||||
|
// 构建错误响应
|
||||||
|
const errorResponse = {
|
||||||
|
error: errorMessage,
|
||||||
|
message: isDevelopment ? error.message : userMessage,
|
||||||
|
requestId,
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
};
|
||||||
|
|
||||||
|
// 在开发环境中包含更多调试信息
|
||||||
|
if (isDevelopment) {
|
||||||
|
errorResponse.stack = error.stack;
|
||||||
|
errorResponse.url = req.originalUrl;
|
||||||
|
errorResponse.method = req.method;
|
||||||
|
}
|
||||||
|
|
||||||
|
res.status(statusCode).json(errorResponse);
|
||||||
|
};
|
||||||
|
|
||||||
|
// 🌐 全局速率限制中间件（延迟初始化）
|
||||||
|
let rateLimiter = null;
|
||||||
|
|
||||||
|
const getRateLimiter = () => {
|
||||||
|
if (!rateLimiter) {
|
||||||
|
try {
|
||||||
|
const client = redis.getClient();
|
||||||
|
if (!client) {
|
||||||
|
logger.warn('⚠️ Redis client not available for rate limiter');
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
rateLimiter = new RateLimiterRedis({
|
||||||
|
storeClient: client,
|
||||||
|
keyPrefix: 'global_rate_limit',
|
||||||
|
points: 1000, // 请求数量
|
||||||
|
duration: 900, // 15分钟 (900秒)
|
||||||
|
blockDuration: 900, // 阻塞时间15分钟
|
||||||
|
});
|
||||||
|
|
||||||
|
logger.info('✅ Rate limiter initialized successfully');
|
||||||
|
} catch (error) {
|
||||||
|
logger.warn('⚠️ Rate limiter initialization failed, using fallback', { error: error.message });
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return rateLimiter;
|
||||||
|
};
|
||||||
|
|
||||||
|
const globalRateLimit = async (req, res, next) => {
|
||||||
|
// 跳过健康检查和内部请求
|
||||||
|
if (req.path === '/health' || req.path === '/api/health') {
|
||||||
|
return next();
|
||||||
|
}
|
||||||
|
|
||||||
|
const limiter = getRateLimiter();
|
||||||
|
if (!limiter) {
|
||||||
|
// 如果Redis不可用，直接跳过速率限制
|
||||||
|
return next();
|
||||||
|
}
|
||||||
|
|
||||||
|
const clientIP = req.ip || req.connection?.remoteAddress || 'unknown';
|
||||||
|
|
||||||
|
try {
|
||||||
|
await limiter.consume(clientIP);
|
||||||
|
next();
|
||||||
|
} catch (rejRes) {
|
||||||
|
const remainingPoints = rejRes.remainingPoints || 0;
|
||||||
|
const msBeforeNext = rejRes.msBeforeNext || 900000;
|
||||||
|
|
||||||
|
logger.security(`🚦 Global rate limit exceeded for IP: ${clientIP}`);
|
||||||
|
|
||||||
|
res.set({
|
||||||
|
'Retry-After': Math.round(msBeforeNext / 1000) || 900,
|
||||||
|
'X-RateLimit-Limit': 1000,
|
||||||
|
'X-RateLimit-Remaining': remainingPoints,
|
||||||
|
'X-RateLimit-Reset': new Date(Date.now() + msBeforeNext).toISOString()
|
||||||
|
});
|
||||||
|
|
||||||
|
res.status(429).json({
|
||||||
|
error: 'Too Many Requests',
|
||||||
|
message: 'Too many requests from this IP, please try again later.',
|
||||||
|
retryAfter: Math.round(msBeforeNext / 1000)
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// 📊 请求大小限制中间件
|
||||||
|
const requestSizeLimit = (req, res, next) => {
|
||||||
|
const maxSize = 10 * 1024 * 1024; // 10MB
|
||||||
|
const contentLength = parseInt(req.headers['content-length'] || '0');
|
||||||
|
|
||||||
|
if (contentLength > maxSize) {
|
||||||
|
logger.security(`🚨 Request too large: ${contentLength} bytes from ${req.ip}`);
|
||||||
|
return res.status(413).json({
|
||||||
|
error: 'Payload Too Large',
|
||||||
|
message: 'Request body size exceeds limit',
|
||||||
|
limit: '10MB'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
next();
|
||||||
|
};
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
authenticateApiKey,
|
||||||
|
authenticateAdmin,
|
||||||
|
corsMiddleware,
|
||||||
|
requestLogger,
|
||||||
|
securityMiddleware,
|
||||||
|
errorHandler,
|
||||||
|
globalRateLimit,
|
||||||
|
requestSizeLimit
|
||||||
|
};
|
||||||
678
src/models/redis.js
Normal file
678
src/models/redis.js
Normal file
@@ -0,0 +1,678 @@
|
|||||||
|
const Redis = require('ioredis');
|
||||||
|
const config = require('../../config/config');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
|
||||||
|
class RedisClient {
|
||||||
|
constructor() {
|
||||||
|
this.client = null;
|
||||||
|
this.isConnected = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
async connect() {
|
||||||
|
try {
|
||||||
|
this.client = new Redis({
|
||||||
|
host: config.redis.host,
|
||||||
|
port: config.redis.port,
|
||||||
|
password: config.redis.password,
|
||||||
|
db: config.redis.db,
|
||||||
|
retryDelayOnFailover: config.redis.retryDelayOnFailover,
|
||||||
|
maxRetriesPerRequest: config.redis.maxRetriesPerRequest,
|
||||||
|
lazyConnect: config.redis.lazyConnect
|
||||||
|
});
|
||||||
|
|
||||||
|
this.client.on('connect', () => {
|
||||||
|
this.isConnected = true;
|
||||||
|
logger.info('🔗 Redis connected successfully');
|
||||||
|
});
|
||||||
|
|
||||||
|
this.client.on('error', (err) => {
|
||||||
|
this.isConnected = false;
|
||||||
|
logger.error('❌ Redis connection error:', err);
|
||||||
|
});
|
||||||
|
|
||||||
|
this.client.on('close', () => {
|
||||||
|
this.isConnected = false;
|
||||||
|
logger.warn('⚠️ Redis connection closed');
|
||||||
|
});
|
||||||
|
|
||||||
|
await this.client.connect();
|
||||||
|
return this.client;
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('💥 Failed to connect to Redis:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async disconnect() {
|
||||||
|
if (this.client) {
|
||||||
|
await this.client.quit();
|
||||||
|
this.isConnected = false;
|
||||||
|
logger.info('👋 Redis disconnected');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
getClient() {
|
||||||
|
if (!this.client || !this.isConnected) {
|
||||||
|
logger.warn('⚠️ Redis client is not connected');
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
return this.client;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 安全获取客户端（用于关键操作）
|
||||||
|
getClientSafe() {
|
||||||
|
if (!this.client || !this.isConnected) {
|
||||||
|
throw new Error('Redis client is not connected');
|
||||||
|
}
|
||||||
|
return this.client;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔑 API Key 相关操作
|
||||||
|
async setApiKey(keyId, keyData, hashedKey = null) {
|
||||||
|
const key = `apikey:${keyId}`;
|
||||||
|
const client = this.getClientSafe();
|
||||||
|
|
||||||
|
// 维护哈希映射表（用于快速查找）
|
||||||
|
// hashedKey参数是实际的哈希值，用于建立映射
|
||||||
|
if (hashedKey) {
|
||||||
|
await client.hset('apikey:hash_map', hashedKey, keyId);
|
||||||
|
}
|
||||||
|
|
||||||
|
await client.hset(key, keyData);
|
||||||
|
await client.expire(key, 86400 * 365); // 1年过期
|
||||||
|
}
|
||||||
|
|
||||||
|
async getApiKey(keyId) {
|
||||||
|
const key = `apikey:${keyId}`;
|
||||||
|
return await this.client.hgetall(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
async deleteApiKey(keyId) {
|
||||||
|
const key = `apikey:${keyId}`;
|
||||||
|
|
||||||
|
// 获取要删除的API Key哈希值，以便从映射表中移除
|
||||||
|
const keyData = await this.client.hgetall(key);
|
||||||
|
if (keyData && keyData.apiKey) {
|
||||||
|
// keyData.apiKey现在存储的是哈希值，直接从映射表删除
|
||||||
|
await this.client.hdel('apikey:hash_map', keyData.apiKey);
|
||||||
|
}
|
||||||
|
|
||||||
|
return await this.client.del(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
async getAllApiKeys() {
|
||||||
|
const keys = await this.client.keys('apikey:*');
|
||||||
|
const apiKeys = [];
|
||||||
|
for (const key of keys) {
|
||||||
|
// 过滤掉hash_map，它不是真正的API Key
|
||||||
|
if (key === 'apikey:hash_map') {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
const keyData = await this.client.hgetall(key);
|
||||||
|
if (keyData && Object.keys(keyData).length > 0) {
|
||||||
|
apiKeys.push({ id: key.replace('apikey:', ''), ...keyData });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return apiKeys;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔍 通过哈希值查找API Key（性能优化）
|
||||||
|
async findApiKeyByHash(hashedKey) {
|
||||||
|
// 使用反向映射表：hash -> keyId
|
||||||
|
const keyId = await this.client.hget('apikey:hash_map', hashedKey);
|
||||||
|
if (!keyId) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
const keyData = await this.client.hgetall(`apikey:${keyId}`);
|
||||||
|
if (keyData && Object.keys(keyData).length > 0) {
|
||||||
|
return { id: keyId, ...keyData };
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果数据不存在，清理映射表
|
||||||
|
await this.client.hdel('apikey:hash_map', hashedKey);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📊 使用统计相关操作（支持缓存token统计和模型信息）
|
||||||
|
async incrementTokenUsage(keyId, tokens, inputTokens = 0, outputTokens = 0, cacheCreateTokens = 0, cacheReadTokens = 0, model = 'unknown') {
|
||||||
|
const key = `usage:${keyId}`;
|
||||||
|
const today = new Date().toISOString().split('T')[0];
|
||||||
|
const currentMonth = `${new Date().getFullYear()}-${String(new Date().getMonth() + 1).padStart(2, '0')}`;
|
||||||
|
const daily = `usage:daily:${keyId}:${today}`;
|
||||||
|
const monthly = `usage:monthly:${keyId}:${currentMonth}`;
|
||||||
|
|
||||||
|
// 按模型统计的键
|
||||||
|
const modelDaily = `usage:model:daily:${model}:${today}`;
|
||||||
|
const modelMonthly = `usage:model:monthly:${model}:${currentMonth}`;
|
||||||
|
|
||||||
|
// API Key级别的模型统计
|
||||||
|
const keyModelDaily = `usage:${keyId}:model:daily:${model}:${today}`;
|
||||||
|
const keyModelMonthly = `usage:${keyId}:model:monthly:${model}:${currentMonth}`;
|
||||||
|
|
||||||
|
// 智能处理输入输出token分配
|
||||||
|
const finalInputTokens = inputTokens || 0;
|
||||||
|
const finalOutputTokens = outputTokens || (finalInputTokens > 0 ? 0 : tokens);
|
||||||
|
const finalCacheCreateTokens = cacheCreateTokens || 0;
|
||||||
|
const finalCacheReadTokens = cacheReadTokens || 0;
|
||||||
|
|
||||||
|
// 重新计算真实的总token数（包括缓存token）
|
||||||
|
const totalTokens = finalInputTokens + finalOutputTokens + finalCacheCreateTokens + finalCacheReadTokens;
|
||||||
|
// 核心token（不包括缓存）- 用于与历史数据兼容
|
||||||
|
const coreTokens = finalInputTokens + finalOutputTokens;
|
||||||
|
|
||||||
|
await Promise.all([
|
||||||
|
// 核心token统计（保持向后兼容）
|
||||||
|
this.client.hincrby(key, 'totalTokens', coreTokens),
|
||||||
|
this.client.hincrby(key, 'totalInputTokens', finalInputTokens),
|
||||||
|
this.client.hincrby(key, 'totalOutputTokens', finalOutputTokens),
|
||||||
|
// 缓存token统计（新增）
|
||||||
|
this.client.hincrby(key, 'totalCacheCreateTokens', finalCacheCreateTokens),
|
||||||
|
this.client.hincrby(key, 'totalCacheReadTokens', finalCacheReadTokens),
|
||||||
|
this.client.hincrby(key, 'totalAllTokens', totalTokens), // 包含所有类型的总token
|
||||||
|
// 请求计数
|
||||||
|
this.client.hincrby(key, 'totalRequests', 1),
|
||||||
|
// 每日统计
|
||||||
|
this.client.hincrby(daily, 'tokens', coreTokens),
|
||||||
|
this.client.hincrby(daily, 'inputTokens', finalInputTokens),
|
||||||
|
this.client.hincrby(daily, 'outputTokens', finalOutputTokens),
|
||||||
|
this.client.hincrby(daily, 'cacheCreateTokens', finalCacheCreateTokens),
|
||||||
|
this.client.hincrby(daily, 'cacheReadTokens', finalCacheReadTokens),
|
||||||
|
this.client.hincrby(daily, 'allTokens', totalTokens),
|
||||||
|
this.client.hincrby(daily, 'requests', 1),
|
||||||
|
// 每月统计
|
||||||
|
this.client.hincrby(monthly, 'tokens', coreTokens),
|
||||||
|
this.client.hincrby(monthly, 'inputTokens', finalInputTokens),
|
||||||
|
this.client.hincrby(monthly, 'outputTokens', finalOutputTokens),
|
||||||
|
this.client.hincrby(monthly, 'cacheCreateTokens', finalCacheCreateTokens),
|
||||||
|
this.client.hincrby(monthly, 'cacheReadTokens', finalCacheReadTokens),
|
||||||
|
this.client.hincrby(monthly, 'allTokens', totalTokens),
|
||||||
|
this.client.hincrby(monthly, 'requests', 1),
|
||||||
|
// 按模型统计 - 每日
|
||||||
|
this.client.hincrby(modelDaily, 'inputTokens', finalInputTokens),
|
||||||
|
this.client.hincrby(modelDaily, 'outputTokens', finalOutputTokens),
|
||||||
|
this.client.hincrby(modelDaily, 'cacheCreateTokens', finalCacheCreateTokens),
|
||||||
|
this.client.hincrby(modelDaily, 'cacheReadTokens', finalCacheReadTokens),
|
||||||
|
this.client.hincrby(modelDaily, 'allTokens', totalTokens),
|
||||||
|
this.client.hincrby(modelDaily, 'requests', 1),
|
||||||
|
// 按模型统计 - 每月
|
||||||
|
this.client.hincrby(modelMonthly, 'inputTokens', finalInputTokens),
|
||||||
|
this.client.hincrby(modelMonthly, 'outputTokens', finalOutputTokens),
|
||||||
|
this.client.hincrby(modelMonthly, 'cacheCreateTokens', finalCacheCreateTokens),
|
||||||
|
this.client.hincrby(modelMonthly, 'cacheReadTokens', finalCacheReadTokens),
|
||||||
|
this.client.hincrby(modelMonthly, 'allTokens', totalTokens),
|
||||||
|
this.client.hincrby(modelMonthly, 'requests', 1),
|
||||||
|
// API Key级别的模型统计 - 每日
|
||||||
|
this.client.hincrby(keyModelDaily, 'inputTokens', finalInputTokens),
|
||||||
|
this.client.hincrby(keyModelDaily, 'outputTokens', finalOutputTokens),
|
||||||
|
this.client.hincrby(keyModelDaily, 'cacheCreateTokens', finalCacheCreateTokens),
|
||||||
|
this.client.hincrby(keyModelDaily, 'cacheReadTokens', finalCacheReadTokens),
|
||||||
|
this.client.hincrby(keyModelDaily, 'allTokens', totalTokens),
|
||||||
|
this.client.hincrby(keyModelDaily, 'requests', 1),
|
||||||
|
// API Key级别的模型统计 - 每月
|
||||||
|
this.client.hincrby(keyModelMonthly, 'inputTokens', finalInputTokens),
|
||||||
|
this.client.hincrby(keyModelMonthly, 'outputTokens', finalOutputTokens),
|
||||||
|
this.client.hincrby(keyModelMonthly, 'cacheCreateTokens', finalCacheCreateTokens),
|
||||||
|
this.client.hincrby(keyModelMonthly, 'cacheReadTokens', finalCacheReadTokens),
|
||||||
|
this.client.hincrby(keyModelMonthly, 'allTokens', totalTokens),
|
||||||
|
this.client.hincrby(keyModelMonthly, 'requests', 1),
|
||||||
|
// 设置过期时间
|
||||||
|
this.client.expire(daily, 86400 * 32), // 32天过期
|
||||||
|
this.client.expire(monthly, 86400 * 365), // 1年过期
|
||||||
|
this.client.expire(modelDaily, 86400 * 32), // 模型每日统计32天过期
|
||||||
|
this.client.expire(modelMonthly, 86400 * 365), // 模型每月统计1年过期
|
||||||
|
this.client.expire(keyModelDaily, 86400 * 32), // API Key模型每日统计32天过期
|
||||||
|
this.client.expire(keyModelMonthly, 86400 * 365) // API Key模型每月统计1年过期
|
||||||
|
]);
|
||||||
|
}
|
||||||
|
|
||||||
|
async getUsageStats(keyId) {
|
||||||
|
const totalKey = `usage:${keyId}`;
|
||||||
|
const today = new Date().toISOString().split('T')[0];
|
||||||
|
const dailyKey = `usage:daily:${keyId}:${today}`;
|
||||||
|
const currentMonth = `${new Date().getFullYear()}-${String(new Date().getMonth() + 1).padStart(2, '0')}`;
|
||||||
|
const monthlyKey = `usage:monthly:${keyId}:${currentMonth}`;
|
||||||
|
|
||||||
|
const [total, daily, monthly] = await Promise.all([
|
||||||
|
this.client.hgetall(totalKey),
|
||||||
|
this.client.hgetall(dailyKey),
|
||||||
|
this.client.hgetall(monthlyKey)
|
||||||
|
]);
|
||||||
|
|
||||||
|
// 获取API Key的创建时间来计算平均值
|
||||||
|
const keyData = await this.client.hgetall(`apikey:${keyId}`);
|
||||||
|
const createdAt = keyData.createdAt ? new Date(keyData.createdAt) : new Date();
|
||||||
|
const now = new Date();
|
||||||
|
const daysSinceCreated = Math.max(1, Math.ceil((now - createdAt) / (1000 * 60 * 60 * 24)));
|
||||||
|
|
||||||
|
const totalTokens = parseInt(total.totalTokens) || 0;
|
||||||
|
const totalRequests = parseInt(total.totalRequests) || 0;
|
||||||
|
|
||||||
|
// 计算平均RPM (requests per minute) 和 TPM (tokens per minute)
|
||||||
|
const totalMinutes = Math.max(1, daysSinceCreated * 24 * 60);
|
||||||
|
const avgRPM = totalRequests / totalMinutes;
|
||||||
|
const avgTPM = totalTokens / totalMinutes;
|
||||||
|
|
||||||
|
// 处理旧数据兼容性（支持缓存token）
|
||||||
|
const handleLegacyData = (data) => {
|
||||||
|
// 优先使用total*字段（存储时使用的字段）
|
||||||
|
const tokens = parseInt(data.totalTokens) || parseInt(data.tokens) || 0;
|
||||||
|
const inputTokens = parseInt(data.totalInputTokens) || parseInt(data.inputTokens) || 0;
|
||||||
|
const outputTokens = parseInt(data.totalOutputTokens) || parseInt(data.outputTokens) || 0;
|
||||||
|
const requests = parseInt(data.totalRequests) || parseInt(data.requests) || 0;
|
||||||
|
|
||||||
|
// 新增缓存token字段
|
||||||
|
const cacheCreateTokens = parseInt(data.totalCacheCreateTokens) || parseInt(data.cacheCreateTokens) || 0;
|
||||||
|
const cacheReadTokens = parseInt(data.totalCacheReadTokens) || parseInt(data.cacheReadTokens) || 0;
|
||||||
|
const allTokens = parseInt(data.totalAllTokens) || parseInt(data.allTokens) || 0;
|
||||||
|
|
||||||
|
const totalFromSeparate = inputTokens + outputTokens;
|
||||||
|
|
||||||
|
if (totalFromSeparate === 0 && tokens > 0) {
|
||||||
|
// 旧数据：没有输入输出分离
|
||||||
|
return {
|
||||||
|
tokens,
|
||||||
|
inputTokens: Math.round(tokens * 0.3), // 假设30%为输入
|
||||||
|
outputTokens: Math.round(tokens * 0.7), // 假设70%为输出
|
||||||
|
cacheCreateTokens: 0, // 旧数据没有缓存token
|
||||||
|
cacheReadTokens: 0,
|
||||||
|
allTokens: tokens, // 对于旧数据，allTokens等于tokens
|
||||||
|
requests
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
// 新数据或无数据
|
||||||
|
return {
|
||||||
|
tokens,
|
||||||
|
inputTokens,
|
||||||
|
outputTokens,
|
||||||
|
cacheCreateTokens,
|
||||||
|
cacheReadTokens,
|
||||||
|
allTokens: allTokens || (inputTokens + outputTokens + cacheCreateTokens + cacheReadTokens), // 计算或使用存储的值
|
||||||
|
requests
|
||||||
|
};
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const totalData = handleLegacyData(total);
|
||||||
|
const dailyData = handleLegacyData(daily);
|
||||||
|
const monthlyData = handleLegacyData(monthly);
|
||||||
|
|
||||||
|
return {
|
||||||
|
total: totalData,
|
||||||
|
daily: dailyData,
|
||||||
|
monthly: monthlyData,
|
||||||
|
averages: {
|
||||||
|
rpm: Math.round(avgRPM * 100) / 100, // 保留2位小数
|
||||||
|
tpm: Math.round(avgTPM * 100) / 100,
|
||||||
|
dailyRequests: Math.round((totalRequests / daysSinceCreated) * 100) / 100,
|
||||||
|
dailyTokens: Math.round((totalTokens / daysSinceCreated) * 100) / 100
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🧹 清空所有API Key的使用统计数据
|
||||||
|
async resetAllUsageStats() {
|
||||||
|
const client = this.getClientSafe();
|
||||||
|
const stats = {
|
||||||
|
deletedKeys: 0,
|
||||||
|
deletedDailyKeys: 0,
|
||||||
|
deletedMonthlyKeys: 0,
|
||||||
|
resetApiKeys: 0
|
||||||
|
};
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 获取所有API Key ID
|
||||||
|
const apiKeyIds = [];
|
||||||
|
const apiKeyKeys = await client.keys('apikey:*');
|
||||||
|
|
||||||
|
for (const key of apiKeyKeys) {
|
||||||
|
if (key === 'apikey:hash_map') continue; // 跳过哈希映射表
|
||||||
|
const keyId = key.replace('apikey:', '');
|
||||||
|
apiKeyIds.push(keyId);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 清空每个API Key的使用统计
|
||||||
|
for (const keyId of apiKeyIds) {
|
||||||
|
// 删除总体使用统计
|
||||||
|
const usageKey = `usage:${keyId}`;
|
||||||
|
const deleted = await client.del(usageKey);
|
||||||
|
if (deleted > 0) {
|
||||||
|
stats.deletedKeys++;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除该API Key的每日统计（使用精确的keyId匹配）
|
||||||
|
const dailyKeys = await client.keys(`usage:daily:${keyId}:*`);
|
||||||
|
if (dailyKeys.length > 0) {
|
||||||
|
await client.del(...dailyKeys);
|
||||||
|
stats.deletedDailyKeys += dailyKeys.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 删除该API Key的每月统计（使用精确的keyId匹配）
|
||||||
|
const monthlyKeys = await client.keys(`usage:monthly:${keyId}:*`);
|
||||||
|
if (monthlyKeys.length > 0) {
|
||||||
|
await client.del(...monthlyKeys);
|
||||||
|
stats.deletedMonthlyKeys += monthlyKeys.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 重置API Key的lastUsedAt字段
|
||||||
|
const keyData = await client.hgetall(`apikey:${keyId}`);
|
||||||
|
if (keyData && Object.keys(keyData).length > 0) {
|
||||||
|
keyData.lastUsedAt = '';
|
||||||
|
await client.hset(`apikey:${keyId}`, keyData);
|
||||||
|
stats.resetApiKeys++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 额外清理：删除所有可能遗漏的usage相关键
|
||||||
|
const allUsageKeys = await client.keys('usage:*');
|
||||||
|
if (allUsageKeys.length > 0) {
|
||||||
|
await client.del(...allUsageKeys);
|
||||||
|
stats.deletedKeys += allUsageKeys.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
return stats;
|
||||||
|
} catch (error) {
|
||||||
|
throw new Error(`Failed to reset usage stats: ${error.message}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🏢 Claude 账户管理
|
||||||
|
async setClaudeAccount(accountId, accountData) {
|
||||||
|
const key = `claude:account:${accountId}`;
|
||||||
|
await this.client.hset(key, accountData);
|
||||||
|
}
|
||||||
|
|
||||||
|
async getClaudeAccount(accountId) {
|
||||||
|
const key = `claude:account:${accountId}`;
|
||||||
|
return await this.client.hgetall(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
async getAllClaudeAccounts() {
|
||||||
|
const keys = await this.client.keys('claude:account:*');
|
||||||
|
const accounts = [];
|
||||||
|
for (const key of keys) {
|
||||||
|
const accountData = await this.client.hgetall(key);
|
||||||
|
if (accountData && Object.keys(accountData).length > 0) {
|
||||||
|
accounts.push({ id: key.replace('claude:account:', ''), ...accountData });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return accounts;
|
||||||
|
}
|
||||||
|
|
||||||
|
async deleteClaudeAccount(accountId) {
|
||||||
|
const key = `claude:account:${accountId}`;
|
||||||
|
return await this.client.del(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔐 会话管理
|
||||||
|
async setSession(sessionId, sessionData, ttl = 86400) {
|
||||||
|
const key = `session:${sessionId}`;
|
||||||
|
await this.client.hset(key, sessionData);
|
||||||
|
await this.client.expire(key, ttl);
|
||||||
|
}
|
||||||
|
|
||||||
|
async getSession(sessionId) {
|
||||||
|
const key = `session:${sessionId}`;
|
||||||
|
return await this.client.hgetall(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
async deleteSession(sessionId) {
|
||||||
|
const key = `session:${sessionId}`;
|
||||||
|
return await this.client.del(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🗝️ API Key哈希索引管理
|
||||||
|
async setApiKeyHash(hashedKey, keyData, ttl = 0) {
|
||||||
|
const key = `apikey_hash:${hashedKey}`;
|
||||||
|
await this.client.hset(key, keyData);
|
||||||
|
if (ttl > 0) {
|
||||||
|
await this.client.expire(key, ttl);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async getApiKeyHash(hashedKey) {
|
||||||
|
const key = `apikey_hash:${hashedKey}`;
|
||||||
|
return await this.client.hgetall(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
async deleteApiKeyHash(hashedKey) {
|
||||||
|
const key = `apikey_hash:${hashedKey}`;
|
||||||
|
return await this.client.del(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔗 OAuth会话管理
|
||||||
|
async setOAuthSession(sessionId, sessionData, ttl = 600) { // 10分钟过期
|
||||||
|
const key = `oauth:${sessionId}`;
|
||||||
|
await this.client.hset(key, sessionData);
|
||||||
|
await this.client.expire(key, ttl);
|
||||||
|
}
|
||||||
|
|
||||||
|
async getOAuthSession(sessionId) {
|
||||||
|
const key = `oauth:${sessionId}`;
|
||||||
|
return await this.client.hgetall(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
async deleteOAuthSession(sessionId) {
|
||||||
|
const key = `oauth:${sessionId}`;
|
||||||
|
return await this.client.del(key);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🚦 速率限制
|
||||||
|
async checkRateLimit(identifier, limit = 100, window = 60) {
|
||||||
|
const key = `ratelimit:${identifier}`;
|
||||||
|
const current = await this.client.incr(key);
|
||||||
|
|
||||||
|
if (current === 1) {
|
||||||
|
await this.client.expire(key, window);
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
allowed: current <= limit,
|
||||||
|
current,
|
||||||
|
limit,
|
||||||
|
resetTime: await this.client.ttl(key)
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📈 系统统计
|
||||||
|
async getSystemStats() {
|
||||||
|
const keys = await Promise.all([
|
||||||
|
this.client.keys('apikey:*'),
|
||||||
|
this.client.keys('claude:account:*'),
|
||||||
|
this.client.keys('usage:*')
|
||||||
|
]);
|
||||||
|
|
||||||
|
return {
|
||||||
|
totalApiKeys: keys[0].length,
|
||||||
|
totalClaudeAccounts: keys[1].length,
|
||||||
|
totalUsageRecords: keys[2].length
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📊 获取今日系统统计
|
||||||
|
async getTodayStats() {
|
||||||
|
try {
|
||||||
|
const today = new Date().toISOString().split('T')[0];
|
||||||
|
const dailyKeys = await this.client.keys(`usage:daily:*:${today}`);
|
||||||
|
|
||||||
|
let totalRequestsToday = 0;
|
||||||
|
let totalTokensToday = 0;
|
||||||
|
let totalInputTokensToday = 0;
|
||||||
|
let totalOutputTokensToday = 0;
|
||||||
|
let totalCacheCreateTokensToday = 0;
|
||||||
|
let totalCacheReadTokensToday = 0;
|
||||||
|
|
||||||
|
// 批量获取所有今日数据，提高性能
|
||||||
|
if (dailyKeys.length > 0) {
|
||||||
|
const pipeline = this.client.pipeline();
|
||||||
|
dailyKeys.forEach(key => pipeline.hgetall(key));
|
||||||
|
const results = await pipeline.exec();
|
||||||
|
|
||||||
|
for (const [error, dailyData] of results) {
|
||||||
|
if (error || !dailyData) continue;
|
||||||
|
|
||||||
|
totalRequestsToday += parseInt(dailyData.requests) || 0;
|
||||||
|
const currentDayTokens = parseInt(dailyData.tokens) || 0;
|
||||||
|
totalTokensToday += currentDayTokens;
|
||||||
|
|
||||||
|
// 处理旧数据兼容性：如果有总token但没有输入输出分离，则使用总token作为输出token
|
||||||
|
const inputTokens = parseInt(dailyData.inputTokens) || 0;
|
||||||
|
const outputTokens = parseInt(dailyData.outputTokens) || 0;
|
||||||
|
const cacheCreateTokens = parseInt(dailyData.cacheCreateTokens) || 0;
|
||||||
|
const cacheReadTokens = parseInt(dailyData.cacheReadTokens) || 0;
|
||||||
|
const totalTokensFromSeparate = inputTokens + outputTokens;
|
||||||
|
|
||||||
|
if (totalTokensFromSeparate === 0 && currentDayTokens > 0) {
|
||||||
|
// 旧数据：没有输入输出分离，假设70%为输出，30%为输入（基于一般对话比例）
|
||||||
|
totalOutputTokensToday += Math.round(currentDayTokens * 0.7);
|
||||||
|
totalInputTokensToday += Math.round(currentDayTokens * 0.3);
|
||||||
|
} else {
|
||||||
|
// 新数据：使用实际的输入输出分离
|
||||||
|
totalInputTokensToday += inputTokens;
|
||||||
|
totalOutputTokensToday += outputTokens;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 添加cache token统计
|
||||||
|
totalCacheCreateTokensToday += cacheCreateTokens;
|
||||||
|
totalCacheReadTokensToday += cacheReadTokens;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取今日创建的API Key数量（批量优化）
|
||||||
|
const allApiKeys = await this.client.keys('apikey:*');
|
||||||
|
let apiKeysCreatedToday = 0;
|
||||||
|
|
||||||
|
if (allApiKeys.length > 0) {
|
||||||
|
const pipeline = this.client.pipeline();
|
||||||
|
allApiKeys.forEach(key => pipeline.hget(key, 'createdAt'));
|
||||||
|
const results = await pipeline.exec();
|
||||||
|
|
||||||
|
for (const [error, createdAt] of results) {
|
||||||
|
if (!error && createdAt && createdAt.startsWith(today)) {
|
||||||
|
apiKeysCreatedToday++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
requestsToday: totalRequestsToday,
|
||||||
|
tokensToday: totalTokensToday,
|
||||||
|
inputTokensToday: totalInputTokensToday,
|
||||||
|
outputTokensToday: totalOutputTokensToday,
|
||||||
|
cacheCreateTokensToday: totalCacheCreateTokensToday,
|
||||||
|
cacheReadTokensToday: totalCacheReadTokensToday,
|
||||||
|
apiKeysCreatedToday
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error getting today stats:', error);
|
||||||
|
return {
|
||||||
|
requestsToday: 0,
|
||||||
|
tokensToday: 0,
|
||||||
|
inputTokensToday: 0,
|
||||||
|
outputTokensToday: 0,
|
||||||
|
cacheCreateTokensToday: 0,
|
||||||
|
cacheReadTokensToday: 0,
|
||||||
|
apiKeysCreatedToday: 0
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📈 获取系统总的平均RPM和TPM
|
||||||
|
async getSystemAverages() {
|
||||||
|
try {
|
||||||
|
const allApiKeys = await this.client.keys('apikey:*');
|
||||||
|
let totalRequests = 0;
|
||||||
|
let totalTokens = 0;
|
||||||
|
let totalInputTokens = 0;
|
||||||
|
let totalOutputTokens = 0;
|
||||||
|
let oldestCreatedAt = new Date();
|
||||||
|
|
||||||
|
// 批量获取所有usage数据和key数据，提高性能
|
||||||
|
const usageKeys = allApiKeys.map(key => `usage:${key.replace('apikey:', '')}`);
|
||||||
|
const pipeline = this.client.pipeline();
|
||||||
|
|
||||||
|
// 添加所有usage查询
|
||||||
|
usageKeys.forEach(key => pipeline.hgetall(key));
|
||||||
|
// 添加所有key数据查询
|
||||||
|
allApiKeys.forEach(key => pipeline.hgetall(key));
|
||||||
|
|
||||||
|
const results = await pipeline.exec();
|
||||||
|
const usageResults = results.slice(0, usageKeys.length);
|
||||||
|
const keyResults = results.slice(usageKeys.length);
|
||||||
|
|
||||||
|
for (let i = 0; i < allApiKeys.length; i++) {
|
||||||
|
const totalData = usageResults[i][1] || {};
|
||||||
|
const keyData = keyResults[i][1] || {};
|
||||||
|
|
||||||
|
totalRequests += parseInt(totalData.totalRequests) || 0;
|
||||||
|
totalTokens += parseInt(totalData.totalTokens) || 0;
|
||||||
|
totalInputTokens += parseInt(totalData.totalInputTokens) || 0;
|
||||||
|
totalOutputTokens += parseInt(totalData.totalOutputTokens) || 0;
|
||||||
|
|
||||||
|
const createdAt = keyData.createdAt ? new Date(keyData.createdAt) : new Date();
|
||||||
|
if (createdAt < oldestCreatedAt) {
|
||||||
|
oldestCreatedAt = createdAt;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const now = new Date();
|
||||||
|
// 保持与个人API Key计算一致的算法：按天计算然后转换为分钟
|
||||||
|
const daysSinceOldest = Math.max(1, Math.ceil((now - oldestCreatedAt) / (1000 * 60 * 60 * 24)));
|
||||||
|
const totalMinutes = daysSinceOldest * 24 * 60;
|
||||||
|
|
||||||
|
return {
|
||||||
|
systemRPM: Math.round((totalRequests / totalMinutes) * 100) / 100,
|
||||||
|
systemTPM: Math.round((totalTokens / totalMinutes) * 100) / 100,
|
||||||
|
totalInputTokens,
|
||||||
|
totalOutputTokens,
|
||||||
|
totalTokens
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Error getting system averages:', error);
|
||||||
|
return {
|
||||||
|
systemRPM: 0,
|
||||||
|
systemTPM: 0,
|
||||||
|
totalInputTokens: 0,
|
||||||
|
totalOutputTokens: 0,
|
||||||
|
totalTokens: 0
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🧹 清理过期数据
|
||||||
|
async cleanup() {
|
||||||
|
try {
|
||||||
|
const patterns = [
|
||||||
|
'usage:daily:*',
|
||||||
|
'ratelimit:*',
|
||||||
|
'session:*',
|
||||||
|
'oauth:*'
|
||||||
|
];
|
||||||
|
|
||||||
|
for (const pattern of patterns) {
|
||||||
|
const keys = await this.client.keys(pattern);
|
||||||
|
const pipeline = this.client.pipeline();
|
||||||
|
|
||||||
|
for (const key of keys) {
|
||||||
|
const ttl = await this.client.ttl(key);
|
||||||
|
if (ttl === -1) { // 没有设置过期时间的键
|
||||||
|
if (key.startsWith('oauth:')) {
|
||||||
|
pipeline.expire(key, 600); // OAuth会话设置10分钟过期
|
||||||
|
} else {
|
||||||
|
pipeline.expire(key, 86400); // 其他设置1天过期
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await pipeline.exec();
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.info('🧹 Redis cleanup completed');
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Redis cleanup failed:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = new RedisClient();
|
||||||
901
src/routes/admin.js
Normal file
901
src/routes/admin.js
Normal file
@@ -0,0 +1,901 @@
|
|||||||
|
const express = require('express');
|
||||||
|
const apiKeyService = require('../services/apiKeyService');
|
||||||
|
const claudeAccountService = require('../services/claudeAccountService');
|
||||||
|
const redis = require('../models/redis');
|
||||||
|
const { authenticateAdmin } = require('../middleware/auth');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
const oauthHelper = require('../utils/oauthHelper');
|
||||||
|
const CostCalculator = require('../utils/costCalculator');
|
||||||
|
const pricingService = require('../services/pricingService');
|
||||||
|
|
||||||
|
const router = express.Router();
|
||||||
|
|
||||||
|
// 🔑 API Keys 管理
|
||||||
|
|
||||||
|
// 获取所有API Keys
|
||||||
|
router.get('/api-keys', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const apiKeys = await apiKeyService.getAllApiKeys();
|
||||||
|
res.json({ success: true, data: apiKeys });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get API keys:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to get API keys', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 创建新的API Key
|
||||||
|
router.post('/api-keys', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const {
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
tokenLimit,
|
||||||
|
requestLimit,
|
||||||
|
expiresAt,
|
||||||
|
claudeAccountId
|
||||||
|
} = req.body;
|
||||||
|
|
||||||
|
// 输入验证
|
||||||
|
if (!name || typeof name !== 'string' || name.trim().length === 0) {
|
||||||
|
return res.status(400).json({ error: 'Name is required and must be a non-empty string' });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (name.length > 100) {
|
||||||
|
return res.status(400).json({ error: 'Name must be less than 100 characters' });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (description && (typeof description !== 'string' || description.length > 500)) {
|
||||||
|
return res.status(400).json({ error: 'Description must be a string with less than 500 characters' });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (tokenLimit && (!Number.isInteger(Number(tokenLimit)) || Number(tokenLimit) < 0)) {
|
||||||
|
return res.status(400).json({ error: 'Token limit must be a non-negative integer' });
|
||||||
|
}
|
||||||
|
|
||||||
|
if (requestLimit && (!Number.isInteger(Number(requestLimit)) || Number(requestLimit) < 0)) {
|
||||||
|
return res.status(400).json({ error: 'Request limit must be a non-negative integer' });
|
||||||
|
}
|
||||||
|
|
||||||
|
const newKey = await apiKeyService.generateApiKey({
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
tokenLimit,
|
||||||
|
requestLimit,
|
||||||
|
expiresAt,
|
||||||
|
claudeAccountId
|
||||||
|
});
|
||||||
|
|
||||||
|
logger.success(`🔑 Admin created new API key: ${name}`);
|
||||||
|
res.json({ success: true, data: newKey });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to create API key:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to create API key', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 更新API Key
|
||||||
|
router.put('/api-keys/:keyId', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { keyId } = req.params;
|
||||||
|
const updates = req.body;
|
||||||
|
|
||||||
|
await apiKeyService.updateApiKey(keyId, updates);
|
||||||
|
|
||||||
|
logger.success(`📝 Admin updated API key: ${keyId}`);
|
||||||
|
res.json({ success: true, message: 'API key updated successfully' });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to update API key:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to update API key', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 删除API Key
|
||||||
|
router.delete('/api-keys/:keyId', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { keyId } = req.params;
|
||||||
|
|
||||||
|
await apiKeyService.deleteApiKey(keyId);
|
||||||
|
|
||||||
|
logger.success(`🗑️ Admin deleted API key: ${keyId}`);
|
||||||
|
res.json({ success: true, message: 'API key deleted successfully' });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to delete API key:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to delete API key', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🏢 Claude 账户管理
|
||||||
|
|
||||||
|
// 生成OAuth授权URL
|
||||||
|
router.post('/claude-accounts/generate-auth-url', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { proxy } = req.body; // 接收代理配置
|
||||||
|
const oauthParams = await oauthHelper.generateOAuthParams();
|
||||||
|
|
||||||
|
// 将codeVerifier和state临时存储到Redis，用于后续验证
|
||||||
|
const sessionId = require('crypto').randomUUID();
|
||||||
|
await redis.setOAuthSession(sessionId, {
|
||||||
|
codeVerifier: oauthParams.codeVerifier,
|
||||||
|
state: oauthParams.state,
|
||||||
|
codeChallenge: oauthParams.codeChallenge,
|
||||||
|
proxy: proxy || null, // 存储代理配置
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
expiresAt: new Date(Date.now() + 10 * 60 * 1000).toISOString() // 10分钟过期
|
||||||
|
});
|
||||||
|
|
||||||
|
logger.success('🔗 Generated OAuth authorization URL with proxy support');
|
||||||
|
res.json({
|
||||||
|
success: true,
|
||||||
|
data: {
|
||||||
|
authUrl: oauthParams.authUrl,
|
||||||
|
sessionId: sessionId,
|
||||||
|
instructions: [
|
||||||
|
'1. 复制上面的链接到浏览器中打开',
|
||||||
|
'2. 登录您的 Anthropic 账户',
|
||||||
|
'3. 同意应用权限',
|
||||||
|
'4. 复制浏览器地址栏中的完整 URL',
|
||||||
|
'5. 在添加账户表单中粘贴完整的回调 URL 和授权码'
|
||||||
|
]
|
||||||
|
}
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to generate OAuth URL:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to generate OAuth URL', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 验证授权码并获取token
|
||||||
|
router.post('/claude-accounts/exchange-code', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { sessionId, authorizationCode, callbackUrl } = req.body;
|
||||||
|
|
||||||
|
if (!sessionId || (!authorizationCode && !callbackUrl)) {
|
||||||
|
return res.status(400).json({ error: 'Session ID and authorization code (or callback URL) are required' });
|
||||||
|
}
|
||||||
|
|
||||||
|
// 从Redis获取OAuth会话信息
|
||||||
|
const oauthSession = await redis.getOAuthSession(sessionId);
|
||||||
|
if (!oauthSession) {
|
||||||
|
return res.status(400).json({ error: 'Invalid or expired OAuth session' });
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查会话是否过期
|
||||||
|
if (new Date() > new Date(oauthSession.expiresAt)) {
|
||||||
|
await redis.deleteOAuthSession(sessionId);
|
||||||
|
return res.status(400).json({ error: 'OAuth session has expired, please generate a new authorization URL' });
|
||||||
|
}
|
||||||
|
|
||||||
|
// 统一处理授权码输入（可能是直接的code或完整的回调URL）
|
||||||
|
let finalAuthCode;
|
||||||
|
const inputValue = callbackUrl || authorizationCode;
|
||||||
|
|
||||||
|
try {
|
||||||
|
finalAuthCode = oauthHelper.parseCallbackUrl(inputValue);
|
||||||
|
} catch (parseError) {
|
||||||
|
return res.status(400).json({ error: 'Failed to parse authorization input', message: parseError.message });
|
||||||
|
}
|
||||||
|
|
||||||
|
// 交换访问令牌
|
||||||
|
const tokenData = await oauthHelper.exchangeCodeForTokens(
|
||||||
|
finalAuthCode,
|
||||||
|
oauthSession.codeVerifier,
|
||||||
|
oauthSession.state,
|
||||||
|
oauthSession.proxy // 传递代理配置
|
||||||
|
);
|
||||||
|
|
||||||
|
// 清理OAuth会话
|
||||||
|
await redis.deleteOAuthSession(sessionId);
|
||||||
|
|
||||||
|
logger.success('🎉 Successfully exchanged authorization code for tokens');
|
||||||
|
res.json({
|
||||||
|
success: true,
|
||||||
|
data: {
|
||||||
|
claudeAiOauth: tokenData
|
||||||
|
}
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to exchange authorization code:', {
|
||||||
|
error: error.message,
|
||||||
|
sessionId: req.body.sessionId,
|
||||||
|
// 不记录完整的授权码，只记录长度和前几个字符
|
||||||
|
codeLength: req.body.callbackUrl ? req.body.callbackUrl.length : (req.body.authorizationCode ? req.body.authorizationCode.length : 0),
|
||||||
|
codePrefix: req.body.callbackUrl ? req.body.callbackUrl.substring(0, 10) + '...' : (req.body.authorizationCode ? req.body.authorizationCode.substring(0, 10) + '...' : 'N/A')
|
||||||
|
});
|
||||||
|
res.status(500).json({ error: 'Failed to exchange authorization code', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 获取所有Claude账户
|
||||||
|
router.get('/claude-accounts', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const accounts = await claudeAccountService.getAllAccounts();
|
||||||
|
res.json({ success: true, data: accounts });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get Claude accounts:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to get Claude accounts', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 创建新的Claude账户
|
||||||
|
router.post('/claude-accounts', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const {
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
email,
|
||||||
|
password,
|
||||||
|
refreshToken,
|
||||||
|
claudeAiOauth,
|
||||||
|
proxy
|
||||||
|
} = req.body;
|
||||||
|
|
||||||
|
if (!name) {
|
||||||
|
return res.status(400).json({ error: 'Name is required' });
|
||||||
|
}
|
||||||
|
|
||||||
|
const newAccount = await claudeAccountService.createAccount({
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
email,
|
||||||
|
password,
|
||||||
|
refreshToken,
|
||||||
|
claudeAiOauth,
|
||||||
|
proxy
|
||||||
|
});
|
||||||
|
|
||||||
|
logger.success(`🏢 Admin created new Claude account: ${name}`);
|
||||||
|
res.json({ success: true, data: newAccount });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to create Claude account:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to create Claude account', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 更新Claude账户
|
||||||
|
router.put('/claude-accounts/:accountId', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { accountId } = req.params;
|
||||||
|
const updates = req.body;
|
||||||
|
|
||||||
|
await claudeAccountService.updateAccount(accountId, updates);
|
||||||
|
|
||||||
|
logger.success(`📝 Admin updated Claude account: ${accountId}`);
|
||||||
|
res.json({ success: true, message: 'Claude account updated successfully' });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to update Claude account:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to update Claude account', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 删除Claude账户
|
||||||
|
router.delete('/claude-accounts/:accountId', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { accountId } = req.params;
|
||||||
|
|
||||||
|
await claudeAccountService.deleteAccount(accountId);
|
||||||
|
|
||||||
|
logger.success(`🗑️ Admin deleted Claude account: ${accountId}`);
|
||||||
|
res.json({ success: true, message: 'Claude account deleted successfully' });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to delete Claude account:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to delete Claude account', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 刷新Claude账户token
|
||||||
|
router.post('/claude-accounts/:accountId/refresh', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { accountId } = req.params;
|
||||||
|
|
||||||
|
const result = await claudeAccountService.refreshAccountToken(accountId);
|
||||||
|
|
||||||
|
logger.success(`🔄 Admin refreshed token for Claude account: ${accountId}`);
|
||||||
|
res.json({ success: true, data: result });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to refresh Claude account token:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to refresh token', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 📊 系统统计
|
||||||
|
|
||||||
|
// 获取系统概览
|
||||||
|
router.get('/dashboard', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const [, apiKeys, accounts, todayStats, systemAverages] = await Promise.all([
|
||||||
|
redis.getSystemStats(),
|
||||||
|
apiKeyService.getAllApiKeys(),
|
||||||
|
claudeAccountService.getAllAccounts(),
|
||||||
|
redis.getTodayStats(),
|
||||||
|
redis.getSystemAverages()
|
||||||
|
]);
|
||||||
|
|
||||||
|
// 计算使用统计（包含cache tokens）
|
||||||
|
const totalTokensUsed = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.tokens || 0), 0);
|
||||||
|
const totalRequestsUsed = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.requests || 0), 0);
|
||||||
|
const totalInputTokensUsed = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.inputTokens || 0), 0);
|
||||||
|
const totalOutputTokensUsed = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.outputTokens || 0), 0);
|
||||||
|
const totalCacheCreateTokensUsed = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.cacheCreateTokens || 0), 0);
|
||||||
|
const totalCacheReadTokensUsed = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.cacheReadTokens || 0), 0);
|
||||||
|
const totalAllTokensUsed = apiKeys.reduce((sum, key) => sum + (key.usage?.total?.allTokens || 0), 0);
|
||||||
|
|
||||||
|
const activeApiKeys = apiKeys.filter(key => key.isActive).length;
|
||||||
|
const activeAccounts = accounts.filter(acc => acc.isActive && acc.status === 'active').length;
|
||||||
|
|
||||||
|
const dashboard = {
|
||||||
|
overview: {
|
||||||
|
totalApiKeys: apiKeys.length,
|
||||||
|
activeApiKeys,
|
||||||
|
totalClaudeAccounts: accounts.length,
|
||||||
|
activeClaudeAccounts: activeAccounts,
|
||||||
|
totalTokensUsed,
|
||||||
|
totalRequestsUsed,
|
||||||
|
totalInputTokensUsed,
|
||||||
|
totalOutputTokensUsed,
|
||||||
|
totalCacheCreateTokensUsed,
|
||||||
|
totalCacheReadTokensUsed,
|
||||||
|
totalAllTokensUsed
|
||||||
|
},
|
||||||
|
recentActivity: {
|
||||||
|
apiKeysCreatedToday: todayStats.apiKeysCreatedToday,
|
||||||
|
requestsToday: todayStats.requestsToday,
|
||||||
|
tokensToday: todayStats.tokensToday,
|
||||||
|
inputTokensToday: todayStats.inputTokensToday,
|
||||||
|
outputTokensToday: todayStats.outputTokensToday,
|
||||||
|
cacheCreateTokensToday: todayStats.cacheCreateTokensToday || 0,
|
||||||
|
cacheReadTokensToday: todayStats.cacheReadTokensToday || 0
|
||||||
|
},
|
||||||
|
systemAverages: {
|
||||||
|
rpm: systemAverages.systemRPM,
|
||||||
|
tpm: systemAverages.systemTPM
|
||||||
|
},
|
||||||
|
systemHealth: {
|
||||||
|
redisConnected: redis.isConnected,
|
||||||
|
claudeAccountsHealthy: activeAccounts > 0,
|
||||||
|
uptime: process.uptime()
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
res.json({ success: true, data: dashboard });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get dashboard data:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to get dashboard data', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 获取使用统计
|
||||||
|
router.get('/usage-stats', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { period = 'daily' } = req.query; // daily, monthly
|
||||||
|
|
||||||
|
// 获取基础API Key统计
|
||||||
|
const apiKeys = await apiKeyService.getAllApiKeys();
|
||||||
|
|
||||||
|
const stats = apiKeys.map(key => ({
|
||||||
|
keyId: key.id,
|
||||||
|
keyName: key.name,
|
||||||
|
usage: key.usage
|
||||||
|
}));
|
||||||
|
|
||||||
|
res.json({ success: true, data: { period, stats } });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get usage stats:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to get usage stats', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 获取按模型的使用统计和费用
|
||||||
|
router.get('/model-stats', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { period = 'daily' } = req.query; // daily, monthly
|
||||||
|
const today = new Date().toISOString().split('T')[0];
|
||||||
|
const currentMonth = `${new Date().getFullYear()}-${String(new Date().getMonth() + 1).padStart(2, '0')}`;
|
||||||
|
|
||||||
|
logger.info(`📊 Getting global model stats, period: ${period}, today: ${today}, currentMonth: ${currentMonth}`);
|
||||||
|
|
||||||
|
const client = redis.getClientSafe();
|
||||||
|
|
||||||
|
// 获取所有模型的统计数据
|
||||||
|
const pattern = period === 'daily' ? `usage:model:daily:*:${today}` : `usage:model:monthly:*:${currentMonth}`;
|
||||||
|
logger.info(`📊 Searching pattern: ${pattern}`);
|
||||||
|
|
||||||
|
const keys = await client.keys(pattern);
|
||||||
|
logger.info(`📊 Found ${keys.length} matching keys:`, keys);
|
||||||
|
|
||||||
|
const modelStats = [];
|
||||||
|
|
||||||
|
for (const key of keys) {
|
||||||
|
const match = key.match(period === 'daily' ?
|
||||||
|
/usage:model:daily:(.+):\d{4}-\d{2}-\d{2}$/ :
|
||||||
|
/usage:model:monthly:(.+):\d{4}-\d{2}$/
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!match) {
|
||||||
|
logger.warn(`📊 Pattern mismatch for key: ${key}`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
const model = match[1];
|
||||||
|
const data = await client.hgetall(key);
|
||||||
|
|
||||||
|
logger.info(`📊 Model ${model} data:`, data);
|
||||||
|
|
||||||
|
if (data && Object.keys(data).length > 0) {
|
||||||
|
const usage = {
|
||||||
|
input_tokens: parseInt(data.inputTokens) || 0,
|
||||||
|
output_tokens: parseInt(data.outputTokens) || 0,
|
||||||
|
cache_creation_input_tokens: parseInt(data.cacheCreateTokens) || 0,
|
||||||
|
cache_read_input_tokens: parseInt(data.cacheReadTokens) || 0
|
||||||
|
};
|
||||||
|
|
||||||
|
// 计算费用
|
||||||
|
const costData = CostCalculator.calculateCost(usage, model);
|
||||||
|
|
||||||
|
modelStats.push({
|
||||||
|
model,
|
||||||
|
period,
|
||||||
|
requests: parseInt(data.requests) || 0,
|
||||||
|
inputTokens: usage.input_tokens,
|
||||||
|
outputTokens: usage.output_tokens,
|
||||||
|
cacheCreateTokens: usage.cache_creation_input_tokens,
|
||||||
|
cacheReadTokens: usage.cache_read_input_tokens,
|
||||||
|
allTokens: parseInt(data.allTokens) || 0,
|
||||||
|
usage: {
|
||||||
|
requests: parseInt(data.requests) || 0,
|
||||||
|
inputTokens: usage.input_tokens,
|
||||||
|
outputTokens: usage.output_tokens,
|
||||||
|
cacheCreateTokens: usage.cache_creation_input_tokens,
|
||||||
|
cacheReadTokens: usage.cache_read_input_tokens,
|
||||||
|
totalTokens: usage.input_tokens + usage.output_tokens + usage.cache_creation_input_tokens + usage.cache_read_input_tokens
|
||||||
|
},
|
||||||
|
costs: costData.costs,
|
||||||
|
formatted: costData.formatted,
|
||||||
|
pricing: costData.pricing
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 按总费用排序
|
||||||
|
modelStats.sort((a, b) => b.costs.total - a.costs.total);
|
||||||
|
|
||||||
|
logger.info(`📊 Returning ${modelStats.length} global model stats for period ${period}:`, modelStats);
|
||||||
|
|
||||||
|
res.json({ success: true, data: modelStats });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get model stats:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to get model stats', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🔧 系统管理
|
||||||
|
|
||||||
|
// 清理过期数据
|
||||||
|
router.post('/cleanup', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const [expiredKeys, errorAccounts] = await Promise.all([
|
||||||
|
apiKeyService.cleanupExpiredKeys(),
|
||||||
|
claudeAccountService.cleanupErrorAccounts()
|
||||||
|
]);
|
||||||
|
|
||||||
|
await redis.cleanup();
|
||||||
|
|
||||||
|
logger.success(`🧹 Admin triggered cleanup: ${expiredKeys} expired keys, ${errorAccounts} error accounts`);
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
success: true,
|
||||||
|
message: 'Cleanup completed',
|
||||||
|
data: {
|
||||||
|
expiredKeysRemoved: expiredKeys,
|
||||||
|
errorAccountsReset: errorAccounts
|
||||||
|
}
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Cleanup failed:', error);
|
||||||
|
res.status(500).json({ error: 'Cleanup failed', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 获取使用趋势数据
|
||||||
|
router.get('/usage-trend', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { days = 7 } = req.query;
|
||||||
|
const daysCount = parseInt(days) || 7;
|
||||||
|
const client = redis.getClientSafe();
|
||||||
|
|
||||||
|
const trendData = [];
|
||||||
|
const today = new Date();
|
||||||
|
|
||||||
|
// 获取过去N天的数据
|
||||||
|
for (let i = 0; i < daysCount; i++) {
|
||||||
|
const date = new Date(today);
|
||||||
|
date.setDate(date.getDate() - i);
|
||||||
|
const dateStr = date.toISOString().split('T')[0];
|
||||||
|
|
||||||
|
// 汇总当天所有API Key的使用数据
|
||||||
|
const pattern = `usage:daily:*:${dateStr}`;
|
||||||
|
const keys = await client.keys(pattern);
|
||||||
|
|
||||||
|
let dayInputTokens = 0;
|
||||||
|
let dayOutputTokens = 0;
|
||||||
|
let dayRequests = 0;
|
||||||
|
let dayCacheCreateTokens = 0;
|
||||||
|
let dayCacheReadTokens = 0;
|
||||||
|
let dayCost = 0;
|
||||||
|
|
||||||
|
for (const key of keys) {
|
||||||
|
const data = await client.hgetall(key);
|
||||||
|
if (data) {
|
||||||
|
dayInputTokens += parseInt(data.inputTokens) || 0;
|
||||||
|
dayOutputTokens += parseInt(data.outputTokens) || 0;
|
||||||
|
dayRequests += parseInt(data.requests) || 0;
|
||||||
|
dayCacheCreateTokens += parseInt(data.cacheCreateTokens) || 0;
|
||||||
|
dayCacheReadTokens += parseInt(data.cacheReadTokens) || 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 计算当天费用（使用通用模型价格估算）
|
||||||
|
const usage = {
|
||||||
|
input_tokens: dayInputTokens,
|
||||||
|
output_tokens: dayOutputTokens,
|
||||||
|
cache_creation_input_tokens: dayCacheCreateTokens,
|
||||||
|
cache_read_input_tokens: dayCacheReadTokens
|
||||||
|
};
|
||||||
|
const costResult = CostCalculator.calculateCost(usage, 'unknown');
|
||||||
|
dayCost = costResult.costs.total;
|
||||||
|
|
||||||
|
trendData.push({
|
||||||
|
date: dateStr,
|
||||||
|
inputTokens: dayInputTokens,
|
||||||
|
outputTokens: dayOutputTokens,
|
||||||
|
requests: dayRequests,
|
||||||
|
cacheCreateTokens: dayCacheCreateTokens,
|
||||||
|
cacheReadTokens: dayCacheReadTokens,
|
||||||
|
totalTokens: dayInputTokens + dayOutputTokens + dayCacheCreateTokens + dayCacheReadTokens,
|
||||||
|
cost: dayCost,
|
||||||
|
formattedCost: CostCalculator.formatCost(dayCost)
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 按日期正序排列
|
||||||
|
trendData.sort((a, b) => new Date(a.date) - new Date(b.date));
|
||||||
|
|
||||||
|
res.json({ success: true, data: trendData });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get usage trend:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to get usage trend', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 获取单个API Key的模型统计
|
||||||
|
router.get('/api-keys/:keyId/model-stats', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { keyId } = req.params;
|
||||||
|
const { period = 'monthly', startDate, endDate } = req.query;
|
||||||
|
|
||||||
|
logger.info(`📊 Getting model stats for API key: ${keyId}, period: ${period}, startDate: ${startDate}, endDate: ${endDate}`);
|
||||||
|
|
||||||
|
const client = redis.getClientSafe();
|
||||||
|
const today = new Date().toISOString().split('T')[0];
|
||||||
|
const currentMonth = `${new Date().getFullYear()}-${String(new Date().getMonth() + 1).padStart(2, '0')}`;
|
||||||
|
|
||||||
|
let searchPatterns = [];
|
||||||
|
|
||||||
|
if (period === 'custom' && startDate && endDate) {
|
||||||
|
// 自定义日期范围，生成多个日期的搜索模式
|
||||||
|
const start = new Date(startDate);
|
||||||
|
const end = new Date(endDate);
|
||||||
|
|
||||||
|
// 确保日期范围有效
|
||||||
|
if (start > end) {
|
||||||
|
return res.status(400).json({ error: 'Start date must be before or equal to end date' });
|
||||||
|
}
|
||||||
|
|
||||||
|
// 限制最大范围为31天
|
||||||
|
const daysDiff = Math.ceil((end - start) / (1000 * 60 * 60 * 24)) + 1;
|
||||||
|
if (daysDiff > 31) {
|
||||||
|
return res.status(400).json({ error: 'Date range cannot exceed 31 days' });
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成日期范围内所有日期的搜索模式
|
||||||
|
for (let d = new Date(start); d <= end; d.setDate(d.getDate() + 1)) {
|
||||||
|
const dateStr = d.toISOString().split('T')[0];
|
||||||
|
searchPatterns.push(`usage:${keyId}:model:daily:*:${dateStr}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.info(`📊 Custom date range patterns: ${searchPatterns.length} days from ${startDate} to ${endDate}`);
|
||||||
|
} else {
|
||||||
|
// 原有的预设期间逻辑
|
||||||
|
const pattern = period === 'daily' ?
|
||||||
|
`usage:${keyId}:model:daily:*:${today}` :
|
||||||
|
`usage:${keyId}:model:monthly:*:${currentMonth}`;
|
||||||
|
searchPatterns = [pattern];
|
||||||
|
logger.info(`📊 Preset period pattern: ${pattern}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 汇总所有匹配的数据
|
||||||
|
const modelStatsMap = new Map();
|
||||||
|
const modelStats = []; // 定义结果数组
|
||||||
|
|
||||||
|
for (const pattern of searchPatterns) {
|
||||||
|
const keys = await client.keys(pattern);
|
||||||
|
logger.info(`📊 Pattern ${pattern} found ${keys.length} keys`);
|
||||||
|
|
||||||
|
for (const key of keys) {
|
||||||
|
const match = key.match(/usage:.+:model:daily:(.+):\d{4}-\d{2}-\d{2}$/) ||
|
||||||
|
key.match(/usage:.+:model:monthly:(.+):\d{4}-\d{2}$/);
|
||||||
|
|
||||||
|
if (!match) {
|
||||||
|
logger.warn(`📊 Pattern mismatch for key: ${key}`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
const model = match[1];
|
||||||
|
const data = await client.hgetall(key);
|
||||||
|
|
||||||
|
if (data && Object.keys(data).length > 0) {
|
||||||
|
// 累加同一模型的数据
|
||||||
|
if (!modelStatsMap.has(model)) {
|
||||||
|
modelStatsMap.set(model, {
|
||||||
|
requests: 0,
|
||||||
|
inputTokens: 0,
|
||||||
|
outputTokens: 0,
|
||||||
|
cacheCreateTokens: 0,
|
||||||
|
cacheReadTokens: 0,
|
||||||
|
allTokens: 0
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const stats = modelStatsMap.get(model);
|
||||||
|
stats.requests += parseInt(data.requests) || 0;
|
||||||
|
stats.inputTokens += parseInt(data.inputTokens) || 0;
|
||||||
|
stats.outputTokens += parseInt(data.outputTokens) || 0;
|
||||||
|
stats.cacheCreateTokens += parseInt(data.cacheCreateTokens) || 0;
|
||||||
|
stats.cacheReadTokens += parseInt(data.cacheReadTokens) || 0;
|
||||||
|
stats.allTokens += parseInt(data.allTokens) || 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 将汇总的数据转换为最终结果
|
||||||
|
for (const [model, stats] of modelStatsMap) {
|
||||||
|
logger.info(`📊 Model ${model} aggregated data:`, stats);
|
||||||
|
|
||||||
|
const usage = {
|
||||||
|
input_tokens: stats.inputTokens,
|
||||||
|
output_tokens: stats.outputTokens,
|
||||||
|
cache_creation_input_tokens: stats.cacheCreateTokens,
|
||||||
|
cache_read_input_tokens: stats.cacheReadTokens
|
||||||
|
};
|
||||||
|
|
||||||
|
// 使用CostCalculator计算费用
|
||||||
|
const costData = CostCalculator.calculateCost(usage, model);
|
||||||
|
|
||||||
|
modelStats.push({
|
||||||
|
model,
|
||||||
|
requests: stats.requests,
|
||||||
|
inputTokens: stats.inputTokens,
|
||||||
|
outputTokens: stats.outputTokens,
|
||||||
|
cacheCreateTokens: stats.cacheCreateTokens,
|
||||||
|
cacheReadTokens: stats.cacheReadTokens,
|
||||||
|
allTokens: stats.allTokens,
|
||||||
|
// 添加费用信息
|
||||||
|
costs: costData.costs,
|
||||||
|
formatted: costData.formatted,
|
||||||
|
pricing: costData.pricing,
|
||||||
|
usingDynamicPricing: costData.usingDynamicPricing
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果没有找到模型级别的详细数据，尝试从汇总数据中生成展示
|
||||||
|
if (modelStats.length === 0) {
|
||||||
|
logger.info(`📊 No detailed model stats found, trying to get aggregate data for API key ${keyId}`);
|
||||||
|
|
||||||
|
// 尝试从API Keys列表中获取usage数据作为备选方案
|
||||||
|
try {
|
||||||
|
const apiKeys = await apiKeyService.getAllApiKeys();
|
||||||
|
const targetApiKey = apiKeys.find(key => key.id === keyId);
|
||||||
|
|
||||||
|
if (targetApiKey && targetApiKey.usage) {
|
||||||
|
logger.info(`📊 Found API key usage data from getAllApiKeys for ${keyId}:`, targetApiKey.usage);
|
||||||
|
|
||||||
|
// 从汇总数据创建展示条目
|
||||||
|
let usageData;
|
||||||
|
if (period === 'custom' || period === 'daily') {
|
||||||
|
// 对于自定义或日统计，使用daily数据或total数据
|
||||||
|
usageData = targetApiKey.usage.daily || targetApiKey.usage.total;
|
||||||
|
} else {
|
||||||
|
// 对于月统计，使用monthly数据或total数据
|
||||||
|
usageData = targetApiKey.usage.monthly || targetApiKey.usage.total;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (usageData && usageData.allTokens > 0) {
|
||||||
|
const usage = {
|
||||||
|
input_tokens: usageData.inputTokens || 0,
|
||||||
|
output_tokens: usageData.outputTokens || 0,
|
||||||
|
cache_creation_input_tokens: usageData.cacheCreateTokens || 0,
|
||||||
|
cache_read_input_tokens: usageData.cacheReadTokens || 0
|
||||||
|
};
|
||||||
|
|
||||||
|
// 对于汇总数据，使用默认模型计算费用
|
||||||
|
const costData = CostCalculator.calculateCost(usage, 'claude-3-5-sonnet-20241022');
|
||||||
|
|
||||||
|
modelStats.push({
|
||||||
|
model: '总体使用 (历史数据)',
|
||||||
|
requests: usageData.requests || 0,
|
||||||
|
inputTokens: usageData.inputTokens || 0,
|
||||||
|
outputTokens: usageData.outputTokens || 0,
|
||||||
|
cacheCreateTokens: usageData.cacheCreateTokens || 0,
|
||||||
|
cacheReadTokens: usageData.cacheReadTokens || 0,
|
||||||
|
allTokens: usageData.allTokens || 0,
|
||||||
|
// 添加费用信息
|
||||||
|
costs: costData.costs,
|
||||||
|
formatted: costData.formatted,
|
||||||
|
pricing: costData.pricing,
|
||||||
|
usingDynamicPricing: costData.usingDynamicPricing
|
||||||
|
});
|
||||||
|
|
||||||
|
logger.info('📊 Generated display data from API key usage stats');
|
||||||
|
} else {
|
||||||
|
logger.info(`📊 No usage data found for period ${period} in API key data`);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
logger.info(`📊 API key ${keyId} not found or has no usage data`);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Error fetching API key usage data:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 按总token数降序排列
|
||||||
|
modelStats.sort((a, b) => b.allTokens - a.allTokens);
|
||||||
|
|
||||||
|
logger.info(`📊 Returning ${modelStats.length} model stats for API key ${keyId}:`, modelStats);
|
||||||
|
|
||||||
|
res.json({ success: true, data: modelStats });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get API key model stats:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to get API key model stats', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
|
||||||
|
// 计算总体使用费用
|
||||||
|
router.get('/usage-costs', authenticateAdmin, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { period = 'all' } = req.query; // all, today, monthly
|
||||||
|
|
||||||
|
logger.info(`💰 Calculating usage costs for period: ${period}`);
|
||||||
|
|
||||||
|
// 获取所有API Keys的使用统计
|
||||||
|
const apiKeys = await apiKeyService.getAllApiKeys();
|
||||||
|
|
||||||
|
let totalCosts = {
|
||||||
|
inputCost: 0,
|
||||||
|
outputCost: 0,
|
||||||
|
cacheCreateCost: 0,
|
||||||
|
cacheReadCost: 0,
|
||||||
|
totalCost: 0
|
||||||
|
};
|
||||||
|
|
||||||
|
let modelCosts = {};
|
||||||
|
|
||||||
|
// 按模型统计费用
|
||||||
|
const client = redis.getClientSafe();
|
||||||
|
const today = new Date().toISOString().split('T')[0];
|
||||||
|
const currentMonth = `${new Date().getFullYear()}-${String(new Date().getMonth() + 1).padStart(2, '0')}`;
|
||||||
|
|
||||||
|
let pattern;
|
||||||
|
if (period === 'today') {
|
||||||
|
pattern = `usage:model:daily:*:${today}`;
|
||||||
|
} else if (period === 'monthly') {
|
||||||
|
pattern = `usage:model:monthly:*:${currentMonth}`;
|
||||||
|
} else {
|
||||||
|
// 全部时间，使用API Key汇总数据
|
||||||
|
for (const apiKey of apiKeys) {
|
||||||
|
if (apiKey.usage && apiKey.usage.total) {
|
||||||
|
const usage = {
|
||||||
|
input_tokens: apiKey.usage.total.inputTokens || 0,
|
||||||
|
output_tokens: apiKey.usage.total.outputTokens || 0,
|
||||||
|
cache_creation_input_tokens: apiKey.usage.total.cacheCreateTokens || 0,
|
||||||
|
cache_read_input_tokens: apiKey.usage.total.cacheReadTokens || 0
|
||||||
|
};
|
||||||
|
|
||||||
|
// 计算未知模型的费用（汇总数据）
|
||||||
|
const costResult = CostCalculator.calculateCost(usage, 'unknown');
|
||||||
|
totalCosts.inputCost += costResult.costs.input;
|
||||||
|
totalCosts.outputCost += costResult.costs.output;
|
||||||
|
totalCosts.cacheCreateCost += costResult.costs.cacheWrite;
|
||||||
|
totalCosts.cacheReadCost += costResult.costs.cacheRead;
|
||||||
|
totalCosts.totalCost += costResult.costs.total;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
success: true,
|
||||||
|
data: {
|
||||||
|
period,
|
||||||
|
totalCosts: {
|
||||||
|
...totalCosts,
|
||||||
|
formatted: {
|
||||||
|
inputCost: CostCalculator.formatCost(totalCosts.inputCost),
|
||||||
|
outputCost: CostCalculator.formatCost(totalCosts.outputCost),
|
||||||
|
cacheCreateCost: CostCalculator.formatCost(totalCosts.cacheCreateCost),
|
||||||
|
cacheReadCost: CostCalculator.formatCost(totalCosts.cacheReadCost),
|
||||||
|
totalCost: CostCalculator.formatCost(totalCosts.totalCost)
|
||||||
|
}
|
||||||
|
},
|
||||||
|
modelCosts: [],
|
||||||
|
pricingServiceStatus: pricingService.getStatus()
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 对于今日或本月，从Redis获取详细的模型统计
|
||||||
|
const keys = await client.keys(pattern);
|
||||||
|
|
||||||
|
for (const key of keys) {
|
||||||
|
const match = key.match(period === 'today' ?
|
||||||
|
/usage:model:daily:(.+):\d{4}-\d{2}-\d{2}$/ :
|
||||||
|
/usage:model:monthly:(.+):\d{4}-\d{2}$/
|
||||||
|
);
|
||||||
|
|
||||||
|
if (!match) continue;
|
||||||
|
|
||||||
|
const model = match[1];
|
||||||
|
const data = await client.hgetall(key);
|
||||||
|
|
||||||
|
if (data && Object.keys(data).length > 0) {
|
||||||
|
const usage = {
|
||||||
|
input_tokens: parseInt(data.inputTokens) || 0,
|
||||||
|
output_tokens: parseInt(data.outputTokens) || 0,
|
||||||
|
cache_creation_input_tokens: parseInt(data.cacheCreateTokens) || 0,
|
||||||
|
cache_read_input_tokens: parseInt(data.cacheReadTokens) || 0
|
||||||
|
};
|
||||||
|
|
||||||
|
const costResult = CostCalculator.calculateCost(usage, model);
|
||||||
|
|
||||||
|
// 累加总费用
|
||||||
|
totalCosts.inputCost += costResult.costs.input;
|
||||||
|
totalCosts.outputCost += costResult.costs.output;
|
||||||
|
totalCosts.cacheCreateCost += costResult.costs.cacheWrite;
|
||||||
|
totalCosts.cacheReadCost += costResult.costs.cacheRead;
|
||||||
|
totalCosts.totalCost += costResult.costs.total;
|
||||||
|
|
||||||
|
// 记录模型费用
|
||||||
|
modelCosts[model] = {
|
||||||
|
model,
|
||||||
|
requests: parseInt(data.requests) || 0,
|
||||||
|
usage,
|
||||||
|
costs: costResult.costs,
|
||||||
|
formatted: costResult.formatted,
|
||||||
|
usingDynamicPricing: costResult.usingDynamicPricing
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
success: true,
|
||||||
|
data: {
|
||||||
|
period,
|
||||||
|
totalCosts: {
|
||||||
|
...totalCosts,
|
||||||
|
formatted: {
|
||||||
|
inputCost: CostCalculator.formatCost(totalCosts.inputCost),
|
||||||
|
outputCost: CostCalculator.formatCost(totalCosts.outputCost),
|
||||||
|
cacheCreateCost: CostCalculator.formatCost(totalCosts.cacheCreateCost),
|
||||||
|
cacheReadCost: CostCalculator.formatCost(totalCosts.cacheReadCost),
|
||||||
|
totalCost: CostCalculator.formatCost(totalCosts.totalCost)
|
||||||
|
}
|
||||||
|
},
|
||||||
|
modelCosts: Object.values(modelCosts).sort((a, b) => b.costs.total - a.costs.total),
|
||||||
|
pricingServiceStatus: pricingService.getStatus()
|
||||||
|
}
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to calculate usage costs:', error);
|
||||||
|
res.status(500).json({ error: 'Failed to calculate usage costs', message: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = router;
|
||||||
225
src/routes/api.js
Normal file
225
src/routes/api.js
Normal file
@@ -0,0 +1,225 @@
|
|||||||
|
const express = require('express');
|
||||||
|
const claudeRelayService = require('../services/claudeRelayService');
|
||||||
|
const apiKeyService = require('../services/apiKeyService');
|
||||||
|
const { authenticateApiKey } = require('../middleware/auth');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
|
||||||
|
const router = express.Router();
|
||||||
|
|
||||||
|
// 🚀 Claude API messages 端点
|
||||||
|
router.post('/v1/messages', authenticateApiKey, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const startTime = Date.now();
|
||||||
|
|
||||||
|
// 严格的输入验证
|
||||||
|
if (!req.body || typeof req.body !== 'object') {
|
||||||
|
return res.status(400).json({
|
||||||
|
error: 'Invalid request',
|
||||||
|
message: 'Request body must be a valid JSON object'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!req.body.messages || !Array.isArray(req.body.messages)) {
|
||||||
|
return res.status(400).json({
|
||||||
|
error: 'Invalid request',
|
||||||
|
message: 'Missing or invalid field: messages (must be an array)'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
if (req.body.messages.length === 0) {
|
||||||
|
return res.status(400).json({
|
||||||
|
error: 'Invalid request',
|
||||||
|
message: 'Messages array cannot be empty'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否为流式请求
|
||||||
|
const isStream = req.body.stream === true;
|
||||||
|
|
||||||
|
logger.api(`🚀 Processing ${isStream ? 'stream' : 'non-stream'} request for key: ${req.apiKey.name}`);
|
||||||
|
|
||||||
|
if (isStream) {
|
||||||
|
// 流式响应 - 只使用官方真实usage数据
|
||||||
|
res.setHeader('Content-Type', 'text/event-stream');
|
||||||
|
res.setHeader('Cache-Control', 'no-cache');
|
||||||
|
res.setHeader('Connection', 'keep-alive');
|
||||||
|
res.setHeader('Access-Control-Allow-Origin', '*');
|
||||||
|
|
||||||
|
let usageDataCaptured = false;
|
||||||
|
|
||||||
|
// 使用自定义流处理器来捕获usage数据
|
||||||
|
await claudeRelayService.relayStreamRequestWithUsageCapture(req.body, req.apiKey, res, (usageData) => {
|
||||||
|
// 回调函数：当检测到完整usage数据时记录真实token使用量
|
||||||
|
logger.info('🎯 Usage callback triggered with complete data:', JSON.stringify(usageData, null, 2));
|
||||||
|
|
||||||
|
if (usageData && usageData.input_tokens !== undefined && usageData.output_tokens !== undefined) {
|
||||||
|
const inputTokens = usageData.input_tokens || 0;
|
||||||
|
const outputTokens = usageData.output_tokens || 0;
|
||||||
|
const cacheCreateTokens = usageData.cache_creation_input_tokens || 0;
|
||||||
|
const cacheReadTokens = usageData.cache_read_input_tokens || 0;
|
||||||
|
const model = usageData.model || 'unknown';
|
||||||
|
|
||||||
|
// 记录真实的token使用量（包含模型信息和所有4种token）
|
||||||
|
apiKeyService.recordUsage(req.apiKey.id, inputTokens, outputTokens, cacheCreateTokens, cacheReadTokens, model).catch(error => {
|
||||||
|
logger.error('❌ Failed to record stream usage:', error);
|
||||||
|
});
|
||||||
|
|
||||||
|
usageDataCaptured = true;
|
||||||
|
logger.api(`📊 Stream usage recorded (real) - Model: ${model}, Input: ${inputTokens}, Output: ${outputTokens}, Cache Create: ${cacheCreateTokens}, Cache Read: ${cacheReadTokens}, Total: ${inputTokens + outputTokens + cacheCreateTokens + cacheReadTokens} tokens`);
|
||||||
|
} else {
|
||||||
|
logger.warn('⚠️ Usage callback triggered but data is incomplete:', JSON.stringify(usageData));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 流式请求完成后 - 如果没有捕获到usage数据，记录警告但不进行估算
|
||||||
|
setTimeout(() => {
|
||||||
|
if (!usageDataCaptured) {
|
||||||
|
logger.warn('⚠️ No usage data captured from SSE stream - no statistics recorded (official data only)');
|
||||||
|
}
|
||||||
|
}, 1000); // 1秒后检查
|
||||||
|
} else {
|
||||||
|
// 非流式响应 - 只使用官方真实usage数据
|
||||||
|
logger.info('📄 Starting non-streaming request', {
|
||||||
|
apiKeyId: req.apiKey.id,
|
||||||
|
apiKeyName: req.apiKey.name
|
||||||
|
});
|
||||||
|
|
||||||
|
const response = await claudeRelayService.relayRequest(req.body, req.apiKey);
|
||||||
|
|
||||||
|
logger.info('📡 Claude API response received', {
|
||||||
|
statusCode: response.statusCode,
|
||||||
|
headers: JSON.stringify(response.headers),
|
||||||
|
bodyLength: response.body ? response.body.length : 0
|
||||||
|
});
|
||||||
|
|
||||||
|
res.status(response.statusCode);
|
||||||
|
|
||||||
|
// 设置响应头
|
||||||
|
Object.keys(response.headers).forEach(key => {
|
||||||
|
if (key.toLowerCase() !== 'content-encoding') {
|
||||||
|
res.setHeader(key, response.headers[key]);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
let usageRecorded = false;
|
||||||
|
|
||||||
|
// 尝试解析JSON响应并提取usage信息
|
||||||
|
try {
|
||||||
|
const jsonData = JSON.parse(response.body);
|
||||||
|
|
||||||
|
logger.info('📊 Parsed Claude API response:', JSON.stringify(jsonData, null, 2));
|
||||||
|
|
||||||
|
// 从Claude API响应中提取usage信息（完整的token分类体系）
|
||||||
|
if (jsonData.usage && jsonData.usage.input_tokens !== undefined && jsonData.usage.output_tokens !== undefined) {
|
||||||
|
const inputTokens = jsonData.usage.input_tokens || 0;
|
||||||
|
const outputTokens = jsonData.usage.output_tokens || 0;
|
||||||
|
const cacheCreateTokens = jsonData.usage.cache_creation_input_tokens || 0;
|
||||||
|
const cacheReadTokens = jsonData.usage.cache_read_input_tokens || 0;
|
||||||
|
const model = jsonData.model || req.body.model || 'unknown';
|
||||||
|
|
||||||
|
// 记录真实的token使用量（包含模型信息和所有4种token）
|
||||||
|
await apiKeyService.recordUsage(req.apiKey.id, inputTokens, outputTokens, cacheCreateTokens, cacheReadTokens, model);
|
||||||
|
|
||||||
|
usageRecorded = true;
|
||||||
|
logger.api(`📊 Non-stream usage recorded (real) - Model: ${model}, Input: ${inputTokens}, Output: ${outputTokens}, Cache Create: ${cacheCreateTokens}, Cache Read: ${cacheReadTokens}, Total: ${inputTokens + outputTokens + cacheCreateTokens + cacheReadTokens} tokens`);
|
||||||
|
} else {
|
||||||
|
logger.warn('⚠️ No usage data found in Claude API JSON response');
|
||||||
|
}
|
||||||
|
|
||||||
|
res.json(jsonData);
|
||||||
|
} catch (parseError) {
|
||||||
|
logger.warn('⚠️ Failed to parse Claude API response as JSON:', parseError.message);
|
||||||
|
logger.info('📄 Raw response body:', response.body);
|
||||||
|
res.send(response.body);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 如果没有记录usage，只记录警告，不进行估算
|
||||||
|
if (!usageRecorded) {
|
||||||
|
logger.warn('⚠️ No usage data recorded for non-stream request - no statistics recorded (official data only)');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const duration = Date.now() - startTime;
|
||||||
|
logger.api(`✅ Request completed in ${duration}ms for key: ${req.apiKey.name}`);
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Claude relay error:', error);
|
||||||
|
|
||||||
|
if (!res.headersSent) {
|
||||||
|
res.status(500).json({
|
||||||
|
error: 'Relay service error',
|
||||||
|
message: error.message
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🏥 健康检查端点
|
||||||
|
router.get('/health', async (req, res) => {
|
||||||
|
try {
|
||||||
|
const healthStatus = await claudeRelayService.healthCheck();
|
||||||
|
|
||||||
|
res.status(healthStatus.healthy ? 200 : 503).json({
|
||||||
|
status: healthStatus.healthy ? 'healthy' : 'unhealthy',
|
||||||
|
service: 'claude-relay-service',
|
||||||
|
version: '1.0.0',
|
||||||
|
...healthStatus
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Health check error:', error);
|
||||||
|
res.status(503).json({
|
||||||
|
status: 'unhealthy',
|
||||||
|
service: 'claude-relay-service',
|
||||||
|
error: error.message,
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 📊 API Key状态检查端点
|
||||||
|
router.get('/v1/key-info', authenticateApiKey, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const usage = await apiKeyService.getUsageStats(req.apiKey.id);
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
keyInfo: {
|
||||||
|
id: req.apiKey.id,
|
||||||
|
name: req.apiKey.name,
|
||||||
|
tokenLimit: req.apiKey.tokenLimit,
|
||||||
|
requestLimit: req.apiKey.requestLimit,
|
||||||
|
usage
|
||||||
|
},
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Key info error:', error);
|
||||||
|
res.status(500).json({
|
||||||
|
error: 'Failed to get key info',
|
||||||
|
message: error.message
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 📈 使用统计端点
|
||||||
|
router.get('/v1/usage', authenticateApiKey, async (req, res) => {
|
||||||
|
try {
|
||||||
|
const usage = await apiKeyService.getUsageStats(req.apiKey.id);
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
usage,
|
||||||
|
limits: {
|
||||||
|
tokens: req.apiKey.tokenLimit,
|
||||||
|
requests: req.apiKey.requestLimit
|
||||||
|
},
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Usage stats error:', error);
|
||||||
|
res.status(500).json({
|
||||||
|
error: 'Failed to get usage stats',
|
||||||
|
message: error.message
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = router;
|
||||||
202
src/routes/web.js
Normal file
202
src/routes/web.js
Normal file
@@ -0,0 +1,202 @@
|
|||||||
|
const express = require('express');
|
||||||
|
const bcrypt = require('bcryptjs');
|
||||||
|
const crypto = require('crypto');
|
||||||
|
const path = require('path');
|
||||||
|
const fs = require('fs');
|
||||||
|
const redis = require('../models/redis');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
const config = require('../../config/config');
|
||||||
|
|
||||||
|
const router = express.Router();
|
||||||
|
|
||||||
|
// 🏠 服务静态文件
|
||||||
|
router.use('/assets', express.static(path.join(__dirname, '../../web/assets')));
|
||||||
|
|
||||||
|
// 🔒 Web管理界面文件白名单 - 仅允许这些特定文件
|
||||||
|
const ALLOWED_FILES = {
|
||||||
|
'index.html': {
|
||||||
|
path: path.join(__dirname, '../../web/admin/index.html'),
|
||||||
|
contentType: 'text/html; charset=utf-8'
|
||||||
|
},
|
||||||
|
'app.js': {
|
||||||
|
path: path.join(__dirname, '../../web/admin/app.js'),
|
||||||
|
contentType: 'application/javascript; charset=utf-8'
|
||||||
|
},
|
||||||
|
'style.css': {
|
||||||
|
path: path.join(__dirname, '../../web/admin/style.css'),
|
||||||
|
contentType: 'text/css; charset=utf-8'
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// 🛡️ 安全文件服务函数
|
||||||
|
function serveWhitelistedFile(req, res, filename) {
|
||||||
|
const fileConfig = ALLOWED_FILES[filename];
|
||||||
|
|
||||||
|
if (!fileConfig) {
|
||||||
|
logger.security(`🚨 Attempted access to non-whitelisted file: ${filename}`);
|
||||||
|
return res.status(404).json({ error: 'File not found' });
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 检查文件是否存在
|
||||||
|
if (!fs.existsSync(fileConfig.path)) {
|
||||||
|
logger.error(`❌ Whitelisted file not found: ${fileConfig.path}`);
|
||||||
|
return res.status(404).json({ error: 'File not found' });
|
||||||
|
}
|
||||||
|
|
||||||
|
// 读取并返回文件内容
|
||||||
|
const content = fs.readFileSync(fileConfig.path, 'utf8');
|
||||||
|
res.setHeader('Content-Type', fileConfig.contentType);
|
||||||
|
res.setHeader('Cache-Control', 'no-cache, no-store, must-revalidate');
|
||||||
|
res.setHeader('Pragma', 'no-cache');
|
||||||
|
res.setHeader('Expires', '0');
|
||||||
|
res.send(content);
|
||||||
|
|
||||||
|
logger.info(`📄 Served whitelisted file: ${filename}`);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error(`❌ Error serving file ${filename}:`, error);
|
||||||
|
res.status(500).json({ error: 'Internal server error' });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔐 管理员登录
|
||||||
|
router.post('/auth/login', async (req, res) => {
|
||||||
|
try {
|
||||||
|
const { username, password } = req.body;
|
||||||
|
|
||||||
|
if (!username || !password) {
|
||||||
|
return res.status(400).json({
|
||||||
|
error: 'Missing credentials',
|
||||||
|
message: 'Username and password are required'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 从Redis获取管理员信息
|
||||||
|
const adminData = await redis.getSession('admin_credentials');
|
||||||
|
|
||||||
|
if (!adminData || Object.keys(adminData).length === 0) {
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Invalid credentials',
|
||||||
|
message: 'Invalid username or password'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 验证用户名和密码
|
||||||
|
const isValidUsername = adminData.username === username;
|
||||||
|
const isValidPassword = await bcrypt.compare(password, adminData.passwordHash);
|
||||||
|
|
||||||
|
if (!isValidUsername || !isValidPassword) {
|
||||||
|
logger.security(`🔒 Failed login attempt for username: ${username}`);
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Invalid credentials',
|
||||||
|
message: 'Invalid username or password'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 生成会话token
|
||||||
|
const sessionId = crypto.randomBytes(32).toString('hex');
|
||||||
|
|
||||||
|
// 存储会话
|
||||||
|
const sessionData = {
|
||||||
|
username: adminData.username,
|
||||||
|
loginTime: new Date().toISOString(),
|
||||||
|
lastActivity: new Date().toISOString()
|
||||||
|
};
|
||||||
|
|
||||||
|
await redis.setSession(sessionId, sessionData, config.security.adminSessionTimeout);
|
||||||
|
|
||||||
|
// 更新最后登录时间
|
||||||
|
adminData.lastLogin = new Date().toISOString();
|
||||||
|
await redis.setSession('admin_credentials', adminData);
|
||||||
|
|
||||||
|
logger.success(`🔐 Admin login successful: ${username}`);
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
success: true,
|
||||||
|
token: sessionId,
|
||||||
|
expiresIn: config.security.adminSessionTimeout
|
||||||
|
});
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Login error:', error);
|
||||||
|
res.status(500).json({
|
||||||
|
error: 'Login failed',
|
||||||
|
message: 'Internal server error'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🚪 管理员登出
|
||||||
|
router.post('/auth/logout', async (req, res) => {
|
||||||
|
try {
|
||||||
|
const token = req.headers['authorization']?.replace('Bearer ', '') || req.cookies?.adminToken;
|
||||||
|
|
||||||
|
if (token) {
|
||||||
|
await redis.deleteSession(token);
|
||||||
|
logger.success('🚪 Admin logout successful');
|
||||||
|
}
|
||||||
|
|
||||||
|
res.json({ success: true, message: 'Logout successful' });
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Logout error:', error);
|
||||||
|
res.status(500).json({
|
||||||
|
error: 'Logout failed',
|
||||||
|
message: 'Internal server error'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🔄 刷新token
|
||||||
|
router.post('/auth/refresh', async (req, res) => {
|
||||||
|
try {
|
||||||
|
const token = req.headers['authorization']?.replace('Bearer ', '') || req.cookies?.adminToken;
|
||||||
|
|
||||||
|
if (!token) {
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'No token provided',
|
||||||
|
message: 'Authentication required'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
const sessionData = await redis.getSession(token);
|
||||||
|
|
||||||
|
if (!sessionData) {
|
||||||
|
return res.status(401).json({
|
||||||
|
error: 'Invalid token',
|
||||||
|
message: 'Session expired or invalid'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新最后活动时间
|
||||||
|
sessionData.lastActivity = new Date().toISOString();
|
||||||
|
await redis.setSession(token, sessionData, config.security.adminSessionTimeout);
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
success: true,
|
||||||
|
token: token,
|
||||||
|
expiresIn: config.security.adminSessionTimeout
|
||||||
|
});
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Token refresh error:', error);
|
||||||
|
res.status(500).json({
|
||||||
|
error: 'Token refresh failed',
|
||||||
|
message: 'Internal server error'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🌐 Web管理界面路由 - 使用固定白名单
|
||||||
|
router.get('/', (req, res) => {
|
||||||
|
serveWhitelistedFile(req, res, 'index.html');
|
||||||
|
});
|
||||||
|
|
||||||
|
router.get('/app.js', (req, res) => {
|
||||||
|
serveWhitelistedFile(req, res, 'app.js');
|
||||||
|
});
|
||||||
|
|
||||||
|
router.get('/style.css', (req, res) => {
|
||||||
|
serveWhitelistedFile(req, res, 'style.css');
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = router;
|
||||||
271
src/services/apiKeyService.js
Normal file
271
src/services/apiKeyService.js
Normal file
@@ -0,0 +1,271 @@
|
|||||||
|
const crypto = require('crypto');
|
||||||
|
const { v4: uuidv4 } = require('uuid');
|
||||||
|
const config = require('../../config/config');
|
||||||
|
const redis = require('../models/redis');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
|
||||||
|
class ApiKeyService {
|
||||||
|
constructor() {
|
||||||
|
this.prefix = config.security.apiKeyPrefix;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔑 生成新的API Key
|
||||||
|
async generateApiKey(options = {}) {
|
||||||
|
const {
|
||||||
|
name = 'Unnamed Key',
|
||||||
|
description = '',
|
||||||
|
tokenLimit = config.limits.defaultTokenLimit,
|
||||||
|
requestLimit = config.limits.defaultRequestLimit,
|
||||||
|
expiresAt = null,
|
||||||
|
claudeAccountId = null,
|
||||||
|
isActive = true
|
||||||
|
} = options;
|
||||||
|
|
||||||
|
// 生成简单的API Key (64字符十六进制)
|
||||||
|
const apiKey = `${this.prefix}${this._generateSecretKey()}`;
|
||||||
|
const keyId = uuidv4();
|
||||||
|
const hashedKey = this._hashApiKey(apiKey);
|
||||||
|
|
||||||
|
const keyData = {
|
||||||
|
id: keyId,
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
apiKey: hashedKey,
|
||||||
|
tokenLimit: String(tokenLimit ?? 0),
|
||||||
|
requestLimit: String(requestLimit ?? 0),
|
||||||
|
isActive: String(isActive),
|
||||||
|
claudeAccountId: claudeAccountId || '',
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
lastUsedAt: '',
|
||||||
|
expiresAt: expiresAt || '',
|
||||||
|
createdBy: 'admin' // 可以根据需要扩展用户系统
|
||||||
|
};
|
||||||
|
|
||||||
|
// 保存API Key数据并建立哈希映射
|
||||||
|
await redis.setApiKey(keyId, keyData, hashedKey);
|
||||||
|
|
||||||
|
logger.success(`🔑 Generated new API key: ${name} (${keyId})`);
|
||||||
|
|
||||||
|
return {
|
||||||
|
id: keyId,
|
||||||
|
apiKey, // 只在创建时返回完整的key
|
||||||
|
name: keyData.name,
|
||||||
|
description: keyData.description,
|
||||||
|
tokenLimit: parseInt(keyData.tokenLimit),
|
||||||
|
requestLimit: parseInt(keyData.requestLimit),
|
||||||
|
isActive: keyData.isActive === 'true',
|
||||||
|
claudeAccountId: keyData.claudeAccountId,
|
||||||
|
createdAt: keyData.createdAt,
|
||||||
|
expiresAt: keyData.expiresAt,
|
||||||
|
createdBy: keyData.createdBy
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔍 验证API Key
|
||||||
|
async validateApiKey(apiKey) {
|
||||||
|
try {
|
||||||
|
if (!apiKey || !apiKey.startsWith(this.prefix)) {
|
||||||
|
return { valid: false, error: 'Invalid API key format' };
|
||||||
|
}
|
||||||
|
|
||||||
|
// 计算API Key的哈希值
|
||||||
|
const hashedKey = this._hashApiKey(apiKey);
|
||||||
|
|
||||||
|
// 通过哈希值直接查找API Key（性能优化）
|
||||||
|
const keyData = await redis.findApiKeyByHash(hashedKey);
|
||||||
|
|
||||||
|
if (!keyData) {
|
||||||
|
return { valid: false, error: 'API key not found' };
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否激活
|
||||||
|
if (keyData.isActive !== 'true') {
|
||||||
|
return { valid: false, error: 'API key is disabled' };
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否过期
|
||||||
|
if (keyData.expiresAt && new Date() > new Date(keyData.expiresAt)) {
|
||||||
|
return { valid: false, error: 'API key has expired' };
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查使用限制
|
||||||
|
const usage = await redis.getUsageStats(keyData.id);
|
||||||
|
const tokenLimit = parseInt(keyData.tokenLimit);
|
||||||
|
const requestLimit = parseInt(keyData.requestLimit);
|
||||||
|
|
||||||
|
if (tokenLimit > 0 && usage.total.tokens >= tokenLimit) {
|
||||||
|
return { valid: false, error: 'Token limit exceeded' };
|
||||||
|
}
|
||||||
|
|
||||||
|
if (requestLimit > 0 && usage.total.requests >= requestLimit) {
|
||||||
|
return { valid: false, error: 'Request limit exceeded' };
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新最后使用时间（优化：只在实际API调用时更新，而不是验证时）
|
||||||
|
// 注意：lastUsedAt的更新已移至recordUsage方法中
|
||||||
|
|
||||||
|
logger.api(`🔓 API key validated successfully: ${keyData.id}`);
|
||||||
|
|
||||||
|
return {
|
||||||
|
valid: true,
|
||||||
|
keyData: {
|
||||||
|
id: keyData.id,
|
||||||
|
name: keyData.name,
|
||||||
|
claudeAccountId: keyData.claudeAccountId,
|
||||||
|
tokenLimit: parseInt(keyData.tokenLimit),
|
||||||
|
requestLimit: parseInt(keyData.requestLimit),
|
||||||
|
usage
|
||||||
|
}
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ API key validation error:', error);
|
||||||
|
return { valid: false, error: 'Internal validation error' };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📋 获取所有API Keys
|
||||||
|
async getAllApiKeys() {
|
||||||
|
try {
|
||||||
|
const apiKeys = await redis.getAllApiKeys();
|
||||||
|
|
||||||
|
// 为每个key添加使用统计
|
||||||
|
for (const key of apiKeys) {
|
||||||
|
key.usage = await redis.getUsageStats(key.id);
|
||||||
|
key.tokenLimit = parseInt(key.tokenLimit);
|
||||||
|
key.requestLimit = parseInt(key.requestLimit);
|
||||||
|
key.isActive = key.isActive === 'true';
|
||||||
|
delete key.apiKey; // 不返回哈希后的key
|
||||||
|
}
|
||||||
|
|
||||||
|
return apiKeys;
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get API keys:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📝 更新API Key
|
||||||
|
async updateApiKey(keyId, updates) {
|
||||||
|
try {
|
||||||
|
const keyData = await redis.getApiKey(keyId);
|
||||||
|
if (!keyData || Object.keys(keyData).length === 0) {
|
||||||
|
throw new Error('API key not found');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 允许更新的字段
|
||||||
|
const allowedUpdates = ['name', 'description', 'tokenLimit', 'requestLimit', 'isActive', 'claudeAccountId', 'expiresAt'];
|
||||||
|
const updatedData = { ...keyData };
|
||||||
|
|
||||||
|
for (const [field, value] of Object.entries(updates)) {
|
||||||
|
if (allowedUpdates.includes(field)) {
|
||||||
|
updatedData[field] = (value != null ? value : '').toString();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
updatedData.updatedAt = new Date().toISOString();
|
||||||
|
|
||||||
|
// 更新时不需要重新建立哈希映射，因为API Key本身没有变化
|
||||||
|
await redis.setApiKey(keyId, updatedData);
|
||||||
|
|
||||||
|
logger.success(`📝 Updated API key: ${keyId}`);
|
||||||
|
|
||||||
|
return { success: true };
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to update API key:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🗑️ 删除API Key
|
||||||
|
async deleteApiKey(keyId) {
|
||||||
|
try {
|
||||||
|
const result = await redis.deleteApiKey(keyId);
|
||||||
|
|
||||||
|
if (result === 0) {
|
||||||
|
throw new Error('API key not found');
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.success(`🗑️ Deleted API key: ${keyId}`);
|
||||||
|
|
||||||
|
return { success: true };
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to delete API key:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📊 记录使用情况（支持缓存token）
|
||||||
|
async recordUsage(keyId, inputTokens = 0, outputTokens = 0, cacheCreateTokens = 0, cacheReadTokens = 0, model = 'unknown') {
|
||||||
|
try {
|
||||||
|
const totalTokens = inputTokens + outputTokens + cacheCreateTokens + cacheReadTokens;
|
||||||
|
await redis.incrementTokenUsage(keyId, totalTokens, inputTokens, outputTokens, cacheCreateTokens, cacheReadTokens, model);
|
||||||
|
|
||||||
|
// 更新最后使用时间（性能优化：只在实际使用时更新）
|
||||||
|
const keyData = await redis.getApiKey(keyId);
|
||||||
|
if (keyData && Object.keys(keyData).length > 0) {
|
||||||
|
keyData.lastUsedAt = new Date().toISOString();
|
||||||
|
// 使用记录时不需要重新建立哈希映射
|
||||||
|
await redis.setApiKey(keyId, keyData);
|
||||||
|
}
|
||||||
|
|
||||||
|
const logParts = [`Model: ${model}`, `Input: ${inputTokens}`, `Output: ${outputTokens}`];
|
||||||
|
if (cacheCreateTokens > 0) logParts.push(`Cache Create: ${cacheCreateTokens}`);
|
||||||
|
if (cacheReadTokens > 0) logParts.push(`Cache Read: ${cacheReadTokens}`);
|
||||||
|
logParts.push(`Total: ${totalTokens} tokens`);
|
||||||
|
|
||||||
|
logger.database(`📊 Recorded usage: ${keyId} - ${logParts.join(', ')}`);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to record usage:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔐 生成密钥
|
||||||
|
_generateSecretKey() {
|
||||||
|
return crypto.randomBytes(32).toString('hex');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔒 哈希API Key
|
||||||
|
_hashApiKey(apiKey) {
|
||||||
|
return crypto.createHash('sha256').update(apiKey + config.security.encryptionKey).digest('hex');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📈 获取使用统计
|
||||||
|
async getUsageStats(keyId) {
|
||||||
|
return await redis.getUsageStats(keyId);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🚦 检查速率限制
|
||||||
|
async checkRateLimit(keyId, limit = null) {
|
||||||
|
const rateLimit = limit || config.rateLimit.maxRequests;
|
||||||
|
const window = Math.floor(config.rateLimit.windowMs / 1000);
|
||||||
|
|
||||||
|
return await redis.checkRateLimit(`apikey:${keyId}`, rateLimit, window);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🧹 清理过期的API Keys
|
||||||
|
async cleanupExpiredKeys() {
|
||||||
|
try {
|
||||||
|
const apiKeys = await redis.getAllApiKeys();
|
||||||
|
const now = new Date();
|
||||||
|
let cleanedCount = 0;
|
||||||
|
|
||||||
|
for (const key of apiKeys) {
|
||||||
|
if (key.expiresAt && new Date(key.expiresAt) < now) {
|
||||||
|
await redis.deleteApiKey(key.id);
|
||||||
|
cleanedCount++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (cleanedCount > 0) {
|
||||||
|
logger.success(`🧹 Cleaned up ${cleanedCount} expired API keys`);
|
||||||
|
}
|
||||||
|
|
||||||
|
return cleanedCount;
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to cleanup expired keys:', error);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = new ApiKeyService();
|
||||||
452
src/services/claudeAccountService.js
Normal file
452
src/services/claudeAccountService.js
Normal file
@@ -0,0 +1,452 @@
|
|||||||
|
const { v4: uuidv4 } = require('uuid');
|
||||||
|
const crypto = require('crypto');
|
||||||
|
const { SocksProxyAgent } = require('socks-proxy-agent');
|
||||||
|
const { HttpsProxyAgent } = require('https-proxy-agent');
|
||||||
|
const axios = require('axios');
|
||||||
|
const redis = require('../models/redis');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
const config = require('../../config/config');
|
||||||
|
|
||||||
|
class ClaudeAccountService {
|
||||||
|
constructor() {
|
||||||
|
this.claudeApiUrl = 'https://console.anthropic.com/v1/oauth/token';
|
||||||
|
this.claudeOauthClientId = '9d1c250a-e61b-44d9-88ed-5944d1962f5e';
|
||||||
|
|
||||||
|
// 加密相关常量
|
||||||
|
this.ENCRYPTION_ALGORITHM = 'aes-256-cbc';
|
||||||
|
this.ENCRYPTION_SALT = 'salt';
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🏢 创建Claude账户
|
||||||
|
async createAccount(options = {}) {
|
||||||
|
const {
|
||||||
|
name = 'Unnamed Account',
|
||||||
|
description = '',
|
||||||
|
email = '',
|
||||||
|
password = '',
|
||||||
|
refreshToken = '',
|
||||||
|
claudeAiOauth = null, // Claude标准格式的OAuth数据
|
||||||
|
proxy = null, // { type: 'socks5', host: 'localhost', port: 1080, username: '', password: '' }
|
||||||
|
isActive = true
|
||||||
|
} = options;
|
||||||
|
|
||||||
|
const accountId = uuidv4();
|
||||||
|
|
||||||
|
let accountData;
|
||||||
|
|
||||||
|
if (claudeAiOauth) {
|
||||||
|
// 使用Claude标准格式的OAuth数据
|
||||||
|
accountData = {
|
||||||
|
id: accountId,
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
email: this._encryptSensitiveData(email),
|
||||||
|
password: this._encryptSensitiveData(password),
|
||||||
|
claudeAiOauth: this._encryptSensitiveData(JSON.stringify(claudeAiOauth)),
|
||||||
|
accessToken: this._encryptSensitiveData(claudeAiOauth.accessToken),
|
||||||
|
refreshToken: this._encryptSensitiveData(claudeAiOauth.refreshToken),
|
||||||
|
expiresAt: claudeAiOauth.expiresAt.toString(),
|
||||||
|
scopes: claudeAiOauth.scopes.join(' '),
|
||||||
|
proxy: proxy ? JSON.stringify(proxy) : '',
|
||||||
|
isActive: isActive.toString(),
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
lastUsedAt: '',
|
||||||
|
lastRefreshAt: '',
|
||||||
|
status: 'active', // 有OAuth数据的账户直接设为active
|
||||||
|
errorMessage: ''
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
// 兼容旧格式
|
||||||
|
accountData = {
|
||||||
|
id: accountId,
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
email: this._encryptSensitiveData(email),
|
||||||
|
password: this._encryptSensitiveData(password),
|
||||||
|
refreshToken: this._encryptSensitiveData(refreshToken),
|
||||||
|
accessToken: '',
|
||||||
|
expiresAt: '',
|
||||||
|
scopes: '',
|
||||||
|
proxy: proxy ? JSON.stringify(proxy) : '',
|
||||||
|
isActive: isActive.toString(),
|
||||||
|
createdAt: new Date().toISOString(),
|
||||||
|
lastUsedAt: '',
|
||||||
|
lastRefreshAt: '',
|
||||||
|
status: 'created', // created, active, expired, error
|
||||||
|
errorMessage: ''
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
await redis.setClaudeAccount(accountId, accountData);
|
||||||
|
|
||||||
|
logger.success(`🏢 Created Claude account: ${name} (${accountId})`);
|
||||||
|
|
||||||
|
return {
|
||||||
|
id: accountId,
|
||||||
|
name,
|
||||||
|
description,
|
||||||
|
email,
|
||||||
|
isActive,
|
||||||
|
proxy,
|
||||||
|
status: accountData.status,
|
||||||
|
createdAt: accountData.createdAt,
|
||||||
|
expiresAt: accountData.expiresAt,
|
||||||
|
scopes: claudeAiOauth ? claudeAiOauth.scopes : []
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔄 刷新Claude账户token
|
||||||
|
async refreshAccountToken(accountId) {
|
||||||
|
try {
|
||||||
|
const accountData = await redis.getClaudeAccount(accountId);
|
||||||
|
|
||||||
|
if (!accountData || Object.keys(accountData).length === 0) {
|
||||||
|
throw new Error('Account not found');
|
||||||
|
}
|
||||||
|
|
||||||
|
const refreshToken = this._decryptSensitiveData(accountData.refreshToken);
|
||||||
|
|
||||||
|
if (!refreshToken) {
|
||||||
|
throw new Error('No refresh token available');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建代理agent
|
||||||
|
const agent = this._createProxyAgent(accountData.proxy);
|
||||||
|
|
||||||
|
const response = await axios.post(this.claudeApiUrl, {
|
||||||
|
grant_type: 'refresh_token',
|
||||||
|
refresh_token: refreshToken,
|
||||||
|
client_id: this.claudeOauthClientId
|
||||||
|
}, {
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
'Accept': 'application/json, text/plain, */*',
|
||||||
|
'User-Agent': 'claude-relay-service/1.0.0'
|
||||||
|
},
|
||||||
|
httpsAgent: agent,
|
||||||
|
timeout: 30000
|
||||||
|
});
|
||||||
|
|
||||||
|
if (response.status === 200) {
|
||||||
|
const { access_token, refresh_token, expires_in } = response.data;
|
||||||
|
|
||||||
|
// 更新账户数据
|
||||||
|
accountData.accessToken = this._encryptSensitiveData(access_token);
|
||||||
|
accountData.refreshToken = this._encryptSensitiveData(refresh_token);
|
||||||
|
accountData.expiresAt = (Date.now() + (expires_in * 1000)).toString();
|
||||||
|
accountData.lastRefreshAt = new Date().toISOString();
|
||||||
|
accountData.status = 'active';
|
||||||
|
accountData.errorMessage = '';
|
||||||
|
|
||||||
|
await redis.setClaudeAccount(accountId, accountData);
|
||||||
|
|
||||||
|
logger.success(`🔄 Refreshed token for account: ${accountData.name} (${accountId})`);
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
accessToken: access_token,
|
||||||
|
expiresAt: accountData.expiresAt
|
||||||
|
};
|
||||||
|
} else {
|
||||||
|
throw new Error(`Token refresh failed with status: ${response.status}`);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.error(`❌ Failed to refresh token for account ${accountId}:`, error);
|
||||||
|
|
||||||
|
// 更新错误状态
|
||||||
|
const accountData = await redis.getClaudeAccount(accountId);
|
||||||
|
if (accountData) {
|
||||||
|
accountData.status = 'error';
|
||||||
|
accountData.errorMessage = error.message;
|
||||||
|
await redis.setClaudeAccount(accountId, accountData);
|
||||||
|
}
|
||||||
|
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🎯 获取有效的访问token
|
||||||
|
async getValidAccessToken(accountId) {
|
||||||
|
try {
|
||||||
|
const accountData = await redis.getClaudeAccount(accountId);
|
||||||
|
|
||||||
|
if (!accountData || Object.keys(accountData).length === 0) {
|
||||||
|
throw new Error('Account not found');
|
||||||
|
}
|
||||||
|
|
||||||
|
if (accountData.isActive !== 'true') {
|
||||||
|
throw new Error('Account is disabled');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查token是否过期
|
||||||
|
const expiresAt = parseInt(accountData.expiresAt);
|
||||||
|
const now = Date.now();
|
||||||
|
|
||||||
|
if (!expiresAt || now >= (expiresAt - 10000)) { // 10秒提前刷新
|
||||||
|
logger.info(`🔄 Token expired/expiring for account ${accountId}, refreshing...`);
|
||||||
|
const refreshResult = await this.refreshAccountToken(accountId);
|
||||||
|
return refreshResult.accessToken;
|
||||||
|
}
|
||||||
|
|
||||||
|
const accessToken = this._decryptSensitiveData(accountData.accessToken);
|
||||||
|
|
||||||
|
if (!accessToken) {
|
||||||
|
throw new Error('No access token available');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 更新最后使用时间
|
||||||
|
accountData.lastUsedAt = new Date().toISOString();
|
||||||
|
await redis.setClaudeAccount(accountId, accountData);
|
||||||
|
|
||||||
|
return accessToken;
|
||||||
|
} catch (error) {
|
||||||
|
logger.error(`❌ Failed to get valid access token for account ${accountId}:`, error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📋 获取所有Claude账户
|
||||||
|
async getAllAccounts() {
|
||||||
|
try {
|
||||||
|
const accounts = await redis.getAllClaudeAccounts();
|
||||||
|
|
||||||
|
// 处理返回数据，移除敏感信息
|
||||||
|
return accounts.map(account => ({
|
||||||
|
id: account.id,
|
||||||
|
name: account.name,
|
||||||
|
description: account.description,
|
||||||
|
email: account.email ? this._maskEmail(this._decryptSensitiveData(account.email)) : '',
|
||||||
|
isActive: account.isActive === 'true',
|
||||||
|
proxy: account.proxy ? JSON.parse(account.proxy) : null,
|
||||||
|
status: account.status,
|
||||||
|
errorMessage: account.errorMessage,
|
||||||
|
createdAt: account.createdAt,
|
||||||
|
lastUsedAt: account.lastUsedAt,
|
||||||
|
lastRefreshAt: account.lastRefreshAt,
|
||||||
|
expiresAt: account.expiresAt
|
||||||
|
}));
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to get Claude accounts:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 📝 更新Claude账户
|
||||||
|
async updateAccount(accountId, updates) {
|
||||||
|
try {
|
||||||
|
const accountData = await redis.getClaudeAccount(accountId);
|
||||||
|
|
||||||
|
if (!accountData || Object.keys(accountData).length === 0) {
|
||||||
|
throw new Error('Account not found');
|
||||||
|
}
|
||||||
|
|
||||||
|
const allowedUpdates = ['name', 'description', 'email', 'password', 'refreshToken', 'proxy', 'isActive'];
|
||||||
|
const updatedData = { ...accountData };
|
||||||
|
|
||||||
|
for (const [field, value] of Object.entries(updates)) {
|
||||||
|
if (allowedUpdates.includes(field)) {
|
||||||
|
if (['email', 'password', 'refreshToken'].includes(field)) {
|
||||||
|
updatedData[field] = this._encryptSensitiveData(value);
|
||||||
|
} else if (field === 'proxy') {
|
||||||
|
updatedData[field] = value ? JSON.stringify(value) : '';
|
||||||
|
} else {
|
||||||
|
updatedData[field] = value.toString();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
updatedData.updatedAt = new Date().toISOString();
|
||||||
|
|
||||||
|
await redis.setClaudeAccount(accountId, updatedData);
|
||||||
|
|
||||||
|
logger.success(`📝 Updated Claude account: ${accountId}`);
|
||||||
|
|
||||||
|
return { success: true };
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to update Claude account:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🗑️ 删除Claude账户
|
||||||
|
async deleteAccount(accountId) {
|
||||||
|
try {
|
||||||
|
const result = await redis.deleteClaudeAccount(accountId);
|
||||||
|
|
||||||
|
if (result === 0) {
|
||||||
|
throw new Error('Account not found');
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.success(`🗑️ Deleted Claude account: ${accountId}`);
|
||||||
|
|
||||||
|
return { success: true };
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to delete Claude account:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🎯 智能选择可用账户
|
||||||
|
async selectAvailableAccount() {
|
||||||
|
try {
|
||||||
|
const accounts = await redis.getAllClaudeAccounts();
|
||||||
|
|
||||||
|
const activeAccounts = accounts.filter(account =>
|
||||||
|
account.isActive === 'true' &&
|
||||||
|
account.status !== 'error'
|
||||||
|
);
|
||||||
|
|
||||||
|
if (activeAccounts.length === 0) {
|
||||||
|
throw new Error('No active Claude accounts available');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 优先选择最近刷新过token的账户
|
||||||
|
const sortedAccounts = activeAccounts.sort((a, b) => {
|
||||||
|
const aLastRefresh = new Date(a.lastRefreshAt || 0).getTime();
|
||||||
|
const bLastRefresh = new Date(b.lastRefreshAt || 0).getTime();
|
||||||
|
return bLastRefresh - aLastRefresh;
|
||||||
|
});
|
||||||
|
|
||||||
|
return sortedAccounts[0].id;
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to select available account:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🌐 创建代理agent
|
||||||
|
_createProxyAgent(proxyConfig) {
|
||||||
|
if (!proxyConfig) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const proxy = JSON.parse(proxyConfig);
|
||||||
|
|
||||||
|
if (proxy.type === 'socks5') {
|
||||||
|
const auth = proxy.username && proxy.password ? `${proxy.username}:${proxy.password}@` : '';
|
||||||
|
const socksUrl = `socks5://${auth}${proxy.host}:${proxy.port}`;
|
||||||
|
return new SocksProxyAgent(socksUrl);
|
||||||
|
} else if (proxy.type === 'http' || proxy.type === 'https') {
|
||||||
|
const auth = proxy.username && proxy.password ? `${proxy.username}:${proxy.password}@` : '';
|
||||||
|
const httpUrl = `${proxy.type}://${auth}${proxy.host}:${proxy.port}`;
|
||||||
|
return new HttpsProxyAgent(httpUrl);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.warn('⚠️ Invalid proxy configuration:', error);
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔐 加密敏感数据
|
||||||
|
_encryptSensitiveData(data) {
|
||||||
|
if (!data) return '';
|
||||||
|
|
||||||
|
try {
|
||||||
|
const key = this._generateEncryptionKey();
|
||||||
|
const iv = crypto.randomBytes(16);
|
||||||
|
|
||||||
|
const cipher = crypto.createCipheriv(this.ENCRYPTION_ALGORITHM, key, iv);
|
||||||
|
let encrypted = cipher.update(data, 'utf8', 'hex');
|
||||||
|
encrypted += cipher.final('hex');
|
||||||
|
|
||||||
|
// 将IV和加密数据一起返回，用:分隔
|
||||||
|
return iv.toString('hex') + ':' + encrypted;
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Encryption error:', error);
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔓 解密敏感数据
|
||||||
|
_decryptSensitiveData(encryptedData) {
|
||||||
|
if (!encryptedData) return '';
|
||||||
|
|
||||||
|
try {
|
||||||
|
// 检查是否是新格式（包含IV）
|
||||||
|
if (encryptedData.includes(':')) {
|
||||||
|
// 新格式：iv:encryptedData
|
||||||
|
const parts = encryptedData.split(':');
|
||||||
|
if (parts.length === 2) {
|
||||||
|
const key = this._generateEncryptionKey();
|
||||||
|
const iv = Buffer.from(parts[0], 'hex');
|
||||||
|
const encrypted = parts[1];
|
||||||
|
|
||||||
|
const decipher = crypto.createDecipheriv(this.ENCRYPTION_ALGORITHM, key, iv);
|
||||||
|
let decrypted = decipher.update(encrypted, 'hex', 'utf8');
|
||||||
|
decrypted += decipher.final('utf8');
|
||||||
|
return decrypted;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 旧格式或格式错误，尝试旧方式解密（向后兼容）
|
||||||
|
// 注意：在新版本Node.js中这将失败，但我们会捕获错误
|
||||||
|
try {
|
||||||
|
const decipher = crypto.createDecipher('aes-256-cbc', config.security.encryptionKey);
|
||||||
|
let decrypted = decipher.update(encryptedData, 'hex', 'utf8');
|
||||||
|
decrypted += decipher.final('utf8');
|
||||||
|
return decrypted;
|
||||||
|
} catch (oldError) {
|
||||||
|
// 如果旧方式也失败，返回原数据
|
||||||
|
logger.warn('⚠️ Could not decrypt data, returning as-is:', oldError.message);
|
||||||
|
return encryptedData;
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Decryption error:', error);
|
||||||
|
return encryptedData;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔑 生成加密密钥（辅助方法）
|
||||||
|
_generateEncryptionKey() {
|
||||||
|
return crypto.scryptSync(config.security.encryptionKey, this.ENCRYPTION_SALT, 32);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🎭 掩码邮箱地址
|
||||||
|
_maskEmail(email) {
|
||||||
|
if (!email || !email.includes('@')) return email;
|
||||||
|
|
||||||
|
const [username, domain] = email.split('@');
|
||||||
|
const maskedUsername = username.length > 2
|
||||||
|
? `${username.slice(0, 2)}***${username.slice(-1)}`
|
||||||
|
: `${username.slice(0, 1)}***`;
|
||||||
|
|
||||||
|
return `${maskedUsername}@${domain}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🧹 清理错误账户
|
||||||
|
async cleanupErrorAccounts() {
|
||||||
|
try {
|
||||||
|
const accounts = await redis.getAllClaudeAccounts();
|
||||||
|
let cleanedCount = 0;
|
||||||
|
|
||||||
|
for (const account of accounts) {
|
||||||
|
if (account.status === 'error' && account.lastRefreshAt) {
|
||||||
|
const lastRefresh = new Date(account.lastRefreshAt);
|
||||||
|
const now = new Date();
|
||||||
|
const hoursSinceLastRefresh = (now - lastRefresh) / (1000 * 60 * 60);
|
||||||
|
|
||||||
|
// 如果错误状态超过24小时，尝试重新激活
|
||||||
|
if (hoursSinceLastRefresh > 24) {
|
||||||
|
account.status = 'created';
|
||||||
|
account.errorMessage = '';
|
||||||
|
await redis.setClaudeAccount(account.id, account);
|
||||||
|
cleanedCount++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (cleanedCount > 0) {
|
||||||
|
logger.success(`🧹 Reset ${cleanedCount} error accounts`);
|
||||||
|
}
|
||||||
|
|
||||||
|
return cleanedCount;
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to cleanup error accounts:', error);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = new ClaudeAccountService();
|
||||||
526
src/services/claudeRelayService.js
Normal file
526
src/services/claudeRelayService.js
Normal file
@@ -0,0 +1,526 @@
|
|||||||
|
const https = require('https');
|
||||||
|
const { SocksProxyAgent } = require('socks-proxy-agent');
|
||||||
|
const { HttpsProxyAgent } = require('https-proxy-agent');
|
||||||
|
const claudeAccountService = require('./claudeAccountService');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
const config = require('../../config/config');
|
||||||
|
|
||||||
|
class ClaudeRelayService {
|
||||||
|
constructor() {
|
||||||
|
this.claudeApiUrl = config.claude.apiUrl;
|
||||||
|
this.apiVersion = config.claude.apiVersion;
|
||||||
|
this.betaHeader = config.claude.betaHeader;
|
||||||
|
this.systemPrompt = config.claude.systemPrompt;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🚀 转发请求到Claude API
|
||||||
|
async relayRequest(requestBody, apiKeyData) {
|
||||||
|
try {
|
||||||
|
// 选择可用的Claude账户
|
||||||
|
const accountId = apiKeyData.claudeAccountId || await claudeAccountService.selectAvailableAccount();
|
||||||
|
|
||||||
|
logger.info(`📤 Processing API request for key: ${apiKeyData.name || apiKeyData.id}, account: ${accountId}`);
|
||||||
|
|
||||||
|
// 获取有效的访问token
|
||||||
|
const accessToken = await claudeAccountService.getValidAccessToken(accountId);
|
||||||
|
|
||||||
|
// 处理请求体
|
||||||
|
const processedBody = this._processRequestBody(requestBody);
|
||||||
|
|
||||||
|
// 获取代理配置
|
||||||
|
const proxyAgent = await this._getProxyAgent(accountId);
|
||||||
|
|
||||||
|
// 发送请求到Claude API
|
||||||
|
const response = await this._makeClaudeRequest(processedBody, accessToken, proxyAgent);
|
||||||
|
|
||||||
|
// 记录成功的API调用
|
||||||
|
const inputTokens = requestBody.messages ?
|
||||||
|
requestBody.messages.reduce((sum, msg) => sum + (msg.content?.length || 0), 0) / 4 : 0; // 粗略估算
|
||||||
|
const outputTokens = response.content ?
|
||||||
|
response.content.reduce((sum, content) => sum + (content.text?.length || 0), 0) / 4 : 0;
|
||||||
|
|
||||||
|
logger.info(`✅ API request completed - Key: ${apiKeyData.name}, Account: ${accountId}, Model: ${requestBody.model}, Input: ~${Math.round(inputTokens)} tokens, Output: ~${Math.round(outputTokens)} tokens`);
|
||||||
|
|
||||||
|
return response;
|
||||||
|
} catch (error) {
|
||||||
|
logger.error(`❌ Claude relay request failed for key: ${apiKeyData.name || apiKeyData.id}:`, error.message);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔄 处理请求体
|
||||||
|
_processRequestBody(body) {
|
||||||
|
if (!body) return body;
|
||||||
|
|
||||||
|
// 深拷贝请求体
|
||||||
|
const processedBody = JSON.parse(JSON.stringify(body));
|
||||||
|
|
||||||
|
// 移除cache_control中的ttl字段
|
||||||
|
this._stripTtlFromCacheControl(processedBody);
|
||||||
|
|
||||||
|
// 只有在配置了系统提示时才添加
|
||||||
|
if (this.systemPrompt && this.systemPrompt.trim()) {
|
||||||
|
const systemPrompt = {
|
||||||
|
type: 'text',
|
||||||
|
text: this.systemPrompt
|
||||||
|
};
|
||||||
|
|
||||||
|
if (processedBody.system) {
|
||||||
|
if (Array.isArray(processedBody.system)) {
|
||||||
|
// 如果system数组存在但为空，或者没有有效内容，则添加系统提示
|
||||||
|
const hasValidContent = processedBody.system.some(item =>
|
||||||
|
item && item.text && item.text.trim()
|
||||||
|
);
|
||||||
|
if (!hasValidContent) {
|
||||||
|
processedBody.system = [systemPrompt];
|
||||||
|
} else {
|
||||||
|
processedBody.system.unshift(systemPrompt);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
throw new Error('system field must be an array');
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
processedBody.system = [systemPrompt];
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 如果没有配置系统提示，且system字段为空，则删除它
|
||||||
|
if (processedBody.system && Array.isArray(processedBody.system)) {
|
||||||
|
const hasValidContent = processedBody.system.some(item =>
|
||||||
|
item && item.text && item.text.trim()
|
||||||
|
);
|
||||||
|
if (!hasValidContent) {
|
||||||
|
delete processedBody.system;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return processedBody;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🧹 移除TTL字段
|
||||||
|
_stripTtlFromCacheControl(body) {
|
||||||
|
if (!body || typeof body !== 'object') return;
|
||||||
|
|
||||||
|
const processContentArray = (contentArray) => {
|
||||||
|
if (!Array.isArray(contentArray)) return;
|
||||||
|
|
||||||
|
contentArray.forEach(item => {
|
||||||
|
if (item && typeof item === 'object' && item.cache_control) {
|
||||||
|
if (item.cache_control.ttl) {
|
||||||
|
delete item.cache_control.ttl;
|
||||||
|
logger.debug('🧹 Removed ttl from cache_control');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
if (Array.isArray(body.system)) {
|
||||||
|
processContentArray(body.system);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (Array.isArray(body.messages)) {
|
||||||
|
body.messages.forEach(message => {
|
||||||
|
if (message && Array.isArray(message.content)) {
|
||||||
|
processContentArray(message.content);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🌐 获取代理Agent
|
||||||
|
async _getProxyAgent(accountId) {
|
||||||
|
try {
|
||||||
|
const accountData = await claudeAccountService.getAllAccounts();
|
||||||
|
const account = accountData.find(acc => acc.id === accountId);
|
||||||
|
|
||||||
|
if (!account || !account.proxy) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
const proxy = account.proxy;
|
||||||
|
|
||||||
|
if (proxy.type === 'socks5') {
|
||||||
|
const auth = proxy.username && proxy.password ? `${proxy.username}:${proxy.password}@` : '';
|
||||||
|
const socksUrl = `socks5://${auth}${proxy.host}:${proxy.port}`;
|
||||||
|
return new SocksProxyAgent(socksUrl);
|
||||||
|
} else if (proxy.type === 'http' || proxy.type === 'https') {
|
||||||
|
const auth = proxy.username && proxy.password ? `${proxy.username}:${proxy.password}@` : '';
|
||||||
|
const httpUrl = `${proxy.type}://${auth}${proxy.host}:${proxy.port}`;
|
||||||
|
return new HttpsProxyAgent(httpUrl);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.warn('⚠️ Failed to create proxy agent:', error);
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔗 发送请求到Claude API
|
||||||
|
async _makeClaudeRequest(body, accessToken, proxyAgent) {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const url = new URL(this.claudeApiUrl);
|
||||||
|
|
||||||
|
const options = {
|
||||||
|
hostname: url.hostname,
|
||||||
|
port: url.port || 443,
|
||||||
|
path: url.pathname,
|
||||||
|
method: 'POST',
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
'Authorization': `Bearer ${accessToken}`,
|
||||||
|
'anthropic-version': this.apiVersion,
|
||||||
|
'User-Agent': 'claude-relay-service/1.0.0'
|
||||||
|
},
|
||||||
|
agent: proxyAgent,
|
||||||
|
timeout: config.proxy.timeout
|
||||||
|
};
|
||||||
|
|
||||||
|
if (this.betaHeader) {
|
||||||
|
options.headers['anthropic-beta'] = this.betaHeader;
|
||||||
|
}
|
||||||
|
|
||||||
|
const req = https.request(options, (res) => {
|
||||||
|
let responseData = '';
|
||||||
|
|
||||||
|
res.on('data', (chunk) => {
|
||||||
|
responseData += chunk;
|
||||||
|
});
|
||||||
|
|
||||||
|
res.on('end', () => {
|
||||||
|
try {
|
||||||
|
const response = {
|
||||||
|
statusCode: res.statusCode,
|
||||||
|
headers: res.headers,
|
||||||
|
body: responseData
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.debug(`🔗 Claude API response: ${res.statusCode}`);
|
||||||
|
|
||||||
|
resolve(response);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to parse Claude API response:', error);
|
||||||
|
reject(error);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
req.on('error', (error) => {
|
||||||
|
logger.error('❌ Claude API request error:', error);
|
||||||
|
reject(error);
|
||||||
|
});
|
||||||
|
|
||||||
|
req.on('timeout', () => {
|
||||||
|
req.destroy();
|
||||||
|
logger.error('❌ Claude API request timeout');
|
||||||
|
reject(new Error('Request timeout'));
|
||||||
|
});
|
||||||
|
|
||||||
|
// 写入请求体
|
||||||
|
req.write(JSON.stringify(body));
|
||||||
|
req.end();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🌊 处理流式响应（带usage数据捕获）
|
||||||
|
async relayStreamRequestWithUsageCapture(requestBody, apiKeyData, responseStream, usageCallback) {
|
||||||
|
try {
|
||||||
|
// 选择可用的Claude账户
|
||||||
|
const accountId = apiKeyData.claudeAccountId || await claudeAccountService.selectAvailableAccount();
|
||||||
|
|
||||||
|
logger.info(`📡 Processing streaming API request with usage capture for key: ${apiKeyData.name || apiKeyData.id}, account: ${accountId}`);
|
||||||
|
|
||||||
|
// 获取有效的访问token
|
||||||
|
const accessToken = await claudeAccountService.getValidAccessToken(accountId);
|
||||||
|
|
||||||
|
// 处理请求体
|
||||||
|
const processedBody = this._processRequestBody(requestBody);
|
||||||
|
|
||||||
|
// 获取代理配置
|
||||||
|
const proxyAgent = await this._getProxyAgent(accountId);
|
||||||
|
|
||||||
|
// 发送流式请求并捕获usage数据
|
||||||
|
return await this._makeClaudeStreamRequestWithUsageCapture(processedBody, accessToken, proxyAgent, responseStream, usageCallback);
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Claude stream relay with usage capture failed:', error);
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🌊 发送流式请求到Claude API（带usage数据捕获）
|
||||||
|
async _makeClaudeStreamRequestWithUsageCapture(body, accessToken, proxyAgent, responseStream, usageCallback) {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const url = new URL(this.claudeApiUrl);
|
||||||
|
|
||||||
|
const options = {
|
||||||
|
hostname: url.hostname,
|
||||||
|
port: url.port || 443,
|
||||||
|
path: url.pathname,
|
||||||
|
method: 'POST',
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
'Authorization': `Bearer ${accessToken}`,
|
||||||
|
'anthropic-version': this.apiVersion,
|
||||||
|
'User-Agent': 'claude-relay-service/1.0.0'
|
||||||
|
},
|
||||||
|
agent: proxyAgent,
|
||||||
|
timeout: config.proxy.timeout
|
||||||
|
};
|
||||||
|
|
||||||
|
if (this.betaHeader) {
|
||||||
|
options.headers['anthropic-beta'] = this.betaHeader;
|
||||||
|
}
|
||||||
|
|
||||||
|
const req = https.request(options, (res) => {
|
||||||
|
// 设置响应头
|
||||||
|
responseStream.statusCode = res.statusCode;
|
||||||
|
Object.keys(res.headers).forEach(key => {
|
||||||
|
responseStream.setHeader(key, res.headers[key]);
|
||||||
|
});
|
||||||
|
|
||||||
|
let buffer = '';
|
||||||
|
let finalUsageReported = false; // 防止重复统计的标志
|
||||||
|
let collectedUsageData = {}; // 收集来自不同事件的usage数据
|
||||||
|
|
||||||
|
// 监听数据块，解析SSE并寻找usage信息
|
||||||
|
res.on('data', (chunk) => {
|
||||||
|
const chunkStr = chunk.toString();
|
||||||
|
|
||||||
|
// 记录原始SSE数据块
|
||||||
|
logger.info('📡 Raw SSE chunk received:', {
|
||||||
|
length: chunkStr.length,
|
||||||
|
content: chunkStr
|
||||||
|
});
|
||||||
|
|
||||||
|
buffer += chunkStr;
|
||||||
|
|
||||||
|
// 处理完整的SSE行
|
||||||
|
const lines = buffer.split('\n');
|
||||||
|
buffer = lines.pop() || ''; // 保留最后的不完整行
|
||||||
|
|
||||||
|
// 转发已处理的完整行到客户端
|
||||||
|
if (lines.length > 0) {
|
||||||
|
const linesToForward = lines.join('\n') + (lines.length > 0 ? '\n' : '');
|
||||||
|
responseStream.write(linesToForward);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const line of lines) {
|
||||||
|
// 记录每个SSE行
|
||||||
|
if (line.trim()) {
|
||||||
|
logger.info('📄 SSE Line:', line);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 解析SSE数据寻找usage信息
|
||||||
|
if (line.startsWith('data: ') && line.length > 6) {
|
||||||
|
try {
|
||||||
|
const jsonStr = line.slice(6);
|
||||||
|
const data = JSON.parse(jsonStr);
|
||||||
|
|
||||||
|
// 收集来自不同事件的usage数据
|
||||||
|
if (data.type === 'message_start' && data.message && data.message.usage) {
|
||||||
|
// message_start包含input tokens、cache tokens和模型信息
|
||||||
|
collectedUsageData.input_tokens = data.message.usage.input_tokens || 0;
|
||||||
|
collectedUsageData.cache_creation_input_tokens = data.message.usage.cache_creation_input_tokens || 0;
|
||||||
|
collectedUsageData.cache_read_input_tokens = data.message.usage.cache_read_input_tokens || 0;
|
||||||
|
collectedUsageData.model = data.message.model;
|
||||||
|
|
||||||
|
logger.info('📊 Collected input/cache data from message_start:', JSON.stringify(collectedUsageData));
|
||||||
|
}
|
||||||
|
|
||||||
|
// message_delta包含最终的output tokens
|
||||||
|
if (data.type === 'message_delta' && data.usage && data.usage.output_tokens !== undefined) {
|
||||||
|
collectedUsageData.output_tokens = data.usage.output_tokens || 0;
|
||||||
|
|
||||||
|
logger.info('📊 Collected output data from message_delta:', JSON.stringify(collectedUsageData));
|
||||||
|
|
||||||
|
// 如果已经收集到了input数据，现在有了output数据，可以统计了
|
||||||
|
if (collectedUsageData.input_tokens !== undefined && !finalUsageReported) {
|
||||||
|
logger.info('🎯 Complete usage data collected, triggering callback');
|
||||||
|
usageCallback(collectedUsageData);
|
||||||
|
finalUsageReported = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} catch (parseError) {
|
||||||
|
// 忽略JSON解析错误，继续处理
|
||||||
|
logger.debug('🔍 SSE line not JSON or no usage data:', line.slice(0, 100));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
res.on('end', () => {
|
||||||
|
// 处理缓冲区中剩余的数据
|
||||||
|
if (buffer.trim()) {
|
||||||
|
responseStream.write(buffer);
|
||||||
|
}
|
||||||
|
responseStream.end();
|
||||||
|
|
||||||
|
// 检查是否捕获到usage数据
|
||||||
|
if (!finalUsageReported) {
|
||||||
|
logger.warn('⚠️ Stream completed but no usage data was captured! This indicates a problem with SSE parsing or Claude API response format.');
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.debug('🌊 Claude stream response with usage capture completed');
|
||||||
|
resolve();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
req.on('error', (error) => {
|
||||||
|
logger.error('❌ Claude stream request error:', error);
|
||||||
|
if (!responseStream.headersSent) {
|
||||||
|
responseStream.writeHead(500, { 'Content-Type': 'application/json' });
|
||||||
|
}
|
||||||
|
if (!responseStream.destroyed) {
|
||||||
|
responseStream.end(JSON.stringify({ error: 'Upstream request failed' }));
|
||||||
|
}
|
||||||
|
reject(error);
|
||||||
|
});
|
||||||
|
|
||||||
|
req.on('timeout', () => {
|
||||||
|
req.destroy();
|
||||||
|
logger.error('❌ Claude stream request timeout');
|
||||||
|
if (!responseStream.headersSent) {
|
||||||
|
responseStream.writeHead(504, { 'Content-Type': 'application/json' });
|
||||||
|
}
|
||||||
|
if (!responseStream.destroyed) {
|
||||||
|
responseStream.end(JSON.stringify({ error: 'Request timeout' }));
|
||||||
|
}
|
||||||
|
reject(new Error('Request timeout'));
|
||||||
|
});
|
||||||
|
|
||||||
|
// 处理客户端断开连接
|
||||||
|
responseStream.on('close', () => {
|
||||||
|
logger.debug('🔌 Client disconnected, cleaning up stream');
|
||||||
|
if (!req.destroyed) {
|
||||||
|
req.destroy();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 写入请求体
|
||||||
|
req.write(JSON.stringify(body));
|
||||||
|
req.end();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🌊 发送流式请求到Claude API
|
||||||
|
async _makeClaudeStreamRequest(body, accessToken, proxyAgent, responseStream) {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const url = new URL(this.claudeApiUrl);
|
||||||
|
|
||||||
|
const options = {
|
||||||
|
hostname: url.hostname,
|
||||||
|
port: url.port || 443,
|
||||||
|
path: url.pathname,
|
||||||
|
method: 'POST',
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
'Authorization': `Bearer ${accessToken}`,
|
||||||
|
'anthropic-version': this.apiVersion,
|
||||||
|
'User-Agent': 'claude-relay-service/1.0.0'
|
||||||
|
},
|
||||||
|
agent: proxyAgent,
|
||||||
|
timeout: config.proxy.timeout
|
||||||
|
};
|
||||||
|
|
||||||
|
if (this.betaHeader) {
|
||||||
|
options.headers['anthropic-beta'] = this.betaHeader;
|
||||||
|
}
|
||||||
|
|
||||||
|
const req = https.request(options, (res) => {
|
||||||
|
// 设置响应头
|
||||||
|
responseStream.statusCode = res.statusCode;
|
||||||
|
Object.keys(res.headers).forEach(key => {
|
||||||
|
responseStream.setHeader(key, res.headers[key]);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 管道响应数据
|
||||||
|
res.pipe(responseStream);
|
||||||
|
|
||||||
|
res.on('end', () => {
|
||||||
|
logger.debug('🌊 Claude stream response completed');
|
||||||
|
resolve();
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
req.on('error', (error) => {
|
||||||
|
logger.error('❌ Claude stream request error:', error);
|
||||||
|
if (!responseStream.headersSent) {
|
||||||
|
responseStream.writeHead(500, { 'Content-Type': 'application/json' });
|
||||||
|
}
|
||||||
|
if (!responseStream.destroyed) {
|
||||||
|
responseStream.end(JSON.stringify({ error: 'Upstream request failed' }));
|
||||||
|
}
|
||||||
|
reject(error);
|
||||||
|
});
|
||||||
|
|
||||||
|
req.on('timeout', () => {
|
||||||
|
req.destroy();
|
||||||
|
logger.error('❌ Claude stream request timeout');
|
||||||
|
if (!responseStream.headersSent) {
|
||||||
|
responseStream.writeHead(504, { 'Content-Type': 'application/json' });
|
||||||
|
}
|
||||||
|
if (!responseStream.destroyed) {
|
||||||
|
responseStream.end(JSON.stringify({ error: 'Request timeout' }));
|
||||||
|
}
|
||||||
|
reject(new Error('Request timeout'));
|
||||||
|
});
|
||||||
|
|
||||||
|
// 处理客户端断开连接
|
||||||
|
responseStream.on('close', () => {
|
||||||
|
logger.debug('🔌 Client disconnected, cleaning up stream');
|
||||||
|
if (!req.destroyed) {
|
||||||
|
req.destroy();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 写入请求体
|
||||||
|
req.write(JSON.stringify(body));
|
||||||
|
req.end();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔄 重试逻辑
|
||||||
|
async _retryRequest(requestFunc, maxRetries = 3) {
|
||||||
|
let lastError;
|
||||||
|
|
||||||
|
for (let i = 0; i < maxRetries; i++) {
|
||||||
|
try {
|
||||||
|
return await requestFunc();
|
||||||
|
} catch (error) {
|
||||||
|
lastError = error;
|
||||||
|
|
||||||
|
if (i < maxRetries - 1) {
|
||||||
|
const delay = Math.pow(2, i) * 1000; // 指数退避
|
||||||
|
logger.warn(`⏳ Retry ${i + 1}/${maxRetries} in ${delay}ms: ${error.message}`);
|
||||||
|
await new Promise(resolve => setTimeout(resolve, delay));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
throw lastError;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🎯 健康检查
|
||||||
|
async healthCheck() {
|
||||||
|
try {
|
||||||
|
const accounts = await claudeAccountService.getAllAccounts();
|
||||||
|
const activeAccounts = accounts.filter(acc => acc.isActive && acc.status === 'active');
|
||||||
|
|
||||||
|
return {
|
||||||
|
healthy: activeAccounts.length > 0,
|
||||||
|
activeAccounts: activeAccounts.length,
|
||||||
|
totalAccounts: accounts.length,
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Health check failed:', error);
|
||||||
|
return {
|
||||||
|
healthy: false,
|
||||||
|
error: error.message,
|
||||||
|
timestamp: new Date().toISOString()
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = new ClaudeRelayService();
|
||||||
234
src/services/pricingService.js
Normal file
234
src/services/pricingService.js
Normal file
@@ -0,0 +1,234 @@
|
|||||||
|
const fs = require('fs');
|
||||||
|
const path = require('path');
|
||||||
|
const https = require('https');
|
||||||
|
const logger = require('../utils/logger');
|
||||||
|
|
||||||
|
class PricingService {
|
||||||
|
constructor() {
|
||||||
|
this.dataDir = path.join(process.cwd(), 'data');
|
||||||
|
this.pricingFile = path.join(this.dataDir, 'model_pricing.json');
|
||||||
|
this.pricingUrl = 'https://raw.githubusercontent.com/BerriAI/litellm/main/model_prices_and_context_window.json';
|
||||||
|
this.pricingData = null;
|
||||||
|
this.lastUpdated = null;
|
||||||
|
this.updateInterval = 24 * 60 * 60 * 1000; // 24小时
|
||||||
|
}
|
||||||
|
|
||||||
|
// 初始化价格服务
|
||||||
|
async initialize() {
|
||||||
|
try {
|
||||||
|
// 确保data目录存在
|
||||||
|
if (!fs.existsSync(this.dataDir)) {
|
||||||
|
fs.mkdirSync(this.dataDir, { recursive: true });
|
||||||
|
logger.info('📁 Created data directory');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否需要下载或更新价格数据
|
||||||
|
await this.checkAndUpdatePricing();
|
||||||
|
|
||||||
|
// 设置定时更新
|
||||||
|
setInterval(() => {
|
||||||
|
this.checkAndUpdatePricing();
|
||||||
|
}, this.updateInterval);
|
||||||
|
|
||||||
|
logger.success('💰 Pricing service initialized successfully');
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to initialize pricing service:', error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查并更新价格数据
|
||||||
|
async checkAndUpdatePricing() {
|
||||||
|
try {
|
||||||
|
const needsUpdate = this.needsUpdate();
|
||||||
|
|
||||||
|
if (needsUpdate) {
|
||||||
|
logger.info('🔄 Updating model pricing data...');
|
||||||
|
await this.downloadPricingData();
|
||||||
|
} else {
|
||||||
|
// 如果不需要更新，加载现有数据
|
||||||
|
await this.loadPricingData();
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to check/update pricing:', error);
|
||||||
|
// 如果更新失败，尝试加载现有数据
|
||||||
|
await this.loadPricingData();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 检查是否需要更新
|
||||||
|
needsUpdate() {
|
||||||
|
if (!fs.existsSync(this.pricingFile)) {
|
||||||
|
logger.info('📋 Pricing file not found, will download');
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
const stats = fs.statSync(this.pricingFile);
|
||||||
|
const fileAge = Date.now() - stats.mtime.getTime();
|
||||||
|
|
||||||
|
if (fileAge > this.updateInterval) {
|
||||||
|
logger.info(`📋 Pricing file is ${Math.round(fileAge / (60 * 60 * 1000))} hours old, will update`);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 下载价格数据
|
||||||
|
downloadPricingData() {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const request = https.get(this.pricingUrl, (response) => {
|
||||||
|
if (response.statusCode !== 200) {
|
||||||
|
reject(new Error(`HTTP ${response.statusCode}: ${response.statusMessage}`));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let data = '';
|
||||||
|
response.on('data', (chunk) => {
|
||||||
|
data += chunk;
|
||||||
|
});
|
||||||
|
|
||||||
|
response.on('end', () => {
|
||||||
|
try {
|
||||||
|
const jsonData = JSON.parse(data);
|
||||||
|
|
||||||
|
// 保存到文件
|
||||||
|
fs.writeFileSync(this.pricingFile, JSON.stringify(jsonData, null, 2));
|
||||||
|
|
||||||
|
// 更新内存中的数据
|
||||||
|
this.pricingData = jsonData;
|
||||||
|
this.lastUpdated = new Date();
|
||||||
|
|
||||||
|
logger.success(`💰 Downloaded pricing data for ${Object.keys(jsonData).length} models`);
|
||||||
|
resolve();
|
||||||
|
} catch (error) {
|
||||||
|
reject(new Error(`Failed to parse pricing data: ${error.message}`));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
request.on('error', (error) => {
|
||||||
|
reject(new Error(`Failed to download pricing data: ${error.message}`));
|
||||||
|
});
|
||||||
|
|
||||||
|
request.setTimeout(30000, () => {
|
||||||
|
request.destroy();
|
||||||
|
reject(new Error('Download timeout'));
|
||||||
|
});
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// 加载本地价格数据
|
||||||
|
async loadPricingData() {
|
||||||
|
try {
|
||||||
|
if (fs.existsSync(this.pricingFile)) {
|
||||||
|
const data = fs.readFileSync(this.pricingFile, 'utf8');
|
||||||
|
this.pricingData = JSON.parse(data);
|
||||||
|
|
||||||
|
const stats = fs.statSync(this.pricingFile);
|
||||||
|
this.lastUpdated = stats.mtime;
|
||||||
|
|
||||||
|
logger.info(`💰 Loaded pricing data for ${Object.keys(this.pricingData).length} models from cache`);
|
||||||
|
} else {
|
||||||
|
logger.warn('💰 No pricing data file found');
|
||||||
|
this.pricingData = {};
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Failed to load pricing data:', error);
|
||||||
|
this.pricingData = {};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取模型价格信息
|
||||||
|
getModelPricing(modelName) {
|
||||||
|
if (!this.pricingData || !modelName) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 尝试直接匹配
|
||||||
|
if (this.pricingData[modelName]) {
|
||||||
|
return this.pricingData[modelName];
|
||||||
|
}
|
||||||
|
|
||||||
|
// 尝试模糊匹配（处理版本号等变化）
|
||||||
|
const normalizedModel = modelName.toLowerCase().replace(/[_-]/g, '');
|
||||||
|
|
||||||
|
for (const [key, value] of Object.entries(this.pricingData)) {
|
||||||
|
const normalizedKey = key.toLowerCase().replace(/[_-]/g, '');
|
||||||
|
if (normalizedKey.includes(normalizedModel) || normalizedModel.includes(normalizedKey)) {
|
||||||
|
logger.debug(`💰 Found pricing for ${modelName} using fuzzy match: ${key}`);
|
||||||
|
return value;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
logger.debug(`💰 No pricing found for model: ${modelName}`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 计算使用费用
|
||||||
|
calculateCost(usage, modelName) {
|
||||||
|
const pricing = this.getModelPricing(modelName);
|
||||||
|
|
||||||
|
if (!pricing) {
|
||||||
|
return {
|
||||||
|
inputCost: 0,
|
||||||
|
outputCost: 0,
|
||||||
|
cacheCreateCost: 0,
|
||||||
|
cacheReadCost: 0,
|
||||||
|
totalCost: 0,
|
||||||
|
hasPricing: false
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const inputCost = (usage.input_tokens || 0) * (pricing.input_cost_per_token || 0);
|
||||||
|
const outputCost = (usage.output_tokens || 0) * (pricing.output_cost_per_token || 0);
|
||||||
|
const cacheCreateCost = (usage.cache_creation_input_tokens || 0) * (pricing.cache_creation_input_token_cost || 0);
|
||||||
|
const cacheReadCost = (usage.cache_read_input_tokens || 0) * (pricing.cache_read_input_token_cost || 0);
|
||||||
|
|
||||||
|
return {
|
||||||
|
inputCost,
|
||||||
|
outputCost,
|
||||||
|
cacheCreateCost,
|
||||||
|
cacheReadCost,
|
||||||
|
totalCost: inputCost + outputCost + cacheCreateCost + cacheReadCost,
|
||||||
|
hasPricing: true,
|
||||||
|
pricing: {
|
||||||
|
input: pricing.input_cost_per_token || 0,
|
||||||
|
output: pricing.output_cost_per_token || 0,
|
||||||
|
cacheCreate: pricing.cache_creation_input_token_cost || 0,
|
||||||
|
cacheRead: pricing.cache_read_input_token_cost || 0
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// 格式化价格显示
|
||||||
|
formatCost(cost) {
|
||||||
|
if (cost === 0) return '$0.000000';
|
||||||
|
if (cost < 0.000001) return `$${cost.toExponential(2)}`;
|
||||||
|
if (cost < 0.01) return `$${cost.toFixed(6)}`;
|
||||||
|
if (cost < 1) return `$${cost.toFixed(4)}`;
|
||||||
|
return `$${cost.toFixed(2)}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取服务状态
|
||||||
|
getStatus() {
|
||||||
|
return {
|
||||||
|
initialized: this.pricingData !== null,
|
||||||
|
lastUpdated: this.lastUpdated,
|
||||||
|
modelCount: this.pricingData ? Object.keys(this.pricingData).length : 0,
|
||||||
|
nextUpdate: this.lastUpdated ? new Date(this.lastUpdated.getTime() + this.updateInterval) : null
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
// 强制更新价格数据
|
||||||
|
async forceUpdate() {
|
||||||
|
try {
|
||||||
|
await this.downloadPricingData();
|
||||||
|
return { success: true, message: 'Pricing data updated successfully' };
|
||||||
|
} catch (error) {
|
||||||
|
logger.error('❌ Force update failed:', error);
|
||||||
|
return { success: false, message: error.message };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = new PricingService();
|
||||||
224
src/utils/costCalculator.js
Normal file
224
src/utils/costCalculator.js
Normal file
@@ -0,0 +1,224 @@
|
|||||||
|
const pricingService = require('../services/pricingService');
|
||||||
|
|
||||||
|
// Claude模型价格配置 (USD per 1M tokens) - 备用定价
|
||||||
|
const MODEL_PRICING = {
|
||||||
|
// Claude 3.5 Sonnet
|
||||||
|
'claude-3-5-sonnet-20241022': {
|
||||||
|
input: 3.00,
|
||||||
|
output: 15.00,
|
||||||
|
cacheWrite: 3.75,
|
||||||
|
cacheRead: 0.30
|
||||||
|
},
|
||||||
|
'claude-sonnet-4-20250514': {
|
||||||
|
input: 3.00,
|
||||||
|
output: 15.00,
|
||||||
|
cacheWrite: 3.75,
|
||||||
|
cacheRead: 0.30
|
||||||
|
},
|
||||||
|
|
||||||
|
// Claude 3.5 Haiku
|
||||||
|
'claude-3-5-haiku-20241022': {
|
||||||
|
input: 0.25,
|
||||||
|
output: 1.25,
|
||||||
|
cacheWrite: 0.30,
|
||||||
|
cacheRead: 0.03
|
||||||
|
},
|
||||||
|
|
||||||
|
// Claude 3 Opus
|
||||||
|
'claude-3-opus-20240229': {
|
||||||
|
input: 15.00,
|
||||||
|
output: 75.00,
|
||||||
|
cacheWrite: 18.75,
|
||||||
|
cacheRead: 1.50
|
||||||
|
},
|
||||||
|
|
||||||
|
// Claude 3 Sonnet
|
||||||
|
'claude-3-sonnet-20240229': {
|
||||||
|
input: 3.00,
|
||||||
|
output: 15.00,
|
||||||
|
cacheWrite: 3.75,
|
||||||
|
cacheRead: 0.30
|
||||||
|
},
|
||||||
|
|
||||||
|
// Claude 3 Haiku
|
||||||
|
'claude-3-haiku-20240307': {
|
||||||
|
input: 0.25,
|
||||||
|
output: 1.25,
|
||||||
|
cacheWrite: 0.30,
|
||||||
|
cacheRead: 0.03
|
||||||
|
},
|
||||||
|
|
||||||
|
// 默认定价（用于未知模型）
|
||||||
|
'unknown': {
|
||||||
|
input: 3.00,
|
||||||
|
output: 15.00,
|
||||||
|
cacheWrite: 3.75,
|
||||||
|
cacheRead: 0.30
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
class CostCalculator {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 计算单次请求的费用
|
||||||
|
* @param {Object} usage - 使用量数据
|
||||||
|
* @param {number} usage.input_tokens - 输入token数量
|
||||||
|
* @param {number} usage.output_tokens - 输出token数量
|
||||||
|
* @param {number} usage.cache_creation_input_tokens - 缓存创建token数量
|
||||||
|
* @param {number} usage.cache_read_input_tokens - 缓存读取token数量
|
||||||
|
* @param {string} model - 模型名称
|
||||||
|
* @returns {Object} 费用详情
|
||||||
|
*/
|
||||||
|
static calculateCost(usage, model = 'unknown') {
|
||||||
|
const inputTokens = usage.input_tokens || 0;
|
||||||
|
const outputTokens = usage.output_tokens || 0;
|
||||||
|
const cacheCreateTokens = usage.cache_creation_input_tokens || 0;
|
||||||
|
const cacheReadTokens = usage.cache_read_input_tokens || 0;
|
||||||
|
|
||||||
|
// 优先使用动态价格服务
|
||||||
|
const pricingData = pricingService.getModelPricing(model);
|
||||||
|
let pricing;
|
||||||
|
let usingDynamicPricing = false;
|
||||||
|
|
||||||
|
if (pricingData) {
|
||||||
|
// 转换动态价格格式为内部格式
|
||||||
|
pricing = {
|
||||||
|
input: (pricingData.input_cost_per_token || 0) * 1000000, // 转换为per 1M tokens
|
||||||
|
output: (pricingData.output_cost_per_token || 0) * 1000000,
|
||||||
|
cacheWrite: (pricingData.cache_creation_input_token_cost || 0) * 1000000,
|
||||||
|
cacheRead: (pricingData.cache_read_input_token_cost || 0) * 1000000
|
||||||
|
};
|
||||||
|
usingDynamicPricing = true;
|
||||||
|
} else {
|
||||||
|
// 回退到静态价格
|
||||||
|
pricing = MODEL_PRICING[model] || MODEL_PRICING['unknown'];
|
||||||
|
}
|
||||||
|
|
||||||
|
// 计算各类型token的费用 (USD)
|
||||||
|
const inputCost = (inputTokens / 1000000) * pricing.input;
|
||||||
|
const outputCost = (outputTokens / 1000000) * pricing.output;
|
||||||
|
const cacheWriteCost = (cacheCreateTokens / 1000000) * pricing.cacheWrite;
|
||||||
|
const cacheReadCost = (cacheReadTokens / 1000000) * pricing.cacheRead;
|
||||||
|
|
||||||
|
const totalCost = inputCost + outputCost + cacheWriteCost + cacheReadCost;
|
||||||
|
|
||||||
|
return {
|
||||||
|
model,
|
||||||
|
pricing,
|
||||||
|
usingDynamicPricing,
|
||||||
|
usage: {
|
||||||
|
inputTokens,
|
||||||
|
outputTokens,
|
||||||
|
cacheCreateTokens,
|
||||||
|
cacheReadTokens,
|
||||||
|
totalTokens: inputTokens + outputTokens + cacheCreateTokens + cacheReadTokens
|
||||||
|
},
|
||||||
|
costs: {
|
||||||
|
input: inputCost,
|
||||||
|
output: outputCost,
|
||||||
|
cacheWrite: cacheWriteCost,
|
||||||
|
cacheRead: cacheReadCost,
|
||||||
|
total: totalCost
|
||||||
|
},
|
||||||
|
// 格式化的费用字符串
|
||||||
|
formatted: {
|
||||||
|
input: this.formatCost(inputCost),
|
||||||
|
output: this.formatCost(outputCost),
|
||||||
|
cacheWrite: this.formatCost(cacheWriteCost),
|
||||||
|
cacheRead: this.formatCost(cacheReadCost),
|
||||||
|
total: this.formatCost(totalCost)
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 计算聚合使用量的费用
|
||||||
|
* @param {Object} aggregatedUsage - 聚合使用量数据
|
||||||
|
* @param {string} model - 模型名称
|
||||||
|
* @returns {Object} 费用详情
|
||||||
|
*/
|
||||||
|
static calculateAggregatedCost(aggregatedUsage, model = 'unknown') {
|
||||||
|
const usage = {
|
||||||
|
input_tokens: aggregatedUsage.inputTokens || aggregatedUsage.totalInputTokens || 0,
|
||||||
|
output_tokens: aggregatedUsage.outputTokens || aggregatedUsage.totalOutputTokens || 0,
|
||||||
|
cache_creation_input_tokens: aggregatedUsage.cacheCreateTokens || aggregatedUsage.totalCacheCreateTokens || 0,
|
||||||
|
cache_read_input_tokens: aggregatedUsage.cacheReadTokens || aggregatedUsage.totalCacheReadTokens || 0
|
||||||
|
};
|
||||||
|
|
||||||
|
return this.calculateCost(usage, model);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取模型定价信息
|
||||||
|
* @param {string} model - 模型名称
|
||||||
|
* @returns {Object} 定价信息
|
||||||
|
*/
|
||||||
|
static getModelPricing(model = 'unknown') {
|
||||||
|
return MODEL_PRICING[model] || MODEL_PRICING['unknown'];
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取所有支持的模型和定价
|
||||||
|
* @returns {Object} 所有模型定价
|
||||||
|
*/
|
||||||
|
static getAllModelPricing() {
|
||||||
|
return { ...MODEL_PRICING };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 验证模型是否支持
|
||||||
|
* @param {string} model - 模型名称
|
||||||
|
* @returns {boolean} 是否支持
|
||||||
|
*/
|
||||||
|
static isModelSupported(model) {
|
||||||
|
return !!MODEL_PRICING[model];
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 格式化费用显示
|
||||||
|
* @param {number} cost - 费用金额
|
||||||
|
* @param {number} decimals - 小数位数
|
||||||
|
* @returns {string} 格式化的费用字符串
|
||||||
|
*/
|
||||||
|
static formatCost(cost, decimals = 6) {
|
||||||
|
if (cost >= 1) {
|
||||||
|
return `$${cost.toFixed(2)}`;
|
||||||
|
} else if (cost >= 0.001) {
|
||||||
|
return `$${cost.toFixed(4)}`;
|
||||||
|
} else {
|
||||||
|
return `$${cost.toFixed(decimals)}`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 计算费用节省（使用缓存的节省）
|
||||||
|
* @param {Object} usage - 使用量数据
|
||||||
|
* @param {string} model - 模型名称
|
||||||
|
* @returns {Object} 节省信息
|
||||||
|
*/
|
||||||
|
static calculateCacheSavings(usage, model = 'unknown') {
|
||||||
|
const pricing = this.getModelPricing(model);
|
||||||
|
const cacheReadTokens = usage.cache_read_input_tokens || 0;
|
||||||
|
|
||||||
|
// 如果这些token不使用缓存，需要按正常input价格计费
|
||||||
|
const normalCost = (cacheReadTokens / 1000000) * pricing.input;
|
||||||
|
const cacheCost = (cacheReadTokens / 1000000) * pricing.cacheRead;
|
||||||
|
const savings = normalCost - cacheCost;
|
||||||
|
const savingsPercentage = normalCost > 0 ? (savings / normalCost) * 100 : 0;
|
||||||
|
|
||||||
|
return {
|
||||||
|
normalCost,
|
||||||
|
cacheCost,
|
||||||
|
savings,
|
||||||
|
savingsPercentage,
|
||||||
|
formatted: {
|
||||||
|
normalCost: this.formatCost(normalCost),
|
||||||
|
cacheCost: this.formatCost(cacheCost),
|
||||||
|
savings: this.formatCost(savings),
|
||||||
|
savingsPercentage: `${savingsPercentage.toFixed(1)}%`
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = CostCalculator;
|
||||||
290
src/utils/logger.js
Normal file
290
src/utils/logger.js
Normal file
@@ -0,0 +1,290 @@
|
|||||||
|
const winston = require('winston');
|
||||||
|
const DailyRotateFile = require('winston-daily-rotate-file');
|
||||||
|
const config = require('../../config/config');
|
||||||
|
const path = require('path');
|
||||||
|
const fs = require('fs');
|
||||||
|
const os = require('os');
|
||||||
|
|
||||||
|
// 📝 增强的日志格式
|
||||||
|
const createLogFormat = (colorize = false) => {
|
||||||
|
const formats = [
|
||||||
|
winston.format.timestamp({ format: 'YYYY-MM-DD HH:mm:ss' }),
|
||||||
|
winston.format.errors({ stack: true }),
|
||||||
|
winston.format.metadata({ fillExcept: ['message', 'level', 'timestamp', 'stack'] })
|
||||||
|
];
|
||||||
|
|
||||||
|
if (colorize) {
|
||||||
|
formats.push(winston.format.colorize());
|
||||||
|
}
|
||||||
|
|
||||||
|
formats.push(
|
||||||
|
winston.format.printf(({ level, message, timestamp, stack, metadata, ...rest }) => {
|
||||||
|
const emoji = {
|
||||||
|
error: '❌',
|
||||||
|
warn: '⚠️ ',
|
||||||
|
info: 'ℹ️ ',
|
||||||
|
debug: '🐛',
|
||||||
|
verbose: '📝'
|
||||||
|
};
|
||||||
|
|
||||||
|
let logMessage = `${emoji[level] || '📝'} [${timestamp}] ${level.toUpperCase()}: ${message}`;
|
||||||
|
|
||||||
|
// 添加元数据
|
||||||
|
if (metadata && Object.keys(metadata).length > 0) {
|
||||||
|
logMessage += ` | ${JSON.stringify(metadata)}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 添加其他属性
|
||||||
|
const additionalData = { ...rest };
|
||||||
|
delete additionalData.level;
|
||||||
|
delete additionalData.message;
|
||||||
|
delete additionalData.timestamp;
|
||||||
|
delete additionalData.stack;
|
||||||
|
|
||||||
|
if (Object.keys(additionalData).length > 0) {
|
||||||
|
logMessage += ` | ${JSON.stringify(additionalData)}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
return stack ? `${logMessage}\n${stack}` : logMessage;
|
||||||
|
})
|
||||||
|
);
|
||||||
|
|
||||||
|
return winston.format.combine(...formats);
|
||||||
|
};
|
||||||
|
|
||||||
|
const logFormat = createLogFormat(false);
|
||||||
|
const consoleFormat = createLogFormat(true);
|
||||||
|
|
||||||
|
// 📁 确保日志目录存在并设置权限
|
||||||
|
if (!fs.existsSync(config.logging.dirname)) {
|
||||||
|
fs.mkdirSync(config.logging.dirname, { recursive: true, mode: 0o755 });
|
||||||
|
}
|
||||||
|
|
||||||
|
// 🔄 增强的日志轮转配置
|
||||||
|
const createRotateTransport = (filename, level = null) => {
|
||||||
|
const transport = new DailyRotateFile({
|
||||||
|
filename: path.join(config.logging.dirname, filename),
|
||||||
|
datePattern: 'YYYY-MM-DD',
|
||||||
|
zippedArchive: true,
|
||||||
|
maxSize: config.logging.maxSize,
|
||||||
|
maxFiles: config.logging.maxFiles,
|
||||||
|
auditFile: path.join(config.logging.dirname, `.${filename.replace('%DATE%', 'audit')}.json`),
|
||||||
|
format: logFormat
|
||||||
|
});
|
||||||
|
|
||||||
|
if (level) {
|
||||||
|
transport.level = level;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 监听轮转事件
|
||||||
|
transport.on('rotate', (oldFilename, newFilename) => {
|
||||||
|
console.log(`📦 Log rotated: ${oldFilename} -> ${newFilename}`);
|
||||||
|
});
|
||||||
|
|
||||||
|
transport.on('new', (newFilename) => {
|
||||||
|
console.log(`📄 New log file created: ${newFilename}`);
|
||||||
|
});
|
||||||
|
|
||||||
|
transport.on('archive', (zipFilename) => {
|
||||||
|
console.log(`🗜️ Log archived: ${zipFilename}`);
|
||||||
|
});
|
||||||
|
|
||||||
|
return transport;
|
||||||
|
};
|
||||||
|
|
||||||
|
const dailyRotateFileTransport = createRotateTransport('claude-relay-%DATE%.log');
|
||||||
|
const errorFileTransport = createRotateTransport('claude-relay-error-%DATE%.log', 'error');
|
||||||
|
|
||||||
|
// 🔒 创建专门的安全日志记录器
|
||||||
|
const securityLogger = winston.createLogger({
|
||||||
|
level: 'warn',
|
||||||
|
format: logFormat,
|
||||||
|
transports: [
|
||||||
|
createRotateTransport('claude-relay-security-%DATE%.log', 'warn')
|
||||||
|
],
|
||||||
|
silent: false
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🌟 增强的 Winston logger
|
||||||
|
const logger = winston.createLogger({
|
||||||
|
level: config.logging.level,
|
||||||
|
format: logFormat,
|
||||||
|
transports: [
|
||||||
|
// 📄 文件输出
|
||||||
|
dailyRotateFileTransport,
|
||||||
|
errorFileTransport,
|
||||||
|
|
||||||
|
// 🖥️ 控制台输出
|
||||||
|
new winston.transports.Console({
|
||||||
|
format: consoleFormat,
|
||||||
|
handleExceptions: false,
|
||||||
|
handleRejections: false
|
||||||
|
})
|
||||||
|
],
|
||||||
|
|
||||||
|
// 🚨 异常处理
|
||||||
|
exceptionHandlers: [
|
||||||
|
new winston.transports.File({
|
||||||
|
filename: path.join(config.logging.dirname, 'exceptions.log'),
|
||||||
|
format: logFormat,
|
||||||
|
maxsize: 10485760, // 10MB
|
||||||
|
maxFiles: 5
|
||||||
|
}),
|
||||||
|
new winston.transports.Console({
|
||||||
|
format: consoleFormat
|
||||||
|
})
|
||||||
|
],
|
||||||
|
|
||||||
|
// 🔄 未捕获异常处理
|
||||||
|
rejectionHandlers: [
|
||||||
|
new winston.transports.File({
|
||||||
|
filename: path.join(config.logging.dirname, 'rejections.log'),
|
||||||
|
format: logFormat,
|
||||||
|
maxsize: 10485760, // 10MB
|
||||||
|
maxFiles: 5
|
||||||
|
}),
|
||||||
|
new winston.transports.Console({
|
||||||
|
format: consoleFormat
|
||||||
|
})
|
||||||
|
],
|
||||||
|
|
||||||
|
// 防止进程退出
|
||||||
|
exitOnError: false
|
||||||
|
});
|
||||||
|
|
||||||
|
// 🎯 增强的自定义方法
|
||||||
|
logger.success = (message, metadata = {}) => {
|
||||||
|
logger.info(`✅ ${message}`, { type: 'success', ...metadata });
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.start = (message, metadata = {}) => {
|
||||||
|
logger.info(`🚀 ${message}`, { type: 'startup', ...metadata });
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.request = (method, url, status, duration, metadata = {}) => {
|
||||||
|
const emoji = status >= 400 ? '🔴' : status >= 300 ? '🟡' : '🟢';
|
||||||
|
const level = status >= 400 ? 'error' : status >= 300 ? 'warn' : 'info';
|
||||||
|
|
||||||
|
logger[level](`${emoji} ${method} ${url} - ${status} (${duration}ms)`, {
|
||||||
|
type: 'request',
|
||||||
|
method,
|
||||||
|
url,
|
||||||
|
status,
|
||||||
|
duration,
|
||||||
|
...metadata
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.api = (message, metadata = {}) => {
|
||||||
|
logger.info(`🔗 ${message}`, { type: 'api', ...metadata });
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.security = (message, metadata = {}) => {
|
||||||
|
const securityData = {
|
||||||
|
type: 'security',
|
||||||
|
timestamp: new Date().toISOString(),
|
||||||
|
pid: process.pid,
|
||||||
|
hostname: os.hostname(),
|
||||||
|
...metadata
|
||||||
|
};
|
||||||
|
|
||||||
|
// 记录到主日志
|
||||||
|
logger.warn(`🔒 ${message}`, securityData);
|
||||||
|
|
||||||
|
// 记录到专门的安全日志文件
|
||||||
|
try {
|
||||||
|
securityLogger.warn(`🔒 ${message}`, securityData);
|
||||||
|
} catch (error) {
|
||||||
|
// 如果安全日志文件不可用，只记录到主日志
|
||||||
|
console.warn('Security logger not available:', error.message);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.database = (message, metadata = {}) => {
|
||||||
|
logger.debug(`💾 ${message}`, { type: 'database', ...metadata });
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.performance = (message, metadata = {}) => {
|
||||||
|
logger.info(`⚡ ${message}`, { type: 'performance', ...metadata });
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.audit = (message, metadata = {}) => {
|
||||||
|
logger.info(`📋 ${message}`, {
|
||||||
|
type: 'audit',
|
||||||
|
timestamp: new Date().toISOString(),
|
||||||
|
pid: process.pid,
|
||||||
|
...metadata
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
// 🔧 性能监控方法
|
||||||
|
logger.timer = (label) => {
|
||||||
|
const start = Date.now();
|
||||||
|
return {
|
||||||
|
end: (message = '', metadata = {}) => {
|
||||||
|
const duration = Date.now() - start;
|
||||||
|
logger.performance(`${label} ${message}`, { duration, ...metadata });
|
||||||
|
return duration;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
// 📊 日志统计
|
||||||
|
logger.stats = {
|
||||||
|
requests: 0,
|
||||||
|
errors: 0,
|
||||||
|
warnings: 0
|
||||||
|
};
|
||||||
|
|
||||||
|
// 重写原始方法以统计
|
||||||
|
const originalError = logger.error;
|
||||||
|
const originalWarn = logger.warn;
|
||||||
|
const originalInfo = logger.info;
|
||||||
|
|
||||||
|
logger.error = function(message, metadata = {}) {
|
||||||
|
logger.stats.errors++;
|
||||||
|
return originalError.call(this, message, metadata);
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.warn = function(message, metadata = {}) {
|
||||||
|
logger.stats.warnings++;
|
||||||
|
return originalWarn.call(this, message, metadata);
|
||||||
|
};
|
||||||
|
|
||||||
|
logger.info = function(message, metadata = {}) {
|
||||||
|
if (metadata.type === 'request') {
|
||||||
|
logger.stats.requests++;
|
||||||
|
}
|
||||||
|
return originalInfo.call(this, message, metadata);
|
||||||
|
};
|
||||||
|
|
||||||
|
// 📈 获取日志统计
|
||||||
|
logger.getStats = () => ({ ...logger.stats });
|
||||||
|
|
||||||
|
// 🧹 清理统计
|
||||||
|
logger.resetStats = () => {
|
||||||
|
logger.stats.requests = 0;
|
||||||
|
logger.stats.errors = 0;
|
||||||
|
logger.stats.warnings = 0;
|
||||||
|
};
|
||||||
|
|
||||||
|
// 📡 健康检查
|
||||||
|
logger.healthCheck = () => {
|
||||||
|
try {
|
||||||
|
const testMessage = 'Logger health check';
|
||||||
|
logger.debug(testMessage);
|
||||||
|
return { healthy: true, timestamp: new Date().toISOString() };
|
||||||
|
} catch (error) {
|
||||||
|
return { healthy: false, error: error.message, timestamp: new Date().toISOString() };
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// 🎬 启动日志记录系统
|
||||||
|
logger.start('Logger initialized', {
|
||||||
|
level: config.logging.level,
|
||||||
|
directory: config.logging.dirname,
|
||||||
|
maxSize: config.logging.maxSize,
|
||||||
|
maxFiles: config.logging.maxFiles
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = logger;
|
||||||
307
src/utils/oauthHelper.js
Normal file
307
src/utils/oauthHelper.js
Normal file
@@ -0,0 +1,307 @@
|
|||||||
|
/**
|
||||||
|
* OAuth助手工具
|
||||||
|
* 基于claude-code-login.js中的OAuth流程实现
|
||||||
|
*/
|
||||||
|
|
||||||
|
const crypto = require('crypto');
|
||||||
|
const { SocksProxyAgent } = require('socks-proxy-agent');
|
||||||
|
const { HttpsProxyAgent } = require('https-proxy-agent');
|
||||||
|
const axios = require('axios');
|
||||||
|
const logger = require('./logger');
|
||||||
|
|
||||||
|
// OAuth 配置常量 - 从claude-code-login.js提取
|
||||||
|
const OAUTH_CONFIG = {
|
||||||
|
AUTHORIZE_URL: 'https://claude.ai/oauth/authorize',
|
||||||
|
TOKEN_URL: 'https://console.anthropic.com/v1/oauth/token',
|
||||||
|
CLIENT_ID: '9d1c250a-e61b-44d9-88ed-5944d1962f5e',
|
||||||
|
REDIRECT_URI: 'https://console.anthropic.com/oauth/code/callback',
|
||||||
|
SCOPES: 'org:create_api_key user:profile user:inference'
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成随机的 state 参数
|
||||||
|
* @returns {string} 随机生成的 state (64字符hex)
|
||||||
|
*/
|
||||||
|
function generateState() {
|
||||||
|
return crypto.randomBytes(32).toString('hex');
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成随机的 code verifier（PKCE）
|
||||||
|
* @returns {string} base64url 编码的随机字符串
|
||||||
|
*/
|
||||||
|
function generateCodeVerifier() {
|
||||||
|
return crypto.randomBytes(32).toString('base64url');
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成 code challenge（PKCE）
|
||||||
|
* @param {string} codeVerifier - code verifier 字符串
|
||||||
|
* @returns {string} SHA256 哈希后的 base64url 编码字符串
|
||||||
|
*/
|
||||||
|
function generateCodeChallenge(codeVerifier) {
|
||||||
|
return crypto.createHash('sha256')
|
||||||
|
.update(codeVerifier)
|
||||||
|
.digest('base64url');
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成授权 URL
|
||||||
|
* @param {string} codeChallenge - PKCE code challenge
|
||||||
|
* @param {string} state - state 参数
|
||||||
|
* @returns {string} 完整的授权 URL
|
||||||
|
*/
|
||||||
|
function generateAuthUrl(codeChallenge, state) {
|
||||||
|
const params = new URLSearchParams({
|
||||||
|
code: 'true',
|
||||||
|
client_id: OAUTH_CONFIG.CLIENT_ID,
|
||||||
|
response_type: 'code',
|
||||||
|
redirect_uri: OAUTH_CONFIG.REDIRECT_URI,
|
||||||
|
scope: OAUTH_CONFIG.SCOPES,
|
||||||
|
code_challenge: codeChallenge,
|
||||||
|
code_challenge_method: 'S256',
|
||||||
|
state: state
|
||||||
|
});
|
||||||
|
|
||||||
|
return `${OAUTH_CONFIG.AUTHORIZE_URL}?${params.toString()}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成OAuth授权URL和相关参数
|
||||||
|
* @returns {{authUrl: string, codeVerifier: string, state: string, codeChallenge: string}}
|
||||||
|
*/
|
||||||
|
function generateOAuthParams() {
|
||||||
|
const state = generateState();
|
||||||
|
const codeVerifier = generateCodeVerifier();
|
||||||
|
const codeChallenge = generateCodeChallenge(codeVerifier);
|
||||||
|
|
||||||
|
const authUrl = generateAuthUrl(codeChallenge, state);
|
||||||
|
|
||||||
|
return {
|
||||||
|
authUrl,
|
||||||
|
codeVerifier,
|
||||||
|
state,
|
||||||
|
codeChallenge
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 创建代理agent
|
||||||
|
* @param {object|null} proxyConfig - 代理配置对象
|
||||||
|
* @returns {object|null} 代理agent或null
|
||||||
|
*/
|
||||||
|
function createProxyAgent(proxyConfig) {
|
||||||
|
if (!proxyConfig) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
if (proxyConfig.type === 'socks5') {
|
||||||
|
const auth = proxyConfig.username && proxyConfig.password ? `${proxyConfig.username}:${proxyConfig.password}@` : '';
|
||||||
|
const socksUrl = `socks5://${auth}${proxyConfig.host}:${proxyConfig.port}`;
|
||||||
|
return new SocksProxyAgent(socksUrl);
|
||||||
|
} else if (proxyConfig.type === 'http' || proxyConfig.type === 'https') {
|
||||||
|
const auth = proxyConfig.username && proxyConfig.password ? `${proxyConfig.username}:${proxyConfig.password}@` : '';
|
||||||
|
const httpUrl = `${proxyConfig.type}://${auth}${proxyConfig.host}:${proxyConfig.port}`;
|
||||||
|
return new HttpsProxyAgent(httpUrl);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.warn('⚠️ Invalid proxy configuration:', error);
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 使用授权码交换访问令牌
|
||||||
|
* @param {string} authorizationCode - 授权码
|
||||||
|
* @param {string} codeVerifier - PKCE code verifier
|
||||||
|
* @param {string} state - state 参数
|
||||||
|
* @param {object|null} proxyConfig - 代理配置（可选）
|
||||||
|
* @returns {Promise<object>} Claude格式的token响应
|
||||||
|
*/
|
||||||
|
async function exchangeCodeForTokens(authorizationCode, codeVerifier, state, proxyConfig = null) {
|
||||||
|
// 清理授权码，移除URL片段
|
||||||
|
const cleanedCode = authorizationCode.split('#')[0]?.split('&')[0] ?? authorizationCode;
|
||||||
|
|
||||||
|
const params = {
|
||||||
|
grant_type: 'authorization_code',
|
||||||
|
client_id: OAUTH_CONFIG.CLIENT_ID,
|
||||||
|
code: cleanedCode,
|
||||||
|
redirect_uri: OAUTH_CONFIG.REDIRECT_URI,
|
||||||
|
code_verifier: codeVerifier,
|
||||||
|
state: state
|
||||||
|
};
|
||||||
|
|
||||||
|
// 创建代理agent
|
||||||
|
const agent = createProxyAgent(proxyConfig);
|
||||||
|
|
||||||
|
try {
|
||||||
|
logger.debug('🔄 Attempting OAuth token exchange', {
|
||||||
|
url: OAUTH_CONFIG.TOKEN_URL,
|
||||||
|
codeLength: cleanedCode.length,
|
||||||
|
codePrefix: cleanedCode.substring(0, 10) + '...',
|
||||||
|
hasProxy: !!proxyConfig,
|
||||||
|
proxyType: proxyConfig?.type || 'none'
|
||||||
|
});
|
||||||
|
|
||||||
|
const response = await axios.post(OAUTH_CONFIG.TOKEN_URL, params, {
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36',
|
||||||
|
'Accept': 'application/json, text/plain, */*',
|
||||||
|
'Accept-Language': 'en-US,en;q=0.9',
|
||||||
|
'Referer': 'https://claude.ai/',
|
||||||
|
'Origin': 'https://claude.ai'
|
||||||
|
},
|
||||||
|
httpsAgent: agent,
|
||||||
|
timeout: 30000
|
||||||
|
});
|
||||||
|
|
||||||
|
logger.success('✅ OAuth token exchange successful', {
|
||||||
|
status: response.status,
|
||||||
|
hasAccessToken: !!response.data?.access_token,
|
||||||
|
hasRefreshToken: !!response.data?.refresh_token,
|
||||||
|
scopes: response.data?.scope
|
||||||
|
});
|
||||||
|
|
||||||
|
const data = response.data;
|
||||||
|
|
||||||
|
// 返回Claude格式的token数据
|
||||||
|
return {
|
||||||
|
accessToken: data.access_token,
|
||||||
|
refreshToken: data.refresh_token,
|
||||||
|
expiresAt: (Math.floor(Date.now() / 1000) + data.expires_in) * 1000,
|
||||||
|
scopes: data.scope ? data.scope.split(' ') : ['user:inference', 'user:profile'],
|
||||||
|
isMax: true
|
||||||
|
};
|
||||||
|
} catch (error) {
|
||||||
|
// 处理axios错误响应
|
||||||
|
if (error.response) {
|
||||||
|
// 服务器返回了错误状态码
|
||||||
|
const status = error.response.status;
|
||||||
|
const errorData = error.response.data;
|
||||||
|
|
||||||
|
logger.error('❌ OAuth token exchange failed with server error', {
|
||||||
|
status: status,
|
||||||
|
statusText: error.response.statusText,
|
||||||
|
headers: error.response.headers,
|
||||||
|
data: errorData,
|
||||||
|
codeLength: cleanedCode.length,
|
||||||
|
codePrefix: cleanedCode.substring(0, 10) + '...'
|
||||||
|
});
|
||||||
|
|
||||||
|
// 尝试从错误响应中提取有用信息
|
||||||
|
let errorMessage = `HTTP ${status}`;
|
||||||
|
|
||||||
|
if (errorData) {
|
||||||
|
if (typeof errorData === 'string') {
|
||||||
|
errorMessage += `: ${errorData}`;
|
||||||
|
} else if (errorData.error) {
|
||||||
|
errorMessage += `: ${errorData.error}`;
|
||||||
|
if (errorData.error_description) {
|
||||||
|
errorMessage += ` - ${errorData.error_description}`;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
errorMessage += `: ${JSON.stringify(errorData)}`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new Error(`Token exchange failed: ${errorMessage}`);
|
||||||
|
} else if (error.request) {
|
||||||
|
// 请求被发送但没有收到响应
|
||||||
|
logger.error('❌ OAuth token exchange failed with network error', {
|
||||||
|
message: error.message,
|
||||||
|
code: error.code,
|
||||||
|
hasProxy: !!proxyConfig
|
||||||
|
});
|
||||||
|
throw new Error('Token exchange failed: No response from server (network error or timeout)');
|
||||||
|
} else {
|
||||||
|
// 其他错误
|
||||||
|
logger.error('❌ OAuth token exchange failed with unknown error', {
|
||||||
|
message: error.message,
|
||||||
|
stack: error.stack
|
||||||
|
});
|
||||||
|
throw new Error(`Token exchange failed: ${error.message}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 解析回调 URL 或授权码
|
||||||
|
* @param {string} input - 完整的回调 URL 或直接的授权码
|
||||||
|
* @returns {string} 授权码
|
||||||
|
*/
|
||||||
|
function parseCallbackUrl(input) {
|
||||||
|
if (!input || typeof input !== 'string') {
|
||||||
|
throw new Error('请提供有效的授权码或回调 URL');
|
||||||
|
}
|
||||||
|
|
||||||
|
const trimmedInput = input.trim();
|
||||||
|
|
||||||
|
// 情况1: 尝试作为完整URL解析
|
||||||
|
if (trimmedInput.startsWith('http://') || trimmedInput.startsWith('https://')) {
|
||||||
|
try {
|
||||||
|
const urlObj = new URL(trimmedInput);
|
||||||
|
const authorizationCode = urlObj.searchParams.get('code');
|
||||||
|
|
||||||
|
if (!authorizationCode) {
|
||||||
|
throw new Error('回调 URL 中未找到授权码 (code 参数)');
|
||||||
|
}
|
||||||
|
|
||||||
|
return authorizationCode;
|
||||||
|
} catch (error) {
|
||||||
|
if (error.message.includes('回调 URL 中未找到授权码')) {
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
throw new Error('无效的 URL 格式，请检查回调 URL 是否正确');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 情况2: 直接的授权码（可能包含URL fragments）
|
||||||
|
// 参考claude-code-login.js的处理方式：移除URL fragments和参数
|
||||||
|
const cleanedCode = trimmedInput.split('#')[0]?.split('&')[0] ?? trimmedInput;
|
||||||
|
|
||||||
|
// 验证授权码格式（Claude的授权码通常是base64url格式）
|
||||||
|
if (!cleanedCode || cleanedCode.length < 10) {
|
||||||
|
throw new Error('授权码格式无效，请确保复制了完整的 Authorization Code');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 基本格式验证：授权码应该只包含字母、数字、下划线、连字符
|
||||||
|
const validCodePattern = /^[A-Za-z0-9_-]+$/;
|
||||||
|
if (!validCodePattern.test(cleanedCode)) {
|
||||||
|
throw new Error('授权码包含无效字符，请检查是否复制了正确的 Authorization Code');
|
||||||
|
}
|
||||||
|
|
||||||
|
return cleanedCode;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 格式化为Claude标准格式
|
||||||
|
* @param {object} tokenData - token数据
|
||||||
|
* @returns {object} claudeAiOauth格式的数据
|
||||||
|
*/
|
||||||
|
function formatClaudeCredentials(tokenData) {
|
||||||
|
return {
|
||||||
|
claudeAiOauth: {
|
||||||
|
accessToken: tokenData.accessToken,
|
||||||
|
refreshToken: tokenData.refreshToken,
|
||||||
|
expiresAt: tokenData.expiresAt,
|
||||||
|
scopes: tokenData.scopes,
|
||||||
|
isMax: tokenData.isMax
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
OAUTH_CONFIG,
|
||||||
|
generateOAuthParams,
|
||||||
|
exchangeCodeForTokens,
|
||||||
|
parseCallbackUrl,
|
||||||
|
formatClaudeCredentials,
|
||||||
|
generateState,
|
||||||
|
generateCodeVerifier,
|
||||||
|
generateCodeChallenge,
|
||||||
|
generateAuthUrl,
|
||||||
|
createProxyAgent
|
||||||
|
};
|
||||||
1989
web/admin/app.js
Normal file
1989
web/admin/app.js
Normal file
File diff suppressed because it is too large
Load Diff
2058
web/admin/index.html
Normal file
2058
web/admin/index.html
Normal file
File diff suppressed because it is too large
Load Diff
395
web/admin/style.css
Normal file
395
web/admin/style.css
Normal file
@@ -0,0 +1,395 @@
|
|||||||
|
:root {
|
||||||
|
--primary-color: #667eea;
|
||||||
|
--secondary-color: #764ba2;
|
||||||
|
--accent-color: #f093fb;
|
||||||
|
--success-color: #10b981;
|
||||||
|
--warning-color: #f59e0b;
|
||||||
|
--error-color: #ef4444;
|
||||||
|
--surface-color: rgba(255, 255, 255, 0.95);
|
||||||
|
--glass-color: rgba(255, 255, 255, 0.1);
|
||||||
|
--text-primary: #1f2937;
|
||||||
|
--text-secondary: #6b7280;
|
||||||
|
--border-color: rgba(255, 255, 255, 0.2);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 通用transition - 仅应用于特定元素 */
|
||||||
|
body, div, button, input, select, textarea, table, tr, td, th, span, p, h1, h2, h3, h4, h5, h6 {
|
||||||
|
transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
|
||||||
|
background: linear-gradient(135deg, var(--primary-color) 0%, var(--secondary-color) 50%, var(--accent-color) 100%);
|
||||||
|
background-attachment: fixed;
|
||||||
|
min-height: 100vh;
|
||||||
|
margin: 0;
|
||||||
|
overflow-x: hidden;
|
||||||
|
}
|
||||||
|
|
||||||
|
body::before {
|
||||||
|
content: '';
|
||||||
|
position: fixed;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
right: 0;
|
||||||
|
bottom: 0;
|
||||||
|
background:
|
||||||
|
radial-gradient(circle at 20% 80%, rgba(240, 147, 251, 0.2) 0%, transparent 50%),
|
||||||
|
radial-gradient(circle at 80% 20%, rgba(102, 126, 234, 0.2) 0%, transparent 50%),
|
||||||
|
radial-gradient(circle at 40% 40%, rgba(118, 75, 162, 0.1) 0%, transparent 50%);
|
||||||
|
pointer-events: none;
|
||||||
|
z-index: -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.glass {
|
||||||
|
background: var(--glass-color);
|
||||||
|
backdrop-filter: blur(20px);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
box-shadow:
|
||||||
|
0 20px 25px -5px rgba(0, 0, 0, 0.1),
|
||||||
|
0 10px 10px -5px rgba(0, 0, 0, 0.04),
|
||||||
|
inset 0 1px 0 rgba(255, 255, 255, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.glass-strong {
|
||||||
|
background: var(--surface-color);
|
||||||
|
backdrop-filter: blur(25px);
|
||||||
|
border: 1px solid rgba(255, 255, 255, 0.3);
|
||||||
|
box-shadow:
|
||||||
|
0 25px 50px -12px rgba(0, 0, 0, 0.25),
|
||||||
|
0 0 0 1px rgba(255, 255, 255, 0.05),
|
||||||
|
inset 0 1px 0 rgba(255, 255, 255, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.tab-btn {
|
||||||
|
position: relative;
|
||||||
|
overflow: hidden;
|
||||||
|
border-radius: 12px;
|
||||||
|
font-weight: 500;
|
||||||
|
letter-spacing: 0.025em;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tab-btn::before {
|
||||||
|
content: '';
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
left: -100%;
|
||||||
|
width: 100%;
|
||||||
|
height: 100%;
|
||||||
|
background: linear-gradient(90deg, transparent, rgba(255, 255, 255, 0.2), transparent);
|
||||||
|
transition: left 0.5s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tab-btn:hover::before {
|
||||||
|
left: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tab-btn.active {
|
||||||
|
background: linear-gradient(135deg, var(--primary-color) 0%, var(--secondary-color) 100%);
|
||||||
|
color: white;
|
||||||
|
box-shadow:
|
||||||
|
0 10px 15px -3px rgba(102, 126, 234, 0.3),
|
||||||
|
0 4px 6px -2px rgba(102, 126, 234, 0.05);
|
||||||
|
transform: translateY(-1px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.card {
|
||||||
|
background: var(--surface-color);
|
||||||
|
border-radius: 16px;
|
||||||
|
border: 1px solid rgba(255, 255, 255, 0.2);
|
||||||
|
box-shadow:
|
||||||
|
0 10px 15px -3px rgba(0, 0, 0, 0.1),
|
||||||
|
0 4px 6px -2px rgba(0, 0, 0, 0.05);
|
||||||
|
overflow: hidden;
|
||||||
|
position: relative;
|
||||||
|
}
|
||||||
|
|
||||||
|
.card::before {
|
||||||
|
content: '';
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
left: 0;
|
||||||
|
right: 0;
|
||||||
|
height: 1px;
|
||||||
|
background: linear-gradient(90deg, transparent, rgba(255, 255, 255, 0.5), transparent);
|
||||||
|
}
|
||||||
|
|
||||||
|
.stat-card {
|
||||||
|
background: linear-gradient(135deg, rgba(255, 255, 255, 0.95) 0%, rgba(255, 255, 255, 0.8) 100%);
|
||||||
|
border-radius: 20px;
|
||||||
|
border: 1px solid rgba(255, 255, 255, 0.3);
|
||||||
|
padding: 24px;
|
||||||
|
position: relative;
|
||||||
|
overflow: hidden;
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.stat-card::before {
|
||||||
|
content: '';
|
||||||
|
position: absolute;
|
||||||
|
top: -50%;
|
||||||
|
left: -50%;
|
||||||
|
width: 200%;
|
||||||
|
height: 200%;
|
||||||
|
background: radial-gradient(circle, rgba(255, 255, 255, 0.1) 0%, transparent 70%);
|
||||||
|
opacity: 0;
|
||||||
|
transition: opacity 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.stat-card:hover {
|
||||||
|
transform: translateY(-4px);
|
||||||
|
box-shadow:
|
||||||
|
0 20px 25px -5px rgba(0, 0, 0, 0.1),
|
||||||
|
0 10px 10px -5px rgba(0, 0, 0, 0.04);
|
||||||
|
}
|
||||||
|
|
||||||
|
.stat-card:hover::before {
|
||||||
|
opacity: 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.stat-icon {
|
||||||
|
width: 56px;
|
||||||
|
height: 56px;
|
||||||
|
border-radius: 16px;
|
||||||
|
display: flex;
|
||||||
|
align-items: center;
|
||||||
|
justify-content: center;
|
||||||
|
font-size: 24px;
|
||||||
|
color: white;
|
||||||
|
box-shadow:
|
||||||
|
0 10px 15px -3px rgba(0, 0, 0, 0.1),
|
||||||
|
0 4px 6px -2px rgba(0, 0, 0, 0.05);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn {
|
||||||
|
font-weight: 500;
|
||||||
|
border-radius: 12px;
|
||||||
|
border: none;
|
||||||
|
cursor: pointer;
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
position: relative;
|
||||||
|
overflow: hidden;
|
||||||
|
letter-spacing: 0.025em;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn::before {
|
||||||
|
content: '';
|
||||||
|
position: absolute;
|
||||||
|
top: 50%;
|
||||||
|
left: 50%;
|
||||||
|
width: 0;
|
||||||
|
height: 0;
|
||||||
|
background: rgba(255, 255, 255, 0.2);
|
||||||
|
border-radius: 50%;
|
||||||
|
transform: translate(-50%, -50%);
|
||||||
|
transition: width 0.3s ease, height 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn:active::before {
|
||||||
|
width: 300px;
|
||||||
|
height: 300px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-primary {
|
||||||
|
background: linear-gradient(135deg, var(--primary-color) 0%, var(--secondary-color) 100%);
|
||||||
|
color: white;
|
||||||
|
box-shadow:
|
||||||
|
0 10px 15px -3px rgba(102, 126, 234, 0.3),
|
||||||
|
0 4px 6px -2px rgba(102, 126, 234, 0.05);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-primary:hover {
|
||||||
|
transform: translateY(-1px);
|
||||||
|
box-shadow:
|
||||||
|
0 20px 25px -5px rgba(102, 126, 234, 0.3),
|
||||||
|
0 10px 10px -5px rgba(102, 126, 234, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-success {
|
||||||
|
background: linear-gradient(135deg, var(--success-color) 0%, #059669 100%);
|
||||||
|
color: white;
|
||||||
|
box-shadow:
|
||||||
|
0 10px 15px -3px rgba(16, 185, 129, 0.3),
|
||||||
|
0 4px 6px -2px rgba(16, 185, 129, 0.05);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-success:hover {
|
||||||
|
transform: translateY(-1px);
|
||||||
|
box-shadow:
|
||||||
|
0 20px 25px -5px rgba(16, 185, 129, 0.3),
|
||||||
|
0 10px 10px -5px rgba(16, 185, 129, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-danger {
|
||||||
|
background: linear-gradient(135deg, var(--error-color) 0%, #dc2626 100%);
|
||||||
|
color: white;
|
||||||
|
box-shadow:
|
||||||
|
0 10px 15px -3px rgba(239, 68, 68, 0.3),
|
||||||
|
0 4px 6px -2px rgba(239, 68, 68, 0.05);
|
||||||
|
}
|
||||||
|
|
||||||
|
.btn-danger:hover {
|
||||||
|
transform: translateY(-1px);
|
||||||
|
box-shadow:
|
||||||
|
0 20px 25px -5px rgba(239, 68, 68, 0.3),
|
||||||
|
0 10px 10px -5px rgba(239, 68, 68, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-input {
|
||||||
|
background: rgba(255, 255, 255, 0.9);
|
||||||
|
border: 2px solid rgba(255, 255, 255, 0.3);
|
||||||
|
border-radius: 12px;
|
||||||
|
padding: 16px;
|
||||||
|
font-size: 16px;
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
backdrop-filter: blur(10px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.form-input:focus {
|
||||||
|
outline: none;
|
||||||
|
border-color: var(--primary-color);
|
||||||
|
box-shadow:
|
||||||
|
0 0 0 3px rgba(102, 126, 234, 0.1),
|
||||||
|
0 10px 15px -3px rgba(0, 0, 0, 0.1);
|
||||||
|
background: rgba(255, 255, 255, 0.95);
|
||||||
|
}
|
||||||
|
|
||||||
|
.table-container {
|
||||||
|
background: rgba(255, 255, 255, 0.95);
|
||||||
|
border-radius: 16px;
|
||||||
|
overflow: hidden;
|
||||||
|
box-shadow:
|
||||||
|
0 10px 15px -3px rgba(0, 0, 0, 0.1),
|
||||||
|
0 4px 6px -2px rgba(0, 0, 0, 0.05);
|
||||||
|
}
|
||||||
|
|
||||||
|
.table-row {
|
||||||
|
transition: all 0.2s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.table-row:hover {
|
||||||
|
background: rgba(102, 126, 234, 0.05);
|
||||||
|
transform: scale(1.005);
|
||||||
|
}
|
||||||
|
|
||||||
|
.modal {
|
||||||
|
backdrop-filter: blur(8px);
|
||||||
|
background: rgba(0, 0, 0, 0.4);
|
||||||
|
}
|
||||||
|
|
||||||
|
.modal-content {
|
||||||
|
background: rgba(255, 255, 255, 0.95);
|
||||||
|
border-radius: 24px;
|
||||||
|
border: 1px solid rgba(255, 255, 255, 0.3);
|
||||||
|
box-shadow:
|
||||||
|
0 25px 50px -12px rgba(0, 0, 0, 0.25),
|
||||||
|
0 0 0 1px rgba(255, 255, 255, 0.05);
|
||||||
|
backdrop-filter: blur(20px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.header-title {
|
||||||
|
background: linear-gradient(135deg, var(--primary-color) 0%, var(--secondary-color) 100%);
|
||||||
|
-webkit-background-clip: text;
|
||||||
|
-webkit-text-fill-color: transparent;
|
||||||
|
background-clip: text;
|
||||||
|
font-weight: 700;
|
||||||
|
letter-spacing: -0.025em;
|
||||||
|
}
|
||||||
|
|
||||||
|
.loading-spinner {
|
||||||
|
display: inline-block;
|
||||||
|
width: 20px;
|
||||||
|
height: 20px;
|
||||||
|
border: 2px solid rgba(255, 255, 255, 0.3);
|
||||||
|
border-radius: 50%;
|
||||||
|
border-top: 2px solid white;
|
||||||
|
animation: spin 1s linear infinite;
|
||||||
|
}
|
||||||
|
|
||||||
|
@keyframes spin {
|
||||||
|
0% { transform: rotate(0deg); }
|
||||||
|
100% { transform: rotate(360deg); }
|
||||||
|
}
|
||||||
|
|
||||||
|
.fade-enter-active, .fade-leave-active {
|
||||||
|
transition: opacity 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.fade-enter-from, .fade-leave-to {
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.slide-up-enter-active, .slide-up-leave-active {
|
||||||
|
transition: all 0.3s ease;
|
||||||
|
}
|
||||||
|
|
||||||
|
.slide-up-enter-from {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateY(30px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.slide-up-leave-to {
|
||||||
|
opacity: 0;
|
||||||
|
transform: translateY(-30px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast {
|
||||||
|
position: fixed;
|
||||||
|
top: 80px;
|
||||||
|
right: 20px;
|
||||||
|
z-index: 1000;
|
||||||
|
min-width: 320px;
|
||||||
|
max-width: 500px;
|
||||||
|
transform: translateX(100%);
|
||||||
|
transition: transform 0.3s ease-in-out;
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast.show {
|
||||||
|
transform: translateX(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast-success {
|
||||||
|
background: linear-gradient(135deg, #10b981 0%, #059669 100%);
|
||||||
|
color: white;
|
||||||
|
border: 1px solid rgba(16, 185, 129, 0.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast-error {
|
||||||
|
background: linear-gradient(135deg, #ef4444 0%, #dc2626 100%);
|
||||||
|
color: white;
|
||||||
|
border: 1px solid rgba(239, 68, 68, 0.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast-info {
|
||||||
|
background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%);
|
||||||
|
color: white;
|
||||||
|
border: 1px solid rgba(59, 130, 246, 0.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
.toast-warning {
|
||||||
|
background: linear-gradient(135deg, #f59e0b 0%, #d97706 100%);
|
||||||
|
color: white;
|
||||||
|
border: 1px solid rgba(245, 158, 11, 0.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
[v-cloak] {
|
||||||
|
display: none;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@media (max-width: 768px) {
|
||||||
|
.glass, .glass-strong {
|
||||||
|
margin: 16px;
|
||||||
|
border-radius: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.stat-card {
|
||||||
|
padding: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tab-btn {
|
||||||
|
font-size: 14px;
|
||||||
|
padding: 12px 8px;
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user