feat(phase-27): add advanced ATT&CK Navigator-style heatmap with layers, filters and export (T-221 to T-223)
This commit is contained in:
@@ -2,6 +2,7 @@ import { Routes, Route, Navigate } from "react-router-dom";
|
||||
import LoginPage from "./pages/LoginPage";
|
||||
import DashboardPage from "./pages/DashboardPage";
|
||||
import TechniquesPage from "./pages/TechniquesPage";
|
||||
import MatrixPage from "./pages/MatrixPage";
|
||||
import TechniqueDetailPage from "./pages/TechniqueDetailPage";
|
||||
import TestsPage from "./pages/TestsPage";
|
||||
import TestCreatePage from "./pages/TestCreatePage";
|
||||
@@ -35,6 +36,7 @@ export default function App() {
|
||||
>
|
||||
<Route path="/dashboard" element={<DashboardPage />} />
|
||||
<Route path="/techniques" element={<TechniquesPage />} />
|
||||
<Route path="/matrix" element={<MatrixPage />} />
|
||||
<Route path="/techniques/:mitreId" element={<TechniqueDetailPage />} />
|
||||
<Route path="/tests" element={<TestsPage />} />
|
||||
<Route path="/tests/new" element={<TestCreatePage />} />
|
||||
|
||||
98
frontend/src/api/heatmap.ts
Normal file
98
frontend/src/api/heatmap.ts
Normal file
@@ -0,0 +1,98 @@
|
||||
import client from "./client";
|
||||
|
||||
// ── Types ────────────────────────────────────────────────────────────
|
||||
|
||||
export interface HeatmapMetadata {
|
||||
name: string;
|
||||
value: string;
|
||||
}
|
||||
|
||||
export interface HeatmapTechnique {
|
||||
techniqueID: string;
|
||||
tactic: string;
|
||||
color: string;
|
||||
score: number;
|
||||
comment: string;
|
||||
enabled: boolean;
|
||||
metadata: HeatmapMetadata[];
|
||||
}
|
||||
|
||||
export interface HeatmapLayer {
|
||||
name: string;
|
||||
versions: {
|
||||
attack: string;
|
||||
navigator: string;
|
||||
layer: string;
|
||||
};
|
||||
domain: string;
|
||||
description: string;
|
||||
filters: {
|
||||
platforms: string[];
|
||||
};
|
||||
gradient: {
|
||||
colors: string[];
|
||||
minValue: number;
|
||||
maxValue: number;
|
||||
};
|
||||
techniques: HeatmapTechnique[];
|
||||
}
|
||||
|
||||
export interface HeatmapFilters {
|
||||
platforms?: string;
|
||||
tactics?: string;
|
||||
min_score?: number;
|
||||
}
|
||||
|
||||
// ── API Functions ────────────────────────────────────────────────────
|
||||
|
||||
/** Fetch the coverage heatmap layer. */
|
||||
export async function getHeatmapCoverage(filters?: HeatmapFilters): Promise<HeatmapLayer> {
|
||||
const { data } = await client.get<HeatmapLayer>("/heatmap/coverage", { params: filters });
|
||||
return data;
|
||||
}
|
||||
|
||||
/** Fetch the threat actor heatmap layer. */
|
||||
export async function getHeatmapThreatActor(
|
||||
actorId: string,
|
||||
filters?: HeatmapFilters,
|
||||
): Promise<HeatmapLayer> {
|
||||
const { data } = await client.get<HeatmapLayer>(`/heatmap/threat-actor/${actorId}`, {
|
||||
params: filters,
|
||||
});
|
||||
return data;
|
||||
}
|
||||
|
||||
/** Fetch the detection rules heatmap layer. */
|
||||
export async function getHeatmapDetectionRules(filters?: HeatmapFilters): Promise<HeatmapLayer> {
|
||||
const { data } = await client.get<HeatmapLayer>("/heatmap/detection-rules", { params: filters });
|
||||
return data;
|
||||
}
|
||||
|
||||
/** Fetch the campaign heatmap layer. */
|
||||
export async function getHeatmapCampaign(
|
||||
campaignId: string,
|
||||
filters?: HeatmapFilters,
|
||||
): Promise<HeatmapLayer> {
|
||||
const { data } = await client.get<HeatmapLayer>(`/heatmap/campaign/${campaignId}`, {
|
||||
params: filters,
|
||||
});
|
||||
return data;
|
||||
}
|
||||
|
||||
/** Export a heatmap layer as a Navigator JSON file (returns blob URL). */
|
||||
export async function exportNavigatorJSON(
|
||||
layerType: string,
|
||||
layerId?: string,
|
||||
filters?: HeatmapFilters,
|
||||
): Promise<Blob> {
|
||||
const params: Record<string, string | number | undefined> = {
|
||||
layer: layerType,
|
||||
layer_id: layerId,
|
||||
...filters,
|
||||
};
|
||||
const { data } = await client.get("/heatmap/export-navigator", {
|
||||
params,
|
||||
responseType: "blob",
|
||||
});
|
||||
return data;
|
||||
}
|
||||
@@ -15,6 +15,7 @@ import {
|
||||
Database,
|
||||
Crosshair,
|
||||
Zap,
|
||||
Grid3X3,
|
||||
} from "lucide-react";
|
||||
import { useAuth } from "../context/AuthContext";
|
||||
|
||||
@@ -28,6 +29,7 @@ interface NavItem {
|
||||
const mainLinks: NavItem[] = [
|
||||
{ to: "/dashboard", label: "Dashboard", icon: LayoutDashboard },
|
||||
{ to: "/techniques", label: "ATT&CK Matrix", icon: Shield },
|
||||
{ to: "/matrix", label: "Advanced Heatmap", icon: Grid3X3 },
|
||||
{
|
||||
to: "/tests",
|
||||
label: "Tests",
|
||||
|
||||
185
frontend/src/components/heatmap/AdvancedHeatmap.tsx
Normal file
185
frontend/src/components/heatmap/AdvancedHeatmap.tsx
Normal file
@@ -0,0 +1,185 @@
|
||||
import { useMemo, useRef } from "react";
|
||||
import { useVirtualizer } from "@tanstack/react-virtual";
|
||||
import type { HeatmapTechnique } from "../../api/heatmap";
|
||||
import HeatmapCell from "./HeatmapCell";
|
||||
|
||||
// MITRE ATT&CK Enterprise tactics in canonical order
|
||||
const TACTIC_ORDER = [
|
||||
"reconnaissance",
|
||||
"resource-development",
|
||||
"initial-access",
|
||||
"execution",
|
||||
"persistence",
|
||||
"privilege-escalation",
|
||||
"defense-evasion",
|
||||
"credential-access",
|
||||
"discovery",
|
||||
"lateral-movement",
|
||||
"collection",
|
||||
"command-and-control",
|
||||
"exfiltration",
|
||||
"impact",
|
||||
];
|
||||
|
||||
const formatTacticName = (tactic: string): string =>
|
||||
tactic
|
||||
.split("-")
|
||||
.map((w) => w.charAt(0).toUpperCase() + w.slice(1))
|
||||
.join(" ");
|
||||
|
||||
interface AdvancedHeatmapProps {
|
||||
techniques: HeatmapTechnique[];
|
||||
onCellClick: (techniqueId: string) => void;
|
||||
zoom: "compact" | "normal" | "expanded";
|
||||
}
|
||||
|
||||
/** Virtualised tactic column — renders only visible rows. */
|
||||
function TacticColumn({
|
||||
tactic,
|
||||
techniques,
|
||||
zoom,
|
||||
onCellClick,
|
||||
}: {
|
||||
tactic: string;
|
||||
techniques: HeatmapTechnique[];
|
||||
zoom: "compact" | "normal" | "expanded";
|
||||
onCellClick: (techniqueId: string) => void;
|
||||
}) {
|
||||
const parentRef = useRef<HTMLDivElement>(null);
|
||||
|
||||
const rowHeight = zoom === "compact" ? 28 : zoom === "normal" ? 40 : 60;
|
||||
|
||||
const rowVirtualizer = useVirtualizer({
|
||||
count: techniques.length,
|
||||
getScrollElement: () => parentRef.current,
|
||||
estimateSize: () => rowHeight,
|
||||
overscan: 10,
|
||||
});
|
||||
|
||||
const columnWidth =
|
||||
zoom === "compact" ? "w-32" : zoom === "normal" ? "w-44" : "w-56";
|
||||
|
||||
return (
|
||||
<div className={`${columnWidth} flex-shrink-0`}>
|
||||
{/* Tactic header */}
|
||||
<div className="mb-2 rounded-lg bg-gray-800 px-2 py-2">
|
||||
<h3 className="text-center text-xs font-semibold text-cyan-400">
|
||||
{formatTacticName(tactic)}
|
||||
</h3>
|
||||
<p className="mt-0.5 text-center text-[10px] text-gray-500">
|
||||
{techniques.length} techniques
|
||||
</p>
|
||||
</div>
|
||||
|
||||
{/* Virtualised list */}
|
||||
<div
|
||||
ref={parentRef}
|
||||
className="overflow-y-auto scrollbar-thin scrollbar-thumb-gray-700 scrollbar-track-gray-900"
|
||||
style={{ maxHeight: "calc(100vh - 320px)" }}
|
||||
>
|
||||
<div
|
||||
style={{
|
||||
height: `${rowVirtualizer.getTotalSize()}px`,
|
||||
width: "100%",
|
||||
position: "relative",
|
||||
}}
|
||||
>
|
||||
{rowVirtualizer.getVirtualItems().map((virtualRow) => {
|
||||
const tech = techniques[virtualRow.index];
|
||||
return (
|
||||
<div
|
||||
key={tech.techniqueID + tactic}
|
||||
style={{
|
||||
position: "absolute",
|
||||
top: 0,
|
||||
left: 0,
|
||||
width: "100%",
|
||||
height: `${virtualRow.size}px`,
|
||||
transform: `translateY(${virtualRow.start}px)`,
|
||||
padding: "2px 0",
|
||||
}}
|
||||
>
|
||||
<HeatmapCell
|
||||
technique={tech}
|
||||
size={zoom}
|
||||
onClick={onCellClick}
|
||||
/>
|
||||
</div>
|
||||
);
|
||||
})}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
export default function AdvancedHeatmap({
|
||||
techniques,
|
||||
onCellClick,
|
||||
zoom,
|
||||
}: AdvancedHeatmapProps) {
|
||||
// Group techniques by tactic
|
||||
const groupedByTactic = useMemo(() => {
|
||||
const groups: Record<string, HeatmapTechnique[]> = {};
|
||||
|
||||
for (const tech of techniques) {
|
||||
// Normalize tactic names
|
||||
const tacticRaw = tech.tactic || "unknown";
|
||||
const tacticNormalized = tacticRaw
|
||||
.trim()
|
||||
.toLowerCase()
|
||||
.replace(/\s+/g, "-")
|
||||
.replace(/_/g, "-");
|
||||
|
||||
if (!groups[tacticNormalized]) {
|
||||
groups[tacticNormalized] = [];
|
||||
}
|
||||
groups[tacticNormalized].push(tech);
|
||||
}
|
||||
|
||||
// Sort techniques within each tactic by techniqueID
|
||||
for (const tactic of Object.keys(groups)) {
|
||||
groups[tactic].sort((a, b) =>
|
||||
a.techniqueID.localeCompare(b.techniqueID),
|
||||
);
|
||||
}
|
||||
|
||||
return groups;
|
||||
}, [techniques]);
|
||||
|
||||
// Get ordered tactics
|
||||
const orderedTactics = useMemo(() => {
|
||||
const tacticSet = new Set(Object.keys(groupedByTactic));
|
||||
const ordered = TACTIC_ORDER.filter((t) => tacticSet.has(t));
|
||||
const remaining = Array.from(tacticSet).filter(
|
||||
(t) => !TACTIC_ORDER.includes(t),
|
||||
);
|
||||
return [...ordered, ...remaining];
|
||||
}, [groupedByTactic]);
|
||||
|
||||
if (techniques.length === 0) {
|
||||
return (
|
||||
<div className="rounded-xl border border-gray-800 bg-gray-900 p-8 text-center">
|
||||
<p className="text-gray-400">No techniques found for the selected layer</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="overflow-x-auto rounded-xl border border-gray-800 bg-gray-900">
|
||||
<div className="min-w-max p-3">
|
||||
<div className="flex gap-2">
|
||||
{orderedTactics.map((tactic) => (
|
||||
<TacticColumn
|
||||
key={tactic}
|
||||
tactic={tactic}
|
||||
techniques={groupedByTactic[tactic] || []}
|
||||
zoom={zoom}
|
||||
onCellClick={onCellClick}
|
||||
/>
|
||||
))}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
81
frontend/src/components/heatmap/HeatmapCell.tsx
Normal file
81
frontend/src/components/heatmap/HeatmapCell.tsx
Normal file
@@ -0,0 +1,81 @@
|
||||
import { useState } from "react";
|
||||
import type { HeatmapTechnique } from "../../api/heatmap";
|
||||
import HeatmapTooltip from "./HeatmapTooltip";
|
||||
|
||||
interface HeatmapCellProps {
|
||||
technique: HeatmapTechnique;
|
||||
size: "compact" | "normal" | "expanded";
|
||||
onClick: (techniqueId: string) => void;
|
||||
}
|
||||
|
||||
export default function HeatmapCell({ technique, size, onClick }: HeatmapCellProps) {
|
||||
const [showTooltip, setShowTooltip] = useState(false);
|
||||
|
||||
const sizeClasses = {
|
||||
compact: "h-6 text-[9px] px-1",
|
||||
normal: "h-9 text-[11px] px-1.5",
|
||||
expanded: "h-14 text-xs px-2",
|
||||
};
|
||||
|
||||
const bgColor = technique.enabled ? technique.color : "transparent";
|
||||
const isDisabled = !technique.enabled;
|
||||
|
||||
// Determine text color based on background brightness
|
||||
const getTextColor = (hex: string): string => {
|
||||
if (!hex || hex === "transparent" || hex === "") return "text-gray-600";
|
||||
const r = parseInt(hex.slice(1, 3), 16);
|
||||
const g = parseInt(hex.slice(3, 5), 16);
|
||||
const b = parseInt(hex.slice(5, 7), 16);
|
||||
const brightness = (r * 299 + g * 587 + b * 114) / 1000;
|
||||
return brightness > 128 ? "text-gray-900" : "text-white";
|
||||
};
|
||||
|
||||
// Status indicators
|
||||
const hasTests = technique.metadata.find((m) => m.name === "tests_count");
|
||||
const testsCount = hasTests ? parseInt(hasTests.value, 10) : 0;
|
||||
const reviewRequired = technique.comment?.toLowerCase().includes("review");
|
||||
const isValidated = technique.score >= 100;
|
||||
|
||||
return (
|
||||
<div
|
||||
className="relative"
|
||||
onMouseEnter={() => setShowTooltip(true)}
|
||||
onMouseLeave={() => setShowTooltip(false)}
|
||||
>
|
||||
<button
|
||||
onClick={() => onClick(technique.techniqueID)}
|
||||
disabled={isDisabled}
|
||||
className={`
|
||||
w-full rounded border transition-all duration-150
|
||||
${sizeClasses[size]}
|
||||
${isDisabled
|
||||
? "cursor-default border-gray-800/30 bg-gray-900/20 opacity-30"
|
||||
: "cursor-pointer border-gray-700/50 hover:brightness-110 hover:ring-1 hover:ring-cyan-400/40"
|
||||
}
|
||||
${reviewRequired && !isDisabled ? "ring-1 ring-amber-400/60" : ""}
|
||||
flex items-center gap-1 overflow-hidden
|
||||
`}
|
||||
style={{
|
||||
backgroundColor: isDisabled ? undefined : bgColor,
|
||||
}}
|
||||
>
|
||||
<span className={`truncate font-mono font-medium leading-tight ${getTextColor(bgColor)}`}>
|
||||
{technique.techniqueID}
|
||||
</span>
|
||||
{size !== "compact" && !isDisabled && (
|
||||
<span className="ml-auto flex items-center gap-0.5 flex-shrink-0">
|
||||
{testsCount === 0 && <span className="text-[8px]" title="No tests">🔴</span>}
|
||||
{reviewRequired && <span className="text-[8px]" title="Review required">⚠️</span>}
|
||||
{isValidated && <span className="text-[8px]" title="Validated">✅</span>}
|
||||
</span>
|
||||
)}
|
||||
</button>
|
||||
|
||||
{showTooltip && technique.enabled && (
|
||||
<div className="absolute left-full top-0 z-50 ml-2">
|
||||
<HeatmapTooltip technique={technique} />
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
148
frontend/src/components/heatmap/HeatmapFilters.tsx
Normal file
148
frontend/src/components/heatmap/HeatmapFilters.tsx
Normal file
@@ -0,0 +1,148 @@
|
||||
import { Filter, X } from "lucide-react";
|
||||
|
||||
interface HeatmapFiltersProps {
|
||||
platforms: string[];
|
||||
onPlatformsChange: (platforms: string[]) => void;
|
||||
selectedTactics: string[];
|
||||
onTacticsChange: (tactics: string[]) => void;
|
||||
minScore: number;
|
||||
onMinScoreChange: (score: number) => void;
|
||||
availableTactics: string[];
|
||||
}
|
||||
|
||||
const PLATFORMS = ["windows", "linux", "macos"];
|
||||
|
||||
const formatTacticName = (tactic: string): string =>
|
||||
tactic
|
||||
.split("-")
|
||||
.map((w) => w.charAt(0).toUpperCase() + w.slice(1))
|
||||
.join(" ");
|
||||
|
||||
export default function HeatmapFilters({
|
||||
platforms,
|
||||
onPlatformsChange,
|
||||
selectedTactics,
|
||||
onTacticsChange,
|
||||
minScore,
|
||||
onMinScoreChange,
|
||||
availableTactics,
|
||||
}: HeatmapFiltersProps) {
|
||||
const togglePlatform = (platform: string) => {
|
||||
if (platforms.includes(platform)) {
|
||||
onPlatformsChange(platforms.filter((p) => p !== platform));
|
||||
} else {
|
||||
onPlatformsChange([...platforms, platform]);
|
||||
}
|
||||
};
|
||||
|
||||
const toggleTactic = (tactic: string) => {
|
||||
if (selectedTactics.includes(tactic)) {
|
||||
onTacticsChange(selectedTactics.filter((t) => t !== tactic));
|
||||
} else {
|
||||
onTacticsChange([...selectedTactics, tactic]);
|
||||
}
|
||||
};
|
||||
|
||||
const hasActiveFilters = platforms.length > 0 || selectedTactics.length > 0 || minScore > 0;
|
||||
|
||||
const clearAll = () => {
|
||||
onPlatformsChange([]);
|
||||
onTacticsChange([]);
|
||||
onMinScoreChange(0);
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="flex flex-wrap items-center gap-4">
|
||||
<div className="flex items-center gap-2">
|
||||
<Filter className="h-4 w-4 text-gray-400" />
|
||||
<span className="text-xs font-medium text-gray-400">Filters:</span>
|
||||
</div>
|
||||
|
||||
{/* Platform checkboxes */}
|
||||
<div className="flex items-center gap-2">
|
||||
{PLATFORMS.map((platform) => (
|
||||
<label
|
||||
key={platform}
|
||||
className="flex cursor-pointer items-center gap-1.5"
|
||||
>
|
||||
<input
|
||||
type="checkbox"
|
||||
checked={platforms.includes(platform)}
|
||||
onChange={() => togglePlatform(platform)}
|
||||
className="h-3.5 w-3.5 rounded border-gray-600 bg-gray-800 text-cyan-500 focus:ring-cyan-500/40"
|
||||
/>
|
||||
<span className="text-xs text-gray-300 capitalize">{platform}</span>
|
||||
</label>
|
||||
))}
|
||||
</div>
|
||||
|
||||
{/* Tactic multi-select */}
|
||||
<div className="relative">
|
||||
<select
|
||||
value=""
|
||||
onChange={(e) => {
|
||||
if (e.target.value) toggleTactic(e.target.value);
|
||||
}}
|
||||
className="rounded-lg border border-gray-700 bg-gray-800 px-2 py-1 text-xs text-gray-200 focus:border-cyan-500 focus:outline-none"
|
||||
>
|
||||
<option value="">
|
||||
{selectedTactics.length > 0
|
||||
? `${selectedTactics.length} Tactics`
|
||||
: "All Tactics"}
|
||||
</option>
|
||||
{availableTactics
|
||||
.filter((t) => !selectedTactics.includes(t))
|
||||
.map((tactic) => (
|
||||
<option key={tactic} value={tactic}>
|
||||
{formatTacticName(tactic)}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
</div>
|
||||
|
||||
{/* Selected tactic pills */}
|
||||
{selectedTactics.length > 0 && (
|
||||
<div className="flex flex-wrap items-center gap-1">
|
||||
{selectedTactics.map((tactic) => (
|
||||
<button
|
||||
key={tactic}
|
||||
onClick={() => toggleTactic(tactic)}
|
||||
className="flex items-center gap-1 rounded-full bg-cyan-500/10 px-2 py-0.5 text-[10px] text-cyan-400 hover:bg-cyan-500/20"
|
||||
>
|
||||
{formatTacticName(tactic)}
|
||||
<X className="h-2.5 w-2.5" />
|
||||
</button>
|
||||
))}
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Min score slider */}
|
||||
<div className="flex items-center gap-2">
|
||||
<span className="text-xs text-gray-400">Min Score:</span>
|
||||
<input
|
||||
type="range"
|
||||
min={0}
|
||||
max={100}
|
||||
step={5}
|
||||
value={minScore}
|
||||
onChange={(e) => onMinScoreChange(parseInt(e.target.value, 10))}
|
||||
className="h-1 w-20 cursor-pointer accent-cyan-500"
|
||||
/>
|
||||
<span className="w-6 text-right text-xs font-medium text-gray-300">
|
||||
{minScore}
|
||||
</span>
|
||||
</div>
|
||||
|
||||
{/* Clear all */}
|
||||
{hasActiveFilters && (
|
||||
<button
|
||||
onClick={clearAll}
|
||||
className="flex items-center gap-1 rounded-lg border border-gray-700 bg-gray-800 px-2 py-1 text-xs text-gray-400 hover:border-red-500/50 hover:text-red-400"
|
||||
>
|
||||
<X className="h-3 w-3" />
|
||||
Clear
|
||||
</button>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
120
frontend/src/components/heatmap/HeatmapLayerSelector.tsx
Normal file
120
frontend/src/components/heatmap/HeatmapLayerSelector.tsx
Normal file
@@ -0,0 +1,120 @@
|
||||
import { useState, useEffect } from "react";
|
||||
import { useQuery } from "@tanstack/react-query";
|
||||
import { Shield, User, Search, ClipboardList } from "lucide-react";
|
||||
import { getThreatActors, type ThreatActorSummary } from "../../api/threat-actors";
|
||||
import { listCampaigns, type CampaignSummary } from "../../api/campaigns";
|
||||
|
||||
export type LayerType = "coverage" | "threat-actor" | "detection-rules" | "campaign";
|
||||
|
||||
interface HeatmapLayerSelectorProps {
|
||||
activeLayer: LayerType;
|
||||
onLayerChange: (layer: LayerType) => void;
|
||||
selectedActorId: string | null;
|
||||
onActorChange: (actorId: string | null) => void;
|
||||
selectedCampaignId: string | null;
|
||||
onCampaignChange: (campaignId: string | null) => void;
|
||||
}
|
||||
|
||||
const LAYERS: {
|
||||
id: LayerType;
|
||||
label: string;
|
||||
icon: React.FC<{ className?: string }>;
|
||||
}[] = [
|
||||
{ id: "coverage", label: "Coverage", icon: Shield },
|
||||
{ id: "threat-actor", label: "Threat Actor", icon: User },
|
||||
{ id: "detection-rules", label: "Detection Rules", icon: Search },
|
||||
{ id: "campaign", label: "Campaign", icon: ClipboardList },
|
||||
];
|
||||
|
||||
export default function HeatmapLayerSelector({
|
||||
activeLayer,
|
||||
onLayerChange,
|
||||
selectedActorId,
|
||||
onActorChange,
|
||||
selectedCampaignId,
|
||||
onCampaignChange,
|
||||
}: HeatmapLayerSelectorProps) {
|
||||
// Fetch actors for dropdown
|
||||
const { data: actorsData } = useQuery({
|
||||
queryKey: ["threat-actors-selector"],
|
||||
queryFn: () => getThreatActors({ limit: 200 }),
|
||||
enabled: activeLayer === "threat-actor",
|
||||
});
|
||||
|
||||
// Fetch campaigns for dropdown
|
||||
const { data: campaignsData } = useQuery({
|
||||
queryKey: ["campaigns-selector"],
|
||||
queryFn: () => listCampaigns({ limit: 200 }),
|
||||
enabled: activeLayer === "campaign",
|
||||
});
|
||||
|
||||
const actors: ThreatActorSummary[] = actorsData?.items || [];
|
||||
const campaigns: CampaignSummary[] = campaignsData?.items || [];
|
||||
|
||||
// Auto-select first actor/campaign if none selected
|
||||
useEffect(() => {
|
||||
if (activeLayer === "threat-actor" && !selectedActorId && actors.length > 0) {
|
||||
onActorChange(actors[0].id);
|
||||
}
|
||||
}, [activeLayer, actors, selectedActorId, onActorChange]);
|
||||
|
||||
useEffect(() => {
|
||||
if (activeLayer === "campaign" && !selectedCampaignId && campaigns.length > 0) {
|
||||
onCampaignChange(campaigns[0].id);
|
||||
}
|
||||
}, [activeLayer, campaigns, selectedCampaignId, onCampaignChange]);
|
||||
|
||||
return (
|
||||
<div className="flex flex-wrap items-center gap-3">
|
||||
{/* Layer type tabs */}
|
||||
<div className="flex rounded-lg border border-gray-700 bg-gray-900 p-0.5">
|
||||
{LAYERS.map((layer) => (
|
||||
<button
|
||||
key={layer.id}
|
||||
onClick={() => onLayerChange(layer.id)}
|
||||
className={`flex items-center gap-1.5 rounded-md px-3 py-1.5 text-xs font-medium transition-colors ${
|
||||
activeLayer === layer.id
|
||||
? "bg-cyan-500/20 text-cyan-400"
|
||||
: "text-gray-400 hover:bg-gray-800 hover:text-gray-200"
|
||||
}`}
|
||||
>
|
||||
<layer.icon className="h-3.5 w-3.5" />
|
||||
{layer.label}
|
||||
</button>
|
||||
))}
|
||||
</div>
|
||||
|
||||
{/* Actor dropdown */}
|
||||
{activeLayer === "threat-actor" && (
|
||||
<select
|
||||
value={selectedActorId || ""}
|
||||
onChange={(e) => onActorChange(e.target.value || null)}
|
||||
className="rounded-lg border border-gray-700 bg-gray-800 px-3 py-1.5 text-sm text-gray-200 focus:border-cyan-500 focus:outline-none"
|
||||
>
|
||||
<option value="">Select Threat Actor...</option>
|
||||
{actors.map((actor) => (
|
||||
<option key={actor.id} value={actor.id}>
|
||||
{actor.name} {actor.country ? `(${actor.country})` : ""}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
)}
|
||||
|
||||
{/* Campaign dropdown */}
|
||||
{activeLayer === "campaign" && (
|
||||
<select
|
||||
value={selectedCampaignId || ""}
|
||||
onChange={(e) => onCampaignChange(e.target.value || null)}
|
||||
className="rounded-lg border border-gray-700 bg-gray-800 px-3 py-1.5 text-sm text-gray-200 focus:border-cyan-500 focus:outline-none"
|
||||
>
|
||||
<option value="">Select Campaign...</option>
|
||||
{campaigns.map((campaign) => (
|
||||
<option key={campaign.id} value={campaign.id}>
|
||||
{campaign.name} ({campaign.status})
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
79
frontend/src/components/heatmap/HeatmapLegend.tsx
Normal file
79
frontend/src/components/heatmap/HeatmapLegend.tsx
Normal file
@@ -0,0 +1,79 @@
|
||||
interface HeatmapLegendProps {
|
||||
layerType: "coverage" | "threat-actor" | "detection-rules" | "campaign";
|
||||
}
|
||||
|
||||
const LEGENDS: Record<
|
||||
string,
|
||||
{ label: string; colors: { color: string; label: string }[] }
|
||||
> = {
|
||||
coverage: {
|
||||
label: "Coverage Status",
|
||||
colors: [
|
||||
{ color: "#d3d3d3", label: "Not Evaluated (0)" },
|
||||
{ color: "#ff6666", label: "Not Covered (10)" },
|
||||
{ color: "#ff9933", label: "In Progress (30)" },
|
||||
{ color: "#ffff66", label: "Partial (60)" },
|
||||
{ color: "#66ff66", label: "Validated (100)" },
|
||||
],
|
||||
},
|
||||
"threat-actor": {
|
||||
label: "Threat Actor Coverage",
|
||||
colors: [
|
||||
{ color: "#d3d3d3", label: "Not Used by Actor" },
|
||||
{ color: "#ff6666", label: "Not Covered (10)" },
|
||||
{ color: "#ff9933", label: "In Progress (30)" },
|
||||
{ color: "#ffff66", label: "Partial (60)" },
|
||||
{ color: "#66ff66", label: "Covered (100)" },
|
||||
],
|
||||
},
|
||||
"detection-rules": {
|
||||
label: "Detection Rules Coverage",
|
||||
colors: [
|
||||
{ color: "#d3d3d3", label: "No Rules (0)" },
|
||||
{ color: "#ff6666", label: "Few Rules (<25)" },
|
||||
{ color: "#ff9933", label: "Some Rules (25-50)" },
|
||||
{ color: "#ffff66", label: "Good Coverage (50-75)" },
|
||||
{ color: "#66ff66", label: "Full Coverage (75-100)" },
|
||||
],
|
||||
},
|
||||
campaign: {
|
||||
label: "Campaign Progress",
|
||||
colors: [
|
||||
{ color: "#ff6666", label: "Draft / Rejected" },
|
||||
{ color: "#ff9933", label: "Red Executing (30)" },
|
||||
{ color: "#ffff66", label: "Blue Evaluating (50)" },
|
||||
{ color: "#66ff66", label: "Validated (100)" },
|
||||
],
|
||||
},
|
||||
};
|
||||
|
||||
export default function HeatmapLegend({ layerType }: HeatmapLegendProps) {
|
||||
const legend = LEGENDS[layerType] || LEGENDS.coverage;
|
||||
|
||||
return (
|
||||
<div className="flex flex-wrap items-center gap-4 rounded-xl border border-gray-800 bg-gray-900 p-4">
|
||||
<span className="text-sm font-medium text-gray-400">{legend.label}:</span>
|
||||
|
||||
{/* Gradient bar */}
|
||||
<div className="flex items-center gap-1">
|
||||
<div
|
||||
className="h-3 w-40 rounded"
|
||||
style={{
|
||||
background: `linear-gradient(to right, ${legend.colors.map((c) => c.color).join(", ")})`,
|
||||
}}
|
||||
/>
|
||||
</div>
|
||||
|
||||
{/* Individual labels */}
|
||||
{legend.colors.map((item) => (
|
||||
<div key={item.label} className="flex items-center gap-1.5">
|
||||
<div
|
||||
className="h-3 w-3 rounded border border-gray-700"
|
||||
style={{ backgroundColor: item.color }}
|
||||
/>
|
||||
<span className="text-xs text-gray-400">{item.label}</span>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
109
frontend/src/components/heatmap/HeatmapTooltip.tsx
Normal file
109
frontend/src/components/heatmap/HeatmapTooltip.tsx
Normal file
@@ -0,0 +1,109 @@
|
||||
import type { HeatmapTechnique } from "../../api/heatmap";
|
||||
|
||||
interface HeatmapTooltipProps {
|
||||
technique: HeatmapTechnique;
|
||||
}
|
||||
|
||||
export default function HeatmapTooltip({ technique }: HeatmapTooltipProps) {
|
||||
const getMeta = (name: string): string | null => {
|
||||
const item = technique.metadata.find((m) => m.name === name);
|
||||
return item?.value ?? null;
|
||||
};
|
||||
|
||||
const testsCount = getMeta("tests_count");
|
||||
const detectionRules = getMeta("detection_rules");
|
||||
const totalRules = getMeta("total_rules");
|
||||
const evaluatedRules = getMeta("evaluated_rules");
|
||||
const lastValidated = getMeta("last_validated");
|
||||
const campaignTests = getMeta("campaign_tests");
|
||||
|
||||
// Determine status label from score
|
||||
const getStatusLabel = (score: number): { label: string; color: string } => {
|
||||
if (score >= 100) return { label: "Validated", color: "text-green-400" };
|
||||
if (score >= 60) return { label: "Partial", color: "text-yellow-400" };
|
||||
if (score >= 30) return { label: "In Progress", color: "text-blue-400" };
|
||||
if (score > 0) return { label: "Not Covered", color: "text-red-400" };
|
||||
return { label: "Not Evaluated", color: "text-gray-400" };
|
||||
};
|
||||
|
||||
const status = getStatusLabel(technique.score);
|
||||
|
||||
return (
|
||||
<div className="w-72 rounded-lg border border-gray-700 bg-gray-900 p-3 shadow-xl">
|
||||
{/* Header */}
|
||||
<div className="mb-2 border-b border-gray-800 pb-2">
|
||||
<p className="font-mono text-sm font-bold text-white">
|
||||
{technique.techniqueID}
|
||||
</p>
|
||||
{technique.tactic && (
|
||||
<p className="mt-0.5 text-[10px] uppercase tracking-wider text-gray-500">
|
||||
{technique.tactic.replace(/-/g, " ")}
|
||||
</p>
|
||||
)}
|
||||
</div>
|
||||
|
||||
{/* Status & Score */}
|
||||
<div className="space-y-1.5 text-xs">
|
||||
<div className="flex items-center justify-between">
|
||||
<span className="text-gray-400">Status:</span>
|
||||
<span className={`font-medium ${status.color}`}>{status.label}</span>
|
||||
</div>
|
||||
<div className="flex items-center justify-between">
|
||||
<span className="text-gray-400">Score:</span>
|
||||
<span className="font-medium text-white">{technique.score}/100</span>
|
||||
</div>
|
||||
|
||||
{/* Score bar */}
|
||||
<div className="h-1.5 w-full overflow-hidden rounded-full bg-gray-800">
|
||||
<div
|
||||
className="h-full rounded-full transition-all"
|
||||
style={{
|
||||
width: `${technique.score}%`,
|
||||
backgroundColor: technique.color || "#666",
|
||||
}}
|
||||
/>
|
||||
</div>
|
||||
|
||||
{testsCount !== null && (
|
||||
<div className="flex items-center justify-between">
|
||||
<span className="text-gray-400">Tests:</span>
|
||||
<span className="text-gray-200">{testsCount} validated</span>
|
||||
</div>
|
||||
)}
|
||||
{detectionRules !== null && (
|
||||
<div className="flex items-center justify-between">
|
||||
<span className="text-gray-400">Detection Rules:</span>
|
||||
<span className="text-gray-200">{detectionRules} available</span>
|
||||
</div>
|
||||
)}
|
||||
{totalRules !== null && (
|
||||
<div className="flex items-center justify-between">
|
||||
<span className="text-gray-400">Rules:</span>
|
||||
<span className="text-gray-200">
|
||||
{evaluatedRules || 0} evaluated / {totalRules} total
|
||||
</span>
|
||||
</div>
|
||||
)}
|
||||
{campaignTests !== null && (
|
||||
<div className="flex items-center justify-between">
|
||||
<span className="text-gray-400">Campaign Tests:</span>
|
||||
<span className="text-gray-200">{campaignTests}</span>
|
||||
</div>
|
||||
)}
|
||||
{lastValidated && (
|
||||
<div className="flex items-center justify-between">
|
||||
<span className="text-gray-400">Last validated:</span>
|
||||
<span className="text-gray-200">{lastValidated}</span>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
||||
{/* Comment */}
|
||||
{technique.comment && (
|
||||
<p className="mt-2 border-t border-gray-800 pt-2 text-[10px] leading-relaxed text-gray-500">
|
||||
{technique.comment}
|
||||
</p>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
@@ -1,197 +1,287 @@
|
||||
import { useState, useMemo } from "react";
|
||||
import { useState, useMemo, useCallback } from "react";
|
||||
import { useQuery } from "@tanstack/react-query";
|
||||
import { Loader2, AlertCircle, Filter, X } from "lucide-react";
|
||||
import { getTechniques, type TechniqueSummary } from "../api/techniques";
|
||||
import AttackMatrix from "../components/AttackMatrix";
|
||||
import type { TechniqueStatus } from "../types/models";
|
||||
import { useNavigate } from "react-router-dom";
|
||||
import { Loader2, AlertCircle, Download, ZoomIn, ZoomOut } from "lucide-react";
|
||||
import {
|
||||
getHeatmapCoverage,
|
||||
getHeatmapThreatActor,
|
||||
getHeatmapDetectionRules,
|
||||
getHeatmapCampaign,
|
||||
exportNavigatorJSON,
|
||||
type HeatmapLayer,
|
||||
type HeatmapFilters as HeatmapFilterParams,
|
||||
} from "../api/heatmap";
|
||||
import AdvancedHeatmap from "../components/heatmap/AdvancedHeatmap";
|
||||
import HeatmapLayerSelector, {
|
||||
type LayerType,
|
||||
} from "../components/heatmap/HeatmapLayerSelector";
|
||||
import HeatmapFiltersComponent from "../components/heatmap/HeatmapFilters";
|
||||
import HeatmapLegend from "../components/heatmap/HeatmapLegend";
|
||||
|
||||
const STATUS_OPTIONS: { value: TechniqueStatus | "all"; label: string; color: string }[] = [
|
||||
{ value: "all", label: "All Statuses", color: "text-gray-400" },
|
||||
{ value: "validated", label: "Validated", color: "text-green-400" },
|
||||
{ value: "partial", label: "Partial", color: "text-yellow-400" },
|
||||
{ value: "in_progress", label: "In Progress", color: "text-blue-400" },
|
||||
{ value: "not_covered", label: "Not Covered", color: "text-red-400" },
|
||||
{ value: "not_evaluated", label: "Not Evaluated", color: "text-gray-400" },
|
||||
const TACTIC_ORDER = [
|
||||
"reconnaissance",
|
||||
"resource-development",
|
||||
"initial-access",
|
||||
"execution",
|
||||
"persistence",
|
||||
"privilege-escalation",
|
||||
"defense-evasion",
|
||||
"credential-access",
|
||||
"discovery",
|
||||
"lateral-movement",
|
||||
"collection",
|
||||
"command-and-control",
|
||||
"exfiltration",
|
||||
"impact",
|
||||
];
|
||||
|
||||
const PLATFORM_OPTIONS = ["all", "windows", "linux", "macos", "cloud", "network"] as const;
|
||||
type ZoomLevel = "compact" | "normal" | "expanded";
|
||||
|
||||
export default function MatrixPage() {
|
||||
const [statusFilter, setStatusFilter] = useState<TechniqueStatus | "all">("all");
|
||||
const [platformFilter, setPlatformFilter] = useState<string>("all");
|
||||
const [tacticFilter, setTacticFilter] = useState<string>("all");
|
||||
const navigate = useNavigate();
|
||||
|
||||
// Layer selection state
|
||||
const [activeLayer, setActiveLayer] = useState<LayerType>("coverage");
|
||||
const [selectedActorId, setSelectedActorId] = useState<string | null>(null);
|
||||
const [selectedCampaignId, setSelectedCampaignId] = useState<string | null>(null);
|
||||
|
||||
// Filter state
|
||||
const [platforms, setPlatforms] = useState<string[]>([]);
|
||||
const [selectedTactics, setSelectedTactics] = useState<string[]>([]);
|
||||
const [minScore, setMinScore] = useState(0);
|
||||
|
||||
// Zoom
|
||||
const [zoom, setZoom] = useState<ZoomLevel>("normal");
|
||||
|
||||
// Export dropdown
|
||||
const [showExportMenu, setShowExportMenu] = useState(false);
|
||||
|
||||
// Build filter params
|
||||
const filterParams: HeatmapFilterParams = useMemo(
|
||||
() => ({
|
||||
platforms: platforms.length > 0 ? platforms.join(",") : undefined,
|
||||
tactics: selectedTactics.length > 0 ? selectedTactics.join(",") : undefined,
|
||||
min_score: minScore > 0 ? minScore : undefined,
|
||||
}),
|
||||
[platforms, selectedTactics, minScore],
|
||||
);
|
||||
|
||||
// Build query key based on active layer + selection
|
||||
const queryKey = useMemo(() => {
|
||||
const base = ["heatmap", activeLayer, filterParams];
|
||||
if (activeLayer === "threat-actor") return [...base, selectedActorId];
|
||||
if (activeLayer === "campaign") return [...base, selectedCampaignId];
|
||||
return base;
|
||||
}, [activeLayer, filterParams, selectedActorId, selectedCampaignId]);
|
||||
|
||||
// Fetch the active layer data
|
||||
const {
|
||||
data: techniques,
|
||||
data: layerData,
|
||||
isLoading,
|
||||
error,
|
||||
} = useQuery({
|
||||
queryKey: ["techniques"],
|
||||
queryFn: () => getTechniques(),
|
||||
} = useQuery<HeatmapLayer>({
|
||||
queryKey,
|
||||
queryFn: () => {
|
||||
switch (activeLayer) {
|
||||
case "coverage":
|
||||
return getHeatmapCoverage(filterParams);
|
||||
case "threat-actor":
|
||||
if (!selectedActorId) return Promise.resolve({ name: "", versions: { attack: "", navigator: "", layer: "" }, domain: "", description: "", filters: { platforms: [] }, gradient: { colors: [], minValue: 0, maxValue: 0 }, techniques: [] } as HeatmapLayer);
|
||||
return getHeatmapThreatActor(selectedActorId, filterParams);
|
||||
case "detection-rules":
|
||||
return getHeatmapDetectionRules(filterParams);
|
||||
case "campaign":
|
||||
if (!selectedCampaignId) return Promise.resolve({ name: "", versions: { attack: "", navigator: "", layer: "" }, domain: "", description: "", filters: { platforms: [] }, gradient: { colors: [], minValue: 0, maxValue: 0 }, techniques: [] } as HeatmapLayer);
|
||||
return getHeatmapCampaign(selectedCampaignId, filterParams);
|
||||
default:
|
||||
return getHeatmapCoverage(filterParams);
|
||||
}
|
||||
},
|
||||
enabled:
|
||||
activeLayer === "coverage" ||
|
||||
activeLayer === "detection-rules" ||
|
||||
(activeLayer === "threat-actor" && !!selectedActorId) ||
|
||||
(activeLayer === "campaign" && !!selectedCampaignId),
|
||||
});
|
||||
|
||||
// Extract unique tactics from techniques
|
||||
const availableTactics = useMemo(() => {
|
||||
if (!techniques) return [];
|
||||
const tactics = new Set<string>();
|
||||
for (const tech of techniques) {
|
||||
if (tech.tactic) {
|
||||
tech.tactic.split(",").forEach((t) => tactics.add(t.trim().toLowerCase()));
|
||||
const techniques = layerData?.techniques || [];
|
||||
|
||||
// Handle cell click - navigate to technique detail
|
||||
const handleCellClick = useCallback(
|
||||
(techniqueId: string) => {
|
||||
navigate(`/techniques/${techniqueId}`);
|
||||
},
|
||||
[navigate],
|
||||
);
|
||||
|
||||
// Handle export
|
||||
const handleExport = async (type: "download" | "url") => {
|
||||
setShowExportMenu(false);
|
||||
|
||||
const layerId =
|
||||
activeLayer === "threat-actor"
|
||||
? selectedActorId ?? undefined
|
||||
: activeLayer === "campaign"
|
||||
? selectedCampaignId ?? undefined
|
||||
: undefined;
|
||||
|
||||
if (type === "download") {
|
||||
try {
|
||||
const blob = await exportNavigatorJSON(activeLayer, layerId, filterParams);
|
||||
const url = URL.createObjectURL(blob);
|
||||
const a = document.createElement("a");
|
||||
a.href = url;
|
||||
a.download = `aegis_${activeLayer}_layer.json`;
|
||||
document.body.appendChild(a);
|
||||
a.click();
|
||||
document.body.removeChild(a);
|
||||
URL.revokeObjectURL(url);
|
||||
} catch {
|
||||
console.error("Failed to export Navigator JSON");
|
||||
}
|
||||
} else {
|
||||
// Copy Navigator URL
|
||||
const navigatorUrl = `https://mitre-attack.github.io/attack-navigator/#layerURL=${encodeURIComponent(
|
||||
window.location.origin + `/api/v1/heatmap/export-navigator?layer=${activeLayer}${layerId ? `&layer_id=${layerId}` : ""}`
|
||||
)}`;
|
||||
navigator.clipboard.writeText(navigatorUrl);
|
||||
}
|
||||
return Array.from(tactics).sort();
|
||||
}, [techniques]);
|
||||
|
||||
// Apply filters
|
||||
const filteredTechniques = useMemo(() => {
|
||||
if (!techniques) return [];
|
||||
|
||||
return techniques.filter((tech: TechniqueSummary) => {
|
||||
// Status filter
|
||||
if (statusFilter !== "all" && tech.status_global !== statusFilter) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// Tactic filter
|
||||
if (tacticFilter !== "all") {
|
||||
const techTactics = tech.tactic?.split(",").map((t) => t.trim().toLowerCase()) || [];
|
||||
if (!techTactics.includes(tacticFilter)) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
// Platform filter is handled client-side since we don't have platform in summary
|
||||
// For now we show all - platform filtering would need the full technique data
|
||||
|
||||
return true;
|
||||
});
|
||||
}, [techniques, statusFilter, tacticFilter]);
|
||||
|
||||
const hasActiveFilters = statusFilter !== "all" || tacticFilter !== "all" || platformFilter !== "all";
|
||||
|
||||
const clearFilters = () => {
|
||||
setStatusFilter("all");
|
||||
setPlatformFilter("all");
|
||||
setTacticFilter("all");
|
||||
};
|
||||
|
||||
if (isLoading) {
|
||||
return (
|
||||
<div className="flex h-64 items-center justify-center">
|
||||
<Loader2 className="h-8 w-8 animate-spin text-cyan-400" />
|
||||
</div>
|
||||
);
|
||||
}
|
||||
// Zoom controls
|
||||
const zoomIn = () => {
|
||||
if (zoom === "compact") setZoom("normal");
|
||||
else if (zoom === "normal") setZoom("expanded");
|
||||
};
|
||||
|
||||
if (error) {
|
||||
return (
|
||||
<div className="flex h-64 flex-col items-center justify-center gap-2">
|
||||
<AlertCircle className="h-10 w-10 text-red-400" />
|
||||
<p className="text-red-400">Failed to load techniques</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
const zoomOut = () => {
|
||||
if (zoom === "expanded") setZoom("normal");
|
||||
else if (zoom === "normal") setZoom("compact");
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="space-y-6">
|
||||
<div className="space-y-4">
|
||||
{/* Header */}
|
||||
<div>
|
||||
<h1 className="text-2xl font-bold text-white">ATT&CK Matrix</h1>
|
||||
<p className="mt-1 text-sm text-gray-400">
|
||||
Interactive MITRE ATT&CK coverage matrix — click any technique for details
|
||||
Advanced heatmap with multiple layers, filters, and ATT&CK Navigator export
|
||||
</p>
|
||||
</div>
|
||||
|
||||
{/* Filters */}
|
||||
<div className="flex flex-wrap items-center gap-4 rounded-xl border border-gray-800 bg-gray-900 p-4">
|
||||
<div className="flex items-center gap-2">
|
||||
<Filter className="h-4 w-4 text-gray-400" />
|
||||
<span className="text-sm font-medium text-gray-400">Filters:</span>
|
||||
{/* Toolbar: Layer Selector + Filters + Export + Zoom */}
|
||||
<div className="rounded-xl border border-gray-800 bg-gray-900 p-4 space-y-3">
|
||||
<div className="flex flex-wrap items-center justify-between gap-4">
|
||||
{/* Layer selector */}
|
||||
<HeatmapLayerSelector
|
||||
activeLayer={activeLayer}
|
||||
onLayerChange={setActiveLayer}
|
||||
selectedActorId={selectedActorId}
|
||||
onActorChange={setSelectedActorId}
|
||||
selectedCampaignId={selectedCampaignId}
|
||||
onCampaignChange={setSelectedCampaignId}
|
||||
/>
|
||||
|
||||
{/* Right side: Export + Zoom */}
|
||||
<div className="flex items-center gap-2">
|
||||
{/* Export dropdown */}
|
||||
<div className="relative">
|
||||
<button
|
||||
onClick={() => setShowExportMenu(!showExportMenu)}
|
||||
className="flex items-center gap-1.5 rounded-lg border border-gray-700 bg-gray-800 px-3 py-1.5 text-xs font-medium text-gray-300 hover:border-cyan-500/50 hover:text-white"
|
||||
>
|
||||
<Download className="h-3.5 w-3.5" />
|
||||
Export
|
||||
</button>
|
||||
{showExportMenu && (
|
||||
<div className="absolute right-0 top-full z-30 mt-1 w-52 rounded-lg border border-gray-700 bg-gray-900 py-1 shadow-xl">
|
||||
<button
|
||||
onClick={() => handleExport("download")}
|
||||
className="w-full px-3 py-2 text-left text-xs text-gray-300 hover:bg-gray-800"
|
||||
>
|
||||
Export Navigator JSON
|
||||
</button>
|
||||
<button
|
||||
onClick={() => handleExport("url")}
|
||||
className="w-full px-3 py-2 text-left text-xs text-gray-300 hover:bg-gray-800"
|
||||
>
|
||||
Copy Navigator URL
|
||||
</button>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
|
||||
{/* Zoom controls */}
|
||||
<div className="flex items-center rounded-lg border border-gray-700 bg-gray-800">
|
||||
<button
|
||||
onClick={zoomOut}
|
||||
disabled={zoom === "compact"}
|
||||
className="px-2 py-1.5 text-gray-400 hover:text-white disabled:opacity-30"
|
||||
>
|
||||
<ZoomOut className="h-3.5 w-3.5" />
|
||||
</button>
|
||||
<span className="border-x border-gray-700 px-2 py-1 text-[10px] font-medium uppercase text-gray-400">
|
||||
{zoom}
|
||||
</span>
|
||||
<button
|
||||
onClick={zoomIn}
|
||||
disabled={zoom === "expanded"}
|
||||
className="px-2 py-1.5 text-gray-400 hover:text-white disabled:opacity-30"
|
||||
>
|
||||
<ZoomIn className="h-3.5 w-3.5" />
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* Status filter */}
|
||||
<select
|
||||
value={statusFilter}
|
||||
onChange={(e) => setStatusFilter(e.target.value as TechniqueStatus | "all")}
|
||||
className="rounded-lg border border-gray-700 bg-gray-800 px-3 py-1.5 text-sm text-gray-200 focus:border-cyan-500 focus:outline-none"
|
||||
>
|
||||
{STATUS_OPTIONS.map((opt) => (
|
||||
<option key={opt.value} value={opt.value}>
|
||||
{opt.label}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
|
||||
{/* Tactic filter */}
|
||||
<select
|
||||
value={tacticFilter}
|
||||
onChange={(e) => setTacticFilter(e.target.value)}
|
||||
className="rounded-lg border border-gray-700 bg-gray-800 px-3 py-1.5 text-sm text-gray-200 focus:border-cyan-500 focus:outline-none"
|
||||
>
|
||||
<option value="all">All Tactics</option>
|
||||
{availableTactics.map((tactic) => (
|
||||
<option key={tactic} value={tactic}>
|
||||
{tactic
|
||||
.split("-")
|
||||
.map((w) => w.charAt(0).toUpperCase() + w.slice(1))
|
||||
.join(" ")}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
|
||||
{/* Platform filter */}
|
||||
<select
|
||||
value={platformFilter}
|
||||
onChange={(e) => setPlatformFilter(e.target.value)}
|
||||
className="rounded-lg border border-gray-700 bg-gray-800 px-3 py-1.5 text-sm text-gray-200 focus:border-cyan-500 focus:outline-none"
|
||||
>
|
||||
{PLATFORM_OPTIONS.map((platform) => (
|
||||
<option key={platform} value={platform}>
|
||||
{platform === "all" ? "All Platforms" : platform.charAt(0).toUpperCase() + platform.slice(1)}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
|
||||
{hasActiveFilters && (
|
||||
<button
|
||||
onClick={clearFilters}
|
||||
className="flex items-center gap-1 rounded-lg border border-gray-700 bg-gray-800 px-3 py-1.5 text-sm text-gray-400 hover:border-red-500/50 hover:text-red-400"
|
||||
>
|
||||
<X className="h-3.5 w-3.5" />
|
||||
Clear
|
||||
</button>
|
||||
)}
|
||||
|
||||
<div className="ml-auto text-sm text-gray-500">
|
||||
Showing {filteredTechniques.length} of {techniques?.length || 0} techniques
|
||||
</div>
|
||||
{/* Filters */}
|
||||
<HeatmapFiltersComponent
|
||||
platforms={platforms}
|
||||
onPlatformsChange={setPlatforms}
|
||||
selectedTactics={selectedTactics}
|
||||
onTacticsChange={setSelectedTactics}
|
||||
minScore={minScore}
|
||||
onMinScoreChange={setMinScore}
|
||||
availableTactics={TACTIC_ORDER}
|
||||
/>
|
||||
</div>
|
||||
|
||||
{/* Matrix */}
|
||||
<AttackMatrix techniques={filteredTechniques} />
|
||||
{/* Stats bar */}
|
||||
{layerData && (
|
||||
<div className="flex items-center gap-4 text-xs text-gray-500">
|
||||
<span>
|
||||
Layer: <span className="text-gray-300">{layerData.name}</span>
|
||||
</span>
|
||||
<span>
|
||||
Techniques:{" "}
|
||||
<span className="text-gray-300">
|
||||
{techniques.filter((t) => t.enabled).length} active
|
||||
</span>{" "}
|
||||
/ {techniques.length} total
|
||||
</span>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{/* Loading / Error / Heatmap */}
|
||||
{isLoading ? (
|
||||
<div className="flex h-64 items-center justify-center">
|
||||
<Loader2 className="h-8 w-8 animate-spin text-cyan-400" />
|
||||
</div>
|
||||
) : error ? (
|
||||
<div className="flex h-64 flex-col items-center justify-center gap-2">
|
||||
<AlertCircle className="h-10 w-10 text-red-400" />
|
||||
<p className="text-red-400">Failed to load heatmap data</p>
|
||||
</div>
|
||||
) : (
|
||||
<AdvancedHeatmap
|
||||
techniques={techniques}
|
||||
onCellClick={handleCellClick}
|
||||
zoom={zoom}
|
||||
/>
|
||||
)}
|
||||
|
||||
{/* Legend */}
|
||||
<div className="flex flex-wrap items-center gap-4 rounded-xl border border-gray-800 bg-gray-900 p-4">
|
||||
<span className="text-sm font-medium text-gray-400">Legend:</span>
|
||||
{STATUS_OPTIONS.filter((s) => s.value !== "all").map((status) => (
|
||||
<div key={status.value} className="flex items-center gap-2">
|
||||
<div
|
||||
className={`h-3 w-3 rounded ${
|
||||
status.value === "validated"
|
||||
? "bg-green-500"
|
||||
: status.value === "partial"
|
||||
? "bg-yellow-500"
|
||||
: status.value === "in_progress"
|
||||
? "bg-blue-500"
|
||||
: status.value === "not_covered"
|
||||
? "bg-red-500"
|
||||
: "bg-gray-600"
|
||||
}`}
|
||||
/>
|
||||
<span className="text-xs text-gray-400">{status.label}</span>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
<HeatmapLegend layerType={activeLayer} />
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user