agent ui: live event panel via SSE + stream-json
This commit is contained in:
parent
3c9d42b2a7
commit
9eab28a716
8 changed files with 277 additions and 33 deletions
18
CLAUDE.md
18
CLAUDE.md
|
|
@ -182,6 +182,24 @@ and the MCP tools. Claude drives any further `recv`/`send` itself —
|
||||||
harness no longer relays claude's stdout as a reply. Stdout is logged for
|
harness no longer relays claude's stdout as a reply. Stdout is logged for
|
||||||
debugging; the side effects (sends via MCP) are what matter.
|
debugging; the side effects (sends via MCP) are what matter.
|
||||||
|
|
||||||
|
**Live view.** Each agent runs a `hive_ag3nt::events::Bus` (a
|
||||||
|
`tokio::sync::broadcast<LiveEvent>` wrapper). The harness emits:
|
||||||
|
- `TurnStart { from, body }` when a wake-up message is popped.
|
||||||
|
- `Stream(value)` for every line claude prints on stdout (parsed
|
||||||
|
stream-json; flattened under `{kind: "stream", type: ...}` via serde
|
||||||
|
internal tagging).
|
||||||
|
- `Note(text)` for stderr lines and non-JSON stdout (so nothing's lost).
|
||||||
|
- `TurnEnd { ok, note }` when claude exits.
|
||||||
|
|
||||||
|
The web UI subscribes via `/events/stream` (SSE) and a small JS panel on
|
||||||
|
`/` appends rows. No full-page reload — the login form (and anything else
|
||||||
|
the operator is typing into) stays put.
|
||||||
|
|
||||||
|
claude is invoked with `--print --verbose --output-format stream-json` so
|
||||||
|
tool calls + assistant text + tool results all land as structured events.
|
||||||
|
The harness no longer reads claude's text stdout into a reply; claude
|
||||||
|
calls `mcp__hyperhive__send` itself.
|
||||||
|
|
||||||
**Tool envelope.** Every MCP tool handler in `hive_ag3nt::mcp::AgentServer`
|
**Tool envelope.** Every MCP tool handler in `hive_ag3nt::mcp::AgentServer`
|
||||||
wraps its logic in `run_tool(name, args_debug, async { ... })`. The
|
wraps its logic in `run_tool(name, args_debug, async { ... })`. The
|
||||||
envelope guarantees:
|
envelope guarantees:
|
||||||
|
|
|
||||||
1
Cargo.lock
generated
1
Cargo.lock
generated
|
|
@ -457,6 +457,7 @@ dependencies = [
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"tokio",
|
"tokio",
|
||||||
|
"tokio-stream",
|
||||||
"tracing",
|
"tracing",
|
||||||
"tracing-subscriber",
|
"tracing-subscriber",
|
||||||
]
|
]
|
||||||
|
|
|
||||||
|
|
@ -16,6 +16,7 @@ schemars.workspace = true
|
||||||
serde.workspace = true
|
serde.workspace = true
|
||||||
serde_json.workspace = true
|
serde_json.workspace = true
|
||||||
tokio.workspace = true
|
tokio.workspace = true
|
||||||
|
tokio-stream.workspace = true
|
||||||
tracing.workspace = true
|
tracing.workspace = true
|
||||||
tracing-subscriber.workspace = true
|
tracing-subscriber.workspace = true
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,12 +1,15 @@
|
||||||
use std::path::{Path, PathBuf};
|
use std::path::{Path, PathBuf};
|
||||||
|
use std::process::Stdio;
|
||||||
use std::sync::{Arc, Mutex};
|
use std::sync::{Arc, Mutex};
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
use anyhow::{Result, bail};
|
use anyhow::{Result, bail};
|
||||||
use clap::{Parser, Subcommand};
|
use clap::{Parser, Subcommand};
|
||||||
|
use hive_ag3nt::events::{Bus, LiveEvent};
|
||||||
use hive_ag3nt::login::{self, LoginState};
|
use hive_ag3nt::login::{self, LoginState};
|
||||||
use hive_ag3nt::{DEFAULT_SOCKET, DEFAULT_WEB_PORT, client, mcp, web_ui};
|
use hive_ag3nt::{DEFAULT_SOCKET, DEFAULT_WEB_PORT, client, mcp, web_ui};
|
||||||
use hive_sh4re::{AgentRequest, AgentResponse};
|
use hive_sh4re::{AgentRequest, AgentResponse};
|
||||||
|
use tokio::io::{AsyncBufReadExt, BufReader};
|
||||||
use tokio::process::Command;
|
use tokio::process::Command;
|
||||||
|
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
|
|
@ -61,14 +64,16 @@ async fn main() -> Result<()> {
|
||||||
tracing::info!(state = ?initial, claude_dir = %claude_dir.display(), "harness boot");
|
tracing::info!(state = ?initial, claude_dir = %claude_dir.display(), "harness boot");
|
||||||
let login_state = Arc::new(Mutex::new(initial));
|
let login_state = Arc::new(Mutex::new(initial));
|
||||||
let ui_state = login_state.clone();
|
let ui_state = login_state.clone();
|
||||||
|
let bus = Bus::new();
|
||||||
|
let ui_bus = bus.clone();
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
if let Err(e) = web_ui::serve(label, port, ui_state).await {
|
if let Err(e) = web_ui::serve(label, port, ui_state, ui_bus).await {
|
||||||
tracing::error!(error = ?e, "web ui failed");
|
tracing::error!(error = ?e, "web ui failed");
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
match initial {
|
match initial {
|
||||||
LoginState::Online => {
|
LoginState::Online => {
|
||||||
serve(&cli.socket, Duration::from_millis(poll_ms), login_state).await
|
serve(&cli.socket, Duration::from_millis(poll_ms), login_state, bus).await
|
||||||
}
|
}
|
||||||
LoginState::NeedsLogin => {
|
LoginState::NeedsLogin => {
|
||||||
// Partial-run mode: keep the harness alive (so the web UI
|
// Partial-run mode: keep the harness alive (so the web UI
|
||||||
|
|
@ -77,7 +82,7 @@ async fn main() -> Result<()> {
|
||||||
// from the dashboard PTY path in step 4 or via
|
// from the dashboard PTY path in step 4 or via
|
||||||
// `root-login` + `claude auth login` in the meantime)
|
// `root-login` + `claude auth login` in the meantime)
|
||||||
// transitions us into the turn loop without a restart.
|
// transitions us into the turn loop without a restart.
|
||||||
needs_login_loop(&cli.socket, &claude_dir, login_state, poll_ms).await
|
needs_login_loop(&cli.socket, &claude_dir, login_state, poll_ms, bus).await
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -103,6 +108,7 @@ async fn needs_login_loop(
|
||||||
claude_dir: &Path,
|
claude_dir: &Path,
|
||||||
state: Arc<Mutex<LoginState>>,
|
state: Arc<Mutex<LoginState>>,
|
||||||
poll_ms: u64,
|
poll_ms: u64,
|
||||||
|
bus: Bus,
|
||||||
) -> Result<()> {
|
) -> Result<()> {
|
||||||
tracing::warn!(
|
tracing::warn!(
|
||||||
claude_dir = %claude_dir.display(),
|
claude_dir = %claude_dir.display(),
|
||||||
|
|
@ -114,12 +120,17 @@ async fn needs_login_loop(
|
||||||
if login::has_session(claude_dir) {
|
if login::has_session(claude_dir) {
|
||||||
tracing::info!("claude session detected — entering turn loop");
|
tracing::info!("claude session detected — entering turn loop");
|
||||||
*state.lock().unwrap() = LoginState::Online;
|
*state.lock().unwrap() = LoginState::Online;
|
||||||
return serve(socket, Duration::from_millis(poll_ms), state).await;
|
return serve(socket, Duration::from_millis(poll_ms), state, bus).await;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn serve(socket: &Path, interval: Duration, state: Arc<Mutex<LoginState>>) -> Result<()> {
|
async fn serve(
|
||||||
|
socket: &Path,
|
||||||
|
interval: Duration,
|
||||||
|
state: Arc<Mutex<LoginState>>,
|
||||||
|
bus: Bus,
|
||||||
|
) -> Result<()> {
|
||||||
tracing::info!(socket = %socket.display(), "hive-ag3nt serve");
|
tracing::info!(socket = %socket.display(), "hive-ag3nt serve");
|
||||||
let _ = state; // reserved for future state transitions (turn-loop -> needs-login)
|
let _ = state; // reserved for future state transitions (turn-loop -> needs-login)
|
||||||
let mcp_config = write_mcp_config(socket).await?;
|
let mcp_config = write_mcp_config(socket).await?;
|
||||||
|
|
@ -129,10 +140,28 @@ async fn serve(socket: &Path, interval: Duration, state: Arc<Mutex<LoginState>>)
|
||||||
match recv {
|
match recv {
|
||||||
Ok(AgentResponse::Message { from, body }) => {
|
Ok(AgentResponse::Message { from, body }) => {
|
||||||
tracing::info!(%from, %body, "inbox");
|
tracing::info!(%from, %body, "inbox");
|
||||||
|
bus.emit(LiveEvent::TurnStart {
|
||||||
|
from: from.clone(),
|
||||||
|
body: body.clone(),
|
||||||
|
});
|
||||||
let prompt = format_wake_prompt(&label, &from, &body);
|
let prompt = format_wake_prompt(&label, &from, &body);
|
||||||
match invoke_claude(&prompt, &mcp_config).await {
|
let outcome = run_turn(&prompt, &mcp_config, &bus).await;
|
||||||
Ok(out) => tracing::info!(stdout = %out.trim(), "claude turn finished"),
|
match outcome {
|
||||||
Err(e) => tracing::warn!(error = %format!("{e:#}"), "claude turn failed"),
|
Ok(()) => {
|
||||||
|
bus.emit(LiveEvent::TurnEnd {
|
||||||
|
ok: true,
|
||||||
|
note: None,
|
||||||
|
});
|
||||||
|
tracing::info!("claude turn finished");
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
let note = format!("{e:#}");
|
||||||
|
bus.emit(LiveEvent::TurnEnd {
|
||||||
|
ok: false,
|
||||||
|
note: Some(note.clone()),
|
||||||
|
});
|
||||||
|
tracing::warn!(error = %note, "claude turn failed");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Ok(AgentResponse::Empty) => {}
|
Ok(AgentResponse::Empty) => {}
|
||||||
|
|
@ -173,15 +202,20 @@ fn format_wake_prompt(label: &str, from: &str, body: &str) -> String {
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn invoke_claude(prompt: &str, mcp_config: &Path) -> Result<String> {
|
/// Spawn `claude` for one turn and stream its `stream-json` stdout into
|
||||||
// Whitelist model: `--tools` restricts which built-ins exist in the
|
/// the live event bus. `--verbose` is required by claude-code when pairing
|
||||||
// session (omitting WebFetch/WebSearch/Task means claude literally
|
/// `--print` with `--output-format stream-json`. Each stdout line is one
|
||||||
// can't invoke them); `--allowedTools` auto-approves the same set
|
/// JSON event; we broadcast the parsed value (or a `Note` fallback on
|
||||||
// plus the hyperhive MCP surface so there's no permission prompt
|
/// parse error so the UI doesn't silently lose information). The tool
|
||||||
// mid-turn. A finer-grained allow-list system for Bash command
|
/// whitelist is the same as before: omit WebFetch/WebSearch/Task; allow
|
||||||
// patterns is on the backlog (PLAN.md polish).
|
/// the hyperhive MCP surface auto-approved. Bash pattern allow-list is on
|
||||||
let out = Command::new("claude")
|
/// the backlog (CLAUDE.md).
|
||||||
|
async fn run_turn(prompt: &str, mcp_config: &Path, bus: &Bus) -> Result<()> {
|
||||||
|
let mut child = Command::new("claude")
|
||||||
.arg("--print")
|
.arg("--print")
|
||||||
|
.arg("--verbose")
|
||||||
|
.arg("--output-format")
|
||||||
|
.arg("stream-json")
|
||||||
.arg("--model")
|
.arg("--model")
|
||||||
.arg("haiku")
|
.arg("haiku")
|
||||||
.arg("--mcp-config")
|
.arg("--mcp-config")
|
||||||
|
|
@ -191,20 +225,38 @@ async fn invoke_claude(prompt: &str, mcp_config: &Path) -> Result<String> {
|
||||||
.arg("--allowedTools")
|
.arg("--allowedTools")
|
||||||
.arg(mcp::allowed_tools_arg())
|
.arg(mcp::allowed_tools_arg())
|
||||||
.arg(prompt)
|
.arg(prompt)
|
||||||
.output()
|
.stdout(Stdio::piped())
|
||||||
.await?;
|
.stderr(Stdio::piped())
|
||||||
if !out.status.success() {
|
.spawn()?;
|
||||||
bail!(
|
|
||||||
"claude exited {}: {}",
|
let stdout = child.stdout.take().expect("piped stdout");
|
||||||
out.status,
|
let stderr = child.stderr.take().expect("piped stderr");
|
||||||
String::from_utf8_lossy(&out.stderr).trim()
|
|
||||||
);
|
let bus_out = bus.clone();
|
||||||
|
let bus_err = bus.clone();
|
||||||
|
let pump_stdout = tokio::spawn(async move {
|
||||||
|
let mut reader = BufReader::new(stdout).lines();
|
||||||
|
while let Ok(Some(line)) = reader.next_line().await {
|
||||||
|
match serde_json::from_str::<serde_json::Value>(&line) {
|
||||||
|
Ok(v) => bus_out.emit(LiveEvent::Stream(v)),
|
||||||
|
Err(_) => bus_out.emit(LiveEvent::Note(format!("(non-json) {line}"))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
|
let pump_stderr = tokio::spawn(async move {
|
||||||
|
let mut reader = BufReader::new(stderr).lines();
|
||||||
|
while let Ok(Some(line)) = reader.next_line().await {
|
||||||
|
bus_err.emit(LiveEvent::Note(format!("stderr: {line}")));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
let status = child.wait().await?;
|
||||||
|
let _ = pump_stdout.await;
|
||||||
|
let _ = pump_stderr.await;
|
||||||
|
if !status.success() {
|
||||||
|
bail!("claude exited {status}");
|
||||||
}
|
}
|
||||||
let text = String::from_utf8_lossy(&out.stdout).trim().to_owned();
|
Ok(())
|
||||||
if text.is_empty() {
|
|
||||||
bail!("claude produced empty output");
|
|
||||||
}
|
|
||||||
Ok(text)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Drop the per-agent MCP config on disk so the turn loop can hand its path
|
/// Drop the per-agent MCP config on disk so the turn loop can hand its path
|
||||||
|
|
|
||||||
|
|
@ -9,6 +9,7 @@ use std::time::Duration;
|
||||||
|
|
||||||
use anyhow::{Result, bail};
|
use anyhow::{Result, bail};
|
||||||
use clap::{Parser, Subcommand};
|
use clap::{Parser, Subcommand};
|
||||||
|
use hive_ag3nt::events::Bus;
|
||||||
use hive_ag3nt::login::{self, LoginState};
|
use hive_ag3nt::login::{self, LoginState};
|
||||||
use hive_ag3nt::{DEFAULT_SOCKET, DEFAULT_WEB_PORT, client, web_ui};
|
use hive_ag3nt::{DEFAULT_SOCKET, DEFAULT_WEB_PORT, client, web_ui};
|
||||||
use hive_sh4re::{HelperEvent, ManagerRequest, ManagerResponse, SYSTEM_SENDER};
|
use hive_sh4re::{HelperEvent, ManagerRequest, ManagerResponse, SYSTEM_SENDER};
|
||||||
|
|
@ -66,11 +67,14 @@ async fn main() -> Result<()> {
|
||||||
tracing::info!(state = ?initial, claude_dir = %claude_dir.display(), "hm1nd boot");
|
tracing::info!(state = ?initial, claude_dir = %claude_dir.display(), "hm1nd boot");
|
||||||
let login_state = Arc::new(Mutex::new(initial));
|
let login_state = Arc::new(Mutex::new(initial));
|
||||||
let ui_state = login_state.clone();
|
let ui_state = login_state.clone();
|
||||||
|
let bus = Bus::new();
|
||||||
|
let ui_bus = bus.clone();
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
if let Err(e) = web_ui::serve(label, port, ui_state).await {
|
if let Err(e) = web_ui::serve(label, port, ui_state, ui_bus).await {
|
||||||
tracing::error!(error = ?e, "web ui failed");
|
tracing::error!(error = ?e, "web ui failed");
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
let _ = bus; // manager turn loop not wired to events yet
|
||||||
match initial {
|
match initial {
|
||||||
LoginState::Online => serve(&cli.socket, Duration::from_millis(poll_ms)).await,
|
LoginState::Online => serve(&cli.socket, Duration::from_millis(poll_ms)).await,
|
||||||
LoginState::NeedsLogin => {
|
LoginState::NeedsLogin => {
|
||||||
|
|
|
||||||
63
hive-ag3nt/src/events.rs
Normal file
63
hive-ag3nt/src/events.rs
Normal file
|
|
@ -0,0 +1,63 @@
|
||||||
|
//! Live event stream for the per-agent web UI. The harness emits one
|
||||||
|
//! `LiveEvent` per interesting thing that happens during a turn — wake-up
|
||||||
|
//! (the popped inbox message), every line claude prints on stdout
|
||||||
|
//! (parsed from `--output-format stream-json`), and the turn-end summary.
|
||||||
|
//! The web UI subscribes via SSE and renders rows live.
|
||||||
|
//!
|
||||||
|
//! Channel type is `tokio::sync::broadcast`. New subscribers see only
|
||||||
|
//! future events; the dashboard JS deals with the cold-start case by
|
||||||
|
//! showing "connecting…" until the first event arrives.
|
||||||
|
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use tokio::sync::broadcast;
|
||||||
|
|
||||||
|
const CHANNEL_CAPACITY: usize = 256;
|
||||||
|
|
||||||
|
/// One row of the agent's live stream. Serialised to JSON for SSE delivery.
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||||
|
#[serde(tag = "kind", rename_all = "snake_case")]
|
||||||
|
pub enum LiveEvent {
|
||||||
|
/// Harness popped a wake-up message and is about to invoke claude.
|
||||||
|
TurnStart { from: String, body: String },
|
||||||
|
/// One line of claude's `--output-format stream-json` stdout, parsed as
|
||||||
|
/// a generic JSON value (so we don't have to track every claude-code
|
||||||
|
/// event variant). The frontend pretty-prints by `type` field.
|
||||||
|
Stream(serde_json::Value),
|
||||||
|
/// Free-form note from the harness (e.g. "claude exited 0",
|
||||||
|
/// "stream-json parse error: ..."). Useful when stream-json itself
|
||||||
|
/// fails so the UI doesn't just go silent.
|
||||||
|
Note(String),
|
||||||
|
/// Turn finished. `ok=false` means claude exited non-zero or the
|
||||||
|
/// harness hit a transport error.
|
||||||
|
TurnEnd { ok: bool, note: Option<String> },
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Bus {
|
||||||
|
tx: Arc<broadcast::Sender<LiveEvent>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Bus {
|
||||||
|
#[must_use]
|
||||||
|
pub fn new() -> Self {
|
||||||
|
let (tx, _) = broadcast::channel(CHANNEL_CAPACITY);
|
||||||
|
Self { tx: Arc::new(tx) }
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn emit(&self, event: LiveEvent) {
|
||||||
|
// Lagged subscribers drop events — fine; the UI is a tail, not a log.
|
||||||
|
let _ = self.tx.send(event);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn subscribe(&self) -> broadcast::Receiver<LiveEvent> {
|
||||||
|
self.tx.subscribe()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Bus {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -2,6 +2,7 @@
|
||||||
//! `hive-m1nd` (manager) binaries.
|
//! `hive-m1nd` (manager) binaries.
|
||||||
|
|
||||||
pub mod client;
|
pub mod client;
|
||||||
|
pub mod events;
|
||||||
pub mod login;
|
pub mod login;
|
||||||
pub mod login_session;
|
pub mod login_session;
|
||||||
pub mod mcp;
|
pub mod mcp;
|
||||||
|
|
|
||||||
|
|
@ -3,6 +3,7 @@
|
||||||
//! each instance must bind a distinct port. `HIVE_PORT` is set per agent by
|
//! each instance must bind a distinct port. `HIVE_PORT` is set per agent by
|
||||||
//! `hive-c0re`'s generated per-agent flake (deterministic from agent name).
|
//! `hive-c0re`'s generated per-agent flake (deterministic from agent name).
|
||||||
|
|
||||||
|
use std::convert::Infallible;
|
||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
use std::sync::{Arc, Mutex};
|
use std::sync::{Arc, Mutex};
|
||||||
|
|
||||||
|
|
@ -10,11 +11,16 @@ use anyhow::{Context, Result};
|
||||||
use axum::{
|
use axum::{
|
||||||
Form, Router,
|
Form, Router,
|
||||||
extract::State,
|
extract::State,
|
||||||
response::{Html, IntoResponse, Redirect, Response},
|
response::{
|
||||||
|
Html, IntoResponse, Redirect, Response,
|
||||||
|
sse::{Event, KeepAlive, Sse},
|
||||||
|
},
|
||||||
routing::{get, post},
|
routing::{get, post},
|
||||||
};
|
};
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
|
use tokio_stream::{Stream, StreamExt, wrappers::BroadcastStream};
|
||||||
|
|
||||||
|
use crate::events::Bus;
|
||||||
use crate::login::LoginState;
|
use crate::login::LoginState;
|
||||||
use crate::login_session::{LoginSession, drop_if_finished};
|
use crate::login_session::{LoginSession, drop_if_finished};
|
||||||
|
|
||||||
|
|
@ -28,16 +34,19 @@ struct AppState {
|
||||||
label: String,
|
label: String,
|
||||||
login: LoginStateCell,
|
login: LoginStateCell,
|
||||||
session: Arc<Mutex<Option<Arc<LoginSession>>>>,
|
session: Arc<Mutex<Option<Arc<LoginSession>>>>,
|
||||||
|
bus: Bus,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn serve(label: String, port: u16, login: LoginStateCell) -> Result<()> {
|
pub async fn serve(label: String, port: u16, login: LoginStateCell, bus: Bus) -> Result<()> {
|
||||||
let state = AppState {
|
let state = AppState {
|
||||||
label,
|
label,
|
||||||
login,
|
login,
|
||||||
session: Arc::new(Mutex::new(None)),
|
session: Arc::new(Mutex::new(None)),
|
||||||
|
bus,
|
||||||
};
|
};
|
||||||
let app = Router::new()
|
let app = Router::new()
|
||||||
.route("/", get(index))
|
.route("/", get(index))
|
||||||
|
.route("/events/stream", get(events_stream))
|
||||||
.route("/login/start", post(post_login_start))
|
.route("/login/start", post(post_login_start))
|
||||||
.route("/login/code", post(post_login_code))
|
.route("/login/code", post(post_login_code))
|
||||||
.route("/login/cancel", post(post_login_cancel))
|
.route("/login/cancel", post(post_login_cancel))
|
||||||
|
|
@ -67,9 +76,87 @@ async fn index(State(state): State<AppState>) -> Html<String> {
|
||||||
}
|
}
|
||||||
|
|
||||||
fn render_online() -> String {
|
fn render_online() -> String {
|
||||||
"<p class=\"status-online\">▓█▓▒░ harness alive — turn loop running ▓█▓▒░</p>\n<p class=\"meta\">phase 6a placeholder — turn-loop status / inbox / xterm.js coming in 6b+</p>".into()
|
format!(
|
||||||
|
"<p class=\"status-online\">▓█▓▒░ harness alive — turn loop running ▓█▓▒░</p>\n{LIVE_PANEL}",
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Live event tail rendered into every `/` response when the agent is online.
|
||||||
|
/// JS opens an `EventSource` on `/events/stream` and appends rows; no full-page
|
||||||
|
/// reload, so the login flow and other forms aren't clobbered.
|
||||||
|
const LIVE_PANEL: &str = r#"
|
||||||
|
<h3>live</h3>
|
||||||
|
<pre id="live" class="diff"><span class="meta">connecting…</span></pre>
|
||||||
|
<script>
|
||||||
|
(function() {
|
||||||
|
const log = document.getElementById('live');
|
||||||
|
function appendRow(text, cls) {
|
||||||
|
log.innerHTML = '';
|
||||||
|
const row = document.createElement('span');
|
||||||
|
if (cls) row.className = cls;
|
||||||
|
row.textContent = text + '\n';
|
||||||
|
log.appendChild(row);
|
||||||
|
}
|
||||||
|
function appendLine(text, cls) {
|
||||||
|
if (log.firstChild && log.firstChild.className === 'meta') log.innerHTML = '';
|
||||||
|
const row = document.createElement('span');
|
||||||
|
if (cls) row.className = cls;
|
||||||
|
row.textContent = text + '\n';
|
||||||
|
log.appendChild(row);
|
||||||
|
log.scrollTop = log.scrollHeight;
|
||||||
|
}
|
||||||
|
function fmt(ev) {
|
||||||
|
if (ev.kind === 'turn_start') return '◆ TURN ← ' + ev.from + ': ' + ev.body;
|
||||||
|
if (ev.kind === 'turn_end') return '◇ TURN END ' + (ev.ok ? 'ok' : 'fail') + (ev.note ? ' — ' + ev.note : '');
|
||||||
|
if (ev.kind === 'note') return '· ' + ev.text;
|
||||||
|
if (ev.kind === 'stream') {
|
||||||
|
// serde internal tagging flattens the inner json next to `kind`,
|
||||||
|
// so the original stream-json event sits under `ev` minus `kind`.
|
||||||
|
const v = Object.assign({}, ev); delete v.kind;
|
||||||
|
if (v.type === 'system' && v.subtype === 'init') return '[init] tools=' + (v.tools||[]).length;
|
||||||
|
if (v.type === 'assistant' && v.message && v.message.content) {
|
||||||
|
const parts = v.message.content.map(c => {
|
||||||
|
if (c.type === 'text') return c.text;
|
||||||
|
if (c.type === 'tool_use') return '→ ' + c.name + '(' + JSON.stringify(c.input) + ')';
|
||||||
|
return c.type;
|
||||||
|
});
|
||||||
|
return parts.join('\n');
|
||||||
|
}
|
||||||
|
if (v.type === 'user' && v.message && v.message.content) {
|
||||||
|
const parts = v.message.content.map(c => {
|
||||||
|
if (c.type === 'tool_result') {
|
||||||
|
const txt = Array.isArray(c.content) ? c.content.map(p => p.text || '').join(' ') : (c.content || '');
|
||||||
|
return '← ' + (txt.length > 200 ? txt.slice(0,200) + '…' : txt);
|
||||||
|
}
|
||||||
|
return c.type;
|
||||||
|
});
|
||||||
|
return parts.join('\n');
|
||||||
|
}
|
||||||
|
if (v.type === 'result') return '[done] ' + (v.subtype || '') + (v.is_error ? ' error' : '');
|
||||||
|
return JSON.stringify(v);
|
||||||
|
}
|
||||||
|
return JSON.stringify(ev);
|
||||||
|
}
|
||||||
|
function cls(ev) {
|
||||||
|
if (ev.kind === 'turn_start') return 'turnstart';
|
||||||
|
if (ev.kind === 'turn_end') return ev.ok ? 'turnok' : 'turnfail';
|
||||||
|
if (ev.kind === 'note') return 'meta';
|
||||||
|
return '';
|
||||||
|
}
|
||||||
|
const es = new EventSource('/events/stream');
|
||||||
|
es.onmessage = function(e) {
|
||||||
|
try {
|
||||||
|
const ev = JSON.parse(e.data);
|
||||||
|
appendLine(fmt(ev), cls(ev));
|
||||||
|
} catch (err) {
|
||||||
|
appendLine('[parse err] ' + e.data, 'meta');
|
||||||
|
}
|
||||||
|
};
|
||||||
|
es.onerror = function() { appendLine('[disconnected — retrying]', 'meta'); };
|
||||||
|
})();
|
||||||
|
</script>
|
||||||
|
"#;
|
||||||
|
|
||||||
fn render_needs_login_idle() -> String {
|
fn render_needs_login_idle() -> String {
|
||||||
"<p class=\"status-needs-login\">▓█▓▒░ NEEDS L0G1N ▓█▓▒░</p>\n<p>No Claude session in <code>~/.claude/</code>. The harness is up but the turn loop is paused until you log in.</p>\n<form method=\"POST\" action=\"/login/start\">\n <button type=\"submit\" class=\"btn btn-login\">◆ ST4RT L0G1N</button>\n</form>\n<p class=\"meta\">Spawns <code>claude auth login</code> over plain stdio pipes. The OAuth URL will appear here when claude emits it; paste the resulting code back into the form below.</p>".into()
|
"<p class=\"status-needs-login\">▓█▓▒░ NEEDS L0G1N ▓█▓▒░</p>\n<p>No Claude session in <code>~/.claude/</code>. The harness is up but the turn loop is paused until you log in.</p>\n<form method=\"POST\" action=\"/login/start\">\n <button type=\"submit\" class=\"btn btn-login\">◆ ST4RT L0G1N</button>\n</form>\n<p class=\"meta\">Spawns <code>claude auth login</code> over plain stdio pipes. The OAuth URL will appear here when claude emits it; paste the resulting code back into the form below.</p>".into()
|
||||||
}
|
}
|
||||||
|
|
@ -104,6 +191,18 @@ fn render_login_in_progress(session: &Arc<LoginSession>) -> String {
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn events_stream(
|
||||||
|
State(state): State<AppState>,
|
||||||
|
) -> Sse<impl Stream<Item = Result<Event, Infallible>>> {
|
||||||
|
let rx = state.bus.subscribe();
|
||||||
|
let stream = BroadcastStream::new(rx).filter_map(|res| {
|
||||||
|
let ev = res.ok()?;
|
||||||
|
let json = serde_json::to_string(&ev).ok()?;
|
||||||
|
Some(Ok(Event::default().data(json)))
|
||||||
|
});
|
||||||
|
Sse::new(stream).keep_alive(KeepAlive::default())
|
||||||
|
}
|
||||||
|
|
||||||
async fn post_login_start(State(state): State<AppState>) -> Response {
|
async fn post_login_start(State(state): State<AppState>) -> Response {
|
||||||
drop_if_finished(&state.session);
|
drop_if_finished(&state.session);
|
||||||
{
|
{
|
||||||
|
|
@ -244,5 +343,10 @@ const STYLE: &str = r#"
|
||||||
word-break: break-all;
|
word-break: break-all;
|
||||||
max-height: 30em;
|
max-height: 30em;
|
||||||
}
|
}
|
||||||
|
#live { max-height: 24em; overflow-y: auto; }
|
||||||
|
#live span { display: block; }
|
||||||
|
#live .turnstart { color: var(--amber); }
|
||||||
|
#live .turnok { color: var(--green); }
|
||||||
|
#live .turnfail { color: #ff6b6b; }
|
||||||
</style>
|
</style>
|
||||||
"#;
|
"#;
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue