web_ui: support unix:<path> upstreams in extraWebProxies

reqwest has no UDS transport, so unix: upstreams dial the socket
directly with a raw hyper/1.1 client per request instead. http(s)://
upstreams are unaffected (still go through the existing reqwest path).

Adds hyper (client, http1), hyper-util (tokio IO adapter), and
http-body-util as direct hive-ag3nt dependencies - all three were
already present transitively via reqwest, this just uses them
directly for the new code path.
This commit is contained in:
iris 2026-07-10 12:11:04 +02:00 committed by mara
commit 476a7a3c9f
6 changed files with 127 additions and 5 deletions

3
Cargo.lock generated
View file

@ -1383,6 +1383,9 @@ dependencies = [
"futures-util",
"hive-claude",
"hive-sh4re",
"http-body-util",
"hyper",
"hyper-util",
"reqwest",
"rmcp",
"rusqlite",

View file

@ -68,6 +68,9 @@ reqwest = { version = "0.12", default-features = false, features = [
"json",
"rustls-tls",
] }
hyper = { version = "1", features = ["client", "http1"] }
hyper-util = { version = "0.1", features = ["tokio"] }
http-body-util = "0.1"
forgejo-api = { version = "0.11", default-features = false, features = [
"rustls-tls",
] }

View file

@ -411,7 +411,11 @@ shaped).
on both sides, and buffers the full response body (MVP — SSE
connections will appear as one large response rather than streaming).
The `/extra/` namespace ensures user-declared proxies can never
conflict with native agent endpoints. Implemented in `web_ui/proxy.rs`.
conflict with native agent endpoints. Upstream values are either an
`http(s)://` URL (forwarded via `reqwest`) or a Unix domain socket,
spelled `unix:<path>` (e.g. `unix:/run/myapp/http.sock`) — dialed
directly with a raw HTTP/1.1 client per request, since `reqwest` has
no UDS transport. Implemented in `web_ui/proxy.rs`.
Bus events (new vocabulary on `/events/stream`):

View file

@ -10,6 +10,9 @@ workspace = true
anyhow.workspace = true
axum.workspace = true
reqwest.workspace = true
hyper.workspace = true
hyper-util.workspace = true
http-body-util.workspace = true
forgejo-api.workspace = true
url.workspace = true
time.workspace = true

View file

