From 9254542f3ddccdc95bea193ccccabc9bd4031d14 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=85=AC=E6=98=8E?=
 <83812544+Ed1s0nZ@users.noreply.github.com>
Date: Tue, 25 Nov 2025 21:28:12 +0800
Subject: [PATCH] Update run.sh

---
 run.sh | 45 +++++++++++++++++++++++++++++++++++----------
 1 file changed, 35 insertions(+), 10 deletions(-)

diff --git a/run.sh b/run.sh
index a752063c..c2d77608 100644
--- a/run.sh
+++ b/run.sh
@@ -1,35 +1,60 @@
 #!/bin/bash
 
+set -euo pipefail
+
 # CyberStrikeAI 启动脚本
+ROOT_DIR="$(cd "$(dirname "$0")" && pwd)"
+cd "$ROOT_DIR"
 
 echo "🚀 启动 CyberStrikeAI..."
 
+CONFIG_FILE="$ROOT_DIR/config.yaml"
+VENV_DIR="$ROOT_DIR/venv"
+REQUIREMENTS_FILE="$ROOT_DIR/requirements.txt"
+
 # 检查配置文件
-if [ ! -f "config.yaml" ]; then
+if [ ! -f "$CONFIG_FILE" ]; then
     echo "❌ 配置文件 config.yaml 不存在"
     exit 1
 fi
 
-# 检查Go环境
-if ! command -v go &> /dev/null; then
+# 检查 Python 环境
+if ! command -v python3 >/dev/null 2>&1; then
+    echo "❌ 未找到 python3，请先安装 Python 3.10+"
+    exit 1
+fi
+
+# 创建并激活虚拟环境
+if [ ! -d "$VENV_DIR" ]; then
+    echo "🐍 创建 Python 虚拟环境..."
+    python3 -m venv "$VENV_DIR"
+fi
+
+echo "🐍 激活虚拟环境..."
+# shellcheck disable=SC1091
+source "$VENV_DIR/bin/activate"
+
+if [ -f "$REQUIREMENTS_FILE" ]; then
+    echo "📦 安装/更新 Python 依赖..."
+    pip install -r "$REQUIREMENTS_FILE"
+else
+    echo "⚠️ 未找到 requirements.txt，跳过 Python 依赖安装"
+fi
+
+# 检查 Go 环境
+if ! command -v go >/dev/null 2>&1; then
     echo "❌ Go 未安装，请先安装 Go 1.21 或更高版本"
     exit 1
 fi
 
 # 下载依赖
-echo "📦 下载依赖..."
+echo "📦 下载 Go 依赖..."
 go mod download
 
 # 构建项目
 echo "🔨 构建项目..."
 go build -o cyberstrike-ai cmd/server/main.go
 
-if [ $? -ne 0 ]; then
-    echo "❌ 构建失败"
-    exit 1
-fi
-
 # 运行服务器
 echo "✅ 启动服务器..."
 ./cyberstrike-ai
-