@countermeasure-platform/web-components 0.2.0

package/src/react/primitives/mitre-matrix.tsx

@@ -0,0 +1,203 @@
+import * as React from 'react'
+import { cva } from 'class-variance-authority'
+
+import { cn } from './utils'
+
+/* ------------------------------------------------------------------ */
+/*  Types                                                              */
+/* ------------------------------------------------------------------ */
+
+export type MitreSeverity = 'critical' | 'high' | 'medium' | 'low'
+
+export interface MitreTechnique {
+  id: string
+  name: string
+  tactic: string
+  detected?: boolean
+  count?: number
+  severity?: MitreSeverity
+}
+
+/* ------------------------------------------------------------------ */
+/*  Constants                                                          */
+/* ------------------------------------------------------------------ */
+
+const DEFAULT_TACTICS: readonly string[] = [
+  'Initial Access',
+  'Execution',
+  'Persistence',
+  'Privilege Escalation',
+  'Defense Evasion',
+  'Credential Access',
+  'Discovery',
+  'Lateral Movement',
+  'Collection',
+  'Command and Control',
+  'Exfiltration',
+  'Impact',
+]
+
+/* ------------------------------------------------------------------ */
+/*  Variants                                                           */
+/* ------------------------------------------------------------------ */
+
+const techniqueVariants = cva(
+  'group relative flex flex-col gap-0.5 rounded-lg border px-2.5 py-2 text-left transition-all duration-200',
+  {
+    variants: {
+      severity: {
+        critical: 'border-accent-rose/30 hover:border-accent-rose/50',
+        high: 'border-accent-rose/20 hover:border-accent-rose/40',
+        medium: 'border-accent-amber/20 hover:border-accent-amber/40',
+        low: 'border-accent-cyan/20 hover:border-accent-cyan/40',
+      },
+      detected: {
+        true: 'bg-accent-emerald/10 border-accent-emerald/30 hover:bg-accent-emerald/15',
+        false: 'bg-glass border-border-glass hover:bg-glass-hover',
+      },
+    },
+    defaultVariants: {
+      detected: false,
+    },
+  }
+)
+
+/* ------------------------------------------------------------------ */
+/*  TechniqueCell                                                      */
+/* ------------------------------------------------------------------ */
+
+interface TechniqueCellProps {
+  technique: MitreTechnique
+  showCounts?: boolean
+  highlightDetected?: boolean
+  onClick?: ((technique: MitreTechnique) => void) | undefined
+}
+
+function TechniqueCell({
+  technique,
+  showCounts = false,
+  highlightDetected = true,
+  onClick,
+}: TechniqueCellProps) {
+  const detected = highlightDetected && technique.detected === true
+
+  return (
+    <div
+      className={cn(
+        detected
+          ? techniqueVariants({ detected: true })
+          : technique.severity
+            ? techniqueVariants({ severity: technique.severity, detected: false })
+            : techniqueVariants({ detected: false }),
+        onClick && 'cursor-pointer'
+      )}
+      title={`${technique.id}: ${technique.name}`}
+      onClick={onClick ? () => onClick(technique) : undefined}
+    >
+      {/* Name */}
+      <span className="truncate text-[11px] font-medium leading-tight text-foreground">
+        {technique.name}
+      </span>
+
+      {/* ID + optional count */}
+      <div className="flex items-center justify-between gap-1">
+        <span className="text-[10px] font-mono text-foreground-muted">{technique.id}</span>
+        {showCounts && technique.count !== undefined && technique.count > 0 && (
+          <span className="rounded-full bg-accent-violet/15 px-1.5 py-0.5 text-[9px] font-bold tabular-nums text-accent-violet">
+            {technique.count}
+          </span>
+        )}
+      </div>
+
+      {/* Hover tooltip */}
+      <div className="pointer-events-none absolute left-1/2 top-full z-20 mt-1 hidden w-44 -translate-x-1/2 rounded-lg border border-border-glass bg-surface-secondary/95 p-2 shadow-lg backdrop-blur-xl group-hover:block">
+        <div className="text-[10px] font-mono text-foreground-muted">{technique.id}</div>
+        <div className="text-xs font-medium text-foreground">{technique.name}</div>
+        {detected && <div className="mt-1 text-[10px] font-bold text-accent-emerald">Detected</div>}
+        {technique.count !== undefined && technique.count > 0 && (
+          <div className="text-[10px] text-foreground-secondary">Detections: {technique.count}</div>
+        )}
+      </div>
+    </div>
+  )
+}
+
+/* ------------------------------------------------------------------ */
+/*  MitreMatrix                                                        */
+/* ------------------------------------------------------------------ */
+
+export interface MitreMatrixProps extends React.ComponentProps<'div'> {
+  techniques: MitreTechnique[]
+  /** Tactic column headers. Defaults to the standard 12 ATT&CK tactics. */
+  tactics?: string[]
+  /** Show detection count badges on technique cells. */
+  showCounts?: boolean
+  /** Highlight detected techniques with green accent. Defaults to true. */
+  highlightDetected?: boolean
+  /** Callback when a technique cell is clicked. */
+  onTechniqueClick?: (technique: MitreTechnique) => void
+}
+
+function MitreMatrix({
+  className,
+  techniques,
+  tactics = DEFAULT_TACTICS as string[],
+  showCounts = false,
+  highlightDetected = true,
+  onTechniqueClick,
+  ...props
+}: MitreMatrixProps) {
+  /* Group techniques by tactic */
+  const tacticMap = React.useMemo(() => {
+    const map = new Map<string, MitreTechnique[]>()
+    for (const tactic of tactics) {
+      map.set(tactic, [])
+    }
+    for (const tech of techniques) {
+      const list = map.get(tech.tactic)
+      if (list) {
+        list.push(tech)
+      }
+    }
+    return map
+  }, [tactics, techniques])
+
+  return (
+    <div
+      data-slot="mitre-matrix"
+      role="grid"
+      aria-label="MITRE ATT&CK Matrix"
+      className={cn('overflow-x-auto', className)}
+      {...props}
+    >
+      <div className="inline-flex min-w-full gap-2">
+        {tactics.map(tactic => {
+          const techs = tacticMap.get(tactic) ?? []
+          return (
+            <div key={tactic} role="gridcell" className="flex w-36 shrink-0 flex-col gap-1.5">
+              {/* Tactic header */}
+              <div className="rounded-lg border border-border-glass bg-surface-secondary px-2.5 py-2 text-center">
+                <span className="text-[11px] font-bold uppercase tracking-wider text-foreground-secondary">
+                  {tactic}
+                </span>
+              </div>
+
+              {/* Technique cells */}
+              {techs.map(tech => (
+                <TechniqueCell
+                  key={tech.id}
+                  technique={tech}
+                  showCounts={showCounts}
+                  highlightDetected={highlightDetected}
+                  onClick={onTechniqueClick}
+                />
+              ))}
+            </div>
+          )
+        })}
+      </div>
+    </div>
+  )
+}
+
+export { MitreMatrix, techniqueVariants }

