哪吒网络安全 (Nezha Cyber)

DeepSeek 驱动的终端 UI 红队辅助工具 —— 哪吒之魔童降世角色人格

项目概述

核心特性

角色人格 AI

默认 Agent「哪吒」—— 魔丸降世，自称"小爷"，口头禅"我是小妖怪，逍遥又自在"
纯中文输出，绝不说英文
按 Ctrl+1/2/3 或 Ctrl+Tab 快速切换 Agent（哪吒 / 代码审计专家 / 威胁情报分析师）

实时流式对话

SSE 流式通信，逐 Token 渲染
AI 思考中动画指示器（"[哪吒] 正在思考..."）
自动滚动到底部，手动上滚查看历史时不自动滚动

文件工具与项目创建（Claude Code 风格）

4 个本地文件操作工具：write_file / read_file / create_directory / list_directory
AI 请求工具调用 → 弹出居中确认对话框 → 用户按 Y 确认 / N 拒绝
路径沙箱保护，禁止越界访问当前目录
read_file 输出自动截断 200 行

对话持久化（SQLite）

基于 rusqlite 内嵌 SQLite，无需系统安装
/save 保存当前对话，/load <id> 加载历史对话
/history 列出所有已保存对话
退出时自动保存（可配置 auto_save 偏好）

命令面板（Ctrl+K）

模糊搜索所有命令：/model /theme /agent /save /load /history /new /close
实时过滤，Enter 执行

Token 用量与账户余额

实时 Token 计数（提示 / 生成 / 总计）
费用计算（基于 DeepSeek 定价 ¥2.0 / ¥8.0 每百万 Token）
侧边栏实时显示账户余额（¥）

终端 UI

分屏布局：左侧可折叠侧边栏 + 右侧聊天区
消息自动宽度换行，中文友好
两套内置主题：赛博暗色 + 日光亮色 (/theme daylight)
多标签页会话管理（Ctrl+N）

技术架构

Elm Architecture

用户按键 / API 事件 / Tick
     │
     ▼
Action (mpsc 通道)  ──►  update()  ──►  修改 App 状态
                               │
                               ▼
                         render() 重绘界面

Model: App 全局状态（标签页、Agent 配置、记忆库、余额、确认对话框）
Update: update() 处理 30+ Action 变体
View: 9 个纯渲染函数

依赖栈

依赖	版本	用途
`ratatui`	0.29	终端 UI 框架
`crossterm`	0.28	终端控制
`tokio`	1	异步运行时
`reqwest`	0.12	HTTP + SSE 流
`rusqlite`	0.31 (bundled)	对话持久化
`serde` + `serde_json` + `serde_yaml`	1 / 1 / 0.9	序列化
`uuid`	1	消息唯一标识
`chrono`	0.4	时间戳
`unicode-width`	0.2	中日韩字符宽度
`dotenvy`	0.15	.env 文件加载

总计 14 个直接依赖，编译为单一二进制文件。

安装指南

前置条件

Rust toolchain（stable，2021 edition+）：curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh
DeepSeek API Token（https://platform.deepseek.com）

从源码构建

git clone https://gitcode.com/ctkqiang_sr/nezha_cyber.git
cd nezha_cyber
echo 'DEEPSEEK_TOKEN=sk-你的token' > .env
cargo build --release
./target/release/nezha_cyber

通过 cargo install 安装

cargo install --git https://gitcode.com/ctkqiang_sr/nezha_cyber.git

配置说明

三种方式设置 API Key

方式	说明
`.env` 文件	项目根目录创建 `.env` → `DEEPSEEK_TOKEN=sk-xxx`
`export`	`export DEEPSEEK_TOKEN=sk-xxx && nezha_cyber`
`config.yaml`	YAML 配置文件中的 `api_key` 字段

优先级：.env → config.yaml → 环境变量。同时支持旧版 DEEPSEEK_API_KEY。

config.yaml（可选）

api_base: 'https://api.deepseek.com'
default_model: 'deepseek-v4-pro'
default_pricing:
  prompt_price_per_m: 2.0
  completion_price_per_m: 8.0

agents.yaml（可选）

agents:
  - name: '自定义Agent'
    description: '功能描述'
    system_prompt: '系统提示词...'
    model: 'deepseek-v4-pro'
    tools: []

使用指南

基本交互

在底部输入区键入消息
按 Enter 发送
AI 回复逐 Token 流式显示
左侧边栏查看 Token 用量、费用和余额

Agent 切换

方式	操作
快捷键	`Ctrl+1` `Ctrl+2` `Ctrl+3` 或 `Ctrl+Tab` 循环
命令面板	`Ctrl+K` → `/agent 哪吒`

文件工具工作流

用户: "帮我写一个 Python FastAPI 项目"
→ 哪吒: 调用 list_directory 查看当前目录
→ 弹出确认对话框 ─ 按 Y 确认
→ 哪吒规划架构 → 调用 create_directory + write_file × N
→ 每次写入弹出确认 ─ 按 Y/N 决定是否落盘
→ 项目创建完成

