CalvinBackup/CyberStrikeAI
1
0
Fork 0
mirror of https://github.com/Ed1s0nZ/CyberStrikeAI.git synced 2026-07-07 04:58:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
公明
2026-07-06 13:59:13 +08:00
committed by GitHub
parent a63b535a68
commit 38ebdebd14
1 changed files with 42 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/app/app.go
+42 -2
View File
@@ -55,11 +55,15 @@ type App struct {
knowledgeIndexer *knowledge.Indexer // 知识库索引器（用于动态初始化）
knowledgeHandler *handler.KnowledgeHandler // 知识库处理器（用于动态初始化）
agentHandler *handler.AgentHandler // Agent处理器（用于更新知识库管理器）
robotHandler *handler.RobotHandler // 机器人处理器（钉钉/飞书/企业微信）
robotMu sync.Mutex // 保护钉钉/飞书长连接的 cancel
robotHandler *handler.RobotHandler // 机器人处理器（钉钉/飞书/企业微信等）
robotMu sync.Mutex // 保护机器人长连接的 cancel
dingCancel context.CancelFunc // 钉钉 Stream 取消函数，用于配置变更时重启
larkCancel context.CancelFunc // 飞书长连接取消函数，用于配置变更时重启
wechatCancel context.CancelFunc // 微信 iLink 长轮询取消函数
telegramCancel context.CancelFunc // Telegram 长轮询取消函数
slackCancel context.CancelFunc // Slack Socket Mode 取消函数
discordCancel context.CancelFunc // Discord Gateway 取消函数
qqCancel context.CancelFunc // QQ WebSocket 取消函数
c2Manager *c2.Manager // C2 管理器（未启用 C2 时为 nil）
c2Watchdog *c2.SessionWatchdog // C2 会话看门狗
c2WatchdogCancel context.CancelFunc // 看门狗取消函数
@@ -728,6 +732,26 @@ func (a *App) startRobotConnections() {
a.wechatCancel = cancel
go robot.StartWechat(ctx, cfg.Robots, a.robotHandler, cfg.Version, a.logger.Logger)
}
if cfg.Robots.Telegram.Enabled && strings.TrimSpace(cfg.Robots.Telegram.BotToken) != "" {
ctx, cancel := context.WithCancel(context.Background())
a.telegramCancel = cancel
go robot.StartTelegram(ctx, cfg.Robots, a.robotHandler, a.logger.Logger)
}
if cfg.Robots.Slack.Enabled && strings.TrimSpace(cfg.Robots.Slack.BotToken) != "" && strings.TrimSpace(cfg.Robots.Slack.AppToken) != "" {
ctx, cancel := context.WithCancel(context.Background())
a.slackCancel = cancel
go robot.StartSlack(ctx, cfg.Robots, a.robotHandler, a.logger.Logger)
}
if cfg.Robots.Discord.Enabled && strings.TrimSpace(cfg.Robots.Discord.BotToken) != "" {
ctx, cancel := context.WithCancel(context.Background())
a.discordCancel = cancel
go robot.StartDiscord(ctx, cfg.Robots, a.robotHandler, a.logger.Logger)
}
if cfg.Robots.QQ.Enabled && strings.TrimSpace(cfg.Robots.QQ.AppID) != "" && strings.TrimSpace(cfg.Robots.QQ.ClientSecret) != "" {
ctx, cancel := context.WithCancel(context.Background())
a.qqCancel = cancel
go robot.StartQQ(ctx, cfg.Robots, a.robotHandler, a.logger.Logger)
}
}
// RestartRobotConnections 重启钉钉/飞书/微信长连接，使前端应用配置后立即生效（实现 handler.RobotRestarter）
@@ -745,6 +769,22 @@ func (a *App) RestartRobotConnections() {
a.wechatCancel()
a.wechatCancel = nil
}
if a.telegramCancel != nil {
a.telegramCancel()
a.telegramCancel = nil
}
if a.slackCancel != nil {
a.slackCancel()
a.slackCancel = nil
}
if a.discordCancel != nil {
a.discordCancel()
a.discordCancel = nil
}
if a.qqCancel != nil {
a.qqCancel()
a.qqCancel = nil
}
a.robotMu.Unlock()
// 给旧 goroutine 一点时间退出
time.Sleep(200 * time.Millisecond)