package/src/react/primitives/modal.tsx

@@ -0,0 +1,191 @@
+import * as DialogPrimitive from '@radix-ui/react-dialog'
+import { XIcon } from 'lucide-react'
+import * as React from 'react'
+
+import { cn } from './utils'
+
+/* ------------------------------------------------------------------ */
+/* Types                                                               */
+/* ------------------------------------------------------------------ */
+
+export type ModalVariant = 'confirm' | 'form' | 'alert'
+
+export interface ModalProps extends React.ComponentProps<typeof DialogPrimitive.Root> {
+  children: React.ReactNode
+}
+
+export interface ModalContentProps extends React.ComponentProps<typeof DialogPrimitive.Content> {
+  /** Whether to show the close button */
+  showCloseButton?: boolean
+}
+
+/* ------------------------------------------------------------------ */
+/* Root                                                                */
+/* ------------------------------------------------------------------ */
+
+function Modal({ ...props }: ModalProps) {
+  return <DialogPrimitive.Root data-slot="modal" {...props} />
+}
+
+/* ------------------------------------------------------------------ */
+/* Trigger                                                             */
+/* ------------------------------------------------------------------ */
+
+function ModalTrigger({ ...props }: React.ComponentProps<typeof DialogPrimitive.Trigger>) {
+  return <DialogPrimitive.Trigger data-slot="modal-trigger" {...props} />
+}
+
+/* ------------------------------------------------------------------ */
+/* Close                                                               */
+/* ------------------------------------------------------------------ */
+
+function ModalClose({ ...props }: React.ComponentProps<typeof DialogPrimitive.Close>) {
+  return <DialogPrimitive.Close data-slot="modal-close" {...props} />
+}
+
+/* ------------------------------------------------------------------ */
+/* Overlay                                                             */
+/* ------------------------------------------------------------------ */
+
+function ModalOverlay({
+  className,
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Overlay>) {
+  return (
+    <DialogPrimitive.Overlay
+      data-slot="modal-overlay"
+      className={cn(
+        'fixed inset-0 z-50 bg-black/60 backdrop-blur-sm',
+        'data-[state=open]:animate-in data-[state=closed]:animate-out',
+        'data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0',
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+/* ------------------------------------------------------------------ */
+/* Content                                                             */
+/* ------------------------------------------------------------------ */
+
+function ModalContent({
+  className,
+  children,
+  showCloseButton = true,
+  ...props
+}: ModalContentProps) {
+  return (
+    <DialogPrimitive.Portal>
+      <ModalOverlay />
+      <DialogPrimitive.Content
+        data-slot="modal-content"
+        className={cn(
+          'fixed top-[50%] left-[50%] z-50 w-full max-w-lg translate-x-[-50%] translate-y-[-50%]',
+          'grid gap-0 overflow-hidden rounded-xl border border-border-glass shadow-xl',
+          'bg-glass backdrop-blur-xl',
+          'data-[state=open]:animate-in data-[state=closed]:animate-out',
+          'data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0',
+          'data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95',
+          'duration-200 outline-none',
+          className
+        )}
+        {...props}
+      >
+        {children}
+        {showCloseButton && (
+          <DialogPrimitive.Close
+            data-slot="modal-close"
+            className="absolute top-4 right-4 rounded-lg p-1 text-foreground-muted opacity-70 transition-opacity hover:opacity-100 focus:ring-2 focus:ring-primary/50 focus:outline-none"
+          >
+            <XIcon className="size-4" />
+            <span className="sr-only">Close</span>
+          </DialogPrimitive.Close>
+        )}
+      </DialogPrimitive.Content>
+    </DialogPrimitive.Portal>
+  )
+}
+
+/* ------------------------------------------------------------------ */
+/* Header                                                              */
+/* ------------------------------------------------------------------ */
+
+function ModalHeader({ className, ...props }: React.ComponentProps<'div'>) {
+  return (
+    <div
+      data-slot="modal-header"
+      className={cn('flex flex-col gap-1.5 border-b border-border-glass px-6 py-4', className)}
+      {...props}
+    />
+  )
+}
+
+/* ------------------------------------------------------------------ */
+/* Title                                                               */
+/* ------------------------------------------------------------------ */
+
+function ModalTitle({ className, ...props }: React.ComponentProps<typeof DialogPrimitive.Title>) {
+  return (
+    <DialogPrimitive.Title
+      data-slot="modal-title"
+      className={cn('text-lg font-semibold leading-none text-foreground', className)}
+      {...props}
+    />
+  )
+}
+
+/* ------------------------------------------------------------------ */
+/* Description                                                         */
+/* ------------------------------------------------------------------ */
+
+function ModalDescription({
+  className,
+  ...props
+}: React.ComponentProps<typeof DialogPrimitive.Description>) {
+  return (
+    <DialogPrimitive.Description
+      data-slot="modal-description"
+      className={cn('text-sm text-foreground-muted', className)}
+      {...props}
+    />
+  )
+}
+
+/* ------------------------------------------------------------------ */
+/* Body                                                                */
+/* ------------------------------------------------------------------ */
+
+function ModalBody({ className, ...props }: React.ComponentProps<'div'>) {
+  return <div data-slot="modal-body" className={cn('px-6 py-4', className)} {...props} />
+}
+
+/* ------------------------------------------------------------------ */
+/* Footer                                                              */
+/* ------------------------------------------------------------------ */
+
+function ModalFooter({ className, ...props }: React.ComponentProps<'div'>) {
+  return (
+    <div
+      data-slot="modal-footer"
+      className={cn(
+        'flex items-center justify-end gap-2 border-t border-border-glass px-6 py-4',
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+export {
+  Modal,
+  ModalTrigger,
+  ModalClose,
+  ModalOverlay,
+  ModalContent,
+  ModalHeader,
+  ModalTitle,
+  ModalDescription,
+  ModalBody,
+  ModalFooter,
+}

package/src/react/primitives/navigation-menu.tsx

@@ -0,0 +1,144 @@
+import * as NavigationMenuPrimitive from '@radix-ui/react-navigation-menu'
+import { ChevronDownIcon } from 'lucide-react'
+import * as React from 'react'
+
+import { cn } from './utils'
+
+function NavigationMenu({
+  className,
+  children,
+  ...props
+}: React.ComponentProps<typeof NavigationMenuPrimitive.Root>) {
+  return (
+    <NavigationMenuPrimitive.Root
+      data-slot="navigation-menu"
+      className={cn('relative z-10 flex max-w-max flex-1 items-center justify-center', className)}
+      {...props}
+    >
+      {children}
+      <NavigationMenuViewport />
+    </NavigationMenuPrimitive.Root>
+  )
+}
+
+function NavigationMenuList({
+  className,
+  ...props
+}: React.ComponentProps<typeof NavigationMenuPrimitive.List>) {
+  return (
+    <NavigationMenuPrimitive.List
+      data-slot="navigation-menu-list"
+      className={cn('group flex flex-1 list-none items-center justify-center gap-1', className)}
+      {...props}
+    />
+  )
+}
+
+function NavigationMenuItem({
+  ...props
+}: React.ComponentProps<typeof NavigationMenuPrimitive.Item>) {
+  return <NavigationMenuPrimitive.Item data-slot="navigation-menu-item" {...props} />
+}
+
+const navigationMenuTriggerStyle =
+  'group inline-flex h-9 w-max items-center justify-center rounded-lg px-4 py-2 text-sm font-medium transition-colors hover:bg-accent hover:text-foreground focus:bg-accent focus:text-foreground focus:outline-none disabled:pointer-events-none disabled:opacity-50 data-[state=open]:bg-accent/50 data-[active]:bg-accent/50 text-foreground-secondary'
+
+function NavigationMenuTrigger({
+  className,
+  children,
+  ...props
+}: React.ComponentProps<typeof NavigationMenuPrimitive.Trigger>) {
+  return (
+    <NavigationMenuPrimitive.Trigger
+      data-slot="navigation-menu-trigger"
+      className={cn(navigationMenuTriggerStyle, 'group', className)}
+      {...props}
+    >
+      {children}
+      <ChevronDownIcon
+        className="relative top-px ml-1 size-3 transition-transform duration-200 group-data-[state=open]:rotate-180"
+        aria-hidden="true"
+      />
+    </NavigationMenuPrimitive.Trigger>
+  )
+}
+
+function NavigationMenuContent({
+  className,
+  ...props
+}: React.ComponentProps<typeof NavigationMenuPrimitive.Content>) {
+  return (
+    <NavigationMenuPrimitive.Content
+      data-slot="navigation-menu-content"
+      className={cn(
+        'data-[motion^=from-]:animate-in data-[motion^=to-]:animate-out data-[motion^=from-]:fade-in data-[motion^=to-]:fade-out data-[motion=from-end]:slide-in-from-right-52 data-[motion=from-start]:slide-in-from-left-52 data-[motion=to-end]:slide-out-to-right-52 data-[motion=to-start]:slide-out-to-left-52',
+        'left-0 top-0 w-full md:absolute md:w-auto',
+        className
+      )}
+      {...props}
+    />
+  )
+}
+
+function NavigationMenuLink({
+  className,
+  ...props
+}: React.ComponentProps<typeof NavigationMenuPrimitive.Link>) {
+  return (
+    <NavigationMenuPrimitive.Link
+      data-slot="navigation-menu-link"
+      className={cn(navigationMenuTriggerStyle, className)}
+      {...props}
+    />
+  )
+}
+
+function NavigationMenuViewport({
+  className,
+  ...props
+}: React.ComponentProps<typeof NavigationMenuPrimitive.Viewport>) {
+  return (
+    <div className="absolute top-full left-0 flex justify-center">
+      <NavigationMenuPrimitive.Viewport
+        data-slot="navigation-menu-viewport"
+        className={cn(
+          'origin-top-center data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-90 relative mt-1.5 h-[var(--radix-navigation-menu-viewport-height)] w-full overflow-hidden rounded-xl border shadow-xl md:w-[var(--radix-navigation-menu-viewport-width)]',
+          'border-border-glass bg-popover backdrop-blur-xl',
+          'transition-[width,height] duration-300',
+          className
+        )}
+        {...props}
+      />
+    </div>
+  )
+}
+
+function NavigationMenuIndicator({
+  className,
+  ...props
+}: React.ComponentProps<typeof NavigationMenuPrimitive.Indicator>) {
+  return (
+    <NavigationMenuPrimitive.Indicator
+      data-slot="navigation-menu-indicator"
+      className={cn(
+        'data-[state=visible]:animate-in data-[state=hidden]:animate-out data-[state=hidden]:fade-out data-[state=visible]:fade-in top-full z-[1] flex h-1.5 items-end justify-center overflow-hidden',
+        className
+      )}
+      {...props}
+    >
+      <div className="bg-border-glass relative top-[60%] size-2 rotate-45 rounded-tl-sm shadow-md" />
+    </NavigationMenuPrimitive.Indicator>
+  )
+}
+
+export {
+  NavigationMenu,
+  NavigationMenuContent,
+  NavigationMenuIndicator,
+  NavigationMenuItem,
+  NavigationMenuLink,
+  NavigationMenuList,
+  NavigationMenuTrigger,
+  NavigationMenuViewport,
+  navigationMenuTriggerStyle,
+}

package/src/react/primitives/network-graph.tsx

@@ -0,0 +1,95 @@
+import { useEffect, useRef } from 'react'
+
+import { cn } from './utils'
+import { NetworkGraph as NetworkGraphVanilla } from '../../visualization/network-graph'
+import type {
+  GraphNode,
+  GraphEdge,
+  NetworkGraphConfig,
+} from '../../visualization/network-graph/types'
+
+export type { GraphNode, GraphEdge } from '../../visualization/network-graph/types'
+
+export interface NetworkGraphProps {
+  nodes: GraphNode[]
+  edges: GraphEdge[]
+  layout?: 'force' | 'circular' | 'grid'
+  zoomable?: boolean
+  pannable?: boolean
+  nodeRadius?: number
+  showLabels?: boolean
+  onNodeClick?: (node: GraphNode) => void
+  onEdgeClick?: (edge: GraphEdge) => void
+  className?: string
+}
+
+function NetworkGraph({
+  nodes,
+  edges,
+  layout = 'force',
+  zoomable = true,
+  pannable = true,
+  nodeRadius,
+  showLabels = true,
+  onNodeClick,
+  onEdgeClick,
+  className,
+}: NetworkGraphProps) {
+  const containerRef = useRef<HTMLDivElement>(null)
+  const instanceRef = useRef<NetworkGraphVanilla | null>(null)
+
+  const onNodeClickRef = useRef(onNodeClick)
+  onNodeClickRef.current = onNodeClick
+  const onEdgeClickRef = useRef(onEdgeClick)
+  onEdgeClickRef.current = onEdgeClick
+
+  useEffect(() => {
+    const container = containerRef.current
+    if (!container) return
+
+    const cfg: NetworkGraphConfig = {
+      container,
+      nodes,
+      edges,
+      layout,
+      zoomable,
+      pannable,
+      showLabels,
+      onNodeClick: node => onNodeClickRef.current?.(node),
+      onEdgeClick: edge => onEdgeClickRef.current?.(edge),
+    }
+    if (nodeRadius !== undefined) cfg.nodeRadius = nodeRadius
+
+    instanceRef.current = new NetworkGraphVanilla(cfg)
+
+    return () => {
+      instanceRef.current?.destroy()
+      instanceRef.current = null
+    }
+  }, [zoomable, pannable, nodeRadius, showLabels])
+
+  // Update data
+  useEffect(() => {
+    if (instanceRef.current) {
+      instanceRef.current.update(nodes, edges)
+    }
+  }, [nodes, edges])
+
+  // Update layout
+  useEffect(() => {
+    if (instanceRef.current) {
+      instanceRef.current.setLayout(layout)
+    }
+  }, [layout])
+
+  return (
+    <div
+      ref={containerRef}
+      data-slot="network-graph"
+      className={cn('relative h-full w-full overflow-hidden', className)}
+    />
+  )
+}
+NetworkGraph.displayName = 'NetworkGraph'
+
+export { NetworkGraph }