nofx/start.sh at dev · dominic0312/nofx

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

261

262

263

264

265

266

267

268

269

270

271

272

273

274

275

276

277

278

279

280

281

282

283

284

285

286

287

288

289

290

291

292

293

294

295

296

297

298

299

300

301

302

303

304

305

306

307

308

309

310

311

312

313

314

315

316

317

318

319

320

321

322

323

324

325

326

327

328

329

330

331

332

333

334

335

336

337

338

339

340

341

342

343

344

345

346

347

348

349

350

351

352

353

354

355

356

357

358

359

360

361

362

363

364

365

366

367

368

369

370

371

372

373

374

375

376

377

378

379

380

381

382

383

384

385

386

387

388

389

390

391

392

393

394

395

396

397

398

399

400

401

402

403

404

405

406

407

408

409

410

411

412

413

414

415

416

417

418

419

420

421

422

423

424

425

426

427

428

429

430

431

432

433

434

435

436

437

438

439

440

#!/bin/bash

# ═══════════════════════════════════════════════════════════════

# NOFX AI Trading System - Docker Quick Start Script

# Usage: ./start.sh [command]

# ═══════════════════════════════════════════════════════════════

set -e

# ------------------------------------------------------------------------

# Color Definitions

# ------------------------------------------------------------------------

RED='\033[0;31m'

GREEN='\033[0;32m'

YELLOW='\033[1;33m'

BLUE='\033[0;34m'

CYAN='\033[0;36m'

NC='\033[0m' # No Color

# ------------------------------------------------------------------------

# Utility Functions: Colored Output

# ------------------------------------------------------------------------

print_info() {

echo -e "${BLUE}[INFO]${NC} $1"

}

print_success() {

echo -e "${GREEN}[SUCCESS]${NC} $1"

}

print_warning() {

echo -e "${YELLOW}[WARNING]${NC} $1"

}

print_error() {

echo -e "${RED}[ERROR]${NC} $1"

}

# ------------------------------------------------------------------------

# Detection: Docker Compose Command (Backward Compatible)

# ------------------------------------------------------------------------

detect_compose_cmd() {

if command -v docker compose &> /dev/null; then

COMPOSE_CMD="docker compose"

elif command -v docker-compose &> /dev/null; then

COMPOSE_CMD="docker-compose"

else

print_error "Docker Compose 未安装！请先安装 Docker Compose"

exit 1

print_info "使用 Docker Compose 命令: $COMPOSE_CMD"

}

# ------------------------------------------------------------------------

# Validation: Docker Installation

# ------------------------------------------------------------------------

check_docker() {

if ! command -v docker &> /dev/null; then

print_error "Docker 未安装！请先安装 Docker: https://docs.docker.com/get-docker/"

exit 1

detect_compose_cmd

print_success "Docker 和 Docker Compose 已安装"

}

# ------------------------------------------------------------------------

# Validation: Environment File (.env)

# ------------------------------------------------------------------------

check_env() {

if [ ! -f ".env" ]; then

print_warning ".env 不存在，从模板复制..."

cp .env.example .env

print_info "已创建 .env 文件"

print_success "环境变量文件存在"

}

# ------------------------------------------------------------------------

# Helper: Check if env var is set and not placeholder

# ------------------------------------------------------------------------

is_env_configured() {

local var_name="$1"

local value=$(grep "^${var_name}=" .env 2>/dev/null | cut -d'=' -f2-)

# 去除引号

value=$(echo "$value" | tr -d '"'"'")

# 检查是否为空或占位符

if [ -z "$value" ]; then

return 1

# 检查是否是示例值

case "$value" in

return 1

;;

esac

return 0

}

# ------------------------------------------------------------------------

# Helper: Generate and set env var in .env file

# ------------------------------------------------------------------------

set_env_var() {

local var_name="$1"

local var_value="$2"

# 如果变量已存在（即使是占位符），替换它

if grep -q "^${var_name}=" .env 2>/dev/null; then

# macOS 和 Linux 兼容的 sed

if [[ "$OSTYPE" == "darwin"* ]]; then

sed -i '' "s|^${var_name}=.*|${var_name}=${var_value}|" .env

else

sed -i "s|^${var_name}=.*|${var_name}=${var_value}|" .env

else

# 变量不存在，追加

echo "${var_name}=${var_value}" >> .env

}

# ------------------------------------------------------------------------

# Validation: Encryption Keys in .env

# ------------------------------------------------------------------------

