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

Add files via upload

Browse Source
This commit is contained in:
公明
2025-11-19 00:18:10 +08:00
committed by GitHub
parent 0ce3364635
commit c33b293013
3 changed files with 43 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/static/css/style.css
+20
View File
@@ -141,6 +141,26 @@ header {
color: #cfe4ff;
}
.attack-chain-btn {
color: #ffe08a;
border-color: rgba(255, 255, 255, 0.3);
background: rgba(255, 255, 255, 0.08);
}
.attack-chain-btn:not(:disabled):hover {
background: rgba(255, 255, 255, 0.18);
border-color: rgba(255, 255, 255, 0.45);
color: #fff5cc;
}
.attack-chain-btn:disabled {
opacity: 0.55;
cursor: not-allowed;
border-color: rgba(255, 255, 255, 0.15);
background: rgba(255, 255, 255, 0.04);
color: rgba(255, 255, 255, 0.6);
}
.settings-btn {
padding: 8px;
min-width: 44px;
web/static/js/app.js
+16 -25
View File
@@ -247,6 +247,7 @@ function initializeChatUI() {
addMessage('assistant', '系统已就绪。请输入您的测试需求，系统将自动执行相应的安全测试。');
}
addAttackChainButton(currentConversationId);
loadActiveTasks(true);
if (activeTaskInterval) {
clearInterval(activeTaskInterval);
@@ -779,6 +780,7 @@ function handleStreamEvent(event, progressElement, progressId,
updateProgressConversation(progressId, event.data.conversationId);
currentConversationId = event.data.conversationId;
updateActiveConversation();
addAttackChainButton(currentConversationId);
loadActiveTasks();
// 立即刷新对话列表，让新对话显示在历史记录中
loadConversations();
@@ -929,6 +931,7 @@ function handleStreamEvent(event, progressElement, progressId,
if (responseData.conversationId) {
currentConversationId = responseData.conversationId;
updateActiveConversation();
addAttackChainButton(currentConversationId);
updateProgressConversation(progressId, responseData.conversationId);
loadActiveTasks();
}
@@ -1031,6 +1034,7 @@ function handleStreamEvent(event, progressElement, progressId,
if (event.data && event.data.conversationId) {
currentConversationId = event.data.conversationId;
updateActiveConversation();
addAttackChainButton(currentConversationId);
updateProgressConversation(progressId, event.data.conversationId);
}
if (progressTaskState.has(progressId)) {
@@ -1572,6 +1576,7 @@ function startNewConversation() {
currentConversationId = null;
document.getElementById('chat-messages').innerHTML = '';
addMessage('assistant', '系统已就绪。请输入您的测试需求，系统将自动执行相应的安全测试。');
addAttackChainButton(null);
updateActiveConversation();
// 刷新对话列表，确保显示最新的历史对话
loadConversations();
@@ -1783,6 +1788,7 @@ async function deleteConversation(conversationId) {
currentConversationId = null;
document.getElementById('chat-messages').innerHTML = '';
addMessage('assistant', '系统已就绪。请输入您的测试需求，系统将自动执行相应的安全测试。');
addAttackChainButton(null);
}
// 刷新对话列表
@@ -3490,34 +3496,19 @@ let isAttackChainLoading = false; // 防止重复加载
// 添加攻击链按钮
function addAttackChainButton(conversationId) {
// 检查是否已存在按钮
let attackChainBtn = document.getElementById('attack-chain-btn');
const attackChainBtn = document.getElementById('attack-chain-btn');
if (!attackChainBtn) {
attackChainBtn = document.createElement('button');
attackChainBtn.id = 'attack-chain-btn';
attackChainBtn.className = 'btn-secondary';
attackChainBtn.style.marginLeft = '10px';
attackChainBtn.innerHTML = '🔗 攻击链';
return;
}
if (conversationId) {
attackChainBtn.disabled = false;
attackChainBtn.title = '查看当前对话的攻击链';
attackChainBtn.onclick = () => showAttackChain(conversationId);
// 在消息区域上方添加按钮容器
const chatMessages = document.getElementById('chat-messages');
if (chatMessages) {
// 检查是否已有按钮容器
let btnContainer = document.getElementById('attack-chain-btn-container');
if (!btnContainer) {
btnContainer = document.createElement('div');
btnContainer.id = 'attack-chain-btn-container';
btnContainer.style.padding = '10px';
btnContainer.style.borderBottom = '1px solid var(--border-color)';
btnContainer.style.background = 'var(--bg-secondary)';
chatMessages.parentNode.insertBefore(btnContainer, chatMessages);
}
btnContainer.innerHTML = '';
btnContainer.appendChild(attackChainBtn);
}
} else {
attackChainBtn.onclick = () => showAttackChain(conversationId);
attackChainBtn.disabled = true;
attackChainBtn.title = '请选择一个对话以查看攻击链';
attackChainBtn.onclick = null;
}
}
web/templates/index.html
+7 -1
View File
@@ -40,7 +40,13 @@
<svg width="20" height="20" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">
<path d="M3 12h4l3 8 4-16 3 8h4" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
</svg>
<span>监控</span>
<span>MCP监控</span>
</button>
<button id="attack-chain-btn" class="attack-chain-btn" title="请选择一个对话以查看攻击链" disabled>
<svg width="20" height="20" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">
<path d="M10.5 13.5l3-3M8 8H5a4 4 0 1 0 0 8h3m8-8h3a4 4 0 0 1 0 8h-3" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
</svg>
<span>攻击链</span>
</button>
<button class="settings-btn" onclick="openSettings()" title="设置">
<svg width="20" height="20" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">