对话记忆

命令	功能
`/save`	保存当前对话
`/load <id>`	加载历史对话
`/history`	列出所有已保存对话

智能体系统

哪吒（默认）

魔丸降世，性格叛逆又正义，嘴毒心软。自称"小爷"，说话带少年痞气。红队高手。

代码审计专家

源代码安全审查、漏洞分析、安全编码规范。

威胁情报分析师

APT 攻击分析、恶意软件行为、攻击链追踪、IOC 提取。

DeepSeek API 集成

端点: POST https://api.deepseek.com/chat/completions
余额查询: GET https://api.deepseek.com/user/balance
协议: SSE 流式传输
认证: Bearer Token
定价: 输入 ¥2.00 / 百万 Token，输出 ¥8.00 / 百万 Token

快捷键速查

全局

快捷键	功能
`Ctrl+C`	退出
`Ctrl+N`	新建标签页
`Ctrl+K`	命令面板
`Ctrl+B`	折叠侧边栏
`Ctrl+1/2/3`	切换 Agent
`Ctrl+Tab`	下一个 Agent
`Esc`	关闭面板/拒绝工具调用

输入区

快捷键	功能
`Enter`	发送消息
`Backspace`	删除字符
`Up/Down`	滚动消息
`PgUp/PgDn`	滚动 5 行

工具调用确认

按键	操作
`Y`	确认执行（写文件/读文件/建目录）
`N`	拒绝
`Esc`	忽略（同拒绝）

项目结构

nezha_cyber/
├── Cargo.toml
├── .env                    # API Token（不入版本控制）
├── config.yaml             # 应用配置（可选）
├── agents.yaml             # 智能体定义（可选）
├── src/
│   ├── main.rs             # 入口 + 事件循环 + 键盘处理
│   ├── action.rs           # Action 枚举（30+ 变体）
│   ├── app.rs              # App 全局状态 + update()
│   ├── api/
│   │   ├── types.rs        # 数据结构（Message, Usage, Role 等）
│   │   └── deepseek.rs     # DeepSeek SSE 流式客户端
│   ├── agent/
│   │   └── config.rs       # YAML 配置解析 + AppConfig
│   ├── persistence/
│   │   ├── mod.rs
│   │   └── db.rs           # SQLite CRUD（MemoryStore）
│   ├── tools/
│   │   ├── mod.rs
│   │   └── executor.rs     # 文件工具本地执行器 + 路径沙箱
│   └── ui/
│       ├── layout.rs       # 布局计算
│       ├── theme.rs        # 2 套主题
│       └── render.rs       # 9 个渲染函数
└── tests/
    └── integration_test.rs

开发指南

搭建

git clone https://gitcode.com/ctkqiang_sr/nezha_cyber.git
cd nezha_cyber
echo 'DEEPSEEK_TOKEN=sk-xxx' > .env
cargo build && cargo run

运行测试

cargo test   # 107 passed, 0 failed

生产构建

cargo build --release

故障排除

API 返回错误

检查 .env 中的 Token 是否有效且以 sk- 开头
验证端点连通：curl -H "Authorization: Bearer $DEEPSEEK_TOKEN" https://api.deepseek.com/user/balance
确认 API Base 为 https://api.deepseek.com（无 /v1）

余额查询失败

未设置 Token 时显示"检查中…"，设好 Token 后重启
Token 过期/无效时侧边栏显示"余额不足"

终端显示异常

调整窗口大小自动重排
极端情况按 Ctrl+C 退出重进

路线图

已完成 (v0.1.0)

短期 (v0.2.0)

多行输入支持（Ctrl+J 换行）
@ 文件附件自动补全
/compact /fork /export 命令执行引擎
自定义工具插件系统

中期 (v0.3.0)

会话导出为 Markdown / JSON / HTML
鼠标支持：侧边栏点击，滚轮滚动
自定义主题 YAML 配置

长期 (v1.0.0)

gRPC / WebSocket 替代传输
终端录屏与审计回放
跨平台 CI/CD 自动化发布

贡献指南

Fork 仓库 → 从 main 创建分支
遵循项目代码规范（见 .trae/rules/project.md）
确保 cargo build 零 warning，cargo test 全绿
使用 Conventional Commits 格式提交
提交 Pull Request

许可证

MIT License — 详见 LICENSE

Name		Name	Last commit message	Last commit date
.gitcode		.gitcode
.github		.github
.vscode		.vscode
docs		docs
src		src
tests		tests
.dockerignore		.dockerignore
.gitignore		.gitignore
Cargo.lock		Cargo.lock
Cargo.toml		Cargo.toml
Dockerfile		Dockerfile
README.md		README.md
docker-compose.yml		docker-compose.yml

Latest commit

History

Folders and files

Repository files navigation