diff --git a/frontend/src/pages/AuditTasks.tsx b/frontend/src/pages/AuditTasks.tsx
index 7675cf8..ef1ecef 100644
--- a/frontend/src/pages/AuditTasks.tsx
+++ b/frontend/src/pages/AuditTasks.tsx
@@ -24,16 +24,20 @@ import {
Shield,
Terminal,
Bot,
- Zap
+ Zap,
+ Download
} from "lucide-react";
import { api } from "@/shared/config/database";
+import { apiClient } from "@/shared/api/serverClient";
import type { AuditTask } from "@/shared/types";
import { Link, useNavigate } from "react-router-dom";
import { toast } from "sonner";
import CreateTaskDialog from "@/components/audit/CreateTaskDialog";
import TerminalProgressDialog from "@/components/audit/TerminalProgressDialog";
+import ExportReportDialog from "@/components/reports/ExportReportDialog";
import { calculateTaskProgress } from "@/shared/utils/utils";
-import { getAgentTasks, cancelAgentTask, type AgentTask } from "@/shared/api/agentTasks";
+import { getAgentTasks, cancelAgentTask, getAgentFindings, type AgentTask, type AgentFinding } from "@/shared/api/agentTasks";
+import ReportExportDialog from "@/pages/AgentAudit/components/ReportExportDialog";
// Zombie task detection config
const ZOMBIE_TIMEOUT = 180000; // 3 minutes without progress is potentially stuck
@@ -59,6 +63,14 @@ export default function AuditTasks() {
const [agentTasks, setAgentTasks] = useState
([]);
const [agentLoading, setAgentLoading] = useState(true);
const [cancellingAgentTaskId, setCancellingAgentTaskId] = useState(null);
+ const [exportingTaskId, setExportingTaskId] = useState(null);
+ const [showExportDialog, setShowExportDialog] = useState(false);
+ const [exportTask, setExportTask] = useState(null);
+ const [exportIssues, setExportIssues] = useState([]);
+ // Agent 任务导出对话框状态
+ const [showAgentExportDialog, setShowAgentExportDialog] = useState(false);
+ const [exportAgentTask, setExportAgentTask] = useState(null);
+ const [exportAgentFindings, setExportAgentFindings] = useState([]);
// Zombie task detection: track progress and time for each task
const taskProgressRef = useRef>(new Map());
@@ -201,6 +213,40 @@ export default function AuditTasks() {
}
};
+ // 打开快速扫描任务导出对话框
+ const handleOpenExportDialog = async (task: AuditTask) => {
+ try {
+ setExportingTaskId(task.id);
+ // 获取任务的问题列表
+ const issuesResponse = await apiClient.get(`/tasks/${task.id}/issues`);
+ setExportTask(task);
+ setExportIssues(issuesResponse.data || []);
+ setShowExportDialog(true);
+ } catch (error: any) {
+ console.error('获取问题列表失败:', error);
+ toast.error("获取问题列表失败");
+ } finally {
+ setExportingTaskId(null);
+ }
+ };
+
+ // 打开 Agent 任务导出对话框
+ const handleOpenAgentExportDialog = async (task: AgentTask) => {
+ try {
+ setExportingTaskId(task.id);
+ // 获取任务的 findings 列表
+ const findings = await getAgentFindings(task.id);
+ setExportAgentTask(task);
+ setExportAgentFindings(findings);
+ setShowAgentExportDialog(true);
+ } catch (error: any) {
+ console.error('获取 findings 列表失败:', error);
+ toast.error("获取审计结果失败");
+ } finally {
+ setExportingTaskId(null);
+ }
+ };
+
const loadTasks = async () => {
try {
setLoading(true);
@@ -711,6 +757,17 @@ export default function AuditTasks() {
)}
+ {(task.status === 'completed' || (task.findings_count != null && task.findings_count > 0)) && (
+
+ )}
{/* 任务详情按钮 */}
)}
+ {(task.issues_count > 0 || task.status === 'completed') && (
+
+ )}