CodeReview/frontend/src/pages/AgentAudit/components/ReportExportDialog.tsx

850 lines
27 KiB
TypeScript
Raw Normal View History

/**
* Report Export Dialog Component
* Full-featured report export with preview and multi-format support
* Cassette futurism aesthetic
*/
import { useState, useEffect, useCallback, memo } from "react";
import { marked } from "marked";
import {
Dialog,
DialogContent,
DialogHeader,
DialogTitle,
} from "@/components/ui/dialog";
import { Tabs, TabsList, TabsTrigger, TabsContent } from "@/components/ui/tabs";
import { Button } from "@/components/ui/button";
import { Badge } from "@/components/ui/badge";
import { ScrollArea } from "@/components/ui/scroll-area";
import {
FileText,
FileJson,
FileCode,
Download,
Loader2,
Copy,
Check,
AlertTriangle,
RefreshCw,
Eye,
Terminal,
Shield,
Bug,
CheckCircle2,
} from "lucide-react";
import { apiClient } from "@/shared/api/serverClient";
import { downloadAgentReport } from "@/shared/api/agentTasks";
import type { AgentTask, AgentFinding } from "@/shared/api/agentTasks";
// ============ Types ============
type ReportFormat = "markdown" | "json" | "html";
interface ReportExportDialogProps {
open: boolean;
onOpenChange: (open: boolean) => void;
task: AgentTask | null;
findings: AgentFinding[];
}
interface ReportPreview {
content: string;
format: ReportFormat;
loading: boolean;
error: string | null;
}
// ============ Constants ============
const FORMAT_CONFIG: Record<ReportFormat, {
label: string;
icon: React.ReactNode;
extension: string;
mime: string;
}> = {
markdown: {
label: "Markdown",
icon: <FileText className="w-4 h-4" />,
extension: ".md",
mime: "text/markdown",
},
json: {
label: "JSON",
icon: <FileJson className="w-4 h-4" />,
extension: ".json",
mime: "application/json",
},
html: {
label: "HTML",
icon: <FileCode className="w-4 h-4" />,
extension: ".html",
mime: "text/html",
},
};
// ============ Helper Functions ============
function getSeverityColor(severity: string): string {
const colors: Record<string, string> = {
critical: "text-rose-400",
high: "text-orange-400",
medium: "text-amber-400",
low: "text-sky-400",
info: "text-slate-400",
};
return colors[severity.toLowerCase()] || colors.info;
}
function formatBytes(bytes: number): string {
if (bytes === 0) return "0 B";
const k = 1024;
const sizes = ["B", "KB", "MB"];
const i = Math.floor(Math.log(bytes) / Math.log(k));
return `${parseFloat((bytes / Math.pow(k, i)).toFixed(1))} ${sizes[i]}`;
}
// ============ Sub Components ============
// Report stats summary - uses task statistics for reliability
const ReportStats = memo(function ReportStats({
task,
}: {
task: AgentTask;
findings: AgentFinding[]; // Keep for API compatibility, but use task stats
}) {
// Use task's reliable statistics instead of computing from findings array
// This ensures consistency even when findings array is empty or not loaded
const totalFindings = task.findings_count || 0;
const criticalAndHigh = (task.critical_count || 0) + (task.high_count || 0);
const verified = task.verified_count || 0;
return (
<div className="grid grid-cols-4 gap-2 mb-4">
<div className="p-2.5 rounded bg-slate-900/60 border border-slate-700/40">
<div className="flex items-center gap-1.5 mb-1">
<Shield className="w-3 h-3 text-emerald-400" />
<span className="text-[9px] text-slate-500 uppercase tracking-wider">Score</span>
</div>
<div className="text-lg font-bold font-mono text-emerald-400">
{task.security_score?.toFixed(0) || "N/A"}
</div>
</div>
<div className="p-2.5 rounded bg-slate-900/60 border border-slate-700/40">
<div className="flex items-center gap-1.5 mb-1">
<Bug className="w-3 h-3 text-rose-400" />
<span className="text-[9px] text-slate-500 uppercase tracking-wider">Findings</span>
</div>
<div className="text-lg font-bold font-mono text-white">
{totalFindings}
</div>
</div>
<div className="p-2.5 rounded bg-slate-900/60 border border-slate-700/40">
<div className="flex items-center gap-1.5 mb-1">
<AlertTriangle className="w-3 h-3 text-orange-400" />
<span className="text-[9px] text-slate-500 uppercase tracking-wider">Critical</span>
</div>
<div className="text-lg font-bold font-mono text-rose-400">
{criticalAndHigh}
</div>
</div>
<div className="p-2.5 rounded bg-slate-900/60 border border-slate-700/40">
<div className="flex items-center gap-1.5 mb-1">
<CheckCircle2 className="w-3 h-3 text-teal-400" />
<span className="text-[9px] text-slate-500 uppercase tracking-wider">Verified</span>
</div>
<div className="text-lg font-bold font-mono text-teal-400">
{verified}
</div>
</div>
</div>
);
});
// Markdown preview renderer
const MarkdownPreview = memo(function MarkdownPreview({
content,
}: {
content: string;
}) {
// Simple markdown to styled elements renderer
const renderMarkdown = (text: string) => {
const lines = text.split("\n");
const elements: React.ReactNode[] = [];
let inCodeBlock = false;
let codeContent: string[] = [];
let codeLanguage = "";
lines.forEach((line, index) => {
// Code block handling
if (line.startsWith("```")) {
if (inCodeBlock) {
elements.push(
<div key={`code-${index}`} className="my-3 rounded bg-black/50 border border-slate-700/50 overflow-hidden">
<div className="flex items-center justify-between px-3 py-1.5 bg-slate-800/50 border-b border-slate-700/50">
<span className="text-[10px] text-slate-500 uppercase tracking-wider font-mono">
{codeLanguage || "code"}
</span>
</div>
<pre className="p-3 text-xs font-mono text-slate-300 overflow-x-auto">
{codeContent.join("\n")}
</pre>
</div>
);
codeContent = [];
codeLanguage = "";
inCodeBlock = false;
} else {
inCodeBlock = true;
codeLanguage = line.slice(3).trim();
}
return;
}
if (inCodeBlock) {
codeContent.push(line);
return;
}
// Headers
if (line.startsWith("# ")) {
elements.push(
<h1 key={index} className="text-xl font-bold text-white mt-6 mb-3 pb-2 border-b border-slate-700/50">
{line.slice(2)}
</h1>
);
return;
}
if (line.startsWith("## ")) {
elements.push(
<h2 key={index} className="text-lg font-bold text-white mt-5 mb-2">
{line.slice(3)}
</h2>
);
return;
}
if (line.startsWith("### ")) {
elements.push(
<h3 key={index} className="text-base font-semibold text-slate-200 mt-4 mb-2">
{line.slice(4)}
</h3>
);
return;
}
// Horizontal rule
if (line.match(/^---+$/)) {
elements.push(
<hr key={index} className="my-4 border-slate-700/50" />
);
return;
}
// List items
if (line.match(/^[-*]\s/)) {
elements.push(
<div key={index} className="flex gap-2 text-sm text-slate-300 ml-2 my-0.5">
<span className="text-primary"></span>
<span>{line.slice(2)}</span>
</div>
);
return;
}
// Bold text handling
let processedLine = line;
if (line.includes("**")) {
const parts = line.split(/\*\*(.+?)\*\*/g);
const lineElements = parts.map((part, i) => {
if (i % 2 === 1) {
return <strong key={i} className="text-white font-semibold">{part}</strong>;
}
return part;
});
elements.push(
<p key={index} className="text-sm text-slate-300 my-1">
{lineElements}
</p>
);
return;
}
// Empty lines
if (line.trim() === "") {
elements.push(<div key={index} className="h-2" />);
return;
}
// Regular paragraphs
elements.push(
<p key={index} className="text-sm text-slate-300 my-1">
{processedLine}
</p>
);
});
return elements;
};
return (
<div className="prose prose-invert max-w-none">
{renderMarkdown(content)}
</div>
);
});
// JSON preview with syntax highlighting
const JsonPreview = memo(function JsonPreview({
content,
}: {
content: string;
}) {
const highlightJson = (json: string) => {
try {
const parsed = JSON.parse(json);
const formatted = JSON.stringify(parsed, null, 2);
return formatted
.replace(/"([^"]+)":/g, '<span class="text-violet-400">"$1"</span>:')
.replace(/: "([^"]+)"/g, ': <span class="text-emerald-400">"$1"</span>')
.replace(/: (\d+)/g, ': <span class="text-amber-400">$1</span>')
.replace(/: (true|false)/g, ': <span class="text-sky-400">$1</span>')
.replace(/: (null)/g, ': <span class="text-slate-500">$1</span>');
} catch {
return json;
}
};
return (
<pre
className="text-xs font-mono text-slate-300 whitespace-pre-wrap"
dangerouslySetInnerHTML={{ __html: highlightJson(content) }}
/>
);
});
// ============ Main Component ============
export const ReportExportDialog = memo(function ReportExportDialog({
open,
onOpenChange,
task,
findings,
}: ReportExportDialogProps) {
const [activeFormat, setActiveFormat] = useState<ReportFormat>("markdown");
const [preview, setPreview] = useState<ReportPreview>({
content: "",
format: "markdown",
loading: false,
error: null,
});
const [copied, setCopied] = useState(false);
const [downloading, setDownloading] = useState(false);
// Fetch report content for preview
const fetchPreview = useCallback(async (format: ReportFormat) => {
if (!task) return;
setPreview(prev => ({ ...prev, loading: true, error: null }));
try {
// For JSON, fetch from backend API to ensure data consistency
// The backend properly queries findings from the database
if (format === "json") {
const response = await apiClient.get(`/agent-tasks/${task.id}/report`, {
params: { format: "json" },
});
setPreview({
content: JSON.stringify(response.data, null, 2),
format: "json",
loading: false,
error: null,
});
return;
}
// For HTML, generate from markdown
if (format === "html") {
const mdResponse = await apiClient.get(`/agent-tasks/${task.id}/report`, {
params: { format: "markdown" },
responseType: "text",
});
const htmlContent = await generateHtmlReport(mdResponse.data, task);
setPreview({
content: htmlContent,
format: "html",
loading: false,
error: null,
});
return;
}
// For Markdown, fetch from server
const response = await apiClient.get(`/agent-tasks/${task.id}/report`, {
params: { format: "markdown" },
responseType: "text",
});
setPreview({
content: response.data,
format: "markdown",
loading: false,
error: null,
});
} catch (err) {
console.error("Failed to fetch report preview:", err);
setPreview(prev => ({
...prev,
loading: false,
error: "Failed to load report preview",
}));
}
}, [task, findings]);
// Generate HTML report from markdown
const generateHtmlReport = async (markdown: string, task: AgentTask): Promise<string> => {
// Parse markdown to HTML using marked
const contentHtml = await marked.parse(markdown);
return `<!DOCTYPE html>
<html lang="zh-CN">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title> - ${task.name || task.id}</title>
<style>
:root {
--bg-primary: #0a0a0f;
--bg-secondary: #0d0d12;
--bg-tertiary: #1a1a24;
--text-primary: #ffffff;
--text-secondary: #94a3b8;
--accent: #FF6B2C;
--border: #2d2d3d;
--success: #34d399;
--warning: #fbbf24;
--error: #fb7185;
--code-bg: #1e1e2e;
}
* { box-sizing: border-box; margin: 0; padding: 0; }
body {
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, 'Helvetica Neue', Arial, sans-serif;
background: var(--bg-primary);
color: var(--text-secondary);
line-height: 1.6;
padding: 2rem;
max-width: 900px;
margin: 0 auto;
}
/* Typography */
h1, h2, h3, h4, h5, h6 {
color: var(--text-primary);
margin-top: 2rem;
margin-bottom: 1rem;
font-weight: 600;
}
h1 { font-size: 2rem; border-bottom: 2px solid var(--accent); padding-bottom: 0.5rem; }
h2 { font-size: 1.5rem; border-bottom: 1px solid var(--border); padding-bottom: 0.5rem; }
h3 { font-size: 1.25rem; color: var(--text-primary); margin-top: 1.5rem; }
p { margin-bottom: 1rem; }
/* Tables */
table {
width: 100%;
border-collapse: collapse;
margin: 1.5rem 0;
background: var(--bg-secondary);
border-radius: 8px;
overflow: hidden;
border: 1px solid var(--border);
}
th, td {
padding: 0.75rem 1rem;
text-align: left;
border-bottom: 1px solid var(--border);
}
th {
background: var(--bg-tertiary);
color: var(--text-primary);
font-weight: 600;
text-transform: uppercase;
font-size: 0.85rem;
letter-spacing: 0.05em;
}
tr:last-child td { border-bottom: none; }
tr:hover td { background: rgba(255, 255, 255, 0.02); }
/* Code Blocks */
pre {
background: var(--code-bg);
padding: 1rem;
border-radius: 8px;
overflow-x: auto;
margin: 1rem 0;
border: 1px solid var(--border);
}
code {
font-family: 'SF Mono', 'Menlo', 'Monaco', 'Courier New', monospace;
font-size: 0.9em;
color: #e2e8f0;
}
p code, li code, td code {
background: rgba(255, 255, 255, 0.1);
padding: 0.2em 0.4em;
border-radius: 4px;
font-size: 0.85em;
color: #e2e8f0;
}
/* Lists */
ul, ol { margin-left: 1.5rem; margin-bottom: 1rem; }
li { margin-bottom: 0.5rem; }
/* Blockquotes */
blockquote {
border-left: 4px solid var(--accent);
padding-left: 1rem;
margin: 1rem 0;
color: var(--text-secondary);
font-style: italic;
background: rgba(255, 107, 44, 0.05);
padding: 0.5rem 0.5rem 0.5rem 1rem;
border-radius: 0 4px 4px 0;
}
/* Links */
a { color: var(--accent); text-decoration: none; }
a:hover { text-decoration: underline; }
/* Horizontal Rule */
hr {
border: 0;
height: 1px;
background: var(--border);
margin: 2rem 0;
}
/* Utilities */
strong { color: var(--text-primary); font-weight: 600; }
em { color: var(--text-secondary); }
/* Print Styles */
@media print {
body {
background: white;
color: black;
max-width: 100%;
padding: 0;
}
h1, h2, h3, strong { color: black; }
pre, code { background: #f1f5f9; color: black; border-color: #e2e8f0; }
table { border-color: #e2e8f0; }
th { background: #f8fafc; color: black; border-bottom-color: #cbd5e1; }
td { border-bottom-color: #e2e8f0; }
p code, li code, td code { background: #f1f5f9; color: #0f172a; }
a { color: #2563eb; text-decoration: underline; }
}
</style>
</head>
<body>
${contentHtml}
<footer style="margin-top: 4rem; padding-top: 2rem; border-top: 1px solid var(--border); text-align: center; font-size: 0.875rem; color: var(--text-secondary);">
<p> ${new Date().toLocaleString('zh-CN')} &bull; DeepAudit </p>
</footer>
</body>
</html>`;
};
// Load preview when format changes or dialog opens
useEffect(() => {
if (open && task) {
fetchPreview(activeFormat);
}
}, [open, activeFormat, task, fetchPreview]);
// Handle copy to clipboard
const handleCopy = async () => {
try {
await navigator.clipboard.writeText(preview.content);
setCopied(true);
setTimeout(() => setCopied(false), 2000);
} catch (err) {
console.error("Failed to copy:", err);
}
};
// Handle download
const handleDownload = async () => {
if (!task || !findings && activeFormat !== "json") return; // Allow json even if findings empty? Actually backend handles it.
setDownloading(true);
try {
let content = "";
let filename = `audit_report_${task.id.substring(0, 8)}`;
let mimeType = "text/plain";
if (activeFormat === "markdown") {
content = preview.content; // Use cached preview
// Fallback if preview failed? Re-fetch.
if (!content) {
const response = await apiClient.get(`/agent-tasks/${task.id}/report`, {
params: { format: "markdown" },
responseType: "text",
});
content = response.data;
}
filename += ".md";
} else if (activeFormat === "json") {
// Fetch JSON directly from backend
const response = await apiClient.get(`/agent-tasks/${task.id}/report`, {
params: { format: "json" },
responseType: "json", // Important: axios parses JSON
});
// Pretty print JSON
content = JSON.stringify(response.data, null, 2);
filename += ".json";
mimeType = "application/json";
} else if (activeFormat === "html") { // HTML
if (preview.content && preview.format === "html") {
content = preview.content;
} else {
const response = await apiClient.get(`/agent-tasks/${task.id}/report`, {
params: { format: "markdown" },
responseType: "text",
});
content = await generateHtmlReport(response.data, task);
}
filename += ".html";
mimeType = "text/html";
}
// Create download trigger
const blob = new Blob([content], { type: mimeType });
const url = URL.createObjectURL(blob);
const a = document.createElement("a");
a.href = url;
a.download = filename;
document.body.appendChild(a);
a.click();
document.body.removeChild(a);
URL.revokeObjectURL(url);
onOpenChange(false);
} catch (err) {
console.error("Download failed:", err);
// Ideally show toast error here
} finally {
setDownloading(false);
}
};
if (!task) return null;
return (
<Dialog open={open} onOpenChange={onOpenChange}>
<DialogContent className="max-w-4xl h-[85vh] bg-[#0a0a0f] border-slate-700/50 p-0 gap-0 overflow-hidden">
{/* Header */}
<div className="px-5 py-4 border-b border-slate-700/50 bg-gradient-to-r from-[#0d0d12] to-[#0a0a0f]">
<DialogHeader>
<div className="flex items-center gap-3">
<div className="p-2 rounded bg-primary/10 border border-primary/30">
<FileText className="w-5 h-5 text-primary" />
</div>
<div>
<DialogTitle className="text-lg font-bold text-white">
Export Audit Report
</DialogTitle>
<p className="text-xs text-slate-500 mt-0.5 font-mono">
Task: {task.name || task.id.slice(0, 8)}
</p>
</div>
</div>
</DialogHeader>
</div>
{/* Stats Summary */}
<div className="px-5 py-3 border-b border-slate-700/40 bg-[#0b0b10]">
<ReportStats task={task} findings={findings} />
</div>
{/* Format Tabs & Content */}
<Tabs
value={activeFormat}
onValueChange={(v) => setActiveFormat(v as ReportFormat)}
className="flex flex-col flex-1 overflow-hidden"
>
{/* Tab List */}
<div className="px-5 py-2 border-b border-slate-700/40 bg-[#0d0d12]">
<div className="flex items-center justify-between">
<TabsList className="bg-slate-800/50 p-1 h-9">
{(Object.keys(FORMAT_CONFIG) as ReportFormat[]).map((format) => {
const config = FORMAT_CONFIG[format];
return (
<TabsTrigger
key={format}
value={format}
className="px-3 py-1.5 text-xs font-mono uppercase tracking-wider data-[state=active]:bg-primary/20 data-[state=active]:text-primary transition-colors"
>
<span className="flex items-center gap-1.5">
{config.icon}
{config.label}
</span>
</TabsTrigger>
);
})}
</TabsList>
{/* Actions */}
<div className="flex items-center gap-2">
<Button
variant="ghost"
size="sm"
onClick={handleCopy}
disabled={preview.loading || !preview.content}
className="h-8 px-2.5 text-xs text-slate-400 hover:text-white"
>
{copied ? (
<Check className="w-3.5 h-3.5 mr-1.5 text-emerald-400" />
) : (
<Copy className="w-3.5 h-3.5 mr-1.5" />
)}
{copied ? "Copied" : "Copy"}
</Button>
<Button
variant="ghost"
size="sm"
onClick={() => fetchPreview(activeFormat)}
disabled={preview.loading}
className="h-8 px-2.5 text-xs text-slate-400 hover:text-white"
>
<RefreshCw className={`w-3.5 h-3.5 mr-1.5 ${preview.loading ? 'animate-spin' : ''}`} />
Refresh
</Button>
</div>
</div>
</div>
{/* Preview Content */}
<div className="flex-1 overflow-hidden">
{(Object.keys(FORMAT_CONFIG) as ReportFormat[]).map((format) => (
<TabsContent
key={format}
value={format}
className="h-full m-0 data-[state=inactive]:hidden"
>
<ScrollArea className="h-full">
<div className="p-5">
{preview.loading ? (
<div className="flex items-center justify-center py-12">
<div className="flex flex-col items-center gap-3">
<Loader2 className="w-8 h-8 text-primary animate-spin" />
<span className="text-sm text-slate-500">Loading preview...</span>
</div>
</div>
) : preview.error ? (
<div className="flex items-center justify-center py-12">
<div className="flex flex-col items-center gap-3 text-center">
<AlertTriangle className="w-8 h-8 text-amber-400" />
<span className="text-sm text-slate-400">{preview.error}</span>
<Button
variant="outline"
size="sm"
onClick={() => fetchPreview(activeFormat)}
className="mt-2"
>
<RefreshCw className="w-3.5 h-3.5 mr-1.5" />
Retry
</Button>
</div>
</div>
) : (
<div className="bg-[#0d0d12] rounded-lg border border-slate-700/40 overflow-hidden">
{/* Preview header */}
<div className="flex items-center justify-between px-4 py-2 bg-slate-800/30 border-b border-slate-700/40">
<div className="flex items-center gap-2">
<Eye className="w-3.5 h-3.5 text-slate-500" />
<span className="text-[10px] text-slate-500 uppercase tracking-wider font-mono">
Preview
</span>
</div>
<Badge className="text-[9px] bg-slate-700/50 text-slate-400 border-0">
{formatBytes(preview.content.length)}
</Badge>
</div>
{/* Preview content */}
<div className="p-4 max-h-[45vh] overflow-y-auto custom-scrollbar">
{format === "markdown" && <MarkdownPreview content={preview.content} />}
{format === "json" && <JsonPreview content={preview.content} />}
{format === "html" && (
<div className="text-xs font-mono text-slate-400 whitespace-pre-wrap">
{preview.content.slice(0, 3000)}
{preview.content.length > 3000 && (
<span className="text-slate-600">... (truncated for preview)</span>
)}
</div>
)}
</div>
</div>
)}
</div>
</ScrollArea>
</TabsContent>
))}
</div>
</Tabs>
{/* Footer */}
<div className="px-5 py-4 border-t border-slate-700/50 bg-[#0d0d12]">
<div className="flex items-center justify-between">
<div className="flex items-center gap-2 text-xs text-slate-500">
<Terminal className="w-3.5 h-3.5" />
<span className="font-mono">
Export as {FORMAT_CONFIG[activeFormat].label} ({FORMAT_CONFIG[activeFormat].extension})
</span>
</div>
<div className="flex items-center gap-3">
<Button
variant="ghost"
onClick={() => onOpenChange(false)}
className="h-9 px-4 text-sm text-slate-400 hover:text-white"
>
Cancel
</Button>
<Button
onClick={handleDownload}
disabled={downloading || preview.loading || !preview.content}
className="h-9 px-5 bg-primary hover:bg-primary/90 text-sm font-medium"
>
{downloading ? (
<>
<Loader2 className="w-4 h-4 mr-2 animate-spin" />
Exporting...
</>
) : (
<>
<Download className="w-4 h-4 mr-2" />
Download {FORMAT_CONFIG[activeFormat].label}
</>
)}
</Button>
</div>
</div>
</div>
</DialogContent>
</Dialog>
);
});
export default ReportExportDialog;