pub/spaceshell
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: persisted per-workspace panel zoom (proto + daemon, auto-clear on removal)

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Vassiliy Yegorov
2026-06-10 12:16:44 +07:00
parent defceb1169
commit 2b1ccaf31d
5 changed files with 174 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/spaceshd/src/server.rs
+121 -1
View File
@@ -572,7 +572,12 @@ async fn handle_request(
crate::hooks::cleanup(&surface_id);
crate::hooks::cleanup_shell(&surface_id);
broadcast_evt(clients, &Envelope::Evt(Evt::SurfaceClosed { surface_id: surface_id.clone() }));
if let Some(ws_id) = ws_id { emit_layout(reg, &ws_id, clients); }
if let Some(ws_id) = ws_id {
emit_layout(reg, &ws_id, clients);
if let Some(view) = reg.workspace_view(&ws_id) {
broadcast_evt(clients, &Envelope::Evt(Evt::WorkspaceChanged { workspace: view }));
}
}
persister.mark_dirty(reg.persist_state());
let _ = out.send(ok(id, serde_json::Value::Null)).await;
} else {
@@ -594,6 +599,23 @@ async fn handle_request(
}
}
Cmd::SetZoom { workspace_id, surface_id } => {
let Some(w) = reg.workspace(&workspace_id) else {
let _ = out.send(err(id, "NOT_FOUND", "workspace")).await; return;
};
if let Some(sid) = &surface_id {
if !w.surfaces.contains_key(sid) {
let _ = out.send(err(id, "NOT_FOUND", "surface")).await; return;
}
}
reg.workspace_mut(&workspace_id).expect("workspace validated above").zoomed = surface_id.clone();
if let Some(view) = reg.workspace_view(&workspace_id) {
broadcast_evt(clients, &Envelope::Evt(Evt::WorkspaceChanged { workspace: view }));
}
persister.mark_dirty(reg.persist_state());
let _ = out.send(ok(id, serde_json::Value::Null)).await;
}
Cmd::Health => {
let _ = out.send(ok(id, serde_json::json!({
"version": env!("CARGO_PKG_VERSION"),
@@ -1335,4 +1357,102 @@ mod tests {
let started = d["started_at_ms"].as_u64().unwrap();
assert!(started > 0 && started >= now.saturating_sub(5000) && started <= now + 1000, "started_at_ms plausible: {started} vs now {now}");
}
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
async fn set_zoom_sets_and_clears_and_autoclears() {
let _serial = crate::test_support::serial();
let dir = tempdir_path();
let sock = dir.join("sock");
let store: std::sync::Arc<dyn crate::state_store::StateStore> =
std::sync::Arc::new(crate::state_store::JsonStateStore::new(dir.join("state.json")));
let event_store = make_event_store(&dir);
let sock_for_task = sock.clone();
let store2 = store.clone();
tokio::spawn(async move { let _ = serve(&sock_for_task, store2, event_store).await; });
wait_for_socket(&sock).await;
let mut s = UnixStream::connect(&sock).await.unwrap();
let r = req(&mut s, 1, Cmd::Open { path: std::env::temp_dir().to_string_lossy().into() }).await;
let ws = res_data(&r)["workspace_id"].as_str().unwrap().to_string();
let r = req(&mut s, 2, Cmd::NewSurface {
workspace_id: spacesh_proto::WorkspaceId(ws.clone()),
command: Some("/bin/sh".into()), args: vec!["-c".into(), "sleep 5".into()], cols: 80, rows: 24,
}).await;
let sid = res_data(&r)["surface_id"].as_str().unwrap().to_string();
let _ = req(&mut s, 3, Cmd::SetZoom {
workspace_id: spacesh_proto::WorkspaceId(ws.clone()),
surface_id: Some(spacesh_proto::SurfaceId(sid.clone())),
}).await;
let st = req(&mut s, 4, Cmd::Status).await;
let w0 = res_data(&st)["workspaces"].as_array().unwrap().iter().find(|w| w["id"] == ws).unwrap().clone();
assert_eq!(w0["zoomed"], sid);
let _ = req(&mut s, 5, Cmd::SetZoom {
workspace_id: spacesh_proto::WorkspaceId(ws.clone()), surface_id: None,
}).await;
let st = req(&mut s, 6, Cmd::Status).await;
let w0 = res_data(&st)["workspaces"].as_array().unwrap().iter().find(|w| w["id"] == ws).unwrap().clone();
assert!(w0["zoomed"].is_null());
let _ = req(&mut s, 7, Cmd::SetZoom {
workspace_id: spacesh_proto::WorkspaceId(ws.clone()),
surface_id: Some(spacesh_proto::SurfaceId(sid.clone())),
}).await;
let _ = req(&mut s, 8, Cmd::Close { surface_id: spacesh_proto::SurfaceId(sid.clone()) }).await;
let st = req(&mut s, 9, Cmd::Status).await;
let w0 = res_data(&st)["workspaces"].as_array().unwrap().iter().find(|w| w["id"] == ws).unwrap().clone();
assert!(w0["zoomed"].is_null(), "closing the zoomed surface clears zoom");
}
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
async fn close_zoomed_broadcasts_workspace_changed() {
let _serial = crate::test_support::serial();
let dir = tempdir_path();
let sock = dir.join("sock");
let store: std::sync::Arc<dyn crate::state_store::StateStore> =
std::sync::Arc::new(crate::state_store::JsonStateStore::new(dir.join("state.json")));
let event_store = make_event_store(&dir);
let sock_for_task = sock.clone();
let store2 = store.clone();
tokio::spawn(async move { let _ = serve(&sock_for_task, store2, event_store).await; });
wait_for_socket(&sock).await;
// Control connection: open, spawn, zoom.
let mut ctrl = UnixStream::connect(&sock).await.unwrap();
let r = req(&mut ctrl, 1, Cmd::Open { path: std::env::temp_dir().to_string_lossy().into() }).await;
let ws = res_data(&r)["workspace_id"].as_str().unwrap().to_string();
let r = req(&mut ctrl, 2, Cmd::NewSurface {
workspace_id: spacesh_proto::WorkspaceId(ws.clone()),
command: Some("/bin/sh".into()), args: vec!["-c".into(), "sleep 5".into()], cols: 80, rows: 24,
}).await;
let sid = res_data(&r)["surface_id"].as_str().unwrap().to_string();
let _ = req(&mut ctrl, 3, Cmd::SetZoom {
workspace_id: spacesh_proto::WorkspaceId(ws.clone()),
surface_id: Some(spacesh_proto::SurfaceId(sid.clone())),
}).await;
// Observer connection: must be attached BEFORE the Close so it catches the broadcast.
let mut observer = UnixStream::connect(&sock).await.unwrap();
// Close the zoomed surface on the control connection.
let _ = req(&mut ctrl, 4, Cmd::Close { surface_id: spacesh_proto::SurfaceId(sid.clone()) }).await;
// Observer must receive a WorkspaceChanged for this workspace with zoomed == None.
let mut saw_cleared = false;
let deadline = tokio::time::Instant::now() + tokio::time::Duration::from_secs(2);
while tokio::time::Instant::now() < deadline {
if let Ok(Ok(Some(env))) =
tokio::time::timeout(tokio::time::Duration::from_millis(200), read_frame(&mut observer)).await {
if let Envelope::Evt(Evt::WorkspaceChanged { workspace }) = env {
if workspace.id.0 == ws {
assert!(workspace.zoomed.is_none(), "WorkspaceChanged must report cleared zoom");
saw_cleared = true;
break;
}
}
}
}
assert!(saw_cleared, "expected a WorkspaceChanged broadcast with cleared zoom after closing the zoomed surface");
}
}