check_encryption() {

print_info "检查加密密钥配置..."

local generated=false

# 检查并生成 JWT_SECRET

if ! is_env_configured "JWT_SECRET"; then

print_warning "JWT_SECRET 未配置，正在生成..."

local jwt_secret=$(openssl rand -base64 32)

set_env_var "JWT_SECRET" "$jwt_secret"

print_success "JWT_SECRET 已生成"

generated=true

# 检查并生成 DATA_ENCRYPTION_KEY

if ! is_env_configured "DATA_ENCRYPTION_KEY"; then

print_warning "DATA_ENCRYPTION_KEY 未配置，正在生成..."

local data_key=$(openssl rand -base64 32)

set_env_var "DATA_ENCRYPTION_KEY" "$data_key"

print_success "DATA_ENCRYPTION_KEY 已生成"

generated=true

# 检查并生成 RSA_PRIVATE_KEY

if ! is_env_configured "RSA_PRIVATE_KEY"; then

print_warning "RSA_PRIVATE_KEY 未配置，正在生成..."

# 生成 RSA 密钥并转换为单行格式（\n 替换为 \\n）

local rsa_key=$(openssl genrsa 2048 2>/dev/null | awk '{printf "%s\\n", $0}')

set_env_var "RSA_PRIVATE_KEY" "\"$rsa_key\""

print_success "RSA_PRIVATE_KEY 已生成"

generated=true

if [ "$generated" = true ]; then

echo ""

8000 print_success "所有缺失的密钥已自动生成并保存到 .env"

print_warning "请妥善保管 .env 文件，不要提交到版本控制系统"

echo ""

print_success "加密密钥检查完成"

print_info " • JWT_SECRET: OK"

print_info " • DATA_ENCRYPTION_KEY: OK"

print_info " • RSA_PRIVATE_KEY: OK"

# 修复 .env 文件权限

chmod 600 .env 2>/dev/null || true

}

# ------------------------------------------------------------------------

# Validation: Configuration File (config.json) - BASIC SETTINGS ONLY

# ------------------------------------------------------------------------

check_config() {

if [ ! -f "config.json" ]; then

print_warning "config.json 不存在，从模板复制..."

cp config.json.example config.json

print_info "已使用默认配置创建 config.json"

print_success "配置文件存在"

}

# ------------------------------------------------------------------------

# Utility: Read Environment Variables

# ------------------------------------------------------------------------

read_env_vars() {

if [ -f ".env" ]; then

NOFX_FRONTEND_PORT=$(grep "^NOFX_FRONTEND_PORT=" .env 2>/dev/null | cut -d'=' -f2 || echo "3000")

NOFX_BACKEND_PORT=$(grep "^NOFX_BACKEND_PORT=" .env 2>/dev/null | cut -d'=' -f2 || echo "8080")

NOFX_FRONTEND_PORT=$(echo "$NOFX_FRONTEND_PORT" | tr -d '"'"'" | tr -d ' ')

NOFX_BACKEND_PORT=$(echo "$NOFX_BACKEND_PORT" | tr -d '"'"'" | tr -d ' ')

NOFX_FRONTEND_PORT=${NOFX_FRONTEND_PORT:-3000}

NOFX_BACKEND_PORT=${NOFX_BACKEND_PORT:-8080}

else

NOFX_FRONTEND_PORT=3000

NOFX_BACKEND_PORT=8080

}

# ------------------------------------------------------------------------

# Validation: Database File (data.db)

# ------------------------------------------------------------------------

check_database() {

if [ -d "data.db" ]; then

print_warning "data.db 是目录而非文件，正在删除目录..."

rm -rf data.db

install -m 600 /dev/null data.db

print_success "已创建空数据库文件"

elif [ ! -f "data.db" ]; then

print_warning "数据库文件不存在，创建空数据库文件..."

install -m 600 /dev/null data.db

print_info "已创建空数据库文件，系统将在启动时初始化"

else

print_success "数据库文件存在"

}

# ------------------------------------------------------------------------

# Service Management: Start

# ------------------------------------------------------------------------

start() {

print_info "正在启动 NOFX AI Trading System..."

read_env_vars

if [ ! -f "data.db" ]; then

print_info "创建数据库文件..."

install -m 600 /dev/null data.db

if [ ! -d "decision_logs" ]; then

print_info "创建日志目录..."

install -m 700 -d decision_logs

if [ "$1" == "--build" ]; then

print_info "重新构建镜像..."

$COMPOSE_CMD up -d --build

else

print_info "启动容器..."

$COMPOSE_CMD up -d

print_success "服务已启动！"

print_info "Web 界面: http://localhost:${NOFX_FRONTEND_PORT}"

print_info "API 端点: http://localhost:${NOFX_BACKEND_PORT}"

print_info ""

print_info "查看日志: ./start.sh logs"

print_info "停止服务: ./start.sh stop"

}

# ------------------------------------------------------------------------

