4runr-os 2.10.74 → 2.10.75

package/mk3-tui/src/ui/shield_dashboard.rs

@@ -0,0 +1,549 @@
+use crate::app::AppState;
+use ratatui::prelude::*;
+use ratatui::widgets::{Block, Borders, List, ListItem, Paragraph, Wrap};
+use serde_json::Value;
+
+const BRAND_PURPLE: Color = Color::Rgb(138, 43, 226);
+const CYBER_CYAN: Color = Color::Rgb(0, 255, 255);
+const NEON_GREEN: Color = Color::Rgb(57, 255, 20);
+const AMBER_WARN: Color = Color::Rgb(255, 191, 0);
+const SHIELD_ORANGE: Color = Color::Rgb(255, 140, 0);
+const TEXT_PRIMARY: Color = Color::Rgb(230, 230, 240);
+const TEXT_DIM: Color = Color::Rgb(100, 100, 120);
+const TEXT_MUTED: Color = Color::Rgb(60, 60, 80);
+const BG_PANEL: Color = Color::Rgb(18, 18, 25);
+
+const ABOUT_SHIELD: &[&str] = &[
+    "Shield is 4Runr's input/output safety layer on the Gateway run path.",
+    "",
+    "Production enforcement (every real run):",
+    "  • Input check before agent execution starts",
+    "  • Injection patterns → block (failed run)",
+    "  • PII in prompts → mask or block per policy",
+    "  • Output check after agent returns",
+    "",
+    "This screen shows live Gateway state — no test runs required.",
+    "Counters increment as real agent/API traffic is evaluated.",
+];
+
+#[derive(Debug, Clone)]
+pub struct ShieldBlockRow {
+    pub id: String,
+    pub name: String,
+    pub reason: String,
+    pub created_at: String,
+}
+
+#[derive(Debug, Clone)]
+pub struct ShieldDashboardState {
+    pub enabled: bool,
+    pub mode: String,
+    pub health_status: String,
+    pub pii_enabled: bool,
+    pub injection_enabled: bool,
+    pub injection_action: String,
+    pub hallucination_enabled: bool,
+    pub hallucination_action: String,
+    pub decisions_total: u64,
+    pub blocks_total: u64,
+    pub masks_total: u64,
+    pub rewrites_total: u64,
+    pub warnings: Vec<String>,
+    pub recent_blocks: Vec<ShieldBlockRow>,
+    pub loading: bool,
+    pub error: Option<String>,
+    pub last_refresh: Option<std::time::Instant>,
+    pub auto_refresh_enabled: bool,
+    pub auto_refresh_interval: std::time::Duration,
+}
+
+impl Default for ShieldDashboardState {
+    fn default() -> Self {
+        Self {
+            enabled: false,
+            mode: "off".to_string(),
+            health_status: String::new(),
+            pii_enabled: false,
+            injection_enabled: false,
+            injection_action: String::new(),
+            hallucination_enabled: false,
+            hallucination_action: String::new(),
+            decisions_total: 0,
+            blocks_total: 0,
+            masks_total: 0,
+            rewrites_total: 0,
+            warnings: Vec::new(),
+            recent_blocks: Vec::new(),
+            loading: false,
+            error: None,
+            last_refresh: None,
+            auto_refresh_enabled: true,
+            auto_refresh_interval: std::time::Duration::from_secs(5),
+        }
+    }
+}
+
+impl ShieldDashboardState {
+
+    pub fn apply_load(
+        &mut self,
+        health: &ShieldHealthSnap,
+        config: &ShieldConfigSnap,
+        metrics: &ShieldMetricsSnap,
+        warnings: Vec<String>,
+        recent: Vec<ShieldBlockRow>,
+    ) {
+        self.loading = false;
+        self.error = None;
+        self.enabled = health.enabled;
+        self.mode = health.mode.clone();
+        self.health_status = health.status.clone();
+        self.pii_enabled = config.pii_enabled;
+        self.injection_enabled = config.injection_enabled;
+        self.injection_action = config.injection_action.clone();
+        self.hallucination_enabled = config.hallucination_enabled;
+        self.hallucination_action = config.hallucination_action.clone();
+        self.decisions_total = metrics.decisions;
+        self.blocks_total = metrics.blocks;
+        self.masks_total = metrics.masks;
+        self.rewrites_total = metrics.rewrites;
+        self.warnings = warnings;
+        self.recent_blocks = recent;
+        self.last_refresh = Some(std::time::Instant::now());
+    }
+}
+
+#[derive(Debug, Clone, Default)]
+pub struct ShieldHealthSnap {
+    pub enabled: bool,
+    pub mode: String,
+    pub status: String,
+}
+
+#[derive(Debug, Clone, Default)]
+pub struct ShieldConfigSnap {
+    pub pii_enabled: bool,
+    pub injection_enabled: bool,
+    pub injection_action: String,
+    pub hallucination_enabled: bool,
+    pub hallucination_action: String,
+}
+
+#[derive(Debug, Clone, Default)]
+pub struct ShieldMetricsSnap {
+    pub decisions: u64,
+    pub blocks: u64,
+    pub masks: u64,
+    pub rewrites: u64,
+}
+
+pub fn parse_shield_load(
+    health_val: &Value,
+    config_val: &Value,
+    metrics_val: &Value,
+    gateway_health_val: &Value,
+    runs_val: &Value,
+) -> (
+    ShieldHealthSnap,
+    ShieldConfigSnap,
+    ShieldMetricsSnap,
+    Vec<String>,
+    Vec<ShieldBlockRow>,
+) {
+    let health_obj = health_val.as_object();
+    let health = ShieldHealthSnap {
+        enabled: health_obj
+            .and_then(|o| o.get("enabled"))
+            .and_then(|v| v.as_bool())
+            .unwrap_or(false),
+        mode: health_obj
+            .and_then(|o| o.get("mode"))
+            .and_then(|v| v.as_str())
+            .unwrap_or("off")
+            .to_string(),
+        status: if health_obj
+            .and_then(|o| o.get("enabled"))
+            .and_then(|v| v.as_bool())
+            .unwrap_or(false)
+        {
+            "active".to_string()
+        } else {
+            "disabled".to_string()
+        },
+    };
+
+    let cfg = config_val
+        .get("config")
+        .or(Some(config_val))
+        .and_then(|v| v.as_object());
+    let pii = cfg.and_then(|c| c.get("pii")).and_then(|v| v.as_object());
+    let injection = cfg.and_then(|c| c.get("injection")).and_then(|v| v.as_object());
+    let hallucination = cfg
+        .and_then(|c| c.get("hallucination"))
+        .and_then(|v| v.as_object());
+
+    let config = ShieldConfigSnap {
+        pii_enabled: pii
+            .and_then(|o| o.get("enabled"))
+            .and_then(|v| v.as_bool())
+            .unwrap_or(false),
+        injection_enabled: injection
+            .and_then(|o| o.get("enabled"))
+            .and_then(|v| v.as_bool())
+            .unwrap_or(false),
+        injection_action: injection
+            .and_then(|o| o.get("action"))
+            .and_then(|v| v.as_str())
+            .unwrap_or("—")
+            .to_string(),
+        hallucination_enabled: hallucination
+            .and_then(|o| o.get("enabled"))
+            .and_then(|v| v.as_bool())
+            .unwrap_or(false),
+        hallucination_action: hallucination
+            .and_then(|o| o.get("action"))
+            .and_then(|v| v.as_str())
+            .unwrap_or("—")
+            .to_string(),
+    };
+
+    let metrics = ShieldMetricsSnap {
+        decisions: metrics_val
+            .get("decisions")
+            .and_then(|v| v.as_u64())
+            .unwrap_or(0),
+        blocks: metrics_val
+            .get("blocks")
+            .and_then(|v| v.as_u64())
+            .unwrap_or(0),
+        masks: metrics_val
+            .get("masks")
+            .and_then(|v| v.as_u64())
+            .unwrap_or(0),
+        rewrites: metrics_val
+            .get("rewrites")
+            .and_then(|v| v.as_u64())
+            .unwrap_or(0),
+    };
+
+    let mut warnings: Vec<String> = Vec::new();
+    if let Some(sh) = gateway_health_val
+        .get("checks")
+        .and_then(|c| c.get("shield"))
+        .and_then(|v| v.as_object())
+    {
+        if let Some(st) = sh.get("status").and_then(|v| v.as_str()) {
+            if st != "ok" && st != "up" {
+                warnings.push(format!("Gateway health: {}", st));
+            }
+        }
+        if let Some(arr) = sh.get("warnings").and_then(|v| v.as_array()) {
+            for w in arr {
+                if let Some(s) = w.as_str() {
+                    warnings.push(s.to_string());
+                }
+            }
+        }
+    }
+
+    let mut recent: Vec<ShieldBlockRow> = Vec::new();
+    let runs_arr = runs_val
+        .as_array()
+        .or_else(|| runs_val.get("runs").and_then(|v| v.as_array()));
+    if let Some(arr) = runs_arr {
+        for item in arr {
+            let o = match item.as_object() {
+                Some(o) => o,
+                None => continue,
+            };
+            let output = o.get("output").and_then(|v| v.as_object());
+            let err = output
+                .and_then(|oo| oo.get("error"))
+                .and_then(|v| v.as_str())
+                .unwrap_or("");
+            let logs_shield = o
+                .get("logs")
+                .and_then(|l| l.as_array())
+                .map(|arr| {
+                    arr.iter().any(|e| {
+                        e.get("message")
+                            .and_then(|m| m.as_str())
+                            .map(|s| s.to_lowercase().contains("shield"))
+                            .unwrap_or(false)
+                    })
+                })
+                .unwrap_or(false);
+            if !err.to_lowercase().contains("shield") && !logs_shield {
+                continue;
+            }
+            let id = o
+                .get("id")
+                .and_then(|v| v.as_str())
+                .unwrap_or("?")
+                .to_string();
+            let name = o
+                .get("name")
+                .and_then(|v| v.as_str())
+                .unwrap_or("Run")
+                .to_string();
+            let reason = output
+                .and_then(|oo| oo.get("reason"))
+                .and_then(|v| v.as_str())
+                .unwrap_or(err)
+                .to_string();
+            let created_at = o
+                .get("createdAt")
+                .and_then(|v| v.as_str())
+                .map(|s| s.chars().take(19).collect())
+                .unwrap_or_else(|| "—".to_string());
+            recent.push(ShieldBlockRow {
+                id,
+                name,
+                reason,
+                created_at,
+            });
+        }
+    }
+    recent.truncate(12);
+
+    (health, config, metrics, warnings, recent)
+}
+
+pub fn render(f: &mut Frame, state: &AppState) {
+    let area = f.size();
+    let sd = &state.shield_dashboard;
+
+    use ratatui::layout::{Constraint, Direction, Layout};
+
+    let live = if sd.auto_refresh_enabled { " · LIVE" } else { "" };
+    let header_title = format!(
+        " 🛡️  Shield — {} · {} block(s) · {} mask(s){} ",
+        sd.mode.to_uppercase(),
+        sd.blocks_total,
+        sd.masks_total,
+        live
+    );
+
+    let chunks = Layout::default()
+        .direction(Direction::Vertical)
+        .constraints([
+            Constraint::Length(3),
+            Constraint::Min(10),
+            Constraint::Length(8),
+            Constraint::Length(3),
+        ])
+        .split(area);
+
+    let header = Block::default()
+        .title(header_title)
+        .borders(Borders::ALL)
+        .border_style(Style::default().fg(SHIELD_ORANGE))
+        .style(Style::default().bg(BG_PANEL));
+    f.render_widget(header, chunks[0]);
+
+    let mid = Layout::default()
+        .direction(Direction::Horizontal)
+        .constraints([Constraint::Percentage(50), Constraint::Percentage(50)])
+        .split(chunks[1]);
+
+    render_status_panel(f, mid[0], sd);
+    render_metrics_panel(f, mid[1], sd);
+    render_enforcement_panel(f, chunks[2], sd);
+
+    let footer_msg = if sd.loading {
+        "Loading live Shield state from Gateway…".to_string()
+    } else if let Some(e) = &sd.error {
+        e.clone()
+    } else {
+        "R Refresh · A Live toggle · ESC Close".to_string()
+    };
+    let footer_color = if sd.error.is_some() {
+        Color::Rgb(255, 69, 69)
+    } else {
+        TEXT_DIM
+    };
+    let footer = Block::default()
+        .title(" ⌨️  Actions ")
+        .borders(Borders::ALL)
+        .border_style(Style::default().fg(TEXT_MUTED));
+    let inner = footer.inner(chunks[3]);
+    f.render_widget(footer, chunks[3]);
+    f.render_widget(
+        Paragraph::new(footer_msg)
+            .style(Style::default().fg(footer_color))
+            .alignment(Alignment::Center),
+        inner,
+    );
+}
+
+fn render_status_panel(f: &mut Frame, area: Rect, sd: &ShieldDashboardState) {
+    let block = Block::default()
+        .title(" Status & detectors ")
+        .borders(Borders::ALL)
+        .border_style(Style::default().fg(CYBER_CYAN))
+        .style(Style::default().bg(BG_PANEL));
+    let inner = block.inner(area);
+    f.render_widget(block, area);
+
+    let mode_color = match sd.mode.as_str() {
+        "enforce" => NEON_GREEN,
+        "monitor" => AMBER_WARN,
+        _ => TEXT_MUTED,
+    };
+
+    let mut lines = vec![
+        Line::from(vec![
+            Span::styled("Enabled:  ", Style::default().fg(TEXT_DIM)),
+            Span::styled(
+                if sd.enabled { "YES" } else { "NO" },
+                Style::default()
+                    .fg(if sd.enabled { NEON_GREEN } else { TEXT_MUTED })
+                    .bold(),
+            ),
+        ]),
+        Line::from(vec![
+            Span::styled("Mode:     ", Style::default().fg(TEXT_DIM)),
+            Span::styled(
+                sd.mode.to_uppercase(),
+                Style::default().fg(mode_color).bold(),
+            ),
+        ]),
+        Line::from(""),
+        Line::from("Detectors (production path):").style(Style::default().fg(TEXT_DIM)),
+        detector_line("PII", sd.pii_enabled, "mask on input/output".to_string()),
+        detector_line(
+            "Injection",
+            sd.injection_enabled,
+            format!("action: {}", sd.injection_action),
+        ),
+        detector_line(
+            "Hallucination",
+            sd.hallucination_enabled,
+            format!("action: {}", sd.hallucination_action),
+        ),
+    ];
+
+    if !sd.warnings.is_empty() {
+        lines.push(Line::from(""));
+        lines.push(
+            Line::from("Warnings:")
+                .style(Style::default().fg(AMBER_WARN).bold()),
+        );
+        for w in &sd.warnings {
+            lines.push(Line::from(format!("  ⚠ {}", w)).style(Style::default().fg(AMBER_WARN)));
+        }
+    }
+
+    lines.push(Line::from(""));
+    for line in ABOUT_SHIELD.iter().take(4) {
+        lines.push(Line::from(*line).style(Style::default().fg(TEXT_MUTED)));
+    }
+
+    f.render_widget(Paragraph::new(lines).wrap(Wrap { trim: false }), inner);
+}
+
+fn detector_line(name: &str, on: bool, detail: String) -> Line<'static> {
+    Line::from(vec![
+        Span::raw("  "),
+        Span::styled(
+            if on { "●" } else { "○" },
+            Style::default().fg(if on { NEON_GREEN } else { TEXT_MUTED }),
+        ),
+        Span::styled(format!(" {:<14}", name), Style::default().fg(TEXT_PRIMARY)),
+        Span::styled(detail, Style::default().fg(TEXT_DIM)),
+    ])
+}
+
+fn render_metrics_panel(f: &mut Frame, area: Rect, sd: &ShieldDashboardState) {
+    let block = Block::default()
+        .title(" Gateway metrics (cumulative) ")
+        .borders(Borders::ALL)
+        .border_style(Style::default().fg(BRAND_PURPLE))
+        .style(Style::default().bg(BG_PANEL));
+    let inner = block.inner(area);
+    f.render_widget(block, area);
+
+    let lines = vec![
+        Line::from(""),
+        metric_line("Decisions total", sd.decisions_total, TEXT_PRIMARY),
+        metric_line("Blocks total", sd.blocks_total, SHIELD_ORANGE),
+        metric_line("Masks total", sd.masks_total, CYBER_CYAN),
+        metric_line("Rewrites total", sd.rewrites_total, AMBER_WARN),
+        Line::from(""),
+        Line::from("Source: Gateway Prometheus /metrics").style(Style::default().fg(TEXT_MUTED)),
+        Line::from("Updates on refresh — counts all production runs.").style(Style::default().fg(TEXT_MUTED)),
+        Line::from(""),
+        Line::from(ABOUT_SHIELD[4]).style(Style::default().fg(TEXT_DIM)),
+        Line::from(ABOUT_SHIELD[5]).style(Style::default().fg(TEXT_DIM)),
+        Line::from(ABOUT_SHIELD[6]).style(Style::default().fg(TEXT_DIM)),
+        Line::from(ABOUT_SHIELD[7]).style(Style::default().fg(TEXT_DIM)),
+    ];
+
+    f.render_widget(Paragraph::new(lines), inner);
+}
+
+fn metric_line(label: &str, value: u64, color: Color) -> Line<'static> {
+    Line::from(vec![
+        Span::styled(format!("  {:<18}", label), Style::default().fg(TEXT_DIM)),
+        Span::styled(
+            value.to_string(),
+            Style::default().fg(color).bold(),
+        ),
+    ])
+}
+
+fn render_enforcement_panel(f: &mut Frame, area: Rect, sd: &ShieldDashboardState) {
+    let block = Block::default()
+        .title(format!(
+            " Recent enforcement ({} blocked run(s) in history) ",
+            sd.recent_blocks.len()
+        ))
+        .borders(Borders::ALL)
+        .border_style(Style::default().fg(SHIELD_ORANGE))
+        .style(Style::default().bg(BG_PANEL));
+    let inner = block.inner(area);
+    f.render_widget(block, area);
+
+    if sd.recent_blocks.is_empty() {
+        let empty = vec![
+            Line::from(""),
+            Line::from("No blocked runs in Gateway history yet.").style(Style::default().fg(TEXT_DIM)),
+            Line::from("Shield is active on every run — blocks appear here when production").style(Style::default().fg(TEXT_MUTED)),
+            Line::from("traffic triggers injection/PII policy (failed status + SHIELD in Run Manager).").style(Style::default().fg(TEXT_MUTED)),
+        ];
+        f.render_widget(Paragraph::new(empty), inner);
+        return;
+    }
+
+    let items: Vec<ListItem> = sd
+        .recent_blocks
+        .iter()
+        .map(|r| {
+            let short_id = if r.id.len() > 8 {
+                &r.id[r.id.len() - 8..]
+            } else {
+                &r.id
+            };
+            ListItem::new(Line::from(vec![
+                Span::styled(
+                    format!("{} ", r.created_at),
+                    Style::default().fg(TEXT_MUTED),
+                ),
+                Span::styled(
+                    format!("[{}] ", short_id),
+                    Style::default().fg(CYBER_CYAN),
+                ),
+                Span::styled(
+                    format!("{} — ", r.name),
+                    Style::default().fg(TEXT_PRIMARY),
+                ),
+                Span::styled(
+                    r.reason.chars().take(48).collect::<String>(),
+                    Style::default().fg(SHIELD_ORANGE),
+                ),
+            ]))
+        })
+        .collect();
+
+    f.render_widget(List::new(items), inner);
+}

package/package.json

@@ -1,8 +1,8 @@
 {
   "name": "4runr-os",
-  "version": "2.10.74",
+  "version": "2.10.75",
   "type": "module",
-  "description": "4Runr AI Agent OS - Secure terminal interface for AI agents. v2.10.74: Shield visible in Run Manager, System Status, monitoring; shield demo/probe/status commands; stronger tools smoke.",
+  "description": "4Runr AI Agent OS - Secure terminal interface for AI agents. v2.10.75: Dedicated Shield dashboard (shield command) — live metrics, detectors, enforcement history.",
   "main": "dist/index.js",
   "bin": {
     "4runr": "dist/index.js",