Skip to content

Commit d143018

Browse files
committed
Format Rust server code
1 parent 133c380 commit d143018

5 files changed

Lines changed: 186 additions & 110 deletions

File tree

apps/server/src/command/http.rs

Lines changed: 18 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -728,7 +728,7 @@ fn dispatch_rpc(
728728
req.patch,
729729
app.state(),
730730
)
731-
.map_err(rpc_bad_request)?,
731+
.map_err(rpc_bad_request)?,
732732
)
733733
.map_err(|e| rpc_bad_request(e.to_string()))
734734
}
@@ -828,7 +828,7 @@ fn dispatch_rpc(
828828
req.mutation.target.clone(),
829829
req.file_path,
830830
)
831-
.map_err(rpc_bad_request)?;
831+
.map_err(rpc_bad_request)?;
832832
emit_workspace_artifacts_dirty(
833833
app,
834834
&req.mutation.path,
@@ -866,7 +866,7 @@ fn dispatch_rpc(
866866
req.mutation.target.clone(),
867867
req.file_path,
868868
)
869-
.map_err(rpc_bad_request)?;
869+
.map_err(rpc_bad_request)?;
870870
emit_workspace_artifacts_dirty(
871871
app,
872872
&req.mutation.path,
@@ -930,7 +930,7 @@ fn dispatch_rpc(
930930
req.mutation.target.clone(),
931931
req.message,
932932
)
933-
.map_err(rpc_bad_request)?,
933+
.map_err(rpc_bad_request)?,
934934
)
935935
.map_err(|e| rpc_bad_request(e.to_string()))?;
936936
emit_workspace_artifacts_dirty(
@@ -1056,7 +1056,7 @@ fn dispatch_rpc(
10561056
req.input,
10571057
app.state(),
10581058
)
1059-
.map_err(rpc_bad_request)?;
1059+
.map_err(rpc_bad_request)?;
10601060
Ok(Value::Null)
10611061
}
10621062
"terminal_resize" => {
@@ -1559,8 +1559,14 @@ mod tests {
15591559
.expect("attach rpc should be dispatched");
15601560

15611561
let snapshot: WorkspaceRuntimeSnapshot = serde_json::from_value(value).unwrap();
1562-
assert_eq!(snapshot.controller.controller_device_id.as_deref(), Some("device-a"));
1563-
assert_eq!(snapshot.controller.controller_client_id.as_deref(), Some("client-a"));
1562+
assert_eq!(
1563+
snapshot.controller.controller_device_id.as_deref(),
1564+
Some("device-a")
1565+
);
1566+
assert_eq!(
1567+
snapshot.controller.controller_client_id.as_deref(),
1568+
Some("client-a")
1569+
);
15641570
}
15651571

15661572
#[test]
@@ -1795,12 +1801,11 @@ mod tests {
17951801
scoped.ui_state.active_workspace_id.as_deref(),
17961802
Some(first_launch.snapshot.workspace.workspace_id.as_str())
17971803
);
1798-
assert!(
1799-
scoped
1800-
.workspaces
1801-
.iter()
1802-
.all(|snapshot| snapshot.workspace.workspace_id != second_launch.snapshot.workspace.workspace_id)
1803-
);
1804+
assert!(scoped
1805+
.workspaces
1806+
.iter()
1807+
.all(|snapshot| snapshot.workspace.workspace_id
1808+
!= second_launch.snapshot.workspace.workspace_id));
18041809
}
18051810

18061811
#[test]

apps/server/src/infra/db.rs

Lines changed: 68 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,10 @@ fn load_legacy_ui_state_from_conn(conn: &Connection) -> Result<WorkbenchUiState,
288288
parse_json(&payload)
289289
}
290290