@ -6,8 +6,14 @@
//! proxies from ever colliding with the native agent endpoints (`/api/*`,
//! `/events/*`, …). Response bodies are buffered whole (MVP): an SSE upstream
//! appears as one large response rather than streaming.
//!
//! Upstreams are either a regular `http(s)://` URL (forwarded via `reqwest`)
//! or a Unix domain socket, spelled `unix:<path>` (e.g.
//! `unix:/run/myapp/http.sock`) — forwarded via a raw hyper/1.1 client
//! dialing the socket directly, since `reqwest` has no UDS transport.
use std::collections::HashMap;
use std::path::{Path, PathBuf};
use std::sync::Arc;
use axum::{
@ -17,6 +23,8 @@ use axum::{
http::{HeaderMap, HeaderName, HeaderValue, Method, StatusCode, Uri},
response::{IntoResponse, Response},
};
use http_body_util::{BodyExt, Full};
use hyper_util::rt::TokioIo;
use super::AppState;
@ -55,14 +63,28 @@ pub(super) fn mount_extra_proxies(mut app: Router<AppState>) -> Router<AppState>
}
/// Build a transparent reverse-proxy service forwarding every request to
/// `upstream` (e.g. `http://127.0.0.1:3737`). The caller nests it at a path
/// prefix; axum strips the prefix before the service sees the request.
/// Returns `None` when `upstream` is blank or the reqwest client won't build.
/// `upstream` — either an `http(s)://` URL or a `unix:<path>` Unix domain
/// socket. The caller nests it at a path prefix; axum strips the prefix
/// before the service sees the request. Returns `None` when `upstream` is
/// blank or the underlying client won't build.
fn extra_proxy_service(upstream: &str) -> Option<Router<()>> {
let upstream = upstream.trim();
if upstream.is_empty() {
return None;
}
if let Some(sock_path) = upstream.strip_prefix("unix:") {
let sock_path = sock_path.trim();
if sock_path.is_empty() {
tracing::warn!("extra proxy: `unix:` upstream has an empty socket path — ignoring");
return None;
}
return Some(
Router::new()
.fallback(unix_proxy_handler)
.with_state(Arc::new(PathBuf::from(sock_path))),
);
}
let client = match reqwest::Client::builder()
.timeout(std::time::Duration::from_secs(30))
.build()
@ -148,3 +170,86 @@ async fn proxy_handler(
}
}
}
/// Forward the request to `sock_path` over a Unix domain socket, dialing a
/// fresh connection per request (MVP — no connection pooling, matching the
/// simplicity of the HTTP path above). `reqwest` has no UDS transport, so
/// this speaks raw HTTP/1.1 via `hyper`'s low-level client directly over a
/// `tokio::net::UnixStream`.
async fn unix_proxy_handler(
State(sock_path): State<Arc<PathBuf>>,
method: Method,
uri: Uri,
headers: HeaderMap,
body: Bytes,
) -> Response {
let path_and_query = uri
.path_and_query()
.map_or("/", axum::http::uri::PathAndQuery::as_str)
.to_owned();
match forward_over_unix_socket(&sock_path, method, &path_and_query, headers, body).await {
Ok(resp) => resp,
Err(e) => {
tracing::warn!(
socket = %sock_path.display(),
path = path_and_query,
"extra proxy: unix upstream request failed: {e}"
);
StatusCode::BAD_GATEWAY.into_response()
}
}
}
/// Dial `sock_path`, send one HTTP/1.1 request built from the given parts,
/// and return the axum `Response` built from whatever comes back. Split out
/// of [`unix_proxy_handler`] so the handler can map every failure mode
/// (connect, handshake, send, body read) to the same `BAD_GATEWAY` fallback
/// with one `?`-chain instead of a matching arm per step.
async fn forward_over_unix_socket(
sock_path: &Path,
method: Method,
path_and_query: &str,
mut headers: HeaderMap,
body: Bytes,
) -> anyhow::Result<Response> {
let stream = tokio::net::UnixStream::connect(sock_path).await?;
let io = TokioIo::new(stream);
let (mut sender, conn) = hyper::client::conn::http1::handshake(io).await?;
// The connection future drives I/O in the background; drop-and-forget is
// fine here since we only ever send one request per dialed socket.
tokio::spawn(async move {
if let Err(e) = conn.await {
tracing::debug!("extra proxy: unix connection closed: {e}");
}
});
for h in HOP_BY_HOP {
headers.remove(h);
}
// HTTP/1.1 requires a Host header; a UDS peer has no meaningful
// hostname, so `localhost` is the conventional placeholder (matches
// what tools like `curl --unix-socket` send by default).
if !headers.contains_key(hyper::header::HOST) {
headers.insert(hyper::header::HOST, HeaderValue::from_static("localhost"));
}
let mut req_builder = hyper::Request::builder().method(method).uri(path_and_query);
if let Some(h) = req_builder.headers_mut() {
*h = headers;
}
let req = req_builder.body(Full::new(body))?;
let upstream_resp = sender.send_request(req).await?;
let status = StatusCode::from_u16(upstream_resp.status().as_u16())?;
let mut resp_headers = HeaderMap::new();
for (name, value) in upstream_resp.headers() {
if HOP_BY_HOP.contains(&name.as_str()) {
continue;
}
resp_headers.insert(name.clone(), value.clone());
}
let body_bytes = upstream_resp.into_body().collect().await?.to_bytes();
Ok((status, resp_headers, body_bytes).into_response())
}

View file

@ -819,12 +819,16 @@ in
example = lib.literalExpression ''{ "stats" = "http://127.0.0.1:3737"; }'';
description = ''
Transparent reverse-proxies mounted under `/extra/` in the per-agent web UI.
Each attribute name becomes the sub-path and the value is the upstream URL.
Each attribute name becomes the sub-path and the value is the upstream.
E.g. `{ "stats" = "http://127.0.0.1:3737"; }` mounts a proxy at
`/agent/<name>/extra/stats/` that forwards to port 3737 with the prefix
stripped. All user-declared proxies live under `/extra/` so they can
never conflict with native agent endpoints (`/api/*`, `/events/*`, etc.).
The upstream value is either an `http(s)://` URL or a Unix domain
socket, spelled `unix:<path>` (e.g. `unix:/run/myapp/http.sock`) for
agents whose secondary web server only listens on a UDS.
Intended for agents that run secondary web servers in the same container.
Static assets served by the secondary app must use relative paths to
resolve correctly under the sub-path prefix.