# Service Management: Stop

# ------------------------------------------------------------------------

stop() {

print_info "正在停止服务..."

$COMPOSE_CMD stop

print_success "服务已停止"

}

# ------------------------------------------------------------------------

# Service Management: Restart

# ------------------------------------------------------------------------

restart() {

print_info "正在重启服务..."

$COMPOSE_CMD restart

print_success "服务已重启"

}

# ------------------------------------------------------------------------

# Monitoring: Logs

# ------------------------------------------------------------------------

logs() {

if [ -z "$2" ]; then

$COMPOSE_CMD logs -f

else

$COMPOSE_CMD logs -f "$2"

}

# ------------------------------------------------------------------------

# Monitoring: Status

# ------------------------------------------------------------------------

status() {

read_env_vars

print_info "服务状态:"

$COMPOSE_CMD ps

echo ""

print_info "健康检查:"

curl -s "http://localhost:${NOFX_BACKEND_PORT}/api/health" | jq '.' || echo "后端未响应"

}

# ------------------------------------------------------------------------

# Maintenance: Clean (Destructive)

# ------------------------------------------------------------------------

clean() {

print_warning "这将删除所有容器和数据！"

read -p "确认删除？(yes/no): " confirm

if [ "$confirm" == "yes" ]; then

print_info "正在清理..."

$COMPOSE_CMD down -v

print_success "清理完成"

else

print_info "已取消"

}

# ------------------------------------------------------------------------

# Maintenance: Update

# ------------------------------------------------------------------------

update() {

print_info "正在更新..."

git pull

$COMPOSE_CMD up -d --build

print_success "更新完成"

}

# ------------------------------------------------------------------------

# Command: Regenerate all keys (force)

# ------------------------------------------------------------------------

regenerate_keys() {

print_warning "这将重新生成所有加密密钥！"

print_warning "如果已有加密数据，重新生成后将无法解密！"

echo ""

read -p "确认重新生成？(yes/no): " confirm

if [ "$confirm" != "yes" ]; then

print_info "已取消"

return

check_env

print_info "正在生成新的密钥..."

# 生成 JWT_SECRET

local jwt_secret=$(openssl rand -base64 32)

set_env_var "JWT_SECRET" "$jwt_secret"

print_success "JWT_SECRET 已生成"

# 生成 DATA_ENCRYPTION_KEY

local data_key=$(openssl rand -base64 32)

set_env_var "DATA_ENCRYPTION_KEY" "$data_key"

print_success "DATA_ENCRYPTION_KEY 已生成"

# 生成 RSA_PRIVATE_KEY

local rsa_key=$(openssl genrsa 2048 2>/dev/null | awk '{printf "%s\\n", $0}')

set_env_var "RSA_PRIVATE_KEY" "\"$rsa_key\""

print_success "RSA_PRIVATE_KEY 已生成"

chmod 600 .env 2>/dev/null || true

echo ""

print_success "所有密钥已重新生成并保存到 .env"

print_warning "请妥善保管 .env 文件"

}

# ------------------------------------------------------------------------

# Help: Usage Information

# ------------------------------------------------------------------------

show_help() {

echo "NOFX AI Trading System - Docker 管理脚本"

echo ""

echo "用法: ./start.sh [command] [options]"

echo ""

echo "命令:"

echo " start [--build] 启动服务（可选：重新构建）"

echo " stop 停止服务"

echo " restart 重启服务"

echo " logs [service] 查看日志（可选：指定服务名 backend/frontend）"

echo " status 查看服务状态"

echo " clean 清理所有容器和数据"

echo " update 更新代码并重启"

echo " regenerate-keys 重新生成所有加密密钥（慎用）"

echo " help 显示此帮助信息"

echo ""

echo "示例:"

echo " ./start.sh start --build # 构建并启动"

echo " ./start.sh logs backend # 查看后端日志"

echo " ./start.sh status # 查看状态"

echo ""

echo "首次使用:"

echo " 直接运行 ./start.sh 即可，缺失的密钥会自动生成"

}

# ------------------------------------------------------------------------

# Main: Command Dispatcher

# ------------------------------------------------------------------------

main() {

check_docker

case "${1:-start}" in

start)

check_env

check_encryption

check_config

check_database

start "$2"

;;

stop)

stop

;;

restart)

restart

;;

logs)

logs "$@"

;;

status)

status

;;

clean)

clean

;;

update)

update

;;

regenerate-keys)

regenerate_keys

;;

help|--help|-h)

show_help

;;

print_error "未知命令: $1"

show_help

exit 1

;;

esac

}

# Execute Main

main "$@"

FilesExpand file tree

start.sh

Latest commit

History

start.sh

File metadata and controls