291-
fn save_legacy_ui_state_to_conn(conn: &Connection, ui_state: &WorkbenchUiState) -> Result<(), String> {
291+
fn save_legacy_ui_state_to_conn(
292+
conn: &Connection,
293+
ui_state: &WorkbenchUiState,
294+
) -> Result<(), String> {
292295
let payload = json_string(ui_state)?;
293296
conn.execute(
294297
"INSERT INTO app_ui_state (id, payload, updated_at)
@@ -332,7 +335,8 @@ fn load_device_ui_state_from_conn(
332335
match payload {
333336
Ok(value) => parse_json(&value),
334337
Err(rusqlite::Error::QueryReturnedNoRows) => {
335-
let legacy = load_legacy_ui_state_from_conn(conn).unwrap_or_else(|_| default_ui_state());
338+
let legacy =
339+
load_legacy_ui_state_from_conn(conn).unwrap_or_else(|_| default_ui_state());
336340
Ok(DeviceWorkbenchUiState {
337341
open_workspace_ids: legacy.open_workspace_ids,
338342
layout: legacy.layout,
@@ -391,7 +395,8 @@ fn load_client_ui_state_from_conn(
391395
match payload {
392396
Ok(value) => parse_json(&value),
393397
Err(rusqlite::Error::QueryReturnedNoRows) => {
394-
let legacy = load_legacy_ui_state_from_conn(conn).unwrap_or_else(|_| default_ui_state());
398+
let legacy =
399+
load_legacy_ui_state_from_conn(conn).unwrap_or_else(|_| default_ui_state());
395400
Ok(ClientWorkbenchUiState {
396401
active_workspace_id: legacy.active_workspace_id,
397402
})
@@ -436,15 +441,16 @@ fn compose_workbench_ui_state(
436441
mut device_state: DeviceWorkbenchUiState,
437442
mut client_state: ClientWorkbenchUiState,
438443
) -> WorkbenchUiState {
439-
device_state.open_workspace_ids = device_state
440-
.open_workspace_ids
441-
.into_iter()
442-
.fold(Vec::new(), |mut items, workspace_id| {
443-
if !items.iter().any(|item| item == &workspace_id) {
444-
items.push(workspace_id);
445-
}
446-
items
447-
});
444+
device_state.open_workspace_ids =
445+
device_state
446+
.open_workspace_ids
447+
.into_iter()
448+
.fold(Vec::new(), |mut items, workspace_id| {
449+
if !items.iter().any(|item| item == &workspace_id) {
450+
items.push(workspace_id);
451+
}
452+
items
453+
});
448454
if let Some(active) = client_state.active_workspace_id.as_deref() {
449455
if !device_state
450456
.open_workspace_ids
@@ -612,7 +618,12 @@ fn set_terminal_recoverable(
612618
"UPDATE workspace_terminals
613619
SET recoverable = ?3, updated_at = ?4
614620
WHERE workspace_id = ?1 AND terminal_id = ?2",
615-
params![workspace_id, terminal_id as i64, if recoverable { 1 } else { 0 }, now_ts()],
621+
params![
622+
workspace_id,
623+
terminal_id as i64,
624+
if recoverable { 1 } else { 0 },
625+
now_ts()
626+
],
616627
)
617628
.map_err(|e| e.to_string())?;
618629
Ok(())
@@ -774,7 +785,9 @@ fn load_workspace_controller_lease_from_conn(
774785

775786
match payload {
776787
Ok(value) => parse_json(&value),
777-
Err(rusqlite::Error::QueryReturnedNoRows) => Ok(default_workspace_controller_lease(workspace_id)),
788+
Err(rusqlite::Error::QueryReturnedNoRows) => {
789+
Ok(default_workspace_controller_lease(workspace_id))
790+
}
778791
Err(error) => Err(error.to_string()),
779792
}
780793
}
@@ -1043,7 +1056,9 @@ fn remove_workspace_from_all_ui_state_scopes(
10431056
let (device_id_row, payload) = row.map_err(|e| e.to_string())?;
10441057
let mut ui_state: DeviceWorkbenchUiState = parse_json(&payload)?;
10451058
let before = ui_state.open_workspace_ids.clone();
1046-
ui_state.open_workspace_ids.retain(|item| item != workspace_id);
1059+
ui_state
1060+
.open_workspace_ids
1061+
.retain(|item| item != workspace_id);
10471062
if ui_state.open_workspace_ids != before {
10481063
save_device_ui_state_to_conn(conn, Some(device_id_row.as_str()), &ui_state)?;
10491064
}
@@ -1225,7 +1240,9 @@ pub(crate) fn workbench_bootstrap(
12251240
device_id: Option<&str>,
12261241
client_id: Option<&str>,
12271242
) -> Result<WorkbenchBootstrap, String> {
1228-
with_db(state, |conn| build_bootstrap_from_conn(conn, device_id, client_id))
1243+
with_db(state, |conn| {
1244+
build_bootstrap_from_conn(conn, device_id, client_id)
1245+
})
12291246
}
12301247

12311248
pub(crate) fn workspace_snapshot(
@@ -1538,7 +1555,9 @@ pub(crate) fn patch_workspace_view_state(
15381555
let next = WorkspaceViewState {
15391556
active_session_id: patch.active_session_id.unwrap_or(current.active_session_id),
15401557
active_pane_id: patch.active_pane_id.unwrap_or(current.active_pane_id),
1541-
active_terminal_id: patch.active_terminal_id.unwrap_or(current.active_terminal_id),
1558+
active_terminal_id: patch
1559+
.active_terminal_id
1560+
.unwrap_or(current.active_terminal_id),
15421561
pane_layout: patch.pane_layout.unwrap_or(current.pane_layout),
15431562
file_preview: patch.file_preview.unwrap_or(current.file_preview),
15441563
};
@@ -1551,14 +1570,18 @@ pub(crate) fn load_workspace_controller_lease(
15511570
state: State<'_, AppState>,
15521571
workspace_id: &str,
15531572
) -> Result<WorkspaceControllerLease, String> {
1554-
with_db(state, |conn| load_workspace_controller_lease_from_conn(conn, workspace_id))
1573+
with_db(state, |conn| {
1574+
load_workspace_controller_lease_from_conn(conn, workspace_id)
1575+
})
15551576
}
15561577

15571578
pub(crate) fn save_workspace_controller_lease(
15581579
state: State<'_, AppState>,
15591580
lease: &WorkspaceControllerLease,
15601581
) -> Result<(), String> {
1561-
with_db(state, |conn| save_workspace_controller_lease_to_conn(conn, lease))
1582+
with_db(state, |conn| {
1583+
save_workspace_controller_lease_to_conn(conn, lease)
1584+
})
15621585
}
15631586

15641587
pub(crate) fn upsert_workspace_attachment(
@@ -1582,7 +1605,14 @@ pub(crate) fn append_agent_lifecycle_event(
15821605
data: &str,
15831606
) -> Result<AgentLifecycleHistoryEntry, String> {
15841607
with_db(state, |conn| {
1585-
append_agent_lifecycle_event_to_conn(conn, workspace_id, session_id, kind, source_event, data)
1608+
append_agent_lifecycle_event_to_conn(
1609+
conn,
1610+
workspace_id,
1611+
session_id,
1612+
kind,
1613+
source_event,
1614+
data,
1615+
)
15861616
})
15871617
}
15881618

@@ -1591,7 +1621,9 @@ pub(crate) fn load_agent_lifecycle_events(
15911621
workspace_id: &str,
15921622
limit: usize,
15931623
) -> Result<Vec<AgentLifecycleHistoryEntry>, String> {
1594-
with_db(state, |conn| load_agent_lifecycle_events_from_conn(conn, workspace_id, limit))
1624+
with_db(state, |conn| {
1625+
load_agent_lifecycle_events_from_conn(conn, workspace_id, limit)
1626+
})
15951627
}
15961628

15971629
pub(crate) fn list_workspace_ids_for_workspace_client(
@@ -1609,7 +1641,9 @@ pub(crate) fn mark_workspace_client_detached(
16091641
device_id: &str,
16101642
client_id: &str,
16111643
) -> Result<(), String> {
1612-
with_db(state, |conn| mark_workspace_client_detached_from_conn(conn, device_id, client_id))
1644+
with_db(state, |conn| {
1645+
mark_workspace_client_detached_from_conn(conn, device_id, client_id)
1646+
})
16131647
}
16141648

16151649
pub(crate) fn persist_workspace_terminal(
@@ -1619,7 +1653,9 @@ pub(crate) fn persist_workspace_terminal(
16191653
output: &str,
16201654
recoverable: bool,
16211655
) -> Result<(), String> {
1622-
with_db(state, |conn| persist_terminal_row(conn, workspace_id, terminal_id, output, recoverable))
1656+
with_db(state, |conn| {
1657+
persist_terminal_row(conn, workspace_id, terminal_id, output, recoverable)
1658+
})
16231659
}
16241660

16251661
pub(crate) fn append_workspace_terminal_output(
@@ -1628,7 +1664,9 @@ pub(crate) fn append_workspace_terminal_output(
16281664
terminal_id: u64,
16291665
chunk: &str,
16301666
) -> Result<(), String> {
1631-
with_db(state, |conn| append_terminal_output(conn, workspace_id, terminal_id, chunk))
1667+
with_db(state, |conn| {
1668+
append_terminal_output(conn, workspace_id, terminal_id, chunk)
1669+
})
16321670
}
16331671

16341672
pub(crate) fn set_workspace_terminal_recoverable(
@@ -1637,15 +1675,19 @@ pub(crate) fn set_workspace_terminal_recoverable(
16371675
terminal_id: u64,
16381676
recoverable: bool,
16391677
) -> Result<(), String> {
1640-
with_db(state, |conn| set_terminal_recoverable(conn, workspace_id, terminal_id, recoverable))
1678+
with_db(state, |conn| {
1679+
set_terminal_recoverable(conn, workspace_id, terminal_id, recoverable)
1680+
})
16411681
}
16421682

16431683
pub(crate) fn delete_workspace_terminal(
16441684
state: State<'_, AppState>,
16451685
workspace_id: &str,
16461686
terminal_id: u64,
16471687
) -> Result<(), String> {
1648-
with_db(state, |conn| delete_persisted_terminal(conn, workspace_id, terminal_id))
1688+
with_db(state, |conn| {
1689+
delete_persisted_terminal(conn, workspace_id, terminal_id)
1690+
})
16491691
}
16501692

16511693
pub(crate) fn append_session_stream(

apps/server/src/main.rs

Lines changed: 23 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -44,28 +44,27 @@ pub(crate) use auth::{
4444
admin_auth_status, admin_blocked_ips, admin_config, admin_unblock_ip, admin_update_config,
4545
auth_status, ensure_optional_path_allowed, ensure_path_allowed, filesystem_list_public,
4646
filesystem_roots_public, filter_allowed_worktrees, load_or_initialize_auth_runtime,
47-
lock as auth_lock, login as auth_login, logout as auth_logout,
48-
normalize_path_for_target, path_within_root, require_session, select_clone_root_for_target,
49-
transport_bind_config, AuthorizedRequest, RequestContext,
47+
lock as auth_lock, login as auth_login, logout as auth_logout, normalize_path_for_target,
48+
path_within_root, require_session, select_clone_root_for_target, transport_bind_config,
49+
AuthorizedRequest, RequestContext,
5050
};
5151
pub(crate) use command::http::start_transport_server;
52+
#[cfg(test)]
53+
pub(crate) use infra::db::launch_workspace_record;
5254
pub(crate) use infra::db::{
5355
activate_workspace_ui, append_agent_lifecycle_event, append_session_stream,
54-
append_workspace_terminal_output,
55-
archive_workspace_session, close_workspace_ui, create_workspace_session,
56-
delete_workspace_terminal, init_db, launch_workspace_record_scoped,
56+
append_workspace_terminal_output, archive_workspace_session, close_workspace_ui,
57+
create_workspace_session, delete_workspace_terminal, init_db, launch_workspace_record_scoped,
5758
list_workspace_ids_for_workspace_client, load_agent_lifecycle_events, load_session,
58-
load_workspace_controller_lease,
59-
mark_active_sessions_interrupted_on_boot, mark_workspace_client_detached,
60-
patch_workspace_view_state, persist_workspace_terminal, save_workspace_controller_lease,
61-
set_session_claude_id, set_session_status, set_workspace_terminal_recoverable,
62-
switch_workspace_session, upsert_workspace_attachment,
59+
load_workspace_controller_lease, mark_active_sessions_interrupted_on_boot,
60+
mark_workspace_client_detached, patch_workspace_view_state, persist_workspace_terminal,
61+
save_workspace_controller_lease, set_session_claude_id, set_session_status,
62+
set_workspace_terminal_recoverable, switch_workspace_session,
6363
update_workbench_layout as persist_workbench_layout, update_workspace_idle_policy,
64-
update_workspace_session, workbench_bootstrap as load_workbench_bootstrap,
65-
workspace_access_context, workspace_snapshot as load_workspace_snapshot,
64+
update_workspace_session, upsert_workspace_attachment,
65+
workbench_bootstrap as load_workbench_bootstrap, workspace_access_context,
66+
workspace_snapshot as load_workspace_snapshot,
6667
};
67-
#[cfg(test)]
68-
pub(crate) use infra::db::launch_workspace_record;
6968
pub(crate) use infra::runtime::{
7069
build_agent_pty_command, build_claude_resume_command, build_terminal_pty_command,
7170
repo_name_from_url, resolve_git_repo_path, resolve_target_path, run_cmd, shell_escape,
@@ -79,16 +78,15 @@ pub(crate) use infra::support::{
7978
};
8079
pub(crate) use infra::time::{default_idle_policy, now_label, now_ts, status_label};
8180
pub(crate) use models::{
82-
AgentEvent, AgentLifecycleEvent, AgentLifecycleHistoryEntry, AgentStartResult,
83-
ArchiveEntry, ClaudeSlashSkillEntry, CommandAvailability, ExecTarget, FileNode,
84-
FilePreview, FilesystemEntry, FilesystemListResponse, FilesystemRoot, GitChangeEntry,
85-
GitFileDiffPayload, GitStatus, IdlePolicy, SessionInfo, SessionMessage,
86-
SessionMessageRole, SessionMode, SessionPatch, SessionStatus, TerminalEvent,
87-
TerminalInfo, TransportEvent, WorkbenchBootstrap, WorkbenchLayout, WorkbenchUiState,
88-
WorkspaceControllerLease, WorkspaceLaunchResult, WorkspaceRuntimeSnapshot,
89-
WorkspaceRuntimeStateEvent, WorkspaceSnapshot, WorkspaceSource, WorkspaceSourceKind,
90-
WorkspaceSummary, WorkspaceTree, WorkspaceViewPatch, WorkspaceViewState, WorktreeDetail,
91-
WorktreeInfo,
81+
AgentEvent, AgentLifecycleEvent, AgentLifecycleHistoryEntry, AgentStartResult, ArchiveEntry,
82+
ClaudeSlashSkillEntry, CommandAvailability, ExecTarget, FileNode, FilePreview, FilesystemEntry,
83+
FilesystemListResponse, FilesystemRoot, GitChangeEntry, GitFileDiffPayload, GitStatus,
84+
IdlePolicy, SessionInfo, SessionMessage, SessionMessageRole, SessionMode, SessionPatch,
85+
SessionStatus, TerminalEvent, TerminalInfo, TransportEvent, WorkbenchBootstrap,
86+
WorkbenchLayout, WorkbenchUiState, WorkspaceControllerLease, WorkspaceLaunchResult,
87+
WorkspaceRuntimeSnapshot, WorkspaceRuntimeStateEvent, WorkspaceSnapshot, WorkspaceSource,
88+
WorkspaceSourceKind, WorkspaceSummary, WorkspaceTree, WorkspaceViewPatch, WorkspaceViewState,
89+
WorktreeDetail, WorktreeInfo,
9290
};
9391
pub(crate) use runtime::{AppHandle, State};
9492
pub(crate) use services::agent::{

apps/server/src/services/terminal.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,12 @@ pub(crate) fn terminal_create(
9393
}
9494
emit_terminal(&app_handle, &workspace_id_out, terminal_id, &text);
9595
let state: State<AppState> = state_handle.state();
96-
let _ =
97-
append_workspace_terminal_output(state, &workspace_id_out, terminal_id, &text);
96+
let _ = append_workspace_terminal_output(
97+
state,
98+
&workspace_id_out,
99+
terminal_id,
100+
&text,
101+
);
98102
}
99103
Err(_) => break,
100104
}

0 commit comments

Comments
 (0)