Compare commits
46
Commits
b3d4b864fd
..
v0.3.7
No files matched your search
+10
-1
@@ -1 +1,10 @@
|
|||||||
SQLX_OFFLINE=true
|
# Live mode (local dev): query! macros validate every query against the real
|
||||||
|
# schema on each build, so the .sqlx cache is only needed for DB-less builds.
|
||||||
|
# Run scripts/prepare_sqlx_sqlite.sh once first — it creates and migrates the DB.
|
||||||
|
# Use an absolute path so builds work from any working directory.
|
||||||
|
# Replace the placeholder below with the DATABASE_URL printed by the script.
|
||||||
|
DATABASE_URL=sqlite:///absolute/path/to/wakey-prepare.sqlite3
|
||||||
|
|
||||||
|
# Offline mode (CI / packaging): force the committed .sqlx cache.
|
||||||
|
# Takes precedence over DATABASE_URL when set.
|
||||||
|
#SQLX_OFFLINE=true
|
||||||
@@ -20,3 +20,6 @@ target/
|
|||||||
.env
|
.env
|
||||||
|
|
||||||
.codex
|
.codex
|
||||||
|
|
||||||
|
# sqlx prepare DB (created by scripts/prepare_sqlx_sqlite.sh)
|
||||||
|
wakey-prepare.sqlite3*
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "INSERT INTO audit_events\n (event_key, event_id, ts_unix, actor_type, actor_id, agent_id, request_id,\n event_type, outcome, latency_ms, message, metadata_json)\n VALUES (?1, ?2, ?3, 'test', NULL, NULL, NULL, 'test', 'ok', NULL, 'old', '{}')",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 3
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "6ce45cc94cbfdd948aad5fb986f9b5976c79bb63759bc9ecff1a7873e2d70137"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "UPDATE known_devices SET display_name = ?1, updated_at_unix = ?2 WHERE device_id = ?3",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 3
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "897b794a7f94af42b545d74a5b859850fc9935f40385061b029fab5892a2aded"
|
||||||
|
}
|
||||||
+20
@@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "SELECT COUNT(*) FROM alert_transitions WHERE transition_id = ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [
|
||||||
|
{
|
||||||
|
"name": "COUNT(*)",
|
||||||
|
"ordinal": 0,
|
||||||
|
"type_info": "Integer"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": [
|
||||||
|
false
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"hash": "b324587b94ee2e98dc8eff5f6735303a2784b74f49068ce8e946cedcd8d12c4e"
|
||||||
|
}
|
||||||
+20
@@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "SELECT COUNT(*) FROM audit_events WHERE message = ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [
|
||||||
|
{
|
||||||
|
"name": "COUNT(*)",
|
||||||
|
"ordinal": 0,
|
||||||
|
"type_info": "Integer"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": [
|
||||||
|
false
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"hash": "bd1666cea8840df04ddcdc8a0c42b9b7771f9bf63c45613c4b6f8b86be479d70"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "DELETE FROM audit_events WHERE ts_unix < ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "ca612fee8c80f815b33dbc088cc7e0a7552f1f4e38390959e14ea5cc765bf2c7"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "DELETE FROM agent_devices WHERE last_seen_unix < ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "cc4b885c31f965013ccaa6f8ae33f6c8d68898654fc0873682c8212339088306"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "DELETE FROM alert_transitions WHERE ts_unix < ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "ce850a32c8702595bbc3207bf86d53209c48be145562192b3a8f84903d39eac1"
|
||||||
|
}
|
||||||
+20
@@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "SELECT COUNT(*) FROM audit_events WHERE event_id = ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [
|
||||||
|
{
|
||||||
|
"name": "COUNT(*)",
|
||||||
|
"ordinal": 0,
|
||||||
|
"type_info": "Integer"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": [
|
||||||
|
false
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"hash": "dc7056b7421dcc83648cbd360273ebcce1156ce0f7da2d39f054025ab9b66b6b"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "INSERT INTO alert_transitions\n (transition_key, transition_id, ts_unix, alert_id, kind, agent_id,\n from_status, to_status, message, metadata_json)\n VALUES (?1, ?2, ?3, 'alert-old', 'test', NULL, NULL, 'active', 'old', '{}')",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 3
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "e83a43217a3bfaacd2f5e76df1fd05447558646fb2b47963751a7ab53b3a99b6"
|
||||||
|
}
|
||||||
Generated
+200
-74
@@ -142,6 +142,7 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "a054912289d18629dc78375ba2c3726a3afe3ff71b4edba9dedfca0e3446d1fc"
|
checksum = "a054912289d18629dc78375ba2c3726a3afe3ff71b4edba9dedfca0e3446d1fc"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"aws-lc-sys",
|
"aws-lc-sys",
|
||||||
|
"untrusted 0.7.1",
|
||||||
"zeroize",
|
"zeroize",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -1013,6 +1014,17 @@ dependencies = [
|
|||||||
"pin-project-lite",
|
"pin-project-lite",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "http-cache-semantics"
|
||||||
|
version = "3.0.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "0603b99309a1e404c2c0945650c0f775b50bb72ac90ae570cb93f9ff05d9efe7"
|
||||||
|
dependencies = [
|
||||||
|
"http",
|
||||||
|
"reqwest",
|
||||||
|
"time",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "http-range-header"
|
name = "http-range-header"
|
||||||
version = "0.4.2"
|
version = "0.4.2"
|
||||||
@@ -1031,6 +1043,22 @@ version = "1.0.3"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "df3b46402a9d5adb4c86a0cf463f42e19994e3ee891101b1841f30a545cb49a9"
|
checksum = "df3b46402a9d5adb4c86a0cf463f42e19994e3ee891101b1841f30a545cb49a9"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "humantime"
|
||||||
|
version = "2.4.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "15cdd26707701c53297e2fa6afb323d55fbc1d0810c3aec078ae3ef0424c3c15"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "humantime-serde"
|
||||||
|
version = "1.1.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "57a3db5ea5923d99402c94e9feb261dc5ee9b4efa158b0315f788cf549cc200c"
|
||||||
|
dependencies = [
|
||||||
|
"humantime",
|
||||||
|
"serde",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "hyper"
|
name = "hyper"
|
||||||
version = "1.9.0"
|
version = "1.9.0"
|
||||||
@@ -1375,6 +1403,24 @@ dependencies = [
|
|||||||
"wasm-bindgen",
|
"wasm-bindgen",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "jsonwebtoken"
|
||||||
|
version = "10.4.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "eba32bfb4ffdeaca3e34431072faf01745c9b26d25504aa7a6cf5684334fc4fc"
|
||||||
|
dependencies = [
|
||||||
|
"aws-lc-rs",
|
||||||
|
"base64",
|
||||||
|
"getrandom 0.2.17",
|
||||||
|
"js-sys",
|
||||||
|
"pem",
|
||||||
|
"serde",
|
||||||
|
"serde_json",
|
||||||
|
"signature",
|
||||||
|
"simple_asn1",
|
||||||
|
"zeroize",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "lazy_static"
|
name = "lazy_static"
|
||||||
version = "1.5.0"
|
version = "1.5.0"
|
||||||
@@ -1637,6 +1683,16 @@ dependencies = [
|
|||||||
"windows-sys 0.61.2",
|
"windows-sys 0.61.2",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "num-bigint"
|
||||||
|
version = "0.4.8"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "c89e69e7e0f03bea5ef08013795c25018e101932225a656383bd384495ecc367"
|
||||||
|
dependencies = [
|
||||||
|
"num-integer",
|
||||||
|
"num-traits",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "num-bigint-dig"
|
name = "num-bigint-dig"
|
||||||
version = "0.8.6"
|
version = "0.8.6"
|
||||||
@@ -1727,9 +1783,9 @@ checksum = "7c87def4c32ab89d880effc9e097653c8da5d6ef28e6b539d313baaacfbafcbe"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "opentelemetry"
|
name = "opentelemetry"
|
||||||
version = "0.31.0"
|
version = "0.33.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "b84bcd6ae87133e903af7ef497404dda70c60d0ea14895fc8a5e6722754fc2a0"
|
checksum = "8c1f1a6ac9c875a4941e4c7c085b61e8f4daa326de03b4701768c3491a6ad9ee"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"futures-core",
|
"futures-core",
|
||||||
"futures-sink",
|
"futures-sink",
|
||||||
@@ -1740,42 +1796,39 @@ dependencies = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "opentelemetry-http"
|
name = "opentelemetry-appender-tracing"
|
||||||
version = "0.31.0"
|
version = "0.33.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "d7a6d09a73194e6b66df7c8f1b680f156d916a1a942abf2de06823dd02b7855d"
|
checksum = "1f01f1c53c46ce80e5fd8cb7a646f13766fd8f315a8a5b75b9bad59a08d2c5c7"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"async-trait",
|
|
||||||
"bytes",
|
|
||||||
"http",
|
|
||||||
"opentelemetry",
|
"opentelemetry",
|
||||||
"reqwest 0.12.28",
|
"tracing",
|
||||||
|
"tracing-core",
|
||||||
|
"tracing-subscriber",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "opentelemetry-otlp"
|
name = "opentelemetry-otlp"
|
||||||
version = "0.31.1"
|
version = "0.33.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "1f69cd6acbb9af919df949cd1ec9e5e7fdc2ef15d234b6b795aaa525cc02f71f"
|
checksum = "0756af24841e367128ef31c831999937b8b5c5197d47914590d8f4e2e88c2e1a"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"http",
|
"http",
|
||||||
"opentelemetry",
|
"opentelemetry",
|
||||||
"opentelemetry-http",
|
|
||||||
"opentelemetry-proto",
|
"opentelemetry-proto",
|
||||||
"opentelemetry_sdk",
|
"opentelemetry_sdk",
|
||||||
"prost",
|
"prost",
|
||||||
"reqwest 0.12.28",
|
|
||||||
"thiserror 2.0.18",
|
"thiserror 2.0.18",
|
||||||
"tokio",
|
"tokio",
|
||||||
"tonic",
|
"tonic",
|
||||||
"tracing",
|
"tonic-types",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "opentelemetry-proto"
|
name = "opentelemetry-proto"
|
||||||
version = "0.31.0"
|
version = "0.33.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "a7175df06de5eaee9909d4805a3d07e28bb752c34cab57fa9cff549da596b30f"
|
checksum = "d576f5dddc3299341bcd85a04be17b2564844c0c9563214f02d85f2524d1afae"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"opentelemetry",
|
"opentelemetry",
|
||||||
"opentelemetry_sdk",
|
"opentelemetry_sdk",
|
||||||
@@ -1786,19 +1839,18 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "opentelemetry_sdk"
|
name = "opentelemetry_sdk"
|
||||||
version = "0.31.0"
|
version = "0.33.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "e14ae4f5991976fd48df6d843de219ca6d31b01daaab2dad5af2badeded372bd"
|
checksum = "372a912db2974d1990207006fd0e3aa0ef3fd4ad3be43c65cfaadf2206866061"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"futures-channel",
|
"futures-channel",
|
||||||
"futures-executor",
|
"futures-executor",
|
||||||
"futures-util",
|
"futures-util",
|
||||||
"opentelemetry",
|
"opentelemetry",
|
||||||
"percent-encoding",
|
"percent-encoding",
|
||||||
|
"portable-atomic",
|
||||||
"rand 0.9.4",
|
"rand 0.9.4",
|
||||||
"thiserror 2.0.18",
|
"thiserror 2.0.18",
|
||||||
"tokio",
|
|
||||||
"tokio-stream",
|
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -1842,6 +1894,16 @@ version = "1.0.15"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "57c0d7b74b563b49d38dae00a0c37d4d6de9b432382b2892f0574ddcae73fd0a"
|
checksum = "57c0d7b74b563b49d38dae00a0c37d4d6de9b432382b2892f0574ddcae73fd0a"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "pem"
|
||||||
|
version = "3.0.6"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "1d30c53c26bc5b31a98cd02d20f25a7c8567146caf63ed593a9d87b2775291be"
|
||||||
|
dependencies = [
|
||||||
|
"base64",
|
||||||
|
"serde_core",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "pem-rfc7468"
|
name = "pem-rfc7468"
|
||||||
version = "0.7.0"
|
version = "0.7.0"
|
||||||
@@ -1916,6 +1978,12 @@ version = "0.2.3"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "b4596b6d070b27117e987119b4dac604f3c58cfb0b191112e24771b2faeac1a6"
|
checksum = "b4596b6d070b27117e987119b4dac604f3c58cfb0b191112e24771b2faeac1a6"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "portable-atomic"
|
||||||
|
version = "1.15.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "05c8b63e8d9609db387f0324918f81d68fe27748f084ef092fb35954d0539a85"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "potential_utf"
|
name = "potential_utf"
|
||||||
version = "0.1.5"
|
version = "0.1.5"
|
||||||
@@ -1959,6 +2027,27 @@ dependencies = [
|
|||||||
"unicode-ident",
|
"unicode-ident",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "procfs"
|
||||||
|
version = "0.18.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "25485360a54d6861439d60facef26de713b1e126bf015ec8f98239467a2b82f7"
|
||||||
|
dependencies = [
|
||||||
|
"bitflags",
|
||||||
|
"procfs-core",
|
||||||
|
"rustix",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "procfs-core"
|
||||||
|
version = "0.18.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "e6401bf7b6af22f78b563665d15a22e9aef27775b79b149a66ca022468a4e405"
|
||||||
|
dependencies = [
|
||||||
|
"bitflags",
|
||||||
|
"hex",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "prost"
|
name = "prost"
|
||||||
version = "0.14.3"
|
version = "0.14.3"
|
||||||
@@ -1982,6 +2071,15 @@ dependencies = [
|
|||||||
"syn",
|
"syn",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "prost-types"
|
||||||
|
version = "0.14.3"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "8991c4cbdb8bc5b11f0b074ffe286c30e523de90fee5ba8132f1399f23cb3dd7"
|
||||||
|
dependencies = [
|
||||||
|
"prost",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "pty-process"
|
name = "pty-process"
|
||||||
version = "0.5.3"
|
version = "0.5.3"
|
||||||
@@ -2183,40 +2281,6 @@ version = "0.8.10"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "dc897dd8d9e8bd1ed8cdad82b5966c3e0ecae09fb1907d58efaa013543185d0a"
|
checksum = "dc897dd8d9e8bd1ed8cdad82b5966c3e0ecae09fb1907d58efaa013543185d0a"
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "reqwest"
|
|
||||||
version = "0.12.28"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "eddd3ca559203180a307f12d114c268abf583f59b03cb906fd0b3ff8646c1147"
|
|
||||||
dependencies = [
|
|
||||||
"base64",
|
|
||||||
"bytes",
|
|
||||||
"futures-channel",
|
|
||||||
"futures-core",
|
|
||||||
"futures-util",
|
|
||||||
"http",
|
|
||||||
"http-body",
|
|
||||||
"http-body-util",
|
|
||||||
"hyper",
|
|
||||||
"hyper-util",
|
|
||||||
"js-sys",
|
|
||||||
"log",
|
|
||||||
"percent-encoding",
|
|
||||||
"pin-project-lite",
|
|
||||||
"serde",
|
|
||||||
"serde_json",
|
|
||||||
"serde_urlencoded",
|
|
||||||
"sync_wrapper",
|
|
||||||
"tokio",
|
|
||||||
"tower",
|
|
||||||
"tower-http",
|
|
||||||
"tower-service",
|
|
||||||
"url",
|
|
||||||
"wasm-bindgen",
|
|
||||||
"wasm-bindgen-futures",
|
|
||||||
"web-sys",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "reqwest"
|
name = "reqwest"
|
||||||
version = "0.13.2"
|
version = "0.13.2"
|
||||||
@@ -2264,7 +2328,7 @@ dependencies = [
|
|||||||
"cfg-if",
|
"cfg-if",
|
||||||
"getrandom 0.2.17",
|
"getrandom 0.2.17",
|
||||||
"libc",
|
"libc",
|
||||||
"untrusted",
|
"untrusted 0.9.0",
|
||||||
"windows-sys 0.52.0",
|
"windows-sys 0.52.0",
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -2403,7 +2467,7 @@ dependencies = [
|
|||||||
"aws-lc-rs",
|
"aws-lc-rs",
|
||||||
"ring",
|
"ring",
|
||||||
"rustls-pki-types",
|
"rustls-pki-types",
|
||||||
"untrusted",
|
"untrusted 0.9.0",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -2699,6 +2763,18 @@ dependencies = [
|
|||||||
"rand_core 0.6.4",
|
"rand_core 0.6.4",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "simple_asn1"
|
||||||
|
version = "0.6.4"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "0d585997b0ac10be3c5ee635f1bab02d512760d14b7c468801ac8a01d9ae5f1d"
|
||||||
|
dependencies = [
|
||||||
|
"num-bigint",
|
||||||
|
"num-traits",
|
||||||
|
"thiserror 2.0.18",
|
||||||
|
"time",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "slab"
|
name = "slab"
|
||||||
version = "0.4.12"
|
version = "0.4.12"
|
||||||
@@ -3258,8 +3334,10 @@ source = "registry+https://github.com/rust-lang/crates.io-index"
|
|||||||
checksum = "fec7c61a0695dc1887c1b53952990f3ad2e3a31453e1f49f10e75424943a93ec"
|
checksum = "fec7c61a0695dc1887c1b53952990f3ad2e3a31453e1f49f10e75424943a93ec"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"async-trait",
|
"async-trait",
|
||||||
|
"axum",
|
||||||
"base64",
|
"base64",
|
||||||
"bytes",
|
"bytes",
|
||||||
|
"h2",
|
||||||
"http",
|
"http",
|
||||||
"http-body",
|
"http-body",
|
||||||
"http-body-util",
|
"http-body-util",
|
||||||
@@ -3268,6 +3346,7 @@ dependencies = [
|
|||||||
"hyper-util",
|
"hyper-util",
|
||||||
"percent-encoding",
|
"percent-encoding",
|
||||||
"pin-project",
|
"pin-project",
|
||||||
|
"socket2",
|
||||||
"sync_wrapper",
|
"sync_wrapper",
|
||||||
"tokio",
|
"tokio",
|
||||||
"tokio-stream",
|
"tokio-stream",
|
||||||
@@ -3288,6 +3367,17 @@ dependencies = [
|
|||||||
"tonic",
|
"tonic",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "tonic-types"
|
||||||
|
version = "0.14.5"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "2a875a902255423d34c1f20838ab374126db8eb41625b7947a1d54113b0b7399"
|
||||||
|
dependencies = [
|
||||||
|
"prost",
|
||||||
|
"prost-types",
|
||||||
|
"tonic",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tower"
|
name = "tower"
|
||||||
version = "0.5.3"
|
version = "0.5.3"
|
||||||
@@ -3403,9 +3493,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tracing-opentelemetry"
|
name = "tracing-opentelemetry"
|
||||||
version = "0.32.1"
|
version = "0.34.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "1ac28f2d093c6c477eaa76b23525478f38de514fa9aeb1285738d4b97a9552fc"
|
checksum = "0a904802a1b902f43638b677ff2a650847e3b4404101b6c586d648e8c1e3e8fe"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"js-sys",
|
"js-sys",
|
||||||
"opentelemetry",
|
"opentelemetry",
|
||||||
@@ -3547,6 +3637,12 @@ version = "0.2.6"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "ebc1c04c71510c7f702b52b7c350734c9ff1295c464a03335b00bb84fc54f853"
|
checksum = "ebc1c04c71510c7f702b52b7c350734c9ff1295c464a03335b00bb84fc54f853"
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "untrusted"
|
||||||
|
version = "0.7.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "a156c684c91ea7d62626509bce3cb4e1d9ed5c4d978f7b4352658f96a4c26b4a"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "untrusted"
|
name = "untrusted"
|
||||||
version = "0.9.0"
|
version = "0.9.0"
|
||||||
@@ -3624,7 +3720,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "wakey"
|
name = "wakey"
|
||||||
version = "0.3.3"
|
version = "0.3.7"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"chrono",
|
"chrono",
|
||||||
@@ -3645,90 +3741,106 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "wakey-agent"
|
name = "wakey-agent"
|
||||||
version = "0.3.3"
|
version = "0.3.7"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"clap",
|
"clap",
|
||||||
"futures-util",
|
"futures-util",
|
||||||
"http",
|
"http",
|
||||||
|
"humantime-serde",
|
||||||
"macaddr",
|
"macaddr",
|
||||||
"nix 0.31.2",
|
"nix 0.31.2",
|
||||||
"reqwest 0.13.2",
|
"reqwest",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"time",
|
|
||||||
"tokio",
|
"tokio",
|
||||||
"tokio-tungstenite 0.29.0",
|
"tokio-tungstenite 0.29.0",
|
||||||
"toml",
|
"toml",
|
||||||
"tracing",
|
"tracing",
|
||||||
"tracing-subscriber",
|
|
||||||
"url",
|
"url",
|
||||||
"wakey",
|
"wakey",
|
||||||
"wakey-core",
|
"wakey-core",
|
||||||
|
"wakey-linux",
|
||||||
"wakey-vt100",
|
"wakey-vt100",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "wakey-control-plane"
|
name = "wakey-control-plane"
|
||||||
version = "0.3.3"
|
version = "0.3.7"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"axum",
|
"axum",
|
||||||
"clap",
|
"clap",
|
||||||
"futures-util",
|
"futures-util",
|
||||||
|
"http-cache-semantics",
|
||||||
|
"humantime-serde",
|
||||||
|
"jsonwebtoken",
|
||||||
"macaddr",
|
"macaddr",
|
||||||
"nix 0.31.2",
|
"nix 0.31.2",
|
||||||
"opentelemetry",
|
"reqwest",
|
||||||
"opentelemetry-otlp",
|
|
||||||
"opentelemetry_sdk",
|
|
||||||
"reqwest 0.13.2",
|
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"sqlx",
|
"sqlx",
|
||||||
"time",
|
"thiserror 2.0.18",
|
||||||
"tokio",
|
"tokio",
|
||||||
"toml",
|
"toml",
|
||||||
|
"tower",
|
||||||
"tower-http",
|
"tower-http",
|
||||||
"tracing",
|
"tracing",
|
||||||
"tracing-opentelemetry",
|
|
||||||
"tracing-subscriber",
|
|
||||||
"uuid",
|
"uuid",
|
||||||
"wakey-agent",
|
"wakey-agent",
|
||||||
"wakey-core",
|
"wakey-core",
|
||||||
|
"wakey-linux",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "wakey-core"
|
name = "wakey-core"
|
||||||
version = "0.3.3"
|
version = "0.3.7"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
|
"anyhow",
|
||||||
|
"humantime",
|
||||||
"macaddr",
|
"macaddr",
|
||||||
|
"opentelemetry",
|
||||||
|
"opentelemetry-appender-tracing",
|
||||||
|
"opentelemetry-otlp",
|
||||||
|
"opentelemetry-proto",
|
||||||
|
"opentelemetry_sdk",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"serde_with",
|
"serde_with",
|
||||||
"strum",
|
"strum",
|
||||||
"thiserror 2.0.18",
|
"thiserror 2.0.18",
|
||||||
|
"time",
|
||||||
|
"tokio",
|
||||||
|
"tokio-stream",
|
||||||
|
"tonic",
|
||||||
|
"tracing",
|
||||||
|
"tracing-opentelemetry",
|
||||||
|
"tracing-subscriber",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "wakey-linux"
|
name = "wakey-linux"
|
||||||
version = "0.3.3"
|
version = "0.3.7"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"futures",
|
"futures",
|
||||||
"lda-ipjs",
|
"lda-ipjs",
|
||||||
"macaddr",
|
"macaddr",
|
||||||
"nix 0.31.2",
|
"nix 0.31.2",
|
||||||
|
"procfs",
|
||||||
"pty-process",
|
"pty-process",
|
||||||
"serde",
|
"serde",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"serial_test",
|
"serial_test",
|
||||||
"tokio",
|
"tokio",
|
||||||
|
"tracing",
|
||||||
"wakey-core",
|
"wakey-core",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "wakey-vt100"
|
name = "wakey-vt100"
|
||||||
version = "0.16.2-wakey.1"
|
version = "0.16.2-wakey.2"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"itoa",
|
"itoa",
|
||||||
"unicode-width",
|
"unicode-width",
|
||||||
@@ -4477,6 +4589,20 @@ name = "zeroize"
|
|||||||
version = "1.8.2"
|
version = "1.8.2"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "b97154e67e32c85465826e8bcc1c59429aaaf107c1e4a9e53c8d8ccd5eff88d0"
|
checksum = "b97154e67e32c85465826e8bcc1c59429aaaf107c1e4a9e53c8d8ccd5eff88d0"
|
||||||
|
dependencies = [
|
||||||
|
"zeroize_derive",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "zeroize_derive"
|
||||||
|
version = "1.5.0"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "3c50655cbb0fe3fc43170059e702f1ce5e19b84cec58dc87b037a09935c2f328"
|
||||||
|
dependencies = [
|
||||||
|
"proc-macro2",
|
||||||
|
"quote",
|
||||||
|
"syn",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "zerotrie"
|
name = "zerotrie"
|
||||||
|
|||||||
+1
-1
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "wakey"
|
name = "wakey"
|
||||||
version = "0.3.3"
|
version = "0.3.7"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
publish = ["gitea"]
|
publish = ["gitea"]
|
||||||
|
|
||||||
|
|||||||
@@ -207,10 +207,23 @@ Control-plane admin API includes token management endpoints:
|
|||||||
- `GET /api/v1/control/audit/events?agent_id=<id>&event_type=<type>&limit=<n>`
|
- `GET /api/v1/control/audit/events?agent_id=<id>&event_type=<type>&limit=<n>`
|
||||||
- `GET /api/v1/control/alerts?lookback_seconds=900`
|
- `GET /api/v1/control/alerts?lookback_seconds=900`
|
||||||
- `GET /api/v1/control/alerts/history?since_unix=<ts>&limit=<n>`
|
- `GET /api/v1/control/alerts/history?since_unix=<ts>&limit=<n>`
|
||||||
- `GET /api/v1/control/alerts/ws` (websocket snapshots + recent transitions)
|
- `GET /api/v1/control/alerts/ws` (websocket snapshots, recent transitions, and evaluator health)
|
||||||
|
- `GET /api/v1/control/terminals/events/ws` (websocket session-list invalidations)
|
||||||
- `DELETE /api/v1/control/agents/{agent_id}`
|
- `DELETE /api/v1/control/agents/{agent_id}`
|
||||||
- `PATCH /api/v1/control/agents/{agent_id}/nickname`
|
- `PATCH /api/v1/control/agents/{agent_id}/nickname`
|
||||||
|
|
||||||
|
Alert evaluation runs once per control plane every five seconds. The alerts GET
|
||||||
|
returns `503 alerts_unavailable` until a successful evaluation, after evaluation
|
||||||
|
failure, or when its snapshot is stale. The existing alerts stream also sends
|
||||||
|
`alerts_health` messages with `available` and an optional `reason`; the UI keeps
|
||||||
|
the last results visible with a stale warning until recovery. Custom alert-rule
|
||||||
|
query overrides are rejected with `400 unsupported_alert_override`.
|
||||||
|
|
||||||
|
Terminal session events carry only `sessions_changed` invalidations. Clients
|
||||||
|
refetch the session list after changes or reconnection and retain a 15-second
|
||||||
|
fallback refresh. Events do not grant session ownership; attachment still uses
|
||||||
|
the existing server-side locking and scoped credentials.
|
||||||
|
|
||||||
If commands still appear silent, verify both processes are running with `-v`
|
If commands still appear silent, verify both processes are running with `-v`
|
||||||
and that `RUST_LOG` is not overriding to a stricter level.
|
and that `RUST_LOG` is not overriding to a stricter level.
|
||||||
|
|
||||||
|
|||||||
@@ -2,9 +2,11 @@
|
|||||||
#
|
#
|
||||||
# Security model:
|
# Security model:
|
||||||
# - Public endpoints for agents: health, enrollment, control WS, and scoped terminal relay WS
|
# - Public endpoints for agents: health, enrollment, control WS, and scoped terminal relay WS
|
||||||
# - Private admin surface: /ui/* and /api/v1/control/* (requires CF Access headers)
|
# - Private admin surface: root, /ui/* and /api/v1/control/* (verified Access JWT)
|
||||||
|
# Configure CC's [access] section and loopback bind before enabling this template.
|
||||||
|
# Never expose /internal/auth/verify through this virtual host.
|
||||||
#
|
#
|
||||||
# Replace cp.example.com with your public control-plane domain.
|
# Replace wakey.ldlda.com and the loopback upstream port for your deployment.
|
||||||
|
|
||||||
wakey.ldlda.com {
|
wakey.ldlda.com {
|
||||||
encode zstd gzip
|
encode zstd gzip
|
||||||
@@ -15,15 +17,14 @@ wakey.ldlda.com {
|
|||||||
reverse_proxy 127.0.0.1:6767
|
reverse_proxy 127.0.0.1:6767
|
||||||
}
|
}
|
||||||
|
|
||||||
# Admin surface requires Cloudflare Access headers.
|
# Applies to HTTP requests and operator WebSocket upgrades, not WS frames.
|
||||||
@admin path /ui* /api/v1/control/*
|
@admin path / /ui* /api/v1/control/*
|
||||||
@cf_access header_regexp CFJWT Cf-Access-Jwt-Assertion .+
|
|
||||||
|
|
||||||
handle @admin {
|
handle @admin {
|
||||||
handle @cf_access {
|
forward_auth 127.0.0.1:6767 {
|
||||||
reverse_proxy 127.0.0.1:6767
|
uri /internal/auth/verify
|
||||||
}
|
}
|
||||||
respond "forbidden" 403
|
reverse_proxy 127.0.0.1:6767
|
||||||
}
|
}
|
||||||
|
|
||||||
# Deny unknown paths by default.
|
# Deny unknown paths by default.
|
||||||
|
|||||||
@@ -12,6 +12,7 @@ ExecStart=/opt/wakey/bin/wakey-control-plane serve --config-file /etc/wakey-cont
|
|||||||
Restart=on-failure
|
Restart=on-failure
|
||||||
RestartSec=2
|
RestartSec=2
|
||||||
Environment=RUST_LOG=info
|
Environment=RUST_LOG=info
|
||||||
|
EnvironmentFile=-/etc/wakey-control-plane/environment
|
||||||
|
|
||||||
[Install]
|
[Install]
|
||||||
WantedBy=multi-user.target
|
WantedBy=multi-user.target
|
||||||
@@ -75,8 +75,23 @@ Current DHCP leases are summary-eligible and may support wake if they include a
|
|||||||
|
|
||||||
`Device.ips` and `Device.macs` stay as summaries. Summary IPs come from concrete live-ish sources such as current neighbor rows and current DHCP lease rows. Hook-derived and offline IPs remain visible through endpoints but are collapsed out of primary UI summaries by default.
|
`Device.ips` and `Device.macs` stay as summaries. Summary IPs come from concrete live-ish sources such as current neighbor rows and current DHCP lease rows. Hook-derived and offline IPs remain visible through endpoints but are collapsed out of primary UI summaries by default.
|
||||||
|
|
||||||
|
Fleet display labels prefer the saved device name, then a hostname, then a MAC,
|
||||||
|
then an IP. Generated address labels use `Device: <address>` without presence
|
||||||
|
prefixes; the badge carries presence. An inactive IP is only a naming fallback
|
||||||
|
when no name or MAC exists, never evidence of a current address or a wake route.
|
||||||
|
Offline neighbor evidence does not establish that a DHCP address has expired or
|
||||||
|
been reassigned. If a concrete source observes that endpoint again, its eligible
|
||||||
|
IP returns to the summary; manually remembered identifiers are not removed by this.
|
||||||
|
|
||||||
`Device.presence` is derived from all endpoints, not only summary-eligible endpoints. This lets concrete offline evidence matter while allowing stronger live evidence to override it.
|
`Device.presence` is derived from all endpoints, not only summary-eligible endpoints. This lets concrete offline evidence matter while allowing stronger live evidence to override it.
|
||||||
|
|
||||||
|
The control plane derives stored snapshot presence and fleet presence from the
|
||||||
|
maximum endpoint signal whenever endpoints exist. A cached summary must not
|
||||||
|
override endpoint evidence; endpoint-less legacy rows fall back to their stored
|
||||||
|
summary. Presence describes the last observation, not guaranteed current
|
||||||
|
reachability. Agent disconnection does not rewrite endpoint presence or expire
|
||||||
|
the retained snapshot.
|
||||||
|
|
||||||
The main fleet API embeds endpoints inside each fleet device. API responses expose all endpoints, including offline and unknown endpoints. The UI may collapse offline/unknown endpoints in the primary row, but details must expose them for copy/debug/history.
|
The main fleet API embeds endpoints inside each fleet device. API responses expose all endpoints, including offline and unknown endpoints. The UI may collapse offline/unknown endpoints in the primary row, but details must expose them for copy/debug/history.
|
||||||
|
|
||||||
## Wakeability and Wake Route Ordering
|
## Wakeability and Wake Route Ordering
|
||||||
|
|||||||
@@ -72,7 +72,11 @@ When the operator socket disconnects, the session becomes detached. Navigation,
|
|||||||
|
|
||||||
The operator UI lists live sessions as tabs and stores only the active terminal ID in browser-tab-scoped `sessionStorage`. On reload it prefers that ID and briefly waits for the previous WebSocket to detach before requesting a fresh attachment credential. The stored ID is a navigation hint, not an ownership credential; another attached operator still locks the session at the control plane.
|
The operator UI lists live sessions as tabs and stores only the active terminal ID in browser-tab-scoped `sessionStorage`. On reload it prefers that ID and briefly waits for the previous WebSocket to detach before requesting a fresh attachment credential. The stored ID is a navigation hint, not an ownership credential; another attached operator still locks the session at the control plane.
|
||||||
|
|
||||||
After operator attachment, the control plane sends a `snapshot` request. The agent serializes up to 5,000 retained physical rows followed by its parsed current screen and input modes as terminal escape bytes, sends them before subsequent live output, and then signals the PTY's current foreground process group with `SIGWINCH`. Historical rows retain their terminal attributes and are replayed as ordinary terminal output so xterm builds its native scrollback; the final formatted state restores the exact live screen. The signal lets full-screen applications redraw without assuming that the original shell remains in the foreground. Signal failure is logged but never terminates the session.
|
The protected `/api/v1/control/terminals/events/ws` stream sends `sessions_changed` invalidations when the volatile session catalog or attachment ownership changes. These events contain no terminal bytes or attachment credentials: browsers refetch the ordinary session list, coalescing bursts and repeating a fetch if another event arrives during it. Subscribing and reconnecting also invalidate the list. The 15-second refresh remains a recovery fallback; the stream is a freshness hint, never an authorization or locking mechanism.
|
||||||
|
|
||||||
|
The React page composes three custom hooks rather than owning every resource itself. `useXterm` owns the single renderer, its addons, fitting, fonts, and display operations. `useTerminalSessions` owns the catalog subscription, coalesced refreshes, and fallback timer. `useTerminalSession` owns active attachment, input, restoration, and session operations. The attachment hook waits for renderer readiness; stable renderer operations prevent ordinary state updates from rebuilding the terminal or reconnecting its transport. Each hook cleans up its own resources on unmount, which detaches the browser without deleting the agent-owned PTY.
|
||||||
|
|
||||||
|
After operator attachment, the control plane sends a `snapshot` request. The agent serializes up to 1,000 retained physical rows followed by its parsed current screen and input modes as terminal escape bytes, sends them before subsequent live output, and then signals the PTY's current foreground process group with `SIGWINCH`. Historical rows retain their terminal attributes and are replayed as ordinary terminal output so xterm builds its native scrollback; the final formatted state restores the exact live screen. The signal lets full-screen applications redraw without assuming that the original shell remains in the foreground. Signal failure is logged but never terminates the session. The connected browser independently retains up to 10,000 rows because that memory belongs to the operator device rather than the agent host.
|
||||||
|
|
||||||
Wakey carries a small, read-only extension to `vt100` because its public API exposes only the active grid. A snapshot streams bounded history from the primary grid, redraws that grid with its saved cursor and attributes, and then reconstructs the alternate grid when a full-screen application is active. The parser is never resized or switched while taking a snapshot. Consequently, reconnecting during `btop` restores the application while preserving the hidden shell history and state that reappear when it exits.
|
Wakey carries a small, read-only extension to `vt100` because its public API exposes only the active grid. A snapshot streams bounded history from the primary grid, redraws that grid with its saved cursor and attributes, and then reconstructs the alternate grid when a full-screen application is active. The parser is never resized or switched while taking a snapshot. Consequently, reconnecting during `btop` restores the application while preserving the hidden shell history and state that reappear when it exits.
|
||||||
|
|
||||||
@@ -80,7 +84,7 @@ The agent keeps its terminal manager outside the control-WebSocket reconnect loo
|
|||||||
|
|
||||||
Closing a session must terminate the entire PTY process group, not only the shell process. Cleanup should send a hangup first, then escalate to termination and forced kill after bounded grace periods. Explicit operator close, absolute session timeout, agent process exit, and machine reboot close the session.
|
Closing a session must terminate the entire PTY process group, not only the shell process. Cleanup should send a hangup first, then escalate to termination and forced kill after bounded grace periods. Explicit operator close, absolute session timeout, agent process exit, and machine reboot close the session.
|
||||||
|
|
||||||
The agent configuration sets `terminal.session_ttl_seconds` for newly created PTYs. The default is 43,200 seconds (12 hours); explicit zero disables automatic expiry. The agent owns and enforces this deadline locally because PTYs survive browser and control-plane disconnection. It advertises the policy to the control plane and includes the creation-time TTL in terminal inventory, allowing CC to mirror expiry and preserve it across CC restart. Missing fields from older agents retain the 12-hour default. Positive values are exact and values that cannot fit the platform timer are rejected rather than clamped.
|
The agent configuration sets `terminal.session_ttl` (for example, `"12h"`) for newly created PTYs. The default is 12 hours; `"0s"` disables automatic expiry. Legacy `session_ttl_seconds` is accepted with a deprecation warning. The wire protocol still uses integer seconds, so fractional-second TTLs are rejected. The agent owns and enforces this deadline locally because PTYs survive browser and control-plane disconnection. It advertises the policy to the control plane and includes the creation-time TTL in terminal inventory, allowing CC to mirror expiry and preserve it across CC restart. Missing fields from older agents retain the 12-hour default. Positive values are exact and values that cannot fit the platform timer are rejected rather than clamped.
|
||||||
|
|
||||||
An agent process restart or machine reboot still ends every terminal session because PTY file descriptors cannot be recovered by a new process. Surviving that boundary would require an external session host such as `tmux` or a separate terminal daemon and is not part of this design.
|
An agent process restart or machine reboot still ends every terminal session because PTY file descriptors cannot be recovered by a new process. Surviving that boundary would require an external session host such as `tmux` or a separate terminal daemon and is not part of this design.
|
||||||
|
|
||||||
@@ -90,7 +94,9 @@ The agent reports a normal exit status or terminating signal when available. The
|
|||||||
|
|
||||||
Terminal transport must not use unbounded queues.
|
Terminal transport must not use unbounded queues.
|
||||||
|
|
||||||
Every queue between PTY, agent socket, control-plane relay, and browser socket is bounded. The agent keeps draining and parsing the PTY while detached so a noisy child cannot stall the agent. Parsed state is bounded by the configured terminal dimensions and a 5,000-row scrollback limit. The control plane retains only a bounded queue of operator controls while an agent relay is reconnecting.
|
Every queue between PTY, agent socket, control-plane relay, and browser socket is bounded. The agent keeps draining and parsing the PTY while detached so a noisy child cannot stall the agent. Parsed state is bounded by the configured terminal dimensions and a 1,000-row reconnect scrollback limit. The control plane retains only a bounded queue of operator controls while an agent relay is reconnecting.
|
||||||
|
|
||||||
|
While a relay is attached, saturation applies lossless backpressure instead of dropping terminal bytes or disconnecting the relay. The agent retains the frame that discovered saturation and pauses further PTY reads until bounded relay capacity returns. Input, resize, close, lifecycle, and reconnect handling remain active while output is backpressured. Reconnect snapshots use the same ordered output path and duplicate snapshot requests are coalesced while one is pending. This is required for terminal protocols such as sixel, where dropping one output frame corrupts the rest of the stream.
|
||||||
|
|
||||||
When input/control and output are ready simultaneously, relay loops prioritize input and control. This keeps interrupt, resize, close, and snapshot traffic responsive while commands produce sustained output.
|
When input/control and output are ready simultaneously, relay loops prioritize input and control. This keeps interrupt, resize, close, and snapshot traffic responsive while commands produce sustained output.
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,88 @@
|
|||||||
|
# Configuration Durations
|
||||||
|
|
||||||
|
Agent and control-plane TOML duration settings use strings with explicit units.
|
||||||
|
Runtime code receives `std::time::Duration`; API and WebSocket numeric fields
|
||||||
|
such as `timeout_ms`, `session_ttl_seconds`, and `latency_ms` are unchanged.
|
||||||
|
|
||||||
|
## Agent
|
||||||
|
|
||||||
|
```toml
|
||||||
|
reconnect_base = "1s"
|
||||||
|
reconnect_max = "30s"
|
||||||
|
observation_sync_interval = "60s"
|
||||||
|
observation_retention = "7d"
|
||||||
|
|
||||||
|
[terminal]
|
||||||
|
session_ttl = "12h"
|
||||||
|
```
|
||||||
|
|
||||||
|
`observation_retention` prunes the agent's local hotplug memory during inventory
|
||||||
|
collection. It does not delete the control plane's last snapshot for an agent.
|
||||||
|
`"0d"` disables this local pruning. `terminal.session_ttl = "0s"` disables PTY
|
||||||
|
expiry. Reconnect and sync timers must be positive and fit the platform timer.
|
||||||
|
Terminal TTL and hook retention must resolve to whole seconds because their
|
||||||
|
protocol/storage timestamps use seconds. Reconnect and sync intervals can use
|
||||||
|
subsecond values.
|
||||||
|
Reconnect scheduling retains its existing 100ms minimum backoff.
|
||||||
|
|
||||||
|
## Control Plane
|
||||||
|
|
||||||
|
```toml
|
||||||
|
command_timeout = "30s"
|
||||||
|
enroll_token_ttl = "1d"
|
||||||
|
history_retention = "30d"
|
||||||
|
device_retention = "60d"
|
||||||
|
```
|
||||||
|
|
||||||
|
`command_timeout`, `enroll_token_ttl`, and `history_retention` must be positive.
|
||||||
|
Enrollment TTL and history retention must resolve to whole seconds. Command
|
||||||
|
timeout can use subsecond values.
|
||||||
|
`history_retention` deletes old audit events and alert transitions, not device
|
||||||
|
snapshots or saved identities.
|
||||||
|
|
||||||
|
`device_retention` expires agent-collected device rows and their child endpoints,
|
||||||
|
addresses, names and facts. The default is `"60d"`; `"0d"` disables expiry.
|
||||||
|
It must resolve to whole seconds. Age is measured from the last CC-received
|
||||||
|
snapshot that included the device, not endpoint timestamps or online presence.
|
||||||
|
An offline device included in new snapshots therefore stays retained. Saved
|
||||||
|
devices and their manually attached identifiers are never deleted by this cleanup.
|
||||||
|
Complete snapshots still remove devices omitted by that agent immediately;
|
||||||
|
retention handles old rows left behind when an agent stops syncing.
|
||||||
|
Expired rows are removed by the Unix daemon's existing maintenance loop every
|
||||||
|
five minutes, including an initial tick at startup. Restart CC to apply changes
|
||||||
|
to this setting; the state-only reload does not reload configuration.
|
||||||
|
|
||||||
|
## Legacy Files
|
||||||
|
|
||||||
|
| Old Key | New Key | Numeric Legacy Unit |
|
||||||
|
| --- | --- | --- |
|
||||||
|
| Agent `reconnect_base_ms` | `reconnect_base` | milliseconds |
|
||||||
|
| Agent `reconnect_max_ms` | `reconnect_max` | milliseconds |
|
||||||
|
| Agent `observation_sync_interval_seconds` | `observation_sync_interval` | seconds |
|
||||||
|
| Agent `observation_retention_days` | `observation_retention` | days |
|
||||||
|
| Agent `terminal.session_ttl_seconds` | `terminal.session_ttl` | seconds |
|
||||||
|
| CC `command_timeout_ms` | `command_timeout` | milliseconds |
|
||||||
|
| CC `enroll_token_ttl_seconds` | `enroll_token_ttl` | seconds |
|
||||||
|
| CC `observation_retention_seconds` | `history_retention` | seconds |
|
||||||
|
|
||||||
|
Old keys accept non-negative integers in their original units, or explicit
|
||||||
|
duration strings, and produce a warning on stderr when loaded. New keys require
|
||||||
|
strings; bare numbers are rejected. Supplying both keys for the same setting is
|
||||||
|
an error, even when their values agree. Negative, malformed, and overflowing
|
||||||
|
values are rejected; zero active timers are rejected rather than silently
|
||||||
|
clamped. No secrets are included in deprecation warnings.
|
||||||
|
|
||||||
|
`init-config --from-config ...` reads legacy files and writes only canonical
|
||||||
|
keys and formatted duration strings. Existing numeric CLI flags are unchanged
|
||||||
|
in this slice, and their overrides also serialize in the new format.
|
||||||
|
|
||||||
|
For example, legacy `reconnect_base_ms = 1500` is read as 1500 milliseconds
|
||||||
|
and written back as `reconnect_base = "1s 500ms"`. New-key
|
||||||
|
`reconnect_base = 1500` is rejected: no implicit numeric unit is assigned.
|
||||||
|
Serialization always writes duration strings, regardless of the accepted input
|
||||||
|
format, using `humantime-serde`.
|
||||||
|
|
||||||
|
The parser is [humantime](https://docs.rs/humantime/latest/humantime/fn.parse_duration.html).
|
||||||
|
Examples include `"14d"`, `"168h"`, `"250ms"`, and `"1h 30m"`. These are elapsed
|
||||||
|
durations, not calendar scheduling rules; use days/hours for retention rather
|
||||||
|
than assuming calendar-month semantics.
|
||||||
@@ -0,0 +1,54 @@
|
|||||||
|
# Fleet View Cache
|
||||||
|
|
||||||
|
CC retains one complete, grouped fleet view in memory. List requests still apply
|
||||||
|
their own search, visibility, presence, agent filters and result limit. Neither
|
||||||
|
the database nor the response format changes.
|
||||||
|
|
||||||
|
## Inputs And Invalidation
|
||||||
|
|
||||||
|
The cache key combines the store's fleet revision and connected-agent IDs.
|
||||||
|
Successful snapshot writes, saved-device create/rename/forget/merge, identifier
|
||||||
|
edits, agent enrollment/revocation/nickname edits, and device-retention cleanup
|
||||||
|
advance the revision. Token-only, audit and alert writes do not invalidate it.
|
||||||
|
Connection changes are detected from the live session map on each read.
|
||||||
|
|
||||||
|
The next reader rebuilds a changed view. Concurrent readers wait for that build
|
||||||
|
and reuse it. The helper checks the key again after building: if inputs changed
|
||||||
|
while database reads/grouping were in flight, it discards the result and retries.
|
||||||
|
Build errors are returned, not cached; cancelling a reader releases the lock.
|
||||||
|
The previous view is not served as a fallback after a rebuild error.
|
||||||
|
After three invalidated builds, the reader stops and releases the lock. Fleet
|
||||||
|
list/wake requests return HTTP 503 with code `fleet_busy`; clients can retry
|
||||||
|
later. No stale view or command is used as a fallback. This bounds the number
|
||||||
|
of rebuilds per reader, not database execution time or total concurrent requests.
|
||||||
|
|
||||||
|
Wake searches the complete cached view by device key, without list visibility
|
||||||
|
filters or the list API's 1,000-row limit. It borrows the view and clones only the
|
||||||
|
selected route. Command dispatch still checks the current session;
|
||||||
|
a cached route cannot grant access to a disconnected agent.
|
||||||
|
|
||||||
|
`GET /api/v1/control/fleet/devices/{device_key}` returns one exact row from the
|
||||||
|
complete view, or `404 fleet_device_not_found`. Details do not inherit list
|
||||||
|
filters or limits. After Remember/Attach/Merge, the UI follows the saved ID
|
||||||
|
returned by the mutation, not matching MAC/IP heuristics. It also refreshes the
|
||||||
|
full saved identity independently of list results; Forget still closes details.
|
||||||
|
|
||||||
|
## External Writes
|
||||||
|
|
||||||
|
The revision belongs to the running CC process. If another process directly
|
||||||
|
modifies its SQLite database, run the existing CC reload command (or send
|
||||||
|
SIGHUP) to invalidate the view, or restart CC. The cache does not watch files
|
||||||
|
or poll SQLite for external changes. Normal HTTP/WS store writes invalidate
|
||||||
|
automatically, and there is no time-based cache expiry.
|
||||||
|
|
||||||
|
## Shared Helper And Tracing
|
||||||
|
|
||||||
|
`runtime::view_cache::ViewCache` owns synchronization and stale-build rejection.
|
||||||
|
Fleet code supplies the key and builder. It is intentionally CC-local; alerts
|
||||||
|
keep their existing timer-driven evaluator rather than being forced into this
|
||||||
|
lazy cache.
|
||||||
|
|
||||||
|
`fleet_load` spans remain on requests. A cache miss adds `fleet_rebuild`, with
|
||||||
|
database loads and `fleet_group_devices` nested below it. On a cache hit, list
|
||||||
|
requests clone/filter the view; wake requests look up their device in the borrowed
|
||||||
|
view. Neither path repeats the database reads or grouping.
|
||||||
@@ -0,0 +1,134 @@
|
|||||||
|
# Multi-CC Identity And Configuration
|
||||||
|
|
||||||
|
Status: proposal, 2026-10-12. This is not an implemented config format.
|
||||||
|
|
||||||
|
## Goal
|
||||||
|
|
||||||
|
One agent process can serve production and test CCs simultaneously. Collect
|
||||||
|
locally once, forward to independent connections, and show human names alongside
|
||||||
|
IDs. A failed CC must not block another or kill local terminal processes.
|
||||||
|
|
||||||
|
## Identity And Names
|
||||||
|
|
||||||
|
- **CC ID:** stable identity of a logical control plane, independent of URL,
|
||||||
|
hostname, display name and WebSocket connection. Renaming or moving it must
|
||||||
|
not require agents to enroll again.
|
||||||
|
- **CC name:** configured on CC and advertised to authenticated agents. Start
|
||||||
|
with config-based rename plus reload/restart; no rename API required initially.
|
||||||
|
- **Agent instance ID/name:** local daemon identity and reported human name,
|
||||||
|
shared across its connections. Not the enrollment identity of every CC.
|
||||||
|
- **Enrollment:** each CC issues its own `agent_id` and token. Never send one
|
||||||
|
CC's credentials to another. Existing per-CC agent nicknames override the
|
||||||
|
reported name in that CC's UI; names never become routing/authentication keys.
|
||||||
|
- **Connection alias:** local config key such as `cloud` or `test`, useful for
|
||||||
|
CLI selection and pre-auth logs. It is not the remote CC ID or authoritative
|
||||||
|
remote name.
|
||||||
|
|
||||||
|
CC ID is generated once and retained in SQLite metadata when not configured;
|
||||||
|
an explicit config ID must agree with an existing stored identity, not silently
|
||||||
|
replace it. New init-config output can include a generated ID. Agent instance
|
||||||
|
ID belongs in its local config. Copying either deployment for an independent
|
||||||
|
test instance requires a distinct identity; do not mint a new ID each startup.
|
||||||
|
|
||||||
|
## Proposed Config Shape
|
||||||
|
|
||||||
|
CC keeps existing operational settings and adds:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
[identity]
|
||||||
|
id = "cc-<uuid>"
|
||||||
|
name = "Cloud"
|
||||||
|
```
|
||||||
|
|
||||||
|
Agent separates local identity/settings from per-CC enrollment:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
[identity]
|
||||||
|
id = "node-<uuid>"
|
||||||
|
name = "Router"
|
||||||
|
|
||||||
|
[local]
|
||||||
|
pid_file = "/var/run/wakey-agent.pid"
|
||||||
|
dhcp_leases_path = "/tmp/dhcp.leases"
|
||||||
|
mac_name_cache_path = "/tmp/wakey_mac_names.json"
|
||||||
|
observation_store_path = "/tmp/wakey_observations.json"
|
||||||
|
|
||||||
|
[sync]
|
||||||
|
interval = "60s"
|
||||||
|
hook_retention = "7d"
|
||||||
|
|
||||||
|
[reconnect]
|
||||||
|
base = "1s"
|
||||||
|
max = "30s"
|
||||||
|
|
||||||
|
[control_planes.cloud]
|
||||||
|
server_url = "https://wakey.example.com"
|
||||||
|
agent_id = "agent-<cloud-issued-id>"
|
||||||
|
agent_token = "<cloud-token>"
|
||||||
|
# expected_cc_id = "cc-<uuid>"
|
||||||
|
|
||||||
|
[control_planes.test]
|
||||||
|
server_url = "http://127.0.0.1:8080"
|
||||||
|
agent_id = "agent-<test-issued-id>"
|
||||||
|
agent_token = "<test-token>"
|
||||||
|
```
|
||||||
|
|
||||||
|
Existing `[terminal]` and `[telemetry]` remain daemon-wide. Reconnect defaults
|
||||||
|
apply independently to each connection; avoid per-CC copies of every setting.
|
||||||
|
The field names above are proposed, not a reason to rename unrelated CC settings.
|
||||||
|
|
||||||
|
Keep accepting the legacy single-CC root fields as one `default` connection,
|
||||||
|
with a deprecation warning. Reject mixed legacy/new connection declarations
|
||||||
|
rather than guessing. Init-config emits the new shape; `--from-config` preserves
|
||||||
|
all connections and their credentials. Enrollment targets a connection alias:
|
||||||
|
one connection can be implicit, multiple connections require explicit selection.
|
||||||
|
|
||||||
|
## Connection And Forwarding Rules
|
||||||
|
|
||||||
|
- Add negotiated, post-auth CC identity/name metadata. Current protocol has
|
||||||
|
Hello/Auth but no explicit successful-auth response; sending credentials is
|
||||||
|
not proof of authentication. Old clients must not receive unsupported enum
|
||||||
|
variants unconditionally. Keep legacy single-CC compatibility explicit.
|
||||||
|
- Forward the agent's configured name during hello; CC sends its own name/ID
|
||||||
|
after auth. Show `name (id)` or a short ID in operator/debug views. Retain the
|
||||||
|
local alias and last reported name for disconnected status, marking it offline.
|
||||||
|
- Each CC gets its own worker, retry state, bounded outbound queue and request
|
||||||
|
replies. Reject duplicate authenticated connections to the same CC ID; changed
|
||||||
|
expected IDs require an explicit operator decision, not silent rebinding.
|
||||||
|
- CC ID is identification, not a substitute for TLS/server authentication.
|
||||||
|
- Share local inventory/hook storage and periodic collection. Fan out complete
|
||||||
|
snapshots under each CC's enrollment ID; a slow connection cannot hold the
|
||||||
|
collector hostage. This does not turn snapshots into a durable event journal.
|
||||||
|
- Requests/replies and their trace contexts stay scoped to their connection.
|
||||||
|
Never broadcast command results or terminal bytes to every CC.
|
||||||
|
- Saved devices, nicknames, permissions and history remain independent on each
|
||||||
|
CC. No CC-to-CC synchronization or automatic identity promotion.
|
||||||
|
|
||||||
|
## Terminal And Telemetry Scope
|
||||||
|
|
||||||
|
The agent owns PTYs and one daemon-wide session registry. Session limits apply
|
||||||
|
globally, not once per connection. Initially, sessions belong to their creating
|
||||||
|
CC: another CC cannot attach, resume or close them. Attachment ownership is
|
||||||
|
enforced on the agent, not just inferred from one CC's registry. Disconnecting
|
||||||
|
one CC preserves its PTYs until existing TTL/explicit-close/agent-exit policy acts.
|
||||||
|
Reconnecting the same CC ID can reconcile its own sessions; removing a configured
|
||||||
|
CC must have an explicit terminal cleanup policy before implementation ships.
|
||||||
|
|
||||||
|
Use local agent instance ID and CC ID for `service.instance.id`, addressing the
|
||||||
|
current unspecified CC instance. Add configured display names as resource
|
||||||
|
metadata, not a replacement for stable IDs. Grafana presentation/label mapping
|
||||||
|
needs validation; do not promise names will automatically become `instance`.
|
||||||
|
CPU/RSS/host metrics describe one local process and must not be duplicated per CC.
|
||||||
|
Document whether snapshot counters count collections or per-CC sends before fan-out.
|
||||||
|
|
||||||
|
## Implementation Order
|
||||||
|
|
||||||
|
1. Stable CC/local-agent identities, names, telemetry metadata and config adapters.
|
||||||
|
2. Negotiated authenticated metadata exchange; keep the one-CC path working.
|
||||||
|
3. Shared local managers plus isolated connection workers and snapshot fan-out.
|
||||||
|
4. Global terminal ownership checks and multi-CC operator/debug status.
|
||||||
|
|
||||||
|
Acceptance tests must cover config/enrollment isolation, names changing without
|
||||||
|
IDs changing, one CC failing while another proceeds, shared collection without
|
||||||
|
cross-CC replies, duplicate CC identity rejection, and cross-CC terminal access
|
||||||
|
rejection. Use two test CCs and one agent before deploying this on the router.
|
||||||
@@ -0,0 +1,178 @@
|
|||||||
|
# OpenWrt UCI for Wakey
|
||||||
|
|
||||||
|
Wakey's OpenWrt init script reads `/etc/config/wakey` for the log filter and
|
||||||
|
OTLP authorization header, then passes them to the agent through procd.
|
||||||
|
The agent's application settings remain in `/etc/wakey-agent/config.toml`.
|
||||||
|
The previous `/etc/wakey-agent/environment` file is no longer read.
|
||||||
|
|
||||||
|
## The three names
|
||||||
|
|
||||||
|
```sh
|
||||||
|
uci set wakey.main.otlp_authorization='Basic BASE64_OF_USERNAME_COLON_PASSWORD'
|
||||||
|
```
|
||||||
|
|
||||||
|
| Part | Meaning |
|
||||||
|
| --- | --- |
|
||||||
|
| `wakey` | Package: `/etc/config/wakey` |
|
||||||
|
| `main` | Named section inside that package |
|
||||||
|
| `otlp_authorization` | Option inside the section |
|
||||||
|
|
||||||
|
Quotes keep the complete `Basic ...` value together despite its space.
|
||||||
|
The credential is the Base64 encoding of `username:password`.
|
||||||
|
It is not the htpasswd hash, and Base64 is not encryption.
|
||||||
|
|
||||||
|
## Create and save
|
||||||
|
|
||||||
|
Run as root on the router. `touch` does not truncate an existing file:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
touch /etc/config/wakey
|
||||||
|
chown root:root /etc/config/wakey
|
||||||
|
chmod 600 /etc/config/wakey
|
||||||
|
|
||||||
|
uci set wakey.main=service
|
||||||
|
uci set wakey.main.log_filter='info'
|
||||||
|
uci set wakey.main.otlp_authorization='Basic BASE64_OF_USERNAME_COLON_PASSWORD'
|
||||||
|
uci commit wakey
|
||||||
|
chmod 600 /etc/config/wakey
|
||||||
|
```
|
||||||
|
|
||||||
|
`wakey.main=service` creates a named section of type `service`. The type is our
|
||||||
|
chosen label, not a built-in Wakey feature. Existing options are not cleared.
|
||||||
|
`set` stages changes; `commit wakey` writes that package to disk. Committing does
|
||||||
|
not restart a service.
|
||||||
|
|
||||||
|
The resulting file looks like:
|
||||||
|
|
||||||
|
```text
|
||||||
|
config service 'main'
|
||||||
|
option log_filter 'info'
|
||||||
|
option otlp_authorization 'Basic BASE64_OF_USERNAME_COLON_PASSWORD'
|
||||||
|
```
|
||||||
|
|
||||||
|
The example above uses a placeholder. Avoid typing real credentials into a
|
||||||
|
command that your shell will save in history. In an interactive SSH terminal, you can
|
||||||
|
read the Basic header without echoing it and send it to UCI over stdin:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
(
|
||||||
|
set -e
|
||||||
|
terminal_state=$(stty -g </dev/tty)
|
||||||
|
# Restore the original terminal settings even on failure or interruption.
|
||||||
|
trap 'stty "$terminal_state" </dev/tty' 0
|
||||||
|
trap 'exit 1' HUP INT TERM
|
||||||
|
printf 'Complete Basic authorization header: ' >/dev/tty
|
||||||
|
stty -echo </dev/tty
|
||||||
|
IFS= read -r authorization </dev/tty
|
||||||
|
stty "$terminal_state" </dev/tty
|
||||||
|
printf '\n' >/dev/tty
|
||||||
|
printf "set wakey.main.otlp_authorization='%s'\ncommit wakey\n" "$authorization" | uci batch
|
||||||
|
chmod 600 /etc/config/wakey
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
|
Use a normal `Basic <base64>` header here, without quotes/newlines inside the value.
|
||||||
|
The shell's builtin `printf` avoids passing the secret in `uci`'s command arguments.
|
||||||
|
UCI is not a secret vault: committed files, staged changes, router backups, and
|
||||||
|
root access can expose these credentials. Do not paste their output into logs/chat.
|
||||||
|
|
||||||
|
## Inspect, change, delete
|
||||||
|
|
||||||
|
Inspect the non-secret option:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
uci get wakey.main.log_filter
|
||||||
|
```
|
||||||
|
|
||||||
|
`uci show wakey`, `uci export wakey`, `uci changes wakey`, and
|
||||||
|
`uci get wakey.main.otlp_authorization` **print the credential**. Use them only
|
||||||
|
when you deliberately want to see it.
|
||||||
|
|
||||||
|
Change an option:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
uci set wakey.main.log_filter='wakey_agent=debug,wakey=debug,opentelemetry=warn'
|
||||||
|
uci commit wakey
|
||||||
|
```
|
||||||
|
|
||||||
|
Delete just the authorization option:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
uci -q delete wakey.main.otlp_authorization
|
||||||
|
uci commit wakey
|
||||||
|
```
|
||||||
|
|
||||||
|
If TOML still requires `authorization_env`, removing the variable makes startup
|
||||||
|
fail. Remove that TOML field too only when the receiver no longer requires auth.
|
||||||
|
|
||||||
|
Discard all uncommitted changes to this package:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
uci revert wakey
|
||||||
|
```
|
||||||
|
|
||||||
|
## How the service reads it
|
||||||
|
|
||||||
|
`config_load` and `config_get` are shell helpers from `/lib/functions.sh`, available
|
||||||
|
in the normal `/etc/rc.common` init-script environment. They are not standalone
|
||||||
|
commands to run directly in your SSH shell.
|
||||||
|
|
||||||
|
```sh
|
||||||
|
config_load wakey
|
||||||
|
config_get authorization main otlp_authorization
|
||||||
|
```
|
||||||
|
|
||||||
|
`config_get` arguments are **destination variable, section, option, optional default**.
|
||||||
|
The line above puts the option into the shell variable `authorization`. It does
|
||||||
|
not export it to the agent automatically; procd must receive it explicitly.
|
||||||
|
|
||||||
|
The checked-in [init script](../scripts/init/openwrt/wakey) uses these helpers.
|
||||||
|
Its configuration handling is:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
start_service() {
|
||||||
|
local authorization='' log_filter='info'
|
||||||
|
if [ -f /etc/config/wakey ]; then
|
||||||
|
config_load wakey || return 1
|
||||||
|
config_get authorization main otlp_authorization
|
||||||
|
config_get log_filter main log_filter 'info'
|
||||||
|
fi
|
||||||
|
|
||||||
|
procd_open_instance
|
||||||
|
procd_set_param command "$BIN" serve --config "$CONFIG"
|
||||||
|
procd_set_param file "$CONFIG"
|
||||||
|
[ ! -f /etc/config/wakey ] || procd_append_param file /etc/config/wakey
|
||||||
|
procd_set_param env "RUST_LOG=$log_filter"
|
||||||
|
[ -z "$authorization" ] || procd_append_param env "WAKEY_OTLP_AUTHORIZATION=$authorization"
|
||||||
|
procd_set_param respawn 5 1 0
|
||||||
|
procd_set_param stdout 1
|
||||||
|
procd_set_param stderr 1
|
||||||
|
procd_close_instance
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Keep these fields in the agent's TOML:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
[telemetry]
|
||||||
|
otlp_endpoint = "http://10.5.0.1:4317"
|
||||||
|
authorization_env = "WAKEY_OTLP_AUTHORIZATION"
|
||||||
|
```
|
||||||
|
|
||||||
|
UCI stores the header; procd passes the environment variable; Rust reads it using
|
||||||
|
the name in TOML. The application does not parse UCI itself.
|
||||||
|
|
||||||
|
After installing the updated init script and committing settings:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
/etc/init.d/wakey restart
|
||||||
|
```
|
||||||
|
|
||||||
|
Use **restart**, not the existing HUP-based reload: a running process cannot receive
|
||||||
|
changed environment variables that way. UCI is only for the router; the CC's
|
||||||
|
systemd environment file remains unchanged. See [telemetry.md](telemetry.md).
|
||||||
|
|
||||||
|
## References
|
||||||
|
|
||||||
|
- [OpenWrt UCI CLI source and command usage](https://github.com/openwrt/uci/blob/master/cli.c)
|
||||||
|
- [OpenWrt config_load/config_get helpers](https://github.com/openwrt/openwrt/blob/main/package/base-files/files/lib/functions.sh)
|
||||||
@@ -0,0 +1,118 @@
|
|||||||
|
# Origin Authentication With Caddy
|
||||||
|
|
||||||
|
Cloudflare Access protects the operator surface at the edge. An origin reachable
|
||||||
|
directly must also verify Access tokens: a nonempty assertion header is not
|
||||||
|
authentication. A read-only probe through `mails.ldlda.com`, preserving Wakey's
|
||||||
|
TLS hostname and HTTP host, returned 403 without a header and 200 with a forged
|
||||||
|
header on the same origin IP. Do not retain the old header-presence matcher.
|
||||||
|
|
||||||
|
## Configure CC
|
||||||
|
|
||||||
|
Add this to the deployed CC configuration, using your protected application's
|
||||||
|
settings, not the bypass application's audience:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
bind = "127.0.0.1:23501"
|
||||||
|
|
||||||
|
[access]
|
||||||
|
issuer = "https://ldlda.cloudflareaccess.com"
|
||||||
|
audience = "0c8ca0bce40a48ddc5bc065cea8390eaddd75a84458f147bbe249fcc57c3d781"
|
||||||
|
# Optional; defaults to the issuer's /cdn-cgi/access/certs endpoint.
|
||||||
|
# jwks_url = "https://ldlda.cloudflareaccess.com/cdn-cgi/access/certs"
|
||||||
|
```
|
||||||
|
|
||||||
|
Issuer and key URLs must use HTTPS without user credentials, query or fragment.
|
||||||
|
The issuer is an origin without a trailing slash; it must exactly match the JWT
|
||||||
|
issuer. The audience is not a secret. URLs come only from trusted configuration,
|
||||||
|
never from an incoming token. `init-config --from-config` preserves `[access]`.
|
||||||
|
|
||||||
|
CC requires a loopback bind when `[access]` is configured. With no `[access]`,
|
||||||
|
the verifier returns 401; existing local development routes remain unchanged.
|
||||||
|
Changes to Access configuration require a daemon restart, not a state reload.
|
||||||
|
|
||||||
|
## Configure Caddy
|
||||||
|
|
||||||
|
Use [the Caddy template](../deploy/control-plane.Caddyfile), replacing the domain
|
||||||
|
and every upstream port with your deployment values. Its protected handler calls
|
||||||
|
CC's `GET /internal/auth/verify` through `forward_auth`, then proxies the original
|
||||||
|
request only on success. No additional service or custom Caddy module is needed.
|
||||||
|
|
||||||
|
Protect `/`, `/ui*` and `/api/v1/control/*`. This includes operator WebSocket
|
||||||
|
handshakes; established WebSocket frames are not individually authenticated and
|
||||||
|
token expiry does not automatically disconnect an existing WebSocket.
|
||||||
|
|
||||||
|
Keep `/healthz`, `/api/v1/agents/enroll`, `/api/v1/agent/ws` and
|
||||||
|
`/api/v1/agent/terminals/*` outside Access verification. They retain Wakey's own
|
||||||
|
agent/scoped-relay authentication. Cloudflare's bypass application should cover
|
||||||
|
these paths; terminal child paths must be included. If using HTTP ACME challenges,
|
||||||
|
ensure `/.well-known/acme-challenge/*` is exempt at Cloudflare; Caddy handles its
|
||||||
|
own active challenges, not CC.
|
||||||
|
|
||||||
|
Do not proxy `/internal/auth/verify` publicly and do not put `forward_auth` in
|
||||||
|
front of the verifier itself. Do not replace verification with a static header
|
||||||
|
that Caddy adds to every request. Caddy owns enforcement: direct access to the
|
||||||
|
CC port would bypass it, so loopback binding and host access controls matter.
|
||||||
|
|
||||||
|
## Verification And Cache
|
||||||
|
|
||||||
|
The verifier accepts only RS256, verifies signatures, and requires matching
|
||||||
|
issuer/audience plus expiry. A future `nbf` also rejects the token. No clock-skew
|
||||||
|
leeway is applied; keep the host clock synchronized. Duplicate assertion headers
|
||||||
|
and assertions over 16 KiB are rejected. No tokens, claims or upstream bodies
|
||||||
|
are logged.
|
||||||
|
|
||||||
|
Responses have no identity payload:
|
||||||
|
|
||||||
|
| Status | Meaning |
|
||||||
|
| --- | --- |
|
||||||
|
| 204 | Valid protected-application token |
|
||||||
|
| 401 | Missing/invalid token, unknown key after successful key retrieval, or unconfigured verifier |
|
||||||
|
| 503 | Required signing keys cannot be obtained safely |
|
||||||
|
|
||||||
|
CC reuses one HTTP client and stores parsed keys with their HTTP cache policy.
|
||||||
|
`http-cache-semantics` determines freshness from the issuer's `Cache-Control`,
|
||||||
|
`Age`, `Date` and `Expires`, rather than a built-in key TTL. For example,
|
||||||
|
`public, max-age=14400, must-revalidate, no-transform` permits four hours of
|
||||||
|
freshness minus the response's existing age; `must-revalidate` forbids stale
|
||||||
|
reuse, not every fresh reuse. No heuristic lifetime is invented when freshness
|
||||||
|
is unspecified. A `no-cache` response requires revalidation before reuse;
|
||||||
|
`no-store` retains neither keys nor policy after serving the fetching request.
|
||||||
|
|
||||||
|
Fresh matching keys verify locally. Stale policies produce conditional requests
|
||||||
|
using ETag or Last-Modified when available. A matching 304 renews the policy and
|
||||||
|
reuses parsed keys without downloading/parsing a body; mismatched or unsolicited
|
||||||
|
304 responses fail closed. An unknown key ID forces conditional revalidation
|
||||||
|
even when the stored response remains fresh, to discover rotation.
|
||||||
|
|
||||||
|
Fetches are single-flight and limited to once per minute, including failures.
|
||||||
|
Rotation immediately after a fetch may therefore take up to a minute to become
|
||||||
|
available. If the issuer requires revalidation on every use (or allows no
|
||||||
|
storage), subsequent callers inside that cooldown receive 503 rather than reuse
|
||||||
|
keys contrary to the HTTP policy. Successful 200 refresh replaces the entire
|
||||||
|
set, including removing all keys for a valid empty set. Failed refresh does not
|
||||||
|
extend freshness: matching keys may be used only while the stored policy still
|
||||||
|
permits reuse. There is no indefinitely stale fallback, including on outages.
|
||||||
|
|
||||||
|
Fetches have a five-second timeout, no redirects, a 256 KiB body limit and at
|
||||||
|
most 64 keys. These limits are internal safeguards, not new config knobs.
|
||||||
|
|
||||||
|
## Rollout
|
||||||
|
|
||||||
|
1. Install the new CC binary, add `[access]`, and restart CC. Check the loopback
|
||||||
|
verifier returns 401 for an absent or forged assertion.
|
||||||
|
2. Update Caddy from the template and validate its configuration before reload.
|
||||||
|
The repo template uses port 6767; this deployment uses 23501.
|
||||||
|
3. Confirm a real Access login can use the UI/control APIs and operator terminal
|
||||||
|
WebSocket. Confirm agent connections and enrollment still work.
|
||||||
|
4. Repeat a direct-origin request with Wakey's Host/SNI. Both absent and forged
|
||||||
|
assertions must return 401, not the UI. Verify the internal endpoint itself
|
||||||
|
returns 404 through the public virtual host.
|
||||||
|
|
||||||
|
Do not roll back to header-only acceptance on a verification failure. A UI
|
||||||
|
build or a CC restart alone does not update the deployed Caddy policy.
|
||||||
|
|
||||||
|
References: [Cloudflare token validation](https://developers.cloudflare.com/cloudflare-one/access-controls/applications/http-apps/authorization-cookie/validating-json/)
|
||||||
|
and [Caddy forward_auth](https://caddyserver.com/docs/caddyfile/directives/forward_auth).
|
||||||
|
HTTP policy is provided by [http-cache-semantics](https://docs.rs/http-cache-semantics/latest/http_cache_semantics/struct.CachePolicy.html);
|
||||||
|
only the single-resource storage, bounded fetch and anti-hammering policy are
|
||||||
|
application-owned. No general-purpose disk or Moka response cache is needed.
|
||||||
+137
@@ -0,0 +1,137 @@
|
|||||||
|
# Wakey Roadmap
|
||||||
|
|
||||||
|
Last updated: 2026-10-12. This records agreed work and follow-ups, not a release
|
||||||
|
promise. Checked items are implemented; live validation is called out separately.
|
||||||
|
|
||||||
|
## Completed Observability
|
||||||
|
|
||||||
|
- [x] Shared optional OTLP setup for traces, logs, and metrics in wakey-core.
|
||||||
|
- [x] Basic/Bearer authorization supplied through a named environment variable.
|
||||||
|
- [x] OpenWrt UCI configuration and systemd environment-file support, with setup guides.
|
||||||
|
- [x] Cross-process command trace context; verified live on a CC-to-agent wake.
|
||||||
|
- [x] Inventory source, merge, device-storage, and fleet-operation spans.
|
||||||
|
- [x] Agent connection phase spans and retry events, without entered guards across awaits.
|
||||||
|
- [x] Terminal snapshot generation, transport attachment, resume, and backpressure instrumentation.
|
||||||
|
- [x] Separate hostname-resolution and neighbor-table spans; a live slow wake spent about four seconds in resolution.
|
||||||
|
- [x] Metrics recipes for rates, duration averages/percentiles, terminal activity, and snapshots.
|
||||||
|
- [x] Authenticated live logs, metrics, and traces confirmed through Grafana's read-only API.
|
||||||
|
|
||||||
|
See [telemetry setup](telemetry.md), [metrics recipes](telemetry-recipes.md), and
|
||||||
|
[OpenWrt UCI](openwrt-uci.md). Terminal instrumentation has automated coverage;
|
||||||
|
live browser recovery and shutdown behavior are not established by those checks.
|
||||||
|
|
||||||
|
## Device Management And Presentation
|
||||||
|
|
||||||
|
- [x] Endpoint-derived presence on snapshot writes and fleet reads; endpoint-less legacy rows retain their summary. CLI ordering matches the core signal order.
|
||||||
|
|
||||||
|
- [x] Make **forget remembered device** reachable in the frontend, with confirmation. The existing backend removes the known identity and its identifiers, not agent observations. An observed device may reappear as unknown.
|
||||||
|
- [x] Add remembered-device rename support, with backend validation and frontend editing (`PATCH /api/v1/control/devices/{device_id}`). Pinned state and identifiers are preserved.
|
||||||
|
- [ ] Check and complete editing of remembered-device notes.
|
||||||
|
- [x] Replace generic unnamed-row labels with useful fallbacks. Prefer saved name, hostname, then `Device: <mac>`, then `Device: <ip>`; retain a generic label only when neither address exists. Presence stays in the badge, not the name. Inactive IPs can label IP-only rows without becoming active IPs or wake routes. Unknown-only fleet groups remain unknown.
|
||||||
|
- [ ] Review frontend visibility defaults. Keep an explicit all-rows option; do not hide evidence just to fix poor labels. The user deliberately enabled all-row querying.
|
||||||
|
- [x] Keep device details current after identity mutations change the grouping key or list filters exclude the result. Exact fleet details and full saved-identity reads follow the mutation's explicit ID; no shared-IP matching or stale-control fallback. Remember/Attach/Rename verified with mocked browser flows on desktop, Remember/Rename on mobile; backend tests cover filtering, old-key rejection and lookup beyond the list cap. Forget deliberately closes; detaching stays on the saved identity instead of guessing among split observations.
|
||||||
|
|
||||||
|
Identifier detachment already exists in the current frontend source; it is **not**
|
||||||
|
the requested whole-device removal. No automatic identity promotion or deletion
|
||||||
|
of remembered identifiers based on observation expiry.
|
||||||
|
|
||||||
|
## Telemetry Follow-Ups
|
||||||
|
|
||||||
|
- [x] Human-readable agent/CC config durations with unit-aware legacy loading, stderr deprecation warnings, and canonical init-config output. See [configuration durations](config-durations.md).
|
||||||
|
- [x] CC device retention by last snapshot inclusion (`device_retention = "60d"`, zero disables), independent of audit history and preserving saved identities.
|
||||||
|
- [x] Lazy grouped fleet-view cache with store/connection invalidation and stale-build rejection. See [fleet cache](fleet-cache.md).
|
||||||
|
|
||||||
|
- [x] Separate console, OTLP log, and trace filters using per-layer filtering. Explicit signal config overrides `RUST_LOG`, then CLI verbosity supplies defaults. Unconfigured defaults remain info-level; exporter-failure logs stay local. Live validation pending.
|
||||||
|
- [x] Upgrade the matched OpenTelemetry stack to 0.33 with exporter-managed backoff/jitter retries. Controlled collector tests cover all-signal recovery, bounded retries, and no retry on authentication rejection; live outage validation remains pending.
|
||||||
|
- [ ] Record hostname-resolution success/failure without exporting hostnames or resolved addresses. Current spans show timing but do not explicitly classify the result.
|
||||||
|
- [ ] Investigate the slow resolver with a reproducible device-side check. Timing isolated the phase, not the cause; do not treat instrumentation as a fix.
|
||||||
|
- [x] Propagate optional W3C trace context for periodic/requested WS snapshots so agent collection and CC persistence can be inspected together. Wire compatibility and context linkage tested; live Grafana inspection confirmed CC storage is a child of agent collection in the same trace. Storage can start after collection ends: this is an asynchronous handoff, not a persistence acknowledgement.
|
||||||
|
- [x] Add current-state gauges: retained device/endpoint rows per agent and unfiltered grouped fleet identities split saved/unsaved. Sample independently every 30 seconds with bounded revision checks; replace labels and suppress failed samples. Collector and state lifecycle tests included; dashboard recipes ready.
|
||||||
|
- [ ] Decide whether snapshots need persistence acknowledgements. The existing sent-device counter does not prove CC storage; any confirmed-storage metric needs an authoritative measurement point.
|
||||||
|
- [x] Break down snapshot persistence timing into pool acquisition, transaction acquisition, row replacement, and commit child spans. Shared write acquisition retains SQLx ownership and `BEGIN IMMEDIATE`; snapshot row work stays in the caller's transaction. Existing rollback and cache-refresh tests cover behavior. Live trace screenshot confirmed all phases: about 83 microseconds acquiring a connection, 154 microseconds beginning the transaction, 12.8 ms replacing rows, and 18.94 ms committing. This sample does not justify changing SQLite durability or prove a persistent bottleneck. See [telemetry span semantics](telemetry.md#signals).
|
||||||
|
|
||||||
|
The existing snapshot counter is cumulative collection volume, not unique devices
|
||||||
|
or current fleet size. Metric labels should remain bounded; no per-device labels.
|
||||||
|
|
||||||
|
## Process And Host Visibility
|
||||||
|
|
||||||
|
- [x] Add Linux process age/RSS and host uptime gauges for agent and CC through opt-in OTLP, using the existing collection interval. Core owns OS-independent instruments; wakey-linux supplies the optional procfs reader. Missing data omits observations; no added labels, config fields, health routes or polling tasks. Collector coverage verifies units and values; router/live dashboard validation pending.
|
||||||
|
- [x] Add cumulative process CPU time (counter) and current open-FD count (gauge) through the same Linux reader and OS-independent instruments. Exclude child processes, omit failed samples, and document one-core CPU-rate interpretation and sampling FD overhead. Extend collector/reader coverage rather than adding timing-based tests; live validation pending.
|
||||||
|
- [ ] Add start-time/version metadata if needed. Keep version as resource metadata rather than repeating it on every sample.
|
||||||
|
- [x] Add total/available host memory and 1/5/15-minute load gauges through the optional Linux reader, with byte conversion owned by procfs and fixed load-period labels. Omit unavailable MemAvailable instead of substituting free memory; document kernel views versus container limits. Collector/reader coverage included; live validation pending.
|
||||||
|
- [ ] Complete a small host summary with hostname, OS/kernel and CPU count. Sources must tolerate missing Linux files; not every agent host is a router.
|
||||||
|
- [ ] Display CC and agent process/host information in Fleet Health and agent details: version/name, process/host uptime, RSS, CPU time/usage, open FDs, available/total host memory and load where supported. OTLP export is not the UI's data source; the UI must work without a collector.
|
||||||
|
- [ ] Add typed agent status snapshots over the existing authenticated WebSocket, on a modest timer and when requested. Reuse Linux collection rather than introducing separate parsers; CC retains each agent's latest sample with received time and connection status. Show stale/disconnected or unavailable values explicitly, never as zero/current. Keep routine samples in memory initially, not an unbounded DB history.
|
||||||
|
- [ ] Expose the CC's own sampled status and retained agent summaries through protected control APIs. Keep `/healthz` unchanged and do not expose environment variables, credentials or command lines. CPU percentages require successive cumulative CPU/time samples; do not label raw CPU seconds as utilization. No expensive host reads on every UI request.
|
||||||
|
- [ ] Keep `/healthz` as cheap liveness; consider a separate bounded `/readyz` database-access check for CC. Collector outages and disconnected fleet agents must not make a functioning CC unready.
|
||||||
|
- [ ] Add application resource gauges where useful: DB pool usage, connected agents, pending commands, and active terminal sessions. Check existing instruments first to avoid duplicates; distinguish pool pressure from host/process pressure.
|
||||||
|
|
||||||
|
## Device Diagnostics
|
||||||
|
|
||||||
|
- [ ] Add agent-side ICMP ping through the existing command relay, with an explicit IP and selected agent/network route. Bound probe count, total duration, output and concurrent work; no shell interpolation. Account for missing tools and ICMP permissions, and support IPv4/IPv6 where available.
|
||||||
|
- [ ] Return probe replies, loss and round-trip times; offer Ping alongside Wake in device details. MAC-only devices cannot be pinged without an IP. A reply means that IP responded from that agent's network, not proof of saved identity; no reply must not automatically mark a device offline or remove an identifier.
|
||||||
|
|
||||||
|
## Optional Agent Sources And Capabilities
|
||||||
|
|
||||||
|
These are exploration items, not a commitment to a router-only model or a new
|
||||||
|
message bus. Sparse WSL inventory is legitimate; missing evidence is not failure.
|
||||||
|
|
||||||
|
- [ ] Discover and advertise usable observation sources separately from supported commands and configured permissions. Extend the existing hello/capability mechanism where appropriate; CC/UI must tolerate missing sources and older agents. Do not infer router status from Linux alone.
|
||||||
|
- [ ] Investigate read-only Wi-Fi station data on the actual router: signal in **dBm**, association/interface, rates and traffic counters where available. Evaluate driver-supported `iw`/netlink or local `ubus`/hostapd interfaces before choosing a dependency. Keep measurements scoped to agent/interface/station with source and capture time; they are not permanent identity or globally valid presence. UI may show a signal badge only when supported and sufficiently fresh.
|
||||||
|
- [ ] Investigate interface lifecycle hooks for WAN/link changes, keeping scripts minimal and logic in the binary. Interface up/down does not prove Internet reachability; any active connectivity probe needs an explicit target and bounded policy. Existing DHCP/neighbor hooks remain independent evidence, not a way to enrich or overwrite other sources.
|
||||||
|
- [ ] Keep source support, operator authorization and local agent policy distinct. Capability advertisement is not permission: CC checks authorization and the agent enforces configured limits. New diagnostic APIs should be typed, protected operations, not arbitrary shell, UCI or bus calls exposed to the browser.
|
||||||
|
|
||||||
|
OpenWrt [hotplug](https://openwrt.org/docs/guide-user/base-system/hotplug) provides
|
||||||
|
event hooks; [ubus](https://openwrt.org/docs/techref/ubus) is local IPC/RPC, and
|
||||||
|
[UCI](https://openwrt.org/docs/guide-developer/ubus/uci) manages configuration.
|
||||||
|
None is a Kafka-style retained observation history. Verify what the vendor
|
||||||
|
firmware actually exposes before making any of them an agent requirement.
|
||||||
|
|
||||||
|
## Origin Authentication
|
||||||
|
|
||||||
|
- [x] Replace the nonempty assertion matcher in the Caddy template with `forward_auth`. The initial `lb.ldlda.com` probe reached Cloudflare; the later `mails.ldlda.com` probe reached an origin and confirmed 403 without an assertion versus 200 with a forged assertion on the same IP. This is a reproduced bypass, not just a theoretical concern; production rollout remains required.
|
||||||
|
- [x] Add CC's `/internal/auth/verify` endpoint with configured issuer/audience/key URL, RS256 signature and required claim validation, bounded cached key retrieval, single-flight refresh and outage cooldown. Return 204/401/503 without exposing tokens or identity data. Configured verification requires loopback binding, and `init-config --from-config` preserves it. See [origin authentication](origin-auth.md).
|
||||||
|
- [x] Cover forged/expired/wrong-audience/missing-claim tokens, key rotation/removal (including an empty set), duplicate assertions, unavailable/malformed/oversized key responses, cooldown and concurrent fetches. Router coverage verifies the oracle is separate from public/control handlers; actual Caddy enforcement is deployment-owned and still requires validation.
|
||||||
|
- [x] Use `http-cache-semantics` for issuer-supplied HTTP freshness instead of a hardcoded signing-key TTL. Preserve bounded streaming reads, force revalidation on unknown key IDs, and handle ETag/Last-Modified 304 responses without reparsing a body. Tests cover fresh reuse, Age, no-cache/no-store, unspecified freshness, conditional rotation and invalid 304 rejection; no sleeps required.
|
||||||
|
- [ ] Deploy the new CC verifier and Caddy policy for root/UI/control HTTP and operator WS handshakes. Keep public agent paths outside Access checks and the verifier unexposed publicly. Validate Caddy configuration, real Access login, agent connectivity, protected WebSocket rejection, and forged-header direct-origin rejection. Authentication failures must never fall back to header-only acceptance.
|
||||||
|
|
||||||
|
## Multi-CC Planning
|
||||||
|
|
||||||
|
- [ ] Consolidate stable CC/local-agent IDs, forwarded display names, per-CC enrollments, and agent config restructuring. See [multi-CC proposal](multi-cc-plan.md). Identity/name exchange and terminal ownership must be defined before adding connection workers; no CC-to-CC state synchronization is planned.
|
||||||
|
|
||||||
|
## Validation And Reliability
|
||||||
|
|
||||||
|
- [x] Bound fleet-cache rebuilds to three attempts under sustained mutations; return retryable HTTP 503 `fleet_busy` rather than silently serving stale wake routes. Deterministic limit/recovery tests included.
|
||||||
|
- [x] Separate fleet wake selection from dispatch; wake lookup uses the complete cached view rather than the presentation list's 1,000-row cap. Automated coverage checks explicit-route rejection and lookup beyond the cap; live checks confirm missing-device and invalid-route errors without dispatching wakes.
|
||||||
|
- [x] Share device read assembly within a single SQLite read transaction. Query-plan inspection on the test database with Python SQLite confirmed all six per-agent reads use primary-key searches; full-fleet reads scan intentionally, with covering indexes for MAC/IP projections and no temporary sort. No additional indexes justified by this check.
|
||||||
|
- [x] Separate fleet refresh selection, relay, decoding/storage, and aggregation. Preserve concurrency of eight and per-agent failures; tests cover disconnected agents, sorted partial results, invalid payload preservation, and valid empty snapshot replacement.
|
||||||
|
- [x] Add a refresh integration regression: a simulated agent answers through the production command relay; its inventory reaches SQLite, invalidates a previously populated fleet cache, and appears in the next fleet read. A disconnected agent does not cancel success; duplicate selection sends only one command. This covers the CC application path, not a real agent process or WebSocket transport.
|
||||||
|
- [ ] Reproduce/check Ctrl-C cleanup on the test device: process groups, PTYs, relay tasks, and telemetry shutdown. Leftovers were suspected, not confirmed.
|
||||||
|
- [ ] Exercise collector unavailability and recovery with running agent/CC instances. Verify normal operations continue and telemetry queues remain bounded; outage losses are expected under the current best-effort policy.
|
||||||
|
- [ ] Validate terminal recovery instrumentation live: generation timing/size, transport attach/resume, and backpressure episodes ending in drain or disconnect.
|
||||||
|
- [ ] Validate the revised UCI credential prompt and service environment on the live router.
|
||||||
|
|
||||||
|
## Watch, Not Yet Fix
|
||||||
|
|
||||||
|
- Fleet listing time is mostly device-store reads in observed traces. Measure
|
||||||
|
sustained latency and row counts before changing queries or indexes.
|
||||||
|
- Snapshot write durations vary. Existing store spans include pool/lock waits,
|
||||||
|
transaction work, and result assembly where applicable; they are not pure SQL timings.
|
||||||
|
|
||||||
|
## Deferred
|
||||||
|
|
||||||
|
- Pinned-state editing: currently not needed for the small remembered fleet.
|
||||||
|
- Browser-render acknowledgement: attach acceptance and snapshot generation do
|
||||||
|
not prove xterm finished restoring. This would require additional protocol work.
|
||||||
|
- Retaining terminal images across recovery: text-screen snapshots do not provide image replay.
|
||||||
|
- CPU/heap profiling: use targeted profiling when a measured issue calls for it;
|
||||||
|
OTLP traces are not CPU profiles or flame graphs.
|
||||||
|
- Grafana dashboard construction is the user's exercise; keep improving the recipe book rather than taking over dashboard editing.
|
||||||
|
|
||||||
|
## Release Notes
|
||||||
|
|
||||||
|
- No version bump is made by this roadmap. Additive optional config fields alone
|
||||||
|
do not require 0.4.0; reconsider if compatibility or defaults change materially.
|
||||||
|
- Recheck live telemetry after deployment. A Grafana volume reset can remove
|
||||||
|
history and invalidate service-account tokens or data-source UIDs.
|
||||||
|
- Never commit Grafana tokens, UCI credentials, or router backups.
|
||||||
@@ -0,0 +1,294 @@
|
|||||||
|
# Wakey Metrics Recipes
|
||||||
|
|
||||||
|
Use Grafana Explore with your Prometheus-compatible data source and the Code
|
||||||
|
query editor. These recipes assume standard OTLP-to-Prometheus name conversion.
|
||||||
|
Discover your actual series first:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
count by (__name__, job) ({__name__=~"wakey[._].*"})
|
||||||
|
```
|
||||||
|
|
||||||
|
With Alloy's Prometheus exporter, service names normally become `job`, and agent
|
||||||
|
instance IDs become `instance`. Inspect the returned labels before adding filters.
|
||||||
|
For example, append `{job="wakey-agent"}` to a metric selector only if that is
|
||||||
|
the job value you actually see. Metrics export every 30 seconds; use a time range
|
||||||
|
that includes recent activity and allow at least two exports for rate queries.
|
||||||
|
|
||||||
|
## Metric Types
|
||||||
|
|
||||||
|
| Type | What it records | How to read it |
|
||||||
|
| --- | --- | --- |
|
||||||
|
| Counter | Cumulative events or bytes since process startup | `rate` for per-second throughput; `increase` for a window's estimated total |
|
||||||
|
| UpDownCounter | Adds and subtracts from a current quantity | Read the current value directly; do not use `rate` for active sessions |
|
||||||
|
| Gauge | Measures a current value, such as stored device rows | Read directly; do not use `rate` or `increase` |
|
||||||
|
| Histogram | Distribution of individual measurements | `_sum / _count` for an average; `_bucket` for estimated percentiles |
|
||||||
|
|
||||||
|
Counter resets on restart are handled by `rate` and `increase`. Apply those
|
||||||
|
functions **before** summing across instances so individual resets remain visible.
|
||||||
|
|
||||||
|
## Process And Host
|
||||||
|
|
||||||
|
Discover the new gauges first (these are Linux-only):
|
||||||
|
|
||||||
|
```promql
|
||||||
|
count by (__name__, job, instance) ({__name__=~"wakey_(process|host)_.*"})
|
||||||
|
```
|
||||||
|
|
||||||
|
Process age and host uptime, in seconds:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
wakey_process_uptime_seconds
|
||||||
|
```
|
||||||
|
|
||||||
|
```promql
|
||||||
|
wakey_host_uptime_seconds
|
||||||
|
```
|
||||||
|
|
||||||
|
Set Grafana's unit to a duration. Compare the two: process age resetting while
|
||||||
|
host uptime continues suggests a daemon restart; both resetting suggests a host
|
||||||
|
restart or a changed namespace/view. Missing samples are not a zero uptime or
|
||||||
|
proof of a restart. These are gauges: do not use `rate` or `increase` on them.
|
||||||
|
|
||||||
|
Resident process memory, in bytes:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
wakey_process_memory_resident_bytes
|
||||||
|
```
|
||||||
|
|
||||||
|
Set the unit to **bytes (IEC)**. Keep `job` and `instance` in the legend to compare
|
||||||
|
agent and CC separately. RSS includes shared/file-backed pages and is not heap
|
||||||
|
allocation alone. A missing series can mean disabled telemetry, an unsupported
|
||||||
|
platform, unavailable procfs, or export trouble. No new health endpoint or host
|
||||||
|
inventory is provided here; do not sum host uptime from multiple daemons.
|
||||||
|
|
||||||
|
CPU consumption as a percentage of **one CPU core**:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
100 * rate(wakey_process_cpu_time_seconds_total[5m])
|
||||||
|
```
|
||||||
|
|
||||||
|
Set the unit to **percent (0-100)**. A multithreaded process can exceed 100%;
|
||||||
|
this is not normalized by the host's CPU count. The counter includes daemon
|
||||||
|
user/system CPU, not its child shells or commands. Allow at least two exports;
|
||||||
|
`rate` handles counter resets on process restart.
|
||||||
|
|
||||||
|
Current open descriptors:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
wakey_process_open_fds
|
||||||
|
```
|
||||||
|
|
||||||
|
Read directly as a count, not a rate. The sampling read can contribute temporary
|
||||||
|
descriptors. Persistent growth can warrant investigation, but sockets, terminal
|
||||||
|
sessions and pool connections legitimately change the count too.
|
||||||
|
|
||||||
|
Host usable/available RAM, in bytes:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
wakey_host_memory_total_bytes
|
||||||
|
```
|
||||||
|
|
||||||
|
```promql
|
||||||
|
wakey_host_memory_available_bytes
|
||||||
|
```
|
||||||
|
|
||||||
|
Set units to **bytes (IEC)**. Estimated percentage not available for new work:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
100 * (1 - wakey_host_memory_available_bytes / wakey_host_memory_total_bytes)
|
||||||
|
```
|
||||||
|
|
||||||
|
Available includes reclaimable memory and is not merely free RAM. Missing
|
||||||
|
`MemAvailable` leaves a gap rather than substituting `MemFree`. Do not sum host
|
||||||
|
RAM from multiple daemon instances on the same host, or interpret these kernel
|
||||||
|
values as a container's cgroup limit.
|
||||||
|
|
||||||
|
Load averages by period:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
wakey_host_load
|
||||||
|
```
|
||||||
|
|
||||||
|
Use `{{job}} / {{instance}} / {{period}}` as the legend. Fixed periods are `1m`,
|
||||||
|
`5m`, and `15m`. Read directly as a load count, not a percentage or a rate:
|
||||||
|
Linux includes runnable and uninterruptible tasks; a high load can include I/O
|
||||||
|
waiting rather than CPU work. These values are not normalized by CPU count.
|
||||||
|
|
||||||
|
## Operations
|
||||||
|
|
||||||
|
Completed attempts per second, including errors and timeouts:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, operation, outcome) (rate(wakey_operations_total[5m]))
|
||||||
|
```
|
||||||
|
|
||||||
|
Estimated attempts over the last hour:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, operation, outcome) (increase(wakey_operations_total[1h]))
|
||||||
|
```
|
||||||
|
|
||||||
|
Average duration over the last five minutes:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, operation) (rate(wakey_operation_duration_seconds_sum[5m]))
|
||||||
|
/
|
||||||
|
sum by (job, operation) (rate(wakey_operation_duration_seconds_count[5m]))
|
||||||
|
```
|
||||||
|
|
||||||
|
Set Grafana's unit to **seconds**. This is a weighted average, not an average of
|
||||||
|
instance averages. Both successful and unsuccessful attempts are included;
|
||||||
|
add `outcome="ok"` to both selectors for successful attempts only.
|
||||||
|
|
||||||
|
Estimated p95 duration:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
histogram_quantile(
|
||||||
|
0.95,
|
||||||
|
sum by (le, job, operation) (
|
||||||
|
rate(wakey_operation_duration_seconds_bucket[5m])
|
||||||
|
)
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
|
This needs classic histogram bucket series. Keep `le` in the grouping: it is
|
||||||
|
the bucket boundary, in seconds. p95 is an estimate from those boundaries,
|
||||||
|
not an exact sort of recorded durations. Few operations make percentiles noisy.
|
||||||
|
|
||||||
|
CC relay duration includes waiting for the agent; agent execution duration
|
||||||
|
measures its dispatch separately. Do not sum both and call that user latency.
|
||||||
|
|
||||||
|
## Terminals
|
||||||
|
|
||||||
|
Active PTY workers by agent, including detached sessions:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, instance) (wakey_terminal_active)
|
||||||
|
```
|
||||||
|
|
||||||
|
Read this directly as a count. It is not a count of browser tabs or attachments.
|
||||||
|
|
||||||
|
PTY output bytes per second:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, instance) (rate(wakey_terminal_output_bytes_total[5m]))
|
||||||
|
```
|
||||||
|
|
||||||
|
Input bytes per second:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, instance) (rate(wakey_terminal_input_bytes_total[5m]))
|
||||||
|
```
|
||||||
|
|
||||||
|
Set the unit to **bytes/sec**. Output is measured at the PTY read, even while
|
||||||
|
detached; it is not WebSocket wire traffic or confirmed browser delivery.
|
||||||
|
|
||||||
|
Relay queue-full encounters over the last fifteen minutes:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, instance) (increase(wakey_terminal_backpressure_total[15m]))
|
||||||
|
```
|
||||||
|
|
||||||
|
This counts encounters, not their duration and not dropped chunks. A burst of
|
||||||
|
image output can raise it while output still arrives intact through backpressure.
|
||||||
|
|
||||||
|
Average duration of completed backpressure episodes, in seconds:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, instance, outcome) (
|
||||||
|
rate(wakey_operation_duration_seconds_sum{operation="terminal_backpressure_wait"}[5m])
|
||||||
|
)
|
||||||
|
/
|
||||||
|
sum by (job, instance, outcome) (
|
||||||
|
rate(wakey_operation_duration_seconds_count{operation="terminal_backpressure_wait"}[5m])
|
||||||
|
)
|
||||||
|
```
|
||||||
|
|
||||||
|
An episode starts when the output channel is full and ends when the pending
|
||||||
|
backlog drains or the relay disconnects. It is not per-frame network latency;
|
||||||
|
unfinished episodes do not appear until they end. Keep the outcome grouping so
|
||||||
|
disconnected episodes are not confused with successfully drained ones.
|
||||||
|
|
||||||
|
## Snapshots
|
||||||
|
|
||||||
|
Device entries in successfully sent snapshots over the last hour:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, instance) (increase(wakey_snapshot_devices_total[1h]))
|
||||||
|
```
|
||||||
|
|
||||||
|
Repeated snapshots count the same devices again. This is collection volume,
|
||||||
|
**not unique devices, current fleet size, or confirmed CC persistence**.
|
||||||
|
|
||||||
|
## Current Devices And Endpoints
|
||||||
|
|
||||||
|
These gauges are emitted by CC. Counts are sampled at startup and every 30 seconds,
|
||||||
|
independently of UI requests, and exported on the normal metrics schedule. Allow
|
||||||
|
up to two intervals for a change to appear. The snapshot counter above cannot
|
||||||
|
substitute for these gauges.
|
||||||
|
|
||||||
|
A **device** is a grouped device identity. An **endpoint** is one observed address
|
||||||
|
entry with an IP, optional MAC/interface, source, and presence. One PC can have
|
||||||
|
multiple IPv4/IPv6 endpoints, and different sources or agents can observe the
|
||||||
|
same address. An endpoint here is not an HTTP API route.
|
||||||
|
|
||||||
|
| Panel | Scope | Suggested visualization |
|
||||||
|
| --- | --- | --- |
|
||||||
|
| Stored devices by agent | Device rows currently retained in CC for each agent; not globally unique identities | Bar gauge, one value per agent |
|
||||||
|
| Stored endpoints by agent | Endpoint rows currently retained in CC for each agent, including inactive entries; not unique IPs | Bar gauge, one value per agent |
|
||||||
|
| Fleet devices | Grouped fleet identities, split into saved and unsaved | Stat, one value per category |
|
||||||
|
|
||||||
|
Stored device rows by agent:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, agent_id) (wakey_stored_devices{job="wakey-control-plane"})
|
||||||
|
```
|
||||||
|
|
||||||
|
Stored endpoint rows by agent:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, agent_id) (wakey_stored_endpoints{job="wakey-control-plane"})
|
||||||
|
```
|
||||||
|
|
||||||
|
Grouped fleet devices by manual identity status:
|
||||||
|
|
||||||
|
```promql
|
||||||
|
sum by (job, identity) (wakey_fleet_devices{job="wakey-control-plane"})
|
||||||
|
```
|
||||||
|
|
||||||
|
Adjust the `job` selector to your actual service label. These queries assume one
|
||||||
|
CC owns the fleet; do not sum replicas of the same database as distinct devices.
|
||||||
|
Enrolled agents with no stored rows report zero. After revocation, an agent can
|
||||||
|
remain represented while its device rows are retained. When neither enrollment
|
||||||
|
nor stored rows exist, the sampler stops reporting that agent label.
|
||||||
|
|
||||||
|
Use Grafana's **short** unit with zero decimal places. Read the gauges directly;
|
||||||
|
do not apply `rate` or `increase`. Counts must reflect unfiltered CC state, not
|
||||||
|
the last UI search or visibility filter. An agent going offline does not by
|
||||||
|
itself delete its stored rows, so these counts are not online-device counts.
|
||||||
|
|
||||||
|
Do not sum per-agent device counts and label that value "unique fleet devices":
|
||||||
|
the same device can be observed by more than one agent. Likewise, an endpoint
|
||||||
|
count is not a count of distinct IP addresses. Saved/unsaved describes manual
|
||||||
|
identity management, not online/offline/unknown presence.
|
||||||
|
|
||||||
|
Failed refreshes suppress observations until sampling recovers; they do not report
|
||||||
|
zero or keep exporting an old sample. Prometheus may still show its last value
|
||||||
|
until that series becomes stale. Export callbacks only read the in-memory sample;
|
||||||
|
they never wait on database work. Sampling is disabled when OTLP is disabled.
|
||||||
|
|
||||||
|
## Empty or Strange Results
|
||||||
|
|
||||||
|
- No series: generate the relevant activity, wait for export, and check the metrics pipeline.
|
||||||
|
- Different names: use the discovery query; collector translation can change suffixes and labels.
|
||||||
|
- `NaN` average: no operations occurred in the window, so the denominator is zero. Show a gap, not a fake zero latency.
|
||||||
|
- Empty `rate`: it needs multiple samples. Try a longer window and check the time range.
|
||||||
|
- Fractional `increase`: Prometheus extrapolates to the window boundaries; event counts can therefore appear fractional.
|
||||||
|
- Stale active count: the last exported value is not an immediate liveness check after a crash or network loss.
|
||||||
|
- No `instance` label: CC does not currently set a service instance ID; agent telemetry does.
|
||||||
|
|
||||||
|
## References
|
||||||
|
|
||||||
|
- [Wakey telemetry setup and limitations](telemetry.md)
|
||||||
|
- [Prometheus rate, increase, and histogram_quantile](https://prometheus.io/docs/prometheus/latest/querying/functions/)
|
||||||
|
- [Alloy OTLP-to-Prometheus conversion](https://grafana.com/docs/alloy/latest/reference/components/otelcol/otelcol.exporter.prometheus/)
|
||||||
@@ -0,0 +1,251 @@
|
|||||||
|
# Telemetry
|
||||||
|
|
||||||
|
CC and the agent daemon share the optional `wakey-core/telemetry` module.
|
||||||
|
Local logs remain enabled. Configure an OTLP **gRPC** receiver to additionally
|
||||||
|
export traces, structured logs, and metrics; HTTP/4318 is not supported in this slice.
|
||||||
|
|
||||||
|
CC configuration:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
[telemetry]
|
||||||
|
otlp_endpoint = "http://127.0.0.1:4317"
|
||||||
|
authorization_env = "WAKEY_OTLP_AUTHORIZATION"
|
||||||
|
service_name = "wakey-control-plane"
|
||||||
|
json_logs = true
|
||||||
|
```
|
||||||
|
|
||||||
|
Agent configuration (replace the example with the collector's WireGuard address):
|
||||||
|
|
||||||
|
```toml
|
||||||
|
[telemetry]
|
||||||
|
otlp_endpoint = "http://10.5.0.1:4317"
|
||||||
|
authorization_env = "WAKEY_OTLP_AUTHORIZATION"
|
||||||
|
service_name = "wakey-agent"
|
||||||
|
json_logs = false
|
||||||
|
```
|
||||||
|
|
||||||
|
Omit `otlp_endpoint` to disable exporting. Omit `authorization_env` for a receiver
|
||||||
|
that requires no authentication. Existing configs need no changes.
|
||||||
|
An empty agent service name uses `wakey-agent`; its enrolled agent ID becomes
|
||||||
|
the resource's `service.instance.id`, not a metric label. Telemetry config changes
|
||||||
|
require a daemon restart, not SIGHUP. Short-lived enroll/reload/observe commands
|
||||||
|
only log locally and do not initialize exporters.
|
||||||
|
|
||||||
|
## Receiver and filtering
|
||||||
|
|
||||||
|
The collector must enable OTLP receivers in **traces, logs, and metrics** pipelines.
|
||||||
|
Loopback is sufficient for CC on the same host/network namespace. The router
|
||||||
|
needs a collector listener reachable over WireGuard; do not expose an unauthenticated
|
||||||
|
receiver on the public Internet. This initial setup assumes plaintext inside that
|
||||||
|
tunnel. `json_logs` changes local output,
|
||||||
|
not whether OTLP logs are exported.
|
||||||
|
|
||||||
|
## Credentials and services
|
||||||
|
|
||||||
|
The OpenWrt agent uses `/etc/config/wakey`; see the
|
||||||
|
[UCI setup guide](openwrt-uci.md) for commands and permissions.
|
||||||
|
The CC uses a systemd environment file as described below.
|
||||||
|
|
||||||
|
`authorization_env` names an environment variable containing the complete header:
|
||||||
|
`Basic <base64(username:password)>` or `Bearer <token>`. With exporting enabled,
|
||||||
|
missing/empty/invalid values fail startup; credentials are never written to TOML
|
||||||
|
or included in error messages. Metadata is marked sensitive for Debug output.
|
||||||
|
Base64 is not encryption: use the WireGuard/SSH tunnel or another protected transport.
|
||||||
|
|
||||||
|
The systemd CC reads the optional `/etc/wakey-control-plane/environment` file.
|
||||||
|
|
||||||
|
Use this portable literal format, **without quotes or `export`**:
|
||||||
|
|
||||||
|
```text
|
||||||
|
# Replace this placeholder with the actual encoded credential locally.
|
||||||
|
WAKEY_OTLP_AUTHORIZATION=Basic BASE64_OF_USERNAME_COLON_PASSWORD
|
||||||
|
RUST_LOG=info
|
||||||
|
```
|
||||||
|
|
||||||
|
Protect the file with owner `root:root` and mode `0600`. systemd's manager reads it
|
||||||
|
before switching to the service user. Use the simple format above, with LF endings
|
||||||
|
and no inline comments or multiline values.
|
||||||
|
An absent file is allowed; configuring auth without providing its variable is not.
|
||||||
|
|
||||||
|
After changing the file, **restart** the service. On systemd, run `daemon-reload`
|
||||||
|
once after installing the updated unit. OpenWrt's HUP reload does not reload the
|
||||||
|
process environment. For a foreground WSL test, export the variable in that shell
|
||||||
|
and use `http://127.0.0.1:4317` with the SSH tunnel; do not put credentials in CLI args.
|
||||||
|
|
||||||
|
Console logs, OTLP logs, and traces can have independent filters:
|
||||||
|
|
||||||
|
```toml
|
||||||
|
[telemetry]
|
||||||
|
console_filter = "info"
|
||||||
|
otlp_log_filter = "debug"
|
||||||
|
trace_filter = "info"
|
||||||
|
```
|
||||||
|
|
||||||
|
For each signal, an explicit filter wins over `RUST_LOG`, which wins over CLI
|
||||||
|
verbosity (`-v`). With no filters, environment override, or verbosity flags, the
|
||||||
|
application defaults remain info-level with SDK warnings. Metrics are unaffected.
|
||||||
|
Explicit filters must be valid and nonempty; changing them requires a restart.
|
||||||
|
Without explicit filters, `RUST_LOG=warn` suppresses informational spans and logs.
|
||||||
|
Use `wakey_agent=info,wakey=info,wakey_core=info,opentelemetry=warn` for the agent and
|
||||||
|
`wakey_control_plane=info,wakey_core=info,opentelemetry=warn` for CC. Exporter warnings
|
||||||
|
remain local to prevent export-failure feedback loops. Both OTLP logs and traces
|
||||||
|
exclude OpenTelemetry SDK targets and the `h2`, `hyper`, `hyper_util`, `tonic`
|
||||||
|
and `tower` crate targets, even under broad debug filters. These transport
|
||||||
|
crates also serve application traffic: their diagnostics stay console-only,
|
||||||
|
while Wakey's command/storage/relay spans still export. Console filters remain
|
||||||
|
independent. Increasing verbosity exports application debug events too; inspect
|
||||||
|
them before enabling it broadly.
|
||||||
|
|
||||||
|
## Signals
|
||||||
|
|
||||||
|
- `relay_command` on CC propagates W3C context in an optional command field.
|
||||||
|
`execute_command` on the agent adopts it; older messages/agents remain compatible.
|
||||||
|
- `collect_device_snapshot` and `store_device_snapshot` measure collection and
|
||||||
|
persistence as linked spans across the agent WebSocket. The snapshot carries
|
||||||
|
optional W3C context from collection; CC adopts it after authentication checks.
|
||||||
|
Periodic and requested syncs use the same path. Old messages and empty/invalid
|
||||||
|
context remain valid but do not create a cross-process link. Both processes
|
||||||
|
need tracing enabled and suitable filters/sampling to see both sides.
|
||||||
|
- Terminal create/attach/close/resume spans describe lifecycle operations, not
|
||||||
|
output chunks. Trace-linked OTLP logs use the SDK's tracing-context bridge.
|
||||||
|
- Inventory source spans separate neighbor, DHCP lease, and hook-store reads
|
||||||
|
from merging. `db_*` spans cover snapshot replacement, fleet reads, and manual
|
||||||
|
identity changes, including transaction waits and result assembly where applicable.
|
||||||
|
Snapshot replacement retains `db_replace_agent_snapshot` as its total duration,
|
||||||
|
with four sequential child spans:
|
||||||
|
- `db_pool_acquire`: obtaining a pooled connection, including any connection
|
||||||
|
setup/checks; it is not exclusively queue wait.
|
||||||
|
- `db_write_transaction_begin`: executing `BEGIN IMMEDIATE`, including SQLite
|
||||||
|
writer-lock wait and worker scheduling, not a direct lock-wait measurement.
|
||||||
|
- `db_snapshot_write_rows`: existing-key lookup, normalization, parent/child
|
||||||
|
writes, and deletion of devices missing from the complete snapshot.
|
||||||
|
- `db_snapshot_commit`: transaction commit, including its SQLite/worker work;
|
||||||
|
this does not isolate filesystem flush time.
|
||||||
|
Pool and transaction-begin spans come from the shared write helper, so they
|
||||||
|
also appear under other write operations. No per-row spans or SQL values are
|
||||||
|
recorded. A failed write rolls back through SQLx and has no commit span;
|
||||||
|
fleet cache invalidation still happens only after a successful commit.
|
||||||
|
- Fleet spans cover list/load/grouping, refresh fan-out with per-agent outcomes,
|
||||||
|
and wake dispatch. Counts distinguish grouped, filtered, and returned rows;
|
||||||
|
wake events identify the chosen agent and whether the route was explicit,
|
||||||
|
without recording device identifiers or addresses.
|
||||||
|
- Connectivity uses short `agent_dns`, `agent_connect`, `agent_handshake_send`,
|
||||||
|
and CC `agent_auth_verify` spans, rather than connection-long spans. Connect
|
||||||
|
attempts also contribute to operation metrics. Lifecycle events report retries
|
||||||
|
with backoff and CC disconnect duration. Sending credentials is not an auth ACK;
|
||||||
|
CC's authentication event is the authoritative result. DNS diagnostics are a
|
||||||
|
separate lookup, not a measurement of the resolver inside WebSocket connect.
|
||||||
|
- Terminal recovery uses `terminal_snapshot_generate` (encoded byte count),
|
||||||
|
transport-attachment spans, and `terminal_snapshot_request`. These measure
|
||||||
|
local work, not browser restoration or end-to-end image replay. Resume operation
|
||||||
|
success means credentials were queued for the worker. `terminal_backpressure_wait`
|
||||||
|
operation durations cover one full-queue backlog episode until it drains (`ok`)
|
||||||
|
or disconnects (`disconnected`); cancellation can retain the default `error` outcome.
|
||||||
|
|
||||||
|
Instrument operation boundaries rather than individual rows or terminal chunks.
|
||||||
|
Use `skip_all`, explicitly allow count fields, and instrument async futures rather
|
||||||
|
than holding entered span guards across awaits. Do not record query arguments,
|
||||||
|
SQL parameters, device payloads, or credentials. Span names are fixed across requests.
|
||||||
|
|
||||||
|
Metrics are exported every 30 seconds and at orderly shutdown:
|
||||||
|
|
||||||
|
The table uses OpenTelemetry instrument names. Prometheus conversion typically
|
||||||
|
replaces dots with underscores and adds type/unit suffixes; histograms produce
|
||||||
|
multiple series. See the [metrics recipe book](telemetry-recipes.md) for discovery
|
||||||
|
queries, rates, averages, percentiles, and Grafana units.
|
||||||
|
|
||||||
|
| Metric | Meaning |
|
||||||
|
| --- | --- |
|
||||||
|
| `wakey.operations` | Completed command/snapshot attempts, labeled by operation and outcome |
|
||||||
|
| `wakey.operation.duration` | Corresponding duration histogram in seconds |
|
||||||
|
| `wakey.snapshot.devices` | Cumulative devices in snapshots successfully sent by the agent |
|
||||||
|
| `wakey.stored.devices` | CC gauge of retained device rows, labeled by `agent_id` |
|
||||||
|
| `wakey.stored.endpoints` | CC gauge of retained endpoint rows, including inactive entries, labeled by `agent_id` |
|
||||||
|
| `wakey.fleet.devices` | CC gauge of unfiltered grouped identities, labeled by `identity` (`saved` or `unsaved`) |
|
||||||
|
| `wakey.process.uptime` | Linux gauge: seconds since this process's kernel-reported start, including suspend |
|
||||||
|
| `wakey.process.memory.resident` | Linux gauge: approximate resident memory in bytes, from VmRSS |
|
||||||
|
| `wakey.process.cpu.time` | Linux counter: cumulative user + system CPU seconds for all process threads, excluding children |
|
||||||
|
| `wakey.process.open_fds` | Linux gauge: current count of open process file descriptors |
|
||||||
|
| `wakey.host.uptime` | Linux gauge: uptime in seconds reported by the process's procfs view, including suspend |
|
||||||
|
| `wakey.host.memory.total` | Linux gauge: total usable RAM in bytes from the visible meminfo |
|
||||||
|
| `wakey.host.memory.available` | Linux gauge: estimated available RAM in bytes; omitted if MemAvailable is unavailable |
|
||||||
|
| `wakey.host.load` | Linux gauge: load average, with fixed `period` labels `1m`, `5m`, `15m` |
|
||||||
|
| `wakey.terminal.active` | Agent PTY workers currently alive |
|
||||||
|
| `wakey.terminal.input.bytes` | Input bytes received by the terminal relay |
|
||||||
|
| `wakey.terminal.output.bytes` | Output bytes read from the PTY, including while detached |
|
||||||
|
| `wakey.terminal.backpressure` | Relay queue-full encounters, not dropped output |
|
||||||
|
|
||||||
|
Metric labels do not include request/session IDs, MACs, or IPs. New spans use
|
||||||
|
`skip_all` and explicitly select safe fields; terminal bytes and credentials are
|
||||||
|
not attached. Profiles, host CPU-utilization metrics, connected-agent gauges, and export-loss
|
||||||
|
dashboards are not part of this initial slice.
|
||||||
|
|
||||||
|
Process and host gauges are registered automatically for Linux agent/CC daemons
|
||||||
|
when OTLP is enabled. `wakey-core` owns the OS-independent instruments and
|
||||||
|
`ProcessMetricsSource` interface; `wakey-linux` supplies the procfs reader via
|
||||||
|
its optional `telemetry` feature. Each daemon attaches that reader to its
|
||||||
|
telemetry guard, so core has no Linux dependency or filesystem reads.
|
||||||
|
Their small procfs reads run during the existing 30-second
|
||||||
|
metric collection and final shutdown collection; no independent poller or task
|
||||||
|
is started. Non-Linux builds omit these gauges. Missing/unreadable procfs data
|
||||||
|
omits the affected observation instead of reporting zero or failing the daemon.
|
||||||
|
No hostname, PID, command line or environment values are exported by this slice.
|
||||||
|
Service/instance resource attributes identify each daemon; two daemons on the
|
||||||
|
same host can report the same host uptime, so do not sum it across processes.
|
||||||
|
CPU time is not elapsed time or CPU percentage: use its rate for consumption.
|
||||||
|
Open-FD sampling can temporarily open descriptors counted by the measurement.
|
||||||
|
Process metrics describe the daemon itself, not its child shells/PTY programs
|
||||||
|
or the entire host. CPU/FD collection is also supplied by `wakey-linux`;
|
||||||
|
unsupported platform readers may omit these optional measurements.
|
||||||
|
Host memory/load come from the same optional Linux reader. `MemAvailable`
|
||||||
|
estimates memory reusable without swapping and is not `MemFree`; no fallback
|
||||||
|
is invented on older kernels. Load averages include runnable and uninterruptible
|
||||||
|
tasks, so they are not CPU percentages. These are the visible kernel/procfs
|
||||||
|
figures, not cgroup memory limits or container-specific load measurements;
|
||||||
|
multiple daemons may export duplicate views of the same host.
|
||||||
|
|
||||||
|
RSS includes anonymous, file-backed and shared resident pages, not just heap,
|
||||||
|
and is an approximate kernel measurement. Process age uses procfs start ticks
|
||||||
|
and uptime, not wall-clock subtraction or telemetry initialization time.
|
||||||
|
Container/time namespaces can change the visible host clock; this is the
|
||||||
|
process's Linux view, not an assertion about a physical machine's boot time.
|
||||||
|
The Linux parser is [procfs](https://docs.rs/procfs/0.18.0/procfs/), with optional
|
||||||
|
default features disabled. Public `/healthz` remains unchanged.
|
||||||
|
|
||||||
|
Current-state fleet gauges sample independently every 30 seconds, with bounded
|
||||||
|
revision retries to avoid mixed-state counts. Enrolled empty agents report zero;
|
||||||
|
removed labels disappear when no enrollment or retained rows remain. Failed samples
|
||||||
|
are omitted, not converted to zeros. No per-device labels are emitted. See the
|
||||||
|
recipe book for scopes, queries, and the distinction between devices and endpoints.
|
||||||
|
|
||||||
|
## Failure policy
|
||||||
|
|
||||||
|
Trace and log queues are capped at 256 records each, with export batches of 64
|
||||||
|
and a 3-second request timeout. Export happens off the application path. The
|
||||||
|
OTLP exporters use their recommended retry policy: up to three retries after
|
||||||
|
the first attempt, exponential backoff and jitter, within the exporter timeout
|
||||||
|
budget. Temporary transport/receiver failures can recover; authentication
|
||||||
|
rejection and partial-success responses are not retried. Requests already in
|
||||||
|
flight can exceed the retry budget, so it is not a hard wall-clock deadline.
|
||||||
|
The application does not wrap exports in another retry loop. On an
|
||||||
|
outage, telemetry can be dropped rather than accumulating unbounded router RAM;
|
||||||
|
terminal output follows its existing lossless backpressure policy independently.
|
||||||
|
|
||||||
|
Automated receiver tests reject the first export of each signal and confirm
|
||||||
|
recovery, check the retry-attempt limit on a persistent outage, and confirm that
|
||||||
|
bad credentials cause only one attempt. These tests do not establish durable
|
||||||
|
delivery during a long outage; bounded queues can still drop telemetry.
|
||||||
|
The bound is a record count, not an absolute byte limit on arbitrary log messages.
|
||||||
|
Providers flush on orderly daemon exit (Ctrl-C/SIGTERM), on a blocking worker while
|
||||||
|
Tokio remains alive. Only completed spans are exported; this is not graceful draining
|
||||||
|
of every in-flight request. SIGKILL/power loss cannot flush. This is not a persistent
|
||||||
|
telemetry spool.
|
||||||
|
|
||||||
|
## Validation
|
||||||
|
|
||||||
|
```sh
|
||||||
|
cargo test -p wakey-core --features telemetry --test telemetry
|
||||||
|
```
|
||||||
|
|
||||||
|
The test starts a temporary gRPC collector and verifies all three signals,
|
||||||
|
command parent/child IDs, log correlation, and shutdown delivery without sleeps.
|
||||||
@@ -53,6 +53,9 @@ Use netlink only when:
|
|||||||
|
|
||||||
Today that mainly applies to `neighbor::nl`.
|
Today that mainly applies to `neighbor::nl`.
|
||||||
|
|
||||||
|
`wakey-linux` explicitly uses netlink for neighbor and address inventory on its
|
||||||
|
Unix-gated path, without automatic JSON fallback.
|
||||||
|
|
||||||
## Relationship to wakey
|
## Relationship to wakey
|
||||||
|
|
||||||
`wakey` is the product.
|
`wakey` is the product.
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
//! rtnetlink-based neighbor table query. One syscall, filter in userspace.
|
//! rtnetlink-based neighbor query: dump links and neighbors, then filter in userspace.
|
||||||
#![cfg(unix)]
|
#![cfg(unix)]
|
||||||
use std::{
|
use std::{
|
||||||
collections::{HashMap, HashSet},
|
collections::{HashMap, HashSet},
|
||||||
|
|||||||
Regular → Executable
+16
-1
@@ -14,7 +14,22 @@ pscp.exe -l root -scp -pw $PASSWD -r 192.168.100.1:/etc/ldlda_help $PSScriptRoot
|
|||||||
pscp.exe -l root -scp -pw $PASSWD 192.168.100.1:/etc/rc.local $PSScriptRoot
|
pscp.exe -l root -scp -pw $PASSWD 192.168.100.1:/etc/rc.local $PSScriptRoot
|
||||||
pscp.exe -l root -scp -pw $PASSWD -r 192.168.100.1:/etc/wakey-agent $PSScriptRoot
|
pscp.exe -l root -scp -pw $PASSWD -r 192.168.100.1:/etc/wakey-agent $PSScriptRoot
|
||||||
|
|
||||||
Remove-Item -Recurse (Join-Path $PSScriptRoot "root")
|
$uciDir = Join-Path $PSScriptRoot 'uci-config'
|
||||||
|
New-Item -ItemType Directory -Force -Path $uciDir | Out-Null
|
||||||
|
# This ignored backup contains credentials; do not add it to version control.
|
||||||
|
pscp.exe -l root -scp -pw $PASSWD 192.168.100.1:/etc/config/wakey (Join-Path $uciDir 'wakey')
|
||||||
|
|
||||||
|
$rootBackup = Join-Path $PSScriptRoot 'root'
|
||||||
|
if (Test-Path -LiteralPath $rootBackup) {
|
||||||
|
# Router backups include hidden/read-only files. Fail once if deletion is
|
||||||
|
# genuinely blocked rather than merging a fresh copy into a partial backup.
|
||||||
|
try {
|
||||||
|
Remove-Item -LiteralPath $rootBackup -Recurse -Force -ErrorAction Stop
|
||||||
|
}
|
||||||
|
catch {
|
||||||
|
throw "Cannot remove previous router backup '$rootBackup': $($_.Exception.Message)"
|
||||||
|
}
|
||||||
|
}
|
||||||
pscp.exe -l root -scp -pw $PASSWD -r 192.168.100.1:/root $PSScriptRoot
|
pscp.exe -l root -scp -pw $PASSWD -r 192.168.100.1:/root $PSScriptRoot
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -7,12 +7,26 @@ USE_PROCD=1
|
|||||||
NAME=wakey
|
NAME=wakey
|
||||||
BIN=/root/.bin/wakey-agent
|
BIN=/root/.bin/wakey-agent
|
||||||
CONFIG=/etc/wakey-agent/config.toml
|
CONFIG=/etc/wakey-agent/config.toml
|
||||||
|
UCI_CONFIG=/etc/config/wakey
|
||||||
|
|
||||||
start_service() {
|
start_service() {
|
||||||
|
local authorization='' log_filter='info'
|
||||||
|
if [ -f "$UCI_CONFIG" ]; then
|
||||||
|
config_load wakey || return 1
|
||||||
|
config_get authorization main otlp_authorization
|
||||||
|
config_get log_filter main log_filter 'info'
|
||||||
|
fi
|
||||||
procd_open_instance
|
procd_open_instance
|
||||||
procd_set_param command "$BIN" serve --config "$CONFIG"
|
procd_set_param command "$BIN" serve --config "$CONFIG"
|
||||||
procd_set_param file "$CONFIG"
|
procd_set_param file "$CONFIG"
|
||||||
# procd_set_param env RUST_LOG=wakey_agent=debug,wakey=debug
|
if [ -f "$UCI_CONFIG" ]; then
|
||||||
|
procd_append_param file "$UCI_CONFIG"
|
||||||
|
fi
|
||||||
|
procd_set_param env "RUST_LOG=$log_filter"
|
||||||
|
# Pass the complete header as one argument; never log its value.
|
||||||
|
if [ -n "$authorization" ]; then
|
||||||
|
procd_append_param env "WAKEY_OTLP_AUTHORIZATION=$authorization"
|
||||||
|
fi
|
||||||
procd_set_param respawn 5 1 0
|
procd_set_param respawn 5 1 0
|
||||||
procd_set_param stdout 1
|
procd_set_param stdout 1
|
||||||
procd_set_param stderr 1
|
procd_set_param stderr 1
|
||||||
|
|||||||
@@ -2,7 +2,7 @@
|
|||||||
set -eu
|
set -eu
|
||||||
|
|
||||||
ROOT="$(CDPATH='' cd -- "$(dirname -- "$0")/.." && pwd)"
|
ROOT="$(CDPATH='' cd -- "$(dirname -- "$0")/.." && pwd)"
|
||||||
DB="${SQLX_PREPARE_DB:-/tmp/wakey-sqlx-prepare.sqlite3}"
|
DB="${SQLX_PREPARE_DB:-$ROOT/wakey-prepare.sqlite3}"
|
||||||
DATABASE_URL="sqlite://$DB"
|
DATABASE_URL="sqlite://$DB"
|
||||||
|
|
||||||
if ! cargo sqlx --version >/dev/null 2>&1; then
|
if ! cargo sqlx --version >/dev/null 2>&1; then
|
||||||
@@ -23,3 +23,5 @@ cp .sqlx/*.json wakey-control-plane/.sqlx/
|
|||||||
|
|
||||||
echo "SQLx metadata prepared in $ROOT/.sqlx"
|
echo "SQLx metadata prepared in $ROOT/.sqlx"
|
||||||
echo "SQLx package metadata mirrored in $ROOT/wakey-control-plane/.sqlx"
|
echo "SQLx package metadata mirrored in $ROOT/wakey-control-plane/.sqlx"
|
||||||
|
echo "Prepare DB (referenced by .env for live compile-time checking): $DB"
|
||||||
|
echo "DATABASE_URL=$DATABASE_URL"
|
||||||
Executable
+111
@@ -0,0 +1,111 @@
|
|||||||
|
#!/usr/bin/env -S uv run --script
|
||||||
|
# /// script
|
||||||
|
# requires-python = ">=3.11"
|
||||||
|
# dependencies = [
|
||||||
|
# "brotli==1.2.0",
|
||||||
|
# "fonttools==4.63.0",
|
||||||
|
# ]
|
||||||
|
# ///
|
||||||
|
"""Rebuild Wakey's self-hosted terminal webfont from Nerd Fonts."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import hashlib
|
||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
import tarfile
|
||||||
|
import tempfile
|
||||||
|
import urllib.request
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from fontTools.ttLib import TTFont
|
||||||
|
|
||||||
|
NERD_FONTS_VERSION = "v3.4.0"
|
||||||
|
ARCHIVE_SHA256 = "ef552a3e638f25125c6ad4c51176a6adcdce295ab1d2ffacf0db060caf8c1582"
|
||||||
|
WOFF2_SHA256 = "50680a64466fbcc3eff68e00f84505f55681c86f8450c042416af5a8dbc1cce1"
|
||||||
|
FONT_FILE = "JetBrainsMonoNerdFontMono-Regular"
|
||||||
|
ARCHIVE_URL = (
|
||||||
|
"https://github.com/ryanoasis/nerd-fonts/releases/download/"
|
||||||
|
f"{NERD_FONTS_VERSION}/JetBrainsMono.tar.xz"
|
||||||
|
)
|
||||||
|
|
||||||
|
ROOT = Path(__file__).resolve().parent.parent
|
||||||
|
OUTPUT_DIR = ROOT / "ui/src/assets/fonts/jetbrains-mono-nerd"
|
||||||
|
|
||||||
|
|
||||||
|
def sha256(path: Path) -> str:
|
||||||
|
digest = hashlib.sha256()
|
||||||
|
with path.open("rb") as source:
|
||||||
|
for chunk in iter(lambda: source.read(1024 * 1024), b""):
|
||||||
|
digest.update(chunk)
|
||||||
|
return digest.hexdigest()
|
||||||
|
|
||||||
|
|
||||||
|
def verify_sha256(path: Path, expected: str) -> None:
|
||||||
|
actual = sha256(path)
|
||||||
|
if actual != expected:
|
||||||
|
raise RuntimeError(
|
||||||
|
f"checksum mismatch for {path}\nexpected: {expected}\nactual: {actual}"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def download(url: str, destination: Path) -> None:
|
||||||
|
request = urllib.request.Request(url, headers={"User-Agent": "wakey-font-preparer"})
|
||||||
|
with (
|
||||||
|
urllib.request.urlopen(request, timeout=60) as response,
|
||||||
|
destination.open("wb") as output,
|
||||||
|
):
|
||||||
|
shutil.copyfileobj(response, output)
|
||||||
|
|
||||||
|
|
||||||
|
def extract_member(archive: tarfile.TarFile, name: str, destination: Path) -> None:
|
||||||
|
source = archive.extractfile(name)
|
||||||
|
if source is None:
|
||||||
|
raise RuntimeError(f"missing {name} in Nerd Fonts archive")
|
||||||
|
with source, destination.open("wb") as output:
|
||||||
|
shutil.copyfileobj(source, output)
|
||||||
|
|
||||||
|
|
||||||
|
def install_file(source: Path, destination: Path) -> None:
|
||||||
|
"""Atomically replace an asset without exposing a partial file to Vite."""
|
||||||
|
temporary = destination.with_name(f".{destination.name}.tmp")
|
||||||
|
try:
|
||||||
|
shutil.copyfile(source, temporary)
|
||||||
|
os.chmod(temporary, 0o644)
|
||||||
|
temporary.replace(destination)
|
||||||
|
finally:
|
||||||
|
temporary.unlink(missing_ok=True)
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
|
||||||
|
with tempfile.TemporaryDirectory(prefix="wakey-terminal-font.") as temp:
|
||||||
|
work_dir = Path(temp)
|
||||||
|
archive_path = work_dir / "JetBrainsMono.tar.xz"
|
||||||
|
ttf_path = work_dir / f"{FONT_FILE}.ttf"
|
||||||
|
woff2_path = work_dir / f"{FONT_FILE}.woff2"
|
||||||
|
license_path = work_dir / "OFL.txt"
|
||||||
|
|
||||||
|
print(f"Downloading Nerd Fonts {NERD_FONTS_VERSION} JetBrainsMono archive")
|
||||||
|
download(ARCHIVE_URL, archive_path)
|
||||||
|
verify_sha256(archive_path, ARCHIVE_SHA256)
|
||||||
|
|
||||||
|
with tarfile.open(archive_path, "r:xz") as archive:
|
||||||
|
extract_member(archive, ttf_path.name, ttf_path)
|
||||||
|
extract_member(archive, license_path.name, license_path)
|
||||||
|
|
||||||
|
# FontTools otherwise stamps the current time into the head table,
|
||||||
|
# making equivalent rebuilds produce different WOFF2 files.
|
||||||
|
font = TTFont(ttf_path, recalcTimestamp=False)
|
||||||
|
font.flavor = "woff2"
|
||||||
|
font.save(woff2_path)
|
||||||
|
verify_sha256(woff2_path, WOFF2_SHA256)
|
||||||
|
|
||||||
|
install_file(woff2_path, OUTPUT_DIR / woff2_path.name)
|
||||||
|
install_file(license_path, OUTPUT_DIR / license_path.name)
|
||||||
|
|
||||||
|
print(f"Prepared {OUTPUT_DIR / f'{FONT_FILE}.woff2'}")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
@@ -0,0 +1,69 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
# Host-side contract test; OpenWrt's UCI and procd helpers are stubbed.
|
||||||
|
set -eu
|
||||||
|
ROOT="$(CDPATH='' cd -- "$(dirname -- "$0")/../.." && pwd)"
|
||||||
|
TMP="$(mktemp -d)"
|
||||||
|
trap 'rm -rf "$TMP"' EXIT HUP INT TERM
|
||||||
|
|
||||||
|
config_load() { [ "$1" = wakey ]; loaded=1; return "$load_status"; }
|
||||||
|
config_get() {
|
||||||
|
[ "$2" = main ]
|
||||||
|
case "$3" in
|
||||||
|
otlp_authorization) authorization=$test_authorization ;;
|
||||||
|
log_filter) log_filter=${test_log_filter:-$4} ;;
|
||||||
|
*) return 1 ;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
procd_open_instance() { opened=1; }
|
||||||
|
procd_close_instance() { :; }
|
||||||
|
procd_set_param() {
|
||||||
|
if [ "$1" = env ]; then
|
||||||
|
[ "$#" -eq 2 ]
|
||||||
|
received_filter=$2
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
procd_append_param() {
|
||||||
|
case "$1" in
|
||||||
|
env) [ "$#" -eq 2 ]; received_auth=$2 ;;
|
||||||
|
file) tracked_file=$2 ;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
. "$ROOT/scripts/init/openwrt/wakey"
|
||||||
|
UCI_CONFIG="$TMP/wakey"
|
||||||
|
touch "$UCI_CONFIG"
|
||||||
|
load_status=0
|
||||||
|
loaded=0
|
||||||
|
opened=0
|
||||||
|
test_authorization='Basic dGVzdDp0ZXN0=='
|
||||||
|
test_log_filter='wakey_agent=debug,opentelemetry=warn'
|
||||||
|
start_service
|
||||||
|
[ "$loaded" -eq 1 ]
|
||||||
|
[ "$opened" -eq 1 ]
|
||||||
|
[ "$received_auth" = "WAKEY_OTLP_AUTHORIZATION=$test_authorization" ]
|
||||||
|
[ "$received_filter" = "RUST_LOG=$test_log_filter" ]
|
||||||
|
[ "$tracked_file" = "$UCI_CONFIG" ]
|
||||||
|
|
||||||
|
# Missing options retain the default filter and do not export empty credentials.
|
||||||
|
test_authorization=''
|
||||||
|
test_log_filter=''
|
||||||
|
received_auth=''
|
||||||
|
start_service
|
||||||
|
[ "$received_auth" = '' ]
|
||||||
|
[ "$received_filter" = 'RUST_LOG=info' ]
|
||||||
|
|
||||||
|
load_status=1
|
||||||
|
opened=0
|
||||||
|
if start_service; then
|
||||||
|
echo 'failed UCI load was accepted' >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
[ "$opened" -eq 0 ]
|
||||||
|
|
||||||
|
UCI_CONFIG="$TMP/absent"
|
||||||
|
loaded=0
|
||||||
|
tracked_file=''
|
||||||
|
start_service
|
||||||
|
[ "$loaded" -eq 0 ]
|
||||||
|
[ "$tracked_file" = '' ]
|
||||||
|
[ "$received_filter" = 'RUST_LOG=info' ]
|
||||||
|
echo 'PASS: OpenWrt UCI environment forwarding and optional configuration'
|
||||||
@@ -1,5 +1,5 @@
|
|||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use tracing::{debug, warn};
|
use tracing::{Instrument, debug, warn};
|
||||||
use wakey_core::{
|
use wakey_core::{
|
||||||
Device, DeviceInventory, DeviceObservationFact, DhcpLease, DhcpLeaseWithState, InventoryQuery,
|
Device, DeviceInventory, DeviceObservationFact, DhcpLease, DhcpLeaseWithState, InventoryQuery,
|
||||||
NeighborEntry, Presence, Query,
|
NeighborEntry, Presence, Query,
|
||||||
@@ -19,10 +19,28 @@ pub async fn resolve_devices(input: impl Into<String>) -> Result<Vec<Device>> {
|
|||||||
/// This is the current center of gravity for the service layer. Higher-level
|
/// This is the current center of gravity for the service layer. Higher-level
|
||||||
/// status and wake flows should prefer deriving from inventory rather than
|
/// status and wake flows should prefer deriving from inventory rather than
|
||||||
/// directly from raw Linux source rows.
|
/// directly from raw Linux source rows.
|
||||||
|
#[tracing::instrument(name = "inventory", skip_all, fields(devices = tracing::field::Empty))]
|
||||||
pub async fn inventory(query: InventoryQuery) -> Result<DeviceInventory> {
|
pub async fn inventory(query: InventoryQuery) -> Result<DeviceInventory> {
|
||||||
let neighbors = wakey_linux::devices::query_neighbors(&query).await?;
|
// Instrument futures, not entered guards: each source remains correctly
|
||||||
let leases = get_leases().await?;
|
// parented when Tokio moves the task between workers during an await.
|
||||||
let observations = match wakey_linux::observations::list_local_observations().await {
|
let neighbors = async {
|
||||||
|
let rows = wakey_linux::devices::query_neighbors(&query).await?;
|
||||||
|
debug!(rows = rows.len(), "neighbor source collected");
|
||||||
|
Ok::<_, anyhow::Error>(rows)
|
||||||
|
}
|
||||||
|
.instrument(tracing::info_span!("inventory_neighbors"))
|
||||||
|
.await?;
|
||||||
|
let leases = async {
|
||||||
|
let rows = get_leases().await?;
|
||||||
|
debug!(rows = rows.len(), "DHCP source collected");
|
||||||
|
Ok::<_, anyhow::Error>(rows)
|
||||||
|
}
|
||||||
|
.instrument(tracing::info_span!("inventory_dhcp_leases"))
|
||||||
|
.await?;
|
||||||
|
let observations = match wakey_linux::observations::list_local_observations()
|
||||||
|
.instrument(tracing::info_span!("inventory_hook_store"))
|
||||||
|
.await
|
||||||
|
{
|
||||||
Ok(observations) => observations
|
Ok(observations) => observations
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.filter_map(local_observation_to_fact)
|
.filter_map(local_observation_to_fact)
|
||||||
@@ -39,6 +57,7 @@ pub async fn inventory(query: InventoryQuery) -> Result<DeviceInventory> {
|
|||||||
"building device inventory"
|
"building device inventory"
|
||||||
);
|
);
|
||||||
let devices = merge_devices_with_observations(neighbors, leases, observations, &query);
|
let devices = merge_devices_with_observations(neighbors, leases, observations, &query);
|
||||||
|
tracing::Span::current().record("devices", devices.len());
|
||||||
debug!(devices = devices.len(), "merged device inventory");
|
debug!(devices = devices.len(), "merged device inventory");
|
||||||
Ok(DeviceInventory { devices })
|
Ok(DeviceInventory { devices })
|
||||||
}
|
}
|
||||||
@@ -57,6 +76,7 @@ pub fn merge_devices(
|
|||||||
|
|
||||||
/// Merge raw neighbor entries, DHCP leases, and hook observations into device
|
/// Merge raw neighbor entries, DHCP leases, and hook observations into device
|
||||||
/// aggregates.
|
/// aggregates.
|
||||||
|
#[tracing::instrument(name = "inventory_merge", skip_all, fields(neighbors = neighbors.len(), leases = leases.len(), observations = observations.len()))]
|
||||||
pub fn merge_devices_with_observations(
|
pub fn merge_devices_with_observations(
|
||||||
neighbors: Vec<NeighborEntry>,
|
neighbors: Vec<NeighborEntry>,
|
||||||
leases: Vec<DhcpLeaseWithState>,
|
leases: Vec<DhcpLeaseWithState>,
|
||||||
@@ -169,11 +189,13 @@ pub fn local_observation_to_fact(
|
|||||||
}
|
}
|
||||||
|
|
||||||
const fn presence_rank(presence: Presence) -> u8 {
|
const fn presence_rank(presence: Presence) -> u8 {
|
||||||
|
// Match the core signal order: Unknown has no verdict, while Offline
|
||||||
|
// carries concrete evidence and belongs above it in the inventory list.
|
||||||
match presence {
|
match presence {
|
||||||
Presence::Online => 3,
|
Presence::Online => 3,
|
||||||
Presence::LikelyOnline => 2,
|
Presence::LikelyOnline => 2,
|
||||||
Presence::Unknown => 1,
|
Presence::Offline => 1,
|
||||||
Presence::Offline => 0,
|
Presence::Unknown => 0,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -204,6 +226,23 @@ mod tests {
|
|||||||
}]
|
}]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn inventory_sorts_offline_before_unknown() {
|
||||||
|
let neighbors = [NeighborState::Incomplete, NeighborState::Failed]
|
||||||
|
.into_iter()
|
||||||
|
.enumerate()
|
||||||
|
.map(|(index, state)| NeighborEntry {
|
||||||
|
ip: IpAddr::V4(Ipv4Addr::new(192, 168, 1, index as u8 + 10)),
|
||||||
|
dev: None,
|
||||||
|
mac: None,
|
||||||
|
state,
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
let devices = merge_devices(neighbors, Vec::new(), &InventoryQueryBuilder::new().build());
|
||||||
|
assert_eq!(devices[0].presence, Presence::Offline);
|
||||||
|
assert_eq!(devices[1].presence, Presence::Unknown);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn merge_devices_applies_and_across_categories() {
|
fn merge_devices_applies_and_across_categories() {
|
||||||
let query = InventoryQueryBuilder::new()
|
let query = InventoryQueryBuilder::new()
|
||||||
|
|||||||
+4
-1
@@ -2,7 +2,10 @@
|
|||||||
<html lang="en">
|
<html lang="en">
|
||||||
<head>
|
<head>
|
||||||
<meta charset="UTF-8" />
|
<meta charset="UTF-8" />
|
||||||
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
<meta
|
||||||
|
name="viewport"
|
||||||
|
content="width=device-width, initial-scale=1.0, interactive-widget=resizes-content"
|
||||||
|
/>
|
||||||
<title>Wakey Operator UI</title>
|
<title>Wakey Operator UI</title>
|
||||||
<script>
|
<script>
|
||||||
(function () {
|
(function () {
|
||||||
|
|||||||
+7
-6
@@ -15,12 +15,13 @@
|
|||||||
"@base-ui/react": "^1.4.0",
|
"@base-ui/react": "^1.4.0",
|
||||||
"@fontsource-variable/inter": "^5.2.8",
|
"@fontsource-variable/inter": "^5.2.8",
|
||||||
"@tailwindcss/vite": "^4.2.2",
|
"@tailwindcss/vite": "^4.2.2",
|
||||||
"@xterm/addon-clipboard": "^0.2.0",
|
"@xterm/addon-clipboard": "0.3.0-beta.301",
|
||||||
"@xterm/addon-fit": "^0.11.0",
|
"@xterm/addon-fit": "0.12.0-beta.301",
|
||||||
"@xterm/addon-unicode-graphemes": "^0.4.0",
|
"@xterm/addon-image": "0.10.0-beta.301",
|
||||||
"@xterm/addon-unicode11": "^0.9.0",
|
"@xterm/addon-unicode-graphemes": "0.5.0-beta.301",
|
||||||
"@xterm/addon-web-links": "^0.12.0",
|
"@xterm/addon-unicode11": "0.10.0-beta.301",
|
||||||
"@xterm/xterm": "^6.0.0",
|
"@xterm/addon-web-links": "0.13.0-beta.301",
|
||||||
|
"@xterm/xterm": "6.1.0-beta.301",
|
||||||
"class-variance-authority": "^0.7.1",
|
"class-variance-authority": "^0.7.1",
|
||||||
"clsx": "^2.1.1",
|
"clsx": "^2.1.1",
|
||||||
"lucide-react": "^1.8.0",
|
"lucide-react": "^1.8.0",
|
||||||
|
|||||||
Generated
+61
-36
@@ -18,23 +18,26 @@ importers:
|
|||||||
specifier: ^4.2.2
|
specifier: ^4.2.2
|
||||||
version: 4.2.2([email protected](@types/[email protected])([email protected]))
|
version: 4.2.2([email protected](@types/[email protected])([email protected]))
|
||||||
'@xterm/addon-clipboard':
|
'@xterm/addon-clipboard':
|
||||||
specifier: ^0.2.0
|
specifier: 0.3.0-beta.301
|
||||||
version: 0.2.0
|
version: 0.3.0-beta.301(@xterm/[email protected])
|
||||||
'@xterm/addon-fit':
|
'@xterm/addon-fit':
|
||||||
specifier: ^0.11.0
|
specifier: 0.12.0-beta.301
|
||||||
version: 0.11.0
|
version: 0.12.0-beta.301(@xterm/[email protected])
|
||||||
|
'@xterm/addon-image':
|
||||||
|
specifier: 0.10.0-beta.301
|
||||||
|
version: 0.10.0-beta.301(@xterm/[email protected])
|
||||||
'@xterm/addon-unicode-graphemes':
|
'@xterm/addon-unicode-graphemes':
|
||||||
specifier: ^0.4.0
|
specifier: 0.5.0-beta.301
|
||||||
version: 0.4.0
|
version: 0.5.0-beta.301(@xterm/[email protected])
|
||||||
'@xterm/addon-unicode11':
|
'@xterm/addon-unicode11':
|
||||||
specifier: ^0.9.0
|
specifier: 0.10.0-beta.301
|
||||||
version: 0.9.0
|
version: 0.10.0-beta.301(@xterm/[email protected])
|
||||||
'@xterm/addon-web-links':
|
'@xterm/addon-web-links':
|
||||||
specifier: ^0.12.0
|
specifier: 0.13.0-beta.301
|
||||||
version: 0.12.0
|
version: 0.13.0-beta.301(@xterm/[email protected])
|
||||||
'@xterm/xterm':
|
'@xterm/xterm':
|
||||||
specifier: ^6.0.0
|
specifier: 6.1.0-beta.301
|
||||||
version: 6.0.0
|
version: 6.1.0-beta.301
|
||||||
class-variance-authority:
|
class-variance-authority:
|
||||||
specifier: ^0.7.1
|
specifier: ^0.7.1
|
||||||
version: 0.7.1
|
version: 0.7.1
|
||||||
@@ -833,23 +836,38 @@ packages:
|
|||||||
peerDependencies:
|
peerDependencies:
|
||||||
vite: ^4.2.0 || ^5.0.0 || ^6.0.0 || ^7.0.0
|
vite: ^4.2.0 || ^5.0.0 || ^6.0.0 || ^7.0.0
|
||||||
|
|
||||||
'@xterm/addon-clipboard@0.2.0':
|
'@xterm/addon-clipboard@0.3.0-beta.301':
|
||||||
resolution: {integrity: sha512-Dl31BCtBhLaUEECUbEiVcCLvLBbaeGYdT7NofB8OJkGTD3MWgBsaLjXvfGAD4tQNHhm6mbKyYkR7XD8kiZsdNg==}
|
resolution: {integrity: sha512-DRL0EybPhWzgndaTeakVbh/HvJF7yzxRtfNCnIudFZQaGugFYsEUZNOaV2NZev8z5kLwor6miH1j3DNMI8Paxg==}
|
||||||
|
peerDependencies:
|
||||||
|
'@xterm/xterm': ^6.1.0-beta.301
|
||||||
|
|
||||||
'@xterm/[email protected]1.0':
|
'@xterm/[email protected]2.0-beta.301':
|
||||||
resolution: {integrity: sha512-jYcgT6xtVYhnhgxh3QgYDnnNMYTcf8ElbxxFzX0IZo+vabQqSPAjC3c1wJrKB5E19VwQei89QCiZZP86DCPF7g==}
|
resolution: {integrity: sha512-MukRyJLAFQrW/++aM209jBZ8G9J/Ioe4SGS6c3ba87v+zYuX/I66JrDB7sVfykQ9t7DlT+c1MWl1jTlX2kyx1w==}
|
||||||
|
peerDependencies:
|
||||||
|
'@xterm/xterm': ^6.1.0-beta.304
|
||||||
|
|
||||||
'@xterm/addon-[email protected]':
|
'@xterm/addon-[email protected]':
|
||||||
resolution: {integrity: sha512-9+/CqwbKcnlkJU4d3wIgO+wjsL8f6vyz+UwUWLu6nADQz8Gr8ONqGCJfdDjIdI+yYZLABQqQy47FzEM6AWELjw==}
|
resolution: {integrity: sha512-PJZL4GlQ1ReIgbffFMaOAFxc3wOit21/G2UEM82RpH3hNHjn6MH8ilz3ToDeBjYuOqU7ienBaeIuMbNktX4/HA==}
|
||||||
|
peerDependencies:
|
||||||
|
'@xterm/xterm': ^6.1.0-beta.304
|
||||||
|
|
||||||
'@xterm/addon-unicode[email protected]':
|
'@xterm/addon-unicode[email protected]':
|
||||||
resolution: {integrity: sha512-FxDnYcyuXhNl+XSqGZL/t0U9eiNb/q3EWT5rYkQT/zuig8Gz/VagnQANKHdDWFM2lTMk9ly0EFQxxxtZUoRetw==}
|
resolution: {integrity: sha512-EBZIZYS6RItWTGO/KpLqV6y3IKGYyugbP07lK0x6G7185FVA61L2t0/XVtVEQNpczsbPo8zAxBhpoV72TP8CYw==}
|
||||||
|
peerDependencies:
|
||||||
|
'@xterm/xterm': ^6.1.0-beta.304
|
||||||
|
|
||||||
'@xterm/addon-[email protected]':
|
'@xterm/addon-[email protected]':
|
||||||
resolution: {integrity: sha512-4Smom3RPyVp7ZMYOYDoC/9eGJJJqYhnPLGGqJ6wOBfB8VxPViJNSKdgRYb8NpaM6YSelEKbA2SStD7lGyqaobw==}
|
resolution: {integrity: sha512-SS5BzfbvfFIikE4xZFAtHu6n0Kl5R1ZCPx41RL2WULLSE+HFo/WmYee53tMxNK/QUjpcQWepeVB7HebCPhHnig==}
|
||||||
|
peerDependencies:
|
||||||
|
'@xterm/xterm': ^6.1.0-beta.304
|
||||||
|
|
||||||
'@xterm/[email protected]':
|
'@xterm/[email protected]':
|
||||||
resolution: {integrity: sha512-TQwDdQGtwwDt+2cgKDLn0IRaSxYu1tSUjgKarSDkUM0ZNiSRXFpjxEsvc/Zgc5kq5omJ+V0a8/kIM2WD3sMOYg==}
|
resolution: {integrity: sha512-BpY297K+FGLsTQemnA6uOg6hk85DwVuUvWCqE9rIAcDxOOC7fP9GKAZTIfaEos/EZk2Ei23vD6w0P5Br6+RU7Q==}
|
||||||
|
peerDependencies:
|
||||||
|
'@xterm/xterm': ^6.1.0-beta.304
|
||||||
|
|
||||||
|
'@xterm/[email protected]':
|
||||||
|
resolution: {integrity: sha512-y1pH42+Mx+h5E2xDgyilYXBKYkn8/6+anToiTmIt8gVww/2gHFqheK0vW+m9LADQsZ5AFs+SQvcK+tOyxWfRJA==}
|
||||||
|
|
||||||
[email protected]:
|
[email protected]:
|
||||||
resolution: {integrity: sha512-5cvg6CtKwfgdmVqY1WIiXKc3Q1bkRqGLi+2W/6ao+6Y7gu/RCwRuAhGEzh5B4KlszSuTLgZYuqFqo5bImjNKng==}
|
resolution: {integrity: sha512-5cvg6CtKwfgdmVqY1WIiXKc3Q1bkRqGLi+2W/6ao+6Y7gu/RCwRuAhGEzh5B4KlszSuTLgZYuqFqo5bImjNKng==}
|
||||||
@@ -1439,9 +1457,6 @@ packages:
|
|||||||
[email protected]:
|
[email protected]:
|
||||||
resolution: {integrity: sha512-d7kPDd34KO/YnzaDOlikGpOurfF0ByC2sEV4cANCtdqLlTfBlw2p14O/5d/zv40gJPbIQxfES3nSx1/oYNyuZQ==}
|
resolution: {integrity: sha512-d7kPDd34KO/YnzaDOlikGpOurfF0ByC2sEV4cANCtdqLlTfBlw2p14O/5d/zv40gJPbIQxfES3nSx1/oYNyuZQ==}
|
||||||
|
|
||||||
[email protected]:
|
|
||||||
resolution: {integrity: sha512-5xVjhUZlHHeuO2W7w2rDFj/Kl1xLX+HjZxdOQwCsUOifl6UaoH1o1wsbsTMz+r0aeC7gCijvru02j6TfKZWzKg==}
|
|
||||||
|
|
||||||
[email protected]:
|
[email protected]:
|
||||||
resolution: {integrity: sha512-RdJUflcE3cUzKiMqQgsCu06FPu9UdIJO0beYbPhHN4k6apgJtifcoCtT9bcxOpYBtpD2kCM6Sbzg4CausW/PKQ==}
|
resolution: {integrity: sha512-RdJUflcE3cUzKiMqQgsCu06FPu9UdIJO0beYbPhHN4k6apgJtifcoCtT9bcxOpYBtpD2kCM6Sbzg4CausW/PKQ==}
|
||||||
|
|
||||||
@@ -2847,19 +2862,31 @@ snapshots:
|
|||||||
transitivePeerDependencies:
|
transitivePeerDependencies:
|
||||||
- supports-color
|
- supports-color
|
||||||
|
|
||||||
'@xterm/addon-clipboard@0.2.0':
|
'@xterm/addon-clipboard@0.3.0-beta.301(@xterm/[email protected])':
|
||||||
dependencies:
|
dependencies:
|
||||||
js-base64: 3.8.1
|
'@xterm/xterm': 6.1.0-beta.301
|
||||||
|
|
||||||
'@xterm/[email protected]1.0': {}
|
'@xterm/[email protected]2.0-beta.301(@xterm/[email protected])':
|
||||||
|
dependencies:
|
||||||
|
'@xterm/xterm': 6.1.0-beta.301
|
||||||
|
|
||||||
'@xterm/addon-[email protected]': {}
|
'@xterm/addon-[email protected](@xterm/[email protected])':
|
||||||
|
dependencies:
|
||||||
|
'@xterm/xterm': 6.1.0-beta.301
|
||||||
|
|
||||||
'@xterm/addon-unicode[email protected]': {}
|
'@xterm/addon-unicode[email protected](@xterm/[email protected])':
|
||||||
|
dependencies:
|
||||||
|
'@xterm/xterm': 6.1.0-beta.301
|
||||||
|
|
||||||
'@xterm/addon-web-links@0.12.0': {}
|
'@xterm/addon-[email protected](@xterm/xterm@6.1.0-beta.301)':
|
||||||
|
dependencies:
|
||||||
|
'@xterm/xterm': 6.1.0-beta.301
|
||||||
|
|
||||||
'@xterm/[email protected]': {}
|
'@xterm/[email protected](@xterm/[email protected])':
|
||||||
|
dependencies:
|
||||||
|
'@xterm/xterm': 6.1.0-beta.301
|
||||||
|
|
||||||
|
'@xterm/[email protected]': {}
|
||||||
|
|
||||||
[email protected]:
|
[email protected]:
|
||||||
dependencies:
|
dependencies:
|
||||||
@@ -3403,8 +3430,6 @@ snapshots:
|
|||||||
|
|
||||||
[email protected]: {}
|
[email protected]: {}
|
||||||
|
|
||||||
[email protected]: {}
|
|
||||||
|
|
||||||
[email protected]: {}
|
[email protected]: {}
|
||||||
|
|
||||||
[email protected]:
|
[email protected]:
|
||||||
|
|||||||
+105
-15
@@ -1,4 +1,4 @@
|
|||||||
import { lazy, Suspense, useEffect, useState } from "react";
|
import { lazy, Suspense, useEffect, useRef, useState } from "react";
|
||||||
import { Navigate, Route, Routes } from "react-router-dom";
|
import { Navigate, Route, Routes } from "react-router-dom";
|
||||||
import { toast } from "sonner";
|
import { toast } from "sonner";
|
||||||
|
|
||||||
@@ -40,22 +40,48 @@ export function App() {
|
|||||||
const [history, setHistory] = useState<AlertTransition[]>([]);
|
const [history, setHistory] = useState<AlertTransition[]>([]);
|
||||||
const [audit, setAudit] = useState<AuditEvent[]>([]);
|
const [audit, setAudit] = useState<AuditEvent[]>([]);
|
||||||
const [selectedAgentId, setSelectedAgentId] = useState("");
|
const [selectedAgentId, setSelectedAgentId] = useState("");
|
||||||
|
const [alertsHealth, setAlertsHealth] = useState<{
|
||||||
|
available: boolean;
|
||||||
|
reason?: string;
|
||||||
|
} | null>(null);
|
||||||
|
const [alertsStreamConnected, setAlertsStreamConnected] = useState(false);
|
||||||
|
const streamHealthReceived = useRef(false);
|
||||||
|
const alertsRevision = useRef(0);
|
||||||
|
|
||||||
const [state, setState] = useState<LoadState>("idle");
|
const [state, setState] = useState<LoadState>("idle");
|
||||||
|
|
||||||
|
async function loadAlertsWithoutBlocking() {
|
||||||
|
try {
|
||||||
|
const nextAlerts = await fetchAlerts();
|
||||||
|
if (!streamHealthReceived.current) {
|
||||||
|
setAlertsHealth({ available: true });
|
||||||
|
}
|
||||||
|
return nextAlerts;
|
||||||
|
} catch (err) {
|
||||||
|
if (!streamHealthReceived.current) {
|
||||||
|
setAlertsHealth({ available: false, reason: String(err) });
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async function loadAll() {
|
async function loadAll() {
|
||||||
|
const revision = alertsRevision.current;
|
||||||
setState("loading");
|
setState("loading");
|
||||||
try {
|
try {
|
||||||
const [nextAgents, nextAlerts, nextHistory, nextAudit] =
|
const [nextAgents, nextAlerts, nextHistory, nextAudit] =
|
||||||
await Promise.all([
|
await Promise.all([
|
||||||
fetchAgents(),
|
fetchAgents(),
|
||||||
fetchAlerts(),
|
loadAlertsWithoutBlocking(),
|
||||||
fetchAlertHistory(20),
|
fetchAlertHistory(20),
|
||||||
fetchAudit(30),
|
fetchAudit(30),
|
||||||
]);
|
]);
|
||||||
setAgents(nextAgents);
|
setAgents(nextAgents);
|
||||||
setAlerts(nextAlerts);
|
// Streamed state wins over HTTP requests started before its arrival.
|
||||||
|
if (revision === alertsRevision.current) {
|
||||||
|
if (nextAlerts) setAlerts(nextAlerts);
|
||||||
setHistory(nextHistory);
|
setHistory(nextHistory);
|
||||||
|
}
|
||||||
setAudit(nextAudit);
|
setAudit(nextAudit);
|
||||||
const firstConnectedAgentId =
|
const firstConnectedAgentId =
|
||||||
nextAgents.find((agent) => agent.connected)?.agent_id ?? "";
|
nextAgents.find((agent) => agent.connected)?.agent_id ?? "";
|
||||||
@@ -100,43 +126,103 @@ export function App() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async function refreshAlertsAndHistory() {
|
async function refreshAlertsAndHistory() {
|
||||||
|
const revision = alertsRevision.current;
|
||||||
const [nextAlerts, nextHistory] = await Promise.all([
|
const [nextAlerts, nextHistory] = await Promise.all([
|
||||||
fetchAlerts(),
|
loadAlertsWithoutBlocking(),
|
||||||
fetchAlertHistory(20),
|
fetchAlertHistory(20),
|
||||||
]);
|
]);
|
||||||
setAlerts(nextAlerts);
|
if (revision === alertsRevision.current) {
|
||||||
|
if (nextAlerts) setAlerts(nextAlerts);
|
||||||
setHistory(nextHistory);
|
setHistory(nextHistory);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
void loadAll();
|
void loadAll();
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
const wsUrl = `${window.location.protocol === "https:" ? "wss" : "ws"}://${window.location.host}/api/v1/control/alerts/ws`;
|
let socket: WebSocket | null = null;
|
||||||
const ws = new WebSocket(wsUrl);
|
let disposed = false;
|
||||||
|
let attempt = 0;
|
||||||
|
let openedAt = 0;
|
||||||
|
let reconnectTimer: number | undefined;
|
||||||
|
|
||||||
ws.onmessage = (evt) => {
|
const resync = () => {
|
||||||
|
const generation = alertsRevision.current;
|
||||||
|
void Promise.all([fetchAlerts(), fetchAlertHistory(20)])
|
||||||
|
.then(([nextAlerts, nextHistory]) => {
|
||||||
|
// A WS snapshot delivered while this request was in flight is newer.
|
||||||
|
if (disposed || generation !== alertsRevision.current) return;
|
||||||
|
setAlerts(nextAlerts);
|
||||||
|
setHistory(nextHistory);
|
||||||
|
})
|
||||||
|
.catch(() => undefined);
|
||||||
|
};
|
||||||
|
|
||||||
|
const connect = () => {
|
||||||
|
if (disposed) return;
|
||||||
|
const wsUrl = `${window.location.protocol === "https:" ? "wss" : "ws"}://${window.location.host}/api/v1/control/alerts/ws`;
|
||||||
|
socket = new WebSocket(wsUrl);
|
||||||
|
|
||||||
|
socket.onopen = () => {
|
||||||
|
if (disposed) return;
|
||||||
|
openedAt = Date.now();
|
||||||
|
setAlertsStreamConnected(true);
|
||||||
|
// Pull anything missed while the stream was down.
|
||||||
|
resync();
|
||||||
|
};
|
||||||
|
|
||||||
|
socket.onmessage = (evt) => {
|
||||||
|
if (disposed) return;
|
||||||
|
alertsRevision.current += 1;
|
||||||
try {
|
try {
|
||||||
const payload = JSON.parse(String(evt.data)) as {
|
const payload = JSON.parse(String(evt.data)) as {
|
||||||
|
type?: string;
|
||||||
|
available?: boolean;
|
||||||
|
reason?: string;
|
||||||
alerts?: Alert[];
|
alerts?: Alert[];
|
||||||
recent_transitions?: AlertTransition[];
|
recent_transitions?: AlertTransition[];
|
||||||
};
|
};
|
||||||
|
if (payload.type === "alerts_health") {
|
||||||
|
streamHealthReceived.current = true;
|
||||||
|
setAlertsHealth({
|
||||||
|
available: payload.available === true,
|
||||||
|
reason: payload.reason,
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
if (payload.alerts) setAlerts(payload.alerts);
|
if (payload.alerts) setAlerts(payload.alerts);
|
||||||
if (payload.recent_transitions) setHistory(payload.recent_transitions);
|
if (payload.recent_transitions)
|
||||||
|
setHistory(payload.recent_transitions);
|
||||||
} catch {
|
} catch {
|
||||||
// Ignore malformed stream payloads and keep current UI state.
|
// Ignore malformed stream payloads and keep current UI state.
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
ws.onerror = () => {
|
socket.onclose = () => {
|
||||||
const id = window.setInterval(() => {
|
socket = null;
|
||||||
void refreshAlertsAndHistory().catch(() => undefined);
|
if (disposed) return;
|
||||||
}, 8000);
|
setAlertsStreamConnected(false);
|
||||||
ws.onclose = () => window.clearInterval(id);
|
// Only a connection that stayed up proves the stream is healthy; an
|
||||||
|
// accept-then-close server must not pin us to the first backoff step.
|
||||||
|
if (openedAt && Date.now() - openedAt > 10_000) attempt = 0;
|
||||||
|
openedAt = 0;
|
||||||
|
// Browsers never reconnect a WebSocket on their own.
|
||||||
|
const delay =
|
||||||
|
Math.min(30_000, 1000 * 2 ** attempt) + Math.random() * 1000;
|
||||||
|
attempt += 1;
|
||||||
|
reconnectTimer = window.setTimeout(connect, delay);
|
||||||
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
return () => ws.close();
|
connect();
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
disposed = true;
|
||||||
|
if (reconnectTimer !== undefined) window.clearTimeout(reconnectTimer);
|
||||||
|
socket?.close();
|
||||||
|
};
|
||||||
}, []);
|
}, []);
|
||||||
|
|
||||||
return (
|
return (
|
||||||
@@ -175,6 +261,8 @@ export function App() {
|
|||||||
transitions={history}
|
transitions={history}
|
||||||
audit={audit}
|
audit={audit}
|
||||||
loading={state === "loading"}
|
loading={state === "loading"}
|
||||||
|
alertsHealth={alertsHealth}
|
||||||
|
alertsStreamConnected={alertsStreamConnected}
|
||||||
onRefresh={loadAll}
|
onRefresh={loadAll}
|
||||||
/>
|
/>
|
||||||
}
|
}
|
||||||
@@ -217,6 +305,8 @@ export function App() {
|
|||||||
<AlertsPage
|
<AlertsPage
|
||||||
alerts={alerts}
|
alerts={alerts}
|
||||||
transitions={history}
|
transitions={history}
|
||||||
|
alertsHealth={alertsHealth}
|
||||||
|
alertsStreamConnected={alertsStreamConnected}
|
||||||
onRefresh={refreshAlertsAndHistory}
|
onRefresh={refreshAlertsAndHistory}
|
||||||
/>
|
/>
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -339,6 +339,12 @@ export function fetchKnownDevices(): Promise<KnownDevice[]> {
|
|||||||
return request<KnownDevice[]>("/api/v1/control/devices");
|
return request<KnownDevice[]>("/api/v1/control/devices");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function fetchKnownDevice(deviceId: string): Promise<KnownDevice> {
|
||||||
|
return request<KnownDevice>(
|
||||||
|
`/api/v1/control/devices/${encodeURIComponent(deviceId)}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
export function fetchFleetDevices(opts?: {
|
export function fetchFleetDevices(opts?: {
|
||||||
query?: string;
|
query?: string;
|
||||||
presence?: string;
|
presence?: string;
|
||||||
@@ -361,6 +367,12 @@ export function fetchFleetDevices(opts?: {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function fetchFleetDevice(deviceKey: string): Promise<FleetDevice> {
|
||||||
|
return request<FleetDevice>(
|
||||||
|
`/api/v1/control/fleet/devices/${encodeURIComponent(deviceKey)}`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
export function refreshFleetDevices(input?: {
|
export function refreshFleetDevices(input?: {
|
||||||
agentIds?: string[];
|
agentIds?: string[];
|
||||||
timeoutMs?: number;
|
timeoutMs?: number;
|
||||||
@@ -409,6 +421,27 @@ export function createKnownDevice(input: {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function renameKnownDevice(
|
||||||
|
deviceId: string,
|
||||||
|
displayName: string,
|
||||||
|
): Promise<KnownDevice> {
|
||||||
|
return request<KnownDevice>(
|
||||||
|
`/api/v1/control/devices/${encodeURIComponent(deviceId)}`,
|
||||||
|
{
|
||||||
|
method: "PATCH",
|
||||||
|
body: JSON.stringify({ display_name: displayName }),
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function forgetKnownDevice(
|
||||||
|
deviceId: string,
|
||||||
|
): Promise<{ device_id: string; forgotten: boolean }> {
|
||||||
|
return request(`/api/v1/control/devices/${encodeURIComponent(deviceId)}`, {
|
||||||
|
method: "DELETE",
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
export function attachDeviceIdentifier(
|
export function attachDeviceIdentifier(
|
||||||
deviceId: string,
|
deviceId: string,
|
||||||
identifier: { kind: string; value: string },
|
identifier: { kind: string; value: string },
|
||||||
|
|||||||
Binary file not shown.
@@ -0,0 +1,93 @@
|
|||||||
|
Copyright 2020 The JetBrains Mono Project Authors (https://github.com/JetBrains/JetBrainsMono)
|
||||||
|
|
||||||
|
This Font Software is licensed under the SIL Open Font License, Version 1.1.
|
||||||
|
This license is copied below, and is also available with a FAQ at:
|
||||||
|
https://scripts.sil.org/OFL
|
||||||
|
|
||||||
|
|
||||||
|
-----------------------------------------------------------
|
||||||
|
SIL OPEN FONT LICENSE Version 1.1 - 26 February 2007
|
||||||
|
-----------------------------------------------------------
|
||||||
|
|
||||||
|
PREAMBLE
|
||||||
|
The goals of the Open Font License (OFL) are to stimulate worldwide
|
||||||
|
development of collaborative font projects, to support the font creation
|
||||||
|
efforts of academic and linguistic communities, and to provide a free and
|
||||||
|
open framework in which fonts may be shared and improved in partnership
|
||||||
|
with others.
|
||||||
|
|
||||||
|
The OFL allows the licensed fonts to be used, studied, modified and
|
||||||
|
redistributed freely as long as they are not sold by themselves. The
|
||||||
|
fonts, including any derivative works, can be bundled, embedded,
|
||||||
|
redistributed and/or sold with any software provided that any reserved
|
||||||
|
names are not used by derivative works. The fonts and derivatives,
|
||||||
|
however, cannot be released under any other type of license. The
|
||||||
|
requirement for fonts to remain under this license does not apply
|
||||||
|
to any document created using the fonts or their derivatives.
|
||||||
|
|
||||||
|
DEFINITIONS
|
||||||
|
"Font Software" refers to the set of files released by the Copyright
|
||||||
|
Holder(s) under this license and clearly marked as such. This may
|
||||||
|
include source files, build scripts and documentation.
|
||||||
|
|
||||||
|
"Reserved Font Name" refers to any names specified as such after the
|
||||||
|
copyright statement(s).
|
||||||
|
|
||||||
|
"Original Version" refers to the collection of Font Software components as
|
||||||
|
distributed by the Copyright Holder(s).
|
||||||
|
|
||||||
|
"Modified Version" refers to any derivative made by adding to, deleting,
|
||||||
|
or substituting -- in part or in whole -- any of the components of the
|
||||||
|
Original Version, by changing formats or by porting the Font Software to a
|
||||||
|
new environment.
|
||||||
|
|
||||||
|
"Author" refers to any designer, engineer, programmer, technical
|
||||||
|
writer or other person who contributed to the Font Software.
|
||||||
|
|
||||||
|
PERMISSION & CONDITIONS
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining
|
||||||
|
a copy of the Font Software, to use, study, copy, merge, embed, modify,
|
||||||
|
redistribute, and sell modified and unmodified copies of the Font
|
||||||
|
Software, subject to the following conditions:
|
||||||
|
|
||||||
|
1) Neither the Font Software nor any of its individual components,
|
||||||
|
in Original or Modified Versions, may be sold by itself.
|
||||||
|
|
||||||
|
2) Original or Modified Versions of the Font Software may be bundled,
|
||||||
|
redistributed and/or sold with any software, provided that each copy
|
||||||
|
contains the above copyright notice and this license. These can be
|
||||||
|
included either as stand-alone text files, human-readable headers or
|
||||||
|
in the appropriate machine-readable metadata fields within text or
|
||||||
|
binary files as long as those fields can be easily viewed by the user.
|
||||||
|
|
||||||
|
3) No Modified Version of the Font Software may use the Reserved Font
|
||||||
|
Name(s) unless explicit written permission is granted by the corresponding
|
||||||
|
Copyright Holder. This restriction only applies to the primary font name as
|
||||||
|
presented to the users.
|
||||||
|
|
||||||
|
4) The name(s) of the Copyright Holder(s) or the Author(s) of the Font
|
||||||
|
Software shall not be used to promote, endorse or advertise any
|
||||||
|
Modified Version, except to acknowledge the contribution(s) of the
|
||||||
|
Copyright Holder(s) and the Author(s) or with their explicit written
|
||||||
|
permission.
|
||||||
|
|
||||||
|
5) The Font Software, modified or unmodified, in part or in whole,
|
||||||
|
must be distributed entirely under this license, and must not be
|
||||||
|
distributed under any other license. The requirement for fonts to
|
||||||
|
remain under this license does not apply to any document created
|
||||||
|
using the Font Software.
|
||||||
|
|
||||||
|
TERMINATION
|
||||||
|
This license becomes null and void if any of the above conditions are
|
||||||
|
not met.
|
||||||
|
|
||||||
|
DISCLAIMER
|
||||||
|
THE FONT SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
||||||
|
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO ANY WARRANTIES OF
|
||||||
|
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT
|
||||||
|
OF COPYRIGHT, PATENT, TRADEMARK, OR OTHER RIGHT. IN NO EVENT SHALL THE
|
||||||
|
COPYRIGHT HOLDER BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
||||||
|
INCLUDING ANY GENERAL, SPECIAL, INDIRECT, INCIDENTAL, OR CONSEQUENTIAL
|
||||||
|
DAMAGES, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
||||||
|
FROM, OUT OF THE USE OR INABILITY TO USE THE FONT SOFTWARE OR FROM
|
||||||
|
OTHER DEALINGS IN THE FONT SOFTWARE.
|
||||||
@@ -0,0 +1,16 @@
|
|||||||
|
# JetBrainsMono Nerd Font Mono
|
||||||
|
|
||||||
|
`JetBrainsMonoNerdFontMono-Regular.woff2` is the regular monospaced variant
|
||||||
|
from Nerd Fonts v3.4.0 (JetBrains Mono v2.304), converted from the release TTF
|
||||||
|
to WOFF2 without subsetting.
|
||||||
|
|
||||||
|
Source archive:
|
||||||
|
https://github.com/ryanoasis/nerd-fonts/releases/download/v3.4.0/JetBrainsMono.tar.xz
|
||||||
|
|
||||||
|
The font is distributed under the SIL Open Font License in `OFL.txt`.
|
||||||
|
|
||||||
|
Rebuild the checked-in webfont with:
|
||||||
|
|
||||||
|
```sh
|
||||||
|
./scripts/prepare_terminal_font.py
|
||||||
|
```
|
||||||
@@ -16,10 +16,18 @@ import { Textarea } from "@/components/ui/textarea";
|
|||||||
type Props = {
|
type Props = {
|
||||||
alerts: Alert[];
|
alerts: Alert[];
|
||||||
transitions: AlertTransition[];
|
transitions: AlertTransition[];
|
||||||
|
alertsHealth: { available: boolean; reason?: string } | null;
|
||||||
|
alertsStreamConnected: boolean;
|
||||||
onRefresh: () => Promise<void>;
|
onRefresh: () => Promise<void>;
|
||||||
};
|
};
|
||||||
|
|
||||||
export function AlertsPage({ alerts, transitions, onRefresh }: Props) {
|
export function AlertsPage({
|
||||||
|
alerts,
|
||||||
|
transitions,
|
||||||
|
alertsHealth,
|
||||||
|
alertsStreamConnected,
|
||||||
|
onRefresh,
|
||||||
|
}: Props) {
|
||||||
const [severity, setSeverity] = useState("all");
|
const [severity, setSeverity] = useState("all");
|
||||||
const [status, setStatus] = useState("all");
|
const [status, setStatus] = useState("all");
|
||||||
const [kind, setKind] = useState("all");
|
const [kind, setKind] = useState("all");
|
||||||
@@ -66,9 +74,27 @@ export function AlertsPage({ alerts, transitions, onRefresh }: Props) {
|
|||||||
<Card>
|
<Card>
|
||||||
<CardHeader className="flex items-center justify-between gap-2">
|
<CardHeader className="flex items-center justify-between gap-2">
|
||||||
<CardTitle>Active Alerts</CardTitle>
|
<CardTitle>Active Alerts</CardTitle>
|
||||||
<Button size="sm" variant="outline" onClick={() => void onRefresh()}>
|
<div className="flex flex-wrap items-center justify-end gap-x-3 gap-y-1">
|
||||||
|
<div
|
||||||
|
aria-live="polite"
|
||||||
|
className="text-right text-xs text-muted-foreground"
|
||||||
|
>
|
||||||
|
{!alertsStreamConnected && <div>Alert stream disconnected</div>}
|
||||||
|
{alertsHealth?.available === false && (
|
||||||
|
<div>
|
||||||
|
Alert results may be stale or unavailable
|
||||||
|
{alertsHealth.reason ? `: ${alertsHealth.reason}` : ""}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<Button
|
||||||
|
size="sm"
|
||||||
|
variant="outline"
|
||||||
|
onClick={() => void onRefresh()}
|
||||||
|
>
|
||||||
Refresh
|
Refresh
|
||||||
</Button>
|
</Button>
|
||||||
|
</div>
|
||||||
</CardHeader>
|
</CardHeader>
|
||||||
<CardContent className="space-y-3">
|
<CardContent className="space-y-3">
|
||||||
<div className="grid gap-2 sm:grid-cols-3">
|
<div className="grid gap-2 sm:grid-cols-3">
|
||||||
|
|||||||
@@ -27,6 +27,8 @@ type Props = {
|
|||||||
transitions: AlertTransition[];
|
transitions: AlertTransition[];
|
||||||
audit: AuditEvent[];
|
audit: AuditEvent[];
|
||||||
loading: boolean;
|
loading: boolean;
|
||||||
|
alertsHealth: { available: boolean; reason?: string } | null;
|
||||||
|
alertsStreamConnected: boolean;
|
||||||
onRefresh: () => void;
|
onRefresh: () => void;
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -66,6 +68,8 @@ export function DashboardPage({
|
|||||||
transitions,
|
transitions,
|
||||||
audit,
|
audit,
|
||||||
loading,
|
loading,
|
||||||
|
alertsHealth,
|
||||||
|
alertsStreamConnected,
|
||||||
onRefresh,
|
onRefresh,
|
||||||
}: Props) {
|
}: Props) {
|
||||||
const connected = agents.filter((a) => a.connected).length;
|
const connected = agents.filter((a) => a.connected).length;
|
||||||
@@ -165,6 +169,20 @@ export function DashboardPage({
|
|||||||
</Link>
|
</Link>
|
||||||
</CardHeader>
|
</CardHeader>
|
||||||
<CardContent>
|
<CardContent>
|
||||||
|
{(!alertsStreamConnected || alertsHealth?.available === false) && (
|
||||||
|
<div
|
||||||
|
aria-live="polite"
|
||||||
|
className="mb-2 text-xs text-muted-foreground"
|
||||||
|
>
|
||||||
|
{!alertsStreamConnected && <div>Alert stream disconnected</div>}
|
||||||
|
{alertsHealth?.available === false && (
|
||||||
|
<div>
|
||||||
|
Alert results may be stale or unavailable
|
||||||
|
{alertsHealth.reason ? `: ${alertsHealth.reason}` : ""}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
<div className="text-2xl font-bold tabular-nums">{alerts.length}</div>
|
<div className="text-2xl font-bold tabular-nums">{alerts.length}</div>
|
||||||
{Object.keys(severityBuckets).length > 0 ? (
|
{Object.keys(severityBuckets).length > 0 ? (
|
||||||
<div className="mt-2 flex flex-wrap gap-1.5">
|
<div className="mt-2 flex flex-wrap gap-1.5">
|
||||||
|
|||||||
@@ -1,10 +1,12 @@
|
|||||||
import { useEffect, useMemo, useState } from "react";
|
import { useEffect, useMemo, useRef, useState } from "react";
|
||||||
import { Inbox, MoreHorizontal, RefreshCw, Search, Zap } from "lucide-react";
|
import { Inbox, MoreHorizontal, RefreshCw, Search, Zap } from "lucide-react";
|
||||||
import { toast } from "sonner";
|
import { toast } from "sonner";
|
||||||
|
|
||||||
import {
|
import {
|
||||||
fetchFleetDevices,
|
fetchFleetDevices,
|
||||||
|
fetchFleetDevice,
|
||||||
fetchKnownDevices,
|
fetchKnownDevices,
|
||||||
|
fetchKnownDevice,
|
||||||
refreshFleetDevices,
|
refreshFleetDevices,
|
||||||
wakeFleetDevice,
|
wakeFleetDevice,
|
||||||
type Agent,
|
type Agent,
|
||||||
@@ -69,8 +71,21 @@ export function DevicesPage({ agents, onAfterWake, onRefresh }: Props) {
|
|||||||
const [refreshing, setRefreshing] = useState(false);
|
const [refreshing, setRefreshing] = useState(false);
|
||||||
const [wakeBusyKey, setWakeBusyKey] = useState("");
|
const [wakeBusyKey, setWakeBusyKey] = useState("");
|
||||||
const [details, setDetails] = useState<FleetDevice | null>(null);
|
const [details, setDetails] = useState<FleetDevice | null>(null);
|
||||||
|
const loadSeqRef = useRef(0);
|
||||||
|
const detailsKeyRef = useRef<string | null>(null);
|
||||||
|
const detailsSeqRef = useRef(0);
|
||||||
|
|
||||||
|
function selectDetails(device: FleetDevice | null) {
|
||||||
|
// Closing or selecting another row invalidates in-flight detail refreshes.
|
||||||
|
detailsSeqRef.current += 1;
|
||||||
|
detailsKeyRef.current = device?.device_key ?? null;
|
||||||
|
setDetails(device);
|
||||||
|
}
|
||||||
|
|
||||||
async function loadFleet() {
|
async function loadFleet() {
|
||||||
|
// Latest request wins: filter changes and debounced searches overlap, and
|
||||||
|
// an older response must never overwrite fresher state.
|
||||||
|
const seq = ++loadSeqRef.current;
|
||||||
setLoading(true);
|
setLoading(true);
|
||||||
try {
|
try {
|
||||||
const [nextDevices, nextKnownDevices] = await Promise.all([
|
const [nextDevices, nextKnownDevices] = await Promise.all([
|
||||||
@@ -84,12 +99,59 @@ export function DevicesPage({ agents, onAfterWake, onRefresh }: Props) {
|
|||||||
}),
|
}),
|
||||||
fetchKnownDevices(),
|
fetchKnownDevices(),
|
||||||
]);
|
]);
|
||||||
|
if (seq !== loadSeqRef.current) return;
|
||||||
setDevices(nextDevices);
|
setDevices(nextDevices);
|
||||||
setKnownDevices(nextKnownDevices);
|
setKnownDevices(nextKnownDevices);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
|
if (seq !== loadSeqRef.current) return;
|
||||||
toast.error("Failed to load fleet", { description: String(err) });
|
toast.error("Failed to load fleet", { description: String(err) });
|
||||||
} finally {
|
} finally {
|
||||||
setLoading(false);
|
if (seq === loadSeqRef.current) setLoading(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function refreshView(
|
||||||
|
originKey: string | null = detailsKeyRef.current,
|
||||||
|
knownDeviceId?: string,
|
||||||
|
) {
|
||||||
|
if (!originKey || originKey !== detailsKeyRef.current) {
|
||||||
|
await loadFleet();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Mutations supply an explicit identity; never infer it from a shared IP.
|
||||||
|
// Details use their own read, so list filters and limits cannot close them.
|
||||||
|
const key = knownDeviceId ? `known:${knownDeviceId}` : originKey;
|
||||||
|
detailsKeyRef.current = key;
|
||||||
|
const seq = ++detailsSeqRef.current;
|
||||||
|
try {
|
||||||
|
const [device] = await Promise.all([fetchFleetDevice(key), loadFleet()]);
|
||||||
|
if (seq !== detailsSeqRef.current || detailsKeyRef.current !== key)
|
||||||
|
return;
|
||||||
|
const saved = device.known_device
|
||||||
|
? await fetchKnownDevice(device.known_device.device_id)
|
||||||
|
: null;
|
||||||
|
if (seq === detailsSeqRef.current && detailsKeyRef.current === key) {
|
||||||
|
// The dialog's full identity must stay current even if the list refresh
|
||||||
|
// failed. Dropdown options may come from the list, not mutation controls.
|
||||||
|
if (saved) {
|
||||||
|
setKnownDevices((current) =>
|
||||||
|
current.some((item) => item.device_id === saved.device_id)
|
||||||
|
? current.map((item) =>
|
||||||
|
item.device_id === saved.device_id ? saved : item,
|
||||||
|
)
|
||||||
|
: [...current, saved],
|
||||||
|
);
|
||||||
|
}
|
||||||
|
setDetails(device);
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
if (seq !== detailsSeqRef.current || detailsKeyRef.current !== key)
|
||||||
|
return;
|
||||||
|
// A missing/failed identity read must not leave stale mutation controls.
|
||||||
|
selectDetails(null);
|
||||||
|
toast.error("Failed to refresh device details", {
|
||||||
|
description: String(err),
|
||||||
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -102,7 +164,7 @@ export function DevicesPage({ agents, onAfterWake, onRefresh }: Props) {
|
|||||||
? `Refreshed ${result.total_accepted} devices; ${failed.length} agent(s) failed`
|
? `Refreshed ${result.total_accepted} devices; ${failed.length} agent(s) failed`
|
||||||
: `Refreshed ${result.total_accepted} devices from ${result.agents.length} connected agents`;
|
: `Refreshed ${result.total_accepted} devices from ${result.agents.length} connected agents`;
|
||||||
toast.success(msg);
|
toast.success(msg);
|
||||||
await loadFleet();
|
await refreshView();
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
toast.error("Fleet refresh failed", { description: String(err) });
|
toast.error("Fleet refresh failed", { description: String(err) });
|
||||||
} finally {
|
} finally {
|
||||||
@@ -121,7 +183,7 @@ export function DevicesPage({ agents, onAfterWake, onRefresh }: Props) {
|
|||||||
const msg = `Wake sent via ${agentLabel(result.route.agent_id, result.route.nickname)} (${result.route.mac ?? "no mac"})`;
|
const msg = `Wake sent via ${agentLabel(result.route.agent_id, result.route.nickname)} (${result.route.mac ?? "no mac"})`;
|
||||||
toast.success(msg);
|
toast.success(msg);
|
||||||
if (onAfterWake) await onAfterWake();
|
if (onAfterWake) await onAfterWake();
|
||||||
await loadFleet();
|
await refreshView();
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
toast.error("Wake failed", { description: String(err) });
|
toast.error("Wake failed", { description: String(err) });
|
||||||
} finally {
|
} finally {
|
||||||
@@ -152,7 +214,13 @@ export function DevicesPage({ agents, onAfterWake, onRefresh }: Props) {
|
|||||||
// eslint-disable-next-line react-hooks/exhaustive-deps
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
||||||
}, [presence, known, agentId]);
|
}, [presence, known, agentId]);
|
||||||
|
|
||||||
|
// The filter effect above already loads on mount; only react to query edits.
|
||||||
|
const mountedRef = useRef(false);
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
|
if (!mountedRef.current) {
|
||||||
|
mountedRef.current = true;
|
||||||
|
return;
|
||||||
|
}
|
||||||
const timer = window.setTimeout(() => void loadFleet(), 220);
|
const timer = window.setTimeout(() => void loadFleet(), 220);
|
||||||
return () => window.clearTimeout(timer);
|
return () => window.clearTimeout(timer);
|
||||||
// eslint-disable-next-line react-hooks/exhaustive-deps
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
||||||
@@ -251,6 +319,7 @@ export function DevicesPage({ agents, onAfterWake, onRefresh }: Props) {
|
|||||||
label="Identity"
|
label="Identity"
|
||||||
value={known}
|
value={known}
|
||||||
values={knownFilters}
|
values={knownFilters}
|
||||||
|
labels={{ known: "saved", unknown: "unsaved" }}
|
||||||
onChange={(value) => setKnown(value as KnownFilter)}
|
onChange={(value) => setKnown(value as KnownFilter)}
|
||||||
/>
|
/>
|
||||||
<label
|
<label
|
||||||
@@ -326,7 +395,7 @@ export function DevicesPage({ agents, onAfterWake, onRefresh }: Props) {
|
|||||||
device={device}
|
device={device}
|
||||||
busy={wakeBusyKey === device.device_key}
|
busy={wakeBusyKey === device.device_key}
|
||||||
onWake={() => void wakeDevice(device)}
|
onWake={() => void wakeDevice(device)}
|
||||||
onDetails={() => setDetails(device)}
|
onDetails={() => selectDetails(device)}
|
||||||
onCopy={(label, value) => void copyValue(label, value)}
|
onCopy={(label, value) => void copyValue(label, value)}
|
||||||
/>
|
/>
|
||||||
))}
|
))}
|
||||||
@@ -350,13 +419,15 @@ export function DevicesPage({ agents, onAfterWake, onRefresh }: Props) {
|
|||||||
open={Boolean(details)}
|
open={Boolean(details)}
|
||||||
busy={wakeBusyKey === details?.device_key}
|
busy={wakeBusyKey === details?.device_key}
|
||||||
onOpenChange={(open) => {
|
onOpenChange={(open) => {
|
||||||
if (!open) setDetails(null);
|
if (!open) selectDetails(null);
|
||||||
}}
|
}}
|
||||||
onWake={(routeId) => {
|
onWake={(routeId) => {
|
||||||
if (details) void wakeDevice(details, routeId);
|
if (details) void wakeDevice(details, routeId);
|
||||||
}}
|
}}
|
||||||
onCopy={(label, value) => void copyValue(label, value)}
|
onCopy={(label, value) => void copyValue(label, value)}
|
||||||
onChanged={() => void loadFleet()}
|
onChanged={(knownDeviceId) =>
|
||||||
|
refreshView(details?.device_key ?? null, knownDeviceId)
|
||||||
|
}
|
||||||
/>
|
/>
|
||||||
</section>
|
</section>
|
||||||
);
|
);
|
||||||
@@ -392,9 +463,9 @@ function FleetDeviceRow({
|
|||||||
<div className="truncate font-medium">{device.display_name}</div>
|
<div className="truncate font-medium">{device.display_name}</div>
|
||||||
<div className="mt-1 flex flex-wrap gap-1">
|
<div className="mt-1 flex flex-wrap gap-1">
|
||||||
{device.known_device ? (
|
{device.known_device ? (
|
||||||
<Badge variant="secondary">known</Badge>
|
<Badge variant="secondary">saved</Badge>
|
||||||
) : (
|
) : (
|
||||||
<Badge variant="outline">unknown</Badge>
|
<Badge variant="outline">unsaved</Badge>
|
||||||
)}
|
)}
|
||||||
{device.pinned && <Badge variant="outline">pinned</Badge>}
|
{device.pinned && <Badge variant="outline">pinned</Badge>}
|
||||||
{device.sources.map((source) => (
|
{device.sources.map((source) => (
|
||||||
|
|||||||
+177
-689
@@ -1,22 +1,8 @@
|
|||||||
import { useCallback, useEffect, useRef, useState } from "react";
|
import { useEffect, useRef, useState } from "react";
|
||||||
import { ClipboardAddon } from "@xterm/addon-clipboard";
|
import { Eraser, Keyboard, RotateCcw, Terminal } from "lucide-react";
|
||||||
import { FitAddon } from "@xterm/addon-fit";
|
|
||||||
import { Unicode11Addon } from "@xterm/addon-unicode11";
|
|
||||||
import { UnicodeGraphemesAddon } from "@xterm/addon-unicode-graphemes";
|
|
||||||
import { WebLinksAddon } from "@xterm/addon-web-links";
|
|
||||||
import { Terminal as XTerm } from "@xterm/xterm";
|
|
||||||
import { Eraser, Plus, RotateCcw, Terminal, X } from "lucide-react";
|
|
||||||
import { toast } from "sonner";
|
import { toast } from "sonner";
|
||||||
|
|
||||||
import {
|
import type { Agent } from "@/api";
|
||||||
APIError,
|
|
||||||
type Agent,
|
|
||||||
type TerminalSession,
|
|
||||||
attachTerminal,
|
|
||||||
closeTerminal,
|
|
||||||
createTerminal,
|
|
||||||
listTerminals,
|
|
||||||
} from "@/api";
|
|
||||||
import { AgentSelector, displayAgentLabel } from "@/components/AgentSelector";
|
import { AgentSelector, displayAgentLabel } from "@/components/AgentSelector";
|
||||||
import { Badge } from "@/components/ui/badge";
|
import { Badge } from "@/components/ui/badge";
|
||||||
import { Button } from "@/components/ui/button";
|
import { Button } from "@/components/ui/button";
|
||||||
@@ -25,6 +11,11 @@ import {
|
|||||||
TooltipContent,
|
TooltipContent,
|
||||||
TooltipTrigger,
|
TooltipTrigger,
|
||||||
} from "@/components/ui/tooltip";
|
} from "@/components/ui/tooltip";
|
||||||
|
import { TerminalAccessoryBar } from "@/pages/terminal/TerminalAccessoryBar";
|
||||||
|
import { TerminalSessionTabs } from "@/pages/terminal/TerminalSessionTabs";
|
||||||
|
import { useTerminalSession } from "@/pages/terminal/useTerminalSession";
|
||||||
|
import { useTerminalSessions } from "@/pages/terminal/useTerminalSessions";
|
||||||
|
import { useXterm } from "@/pages/terminal/useXterm";
|
||||||
|
|
||||||
type Props = {
|
type Props = {
|
||||||
agents: Agent[];
|
agents: Agent[];
|
||||||
@@ -32,136 +23,48 @@ type Props = {
|
|||||||
onSelectAgent: (agentId: string) => void;
|
onSelectAgent: (agentId: string) => void;
|
||||||
};
|
};
|
||||||
|
|
||||||
type ConnectionState =
|
type TerminalTouch = {
|
||||||
| "idle"
|
pointerId: number;
|
||||||
| "connecting"
|
startX: number;
|
||||||
| "ready"
|
startY: number;
|
||||||
| "disconnected"
|
moved: boolean;
|
||||||
| "exited";
|
};
|
||||||
|
|
||||||
const REMEMBERED_TERMINAL_KEY = "wakey.active-terminal-id";
|
const TERMINAL_TOUCH_SLOP_PX = 8;
|
||||||
const TERMINAL_OPERATOR_KEY = "wakey.terminal-operator-id";
|
|
||||||
|
|
||||||
// polyfill for testing in browsers that don't support crypto.randomUUID()
|
|
||||||
const thing = () =>
|
|
||||||
(String(1e7) + -1e3 + -4e3 + -8e3 + -1e11).replace(/[018]/g, (c: string) => {
|
|
||||||
const num = Number(c);
|
|
||||||
return (
|
|
||||||
num ^
|
|
||||||
(window.crypto.getRandomValues(new Uint8Array(1))[0] & (15 >> (num / 4)))
|
|
||||||
).toString(16);
|
|
||||||
});
|
|
||||||
|
|
||||||
function terminalOperatorId(): string {
|
|
||||||
// Session storage survives route unmounts but remains scoped to this browser
|
|
||||||
// tab. The ID coordinates attachment ownership; API authentication remains
|
|
||||||
// the security boundary.
|
|
||||||
const remembered = window.sessionStorage.getItem(TERMINAL_OPERATOR_KEY);
|
|
||||||
if (remembered) return remembered;
|
|
||||||
const created = window.crypto.randomUUID?.() ?? thing();
|
|
||||||
window.sessionStorage.setItem(TERMINAL_OPERATOR_KEY, created);
|
|
||||||
return created;
|
|
||||||
}
|
|
||||||
|
|
||||||
function mergeTerminalSession(
|
|
||||||
sessions: TerminalSession[],
|
|
||||||
next: TerminalSession,
|
|
||||||
): TerminalSession[] {
|
|
||||||
const existingIndex = sessions.findIndex(
|
|
||||||
(item) => item.terminal_id === next.terminal_id,
|
|
||||||
);
|
|
||||||
if (existingIndex >= 0) {
|
|
||||||
// Attaching updates session state, not identity or tab position. Removing
|
|
||||||
// and re-inserting here made same-second sessions swap places on click.
|
|
||||||
return sessions.map((item, index) =>
|
|
||||||
index === existingIndex ? next : item,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
return orderTerminalSessions([...sessions, next]);
|
|
||||||
}
|
|
||||||
|
|
||||||
function compareTerminalSessions(
|
|
||||||
left: TerminalSession,
|
|
||||||
right: TerminalSession,
|
|
||||||
): number {
|
|
||||||
return (
|
|
||||||
left.created_at_unix - right.created_at_unix ||
|
|
||||||
left.terminal_id.localeCompare(right.terminal_id)
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
function orderTerminalSessions(sessions: TerminalSession[]): TerminalSession[] {
|
|
||||||
return [...sessions].sort(compareTerminalSessions);
|
|
||||||
}
|
|
||||||
|
|
||||||
function reconcileTerminalSessions(
|
|
||||||
current: TerminalSession[],
|
|
||||||
listed: TerminalSession[],
|
|
||||||
): TerminalSession[] {
|
|
||||||
const listedById = new Map(listed.map((item) => [item.terminal_id, item]));
|
|
||||||
const retained = current.flatMap((item) => {
|
|
||||||
const updated = listedById.get(item.terminal_id);
|
|
||||||
if (!updated) return [];
|
|
||||||
listedById.delete(item.terminal_id);
|
|
||||||
return [updated];
|
|
||||||
});
|
|
||||||
return [...retained, ...orderTerminalSessions([...listedById.values()])];
|
|
||||||
}
|
|
||||||
|
|
||||||
function restoreCandidates(
|
|
||||||
sessions: TerminalSession[],
|
|
||||||
rememberedId: string | null,
|
|
||||||
selectedAgentId: string,
|
|
||||||
): TerminalSession[] {
|
|
||||||
const remembered = sessions.find((item) => item.terminal_id === rememberedId);
|
|
||||||
const available = sessions.filter(
|
|
||||||
(item) => !item.operator_attached && item !== remembered,
|
|
||||||
);
|
|
||||||
available.sort((left, right) => {
|
|
||||||
const leftPreferred = left.agent_id === selectedAgentId ? 1 : 0;
|
|
||||||
const rightPreferred = right.agent_id === selectedAgentId ? 1 : 0;
|
|
||||||
return (
|
|
||||||
rightPreferred - leftPreferred ||
|
|
||||||
right.created_at_unix - left.created_at_unix
|
|
||||||
);
|
|
||||||
});
|
|
||||||
return remembered ? [remembered, ...available] : available;
|
|
||||||
}
|
|
||||||
|
|
||||||
function websocketUrl(path: string): string {
|
|
||||||
const scheme = window.location.protocol === "https:" ? "wss:" : "ws:";
|
|
||||||
return `${scheme}//${window.location.host}${path}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
function terminalCreatedTime(createdAtUnix: number): string {
|
|
||||||
return new Date(createdAtUnix * 1000).toLocaleTimeString([], {
|
|
||||||
hour: "2-digit",
|
|
||||||
minute: "2-digit",
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
export function TerminalPage({
|
export function TerminalPage({
|
||||||
agents,
|
agents,
|
||||||
selectedAgentId,
|
selectedAgentId,
|
||||||
onSelectAgent,
|
onSelectAgent,
|
||||||
}: Props) {
|
}: Props) {
|
||||||
|
const pageRef = useRef<HTMLElement>(null);
|
||||||
const hostRef = useRef<HTMLDivElement>(null);
|
const hostRef = useRef<HTMLDivElement>(null);
|
||||||
const terminalRef = useRef<XTerm | null>(null);
|
const terminalTouchRef = useRef<TerminalTouch | null>(null);
|
||||||
const fitRef = useRef<FitAddon | null>(null);
|
const terminalInputFocusedBeforeAccessoryRef = useRef(false);
|
||||||
const socketRef = useRef<WebSocket | null>(null);
|
const [nowUnix, setNowUnix] = useState(() => Math.floor(Date.now() / 1000));
|
||||||
const activeSessionRef = useRef<TerminalSession | null>(null);
|
const [accessoriesOpen, setAccessoriesOpen] = useState(
|
||||||
const selectedAgentIdRef = useRef(selectedAgentId);
|
() => window.matchMedia("(pointer: coarse), (max-width: 640px)").matches,
|
||||||
const lastTerminalSizeRef = useRef({ rows: 0, cols: 0 });
|
|
||||||
const [sessions, setSessions] = useState<TerminalSession[]>([]);
|
|
||||||
const [session, setSession] = useState<TerminalSession | null>(null);
|
|
||||||
const [connection, setConnection] = useState<ConnectionState>("idle");
|
|
||||||
const [sessionTitles, setSessionTitles] = useState<Record<string, string>>(
|
|
||||||
{},
|
|
||||||
);
|
);
|
||||||
const [operatorId] = useState(terminalOperatorId);
|
|
||||||
|
|
||||||
activeSessionRef.current = session;
|
const catalog = useTerminalSessions();
|
||||||
selectedAgentIdRef.current = selectedAgentId;
|
const { sessions } = catalog;
|
||||||
|
const terminal = useXterm({
|
||||||
|
pageRef,
|
||||||
|
hostRef,
|
||||||
|
});
|
||||||
|
const terminalSession = useTerminalSession({
|
||||||
|
terminal: terminal.terminal,
|
||||||
|
catalog,
|
||||||
|
selectedAgentId,
|
||||||
|
});
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const clock = window.setInterval(
|
||||||
|
() => setNowUnix(Math.floor(Date.now() / 1000)),
|
||||||
|
30_000,
|
||||||
|
);
|
||||||
|
return () => window.clearInterval(clock);
|
||||||
|
}, []);
|
||||||
|
|
||||||
const selectedAgent = agents.find(
|
const selectedAgent = agents.find(
|
||||||
(agent) => agent.agent_id === selectedAgentId,
|
(agent) => agent.agent_id === selectedAgentId,
|
||||||
@@ -175,460 +78,55 @@ export function TerminalPage({
|
|||||||
);
|
);
|
||||||
const agentAtSessionLimit =
|
const agentAtSessionLimit =
|
||||||
selectedAgentSessionCount >= selectedAgentSessionLimit;
|
selectedAgentSessionCount >= selectedAgentSessionLimit;
|
||||||
const canRequestStart =
|
const canRequestStart = Boolean(
|
||||||
selectedAgent?.connected &&
|
selectedAgent?.connected &&
|
||||||
selectedAgent.capabilities.includes("terminal") &&
|
selectedAgent.capabilities.includes("terminal") &&
|
||||||
connection !== "connecting";
|
terminalSession.connection !== "connecting" &&
|
||||||
|
terminal.terminal.ready,
|
||||||
|
);
|
||||||
|
const terminalInputReady = terminalSession.connection === "ready";
|
||||||
|
|
||||||
const detachTransport = useCallback(() => {
|
function start() {
|
||||||
const socket = socketRef.current;
|
|
||||||
socketRef.current = null;
|
|
||||||
socket?.close();
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
const sendResize = useCallback(() => {
|
|
||||||
const terminal = terminalRef.current;
|
|
||||||
const socket = socketRef.current;
|
|
||||||
if (!terminal || socket?.readyState !== WebSocket.OPEN) return;
|
|
||||||
if (
|
|
||||||
lastTerminalSizeRef.current.rows === terminal.rows &&
|
|
||||||
lastTerminalSizeRef.current.cols === terminal.cols
|
|
||||||
) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
lastTerminalSizeRef.current = {
|
|
||||||
rows: terminal.rows,
|
|
||||||
cols: terminal.cols,
|
|
||||||
};
|
|
||||||
socket.send(
|
|
||||||
JSON.stringify({
|
|
||||||
type: "resize",
|
|
||||||
rows: terminal.rows,
|
|
||||||
cols: terminal.cols,
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
}, []);
|
|
||||||
|
|
||||||
const connect = useCallback(
|
|
||||||
(nextSession: TerminalSession) => {
|
|
||||||
if (!nextSession.attachment_token) {
|
|
||||||
throw new Error(
|
|
||||||
"Control plane did not issue a terminal attachment token",
|
|
||||||
);
|
|
||||||
}
|
|
||||||
detachTransport();
|
|
||||||
setSession(nextSession);
|
|
||||||
activeSessionRef.current = nextSession;
|
|
||||||
setSessions((current) =>
|
|
||||||
mergeTerminalSession(current, {
|
|
||||||
...nextSession,
|
|
||||||
operator_attached: true,
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
window.sessionStorage.setItem(
|
|
||||||
REMEMBERED_TERMINAL_KEY,
|
|
||||||
nextSession.terminal_id,
|
|
||||||
);
|
|
||||||
setConnection("connecting");
|
|
||||||
const socket = new WebSocket(websocketUrl(nextSession.websocket_url));
|
|
||||||
socket.binaryType = "arraybuffer";
|
|
||||||
socketRef.current = socket;
|
|
||||||
|
|
||||||
socket.onopen = () => {
|
|
||||||
socket.send(
|
|
||||||
JSON.stringify({
|
|
||||||
type: "attach",
|
|
||||||
attachment_token: nextSession.attachment_token,
|
|
||||||
operator_id: operatorId,
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
fitRef.current?.fit();
|
|
||||||
sendResize();
|
|
||||||
};
|
|
||||||
socket.onmessage = (event) => {
|
|
||||||
if (socketRef.current !== socket) return;
|
|
||||||
if (typeof event.data !== "string") {
|
|
||||||
terminalRef.current?.write(new Uint8Array(event.data as ArrayBuffer));
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
try {
|
|
||||||
const control = JSON.parse(event.data) as {
|
|
||||||
type: string;
|
|
||||||
exit_code?: number | null;
|
|
||||||
message?: string;
|
|
||||||
};
|
|
||||||
if (control.type === "ready") {
|
|
||||||
setConnection("ready");
|
|
||||||
window.requestAnimationFrame(() => {
|
|
||||||
fitRef.current?.fit();
|
|
||||||
lastTerminalSizeRef.current = { rows: 0, cols: 0 };
|
|
||||||
sendResize();
|
|
||||||
terminalRef.current?.focus();
|
|
||||||
});
|
|
||||||
} else if (control.type === "exited") {
|
|
||||||
setConnection("exited");
|
|
||||||
terminalRef.current?.writeln(
|
|
||||||
`\r\n[process exited${control.exit_code == null ? "" : ` ${control.exit_code}`}]`,
|
|
||||||
);
|
|
||||||
} else if (control.type === "error") {
|
|
||||||
setConnection("exited");
|
|
||||||
terminalRef.current?.writeln(
|
|
||||||
`\r\n[terminal error: ${control.message ?? "unknown error"}]`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
} catch {
|
|
||||||
terminalRef.current?.writeln("\r\n[invalid terminal control frame]");
|
|
||||||
}
|
|
||||||
};
|
|
||||||
socket.onerror = () => {
|
|
||||||
if (socketRef.current !== socket) return;
|
|
||||||
terminalRef.current?.writeln("\r\n[terminal transport error]");
|
|
||||||
};
|
|
||||||
socket.onclose = () => {
|
|
||||||
if (socketRef.current !== socket) return;
|
|
||||||
socketRef.current = null;
|
|
||||||
setSessions((current) =>
|
|
||||||
current.map((item) =>
|
|
||||||
item.terminal_id === nextSession.terminal_id
|
|
||||||
? { ...item, operator_attached: false }
|
|
||||||
: item,
|
|
||||||
),
|
|
||||||
);
|
|
||||||
setConnection((current) => {
|
|
||||||
if (current === "exited") return current;
|
|
||||||
return "disconnected";
|
|
||||||
});
|
|
||||||
};
|
|
||||||
},
|
|
||||||
[detachTransport, operatorId, sendResize],
|
|
||||||
);
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
if (!hostRef.current) return;
|
|
||||||
const terminal = new XTerm({
|
|
||||||
cursorBlink: true,
|
|
||||||
convertEol: false,
|
|
||||||
fontFamily: '"SFMono-Regular", Consolas, "Liberation Mono", monospace',
|
|
||||||
fontSize: 14,
|
|
||||||
lineHeight: 1.1,
|
|
||||||
// xterm uses this width for its internal scrollbar as well as the
|
|
||||||
// overview ruler. Reserve a larger touch target on coarse pointers.
|
|
||||||
overviewRuler: {
|
|
||||||
width: window.matchMedia("(pointer: coarse)").matches ? 22 : 16,
|
|
||||||
},
|
|
||||||
scrollback: 5000,
|
|
||||||
theme: {
|
|
||||||
background: "#0b1117",
|
|
||||||
foreground: "#e5e7eb",
|
|
||||||
cursor: "#6ee7a8",
|
|
||||||
scrollbarSliderBackground: "#52617099",
|
|
||||||
scrollbarSliderHoverBackground: "#6b7c8dcc",
|
|
||||||
scrollbarSliderActiveBackground: "#8294a6",
|
|
||||||
},
|
|
||||||
allowProposedApi: true, // this for unicode11 addon. FOR SOME reason xtermjs doesnt tell me this in the readme.
|
|
||||||
});
|
|
||||||
terminal.loadAddon(new Unicode11Addon());
|
|
||||||
terminal.unicode.activeVersion = "11";
|
|
||||||
terminal.loadAddon(new UnicodeGraphemesAddon());
|
|
||||||
terminal.loadAddon(new WebLinksAddon());
|
|
||||||
const fit = new FitAddon();
|
|
||||||
terminal.loadAddon(new ClipboardAddon());
|
|
||||||
terminal.loadAddon(fit);
|
|
||||||
terminal.open(hostRef.current);
|
|
||||||
terminal.attachCustomKeyEventHandler((event) => {
|
|
||||||
const copiesTerminalSelection =
|
|
||||||
event.type === "keydown" &&
|
|
||||||
event.ctrlKey &&
|
|
||||||
event.shiftKey &&
|
|
||||||
event.code === "KeyC";
|
|
||||||
if (!copiesTerminalSelection) return true;
|
|
||||||
|
|
||||||
// Chromium reserves Ctrl+Shift+C for DevTools. Other clipboard shortcuts
|
|
||||||
// fall through to xterm's native copy/paste event handlers.
|
|
||||||
event.preventDefault();
|
|
||||||
event.stopPropagation();
|
|
||||||
if (terminal.hasSelection()) {
|
|
||||||
if (!navigator.clipboard) {
|
|
||||||
toast.error("Clipboard access requires HTTPS or localhost");
|
|
||||||
} else {
|
|
||||||
void navigator.clipboard
|
|
||||||
.writeText(terminal.getSelection())
|
|
||||||
.catch((error) =>
|
|
||||||
toast.error("Clipboard access was denied", {
|
|
||||||
description: String(error),
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false;
|
|
||||||
});
|
|
||||||
const titleChange = terminal.onTitleChange((title) => {
|
|
||||||
const terminalId = activeSessionRef.current?.terminal_id;
|
|
||||||
if (!terminalId) return;
|
|
||||||
const normalized = title.trim();
|
|
||||||
setSessionTitles((current) => {
|
|
||||||
if (current[terminalId] === normalized) return current;
|
|
||||||
if (!normalized) {
|
|
||||||
const next = { ...current };
|
|
||||||
delete next[terminalId];
|
|
||||||
return next;
|
|
||||||
}
|
|
||||||
return { ...current, [terminalId]: normalized };
|
|
||||||
});
|
|
||||||
});
|
|
||||||
fit.fit();
|
|
||||||
terminalRef.current = terminal;
|
|
||||||
fitRef.current = fit;
|
|
||||||
|
|
||||||
async function restoreTerminalSession() {
|
|
||||||
try {
|
|
||||||
const listed = await listTerminals();
|
|
||||||
if (cancelled) return;
|
|
||||||
setSessions(orderTerminalSessions(listed));
|
|
||||||
const rememberedId = window.sessionStorage.getItem(
|
|
||||||
REMEMBERED_TERMINAL_KEY,
|
|
||||||
);
|
|
||||||
const candidates = restoreCandidates(
|
|
||||||
listed,
|
|
||||||
rememberedId,
|
|
||||||
selectedAgentIdRef.current,
|
|
||||||
);
|
|
||||||
|
|
||||||
for (const candidate of candidates) {
|
|
||||||
setConnection("connecting");
|
|
||||||
try {
|
|
||||||
const attached = await attachTerminal(
|
|
||||||
candidate.terminal_id,
|
|
||||||
operatorId,
|
|
||||||
);
|
|
||||||
if (cancelled) return;
|
|
||||||
terminal.reset();
|
|
||||||
connect(attached);
|
|
||||||
return;
|
|
||||||
} catch (error) {
|
|
||||||
if (
|
|
||||||
!(error instanceof APIError) ||
|
|
||||||
error.code !== "terminal_operator_already_attached"
|
|
||||||
) {
|
|
||||||
throw error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
setConnection("idle");
|
|
||||||
} catch (error) {
|
|
||||||
if (cancelled) return;
|
|
||||||
setConnection("idle");
|
|
||||||
toast.error("Could not restore terminal session", {
|
|
||||||
description: String(error),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let cancelled = false;
|
|
||||||
void document.fonts.ready.then(() => {
|
|
||||||
if (cancelled) return;
|
|
||||||
fit.fit();
|
|
||||||
lastTerminalSizeRef.current = { rows: 0, cols: 0 };
|
|
||||||
sendResize();
|
|
||||||
});
|
|
||||||
|
|
||||||
const input = terminal.onData((data) => {
|
|
||||||
const socket = socketRef.current;
|
|
||||||
if (socket?.readyState === WebSocket.OPEN) {
|
|
||||||
socket.send(new TextEncoder().encode(data));
|
|
||||||
}
|
|
||||||
});
|
|
||||||
let resizeFrame = 0;
|
|
||||||
let lastHostWidth = 0;
|
|
||||||
let lastHostHeight = 0;
|
|
||||||
const resizeObserver = new ResizeObserver(([entry]) => {
|
|
||||||
const width = Math.round(entry.contentRect.width);
|
|
||||||
const height = Math.round(entry.contentRect.height);
|
|
||||||
if (width === lastHostWidth && height === lastHostHeight) return;
|
|
||||||
lastHostWidth = width;
|
|
||||||
lastHostHeight = height;
|
|
||||||
window.cancelAnimationFrame(resizeFrame);
|
|
||||||
resizeFrame = window.requestAnimationFrame(() => {
|
|
||||||
fit.fit();
|
|
||||||
sendResize();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
resizeObserver.observe(hostRef.current);
|
|
||||||
void restoreTerminalSession();
|
|
||||||
const sessionRefresh = window.setInterval(() => {
|
|
||||||
void listTerminals()
|
|
||||||
.then((listed) => {
|
|
||||||
if (!cancelled) {
|
|
||||||
setSessions((current) =>
|
|
||||||
reconcileTerminalSessions(current, listed),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.catch(() => {
|
|
||||||
// The attached terminal transport remains authoritative while a
|
|
||||||
// background list refresh is temporarily unavailable.
|
|
||||||
});
|
|
||||||
}, 5000);
|
|
||||||
|
|
||||||
return () => {
|
|
||||||
cancelled = true;
|
|
||||||
input.dispose();
|
|
||||||
titleChange.dispose();
|
|
||||||
resizeObserver.disconnect();
|
|
||||||
window.cancelAnimationFrame(resizeFrame);
|
|
||||||
window.clearInterval(sessionRefresh);
|
|
||||||
detachTransport();
|
|
||||||
terminal.dispose();
|
|
||||||
terminalRef.current = null;
|
|
||||||
fitRef.current = null;
|
|
||||||
};
|
|
||||||
}, [connect, detachTransport, sendResize]);
|
|
||||||
|
|
||||||
async function start() {
|
|
||||||
if (!selectedAgentId || !terminalRef.current) return;
|
|
||||||
if (agentAtSessionLimit) {
|
if (agentAtSessionLimit) {
|
||||||
toast.error("Terminal session limit reached", {
|
toast.error("Terminal session limit reached", {
|
||||||
description: `${selectedAgent ? displayAgentLabel(selectedAgent) : selectedAgentId} already reached the active session limit. Close one before opening another.`,
|
description: `${selectedAgent ? displayAgentLabel(selectedAgent) : selectedAgentId} already reached the active session limit. Close one before opening another.`,
|
||||||
});
|
});
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
const previousSession = activeSessionRef.current;
|
void terminalSession.start();
|
||||||
detachTransport();
|
|
||||||
if (previousSession) {
|
|
||||||
setSessions((current) =>
|
|
||||||
current.map((item) =>
|
|
||||||
item.terminal_id === previousSession.terminal_id
|
|
||||||
? { ...item, operator_attached: false }
|
|
||||||
: item,
|
|
||||||
),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
setConnection("connecting");
|
|
||||||
terminalRef.current.reset();
|
|
||||||
try {
|
|
||||||
fitRef.current?.fit();
|
|
||||||
const created = await createTerminal(
|
|
||||||
selectedAgentId,
|
|
||||||
terminalRef.current.rows,
|
|
||||||
terminalRef.current.cols,
|
|
||||||
);
|
|
||||||
connect(created);
|
|
||||||
} catch (error) {
|
|
||||||
setSession(null);
|
|
||||||
activeSessionRef.current = null;
|
|
||||||
setConnection("idle");
|
|
||||||
toast.error("Could not start terminal", { description: String(error) });
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async function activateSession(nextSession: TerminalSession) {
|
function toggleTerminalKeyboard() {
|
||||||
if (nextSession.terminal_id === session?.terminal_id) return;
|
const terminalInstance = terminal.terminal;
|
||||||
if (nextSession.operator_attached) return;
|
const blurTerminalInput = () => {
|
||||||
|
terminalInstance.blurInput();
|
||||||
const previousSession = activeSessionRef.current;
|
const activeElement = document.activeElement;
|
||||||
detachTransport();
|
if (
|
||||||
if (previousSession) {
|
activeElement instanceof HTMLElement &&
|
||||||
setSessions((current) =>
|
hostRef.current?.contains(activeElement)
|
||||||
current.map((item) =>
|
) {
|
||||||
item.terminal_id === previousSession.terminal_id
|
activeElement.blur();
|
||||||
? { ...item, operator_attached: false }
|
|
||||||
: item,
|
|
||||||
),
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
setSession(null);
|
};
|
||||||
activeSessionRef.current = null;
|
const shouldHide =
|
||||||
setConnection("connecting");
|
terminalInputFocusedBeforeAccessoryRef.current ||
|
||||||
terminalRef.current?.reset();
|
Boolean(hostRef.current?.contains(document.activeElement));
|
||||||
window.sessionStorage.setItem(
|
terminalInputFocusedBeforeAccessoryRef.current = false;
|
||||||
REMEMBERED_TERMINAL_KEY,
|
if (shouldHide) {
|
||||||
nextSession.terminal_id,
|
blurTerminalInput();
|
||||||
);
|
// Base UI may restore the previously focused element after pointer handling.
|
||||||
|
window.requestAnimationFrame(blurTerminalInput);
|
||||||
try {
|
} else {
|
||||||
const attached = await attachTerminal(
|
terminalInstance.focus();
|
||||||
nextSession.terminal_id,
|
|
||||||
operatorId,
|
|
||||||
);
|
|
||||||
connect(attached);
|
|
||||||
} catch (error) {
|
|
||||||
setConnection("idle");
|
|
||||||
toast.error("Could not attach terminal session", {
|
|
||||||
description: String(error),
|
|
||||||
});
|
|
||||||
void listTerminals()
|
|
||||||
.then((listed) =>
|
|
||||||
setSessions((current) => reconcileTerminalSessions(current, listed)),
|
|
||||||
)
|
|
||||||
.catch(() => undefined);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function reconnect() {
|
|
||||||
if (!session) return;
|
|
||||||
detachTransport();
|
|
||||||
try {
|
|
||||||
terminalRef.current?.reset();
|
|
||||||
setConnection("connecting");
|
|
||||||
const attached = await attachTerminal(session.terminal_id, operatorId);
|
|
||||||
connect(attached);
|
|
||||||
} catch (error) {
|
|
||||||
setConnection("disconnected");
|
|
||||||
toast.error("Terminal session is no longer available", {
|
|
||||||
description: String(error),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
async function closeSession(closingSession: TerminalSession) {
|
|
||||||
const closingId = closingSession.terminal_id;
|
|
||||||
const closesActiveSession =
|
|
||||||
closingId === activeSessionRef.current?.terminal_id;
|
|
||||||
const remaining = sessions.filter((item) => item.terminal_id !== closingId);
|
|
||||||
const fallback = remaining.find((item) => !item.operator_attached);
|
|
||||||
|
|
||||||
if (closesActiveSession) {
|
|
||||||
if (socketRef.current?.readyState === WebSocket.OPEN) {
|
|
||||||
socketRef.current.send(JSON.stringify({ type: "close" }));
|
|
||||||
}
|
|
||||||
detachTransport();
|
|
||||||
}
|
|
||||||
try {
|
|
||||||
await closeTerminal(closingId);
|
|
||||||
} catch (error) {
|
|
||||||
toast.error("Terminal cleanup failed", { description: String(error) });
|
|
||||||
void listTerminals()
|
|
||||||
.then((listed) =>
|
|
||||||
setSessions((current) => reconcileTerminalSessions(current, listed)),
|
|
||||||
)
|
|
||||||
.catch(() => undefined);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setSessions((current) =>
|
|
||||||
current.filter((item) => item.terminal_id !== closingId),
|
|
||||||
);
|
|
||||||
setSessionTitles((current) => {
|
|
||||||
const next = { ...current };
|
|
||||||
delete next[closingId];
|
|
||||||
return next;
|
|
||||||
});
|
|
||||||
if (window.sessionStorage.getItem(REMEMBERED_TERMINAL_KEY) === closingId) {
|
|
||||||
window.sessionStorage.removeItem(REMEMBERED_TERMINAL_KEY);
|
|
||||||
}
|
|
||||||
if (closesActiveSession) {
|
|
||||||
setSession(null);
|
|
||||||
activeSessionRef.current = null;
|
|
||||||
setConnection("idle");
|
|
||||||
// xterm.clear() deliberately preserves the active cursor line. Closing
|
|
||||||
// a session should discard its complete screen and terminal modes.
|
|
||||||
terminalRef.current?.reset();
|
|
||||||
if (fallback) void activateSession(fallback);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<section className="terminal-page" aria-label="Remote terminal">
|
<section
|
||||||
|
className="terminal-page"
|
||||||
|
aria-label="Remote terminal"
|
||||||
|
ref={pageRef}
|
||||||
|
>
|
||||||
<header className="terminal-toolbar">
|
<header className="terminal-toolbar">
|
||||||
<div className="terminal-heading">
|
<div className="terminal-heading">
|
||||||
<Terminal className="size-5 text-primary" aria-hidden />
|
<Terminal className="size-5 text-primary" aria-hidden />
|
||||||
@@ -643,13 +141,13 @@ export function TerminalPage({
|
|||||||
)}
|
)}
|
||||||
value={selectedAgentId}
|
value={selectedAgentId}
|
||||||
onChange={onSelectAgent}
|
onChange={onSelectAgent}
|
||||||
disabled={connection === "connecting"}
|
disabled={terminalSession.connection === "connecting"}
|
||||||
className="w-full min-w-0 sm:w-64"
|
className="w-full min-w-0 sm:w-64"
|
||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
</header>
|
</header>
|
||||||
|
|
||||||
{!selectedAgent && connection === "idle" ? (
|
{!selectedAgent && terminalSession.connection === "idle" ? (
|
||||||
<p className="terminal-notice">
|
<p className="terminal-notice">
|
||||||
No connected terminal-capable agent is selected.
|
No connected terminal-capable agent is selected.
|
||||||
</p>
|
</p>
|
||||||
@@ -661,128 +159,36 @@ export function TerminalPage({
|
|||||||
|
|
||||||
<div className="terminal-frame">
|
<div className="terminal-frame">
|
||||||
<div className="terminal-framebar">
|
<div className="terminal-framebar">
|
||||||
<div className="terminal-tabs">
|
<TerminalSessionTabs
|
||||||
<div
|
agents={agents}
|
||||||
className="terminal-tab-list"
|
sessions={sessions}
|
||||||
role="tablist"
|
activeTerminalId={terminalSession.session?.terminal_id}
|
||||||
aria-label="Terminal sessions"
|
connection={terminalSession.connection}
|
||||||
>
|
sessionTitles={terminal.sessionTitles}
|
||||||
{sessions.length === 0 ? (
|
nowUnix={nowUnix}
|
||||||
<div className="terminal-session-label">
|
canRequestStart={canRequestStart}
|
||||||
<Terminal className="size-4" aria-hidden />
|
agentAtSessionLimit={agentAtSessionLimit}
|
||||||
<span>No active session</span>
|
selectedAgentSessionLimit={selectedAgentSessionLimit}
|
||||||
</div>
|
onActivate={(item) => void terminalSession.activate(item)}
|
||||||
) : (
|
onClose={(item) => void terminalSession.close(item)}
|
||||||
sessions.map((item) => {
|
onStart={start}
|
||||||
const agent = agents.find(
|
|
||||||
(candidate) => candidate.agent_id === item.agent_id,
|
|
||||||
);
|
|
||||||
const active = item.terminal_id === session?.terminal_id;
|
|
||||||
const locked = item.operator_attached && !active;
|
|
||||||
const agentLabel = agent
|
|
||||||
? displayAgentLabel(agent)
|
|
||||||
: item.agent_id;
|
|
||||||
const sessionTitle = sessionTitles[item.terminal_id];
|
|
||||||
const tabLabel = sessionTitle
|
|
||||||
? `${agentLabel} · ${sessionTitle}`
|
|
||||||
: agentLabel;
|
|
||||||
return (
|
|
||||||
<div
|
|
||||||
key={item.terminal_id}
|
|
||||||
role="presentation"
|
|
||||||
className="terminal-tab"
|
|
||||||
data-active={active || undefined}
|
|
||||||
data-locked={locked || undefined}
|
|
||||||
>
|
|
||||||
<button
|
|
||||||
type="button"
|
|
||||||
role="tab"
|
|
||||||
aria-selected={active}
|
|
||||||
className="terminal-tab-select"
|
|
||||||
disabled={locked || connection === "connecting"}
|
|
||||||
title={
|
|
||||||
locked ? "Attached in another browser" : tabLabel
|
|
||||||
}
|
|
||||||
onClick={() => void activateSession(item)}
|
|
||||||
>
|
|
||||||
<span
|
|
||||||
className="terminal-tab-dot"
|
|
||||||
data-state={
|
|
||||||
active
|
|
||||||
? connection
|
|
||||||
: locked
|
|
||||||
? "attached"
|
|
||||||
: item.agent_attached
|
|
||||||
? "detached"
|
|
||||||
: "disconnected"
|
|
||||||
}
|
|
||||||
aria-hidden
|
|
||||||
/>
|
/>
|
||||||
<span className="terminal-tab-label">{tabLabel}</span>
|
|
||||||
<time
|
|
||||||
dateTime={new Date(
|
|
||||||
item.created_at_unix * 1000,
|
|
||||||
).toISOString()}
|
|
||||||
>
|
|
||||||
{terminalCreatedTime(item.created_at_unix)}
|
|
||||||
</time>
|
|
||||||
</button>
|
|
||||||
<button
|
|
||||||
type="button"
|
|
||||||
className="terminal-tab-close"
|
|
||||||
disabled={locked}
|
|
||||||
aria-label={`Close ${agentLabel} terminal session`}
|
|
||||||
title={
|
|
||||||
locked
|
|
||||||
? "Attached in another browser"
|
|
||||||
: "Close session"
|
|
||||||
}
|
|
||||||
onClick={() => void closeSession(item)}
|
|
||||||
>
|
|
||||||
<X className="size-3.5" aria-hidden />
|
|
||||||
</button>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
})
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
<Tooltip>
|
|
||||||
<TooltipTrigger>
|
|
||||||
<Button
|
|
||||||
type="button"
|
|
||||||
size="icon"
|
|
||||||
variant="ghost"
|
|
||||||
className="terminal-new-tab"
|
|
||||||
disabled={!canRequestStart}
|
|
||||||
aria-label="New terminal session"
|
|
||||||
onClick={start}
|
|
||||||
>
|
|
||||||
<Plus className="size-4" aria-hidden />
|
|
||||||
</Button>
|
|
||||||
</TooltipTrigger>
|
|
||||||
<TooltipContent>
|
|
||||||
{agentAtSessionLimit
|
|
||||||
? `${selectedAgentSessionLimit}-session limit reached`
|
|
||||||
: "New session"}
|
|
||||||
</TooltipContent>
|
|
||||||
</Tooltip>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className="terminal-frame-actions">
|
<div className="terminal-frame-actions">
|
||||||
<Badge
|
<Badge
|
||||||
variant="outline"
|
variant="outline"
|
||||||
className="terminal-status"
|
className="terminal-status"
|
||||||
data-state={connection}
|
data-state={terminalSession.connection}
|
||||||
>
|
>
|
||||||
<span aria-hidden />
|
<span aria-hidden />
|
||||||
{connection}
|
{terminalSession.connection}
|
||||||
</Badge>
|
</Badge>
|
||||||
{connection === "disconnected" ? (
|
{terminalSession.connection === "disconnected" ? (
|
||||||
<Button
|
<Button
|
||||||
type="button"
|
type="button"
|
||||||
size="sm"
|
size="sm"
|
||||||
variant="ghost"
|
variant="ghost"
|
||||||
onClick={reconnect}
|
onClick={() => void terminalSession.reconnect()}
|
||||||
>
|
>
|
||||||
<RotateCcw className="size-4" aria-hidden />
|
<RotateCcw className="size-4" aria-hidden />
|
||||||
Reconnect
|
Reconnect
|
||||||
@@ -794,9 +200,30 @@ export function TerminalPage({
|
|||||||
type="button"
|
type="button"
|
||||||
size="icon"
|
size="icon"
|
||||||
variant="ghost"
|
variant="ghost"
|
||||||
disabled={connection === "idle"}
|
className="terminal-accessories-toggle"
|
||||||
|
data-active={accessoriesOpen || undefined}
|
||||||
|
aria-label={
|
||||||
|
accessoriesOpen
|
||||||
|
? "Hide terminal controls"
|
||||||
|
: "Show terminal controls"
|
||||||
|
}
|
||||||
|
aria-expanded={accessoriesOpen}
|
||||||
|
onClick={() => setAccessoriesOpen((current) => !current)}
|
||||||
|
>
|
||||||
|
<Keyboard className="size-4" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
</TooltipTrigger>
|
||||||
|
<TooltipContent>Terminal controls</TooltipContent>
|
||||||
|
</Tooltip>
|
||||||
|
<Tooltip>
|
||||||
|
<TooltipTrigger>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="icon"
|
||||||
|
variant="ghost"
|
||||||
|
disabled={terminalSession.connection === "idle"}
|
||||||
aria-label="Clear terminal"
|
aria-label="Clear terminal"
|
||||||
onClick={() => terminalRef.current?.clear()}
|
onClick={terminal.terminal.clear}
|
||||||
>
|
>
|
||||||
<Eraser className="size-4" aria-hidden />
|
<Eraser className="size-4" aria-hidden />
|
||||||
</Button>
|
</Button>
|
||||||
@@ -805,7 +232,68 @@ export function TerminalPage({
|
|||||||
</Tooltip>
|
</Tooltip>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
<div className="terminal-surface" ref={hostRef} />
|
{accessoriesOpen ? (
|
||||||
|
<TerminalAccessoryBar
|
||||||
|
inputReady={terminalInputReady}
|
||||||
|
modifiers={terminalSession.modifiers}
|
||||||
|
fontSize={terminal.fontSize}
|
||||||
|
onRememberInputFocus={() => {
|
||||||
|
terminalInputFocusedBeforeAccessoryRef.current = Boolean(
|
||||||
|
hostRef.current?.contains(document.activeElement),
|
||||||
|
);
|
||||||
|
}}
|
||||||
|
onToggleModifier={terminalSession.toggleModifier}
|
||||||
|
onWriteKey={terminalSession.writeAccessoryKey}
|
||||||
|
onScrollPages={terminal.terminal.scrollPages}
|
||||||
|
onCopy={() => void terminal.terminal.copy()}
|
||||||
|
onPaste={() =>
|
||||||
|
void terminal.terminal.paste(terminalSession.writeTerminalData)
|
||||||
|
}
|
||||||
|
onToggleKeyboard={toggleTerminalKeyboard}
|
||||||
|
onChangeFontSize={terminal.terminal.changeFontSize}
|
||||||
|
/>
|
||||||
|
) : null}
|
||||||
|
<div
|
||||||
|
className="terminal-surface"
|
||||||
|
ref={hostRef}
|
||||||
|
onPointerDown={(event) => {
|
||||||
|
if (event.pointerType !== "touch") {
|
||||||
|
terminal.terminal.focus();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
terminalTouchRef.current = {
|
||||||
|
pointerId: event.pointerId,
|
||||||
|
startX: event.clientX,
|
||||||
|
startY: event.clientY,
|
||||||
|
moved: false,
|
||||||
|
};
|
||||||
|
}}
|
||||||
|
onPointerMove={(event) => {
|
||||||
|
const touch = terminalTouchRef.current;
|
||||||
|
if (!touch || touch.pointerId !== event.pointerId || touch.moved) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
touch.moved =
|
||||||
|
Math.hypot(
|
||||||
|
event.clientX - touch.startX,
|
||||||
|
event.clientY - touch.startY,
|
||||||
|
) > TERMINAL_TOUCH_SLOP_PX;
|
||||||
|
}}
|
||||||
|
onPointerUp={(event) => {
|
||||||
|
const touch = terminalTouchRef.current;
|
||||||
|
terminalTouchRef.current = null;
|
||||||
|
if (
|
||||||
|
touch?.pointerId === event.pointerId &&
|
||||||
|
!touch.moved &&
|
||||||
|
event.pointerType === "touch"
|
||||||
|
) {
|
||||||
|
terminal.terminal.focus();
|
||||||
|
}
|
||||||
|
}}
|
||||||
|
onPointerCancel={() => {
|
||||||
|
terminalTouchRef.current = null;
|
||||||
|
}}
|
||||||
|
/>
|
||||||
</div>
|
</div>
|
||||||
</section>
|
</section>
|
||||||
);
|
);
|
||||||
|
|||||||
@@ -27,11 +27,13 @@ export function FilterSelect({
|
|||||||
label,
|
label,
|
||||||
value,
|
value,
|
||||||
values,
|
values,
|
||||||
|
labels,
|
||||||
onChange,
|
onChange,
|
||||||
}: {
|
}: {
|
||||||
label: string;
|
label: string;
|
||||||
value: string;
|
value: string;
|
||||||
values: readonly string[];
|
values: readonly string[];
|
||||||
|
labels?: Readonly<Record<string, string>>;
|
||||||
onChange: (value: string) => void;
|
onChange: (value: string) => void;
|
||||||
}) {
|
}) {
|
||||||
return (
|
return (
|
||||||
@@ -39,12 +41,12 @@ export function FilterSelect({
|
|||||||
<span>{label}</span>
|
<span>{label}</span>
|
||||||
<Select value={value} onValueChange={(next) => next && onChange(next)}>
|
<Select value={value} onValueChange={(next) => next && onChange(next)}>
|
||||||
<SelectTrigger>
|
<SelectTrigger>
|
||||||
<span>{value.replace("_", " ")}</span>
|
<span>{labels?.[value] ?? value.replace("_", " ")}</span>
|
||||||
</SelectTrigger>
|
</SelectTrigger>
|
||||||
<SelectContent alignItemWithTrigger={false}>
|
<SelectContent alignItemWithTrigger={false}>
|
||||||
{values.map((item) => (
|
{values.map((item) => (
|
||||||
<SelectItem key={item} value={item}>
|
<SelectItem key={item} value={item}>
|
||||||
{item.replace("_", " ")}
|
{labels?.[item] ?? item.replace("_", " ")}
|
||||||
</SelectItem>
|
</SelectItem>
|
||||||
))}
|
))}
|
||||||
</SelectContent>
|
</SelectContent>
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import { useMemo, useState } from "react";
|
import { useMemo, useState } from "react";
|
||||||
import { Copy, Minus, Plus, Zap } from "lucide-react";
|
import { Copy, Minus, Plus, Zap, Pencil, Trash2, Check, X } from "lucide-react";
|
||||||
import { toast } from "sonner";
|
import { toast } from "sonner";
|
||||||
|
|
||||||
import {
|
import {
|
||||||
@@ -7,6 +7,8 @@ import {
|
|||||||
createKnownDevice,
|
createKnownDevice,
|
||||||
detachDeviceIdentifier,
|
detachDeviceIdentifier,
|
||||||
mergeKnownDevice,
|
mergeKnownDevice,
|
||||||
|
renameKnownDevice,
|
||||||
|
forgetKnownDevice,
|
||||||
type FleetDevice,
|
type FleetDevice,
|
||||||
type FleetDeviceEndpoint,
|
type FleetDeviceEndpoint,
|
||||||
type KnownDevice,
|
type KnownDevice,
|
||||||
@@ -46,7 +48,7 @@ type Props = {
|
|||||||
onOpenChange: (open: boolean) => void;
|
onOpenChange: (open: boolean) => void;
|
||||||
onWake: (routeId?: string | null) => void;
|
onWake: (routeId?: string | null) => void;
|
||||||
onCopy: (label: string, value: string) => void;
|
onCopy: (label: string, value: string) => void;
|
||||||
onChanged: () => void;
|
onChanged: (knownDeviceId?: string) => Promise<void>;
|
||||||
};
|
};
|
||||||
|
|
||||||
export function FleetDeviceDetailsDialog({
|
export function FleetDeviceDetailsDialog({
|
||||||
@@ -68,6 +70,8 @@ export function FleetDeviceDetailsDialog({
|
|||||||
const [addValue, setAddValue] = useState("");
|
const [addValue, setAddValue] = useState("");
|
||||||
const [error, setError] = useState("");
|
const [error, setError] = useState("");
|
||||||
const [actionBusy, setActionBusy] = useState(false);
|
const [actionBusy, setActionBusy] = useState(false);
|
||||||
|
const [editingName, setEditingName] = useState(false);
|
||||||
|
const [forgetConfirm, setForgetConfirm] = useState(false);
|
||||||
|
|
||||||
const rowIdentifiers = device ? identifiersFor(device) : [];
|
const rowIdentifiers = device ? identifiersFor(device) : [];
|
||||||
|
|
||||||
@@ -95,18 +99,20 @@ export function FleetDeviceDetailsDialog({
|
|||||||
(kd) => kd.device_id !== currentDevice.known_device?.device_id,
|
(kd) => kd.device_id !== currentDevice.known_device?.device_id,
|
||||||
);
|
);
|
||||||
|
|
||||||
async function wrap(fn: () => Promise<void | { changed?: boolean }>) {
|
async function wrap(
|
||||||
|
fn: () => Promise<void | { changed?: boolean; knownDeviceId?: string }>,
|
||||||
|
) {
|
||||||
setError("");
|
setError("");
|
||||||
setActionBusy(true);
|
setActionBusy(true);
|
||||||
try {
|
try {
|
||||||
const res = await fn();
|
const res = await fn();
|
||||||
if (!res || res.changed !== false) {
|
if (!res || res.changed !== false) {
|
||||||
onChanged();
|
await onChanged(res?.knownDeviceId);
|
||||||
}
|
}
|
||||||
} catch (err: any) {
|
} catch (err: any) {
|
||||||
setError(String(err));
|
setError(String(err));
|
||||||
if (err && (err.changed || err.partial)) {
|
if (err && (err.changed || err.partial)) {
|
||||||
onChanged();
|
await onChanged(err.knownDeviceId);
|
||||||
}
|
}
|
||||||
} finally {
|
} finally {
|
||||||
setActionBusy(false);
|
setActionBusy(false);
|
||||||
@@ -121,6 +127,7 @@ export function FleetDeviceDetailsDialog({
|
|||||||
identifiers: rowIdentifiers,
|
identifiers: rowIdentifiers,
|
||||||
});
|
});
|
||||||
toast.success(`Remembered as "${result.display_name}"`);
|
toast.success(`Remembered as "${result.display_name}"`);
|
||||||
|
return { knownDeviceId: result.device_id };
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -129,7 +136,7 @@ export function FleetDeviceDetailsDialog({
|
|||||||
const target = knownDevices.find((kd) => kd.device_id === targetDeviceId);
|
const target = knownDevices.find((kd) => kd.device_id === targetDeviceId);
|
||||||
await wrap(async () => {
|
await wrap(async () => {
|
||||||
const ids = fullKnown ? unattachedIdentifiers : rowIdentifiers;
|
const ids = fullKnown ? unattachedIdentifiers : rowIdentifiers;
|
||||||
if (ids.length === 0) return;
|
if (ids.length === 0) return { changed: false };
|
||||||
|
|
||||||
const results = await Promise.allSettled(
|
const results = await Promise.allSettled(
|
||||||
ids.map((id) => attachDeviceIdentifier(targetDeviceId, id)),
|
ids.map((id) => attachDeviceIdentifier(targetDeviceId, id)),
|
||||||
@@ -140,14 +147,17 @@ export function FleetDeviceDetailsDialog({
|
|||||||
|
|
||||||
if (errors.length) {
|
if (errors.length) {
|
||||||
const err = errors[0].reason as Error;
|
const err = errors[0].reason as Error;
|
||||||
if (changed) (err as any).changed = true;
|
if (changed) {
|
||||||
|
(err as any).changed = true;
|
||||||
|
(err as any).knownDeviceId = targetDeviceId;
|
||||||
|
}
|
||||||
throw err;
|
throw err;
|
||||||
}
|
}
|
||||||
|
|
||||||
toast.success(
|
toast.success(
|
||||||
`Attached ${ids.length} identifier(s) to "${target?.display_name ?? targetDeviceId}"`,
|
`Attached ${ids.length} identifier(s) to "${target?.display_name ?? targetDeviceId}"`,
|
||||||
);
|
);
|
||||||
return { changed };
|
return { changed, knownDeviceId: targetDeviceId };
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -155,12 +165,12 @@ export function FleetDeviceDetailsDialog({
|
|||||||
if (!targetDeviceId || !currentDevice.known_device) return;
|
if (!targetDeviceId || !currentDevice.known_device) return;
|
||||||
const target = knownDevices.find((kd) => kd.device_id === targetDeviceId);
|
const target = knownDevices.find((kd) => kd.device_id === targetDeviceId);
|
||||||
await wrap(async () => {
|
await wrap(async () => {
|
||||||
await mergeKnownDevice(
|
const merged = await mergeKnownDevice(
|
||||||
targetDeviceId,
|
targetDeviceId,
|
||||||
currentDevice.known_device!.device_id,
|
currentDevice.known_device!.device_id,
|
||||||
);
|
);
|
||||||
toast.success(`Merged into "${target?.display_name ?? targetDeviceId}"`);
|
toast.success(`Merged into "${target?.display_name ?? targetDeviceId}"`);
|
||||||
onOpenChange(false);
|
return { knownDeviceId: merged.device_id };
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -184,6 +194,24 @@ export function FleetDeviceDetailsDialog({
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function handleRename() {
|
||||||
|
if (!fullKnown || !displayName.trim()) return;
|
||||||
|
await wrap(async () => {
|
||||||
|
await renameKnownDevice(fullKnown.device_id, displayName.trim());
|
||||||
|
setEditingName(false);
|
||||||
|
toast.success("Device renamed");
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleForget() {
|
||||||
|
if (!fullKnown) return;
|
||||||
|
await wrap(async () => {
|
||||||
|
await forgetKnownDevice(fullKnown.device_id);
|
||||||
|
toast.success("Remembered device forgotten");
|
||||||
|
onOpenChange(false);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
const selectedRoute =
|
const selectedRoute =
|
||||||
device.route_candidates.find((r) => r.route_id === routeId) ??
|
device.route_candidates.find((r) => r.route_id === routeId) ??
|
||||||
device.recommended_route;
|
device.recommended_route;
|
||||||
@@ -275,7 +303,7 @@ export function FleetDeviceDetailsDialog({
|
|||||||
<Button
|
<Button
|
||||||
size="sm"
|
size="sm"
|
||||||
onClick={() => onWake(selectedRoute?.route_id ?? null)}
|
onClick={() => onWake(selectedRoute?.route_id ?? null)}
|
||||||
disabled={busy || !selectedRoute?.wakeable}
|
disabled={actionBusy || busy || !selectedRoute?.wakeable}
|
||||||
>
|
>
|
||||||
<Zap className="size-4" aria-hidden />
|
<Zap className="size-4" aria-hidden />
|
||||||
{busy ? "Waking" : "Wake"}
|
{busy ? "Waking" : "Wake"}
|
||||||
@@ -309,9 +337,36 @@ export function FleetDeviceDetailsDialog({
|
|||||||
{fullKnown && (
|
{fullKnown && (
|
||||||
<div className="grid gap-3">
|
<div className="grid gap-3">
|
||||||
<div className="flex flex-wrap items-center gap-2 text-sm">
|
<div className="flex flex-wrap items-center gap-2 text-sm">
|
||||||
<Badge variant="secondary">known</Badge>
|
<Badge variant="secondary">saved</Badge>
|
||||||
{fullKnown.pinned && <Badge variant="outline">pinned</Badge>}
|
{fullKnown.pinned && <Badge variant="outline">pinned</Badge>}
|
||||||
<span className="font-medium">{fullKnown.display_name}</span>
|
<span className="font-medium">{fullKnown.display_name}</span>
|
||||||
|
<Button
|
||||||
|
variant="ghost"
|
||||||
|
size="icon-sm"
|
||||||
|
title="Rename device"
|
||||||
|
aria-label="Rename device"
|
||||||
|
disabled={actionBusy}
|
||||||
|
onClick={() => {
|
||||||
|
setDisplayName(fullKnown.display_name);
|
||||||
|
setEditingName(true);
|
||||||
|
setForgetConfirm(false);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<Pencil className="size-3.5" />
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
variant="ghost"
|
||||||
|
size="icon-sm"
|
||||||
|
title="Forget remembered device"
|
||||||
|
aria-label="Forget remembered device"
|
||||||
|
disabled={actionBusy}
|
||||||
|
onClick={() => {
|
||||||
|
setForgetConfirm(true);
|
||||||
|
setEditingName(false);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<Trash2 className="size-3.5" />
|
||||||
|
</Button>
|
||||||
{fullKnown.notes && (
|
{fullKnown.notes && (
|
||||||
<span className="text-muted-foreground">
|
<span className="text-muted-foreground">
|
||||||
: {fullKnown.notes}
|
: {fullKnown.notes}
|
||||||
@@ -319,6 +374,76 @@ export function FleetDeviceDetailsDialog({
|
|||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
|
{editingName && (
|
||||||
|
<form
|
||||||
|
className="flex items-center gap-2"
|
||||||
|
onSubmit={(event) => {
|
||||||
|
event.preventDefault();
|
||||||
|
void handleRename();
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<Input
|
||||||
|
aria-label="Device name"
|
||||||
|
autoFocus
|
||||||
|
value={displayName}
|
||||||
|
disabled={actionBusy}
|
||||||
|
onChange={(event) => setDisplayName(event.target.value)}
|
||||||
|
/>
|
||||||
|
<Button
|
||||||
|
type="submit"
|
||||||
|
size="icon-sm"
|
||||||
|
title="Save name"
|
||||||
|
aria-label="Save name"
|
||||||
|
disabled={actionBusy || !displayName.trim()}
|
||||||
|
>
|
||||||
|
<Check className="size-4" />
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
variant="ghost"
|
||||||
|
size="icon-sm"
|
||||||
|
title="Cancel rename"
|
||||||
|
aria-label="Cancel rename"
|
||||||
|
disabled={actionBusy}
|
||||||
|
onClick={() => setEditingName(false)}
|
||||||
|
>
|
||||||
|
<X className="size-4" />
|
||||||
|
</Button>
|
||||||
|
</form>
|
||||||
|
)}
|
||||||
|
{forgetConfirm && (
|
||||||
|
<div
|
||||||
|
className="grid gap-2"
|
||||||
|
role="group"
|
||||||
|
aria-label="Confirm forget device"
|
||||||
|
>
|
||||||
|
<p className="text-sm">
|
||||||
|
Forget “{fullKnown.display_name}” and its remembered
|
||||||
|
identifiers? Agent observations remain; it may reappear as
|
||||||
|
an unsaved device.
|
||||||
|
</p>
|
||||||
|
<div className="flex flex-wrap gap-2">
|
||||||
|
<Button
|
||||||
|
variant="destructive"
|
||||||
|
size="sm"
|
||||||
|
disabled={actionBusy}
|
||||||
|
onClick={() => void handleForget()}
|
||||||
|
>
|
||||||
|
<Trash2 className="size-4" />
|
||||||
|
Forget device
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
variant="outline"
|
||||||
|
size="sm"
|
||||||
|
disabled={actionBusy}
|
||||||
|
onClick={() => setForgetConfirm(false)}
|
||||||
|
>
|
||||||
|
Cancel
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
{/* Current identifiers with remove buttons */}
|
{/* Current identifiers with remove buttons */}
|
||||||
<div className="rounded-md border bg-muted/30 p-3">
|
<div className="rounded-md border bg-muted/30 p-3">
|
||||||
<div className="mb-2 text-xs font-medium uppercase text-muted-foreground">
|
<div className="mb-2 text-xs font-medium uppercase text-muted-foreground">
|
||||||
@@ -434,7 +559,7 @@ export function FleetDeviceDetailsDialog({
|
|||||||
<Separator />
|
<Separator />
|
||||||
<div className="grid gap-1.5">
|
<div className="grid gap-1.5">
|
||||||
<p className="text-xs text-muted-foreground">
|
<p className="text-xs text-muted-foreground">
|
||||||
Merge this known device into another:
|
Merge this saved device into another:
|
||||||
</p>
|
</p>
|
||||||
<div className="grid grid-cols-[minmax(0,1fr)_auto] gap-2">
|
<div className="grid grid-cols-[minmax(0,1fr)_auto] gap-2">
|
||||||
<Select
|
<Select
|
||||||
@@ -479,7 +604,7 @@ export function FleetDeviceDetailsDialog({
|
|||||||
{/* Remember as new known device */}
|
{/* Remember as new known device */}
|
||||||
<div className="grid gap-1.5">
|
<div className="grid gap-1.5">
|
||||||
<p className="text-xs text-muted-foreground">
|
<p className="text-xs text-muted-foreground">
|
||||||
Remember as a new known device:
|
Remember as a new saved device:
|
||||||
</p>
|
</p>
|
||||||
<div className="grid grid-cols-[minmax(0,1fr)_auto] gap-2">
|
<div className="grid grid-cols-[minmax(0,1fr)_auto] gap-2">
|
||||||
<Input
|
<Input
|
||||||
@@ -505,7 +630,7 @@ export function FleetDeviceDetailsDialog({
|
|||||||
{/* Attach to existing known device */}
|
{/* Attach to existing known device */}
|
||||||
<div className="grid gap-1.5">
|
<div className="grid gap-1.5">
|
||||||
<p className="text-xs text-muted-foreground">
|
<p className="text-xs text-muted-foreground">
|
||||||
Or attach to an existing known device:
|
Or attach to an existing saved device:
|
||||||
</p>
|
</p>
|
||||||
<div className="grid grid-cols-[minmax(0,1fr)_auto] gap-2">
|
<div className="grid grid-cols-[minmax(0,1fr)_auto] gap-2">
|
||||||
<Select
|
<Select
|
||||||
@@ -518,7 +643,7 @@ export function FleetDeviceDetailsDialog({
|
|||||||
? knownDevices.find(
|
? knownDevices.find(
|
||||||
(kd) => kd.device_id === targetDeviceId,
|
(kd) => kd.device_id === targetDeviceId,
|
||||||
)?.display_name
|
)?.display_name
|
||||||
: "Select known device"}
|
: "Select saved device"}
|
||||||
</span>
|
</span>
|
||||||
</SelectTrigger>
|
</SelectTrigger>
|
||||||
<SelectContent
|
<SelectContent
|
||||||
|
|||||||
@@ -0,0 +1,259 @@
|
|||||||
|
import {
|
||||||
|
ArrowDown,
|
||||||
|
ArrowLeft,
|
||||||
|
ArrowRight,
|
||||||
|
ArrowUp,
|
||||||
|
ClipboardCopy,
|
||||||
|
ClipboardPaste,
|
||||||
|
Keyboard,
|
||||||
|
ZoomIn,
|
||||||
|
ZoomOut,
|
||||||
|
} from "lucide-react";
|
||||||
|
import type { LucideIcon } from "lucide-react";
|
||||||
|
import type { MouseEvent, PointerEvent } from "react";
|
||||||
|
|
||||||
|
import { Button } from "@/components/ui/button";
|
||||||
|
import type { TerminalModifiers } from "@/pages/terminal/sessionUtils";
|
||||||
|
|
||||||
|
export const MIN_TERMINAL_FONT_SIZE = 10;
|
||||||
|
export const MAX_TERMINAL_FONT_SIZE = 22;
|
||||||
|
|
||||||
|
type Props = {
|
||||||
|
inputReady: boolean;
|
||||||
|
modifiers: TerminalModifiers;
|
||||||
|
fontSize: number;
|
||||||
|
onRememberInputFocus: () => void;
|
||||||
|
onToggleModifier: (modifier: keyof TerminalModifiers) => void;
|
||||||
|
onWriteKey: (data: string) => void;
|
||||||
|
onScrollPages: (pages: number) => void;
|
||||||
|
onCopy: () => void;
|
||||||
|
onPaste: () => void;
|
||||||
|
onToggleKeyboard: () => void;
|
||||||
|
onChangeFontSize: (delta: number) => void;
|
||||||
|
};
|
||||||
|
|
||||||
|
const CONTROL_KEYS: ReadonlyArray<
|
||||||
|
readonly [label: string, sequence: string, title: string]
|
||||||
|
> = [
|
||||||
|
["^C", "\x03", "Interrupt"],
|
||||||
|
["^U", "\x15", "Clear input line"],
|
||||||
|
["^A", "\x01", "Start of line"],
|
||||||
|
["^E", "\x05", "End of line"],
|
||||||
|
];
|
||||||
|
|
||||||
|
const ARROW_KEYS: ReadonlyArray<
|
||||||
|
readonly [label: string, sequence: string, icon: LucideIcon]
|
||||||
|
> = [
|
||||||
|
["Left arrow", "\x1b[D", ArrowLeft],
|
||||||
|
["Down arrow", "\x1b[B", ArrowDown],
|
||||||
|
["Up arrow", "\x1b[A", ArrowUp],
|
||||||
|
["Right arrow", "\x1b[C", ArrowRight],
|
||||||
|
];
|
||||||
|
|
||||||
|
function preserveInputFocus(
|
||||||
|
event: PointerEvent<HTMLDivElement> | MouseEvent<HTMLDivElement>,
|
||||||
|
onRememberInputFocus: () => void,
|
||||||
|
) {
|
||||||
|
if (event.target instanceof Element && event.target.closest("button")) {
|
||||||
|
onRememberInputFocus();
|
||||||
|
event.preventDefault();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export function TerminalAccessoryBar({
|
||||||
|
inputReady,
|
||||||
|
modifiers,
|
||||||
|
fontSize,
|
||||||
|
onRememberInputFocus,
|
||||||
|
onToggleModifier,
|
||||||
|
onWriteKey,
|
||||||
|
onScrollPages,
|
||||||
|
onCopy,
|
||||||
|
onPaste,
|
||||||
|
onToggleKeyboard,
|
||||||
|
onChangeFontSize,
|
||||||
|
}: Props) {
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
className="terminal-accessories"
|
||||||
|
role="toolbar"
|
||||||
|
aria-label="Terminal controls"
|
||||||
|
onPointerDownCapture={(event) => {
|
||||||
|
// Preserve whether xterm was focused before the button library moves
|
||||||
|
// focus. The keyboard button consumes this to show or hide mobile IME.
|
||||||
|
preserveInputFocus(event, onRememberInputFocus);
|
||||||
|
}}
|
||||||
|
onMouseDownCapture={(event) => {
|
||||||
|
// Firefox can take the mouse-event path without emitting pointerdown.
|
||||||
|
preserveInputFocus(event, onRememberInputFocus);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
<div className="terminal-accessory-group">
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="sm"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key"
|
||||||
|
data-active={modifiers.ctrl || undefined}
|
||||||
|
aria-pressed={modifiers.ctrl}
|
||||||
|
disabled={!inputReady}
|
||||||
|
onClick={() => onToggleModifier("ctrl")}
|
||||||
|
>
|
||||||
|
Ctrl
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="sm"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key"
|
||||||
|
data-active={modifiers.meta || undefined}
|
||||||
|
aria-pressed={modifiers.meta}
|
||||||
|
disabled={!inputReady}
|
||||||
|
onClick={() => onToggleModifier("meta")}
|
||||||
|
>
|
||||||
|
Meta
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="sm"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key"
|
||||||
|
disabled={!inputReady}
|
||||||
|
onClick={() => onWriteKey("\x1b")}
|
||||||
|
>
|
||||||
|
Esc
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="sm"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key"
|
||||||
|
disabled={!inputReady}
|
||||||
|
onClick={() => onWriteKey("\t")}
|
||||||
|
>
|
||||||
|
Tab
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="terminal-accessory-group">
|
||||||
|
{CONTROL_KEYS.map(([label, sequence, title]) => (
|
||||||
|
<Button
|
||||||
|
key={label}
|
||||||
|
type="button"
|
||||||
|
size="sm"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--control"
|
||||||
|
aria-label={title}
|
||||||
|
title={title}
|
||||||
|
disabled={!inputReady}
|
||||||
|
onClick={() => onWriteKey(sequence)}
|
||||||
|
>
|
||||||
|
{label}
|
||||||
|
</Button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="terminal-accessory-group">
|
||||||
|
{ARROW_KEYS.map(([label, sequence, Icon]) => (
|
||||||
|
<Button
|
||||||
|
key={label}
|
||||||
|
type="button"
|
||||||
|
size="icon"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--icon"
|
||||||
|
aria-label={label}
|
||||||
|
disabled={!inputReady}
|
||||||
|
onClick={() => onWriteKey(sequence)}
|
||||||
|
>
|
||||||
|
<Icon className="size-4" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="terminal-accessory-group">
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="sm"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--wide"
|
||||||
|
onClick={() => onScrollPages(-1)}
|
||||||
|
>
|
||||||
|
PgUp
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="sm"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--wide"
|
||||||
|
onClick={() => onScrollPages(1)}
|
||||||
|
>
|
||||||
|
PgDn
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="terminal-accessory-group">
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="icon"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--icon"
|
||||||
|
aria-label="Copy selection or visible screen"
|
||||||
|
title="Copy selection or visible screen"
|
||||||
|
onClick={onCopy}
|
||||||
|
>
|
||||||
|
<ClipboardCopy className="size-4" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="icon"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--icon"
|
||||||
|
aria-label="Paste"
|
||||||
|
title="Paste"
|
||||||
|
disabled={!inputReady}
|
||||||
|
onClick={onPaste}
|
||||||
|
>
|
||||||
|
<ClipboardPaste className="size-4" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="icon"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--icon"
|
||||||
|
aria-label="Toggle terminal keyboard"
|
||||||
|
title="Toggle terminal keyboard"
|
||||||
|
disabled={!inputReady}
|
||||||
|
onClick={onToggleKeyboard}
|
||||||
|
>
|
||||||
|
<Keyboard className="size-4" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="terminal-accessory-group terminal-font-controls">
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="icon"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--icon"
|
||||||
|
disabled={fontSize <= MIN_TERMINAL_FONT_SIZE}
|
||||||
|
aria-label="Decrease terminal text size"
|
||||||
|
onClick={() => onChangeFontSize(-1)}
|
||||||
|
>
|
||||||
|
<ZoomOut className="size-4" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
<output aria-label="Terminal text size">{fontSize}</output>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="icon"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-key terminal-key--icon"
|
||||||
|
disabled={fontSize >= MAX_TERMINAL_FONT_SIZE}
|
||||||
|
aria-label="Increase terminal text size"
|
||||||
|
onClick={() => onChangeFontSize(1)}
|
||||||
|
>
|
||||||
|
<ZoomIn className="size-4" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -0,0 +1,193 @@
|
|||||||
|
import { Plus, Terminal, X } from "lucide-react";
|
||||||
|
|
||||||
|
import type { Agent, TerminalSession } from "@/api";
|
||||||
|
import { displayAgentLabel } from "@/components/AgentSelector";
|
||||||
|
import { Button } from "@/components/ui/button";
|
||||||
|
import {
|
||||||
|
Tooltip,
|
||||||
|
TooltipContent,
|
||||||
|
TooltipTrigger,
|
||||||
|
} from "@/components/ui/tooltip";
|
||||||
|
import type { TerminalConnectionState } from "@/pages/terminal/sessionUtils";
|
||||||
|
|
||||||
|
type Props = {
|
||||||
|
agents: Agent[];
|
||||||
|
sessions: TerminalSession[];
|
||||||
|
activeTerminalId?: string;
|
||||||
|
connection: TerminalConnectionState;
|
||||||
|
sessionTitles: Record<string, string>;
|
||||||
|
nowUnix: number;
|
||||||
|
canRequestStart: boolean;
|
||||||
|
agentAtSessionLimit: boolean;
|
||||||
|
selectedAgentSessionLimit: number;
|
||||||
|
onActivate: (session: TerminalSession) => void;
|
||||||
|
onClose: (session: TerminalSession) => void;
|
||||||
|
onStart: () => void;
|
||||||
|
};
|
||||||
|
|
||||||
|
function terminalCreatedTime(createdAtUnix: number): string {
|
||||||
|
return new Date(createdAtUnix * 1000).toLocaleTimeString([], {
|
||||||
|
hour: "2-digit",
|
||||||
|
minute: "2-digit",
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function terminalTimestamp(timestampUnix: number): string {
|
||||||
|
return new Date(timestampUnix * 1000).toLocaleString();
|
||||||
|
}
|
||||||
|
|
||||||
|
function terminalExpiryDisplay(
|
||||||
|
expiresAtUnix: number | null | undefined,
|
||||||
|
nowUnix: number,
|
||||||
|
): { label: string; warning: boolean } | null {
|
||||||
|
if (expiresAtUnix == null) return null;
|
||||||
|
const remainingSeconds = expiresAtUnix - nowUnix;
|
||||||
|
if (remainingSeconds <= 0) return { label: "expired", warning: true };
|
||||||
|
if (remainingSeconds < 60) return { label: "<1m left", warning: true };
|
||||||
|
if (remainingSeconds < 3600) {
|
||||||
|
return {
|
||||||
|
label: `${Math.floor(remainingSeconds / 60)}m left`,
|
||||||
|
warning: remainingSeconds < 10 * 60,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if (remainingSeconds < 24 * 3600) {
|
||||||
|
return {
|
||||||
|
label: `${Math.floor(remainingSeconds / 3600)}h left`,
|
||||||
|
warning: false,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
label: `${Math.floor(remainingSeconds / (24 * 3600))}d left`,
|
||||||
|
warning: false,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
export function TerminalSessionTabs({
|
||||||
|
agents,
|
||||||
|
sessions,
|
||||||
|
activeTerminalId,
|
||||||
|
connection,
|
||||||
|
sessionTitles,
|
||||||
|
nowUnix,
|
||||||
|
canRequestStart,
|
||||||
|
agentAtSessionLimit,
|
||||||
|
selectedAgentSessionLimit,
|
||||||
|
onActivate,
|
||||||
|
onClose,
|
||||||
|
onStart,
|
||||||
|
}: Props) {
|
||||||
|
return (
|
||||||
|
<div className="terminal-tabs">
|
||||||
|
<div
|
||||||
|
className="terminal-tab-list"
|
||||||
|
role="tablist"
|
||||||
|
aria-label="Terminal sessions"
|
||||||
|
>
|
||||||
|
{sessions.length === 0 ? (
|
||||||
|
<div className="terminal-session-label">
|
||||||
|
<Terminal className="size-4" aria-hidden />
|
||||||
|
<span>No active session</span>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
sessions.map((item) => {
|
||||||
|
const agent = agents.find(
|
||||||
|
(candidate) => candidate.agent_id === item.agent_id,
|
||||||
|
);
|
||||||
|
const active = item.terminal_id === activeTerminalId;
|
||||||
|
const locked = item.operator_attached && !active;
|
||||||
|
const agentLabel = agent ? displayAgentLabel(agent) : item.agent_id;
|
||||||
|
const sessionTitle = sessionTitles[item.terminal_id];
|
||||||
|
const tabLabel = sessionTitle
|
||||||
|
? `${agentLabel} · ${sessionTitle}`
|
||||||
|
: agentLabel;
|
||||||
|
const expiry = terminalExpiryDisplay(item.expires_at_unix, nowUnix);
|
||||||
|
const timingTitle =
|
||||||
|
item.expires_at_unix != null
|
||||||
|
? `Expires ${terminalTimestamp(item.expires_at_unix)}`
|
||||||
|
: "No expiry";
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
key={item.terminal_id}
|
||||||
|
role="presentation"
|
||||||
|
className="terminal-tab"
|
||||||
|
data-active={active || undefined}
|
||||||
|
data-locked={locked || undefined}
|
||||||
|
>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
role="tab"
|
||||||
|
aria-selected={active}
|
||||||
|
className="terminal-tab-select"
|
||||||
|
disabled={locked || connection === "connecting"}
|
||||||
|
title={`${locked ? "Attached in another browser\n" : ""}${tabLabel}\nCreated ${terminalTimestamp(item.created_at_unix)}\n${timingTitle}`}
|
||||||
|
onClick={() => onActivate(item)}
|
||||||
|
>
|
||||||
|
<span
|
||||||
|
className="terminal-tab-dot"
|
||||||
|
data-state={
|
||||||
|
active
|
||||||
|
? connection
|
||||||
|
: locked
|
||||||
|
? "attached"
|
||||||
|
: item.agent_attached
|
||||||
|
? "detached"
|
||||||
|
: "disconnected"
|
||||||
|
}
|
||||||
|
aria-hidden
|
||||||
|
/>
|
||||||
|
<span className="terminal-tab-label">{tabLabel}</span>
|
||||||
|
<time
|
||||||
|
className={
|
||||||
|
expiry?.warning ? "text-amber-400" : "text-[#7f8d9a]"
|
||||||
|
}
|
||||||
|
dateTime={new Date(
|
||||||
|
(item.expires_at_unix ?? item.created_at_unix) * 1000,
|
||||||
|
).toISOString()}
|
||||||
|
>
|
||||||
|
{expiry?.label ?? terminalCreatedTime(item.created_at_unix)}
|
||||||
|
</time>
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className="terminal-tab-close"
|
||||||
|
disabled={locked || connection === "connecting"}
|
||||||
|
aria-label={`Close ${agentLabel} terminal session`}
|
||||||
|
title={
|
||||||
|
locked
|
||||||
|
? "Attached in another browser"
|
||||||
|
: connection === "connecting"
|
||||||
|
? "Connecting — wait before closing"
|
||||||
|
: "Close session"
|
||||||
|
}
|
||||||
|
onClick={() => onClose(item)}
|
||||||
|
>
|
||||||
|
<X className="size-3.5" aria-hidden />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
})
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<Tooltip>
|
||||||
|
<TooltipTrigger>
|
||||||
|
<Button
|
||||||
|
type="button"
|
||||||
|
size="icon"
|
||||||
|
variant="ghost"
|
||||||
|
className="terminal-new-tab"
|
||||||
|
disabled={!canRequestStart}
|
||||||
|
aria-label="New terminal session"
|
||||||
|
onClick={onStart}
|
||||||
|
>
|
||||||
|
<Plus className="size-4" aria-hidden />
|
||||||
|
</Button>
|
||||||
|
</TooltipTrigger>
|
||||||
|
<TooltipContent>
|
||||||
|
{agentAtSessionLimit
|
||||||
|
? `${selectedAgentSessionLimit}-session limit reached`
|
||||||
|
: "New session"}
|
||||||
|
</TooltipContent>
|
||||||
|
</Tooltip>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -0,0 +1,113 @@
|
|||||||
|
import type { Terminal as XTerm } from "@xterm/xterm";
|
||||||
|
|
||||||
|
import type { TerminalSession } from "@/api";
|
||||||
|
|
||||||
|
export type TerminalConnectionState =
|
||||||
|
| "idle"
|
||||||
|
| "connecting"
|
||||||
|
| "ready"
|
||||||
|
| "disconnected"
|
||||||
|
| "exited";
|
||||||
|
|
||||||
|
export type TerminalModifiers = {
|
||||||
|
ctrl: boolean;
|
||||||
|
meta: boolean;
|
||||||
|
};
|
||||||
|
|
||||||
|
export const NO_TERMINAL_MODIFIERS: TerminalModifiers = {
|
||||||
|
ctrl: false,
|
||||||
|
meta: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
export function mergeTerminalSession(
|
||||||
|
sessions: TerminalSession[],
|
||||||
|
next: TerminalSession,
|
||||||
|
): TerminalSession[] {
|
||||||
|
const existingIndex = sessions.findIndex(
|
||||||
|
(item) => item.terminal_id === next.terminal_id,
|
||||||
|
);
|
||||||
|
if (existingIndex >= 0) {
|
||||||
|
// Attaching updates session state, not identity or tab position. Removing
|
||||||
|
// and re-inserting here made same-second sessions swap places on click.
|
||||||
|
return sessions.map((item, index) =>
|
||||||
|
index === existingIndex ? next : item,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return orderTerminalSessions([...sessions, next]);
|
||||||
|
}
|
||||||
|
|
||||||
|
function compareTerminalSessions(
|
||||||
|
left: TerminalSession,
|
||||||
|
right: TerminalSession,
|
||||||
|
): number {
|
||||||
|
return (
|
||||||
|
left.created_at_unix - right.created_at_unix ||
|
||||||
|
left.terminal_id.localeCompare(right.terminal_id)
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function orderTerminalSessions(
|
||||||
|
sessions: TerminalSession[],
|
||||||
|
): TerminalSession[] {
|
||||||
|
return [...sessions].sort(compareTerminalSessions);
|
||||||
|
}
|
||||||
|
|
||||||
|
export function reconcileTerminalSessions(
|
||||||
|
current: TerminalSession[],
|
||||||
|
listed: TerminalSession[],
|
||||||
|
): TerminalSession[] {
|
||||||
|
const listedById = new Map(listed.map((item) => [item.terminal_id, item]));
|
||||||
|
const retained = current.flatMap((item) => {
|
||||||
|
const updated = listedById.get(item.terminal_id);
|
||||||
|
if (!updated) return [];
|
||||||
|
listedById.delete(item.terminal_id);
|
||||||
|
return [updated];
|
||||||
|
});
|
||||||
|
return [...retained, ...orderTerminalSessions([...listedById.values()])];
|
||||||
|
}
|
||||||
|
|
||||||
|
export function restoreCandidates(
|
||||||
|
sessions: TerminalSession[],
|
||||||
|
rememberedId: string | null,
|
||||||
|
selectedAgentId: string,
|
||||||
|
): TerminalSession[] {
|
||||||
|
const remembered = sessions.find((item) => item.terminal_id === rememberedId);
|
||||||
|
const available = sessions.filter(
|
||||||
|
(item) => !item.operator_attached && item !== remembered,
|
||||||
|
);
|
||||||
|
available.sort((left, right) => {
|
||||||
|
const leftPreferred = left.agent_id === selectedAgentId ? 1 : 0;
|
||||||
|
const rightPreferred = right.agent_id === selectedAgentId ? 1 : 0;
|
||||||
|
return (
|
||||||
|
rightPreferred - leftPreferred ||
|
||||||
|
right.created_at_unix - left.created_at_unix
|
||||||
|
);
|
||||||
|
});
|
||||||
|
return remembered ? [remembered, ...available] : available;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function applyTerminalModifiers(
|
||||||
|
data: string,
|
||||||
|
modifiers: TerminalModifiers,
|
||||||
|
): string {
|
||||||
|
let modified = data;
|
||||||
|
if (modifiers.ctrl && modified.length === 1) {
|
||||||
|
const code = modified.toUpperCase().charCodeAt(0);
|
||||||
|
if (code >= 64 && code <= 95) {
|
||||||
|
modified = String.fromCharCode(code & 0x1f);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return modifiers.meta ? `\x1b${modified}` : modified;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function visibleTerminalText(terminal: XTerm): string {
|
||||||
|
const buffer = terminal.buffer.active;
|
||||||
|
const firstLine = buffer.viewportY;
|
||||||
|
const lastLine = Math.min(buffer.length, firstLine + terminal.rows);
|
||||||
|
const lines: string[] = [];
|
||||||
|
for (let index = firstLine; index < lastLine; index += 1) {
|
||||||
|
lines.push(buffer.getLine(index)?.translateToString(true) ?? "");
|
||||||
|
}
|
||||||
|
while (lines.at(-1) === "") lines.pop();
|
||||||
|
return lines.join("\n");
|
||||||
|
}
|
||||||
@@ -0,0 +1,434 @@
|
|||||||
|
import { useCallback, useEffect, useRef, useState } from "react";
|
||||||
|
import { toast } from "sonner";
|
||||||
|
|
||||||
|
import {
|
||||||
|
APIError,
|
||||||
|
attachTerminal,
|
||||||
|
closeTerminal,
|
||||||
|
createTerminal,
|
||||||
|
type TerminalSession,
|
||||||
|
} from "@/api";
|
||||||
|
import {
|
||||||
|
NO_TERMINAL_MODIFIERS,
|
||||||
|
applyTerminalModifiers,
|
||||||
|
orderTerminalSessions,
|
||||||
|
restoreCandidates,
|
||||||
|
type TerminalConnectionState,
|
||||||
|
type TerminalModifiers,
|
||||||
|
} from "@/pages/terminal/sessionUtils";
|
||||||
|
import type { TerminalCatalog } from "@/pages/terminal/useTerminalSessions";
|
||||||
|
import type { TerminalPort } from "@/pages/terminal/useXterm";
|
||||||
|
|
||||||
|
const REMEMBERED_TERMINAL_KEY = "wakey.active-terminal-id";
|
||||||
|
const TERMINAL_OPERATOR_KEY = "wakey.terminal-operator-id";
|
||||||
|
|
||||||
|
// Fallback for browsers that do not implement crypto.randomUUID().
|
||||||
|
const randomId = () =>
|
||||||
|
(String(1e7) + -1e3 + -4e3 + -8e3 + -1e11).replace(/[018]/g, (c: string) => {
|
||||||
|
const num = Number(c);
|
||||||
|
return (
|
||||||
|
num ^
|
||||||
|
(window.crypto.getRandomValues(new Uint8Array(1))[0] & (15 >> (num / 4)))
|
||||||
|
).toString(16);
|
||||||
|
});
|
||||||
|
|
||||||
|
function terminalOperatorId(): string {
|
||||||
|
const remembered = window.sessionStorage.getItem(TERMINAL_OPERATOR_KEY);
|
||||||
|
if (remembered) return remembered;
|
||||||
|
const created = window.crypto.randomUUID?.() ?? randomId();
|
||||||
|
window.sessionStorage.setItem(TERMINAL_OPERATOR_KEY, created);
|
||||||
|
return created;
|
||||||
|
}
|
||||||
|
|
||||||
|
function websocketUrl(path: string): string {
|
||||||
|
const scheme = window.location.protocol === "https:" ? "wss:" : "ws:";
|
||||||
|
return `${scheme}//${window.location.host}${path}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
type Options = {
|
||||||
|
terminal: TerminalPort;
|
||||||
|
catalog: TerminalCatalog;
|
||||||
|
selectedAgentId: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
export function useTerminalSession({
|
||||||
|
terminal,
|
||||||
|
catalog,
|
||||||
|
selectedAgentId,
|
||||||
|
}: Options) {
|
||||||
|
const socketRef = useRef<WebSocket | null>(null);
|
||||||
|
const activeSessionRef = useRef<TerminalSession | null>(null);
|
||||||
|
const mountedRef = useRef(false);
|
||||||
|
const selectedAgentIdRef = useRef(selectedAgentId);
|
||||||
|
const catalogRef = useRef(catalog);
|
||||||
|
const [session, setSession] = useState<TerminalSession | null>(null);
|
||||||
|
const [connection, setConnection] = useState<TerminalConnectionState>("idle");
|
||||||
|
const [modifiers, setModifiers] = useState<TerminalModifiers>(
|
||||||
|
NO_TERMINAL_MODIFIERS,
|
||||||
|
);
|
||||||
|
const [operatorId] = useState(terminalOperatorId);
|
||||||
|
|
||||||
|
selectedAgentIdRef.current = selectedAgentId;
|
||||||
|
catalogRef.current = catalog;
|
||||||
|
activeSessionRef.current = session;
|
||||||
|
|
||||||
|
const detachTransport = useCallback(() => {
|
||||||
|
const socket = socketRef.current;
|
||||||
|
socketRef.current = null;
|
||||||
|
socket?.close();
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const writeTerminalData = useCallback(
|
||||||
|
(data: string, consumeModifiers = false) => {
|
||||||
|
const socket = socketRef.current;
|
||||||
|
if (socket?.readyState !== WebSocket.OPEN) return;
|
||||||
|
const output = consumeModifiers
|
||||||
|
? applyTerminalModifiers(data, modifiersRef.current)
|
||||||
|
: data;
|
||||||
|
socket.send(new TextEncoder().encode(output));
|
||||||
|
if (
|
||||||
|
consumeModifiers &&
|
||||||
|
(modifiersRef.current.ctrl || modifiersRef.current.meta)
|
||||||
|
) {
|
||||||
|
modifiersRef.current = NO_TERMINAL_MODIFIERS;
|
||||||
|
setModifiers(NO_TERMINAL_MODIFIERS);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[],
|
||||||
|
);
|
||||||
|
|
||||||
|
const modifiersRef = useRef(modifiers);
|
||||||
|
modifiersRef.current = modifiers;
|
||||||
|
|
||||||
|
const connect = useCallback(
|
||||||
|
(nextSession: TerminalSession) => {
|
||||||
|
if (!nextSession.attachment_token) {
|
||||||
|
throw new Error(
|
||||||
|
"Control plane did not issue a terminal attachment token",
|
||||||
|
);
|
||||||
|
}
|
||||||
|
detachTransport();
|
||||||
|
setSession(nextSession);
|
||||||
|
activeSessionRef.current = nextSession;
|
||||||
|
terminal.setActiveTerminalId(nextSession.terminal_id);
|
||||||
|
catalogRef.current.merge({ ...nextSession, operator_attached: true });
|
||||||
|
window.sessionStorage.setItem(
|
||||||
|
REMEMBERED_TERMINAL_KEY,
|
||||||
|
nextSession.terminal_id,
|
||||||
|
);
|
||||||
|
setConnection("connecting");
|
||||||
|
const socket = new WebSocket(websocketUrl(nextSession.websocket_url));
|
||||||
|
socket.binaryType = "arraybuffer";
|
||||||
|
socketRef.current = socket;
|
||||||
|
|
||||||
|
socket.onopen = () => {
|
||||||
|
if (socketRef.current !== socket) return;
|
||||||
|
socket.send(
|
||||||
|
JSON.stringify({
|
||||||
|
type: "attach",
|
||||||
|
attachment_token: nextSession.attachment_token,
|
||||||
|
operator_id: operatorId,
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
terminal.fit();
|
||||||
|
terminal.sendResize((message) => socket.send(message));
|
||||||
|
};
|
||||||
|
socket.onmessage = (event) => {
|
||||||
|
if (socketRef.current !== socket) return;
|
||||||
|
if (typeof event.data !== "string") {
|
||||||
|
terminal.write(new Uint8Array(event.data as ArrayBuffer));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const control = JSON.parse(event.data) as {
|
||||||
|
type: string;
|
||||||
|
exit_code?: number | null;
|
||||||
|
message?: string;
|
||||||
|
};
|
||||||
|
if (control.type === "ready") {
|
||||||
|
setConnection("ready");
|
||||||
|
void catalogRef.current.refresh().catch(() => undefined);
|
||||||
|
window.requestAnimationFrame(() => {
|
||||||
|
terminal.fit();
|
||||||
|
terminal.sendResize((message) => socket.send(message), true);
|
||||||
|
terminal.focus();
|
||||||
|
});
|
||||||
|
} else if (control.type === "exited") {
|
||||||
|
setConnection("exited");
|
||||||
|
terminal.writeLine(
|
||||||
|
`\r\n[process exited${control.exit_code == null ? "" : ` ${control.exit_code}`}]`,
|
||||||
|
);
|
||||||
|
} else if (control.type === "error") {
|
||||||
|
setConnection("exited");
|
||||||
|
terminal.writeLine(
|
||||||
|
`\r\n[terminal error: ${control.message ?? "unknown error"}]`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
terminal.writeLine("\r\n[invalid terminal control frame]");
|
||||||
|
}
|
||||||
|
};
|
||||||
|
socket.onerror = () => {
|
||||||
|
if (socketRef.current !== socket) return;
|
||||||
|
terminal.writeLine("\r\n[terminal transport error]");
|
||||||
|
};
|
||||||
|
socket.onclose = () => {
|
||||||
|
if (socketRef.current !== socket) return;
|
||||||
|
socketRef.current = null;
|
||||||
|
catalogRef.current.markDetached(nextSession.terminal_id);
|
||||||
|
setConnection((current) =>
|
||||||
|
current === "exited" ? current : "disconnected",
|
||||||
|
);
|
||||||
|
void catalogRef.current.refresh().catch(() => undefined);
|
||||||
|
};
|
||||||
|
},
|
||||||
|
[detachTransport, operatorId, terminal],
|
||||||
|
);
|
||||||
|
|
||||||
|
const activate = useCallback(
|
||||||
|
async (nextSession: TerminalSession) => {
|
||||||
|
if (nextSession.terminal_id === activeSessionRef.current?.terminal_id)
|
||||||
|
return;
|
||||||
|
if (nextSession.operator_attached) return;
|
||||||
|
|
||||||
|
const previousSession = activeSessionRef.current;
|
||||||
|
detachTransport();
|
||||||
|
if (previousSession)
|
||||||
|
catalogRef.current.markDetached(previousSession.terminal_id);
|
||||||
|
setSession(null);
|
||||||
|
activeSessionRef.current = null;
|
||||||
|
terminal.setActiveTerminalId(undefined);
|
||||||
|
setConnection("connecting");
|
||||||
|
terminal.reset();
|
||||||
|
window.sessionStorage.setItem(
|
||||||
|
REMEMBERED_TERMINAL_KEY,
|
||||||
|
nextSession.terminal_id,
|
||||||
|
);
|
||||||
|
|
||||||
|
try {
|
||||||
|
const attached = await attachTerminal(
|
||||||
|
nextSession.terminal_id,
|
||||||
|
operatorId,
|
||||||
|
);
|
||||||
|
if (!mountedRef.current) return;
|
||||||
|
connect(attached);
|
||||||
|
} catch (error) {
|
||||||
|
if (!mountedRef.current) return;
|
||||||
|
setConnection("idle");
|
||||||
|
toast.error("Could not attach terminal session", {
|
||||||
|
description: String(error),
|
||||||
|
});
|
||||||
|
void catalogRef.current.refresh().catch(() => undefined);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[connect, detachTransport, operatorId, terminal],
|
||||||
|
);
|
||||||
|
|
||||||
|
const start = useCallback(async () => {
|
||||||
|
const dimensions = terminal.dimensions();
|
||||||
|
if (!selectedAgentIdRef.current || !dimensions) return;
|
||||||
|
const previousSession = activeSessionRef.current;
|
||||||
|
detachTransport();
|
||||||
|
if (previousSession)
|
||||||
|
catalogRef.current.markDetached(previousSession.terminal_id);
|
||||||
|
setConnection("connecting");
|
||||||
|
terminal.reset();
|
||||||
|
try {
|
||||||
|
terminal.fit();
|
||||||
|
const fittedDimensions = terminal.dimensions() ?? dimensions;
|
||||||
|
const created = await createTerminal(
|
||||||
|
selectedAgentIdRef.current,
|
||||||
|
fittedDimensions.rows,
|
||||||
|
fittedDimensions.cols,
|
||||||
|
);
|
||||||
|
if (!mountedRef.current) return;
|
||||||
|
connect(created);
|
||||||
|
} catch (error) {
|
||||||
|
if (!mountedRef.current) return;
|
||||||
|
setSession(null);
|
||||||
|
activeSessionRef.current = null;
|
||||||
|
terminal.setActiveTerminalId(undefined);
|
||||||
|
setConnection("idle");
|
||||||
|
toast.error("Could not start terminal", { description: String(error) });
|
||||||
|
}
|
||||||
|
}, [connect, detachTransport, terminal]);
|
||||||
|
|
||||||
|
const reconnect = useCallback(async () => {
|
||||||
|
const currentSession = activeSessionRef.current;
|
||||||
|
if (!currentSession) return;
|
||||||
|
detachTransport();
|
||||||
|
try {
|
||||||
|
terminal.reset();
|
||||||
|
setConnection("connecting");
|
||||||
|
const attached = await attachTerminal(
|
||||||
|
currentSession.terminal_id,
|
||||||
|
operatorId,
|
||||||
|
);
|
||||||
|
if (!mountedRef.current) return;
|
||||||
|
connect(attached);
|
||||||
|
} catch (error) {
|
||||||
|
if (!mountedRef.current) return;
|
||||||
|
setConnection("disconnected");
|
||||||
|
toast.error("Terminal session is no longer available", {
|
||||||
|
description: String(error),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}, [connect, detachTransport, operatorId, terminal]);
|
||||||
|
|
||||||
|
const close = useCallback(
|
||||||
|
async (closingSession: TerminalSession) => {
|
||||||
|
const closingId = closingSession.terminal_id;
|
||||||
|
const closesActiveSession =
|
||||||
|
closingId === activeSessionRef.current?.terminal_id;
|
||||||
|
const remaining = catalogRef.current
|
||||||
|
.getSessions()
|
||||||
|
.filter((item) => item.terminal_id !== closingId);
|
||||||
|
const fallback = remaining.find((item) => !item.operator_attached);
|
||||||
|
|
||||||
|
if (closesActiveSession) {
|
||||||
|
if (socketRef.current?.readyState === WebSocket.OPEN) {
|
||||||
|
socketRef.current.send(JSON.stringify({ type: "close" }));
|
||||||
|
}
|
||||||
|
detachTransport();
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
await closeTerminal(closingId);
|
||||||
|
} catch (error) {
|
||||||
|
if (!mountedRef.current) return;
|
||||||
|
toast.error("Terminal cleanup failed", { description: String(error) });
|
||||||
|
void catalogRef.current.refresh().catch(() => undefined);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (!mountedRef.current) return;
|
||||||
|
|
||||||
|
catalogRef.current.remove(closingId);
|
||||||
|
terminal.forgetTitle(closingId);
|
||||||
|
if (
|
||||||
|
window.sessionStorage.getItem(REMEMBERED_TERMINAL_KEY) === closingId
|
||||||
|
) {
|
||||||
|
window.sessionStorage.removeItem(REMEMBERED_TERMINAL_KEY);
|
||||||
|
}
|
||||||
|
if (closesActiveSession) {
|
||||||
|
setSession(null);
|
||||||
|
activeSessionRef.current = null;
|
||||||
|
terminal.setActiveTerminalId(undefined);
|
||||||
|
setConnection("idle");
|
||||||
|
// reset discards the complete screen and terminal modes; clear() does not.
|
||||||
|
terminal.reset();
|
||||||
|
if (fallback) void activate(fallback);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[activate, detachTransport, terminal],
|
||||||
|
);
|
||||||
|
|
||||||
|
const toggleModifier = useCallback(
|
||||||
|
(modifier: keyof TerminalModifiers) => {
|
||||||
|
setModifiers((current) => {
|
||||||
|
const next = { ...current, [modifier]: !current[modifier] };
|
||||||
|
modifiersRef.current = next;
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
window.requestAnimationFrame(() => terminal.focus());
|
||||||
|
},
|
||||||
|
[terminal],
|
||||||
|
);
|
||||||
|
|
||||||
|
const writeAccessoryKey = useCallback(
|
||||||
|
(data: string) => {
|
||||||
|
writeTerminalData(data);
|
||||||
|
window.requestAnimationFrame(() => terminal.focus());
|
||||||
|
},
|
||||||
|
[terminal, writeTerminalData],
|
||||||
|
);
|
||||||
|
|
||||||
|
// Transport input/resize subscriptions and restoration begin only once xterm is usable.
|
||||||
|
useEffect(() => {
|
||||||
|
if (!terminal.ready) return;
|
||||||
|
mountedRef.current = true;
|
||||||
|
const input = terminal.onInput((data) => writeTerminalData(data, true));
|
||||||
|
const resize = terminal.onResize(() => {
|
||||||
|
const socket = socketRef.current;
|
||||||
|
if (socket?.readyState === WebSocket.OPEN) {
|
||||||
|
terminal.sendResize((message) => socket.send(message));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
let cancelled = false;
|
||||||
|
async function restoreTerminalSession() {
|
||||||
|
try {
|
||||||
|
const listed = await catalogRef.current.refresh();
|
||||||
|
if (cancelled) return;
|
||||||
|
const rememberedId = window.sessionStorage.getItem(
|
||||||
|
REMEMBERED_TERMINAL_KEY,
|
||||||
|
);
|
||||||
|
const candidates = restoreCandidates(
|
||||||
|
orderTerminalSessions(listed),
|
||||||
|
rememberedId,
|
||||||
|
selectedAgentIdRef.current,
|
||||||
|
);
|
||||||
|
|
||||||
|
for (const candidate of candidates) {
|
||||||
|
if (cancelled) return;
|
||||||
|
setConnection("connecting");
|
||||||
|
try {
|
||||||
|
const attached = await attachTerminal(
|
||||||
|
candidate.terminal_id,
|
||||||
|
operatorId,
|
||||||
|
);
|
||||||
|
if (cancelled) return;
|
||||||
|
terminal.reset();
|
||||||
|
connect(attached);
|
||||||
|
return;
|
||||||
|
} catch (error) {
|
||||||
|
if (
|
||||||
|
!(error instanceof APIError) ||
|
||||||
|
error.code !== "terminal_operator_already_attached"
|
||||||
|
) {
|
||||||
|
throw error;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (!cancelled) setConnection("idle");
|
||||||
|
} catch (error) {
|
||||||
|
if (cancelled) return;
|
||||||
|
setConnection("idle");
|
||||||
|
toast.error("Could not restore terminal session", {
|
||||||
|
description: String(error),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void restoreTerminalSession();
|
||||||
|
return () => {
|
||||||
|
cancelled = true;
|
||||||
|
mountedRef.current = false;
|
||||||
|
input.dispose();
|
||||||
|
resize.dispose();
|
||||||
|
detachTransport();
|
||||||
|
};
|
||||||
|
}, [
|
||||||
|
terminal.ready,
|
||||||
|
terminal.onInput,
|
||||||
|
terminal.onResize,
|
||||||
|
terminal.reset,
|
||||||
|
terminal.sendResize,
|
||||||
|
terminal,
|
||||||
|
connect,
|
||||||
|
detachTransport,
|
||||||
|
operatorId,
|
||||||
|
writeTerminalData,
|
||||||
|
]);
|
||||||
|
|
||||||
|
return {
|
||||||
|
session,
|
||||||
|
connection,
|
||||||
|
modifiers,
|
||||||
|
start,
|
||||||
|
activate,
|
||||||
|
reconnect,
|
||||||
|
close,
|
||||||
|
writeTerminalData,
|
||||||
|
writeAccessoryKey,
|
||||||
|
toggleModifier,
|
||||||
|
};
|
||||||
|
}
|
||||||
@@ -0,0 +1,157 @@
|
|||||||
|
import { useCallback, useEffect, useRef, useState } from "react";
|
||||||
|
|
||||||
|
import { listTerminals, type TerminalSession } from "@/api";
|
||||||
|
import {
|
||||||
|
mergeTerminalSession,
|
||||||
|
reconcileTerminalSessions,
|
||||||
|
} from "@/pages/terminal/sessionUtils";
|
||||||
|
|
||||||
|
function websocketUrl(path: string): string {
|
||||||
|
const scheme = window.location.protocol === "https:" ? "wss:" : "ws:";
|
||||||
|
return `${scheme}//${window.location.host}${path}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function useTerminalSessions() {
|
||||||
|
const [sessions, setSessions] = useState<TerminalSession[]>([]);
|
||||||
|
const sessionsRef = useRef(sessions);
|
||||||
|
const refreshRef = useRef<() => Promise<TerminalSession[]>>(async () => []);
|
||||||
|
|
||||||
|
sessionsRef.current = sessions;
|
||||||
|
|
||||||
|
const refresh = useCallback(() => refreshRef.current(), []);
|
||||||
|
const merge = useCallback((session: TerminalSession) => {
|
||||||
|
setSessions((current) => mergeTerminalSession(current, session));
|
||||||
|
}, []);
|
||||||
|
const markDetached = useCallback((terminalId: string) => {
|
||||||
|
setSessions((current) =>
|
||||||
|
current.map((item) =>
|
||||||
|
item.terminal_id === terminalId
|
||||||
|
? { ...item, operator_attached: false }
|
||||||
|
: item,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
}, []);
|
||||||
|
const remove = useCallback((terminalId: string) => {
|
||||||
|
setSessions((current) =>
|
||||||
|
current.filter((item) => item.terminal_id !== terminalId),
|
||||||
|
);
|
||||||
|
}, []);
|
||||||
|
const reconcile = useCallback((listed: TerminalSession[]) => {
|
||||||
|
setSessions((current) => reconcileTerminalSessions(current, listed));
|
||||||
|
}, []);
|
||||||
|
const getSessions = useCallback(() => sessionsRef.current, []);
|
||||||
|
|
||||||
|
// The catalog stream and fallback timer live for the page mount, independent of active sessions.
|
||||||
|
useEffect(() => {
|
||||||
|
let cancelled = false;
|
||||||
|
let refreshInFlight: Promise<TerminalSession[]> | null = null;
|
||||||
|
let refreshRequested = false;
|
||||||
|
let eventsSocket: WebSocket | null = null;
|
||||||
|
let reconnectTimer = 0;
|
||||||
|
let healthyTimer = 0;
|
||||||
|
let reconnectAttempt = 0;
|
||||||
|
|
||||||
|
const refreshSessions = (): Promise<TerminalSession[]> => {
|
||||||
|
if (refreshInFlight) {
|
||||||
|
// Coalesce bursts and fetch again if another invalidation arrives in flight.
|
||||||
|
refreshRequested = true;
|
||||||
|
return refreshInFlight;
|
||||||
|
}
|
||||||
|
refreshInFlight = (async () => {
|
||||||
|
let listed: TerminalSession[] = [];
|
||||||
|
do {
|
||||||
|
refreshRequested = false;
|
||||||
|
listed = await listTerminals();
|
||||||
|
if (!cancelled) {
|
||||||
|
setSessions((current) =>
|
||||||
|
reconcileTerminalSessions(current, listed),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
} while (refreshRequested && !cancelled);
|
||||||
|
return listed;
|
||||||
|
})().finally(() => {
|
||||||
|
refreshInFlight = null;
|
||||||
|
if (refreshRequested && !cancelled) {
|
||||||
|
void refreshSessions().catch(() => undefined);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
return refreshInFlight;
|
||||||
|
};
|
||||||
|
|
||||||
|
refreshRef.current = refreshSessions;
|
||||||
|
|
||||||
|
const connectEvents = () => {
|
||||||
|
if (cancelled) return;
|
||||||
|
const socket = new WebSocket(
|
||||||
|
websocketUrl("/api/v1/control/terminals/events/ws"),
|
||||||
|
);
|
||||||
|
eventsSocket = socket;
|
||||||
|
socket.onmessage = (event) => {
|
||||||
|
if (cancelled || eventsSocket !== socket) return;
|
||||||
|
if (typeof event.data !== "string") return;
|
||||||
|
try {
|
||||||
|
const message = JSON.parse(event.data) as { type?: string };
|
||||||
|
if (message.type === "sessions_changed") {
|
||||||
|
void refreshSessions().catch(() => undefined);
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
// Invalidations are hints; the periodic recovery fetch remains active.
|
||||||
|
}
|
||||||
|
};
|
||||||
|
socket.onopen = () => {
|
||||||
|
void refreshSessions().catch(() => undefined);
|
||||||
|
window.clearTimeout(healthyTimer);
|
||||||
|
healthyTimer = window.setTimeout(() => {
|
||||||
|
if (eventsSocket === socket && socket.readyState === WebSocket.OPEN) {
|
||||||
|
reconnectAttempt = 0;
|
||||||
|
}
|
||||||
|
}, 10_000);
|
||||||
|
};
|
||||||
|
socket.onclose = () => {
|
||||||
|
if (cancelled || eventsSocket !== socket) return;
|
||||||
|
window.clearTimeout(healthyTimer);
|
||||||
|
const ceiling = Math.min(500 * 2 ** reconnectAttempt, 12_000);
|
||||||
|
const delay = Math.min(
|
||||||
|
Math.round(ceiling * (0.8 + Math.random() * 0.4)),
|
||||||
|
15_000,
|
||||||
|
);
|
||||||
|
reconnectAttempt = Math.min(reconnectAttempt + 1, 5);
|
||||||
|
reconnectTimer = window.setTimeout(connectEvents, delay);
|
||||||
|
};
|
||||||
|
socket.onerror = () => socket.close();
|
||||||
|
};
|
||||||
|
|
||||||
|
connectEvents();
|
||||||
|
void refreshSessions().catch(() => undefined);
|
||||||
|
const refreshTimer = window.setInterval(() => {
|
||||||
|
void refreshSessions().catch(() => undefined);
|
||||||
|
}, 15_000);
|
||||||
|
const refreshOnFocus = () => {
|
||||||
|
void refreshSessions().catch(() => undefined);
|
||||||
|
};
|
||||||
|
window.addEventListener("focus", refreshOnFocus);
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
cancelled = true;
|
||||||
|
window.clearInterval(refreshTimer);
|
||||||
|
window.removeEventListener("focus", refreshOnFocus);
|
||||||
|
window.clearTimeout(reconnectTimer);
|
||||||
|
window.clearTimeout(healthyTimer);
|
||||||
|
eventsSocket?.close();
|
||||||
|
eventsSocket = null;
|
||||||
|
refreshRef.current = async () => [];
|
||||||
|
};
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
return {
|
||||||
|
sessions,
|
||||||
|
refresh,
|
||||||
|
merge,
|
||||||
|
markDetached,
|
||||||
|
remove,
|
||||||
|
reconcile,
|
||||||
|
getSessions,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
export type TerminalCatalog = ReturnType<typeof useTerminalSessions>;
|
||||||
@@ -0,0 +1,371 @@
|
|||||||
|
import {
|
||||||
|
useCallback,
|
||||||
|
useEffect,
|
||||||
|
useMemo,
|
||||||
|
useRef,
|
||||||
|
useState,
|
||||||
|
type RefObject,
|
||||||
|
} from "react";
|
||||||
|
import { ClipboardAddon } from "@xterm/addon-clipboard";
|
||||||
|
import { FitAddon } from "@xterm/addon-fit";
|
||||||
|
import { ImageAddon } from "@xterm/addon-image";
|
||||||
|
import { Unicode11Addon } from "@xterm/addon-unicode11";
|
||||||
|
import { UnicodeGraphemesAddon } from "@xterm/addon-unicode-graphemes";
|
||||||
|
import { WebLinksAddon } from "@xterm/addon-web-links";
|
||||||
|
import { Terminal as XTerm } from "@xterm/xterm";
|
||||||
|
import { toast } from "sonner";
|
||||||
|
|
||||||
|
import {
|
||||||
|
MAX_TERMINAL_FONT_SIZE,
|
||||||
|
MIN_TERMINAL_FONT_SIZE,
|
||||||
|
} from "@/pages/terminal/TerminalAccessoryBar";
|
||||||
|
import { visibleTerminalText } from "@/pages/terminal/sessionUtils";
|
||||||
|
import { loadTerminalFontFamily } from "@/terminal/terminalFonts";
|
||||||
|
|
||||||
|
const DEFAULT_TERMINAL_FONT_SIZE = 14;
|
||||||
|
|
||||||
|
// Stable behavioral surface shared with session transport and page controls.
|
||||||
|
export type TerminalPort = {
|
||||||
|
clear: () => void;
|
||||||
|
copy: () => Promise<void>;
|
||||||
|
dimensions: () => { rows: number; cols: number } | null;
|
||||||
|
forgetTitle: (terminalId: string) => void;
|
||||||
|
fit: () => void;
|
||||||
|
focus: () => void;
|
||||||
|
blurInput: () => void;
|
||||||
|
onInput: (handler: (data: string) => void) => { dispose: () => void };
|
||||||
|
onResize: (handler: () => void) => { dispose: () => void };
|
||||||
|
paste: (write: (data: string) => void) => Promise<void>;
|
||||||
|
reset: () => void;
|
||||||
|
scrollPages: (pages: number) => void;
|
||||||
|
sendResize: (send: (message: string) => void, force?: boolean) => void;
|
||||||
|
changeFontSize: (delta: number) => void;
|
||||||
|
setActiveTerminalId: (terminalId: string | undefined) => void;
|
||||||
|
write: (data: Uint8Array) => void;
|
||||||
|
writeLine: (line: string) => void;
|
||||||
|
ready: boolean;
|
||||||
|
};
|
||||||
|
|
||||||
|
type Options = {
|
||||||
|
pageRef: RefObject<HTMLElement | null>;
|
||||||
|
hostRef: RefObject<HTMLDivElement | null>;
|
||||||
|
};
|
||||||
|
|
||||||
|
export function useXterm({ pageRef, hostRef }: Options) {
|
||||||
|
const terminalRef = useRef<XTerm | null>(null);
|
||||||
|
const fitRef = useRef<FitAddon | null>(null);
|
||||||
|
const activeTerminalIdRef = useRef<string | undefined>(undefined);
|
||||||
|
const lastSizeRef = useRef({ rows: 0, cols: 0 });
|
||||||
|
const resizeListenersRef = useRef(new Set<() => void>());
|
||||||
|
const [fontFamily, setFontFamily] = useState<string>();
|
||||||
|
const [ready, setReady] = useState(false);
|
||||||
|
const [fontSize, setFontSize] = useState(DEFAULT_TERMINAL_FONT_SIZE);
|
||||||
|
const [sessionTitles, setSessionTitles] = useState<Record<string, string>>(
|
||||||
|
{},
|
||||||
|
);
|
||||||
|
|
||||||
|
const setActiveTerminalId = useCallback((terminalId: string | undefined) => {
|
||||||
|
activeTerminalIdRef.current = terminalId;
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
let cancelled = false;
|
||||||
|
void loadTerminalFontFamily().then((loadedFontFamily) => {
|
||||||
|
if (!cancelled) setFontFamily(loadedFontFamily);
|
||||||
|
});
|
||||||
|
return () => {
|
||||||
|
cancelled = true;
|
||||||
|
};
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
const viewport = window.visualViewport;
|
||||||
|
const updateViewportHeight = () => {
|
||||||
|
const height = Math.round(viewport?.height ?? window.innerHeight);
|
||||||
|
pageRef.current?.style.setProperty(
|
||||||
|
"--terminal-visual-viewport-height",
|
||||||
|
`${height}px`,
|
||||||
|
);
|
||||||
|
};
|
||||||
|
updateViewportHeight();
|
||||||
|
viewport?.addEventListener("resize", updateViewportHeight);
|
||||||
|
viewport?.addEventListener("scroll", updateViewportHeight);
|
||||||
|
window.addEventListener("resize", updateViewportHeight);
|
||||||
|
return () => {
|
||||||
|
viewport?.removeEventListener("resize", updateViewportHeight);
|
||||||
|
viewport?.removeEventListener("scroll", updateViewportHeight);
|
||||||
|
window.removeEventListener("resize", updateViewportHeight);
|
||||||
|
};
|
||||||
|
}, [pageRef]);
|
||||||
|
|
||||||
|
const sendResize = useCallback(
|
||||||
|
(send: (message: string) => void, force = false) => {
|
||||||
|
const terminal = terminalRef.current;
|
||||||
|
if (!terminal) return;
|
||||||
|
if (
|
||||||
|
!force &&
|
||||||
|
lastSizeRef.current.rows === terminal.rows &&
|
||||||
|
lastSizeRef.current.cols === terminal.cols
|
||||||
|
) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
lastSizeRef.current = { rows: terminal.rows, cols: terminal.cols };
|
||||||
|
send(
|
||||||
|
JSON.stringify({
|
||||||
|
type: "resize",
|
||||||
|
rows: terminal.rows,
|
||||||
|
cols: terminal.cols,
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
},
|
||||||
|
[],
|
||||||
|
);
|
||||||
|
|
||||||
|
const fit = useCallback(() => fitRef.current?.fit(), []);
|
||||||
|
const focus = useCallback(() => terminalRef.current?.focus(), []);
|
||||||
|
const blurInput = useCallback(
|
||||||
|
() => terminalRef.current?.textarea?.blur(),
|
||||||
|
[],
|
||||||
|
);
|
||||||
|
const reset = useCallback(() => terminalRef.current?.reset(), []);
|
||||||
|
const clear = useCallback(() => terminalRef.current?.clear(), []);
|
||||||
|
const write = useCallback(
|
||||||
|
(data: Uint8Array) => terminalRef.current?.write(data),
|
||||||
|
[],
|
||||||
|
);
|
||||||
|
const writeLine = useCallback(
|
||||||
|
(line: string) => terminalRef.current?.writeln(line),
|
||||||
|
[],
|
||||||
|
);
|
||||||
|
const dimensions = useCallback(() => {
|
||||||
|
const terminal = terminalRef.current;
|
||||||
|
return terminal ? { rows: terminal.rows, cols: terminal.cols } : null;
|
||||||
|
}, []);
|
||||||
|
const forgetTitle = useCallback((terminalId: string) => {
|
||||||
|
setSessionTitles((current) => {
|
||||||
|
if (!(terminalId in current)) return current;
|
||||||
|
const next = { ...current };
|
||||||
|
delete next[terminalId];
|
||||||
|
return next;
|
||||||
|
});
|
||||||
|
}, []);
|
||||||
|
const onInput = useCallback(
|
||||||
|
(handler: (data: string) => void) =>
|
||||||
|
terminalRef.current?.onData(handler) ?? { dispose: () => {} },
|
||||||
|
[],
|
||||||
|
);
|
||||||
|
const onResize = useCallback((handler: () => void) => {
|
||||||
|
resizeListenersRef.current.add(handler);
|
||||||
|
return { dispose: () => resizeListenersRef.current.delete(handler) };
|
||||||
|
}, []);
|
||||||
|
const scrollPages = useCallback(
|
||||||
|
(pages: number) => terminalRef.current?.scrollPages(pages),
|
||||||
|
[],
|
||||||
|
);
|
||||||
|
const copy = useCallback(async () => {
|
||||||
|
const terminal = terminalRef.current;
|
||||||
|
if (!terminal) return;
|
||||||
|
const text = terminal.hasSelection()
|
||||||
|
? terminal.getSelection()
|
||||||
|
: visibleTerminalText(terminal);
|
||||||
|
if (!text) {
|
||||||
|
toast.info("Nothing to copy");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (!navigator.clipboard) {
|
||||||
|
toast.error("Clipboard access requires HTTPS or localhost");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
await navigator.clipboard.writeText(text);
|
||||||
|
toast.success(
|
||||||
|
terminal.hasSelection() ? "Selection copied" : "Visible screen copied",
|
||||||
|
);
|
||||||
|
} catch (error) {
|
||||||
|
toast.error("Clipboard access was denied", {
|
||||||
|
description: String(error),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}, []);
|
||||||
|
const paste = useCallback(async (writeData: (data: string) => void) => {
|
||||||
|
if (!navigator.clipboard) {
|
||||||
|
toast.error("Clipboard access requires HTTPS or localhost");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
const text = await navigator.clipboard.readText();
|
||||||
|
if (text) writeData(text);
|
||||||
|
terminalRef.current?.focus();
|
||||||
|
} catch (error) {
|
||||||
|
toast.error("Clipboard access was denied", {
|
||||||
|
description: String(error),
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}, []);
|
||||||
|
const changeFontSize = useCallback((delta: number) => {
|
||||||
|
const terminal = terminalRef.current;
|
||||||
|
if (!terminal) return;
|
||||||
|
const current = terminal.options.fontSize ?? DEFAULT_TERMINAL_FONT_SIZE;
|
||||||
|
const next = Math.min(
|
||||||
|
MAX_TERMINAL_FONT_SIZE,
|
||||||
|
Math.max(MIN_TERMINAL_FONT_SIZE, current + delta),
|
||||||
|
);
|
||||||
|
if (next === current) return;
|
||||||
|
terminal.options.fontSize = next;
|
||||||
|
setFontSize(next);
|
||||||
|
window.requestAnimationFrame(() => {
|
||||||
|
fitRef.current?.fit();
|
||||||
|
resizeListenersRef.current.forEach((handler) => handler());
|
||||||
|
});
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
// Wait for the font before creating xterm; this effect owns the instance and its observers.
|
||||||
|
useEffect(() => {
|
||||||
|
if (!hostRef.current || !fontFamily) return;
|
||||||
|
const terminal = new XTerm({
|
||||||
|
cursorBlink: true,
|
||||||
|
convertEol: false,
|
||||||
|
fontFamily,
|
||||||
|
fontSize: DEFAULT_TERMINAL_FONT_SIZE,
|
||||||
|
lineHeight: 1.1,
|
||||||
|
scrollbar: {
|
||||||
|
width: window.matchMedia("(pointer: coarse)").matches ? 22 : 16,
|
||||||
|
},
|
||||||
|
scrollback: 10_000,
|
||||||
|
theme: {
|
||||||
|
background: "#0b1117",
|
||||||
|
foreground: "#e5e7eb",
|
||||||
|
cursor: "#6ee7a8",
|
||||||
|
scrollbarSliderBackground: "#52617099",
|
||||||
|
scrollbarSliderHoverBackground: "#6b7c8dcc",
|
||||||
|
scrollbarSliderActiveBackground: "#8294a6",
|
||||||
|
},
|
||||||
|
allowProposedApi: true,
|
||||||
|
});
|
||||||
|
terminal.loadAddon(new Unicode11Addon());
|
||||||
|
terminal.unicode.activeVersion = "11";
|
||||||
|
terminal.loadAddon(new UnicodeGraphemesAddon());
|
||||||
|
terminal.loadAddon(new WebLinksAddon());
|
||||||
|
terminal.loadAddon(new ImageAddon({}));
|
||||||
|
const fitAddon = new FitAddon();
|
||||||
|
terminal.loadAddon(new ClipboardAddon());
|
||||||
|
terminal.loadAddon(fitAddon);
|
||||||
|
terminal.open(hostRef.current);
|
||||||
|
terminal.attachCustomKeyEventHandler((event) => {
|
||||||
|
const copiesTerminalSelection =
|
||||||
|
event.type === "keydown" &&
|
||||||
|
event.ctrlKey &&
|
||||||
|
event.shiftKey &&
|
||||||
|
event.code === "KeyC";
|
||||||
|
if (!copiesTerminalSelection) return true;
|
||||||
|
event.preventDefault();
|
||||||
|
event.stopPropagation();
|
||||||
|
if (terminal.hasSelection()) {
|
||||||
|
if (!navigator.clipboard) {
|
||||||
|
toast.error("Clipboard access requires HTTPS or localhost");
|
||||||
|
} else {
|
||||||
|
void navigator.clipboard
|
||||||
|
.writeText(terminal.getSelection())
|
||||||
|
.catch((error) =>
|
||||||
|
toast.error("Clipboard access was denied", {
|
||||||
|
description: String(error),
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
});
|
||||||
|
const titleChange = terminal.onTitleChange((title) => {
|
||||||
|
const terminalId = activeTerminalIdRef.current;
|
||||||
|
if (!terminalId) return;
|
||||||
|
const normalized = title.trim();
|
||||||
|
setSessionTitles((current) => {
|
||||||
|
if (current[terminalId] === normalized) return current;
|
||||||
|
if (!normalized) {
|
||||||
|
const next = { ...current };
|
||||||
|
delete next[terminalId];
|
||||||
|
return next;
|
||||||
|
}
|
||||||
|
return { ...current, [terminalId]: normalized };
|
||||||
|
});
|
||||||
|
});
|
||||||
|
fitAddon.fit();
|
||||||
|
// Consumers subscribe only after this transition; the port methods stay stable.
|
||||||
|
terminalRef.current = terminal;
|
||||||
|
fitRef.current = fitAddon;
|
||||||
|
setReady(true);
|
||||||
|
|
||||||
|
let resizeFrame = 0;
|
||||||
|
let lastHostWidth = 0;
|
||||||
|
let lastHostHeight = 0;
|
||||||
|
const resizeObserver = new ResizeObserver(([entry]) => {
|
||||||
|
const width = Math.round(entry.contentRect.width);
|
||||||
|
const height = Math.round(entry.contentRect.height);
|
||||||
|
if (width === lastHostWidth && height === lastHostHeight) return;
|
||||||
|
lastHostWidth = width;
|
||||||
|
lastHostHeight = height;
|
||||||
|
window.cancelAnimationFrame(resizeFrame);
|
||||||
|
resizeFrame = window.requestAnimationFrame(() => {
|
||||||
|
fitAddon.fit();
|
||||||
|
resizeListenersRef.current.forEach((handler) => handler());
|
||||||
|
});
|
||||||
|
});
|
||||||
|
resizeObserver.observe(hostRef.current);
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
titleChange.dispose();
|
||||||
|
resizeObserver.disconnect();
|
||||||
|
resizeListenersRef.current.clear();
|
||||||
|
window.cancelAnimationFrame(resizeFrame);
|
||||||
|
terminal.dispose();
|
||||||
|
terminalRef.current = null;
|
||||||
|
fitRef.current = null;
|
||||||
|
setReady(false);
|
||||||
|
};
|
||||||
|
}, [fontFamily, hostRef]);
|
||||||
|
|
||||||
|
const terminal = useMemo<TerminalPort>(
|
||||||
|
() => ({
|
||||||
|
clear,
|
||||||
|
copy,
|
||||||
|
dimensions,
|
||||||
|
forgetTitle,
|
||||||
|
fit,
|
||||||
|
focus,
|
||||||
|
blurInput,
|
||||||
|
onInput,
|
||||||
|
onResize,
|
||||||
|
paste,
|
||||||
|
reset,
|
||||||
|
scrollPages,
|
||||||
|
sendResize,
|
||||||
|
changeFontSize,
|
||||||
|
setActiveTerminalId,
|
||||||
|
write,
|
||||||
|
writeLine,
|
||||||
|
ready,
|
||||||
|
}),
|
||||||
|
[
|
||||||
|
clear,
|
||||||
|
copy,
|
||||||
|
dimensions,
|
||||||
|
forgetTitle,
|
||||||
|
fit,
|
||||||
|
focus,
|
||||||
|
blurInput,
|
||||||
|
onInput,
|
||||||
|
onResize,
|
||||||
|
paste,
|
||||||
|
reset,
|
||||||
|
scrollPages,
|
||||||
|
sendResize,
|
||||||
|
changeFontSize,
|
||||||
|
setActiveTerminalId,
|
||||||
|
write,
|
||||||
|
writeLine,
|
||||||
|
ready,
|
||||||
|
],
|
||||||
|
);
|
||||||
|
|
||||||
|
return { terminal, fontSize, sessionTitles };
|
||||||
|
}
|
||||||
+100
-3
@@ -326,7 +326,7 @@
|
|||||||
|
|
||||||
.terminal-page {
|
.terminal-page {
|
||||||
display: flex;
|
display: flex;
|
||||||
height: calc(100dvh - 2.5rem);
|
height: calc(var(--terminal-visual-viewport-height, 100dvh) - 2.5rem);
|
||||||
min-height: 0;
|
min-height: 0;
|
||||||
flex-direction: column;
|
flex-direction: column;
|
||||||
gap: 1rem;
|
gap: 1rem;
|
||||||
@@ -488,7 +488,6 @@
|
|||||||
|
|
||||||
.terminal-tab time {
|
.terminal-tab time {
|
||||||
flex: 0 0 auto;
|
flex: 0 0 auto;
|
||||||
color: #7f8d9a;
|
|
||||||
font-size: 0.6875rem;
|
font-size: 0.6875rem;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -578,6 +577,7 @@
|
|||||||
}
|
}
|
||||||
|
|
||||||
.terminal-surface {
|
.terminal-surface {
|
||||||
|
order: 2;
|
||||||
flex: 1;
|
flex: 1;
|
||||||
min-height: 0;
|
min-height: 0;
|
||||||
overflow: hidden;
|
overflow: hidden;
|
||||||
@@ -585,6 +585,87 @@
|
|||||||
padding: 0;
|
padding: 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.terminal-accessories-toggle[data-active] {
|
||||||
|
color: #6ee7a8;
|
||||||
|
background: rgb(110 231 168 / 10%);
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories {
|
||||||
|
display: flex;
|
||||||
|
order: 3;
|
||||||
|
min-width: 0;
|
||||||
|
min-height: 2.75rem;
|
||||||
|
flex: 0 0 auto;
|
||||||
|
align-items: center;
|
||||||
|
gap: 0.35rem;
|
||||||
|
padding: 0.35rem 0.45rem max(0.35rem, env(safe-area-inset-bottom));
|
||||||
|
overflow-x: auto;
|
||||||
|
overscroll-behavior-x: contain;
|
||||||
|
border-top: 1px solid rgb(255 255 255 / 9%);
|
||||||
|
background: #101820;
|
||||||
|
color: #cbd5e1;
|
||||||
|
scrollbar-width: thin;
|
||||||
|
touch-action: pan-x;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessory-group {
|
||||||
|
display: flex;
|
||||||
|
flex: 0 0 auto;
|
||||||
|
align-items: center;
|
||||||
|
gap: 0.25rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessory-group + .terminal-accessory-group {
|
||||||
|
padding-left: 0.35rem;
|
||||||
|
border-left: 1px solid rgb(255 255 255 / 10%);
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories .terminal-key {
|
||||||
|
width: auto;
|
||||||
|
min-width: 2.4rem;
|
||||||
|
height: 2rem;
|
||||||
|
padding-inline: 0.55rem;
|
||||||
|
border: 1px solid rgb(255 255 255 / 9%);
|
||||||
|
border-radius: calc(var(--radius) - 3px);
|
||||||
|
background: rgb(255 255 255 / 4%);
|
||||||
|
color: #cbd5e1;
|
||||||
|
font-family: ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, monospace;
|
||||||
|
font-size: 0.75rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories .terminal-key:hover {
|
||||||
|
background: rgb(255 255 255 / 9%);
|
||||||
|
color: #f1f5f9;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories .terminal-key[data-active] {
|
||||||
|
border-color: rgb(110 231 168 / 48%);
|
||||||
|
background: rgb(110 231 168 / 14%);
|
||||||
|
color: #9af0bf;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories .terminal-key--icon {
|
||||||
|
width: 2rem;
|
||||||
|
min-width: 2rem;
|
||||||
|
padding: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories .terminal-key--control {
|
||||||
|
min-width: 2.25rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories .terminal-key--wide {
|
||||||
|
min-width: 3.15rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-font-controls output {
|
||||||
|
min-width: 1.5rem;
|
||||||
|
color: #91a0ae;
|
||||||
|
font-family: ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, monospace;
|
||||||
|
font-size: 0.6875rem;
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
.terminal-surface .xterm {
|
.terminal-surface .xterm {
|
||||||
height: 100%;
|
height: 100%;
|
||||||
box-sizing: border-box;
|
box-sizing: border-box;
|
||||||
@@ -609,7 +690,7 @@
|
|||||||
|
|
||||||
@media (max-width: 640px) {
|
@media (max-width: 640px) {
|
||||||
.terminal-page {
|
.terminal-page {
|
||||||
height: calc(100dvh - 2.5rem);
|
height: calc(var(--terminal-visual-viewport-height, 100dvh) - 2.5rem);
|
||||||
}
|
}
|
||||||
|
|
||||||
.terminal-controls {
|
.terminal-controls {
|
||||||
@@ -626,6 +707,22 @@
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@media (pointer: coarse) {
|
||||||
|
.terminal-accessories {
|
||||||
|
min-height: 3.25rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories .terminal-key {
|
||||||
|
min-width: 2.75rem;
|
||||||
|
height: 2.75rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
.terminal-accessories .terminal-key--icon {
|
||||||
|
width: 2.75rem;
|
||||||
|
min-width: 2.75rem;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/* ── Presence badges ──────────────────────────────────────────────────── */
|
/* ── Presence badges ──────────────────────────────────────────────────── */
|
||||||
|
|
||||||
.presence-dot {
|
.presence-dot {
|
||||||
|
|||||||
@@ -0,0 +1,8 @@
|
|||||||
|
@font-face {
|
||||||
|
font-family: "Wakey JetBrainsMono Nerd Font Mono";
|
||||||
|
src: url("../assets/fonts/jetbrains-mono-nerd/JetBrainsMonoNerdFontMono-Regular.woff2")
|
||||||
|
format("woff2");
|
||||||
|
font-style: normal;
|
||||||
|
font-weight: 400;
|
||||||
|
font-display: block;
|
||||||
|
}
|
||||||
@@ -0,0 +1,31 @@
|
|||||||
|
import "./terminalFonts.css";
|
||||||
|
|
||||||
|
const TERMINAL_WEB_FONT = "Wakey JetBrainsMono Nerd Font Mono";
|
||||||
|
const TERMINAL_FALLBACK_FONTS =
|
||||||
|
'"SFMono-Regular", Consolas, "Liberation Mono", monospace';
|
||||||
|
|
||||||
|
let terminalFontsPromise: Promise<string> | undefined;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Resolves the terminal font stack only after browser glyph metrics are ready.
|
||||||
|
* xterm caches those metrics when it opens, so a late webfont load permanently
|
||||||
|
* misaligns cells for that terminal instance.
|
||||||
|
*/
|
||||||
|
export function loadTerminalFontFamily(): Promise<string> {
|
||||||
|
terminalFontsPromise ??= document.fonts
|
||||||
|
.load(`400 14px "${TERMINAL_WEB_FONT}"`, "\uE0B0")
|
||||||
|
.then((faces) => {
|
||||||
|
if (faces.length === 0) {
|
||||||
|
throw new Error("terminal webfont was not registered");
|
||||||
|
}
|
||||||
|
return `"${TERMINAL_WEB_FONT}", ${TERMINAL_FALLBACK_FONTS}`;
|
||||||
|
})
|
||||||
|
.catch((error: unknown) => {
|
||||||
|
console.warn(
|
||||||
|
"Could not load terminal webfont; using system monospace",
|
||||||
|
error,
|
||||||
|
);
|
||||||
|
return TERMINAL_FALLBACK_FONTS;
|
||||||
|
});
|
||||||
|
return terminalFontsPromise;
|
||||||
|
}
|
||||||
+1
-1
@@ -1,7 +1,7 @@
|
|||||||
{
|
{
|
||||||
"compilerOptions": {
|
"compilerOptions": {
|
||||||
"target": "ES2020",
|
"target": "ES2020",
|
||||||
"lib": ["ES2020", "DOM", "DOM.Iterable"],
|
"lib": ["ES2022", "DOM", "DOM.Iterable"],
|
||||||
"module": "ESNext",
|
"module": "ESNext",
|
||||||
"ignoreDeprecations": "5.0",
|
"ignoreDeprecations": "5.0",
|
||||||
"moduleResolution": "Bundler",
|
"moduleResolution": "Bundler",
|
||||||
|
|||||||
Vendored
+1
-1
Submodule vendor/vt100 updated: ea740d5ad1...60d0544b14.
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "wakey-agent"
|
name = "wakey-agent"
|
||||||
version = "0.3.3"
|
version = "0.3.7"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
publish = ["gitea"]
|
publish = ["gitea"]
|
||||||
|
|
||||||
@@ -9,6 +9,7 @@ anyhow = "1"
|
|||||||
clap = { version = "4", features = ["derive"] }
|
clap = { version = "4", features = ["derive"] }
|
||||||
futures-util = "0.3"
|
futures-util = "0.3"
|
||||||
http = "1"
|
http = "1"
|
||||||
|
humantime-serde = "1"
|
||||||
macaddr = { version = "1", features = ["serde", "serde_std"] }
|
macaddr = { version = "1", features = ["serde", "serde_std"] }
|
||||||
nix = { version = "0.31", default-features = false, features = [
|
nix = { version = "0.31", default-features = false, features = [
|
||||||
"signal",
|
"signal",
|
||||||
@@ -32,13 +33,10 @@ tokio = { version = "1", features = [
|
|||||||
tokio-tungstenite = { version = "0.29", features = ["rustls-tls-webpki-roots"] }
|
tokio-tungstenite = { version = "0.29", features = ["rustls-tls-webpki-roots"] }
|
||||||
toml = "1.1"
|
toml = "1.1"
|
||||||
tracing = "0.1"
|
tracing = "0.1"
|
||||||
tracing-subscriber = { version = "0.3", features = [
|
|
||||||
"env-filter",
|
|
||||||
"fmt",
|
|
||||||
"time",
|
|
||||||
] }
|
|
||||||
time = { version = "0.3", features = ["formatting", "local-offset"] }
|
|
||||||
url = "2"
|
url = "2"
|
||||||
vt100 = { package = "wakey-vt100", path = "../vendor/vt100", registry = "gitea", version = "=0.16.2-wakey.1" }
|
vt100 = { package = "wakey-vt100", path = "../vendor/vt100", registry = "gitea", version = "=0.16.2-wakey.2" }
|
||||||
wakey = { path = "..", registry = "gitea", version = "0" }
|
wakey = { path = "..", registry = "gitea", version = "0" }
|
||||||
wakey-core = { path = "../wakey-core", registry = "gitea", version = "0" }
|
wakey-core = { path = "../wakey-core", registry = "gitea", version = "0", features = ["telemetry"] }
|
||||||
|
|
||||||
|
[target.'cfg(target_os = "linux")'.dependencies]
|
||||||
|
wakey-linux = { path = "../wakey-linux", registry = "gitea", version = "0", features = ["telemetry"] }
|
||||||
+234
-51
@@ -3,6 +3,8 @@ use serde::{Deserialize, Serialize};
|
|||||||
use std::fmt;
|
use std::fmt;
|
||||||
use std::path::{Path, PathBuf};
|
use std::path::{Path, PathBuf};
|
||||||
use std::sync::LazyLock;
|
use std::sync::LazyLock;
|
||||||
|
use std::time::Duration;
|
||||||
|
use wakey_core::config_duration::{self, LegacyDuration};
|
||||||
|
|
||||||
use crate::protocol::{
|
use crate::protocol::{
|
||||||
DEFAULT_TERMINAL_MAX_SESSIONS, DEFAULT_TERMINAL_SESSION_TTL_SECONDS,
|
DEFAULT_TERMINAL_MAX_SESSIONS, DEFAULT_TERMINAL_SESSION_TTL_SECONDS,
|
||||||
@@ -19,19 +21,25 @@ const DEFAULT_MAC_NAME_CACHE_PATH: &str = "/tmp/wakey_mac_names.json";
|
|||||||
const DEFAULT_OBSERVATION_STORE_PATH: &str = "/tmp/wakey_observations.json";
|
const DEFAULT_OBSERVATION_STORE_PATH: &str = "/tmp/wakey_observations.json";
|
||||||
pub const DEFAULT_OBSERVATION_RETENTION_DAYS: u64 = 7;
|
pub const DEFAULT_OBSERVATION_RETENTION_DAYS: u64 = 7;
|
||||||
|
|
||||||
|
// Derive canonical serde functions once; the trait adapter below translates
|
||||||
|
// legacy keys without maintaining another copy of every config field.
|
||||||
#[derive(Clone, Serialize, Deserialize, PartialEq, Eq)]
|
#[derive(Clone, Serialize, Deserialize, PartialEq, Eq)]
|
||||||
|
#[serde(remote = "Self")]
|
||||||
pub struct AgentConfig {
|
pub struct AgentConfig {
|
||||||
pub server_url: String,
|
pub server_url: String,
|
||||||
pub agent_id: String,
|
pub agent_id: String,
|
||||||
pub agent_token: String,
|
pub agent_token: String,
|
||||||
#[serde(default = "default_reconnect_base_ms")]
|
#[serde(default = "default_reconnect_base", with = "humantime_serde")]
|
||||||
pub reconnect_base_ms: u64,
|
pub reconnect_base: Duration,
|
||||||
#[serde(default = "default_reconnect_max_ms")]
|
#[serde(default = "default_reconnect_max", with = "humantime_serde")]
|
||||||
pub reconnect_max_ms: u64,
|
pub reconnect_max: Duration,
|
||||||
#[serde(default = "default_observation_sync_interval_seconds")]
|
#[serde(
|
||||||
pub observation_sync_interval_seconds: u64,
|
default = "default_observation_sync_interval",
|
||||||
#[serde(default = "default_observation_retention_days")]
|
with = "humantime_serde"
|
||||||
pub observation_retention_days: u64,
|
)]
|
||||||
|
pub observation_sync_interval: Duration,
|
||||||
|
#[serde(default = "default_observation_retention", with = "humantime_serde")]
|
||||||
|
pub observation_retention: Duration,
|
||||||
#[serde(default = "default_pid_file")]
|
#[serde(default = "default_pid_file")]
|
||||||
pub pid_file: PathBuf,
|
pub pid_file: PathBuf,
|
||||||
#[serde(default = "default_dhcp_leases_path")]
|
#[serde(default = "default_dhcp_leases_path")]
|
||||||
@@ -42,10 +50,12 @@ pub struct AgentConfig {
|
|||||||
pub observation_store_path: PathBuf,
|
pub observation_store_path: PathBuf,
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub terminal: TerminalConfig,
|
pub terminal: TerminalConfig,
|
||||||
|
#[serde(default)]
|
||||||
|
pub telemetry: wakey_core::telemetry::TelemetryConfig,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
|
#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
|
||||||
#[serde(deny_unknown_fields)]
|
#[serde(remote = "Self", deny_unknown_fields)]
|
||||||
pub struct TerminalConfig {
|
pub struct TerminalConfig {
|
||||||
#[serde(default)]
|
#[serde(default)]
|
||||||
pub enabled: bool,
|
pub enabled: bool,
|
||||||
@@ -58,8 +68,8 @@ pub struct TerminalConfig {
|
|||||||
#[serde(default = "default_terminal_max_sessions")]
|
#[serde(default = "default_terminal_max_sessions")]
|
||||||
pub max_sessions: usize,
|
pub max_sessions: usize,
|
||||||
/// Maximum lifetime of a PTY process. Zero disables automatic expiry.
|
/// Maximum lifetime of a PTY process. Zero disables automatic expiry.
|
||||||
#[serde(default = "default_terminal_session_ttl_seconds")]
|
#[serde(default = "default_terminal_session_ttl", with = "humantime_serde")]
|
||||||
pub session_ttl_seconds: u64,
|
pub session_ttl: Duration,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for TerminalConfig {
|
impl Default for TerminalConfig {
|
||||||
@@ -70,7 +80,7 @@ impl Default for TerminalConfig {
|
|||||||
args: Vec::new(),
|
args: Vec::new(),
|
||||||
current_dir: None,
|
current_dir: None,
|
||||||
max_sessions: default_terminal_max_sessions(),
|
max_sessions: default_terminal_max_sessions(),
|
||||||
session_ttl_seconds: default_terminal_session_ttl_seconds(),
|
session_ttl: default_terminal_session_ttl(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -79,15 +89,16 @@ pub static DEFAULT_CONFIG: LazyLock<AgentConfig> = LazyLock::new(|| AgentConfig
|
|||||||
server_url: "https://wakey.ldlda.com".to_string(),
|
server_url: "https://wakey.ldlda.com".to_string(),
|
||||||
agent_id: "REPLACE_ME_AGENT_ID".to_string(),
|
agent_id: "REPLACE_ME_AGENT_ID".to_string(),
|
||||||
agent_token: "REPLACE_ME_AGENT_TOKEN".to_string(),
|
agent_token: "REPLACE_ME_AGENT_TOKEN".to_string(),
|
||||||
reconnect_base_ms: default_reconnect_base_ms(),
|
reconnect_base: default_reconnect_base(),
|
||||||
reconnect_max_ms: default_reconnect_max_ms(),
|
reconnect_max: default_reconnect_max(),
|
||||||
observation_sync_interval_seconds: default_observation_sync_interval_seconds(),
|
observation_sync_interval: default_observation_sync_interval(),
|
||||||
observation_retention_days: default_observation_retention_days(),
|
observation_retention: default_observation_retention(),
|
||||||
pid_file: default_pid_file(),
|
pid_file: default_pid_file(),
|
||||||
dhcp_leases_path: default_dhcp_leases_path(),
|
dhcp_leases_path: default_dhcp_leases_path(),
|
||||||
mac_name_cache_path: default_mac_name_cache_path(),
|
mac_name_cache_path: default_mac_name_cache_path(),
|
||||||
observation_store_path: default_observation_store_path(),
|
observation_store_path: default_observation_store_path(),
|
||||||
terminal: TerminalConfig::default(),
|
terminal: TerminalConfig::default(),
|
||||||
|
telemetry: wakey_core::telemetry::TelemetryConfig::default(),
|
||||||
});
|
});
|
||||||
|
|
||||||
impl fmt::Debug for AgentConfig {
|
impl fmt::Debug for AgentConfig {
|
||||||
@@ -96,39 +107,34 @@ impl fmt::Debug for AgentConfig {
|
|||||||
.field("server_url", &self.server_url)
|
.field("server_url", &self.server_url)
|
||||||
.field("agent_id", &self.agent_id)
|
.field("agent_id", &self.agent_id)
|
||||||
.field("agent_token", &"<redacted>")
|
.field("agent_token", &"<redacted>")
|
||||||
.field("reconnect_base_ms", &self.reconnect_base_ms)
|
.field("reconnect_base", &self.reconnect_base)
|
||||||
.field("reconnect_max_ms", &self.reconnect_max_ms)
|
.field("reconnect_max", &self.reconnect_max)
|
||||||
.field(
|
.field("observation_sync_interval", &self.observation_sync_interval)
|
||||||
"observation_sync_interval_seconds",
|
.field("observation_retention", &self.observation_retention)
|
||||||
&self.observation_sync_interval_seconds,
|
|
||||||
)
|
|
||||||
.field(
|
|
||||||
"observation_retention_days",
|
|
||||||
&self.observation_retention_days,
|
|
||||||
)
|
|
||||||
.field("pid_file", &self.pid_file)
|
.field("pid_file", &self.pid_file)
|
||||||
.field("dhcp_leases_path", &self.dhcp_leases_path)
|
.field("dhcp_leases_path", &self.dhcp_leases_path)
|
||||||
.field("mac_name_cache_path", &self.mac_name_cache_path)
|
.field("mac_name_cache_path", &self.mac_name_cache_path)
|
||||||
.field("observation_store_path", &self.observation_store_path)
|
.field("observation_store_path", &self.observation_store_path)
|
||||||
.field("terminal", &self.terminal)
|
.field("terminal", &self.terminal)
|
||||||
|
.field("telemetry", &self.telemetry)
|
||||||
.finish()
|
.finish()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const fn default_reconnect_base_ms() -> u64 {
|
const fn default_reconnect_base() -> Duration {
|
||||||
1_000
|
Duration::from_secs(1)
|
||||||
}
|
}
|
||||||
|
|
||||||
const fn default_reconnect_max_ms() -> u64 {
|
const fn default_reconnect_max() -> Duration {
|
||||||
30_000
|
Duration::from_secs(30)
|
||||||
}
|
}
|
||||||
|
|
||||||
const fn default_observation_sync_interval_seconds() -> u64 {
|
const fn default_observation_sync_interval() -> Duration {
|
||||||
60
|
Duration::from_secs(60)
|
||||||
}
|
}
|
||||||
|
|
||||||
const fn default_observation_retention_days() -> u64 {
|
const fn default_observation_retention() -> Duration {
|
||||||
DEFAULT_OBSERVATION_RETENTION_DAYS
|
Duration::from_secs(DEFAULT_OBSERVATION_RETENTION_DAYS * 86_400)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn default_pid_file() -> PathBuf {
|
fn default_pid_file() -> PathBuf {
|
||||||
@@ -155,17 +161,43 @@ const fn default_terminal_max_sessions() -> usize {
|
|||||||
DEFAULT_TERMINAL_MAX_SESSIONS
|
DEFAULT_TERMINAL_MAX_SESSIONS
|
||||||
}
|
}
|
||||||
|
|
||||||
const fn default_terminal_session_ttl_seconds() -> u64 {
|
const fn default_terminal_session_ttl() -> Duration {
|
||||||
DEFAULT_TERMINAL_SESSION_TTL_SECONDS
|
Duration::from_secs(DEFAULT_TERMINAL_SESSION_TTL_SECONDS)
|
||||||
}
|
}
|
||||||
|
|
||||||
impl TerminalConfig {
|
impl TerminalConfig {
|
||||||
pub(crate) fn session_ttl(&self) -> Result<Option<std::time::Duration>> {
|
pub(crate) fn session_ttl(&self) -> Result<Option<std::time::Duration>> {
|
||||||
checked_terminal_session_ttl(self.session_ttl_seconds).map_err(anyhow::Error::msg)
|
anyhow::ensure!(
|
||||||
|
self.session_ttl.subsec_nanos() == 0,
|
||||||
|
"terminal.session_ttl must use whole seconds (the terminal protocol uses seconds)"
|
||||||
|
);
|
||||||
|
checked_terminal_session_ttl(self.session_ttl.as_secs()).map_err(anyhow::Error::msg)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl AgentConfig {
|
impl AgentConfig {
|
||||||
|
/// Validate scheduling bounds and second-based storage/protocol boundaries.
|
||||||
|
/// Zero is allowed for hook retention and terminal TTL, not active timers.
|
||||||
|
fn validate(&self) -> Result<()> {
|
||||||
|
for (name, duration) in [
|
||||||
|
("reconnect_base", self.reconnect_base),
|
||||||
|
("reconnect_max", self.reconnect_max),
|
||||||
|
("observation_sync_interval", self.observation_sync_interval),
|
||||||
|
] {
|
||||||
|
anyhow::ensure!(!duration.is_zero(), "{name} must be positive");
|
||||||
|
anyhow::ensure!(
|
||||||
|
std::time::Instant::now().checked_add(duration).is_some(),
|
||||||
|
"{name} exceeds the platform timer range"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
anyhow::ensure!(
|
||||||
|
self.observation_retention.subsec_nanos() == 0,
|
||||||
|
"observation_retention must use whole seconds (hook timestamps use seconds)"
|
||||||
|
);
|
||||||
|
self.terminal.session_ttl()?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
pub fn local_path_envs(&self) -> Vec<(&'static str, &Path)> {
|
pub fn local_path_envs(&self) -> Vec<(&'static str, &Path)> {
|
||||||
vec![
|
vec![
|
||||||
(WAKEY_DHCP_LEASES_ENV, self.dhcp_leases_path.as_path()),
|
(WAKEY_DHCP_LEASES_ENV, self.dhcp_leases_path.as_path()),
|
||||||
@@ -187,13 +219,99 @@ pub fn apply_local_path_env_to_command(cmd: &mut std::process::Command, config:
|
|||||||
pub fn load_config(path: &Path) -> Result<AgentConfig> {
|
pub fn load_config(path: &Path) -> Result<AgentConfig> {
|
||||||
let content = std::fs::read_to_string(path)
|
let content = std::fs::read_to_string(path)
|
||||||
.with_context(|| format!("failed to read agent config {}", path.display()))?;
|
.with_context(|| format!("failed to read agent config {}", path.display()))?;
|
||||||
let config: AgentConfig = toml::from_str(&content)
|
let raw: toml::Value = toml::from_str(&content)
|
||||||
.with_context(|| format!("failed to parse agent config {}", path.display()))?;
|
.with_context(|| format!("failed to parse agent config {}", path.display()))?;
|
||||||
config.terminal.session_ttl()?;
|
let mut value = serde_json::to_value(raw)?;
|
||||||
|
let mut deprecated = config_duration::normalize(&mut value, AGENT_DURATION_KEYS)
|
||||||
|
.map_err(anyhow::Error::msg)?
|
||||||
|
.into_iter()
|
||||||
|
.map(|key| ("", key))
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
if let Some(terminal) = value.get_mut("terminal") {
|
||||||
|
deprecated.extend(
|
||||||
|
config_duration::normalize(terminal, TERMINAL_DURATION_KEYS)
|
||||||
|
.map_err(anyhow::Error::msg)?
|
||||||
|
.into_iter()
|
||||||
|
.map(|key| ("terminal.", key)),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let config: AgentConfig = serde_json::from_value(value)
|
||||||
|
.with_context(|| format!("failed to parse agent config {}", path.display()))?;
|
||||||
|
config.validate()?;
|
||||||
|
for (prefix, key) in deprecated {
|
||||||
|
// Initial config loading precedes tracing setup; stderr also keeps
|
||||||
|
// init-config stdout valid TOML and never prints config secrets.
|
||||||
|
eprintln!(
|
||||||
|
"warning: {}: `{prefix}{}` is deprecated; use `{prefix}{}` with a duration string",
|
||||||
|
path.display(),
|
||||||
|
key.old,
|
||||||
|
key.new
|
||||||
|
);
|
||||||
|
}
|
||||||
Ok(config)
|
Ok(config)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const AGENT_DURATION_KEYS: &[LegacyDuration] = &[
|
||||||
|
LegacyDuration {
|
||||||
|
old: "reconnect_base_ms",
|
||||||
|
new: "reconnect_base",
|
||||||
|
unit: Duration::from_millis(1),
|
||||||
|
},
|
||||||
|
LegacyDuration {
|
||||||
|
old: "reconnect_max_ms",
|
||||||
|
new: "reconnect_max",
|
||||||
|
unit: Duration::from_millis(1),
|
||||||
|
},
|
||||||
|
LegacyDuration {
|
||||||
|
old: "observation_sync_interval_seconds",
|
||||||
|
new: "observation_sync_interval",
|
||||||
|
unit: Duration::from_secs(1),
|
||||||
|
},
|
||||||
|
LegacyDuration {
|
||||||
|
old: "observation_retention_days",
|
||||||
|
new: "observation_retention",
|
||||||
|
unit: Duration::from_secs(86_400),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
const TERMINAL_DURATION_KEYS: &[LegacyDuration] = &[LegacyDuration {
|
||||||
|
old: "session_ttl_seconds",
|
||||||
|
new: "session_ttl",
|
||||||
|
unit: Duration::from_secs(1),
|
||||||
|
}];
|
||||||
|
|
||||||
|
impl Serialize for AgentConfig {
|
||||||
|
fn serialize<S: serde::Serializer>(
|
||||||
|
&self,
|
||||||
|
serializer: S,
|
||||||
|
) -> std::result::Result<S::Ok, S::Error> {
|
||||||
|
Self::serialize(self, serializer)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl<'de> Deserialize<'de> for AgentConfig {
|
||||||
|
fn deserialize<D: serde::Deserializer<'de>>(
|
||||||
|
deserializer: D,
|
||||||
|
) -> std::result::Result<Self, D::Error> {
|
||||||
|
config_duration::deserialize_legacy(deserializer, AGENT_DURATION_KEYS, Self::deserialize)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl Serialize for TerminalConfig {
|
||||||
|
fn serialize<S: serde::Serializer>(
|
||||||
|
&self,
|
||||||
|
serializer: S,
|
||||||
|
) -> std::result::Result<S::Ok, S::Error> {
|
||||||
|
Self::serialize(self, serializer)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl<'de> Deserialize<'de> for TerminalConfig {
|
||||||
|
fn deserialize<D: serde::Deserializer<'de>>(
|
||||||
|
deserializer: D,
|
||||||
|
) -> std::result::Result<Self, D::Error> {
|
||||||
|
config_duration::deserialize_legacy(deserializer, TERMINAL_DURATION_KEYS, Self::deserialize)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub fn save_config(path: &Path, config: &AgentConfig) -> Result<()> {
|
pub fn save_config(path: &Path, config: &AgentConfig) -> Result<()> {
|
||||||
|
config.validate()?;
|
||||||
let content = toml::to_string_pretty(config).context("failed to serialize agent config")?;
|
let content = toml::to_string_pretty(config).context("failed to serialize agent config")?;
|
||||||
write_config_atomically(path, &content)
|
write_config_atomically(path, &content)
|
||||||
}
|
}
|
||||||
@@ -258,6 +376,62 @@ fn write_config_atomically(path: &Path, content: &str) -> Result<()> {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
|
const HEADER: &str =
|
||||||
|
"server_url = 'https://example.com'\nagent_id = 'agent-1'\nagent_token = 'secret'\n";
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn legacy_duration_keys_convert_and_serialize_canonically() {
|
||||||
|
let legacy: AgentConfig = toml::from_str(&format!("{HEADER}\nreconnect_base_ms = 123\nreconnect_max_ms = 456\nobservation_sync_interval_seconds = 7\nobservation_retention_days = 3\n[terminal]\nsession_ttl_seconds = 600\n")).unwrap();
|
||||||
|
let canonical: AgentConfig = toml::from_str(&format!("{HEADER}\nreconnect_base = '123ms'\nreconnect_max = '456ms'\nobservation_sync_interval = '7s'\nobservation_retention = '3d'\n[terminal]\nsession_ttl = '10m'\n")).unwrap();
|
||||||
|
assert_eq!(legacy, canonical);
|
||||||
|
legacy.validate().unwrap();
|
||||||
|
let output = toml::to_string(&legacy).unwrap();
|
||||||
|
for key in AGENT_DURATION_KEYS {
|
||||||
|
assert!(!output.contains(key.old));
|
||||||
|
assert!(output.contains(key.new));
|
||||||
|
}
|
||||||
|
assert!(!output.contains("session_ttl_seconds"));
|
||||||
|
assert_eq!(toml::from_str::<AgentConfig>(&output).unwrap(), canonical);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn config_rejects_ambiguous_units_and_invalid_timer_values() {
|
||||||
|
for fields in [
|
||||||
|
"reconnect_base = '1s'\nreconnect_base_ms = 1000",
|
||||||
|
"observation_retention = '7d'\nobservation_retention_days = 7",
|
||||||
|
"reconnect_base = 1000",
|
||||||
|
"observation_retention_days = -1",
|
||||||
|
"observation_retention_days = 9223372036854775807",
|
||||||
|
"reconnect_base = 'nonsense'",
|
||||||
|
"[terminal]\nsession_ttl = '1h'\nsession_ttl_seconds = 3600",
|
||||||
|
] {
|
||||||
|
assert!(
|
||||||
|
toml::from_str::<AgentConfig>(&format!("{HEADER}{fields}")).is_err(),
|
||||||
|
"{fields}"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
for fields in [
|
||||||
|
"observation_sync_interval = '0s'",
|
||||||
|
"reconnect_base = '0s'",
|
||||||
|
"[terminal]\nsession_ttl = '500ms'",
|
||||||
|
] {
|
||||||
|
let config: AgentConfig = toml::from_str(&format!("{HEADER}{fields}")).unwrap();
|
||||||
|
assert!(config.validate().is_err(), "{fields}");
|
||||||
|
}
|
||||||
|
let config: AgentConfig = toml::from_str(&format!(
|
||||||
|
"{HEADER}observation_retention = '0d'\n[terminal]\nsession_ttl = '0s'"
|
||||||
|
))
|
||||||
|
.unwrap();
|
||||||
|
config.validate().unwrap();
|
||||||
|
assert!(config.observation_retention.is_zero());
|
||||||
|
assert_eq!(config.terminal.session_ttl().unwrap(), None);
|
||||||
|
let oversized = AgentConfig {
|
||||||
|
reconnect_max: Duration::from_secs(u64::MAX),
|
||||||
|
..config
|
||||||
|
};
|
||||||
|
assert!(oversized.validate().is_err());
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn config_roundtrip() {
|
fn config_roundtrip() {
|
||||||
let dir = std::env::temp_dir().join(format!("wakey-agent-config-{}", std::process::id()));
|
let dir = std::env::temp_dir().join(format!("wakey-agent-config-{}", std::process::id()));
|
||||||
@@ -266,21 +440,30 @@ mod tests {
|
|||||||
server_url: "https://example.com".into(),
|
server_url: "https://example.com".into(),
|
||||||
agent_id: "agent-1".into(),
|
agent_id: "agent-1".into(),
|
||||||
agent_token: "secret".into(),
|
agent_token: "secret".into(),
|
||||||
reconnect_base_ms: 123,
|
reconnect_base: Duration::from_millis(123),
|
||||||
reconnect_max_ms: 456,
|
reconnect_max: Duration::from_millis(456),
|
||||||
observation_sync_interval_seconds: 7,
|
observation_sync_interval: Duration::from_secs(7),
|
||||||
observation_retention_days: 3,
|
observation_retention: Duration::from_secs(3 * 86_400),
|
||||||
pid_file: "/tmp/test-wakey-agent.pid".into(),
|
pid_file: "/tmp/test-wakey-agent.pid".into(),
|
||||||
dhcp_leases_path: "/tmp/test-dhcp.leases".into(),
|
dhcp_leases_path: "/tmp/test-dhcp.leases".into(),
|
||||||
mac_name_cache_path: "/tmp/test-names.json".into(),
|
mac_name_cache_path: "/tmp/test-names.json".into(),
|
||||||
observation_store_path: "/tmp/test-observations.json".into(),
|
observation_store_path: "/tmp/test-observations.json".into(),
|
||||||
|
telemetry: wakey_core::telemetry::TelemetryConfig {
|
||||||
|
otlp_endpoint: Some("http://10.0.0.1:4317".into()),
|
||||||
|
authorization_env: Some("WAKEY_OTLP_AUTHORIZATION".into()),
|
||||||
|
service_name: "router-agent".into(),
|
||||||
|
json_logs: true,
|
||||||
|
console_filter: Some("info".into()),
|
||||||
|
otlp_log_filter: Some("debug".into()),
|
||||||
|
trace_filter: Some("info".into()),
|
||||||
|
},
|
||||||
terminal: TerminalConfig {
|
terminal: TerminalConfig {
|
||||||
enabled: true,
|
enabled: true,
|
||||||
shell: "/bin/sh".into(),
|
shell: "/bin/sh".into(),
|
||||||
args: vec!["-l".into()],
|
args: vec!["-l".into()],
|
||||||
current_dir: Some("/tmp".into()),
|
current_dir: Some("/tmp".into()),
|
||||||
max_sessions: 2,
|
max_sessions: 2,
|
||||||
session_ttl_seconds: 600,
|
session_ttl: Duration::from_secs(600),
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -310,8 +493,8 @@ agent_token = "secret"
|
|||||||
.expect("config should parse");
|
.expect("config should parse");
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
config.observation_retention_days,
|
config.observation_retention,
|
||||||
DEFAULT_OBSERVATION_RETENTION_DAYS
|
DEFAULT_CONFIG.observation_retention
|
||||||
);
|
);
|
||||||
assert_eq!(config.terminal, TerminalConfig::default());
|
assert_eq!(config.terminal, TerminalConfig::default());
|
||||||
}
|
}
|
||||||
@@ -335,8 +518,8 @@ max_sessions = 2
|
|||||||
assert!(config.terminal.args.is_empty());
|
assert!(config.terminal.args.is_empty());
|
||||||
assert!(config.terminal.current_dir.is_none());
|
assert!(config.terminal.current_dir.is_none());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
config.terminal.session_ttl_seconds,
|
config.terminal.session_ttl,
|
||||||
DEFAULT_TERMINAL_SESSION_TTL_SECONDS
|
TerminalConfig::default().session_ttl
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -361,18 +544,18 @@ max_session = 67
|
|||||||
#[test]
|
#[test]
|
||||||
fn terminal_ttl_zero_is_unlimited_and_positive_is_exact() {
|
fn terminal_ttl_zero_is_unlimited_and_positive_is_exact() {
|
||||||
let mut terminal = TerminalConfig {
|
let mut terminal = TerminalConfig {
|
||||||
session_ttl_seconds: 0,
|
session_ttl: Duration::ZERO,
|
||||||
..TerminalConfig::default()
|
..TerminalConfig::default()
|
||||||
};
|
};
|
||||||
assert_eq!(terminal.session_ttl().expect("unlimited TTL"), None);
|
assert_eq!(terminal.session_ttl().expect("unlimited TTL"), None);
|
||||||
|
|
||||||
terminal.session_ttl_seconds = 67;
|
terminal.session_ttl = Duration::from_secs(67);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
terminal.session_ttl().expect("finite TTL"),
|
terminal.session_ttl().expect("finite TTL"),
|
||||||
Some(std::time::Duration::from_secs(67))
|
Some(std::time::Duration::from_secs(67))
|
||||||
);
|
);
|
||||||
|
|
||||||
terminal.session_ttl_seconds = u64::MAX;
|
terminal.session_ttl = Duration::from_secs(u64::MAX);
|
||||||
assert!(terminal.session_ttl().is_err());
|
assert!(terminal.session_ttl().is_err());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -1,5 +1,5 @@
|
|||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
use tracing::{debug, info, instrument, warn};
|
use tracing::{Instrument, debug, info, instrument, warn};
|
||||||
|
|
||||||
use crate::config::AgentConfig;
|
use crate::config::AgentConfig;
|
||||||
use crate::protocol::{
|
use crate::protocol::{
|
||||||
@@ -69,22 +69,31 @@ async fn dispatch_inventory(req: InventoryRequest, config: &AgentConfig) -> Resu
|
|||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[instrument(name = "agent_inventory", skip_all, fields(neighbors = tracing::field::Empty, leases = tracing::field::Empty, observations = tracing::field::Empty, devices = tracing::field::Empty))]
|
||||||
pub async fn inventory_for_config(
|
pub async fn inventory_for_config(
|
||||||
req: InventoryRequest,
|
req: InventoryRequest,
|
||||||
config: &AgentConfig,
|
config: &AgentConfig,
|
||||||
) -> Result<wakey_core::DeviceInventory> {
|
) -> Result<wakey_core::DeviceInventory> {
|
||||||
let query = req.into_inventory_query();
|
let query = req.into_inventory_query();
|
||||||
let neighbors = wakey::wakey_linux::devices::query_neighbors(&query).await?;
|
// This configured path serves both commands and periodic snapshots; keep
|
||||||
|
// phase spans here so neither caller bypasses source timing.
|
||||||
|
let neighbors = wakey::wakey_linux::devices::query_neighbors(&query)
|
||||||
|
.instrument(tracing::info_span!("inventory_neighbors"))
|
||||||
|
.await?;
|
||||||
|
tracing::Span::current().record("neighbors", neighbors.len());
|
||||||
let leases = wakey::wakey_linux::dhcp::read_dhcp_leases_with_names_from_paths(
|
let leases = wakey::wakey_linux::dhcp::read_dhcp_leases_with_names_from_paths(
|
||||||
&config.dhcp_leases_path,
|
&config.dhcp_leases_path,
|
||||||
&config.observation_store_path,
|
&config.observation_store_path,
|
||||||
&config.mac_name_cache_path,
|
&config.mac_name_cache_path,
|
||||||
)
|
)
|
||||||
|
.instrument(tracing::info_span!("inventory_dhcp_leases"))
|
||||||
.await?;
|
.await?;
|
||||||
match wakey::wakey_linux::observations::prune_stale_observations_from_path(
|
tracing::Span::current().record("leases", leases.len());
|
||||||
|
match wakey::wakey_linux::observations::prune_expired(
|
||||||
&config.observation_store_path,
|
&config.observation_store_path,
|
||||||
config.observation_retention_days,
|
config.observation_retention,
|
||||||
)
|
)
|
||||||
|
.instrument(tracing::info_span!("inventory_observation_prune"))
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(removed) if removed > 0 => {
|
Ok(removed) if removed > 0 => {
|
||||||
@@ -98,6 +107,7 @@ pub async fn inventory_for_config(
|
|||||||
let observations = match wakey::wakey_linux::observations::list_local_observations_from_path(
|
let observations = match wakey::wakey_linux::observations::list_local_observations_from_path(
|
||||||
&config.observation_store_path,
|
&config.observation_store_path,
|
||||||
)
|
)
|
||||||
|
.instrument(tracing::info_span!("inventory_hook_store"))
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(observations) => observations
|
Ok(observations) => observations
|
||||||
@@ -109,6 +119,7 @@ pub async fn inventory_for_config(
|
|||||||
Vec::new()
|
Vec::new()
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
tracing::Span::current().record("observations", observations.len());
|
||||||
debug!(
|
debug!(
|
||||||
neighbors = neighbors.len(),
|
neighbors = neighbors.len(),
|
||||||
leases = leases.len(),
|
leases = leases.len(),
|
||||||
@@ -123,6 +134,7 @@ pub async fn inventory_for_config(
|
|||||||
&query,
|
&query,
|
||||||
),
|
),
|
||||||
};
|
};
|
||||||
|
tracing::Span::current().record("devices", inventory.devices.len());
|
||||||
debug!(
|
debug!(
|
||||||
rows = inventory.devices.len(),
|
rows = inventory.devices.len(),
|
||||||
"dispatched inventory command"
|
"dispatched inventory command"
|
||||||
|
|||||||
@@ -143,21 +143,28 @@ mod tests {
|
|||||||
server_url: "https://old.example.com".into(),
|
server_url: "https://old.example.com".into(),
|
||||||
agent_id: "old-agent".into(),
|
agent_id: "old-agent".into(),
|
||||||
agent_token: "old-token".into(),
|
agent_token: "old-token".into(),
|
||||||
reconnect_base_ms: 2_000,
|
reconnect_base: std::time::Duration::from_secs(2),
|
||||||
reconnect_max_ms: 60_000,
|
reconnect_max: std::time::Duration::from_secs(60),
|
||||||
observation_sync_interval_seconds: 30,
|
observation_sync_interval: std::time::Duration::from_secs(30),
|
||||||
observation_retention_days: 11,
|
observation_retention: std::time::Duration::from_secs(11 * 86_400),
|
||||||
pid_file: "/tmp/custom-wakey-agent.pid".into(),
|
pid_file: "/tmp/custom-wakey-agent.pid".into(),
|
||||||
dhcp_leases_path: "/tmp/custom-dhcp.leases".into(),
|
dhcp_leases_path: "/tmp/custom-dhcp.leases".into(),
|
||||||
mac_name_cache_path: "/tmp/custom-names.json".into(),
|
mac_name_cache_path: "/tmp/custom-names.json".into(),
|
||||||
observation_store_path: "/tmp/custom-observations.json".into(),
|
observation_store_path: "/tmp/custom-observations.json".into(),
|
||||||
|
telemetry: wakey_core::telemetry::TelemetryConfig {
|
||||||
|
otlp_endpoint: Some("http://10.0.0.1:4317".into()),
|
||||||
|
authorization_env: Some("WAKEY_OTLP_AUTHORIZATION".into()),
|
||||||
|
service_name: "router-agent".into(),
|
||||||
|
json_logs: true,
|
||||||
|
..Default::default()
|
||||||
|
},
|
||||||
terminal: crate::config::TerminalConfig {
|
terminal: crate::config::TerminalConfig {
|
||||||
enabled: true,
|
enabled: true,
|
||||||
shell: "/bin/ash".into(),
|
shell: "/bin/ash".into(),
|
||||||
args: vec!["-l".into()],
|
args: vec!["-l".into()],
|
||||||
current_dir: Some("/root".into()),
|
current_dir: Some("/root".into()),
|
||||||
max_sessions: 2,
|
max_sessions: 2,
|
||||||
session_ttl_seconds: 600,
|
session_ttl: std::time::Duration::from_secs(600),
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -169,7 +176,10 @@ mod tests {
|
|||||||
assert_eq!(config.agent_id, "agent-123");
|
assert_eq!(config.agent_id, "agent-123");
|
||||||
assert_eq!(config.agent_token, "token-xyz");
|
assert_eq!(config.agent_token, "token-xyz");
|
||||||
assert_eq!(config.server_url, "https://control.example.com");
|
assert_eq!(config.server_url, "https://control.example.com");
|
||||||
assert_eq!(config.observation_retention_days, 11);
|
assert_eq!(
|
||||||
|
config.observation_retention,
|
||||||
|
base_config.observation_retention
|
||||||
|
);
|
||||||
assert_eq!(config.pid_file, base_config.pid_file);
|
assert_eq!(config.pid_file, base_config.pid_file);
|
||||||
assert_eq!(config.dhcp_leases_path, base_config.dhcp_leases_path);
|
assert_eq!(config.dhcp_leases_path, base_config.dhcp_leases_path);
|
||||||
assert_eq!(config.terminal, base_config.terminal);
|
assert_eq!(config.terminal, base_config.terminal);
|
||||||
|
|||||||
+33
-2
@@ -15,9 +15,37 @@ use wakey_agent::protocol;
|
|||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> Result<()> {
|
async fn main() -> Result<()> {
|
||||||
let cli = Cli::parse();
|
let cli = Cli::parse();
|
||||||
tracing::init(cli.verbose);
|
|
||||||
let global_config = cli.config.as_deref();
|
let global_config = cli.config.as_deref();
|
||||||
|
|
||||||
|
// Only the long-lived daemon exports: hotplug/enroll invocations stay cheap.
|
||||||
|
let daemon_config = if let Command::Serve(args) = &cli.command {
|
||||||
|
let path = global_config.unwrap_or(&args.config);
|
||||||
|
// Let serve report its enrollment/setup instructions for a missing file.
|
||||||
|
if path.exists() {
|
||||||
|
Some(config::load_config(path)?)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
let local_only = wakey_core::telemetry::TelemetryConfig::default();
|
||||||
|
let telemetry = tracing::init(
|
||||||
|
cli.verbose,
|
||||||
|
daemon_config
|
||||||
|
.as_ref()
|
||||||
|
.map(|c| &c.telemetry)
|
||||||
|
.unwrap_or(&local_only),
|
||||||
|
daemon_config.as_ref().map(|c| c.agent_id.as_str()),
|
||||||
|
)?;
|
||||||
|
|
||||||
|
let result = run(cli).await;
|
||||||
|
tokio::task::spawn_blocking(move || telemetry.shutdown()).await?;
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn run(cli: Cli) -> Result<()> {
|
||||||
|
let global_config = cli.config.as_deref();
|
||||||
match cli.command {
|
match cli.command {
|
||||||
Command::Serve(mut args) => {
|
Command::Serve(mut args) => {
|
||||||
if let Some(config) = global_config {
|
if let Some(config) = global_config {
|
||||||
@@ -262,7 +290,10 @@ fn init_config(args: InitConfigArgs) -> Result<()> {
|
|||||||
cfg.agent_token = agent_token;
|
cfg.agent_token = agent_token;
|
||||||
}
|
}
|
||||||
if let Some(days) = args.observation_retention_days {
|
if let Some(days) = args.observation_retention_days {
|
||||||
cfg.observation_retention_days = days;
|
cfg.observation_retention = std::time::Duration::from_secs(
|
||||||
|
days.checked_mul(86_400)
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("observation retention is too large"))?,
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(path) = &args.config {
|
if let Some(path) = &args.config {
|
||||||
|
|||||||
@@ -301,6 +301,9 @@ pub enum ClientMessage {
|
|||||||
DeviceSnapshot {
|
DeviceSnapshot {
|
||||||
agent_id: String,
|
agent_id: String,
|
||||||
devices: Vec<Device>,
|
devices: Vec<Device>,
|
||||||
|
/// W3C trace context for the collection that produced this snapshot.
|
||||||
|
#[serde(default, skip_serializing_if = "std::collections::BTreeMap::is_empty")]
|
||||||
|
trace_context: std::collections::BTreeMap<String, String>,
|
||||||
},
|
},
|
||||||
Result {
|
Result {
|
||||||
request_id: RequestId,
|
request_id: RequestId,
|
||||||
@@ -325,6 +328,8 @@ pub enum ServerMessage {
|
|||||||
Command {
|
Command {
|
||||||
request_id: RequestId,
|
request_id: RequestId,
|
||||||
command: AgentCommand,
|
command: AgentCommand,
|
||||||
|
#[serde(default, skip_serializing_if = "std::collections::BTreeMap::is_empty")]
|
||||||
|
trace_context: std::collections::BTreeMap<String, String>,
|
||||||
},
|
},
|
||||||
SyncDeviceSnapshot,
|
SyncDeviceSnapshot,
|
||||||
SyncTerminalSessions,
|
SyncTerminalSessions,
|
||||||
@@ -354,6 +359,7 @@ mod tests {
|
|||||||
command: AgentCommand::Leases(LeasesRequest {
|
command: AgentCommand::Leases(LeasesRequest {
|
||||||
include_state: true,
|
include_state: true,
|
||||||
}),
|
}),
|
||||||
|
trace_context: Default::default(),
|
||||||
};
|
};
|
||||||
|
|
||||||
let json = serde_json::to_string(&msg).expect("serialize");
|
let json = serde_json::to_string(&msg).expect("serialize");
|
||||||
@@ -367,6 +373,36 @@ mod tests {
|
|||||||
assert!(err.contains("must not be empty"));
|
assert!(err.contains("must not be empty"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn old_commands_without_trace_context_remain_valid() {
|
||||||
|
let message: ServerMessage = serde_json::from_str(r#"{"type":"command","request_id":"req-old","command":{"kind":"leases","include_state":false}}"#).expect("old command");
|
||||||
|
let ServerMessage::Command { trace_context, .. } = message else {
|
||||||
|
panic!("command expected")
|
||||||
|
};
|
||||||
|
assert!(trace_context.is_empty());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn command_trace_context_roundtrips() {
|
||||||
|
let message = ServerMessage::Command {
|
||||||
|
request_id: RequestId::try_from("req-traced".to_string()).unwrap(),
|
||||||
|
command: AgentCommand::Leases(LeasesRequest {
|
||||||
|
include_state: false,
|
||||||
|
}),
|
||||||
|
trace_context: [(
|
||||||
|
"traceparent".into(),
|
||||||
|
"00-0123456789abcdef0123456789abcdef-0123456789abcdef-01".into(),
|
||||||
|
)]
|
||||||
|
.into(),
|
||||||
|
};
|
||||||
|
let ServerMessage::Command { trace_context, .. } =
|
||||||
|
serde_json::from_str(&serde_json::to_string(&message).unwrap()).unwrap()
|
||||||
|
else {
|
||||||
|
panic!("command expected")
|
||||||
|
};
|
||||||
|
assert!(trace_context.contains_key("traceparent"));
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn terminal_id_deserialization_enforces_validation() {
|
fn terminal_id_deserialization_enforces_validation() {
|
||||||
let valid: TerminalId = serde_json::from_str(r#""term-1""#).expect("valid terminal id");
|
let valid: TerminalId = serde_json::from_str(r#""term-1""#).expect("valid terminal id");
|
||||||
@@ -394,11 +430,13 @@ mod tests {
|
|||||||
let msg = ClientMessage::DeviceSnapshot {
|
let msg = ClientMessage::DeviceSnapshot {
|
||||||
agent_id: "agent-a".into(),
|
agent_id: "agent-a".into(),
|
||||||
devices: vec![],
|
devices: vec![],
|
||||||
|
trace_context: Default::default(),
|
||||||
};
|
};
|
||||||
|
|
||||||
let json = serde_json::to_string(&msg).expect("serialize");
|
let json = serde_json::to_string(&msg).expect("serialize");
|
||||||
assert!(json.contains("\"type\":\"device_snapshot\""));
|
assert!(json.contains("\"type\":\"device_snapshot\""));
|
||||||
assert!(json.contains("\"devices\""));
|
assert!(json.contains("\"devices\""));
|
||||||
|
assert!(!json.contains("trace_context"));
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
|||||||
@@ -29,6 +29,8 @@ pub async fn serve(args: ServeArgs) -> Result<()> {
|
|||||||
{
|
{
|
||||||
use tokio::signal::unix::{SignalKind, signal};
|
use tokio::signal::unix::{SignalKind, signal};
|
||||||
let mut hup = signal(SignalKind::hangup()).context("failed to install SIGHUP handler")?;
|
let mut hup = signal(SignalKind::hangup()).context("failed to install SIGHUP handler")?;
|
||||||
|
let mut terminate =
|
||||||
|
signal(SignalKind::terminate()).context("failed to install SIGTERM handler")?;
|
||||||
|
|
||||||
loop {
|
loop {
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
@@ -37,9 +39,17 @@ pub async fn serve(args: ServeArgs) -> Result<()> {
|
|||||||
worker.abort();
|
worker.abort();
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
_ = terminate.recv() => {
|
||||||
|
info!("SIGTERM received; shutting down wakey-agent");
|
||||||
|
worker.abort();
|
||||||
|
break;
|
||||||
|
}
|
||||||
_ = hup.recv() => {
|
_ = hup.recv() => {
|
||||||
match config::load_config(&args.config) {
|
match config::load_config(&args.config) {
|
||||||
Ok(new_cfg) => {
|
Ok(new_cfg) => {
|
||||||
|
if new_cfg.telemetry != cfg.telemetry {
|
||||||
|
warn!("telemetry configuration changed; restart the daemon to apply it");
|
||||||
|
}
|
||||||
cfg = new_cfg;
|
cfg = new_cfg;
|
||||||
info!(config_path = %args.config.display(), agent_id = %cfg.agent_id, "reload requested; restarting session with updated config");
|
info!(config_path = %args.config.display(), agent_id = %cfg.agent_id, "reload requested; restarting session with updated config");
|
||||||
worker.abort();
|
worker.abort();
|
||||||
@@ -67,6 +77,8 @@ pub async fn serve(args: ServeArgs) -> Result<()> {
|
|||||||
worker.abort();
|
worker.abort();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Await the cancelled session task before flushing its telemetry.
|
||||||
|
let _ = worker.await;
|
||||||
let _ = remove_pid_file(&pid_file);
|
let _ = remove_pid_file(&pid_file);
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
+65
-33
@@ -4,7 +4,7 @@ use std::net::IpAddr;
|
|||||||
use std::time::Instant;
|
use std::time::Instant;
|
||||||
use tokio::time::{Duration, MissedTickBehavior, interval, sleep};
|
use tokio::time::{Duration, MissedTickBehavior, interval, sleep};
|
||||||
use tokio_tungstenite::{connect_async, tungstenite::Message};
|
use tokio_tungstenite::{connect_async, tungstenite::Message};
|
||||||
use tracing::{debug, error, info, info_span, warn};
|
use tracing::{Instrument, debug, error, info, info_span, warn};
|
||||||
|
|
||||||
use crate::config::AgentConfig;
|
use crate::config::AgentConfig;
|
||||||
use crate::dispatch::{dispatch_command, inventory_for_config};
|
use crate::dispatch::{dispatch_command, inventory_for_config};
|
||||||
@@ -18,16 +18,16 @@ pub async fn run(config: AgentConfig) -> Result<()> {
|
|||||||
// Terminal workers belong to the agent process, not a single control socket.
|
// Terminal workers belong to the agent process, not a single control socket.
|
||||||
// Keeping this manager outside the reconnect loop lets PTYs survive CC loss.
|
// Keeping this manager outside the reconnect loop lets PTYs survive CC loss.
|
||||||
let (terminal_manager, mut terminal_events) = TerminalManager::new(&config);
|
let (terminal_manager, mut terminal_events) = TerminalManager::new(&config);
|
||||||
let mut backoff = config.reconnect_base_ms.max(100);
|
let mut backoff = config.reconnect_base.max(Duration::from_millis(100));
|
||||||
loop {
|
loop {
|
||||||
match run_once(&config, &terminal_manager, &mut terminal_events).await {
|
match run_once(&config, &terminal_manager, &mut terminal_events).await {
|
||||||
Ok(()) => {
|
Ok(()) => {
|
||||||
backoff = config.reconnect_base_ms.max(100);
|
backoff = config.reconnect_base.max(Duration::from_millis(100));
|
||||||
}
|
}
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
warn!(error = %err, backoff_ms = backoff, "agent session ended; reconnecting");
|
warn!(agent_id = %config.agent_id, error = %err, backoff_ms = backoff.as_millis() as u64, "agent session ended; reconnecting");
|
||||||
sleep(Duration::from_millis(backoff)).await;
|
sleep(backoff).await;
|
||||||
backoff = next_backoff_ms(backoff, config.reconnect_max_ms);
|
backoff = next_backoff(backoff, config.reconnect_max);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -40,20 +40,14 @@ async fn run_once(
|
|||||||
crate::terminal::TerminalManagerEvent,
|
crate::terminal::TerminalManagerEvent,
|
||||||
>,
|
>,
|
||||||
) -> Result<()> {
|
) -> Result<()> {
|
||||||
let session_id = format!(
|
// Connection-long spans export too late and entered guards are unsafe across
|
||||||
"{}-{}",
|
// awaits. Short phase spans export while a healthy socket is still running.
|
||||||
std::process::id(),
|
|
||||||
std::time::SystemTime::now()
|
|
||||||
.duration_since(std::time::UNIX_EPOCH)
|
|
||||||
.map(|d| d.as_millis())
|
|
||||||
.unwrap_or_default()
|
|
||||||
);
|
|
||||||
let span = info_span!("agent_session", session_id = %session_id, agent_id = %config.agent_id);
|
|
||||||
let _span_guard = span.enter(); // the ENTIRETY of below is agent_session.
|
|
||||||
|
|
||||||
let ws_url = websocket_url(&config.server_url)?;
|
let ws_url = websocket_url(&config.server_url)?;
|
||||||
info!(%ws_url, agent_id = %config.agent_id, "connecting agent websocket");
|
info!(%ws_url, agent_id = %config.agent_id, "connecting agent websocket");
|
||||||
if let Some((dns_resolve_ms, resolved_addrs)) = dns_resolution_diagnostics(&ws_url).await {
|
if let Some((dns_resolve_ms, resolved_addrs)) = dns_resolution_diagnostics(&ws_url)
|
||||||
|
.instrument(info_span!("agent_dns", agent_id = %config.agent_id))
|
||||||
|
.await
|
||||||
|
{
|
||||||
info!(
|
info!(
|
||||||
host = %ws_url.host_str().unwrap_or(""),
|
host = %ws_url.host_str().unwrap_or(""),
|
||||||
dns_resolve_ms,
|
dns_resolve_ms,
|
||||||
@@ -70,7 +64,15 @@ async fn run_once(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
let connect_started = Instant::now();
|
let connect_started = Instant::now();
|
||||||
let (stream, _) = connect_async(ws_url.as_str())
|
let (stream, _) = async {
|
||||||
|
let mut operation = wakey_core::telemetry::Operation::new("agent_connect");
|
||||||
|
let result = connect_async(ws_url.as_str()).await;
|
||||||
|
if result.is_ok() {
|
||||||
|
operation.outcome("ok");
|
||||||
|
}
|
||||||
|
result
|
||||||
|
}
|
||||||
|
.instrument(info_span!("agent_connect", agent_id = %config.agent_id))
|
||||||
.await
|
.await
|
||||||
.context("failed to connect websocket")?;
|
.context("failed to connect websocket")?;
|
||||||
let ws_connect_ms = connect_started.elapsed().as_millis() as u64;
|
let ws_connect_ms = connect_started.elapsed().as_millis() as u64;
|
||||||
@@ -80,6 +82,7 @@ async fn run_once(
|
|||||||
}
|
}
|
||||||
let (mut sink, mut source) = stream.split();
|
let (mut sink, mut source) = stream.split();
|
||||||
|
|
||||||
|
async {
|
||||||
send_json(
|
send_json(
|
||||||
&mut sink,
|
&mut sink,
|
||||||
&ClientMessage::Hello {
|
&ClientMessage::Hello {
|
||||||
@@ -91,7 +94,7 @@ async fn run_once(
|
|||||||
.enabled
|
.enabled
|
||||||
.then_some(TerminalCapabilityOptions {
|
.then_some(TerminalCapabilityOptions {
|
||||||
max_sessions: config.terminal.max_sessions.max(1),
|
max_sessions: config.terminal.max_sessions.max(1),
|
||||||
session_ttl_seconds: config.terminal.session_ttl_seconds,
|
session_ttl_seconds: config.terminal.session_ttl.as_secs(),
|
||||||
}),
|
}),
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -106,13 +109,15 @@ async fn run_once(
|
|||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
send_terminal_sessions(&mut sink, terminal_manager).await?;
|
send_terminal_sessions(&mut sink, terminal_manager).await?;
|
||||||
info!(agent_id = %config.agent_id, "agent websocket session authenticated");
|
Ok::<_, anyhow::Error>(())
|
||||||
|
}
|
||||||
|
.instrument(info_span!("agent_handshake_send", agent_id = %config.agent_id))
|
||||||
|
.await?;
|
||||||
|
info!(agent_id = %config.agent_id, "agent hello and credentials sent; awaiting server traffic");
|
||||||
|
|
||||||
let mut heartbeat = interval(Duration::from_secs(30));
|
let mut heartbeat = interval(Duration::from_secs(30));
|
||||||
heartbeat.set_missed_tick_behavior(MissedTickBehavior::Skip);
|
heartbeat.set_missed_tick_behavior(MissedTickBehavior::Skip);
|
||||||
let mut snapshot_sync = interval(Duration::from_secs(
|
let mut snapshot_sync = interval(config.observation_sync_interval);
|
||||||
config.observation_sync_interval_seconds.max(1),
|
|
||||||
));
|
|
||||||
snapshot_sync.set_missed_tick_behavior(MissedTickBehavior::Skip);
|
snapshot_sync.set_missed_tick_behavior(MissedTickBehavior::Skip);
|
||||||
snapshot_sync.reset();
|
snapshot_sync.reset();
|
||||||
loop {
|
loop {
|
||||||
@@ -195,11 +200,13 @@ where
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "collect_device_snapshot", skip_all)]
|
||||||
async fn send_device_snapshot_ws<S>(sink: &mut S, config: &AgentConfig) -> Result<usize>
|
async fn send_device_snapshot_ws<S>(sink: &mut S, config: &AgentConfig) -> Result<usize>
|
||||||
where
|
where
|
||||||
S: SinkExt<Message> + Unpin,
|
S: SinkExt<Message> + Unpin,
|
||||||
<S as futures_util::Sink<Message>>::Error: std::error::Error + Send + Sync + 'static,
|
<S as futures_util::Sink<Message>>::Error: std::error::Error + Send + Sync + 'static,
|
||||||
{
|
{
|
||||||
|
let mut operation = wakey_core::telemetry::Operation::new("snapshot");
|
||||||
let req = crate::protocol::InventoryRequest {
|
let req = crate::protocol::InventoryRequest {
|
||||||
query: None,
|
query: None,
|
||||||
name: None,
|
name: None,
|
||||||
@@ -217,16 +224,22 @@ where
|
|||||||
&ClientMessage::DeviceSnapshot {
|
&ClientMessage::DeviceSnapshot {
|
||||||
agent_id: config.agent_id.clone(),
|
agent_id: config.agent_id.clone(),
|
||||||
devices: inventory.devices,
|
devices: inventory.devices,
|
||||||
|
trace_context: wakey_core::telemetry::current_trace_context(),
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
.await?;
|
.await?;
|
||||||
debug!(agent_id = %config.agent_id, devices = count, "sent device snapshot over websocket");
|
debug!(agent_id = %config.agent_id, devices = count, "sent device snapshot over websocket");
|
||||||
|
operation.outcome("ok");
|
||||||
|
wakey_core::telemetry::count(
|
||||||
|
wakey_core::telemetry::Counter::SnapshotDevices,
|
||||||
|
count as u64,
|
||||||
|
);
|
||||||
Ok(count)
|
Ok(count)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn next_backoff_ms(current_ms: u64, max_ms: u64) -> u64 {
|
pub fn next_backoff(current: Duration, max: Duration) -> Duration {
|
||||||
let cap = max_ms.max(current_ms);
|
let cap = max.max(current);
|
||||||
current_ms.saturating_mul(2).min(cap)
|
current.saturating_mul(2).min(cap)
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle_server_message<S>(
|
async fn handle_server_message<S>(
|
||||||
@@ -244,11 +257,16 @@ where
|
|||||||
ServerMessage::Command {
|
ServerMessage::Command {
|
||||||
request_id,
|
request_id,
|
||||||
command,
|
command,
|
||||||
|
trace_context,
|
||||||
} => {
|
} => {
|
||||||
let kind = command_kind(&command);
|
let kind = command_kind(&command);
|
||||||
|
let span = info_span!("execute_command", command = kind, request_id = %request_id);
|
||||||
|
wakey_core::telemetry::set_remote_parent(&span, &trace_context);
|
||||||
|
let mut operation = wakey_core::telemetry::Operation::new(kind);
|
||||||
info!(request_id = %request_id, command = %kind, "received command from control-plane");
|
info!(request_id = %request_id, command = %kind, "received command from control-plane");
|
||||||
match dispatch_command(command, config).await {
|
match tracing::Instrument::instrument(dispatch_command(command, config), span).await {
|
||||||
Ok(result) => {
|
Ok(result) => {
|
||||||
|
operation.outcome("ok");
|
||||||
info!(request_id = %request_id, command = %kind, "command execution completed");
|
info!(request_id = %request_id, command = %kind, "command execution completed");
|
||||||
send_command_result(sink, request_id, kind, result).await?;
|
send_command_result(sink, request_id, kind, result).await?;
|
||||||
}
|
}
|
||||||
@@ -455,7 +473,7 @@ mod tests {
|
|||||||
use serde_json::Value;
|
use serde_json::Value;
|
||||||
use tokio_tungstenite::tungstenite::Message;
|
use tokio_tungstenite::tungstenite::Message;
|
||||||
|
|
||||||
use super::{next_backoff_ms, send_command_result, websocket_url};
|
use super::{next_backoff, send_command_result, websocket_url};
|
||||||
use crate::protocol::{CommandResult, RequestId};
|
use crate::protocol::{CommandResult, RequestId};
|
||||||
|
|
||||||
struct RecordingSink {
|
struct RecordingSink {
|
||||||
@@ -514,14 +532,28 @@ mod tests {
|
|||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn backoff_doubles_until_cap() {
|
fn backoff_doubles_until_cap() {
|
||||||
assert_eq!(next_backoff_ms(1_000, 30_000), 2_000);
|
use std::time::Duration;
|
||||||
assert_eq!(next_backoff_ms(16_000, 30_000), 30_000);
|
assert_eq!(
|
||||||
assert_eq!(next_backoff_ms(30_000, 30_000), 30_000);
|
next_backoff(Duration::from_secs(1), Duration::from_secs(30)),
|
||||||
|
Duration::from_secs(2)
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
next_backoff(Duration::from_secs(16), Duration::from_secs(30)),
|
||||||
|
Duration::from_secs(30)
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
next_backoff(Duration::from_secs(30), Duration::from_secs(30)),
|
||||||
|
Duration::from_secs(30)
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn backoff_never_shrinks_when_max_is_lower() {
|
fn backoff_never_shrinks_when_max_is_lower() {
|
||||||
assert_eq!(next_backoff_ms(8_000, 1_000), 8_000);
|
use std::time::Duration;
|
||||||
|
assert_eq!(
|
||||||
|
next_backoff(Duration::from_secs(8), Duration::from_secs(1)),
|
||||||
|
Duration::from_secs(8)
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
|
|||||||
+436
-79
@@ -1,4 +1,4 @@
|
|||||||
use std::collections::HashMap;
|
use std::collections::{HashMap, VecDeque};
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
use std::sync::{Arc, Mutex, Weak};
|
use std::sync::{Arc, Mutex, Weak};
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
@@ -13,8 +13,9 @@ use tokio_tungstenite::tungstenite::Message;
|
|||||||
use tracing::{info, warn};
|
use tracing::{info, warn};
|
||||||
|
|
||||||
const MAX_TERMINAL_FRAME_BYTES: usize = 64 * 1024;
|
const MAX_TERMINAL_FRAME_BYTES: usize = 64 * 1024;
|
||||||
const TERMINAL_SCROLLBACK_ROWS: usize = 5_000;
|
const TERMINAL_SCROLLBACK_ROWS: usize = 1_000;
|
||||||
const RELAY_INPUT_QUEUE: usize = 32;
|
const RELAY_INPUT_QUEUE: usize = 32;
|
||||||
|
const RELAY_OUTPUT_QUEUE: usize = 32;
|
||||||
const PTY_EXIT_DRAIN_TIMEOUT: Duration = Duration::from_secs(1);
|
const PTY_EXIT_DRAIN_TIMEOUT: Duration = Duration::from_secs(1);
|
||||||
const PROCESS_SIGNAL_GRACE: Duration = Duration::from_secs(1);
|
const PROCESS_SIGNAL_GRACE: Duration = Duration::from_secs(1);
|
||||||
|
|
||||||
@@ -26,6 +27,109 @@ struct TerminalState {
|
|||||||
parser: vt100::Parser,
|
parser: vt100::Parser,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Buffers ordered terminal output without letting a slow relay grow memory
|
||||||
|
/// without bound. Once a frame is pending, the PTY read branch is paused until
|
||||||
|
/// the relay channel has capacity; control input remains independently active.
|
||||||
|
#[derive(Default)]
|
||||||
|
struct RelayOutputBuffer {
|
||||||
|
sender: Option<mpsc::Sender<Message>>,
|
||||||
|
pending: VecDeque<Message>,
|
||||||
|
snapshot_pending: bool,
|
||||||
|
backpressure_wait: Option<wakey_core::telemetry::Operation>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl RelayOutputBuffer {
|
||||||
|
fn connect(&mut self, sender: mpsc::Sender<Message>) {
|
||||||
|
self.disconnect();
|
||||||
|
self.sender = Some(sender);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn disconnect(&mut self) {
|
||||||
|
if let Some(mut wait) = self.backpressure_wait.take() {
|
||||||
|
wait.outcome("disconnected");
|
||||||
|
}
|
||||||
|
self.sender = None;
|
||||||
|
self.pending.clear();
|
||||||
|
self.snapshot_pending = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
fn is_connected(&self) -> bool {
|
||||||
|
self.sender.is_some()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn has_pending(&self) -> bool {
|
||||||
|
!self.pending.is_empty()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn pending_sender(&self) -> Option<mpsc::Sender<Message>> {
|
||||||
|
self.has_pending().then(|| self.sender.clone()).flatten()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn queue_live(&mut self, frame: Message) {
|
||||||
|
debug_assert!(
|
||||||
|
self.pending.is_empty(),
|
||||||
|
"PTY reads must pause while relay output is pending"
|
||||||
|
);
|
||||||
|
self.queue_frame(frame);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn queue_frame(&mut self, frame: Message) {
|
||||||
|
let Some(sender) = self.sender.as_ref() else {
|
||||||
|
return;
|
||||||
|
};
|
||||||
|
if !self.pending.is_empty() {
|
||||||
|
self.pending.push_back(frame);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
match sender.try_send(frame) {
|
||||||
|
Ok(()) => {}
|
||||||
|
Err(mpsc::error::TrySendError::Full(frame)) => {
|
||||||
|
// One measurement per backlog episode, not per select poll.
|
||||||
|
self.backpressure_wait = Some(wakey_core::telemetry::Operation::new(
|
||||||
|
"terminal_backpressure_wait",
|
||||||
|
));
|
||||||
|
wakey_core::telemetry::count(
|
||||||
|
wakey_core::telemetry::Counter::TerminalBackpressure,
|
||||||
|
1,
|
||||||
|
);
|
||||||
|
self.pending.push_back(frame);
|
||||||
|
}
|
||||||
|
Err(mpsc::error::TrySendError::Closed(_)) => self.disconnect(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn queue_snapshot(&mut self, snapshot: Vec<u8>) {
|
||||||
|
if self.sender.is_none() || self.snapshot_pending {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
for chunk in snapshot.chunks(MAX_TERMINAL_FRAME_BYTES) {
|
||||||
|
self.queue_frame(Message::Binary(chunk.to_vec().into()));
|
||||||
|
if self.sender.is_none() {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
self.snapshot_pending = self.has_pending();
|
||||||
|
}
|
||||||
|
|
||||||
|
fn send_reserved(&mut self, permit: mpsc::OwnedPermit<Message>) {
|
||||||
|
let frame = self
|
||||||
|
.pending
|
||||||
|
.pop_front()
|
||||||
|
.expect("relay permit requires pending output");
|
||||||
|
permit.send(frame);
|
||||||
|
if self.pending.is_empty() {
|
||||||
|
self.snapshot_pending = false;
|
||||||
|
if let Some(mut wait) = self.backpressure_wait.take() {
|
||||||
|
wait.outcome("ok");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn live_sender(&self) -> Option<mpsc::Sender<Message>> {
|
||||||
|
self.sender.clone()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl TerminalState {
|
impl TerminalState {
|
||||||
fn new(rows: u16, cols: u16) -> Self {
|
fn new(rows: u16, cols: u16) -> Self {
|
||||||
Self {
|
Self {
|
||||||
@@ -41,10 +145,14 @@ impl TerminalState {
|
|||||||
self.parser.screen_mut().set_size(rows, cols);
|
self.parser.screen_mut().set_size(rows, cols);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "terminal_snapshot_generate", skip_all, fields(bytes = tracing::field::Empty))]
|
||||||
fn snapshot(&self) -> Vec<u8> {
|
fn snapshot(&self) -> Vec<u8> {
|
||||||
self.parser
|
let snapshot = self
|
||||||
|
.parser
|
||||||
.screen()
|
.screen()
|
||||||
.snapshot_formatted(TERMINAL_SCROLLBACK_ROWS)
|
.snapshot_formatted(TERMINAL_SCROLLBACK_ROWS);
|
||||||
|
tracing::Span::current().record("bytes", snapshot.len());
|
||||||
|
snapshot
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -80,6 +188,7 @@ impl TerminalManager {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "terminal_open", skip_all, fields(terminal_id = %terminal_id))]
|
||||||
pub fn open(
|
pub fn open(
|
||||||
&self,
|
&self,
|
||||||
config: &AgentConfig,
|
config: &AgentConfig,
|
||||||
@@ -115,7 +224,7 @@ impl TerminalManager {
|
|||||||
cancel: Some(cancel_tx),
|
cancel: Some(cancel_tx),
|
||||||
relay_credentials: relay_tx.clone(),
|
relay_credentials: relay_tx.clone(),
|
||||||
created_at_unix,
|
created_at_unix,
|
||||||
session_ttl_seconds: config.terminal.session_ttl_seconds,
|
session_ttl_seconds: config.terminal.session_ttl.as_secs(),
|
||||||
},
|
},
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
@@ -125,6 +234,7 @@ impl TerminalManager {
|
|||||||
let active = Arc::downgrade(&self.active);
|
let active = Arc::downgrade(&self.active);
|
||||||
let events = self.events.clone();
|
let events = self.events.clone();
|
||||||
tokio::spawn(async move {
|
tokio::spawn(async move {
|
||||||
|
let _active_metric = wakey_core::telemetry::ActiveTerminal::new();
|
||||||
let result = run_terminal(
|
let result = run_terminal(
|
||||||
&config,
|
&config,
|
||||||
&terminal_id,
|
&terminal_id,
|
||||||
@@ -149,6 +259,7 @@ impl TerminalManager {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "terminal_close", skip_all, fields(terminal_id = %terminal_id))]
|
||||||
pub fn close(&self, terminal_id: &TerminalId) -> bool {
|
pub fn close(&self, terminal_id: &TerminalId) -> bool {
|
||||||
let mut active = self.active.lock().expect("terminal manager poisoned");
|
let mut active = self.active.lock().expect("terminal manager poisoned");
|
||||||
let Some(terminal) = active.get_mut(terminal_id.as_str()) else {
|
let Some(terminal) = active.get_mut(terminal_id.as_str()) else {
|
||||||
@@ -164,7 +275,9 @@ impl TerminalManager {
|
|||||||
true
|
true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "terminal_resume", skip_all, fields(terminal_id = %terminal_id))]
|
||||||
pub fn resume(&self, terminal_id: &TerminalId, relay_token: String) -> Result<()> {
|
pub fn resume(&self, terminal_id: &TerminalId, relay_token: String) -> Result<()> {
|
||||||
|
let mut operation = wakey_core::telemetry::Operation::new("terminal_resume");
|
||||||
let active = self.active.lock().expect("terminal manager poisoned");
|
let active = self.active.lock().expect("terminal manager poisoned");
|
||||||
let session = active
|
let session = active
|
||||||
.get(terminal_id.as_str())
|
.get(terminal_id.as_str())
|
||||||
@@ -172,6 +285,8 @@ impl TerminalManager {
|
|||||||
if session.relay_credentials.send(relay_token).is_err() {
|
if session.relay_credentials.send(relay_token).is_err() {
|
||||||
anyhow::bail!("terminal session {terminal_id} has stopped");
|
anyhow::bail!("terminal session {terminal_id} has stopped");
|
||||||
}
|
}
|
||||||
|
operation.outcome("ok");
|
||||||
|
info!("terminal resume credentials queued for worker");
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -250,7 +365,7 @@ async fn run_terminal(
|
|||||||
);
|
);
|
||||||
|
|
||||||
let (relay_input_tx, mut relay_input_rx) = mpsc::channel(RELAY_INPUT_QUEUE);
|
let (relay_input_tx, mut relay_input_rx) = mpsc::channel(RELAY_INPUT_QUEUE);
|
||||||
let mut relay_output: Option<mpsc::Sender<Message>> = None;
|
let mut relay_output = RelayOutputBuffer::default();
|
||||||
let mut relay_task: Option<tokio::task::JoinHandle<()>> = None;
|
let mut relay_task: Option<tokio::task::JoinHandle<()>> = None;
|
||||||
let mut relay_generation = 0_u64;
|
let mut relay_generation = 0_u64;
|
||||||
let mut terminal_state = TerminalState::new(rows, cols);
|
let mut terminal_state = TerminalState::new(rows, cols);
|
||||||
@@ -261,6 +376,7 @@ async fn run_terminal(
|
|||||||
let mut drain_deadline: Option<std::pin::Pin<Box<tokio::time::Sleep>>> = None;
|
let mut drain_deadline: Option<std::pin::Pin<Box<tokio::time::Sleep>>> = None;
|
||||||
let mut ttl_deadline = session_ttl.map(|ttl| Box::pin(tokio::time::sleep(ttl)));
|
let mut ttl_deadline = session_ttl.map(|ttl| Box::pin(tokio::time::sleep(ttl)));
|
||||||
loop {
|
loop {
|
||||||
|
let pending_output_sender = relay_output.pending_sender();
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
biased;
|
biased;
|
||||||
_ = &mut cancel => {
|
_ = &mut cancel => {
|
||||||
@@ -303,12 +419,7 @@ async fn run_terminal(
|
|||||||
terminal_state.resize(rows, cols);
|
terminal_state.resize(rows, cols);
|
||||||
}
|
}
|
||||||
RelayInput::Snapshot { generation } if generation == relay_generation => {
|
RelayInput::Snapshot { generation } if generation == relay_generation => {
|
||||||
if !send_terminal_snapshot(terminal_state.snapshot(), &mut relay_output) {
|
relay_output.queue_snapshot(terminal_state.snapshot());
|
||||||
if let Some(task) = relay_task.take() {
|
|
||||||
task.abort();
|
|
||||||
}
|
|
||||||
warn!(terminal_id = %terminal_id, "terminal relay output queue saturated during snapshot");
|
|
||||||
}
|
|
||||||
if let Err(err) = writer.refresh() {
|
if let Err(err) = writer.refresh() {
|
||||||
warn!(terminal_id = %terminal_id, error = %err, "terminal redraw signal failed");
|
warn!(terminal_id = %terminal_id, error = %err, "terminal redraw signal failed");
|
||||||
}
|
}
|
||||||
@@ -318,10 +429,10 @@ async fn run_terminal(
|
|||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
RelayInput::Connected { generation, output } if generation == relay_generation => {
|
RelayInput::Connected { generation, output } if generation == relay_generation => {
|
||||||
relay_output = Some(output);
|
relay_output.connect(output);
|
||||||
}
|
}
|
||||||
RelayInput::Disconnected { generation } if generation == relay_generation => {
|
RelayInput::Disconnected { generation } if generation == relay_generation => {
|
||||||
relay_output = None;
|
relay_output.disconnect();
|
||||||
}
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
@@ -333,8 +444,8 @@ async fn run_terminal(
|
|||||||
}
|
}
|
||||||
relay_generation = relay_generation.wrapping_add(1);
|
relay_generation = relay_generation.wrapping_add(1);
|
||||||
let generation = relay_generation;
|
let generation = relay_generation;
|
||||||
let (output_tx, output_rx) = mpsc::channel(32);
|
let (output_tx, output_rx) = mpsc::channel(RELAY_OUTPUT_QUEUE);
|
||||||
relay_output = None;
|
relay_output.disconnect();
|
||||||
let config = config.clone();
|
let config = config.clone();
|
||||||
let terminal_id = terminal_id.clone();
|
let terminal_id = terminal_id.clone();
|
||||||
let relay_input_tx = relay_input_tx.clone();
|
let relay_input_tx = relay_input_tx.clone();
|
||||||
@@ -353,21 +464,25 @@ async fn run_terminal(
|
|||||||
let _ = relay_input_tx.send(RelayInput::Disconnected { generation }).await;
|
let _ = relay_input_tx.send(RelayInput::Disconnected { generation }).await;
|
||||||
}));
|
}));
|
||||||
}
|
}
|
||||||
read = reader.read(&mut output) => {
|
permit = reserve_relay_output(pending_output_sender), if relay_output.has_pending() => {
|
||||||
match read {
|
match permit {
|
||||||
Ok(0) => break,
|
Ok(permit) => relay_output.send_reserved(permit),
|
||||||
Ok(count) => {
|
Err(()) => {
|
||||||
terminal_state.process(&output[..count]);
|
relay_output.disconnect();
|
||||||
if !send_terminal_output(
|
|
||||||
Message::Binary(output[..count].to_vec().into()),
|
|
||||||
&mut relay_output,
|
|
||||||
) {
|
|
||||||
if let Some(task) = relay_task.take() {
|
if let Some(task) = relay_task.take() {
|
||||||
task.abort();
|
task.abort();
|
||||||
}
|
}
|
||||||
warn!(terminal_id = %terminal_id, "terminal relay output queue saturated");
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
read = reader.read(&mut output), if !relay_output.has_pending() => {
|
||||||
|
match read {
|
||||||
|
Ok(0) => break,
|
||||||
|
Ok(count) => {
|
||||||
|
wakey_core::telemetry::count(wakey_core::telemetry::Counter::TerminalOutputBytes, count as u64);
|
||||||
|
terminal_state.process(&output[..count]);
|
||||||
|
relay_output.queue_live(Message::Binary(output[..count].to_vec().into()));
|
||||||
|
}
|
||||||
// Linux PTY masters commonly report EIO after the slave closes.
|
// Linux PTY masters commonly report EIO after the slave closes.
|
||||||
Err(err) if err.raw_os_error() == Some(5) => break,
|
Err(err) if err.raw_os_error() == Some(5) => break,
|
||||||
Err(err) => return Err(err).context("failed to read PTY output"),
|
Err(err) => return Err(err).context("failed to read PTY output"),
|
||||||
@@ -380,13 +495,13 @@ async fn run_terminal(
|
|||||||
Some(status) => status,
|
Some(status) => status,
|
||||||
None => terminate_process_group(&mut child, process_group).await?,
|
None => terminate_process_group(&mut child, process_group).await?,
|
||||||
};
|
};
|
||||||
if let Some(output) = relay_output {
|
if relay_output.is_connected() {
|
||||||
let control = TerminalControl::Exited {
|
let exit_frame = serde_json::to_string(&TerminalControl::Exited {
|
||||||
exit_code: status.code(),
|
exit_code: status.code(),
|
||||||
};
|
})
|
||||||
if let Ok(text) = serde_json::to_string(&control) {
|
.ok()
|
||||||
let _ = output.try_send(Message::Text(text.into()));
|
.map(|text| Message::Text(text.into()));
|
||||||
}
|
flush_relay_output(&mut relay_output, exit_frame, PTY_EXIT_DRAIN_TIMEOUT).await;
|
||||||
}
|
}
|
||||||
info!(terminal_id = %terminal_id, exit_code = ?status.code(), requested_close, ttl_expired, "terminal worker exited");
|
info!(terminal_id = %terminal_id, exit_code = ?status.code(), requested_close, ttl_expired, "terminal worker exited");
|
||||||
Ok(())
|
Ok(())
|
||||||
@@ -434,7 +549,7 @@ async fn run_terminal_relay(relay: RelayConnection) -> Result<()> {
|
|||||||
let (stream, _) = tokio_tungstenite::connect_async(ws_url.as_str())
|
let (stream, _) = tokio_tungstenite::connect_async(ws_url.as_str())
|
||||||
.await
|
.await
|
||||||
.context("failed to connect terminal relay websocket")?;
|
.context("failed to connect terminal relay websocket")?;
|
||||||
let (mut sink, mut source) = stream.split();
|
let (mut sink, source) = stream.split();
|
||||||
send_json(
|
send_json(
|
||||||
&mut sink,
|
&mut sink,
|
||||||
&TerminalAgentHandshake::Auth {
|
&TerminalAgentHandshake::Auth {
|
||||||
@@ -453,40 +568,62 @@ async fn run_terminal_relay(relay: RelayConnection) -> Result<()> {
|
|||||||
.await
|
.await
|
||||||
.map_err(|_| anyhow::anyhow!("terminal worker stopped"))?;
|
.map_err(|_| anyhow::anyhow!("terminal worker stopped"))?;
|
||||||
|
|
||||||
let mut output = relay.output_rx;
|
relay_loop(source, sink, relay.input, relay.generation, relay.output_rx).await
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn relay_loop<S, St>(
|
||||||
|
mut source: St,
|
||||||
|
mut sink: S,
|
||||||
|
input: mpsc::Sender<RelayInput>,
|
||||||
|
generation: u64,
|
||||||
|
mut output: mpsc::Receiver<Message>,
|
||||||
|
) -> Result<()>
|
||||||
|
where
|
||||||
|
S: futures_util::Sink<Message, Error = tokio_tungstenite::tungstenite::Error> + Unpin,
|
||||||
|
St: futures_util::Stream<
|
||||||
|
Item = std::result::Result<Message, tokio_tungstenite::tungstenite::Error>,
|
||||||
|
> + Unpin,
|
||||||
|
{
|
||||||
|
let mut worker_stopped = false;
|
||||||
|
let mut worker_stop_deadline: Option<tokio::time::Instant> = None;
|
||||||
|
// Reserve at most one control so a full worker queue cannot block output.
|
||||||
|
let mut pending_input = None;
|
||||||
|
let mut close_after_input = false;
|
||||||
loop {
|
loop {
|
||||||
tokio::select! {
|
tokio::select! {
|
||||||
biased;
|
biased;
|
||||||
incoming = source.next() => {
|
_ = async {
|
||||||
|
match worker_stop_deadline {
|
||||||
|
Some(deadline) => tokio::time::sleep_until(deadline).await,
|
||||||
|
None => std::future::pending().await,
|
||||||
|
}
|
||||||
|
} => break,
|
||||||
|
incoming = source.next(), if !worker_stopped && pending_input.is_none() => {
|
||||||
let Some(message) = incoming else { break; };
|
let Some(message) = incoming else { break; };
|
||||||
match message.context("terminal relay websocket receive failed")? {
|
match message.context("terminal relay websocket receive failed")? {
|
||||||
Message::Binary(bytes) => {
|
Message::Binary(bytes) => {
|
||||||
relay.input.send(RelayInput::Binary {
|
wakey_core::telemetry::count(wakey_core::telemetry::Counter::TerminalInputBytes, bytes.len() as u64);
|
||||||
generation: relay.generation,
|
pending_input = Some(RelayInput::Binary {
|
||||||
|
generation,
|
||||||
bytes: bytes.to_vec(),
|
bytes: bytes.to_vec(),
|
||||||
}).await.map_err(|_| anyhow::anyhow!("terminal worker stopped"))?;
|
});
|
||||||
}
|
}
|
||||||
Message::Text(text) => match serde_json::from_str::<TerminalControl>(&text)
|
Message::Text(text) => match serde_json::from_str::<TerminalControl>(&text)
|
||||||
.context("invalid terminal control frame")?
|
.context("invalid terminal control frame")?
|
||||||
{
|
{
|
||||||
TerminalControl::Resize { rows, cols } => {
|
TerminalControl::Resize { rows, cols } => {
|
||||||
relay.input.send(RelayInput::Resize {
|
pending_input = Some(RelayInput::Resize {
|
||||||
generation: relay.generation,
|
generation,
|
||||||
rows,
|
rows,
|
||||||
cols,
|
cols,
|
||||||
}).await
|
});
|
||||||
.map_err(|_| anyhow::anyhow!("terminal worker stopped"))?;
|
|
||||||
}
|
}
|
||||||
TerminalControl::Snapshot => {
|
TerminalControl::Snapshot => {
|
||||||
relay.input.send(RelayInput::Snapshot {
|
pending_input = Some(RelayInput::Snapshot { generation });
|
||||||
generation: relay.generation,
|
|
||||||
}).await.map_err(|_| anyhow::anyhow!("terminal worker stopped"))?;
|
|
||||||
}
|
}
|
||||||
TerminalControl::Close => {
|
TerminalControl::Close => {
|
||||||
let _ = relay.input.send(RelayInput::Close {
|
pending_input = Some(RelayInput::Close { generation });
|
||||||
generation: relay.generation,
|
close_after_input = true;
|
||||||
}).await;
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
_ => anyhow::bail!("terminal control frame has invalid direction"),
|
_ => anyhow::bail!("terminal control frame has invalid direction"),
|
||||||
},
|
},
|
||||||
@@ -498,36 +635,82 @@ async fn run_terminal_relay(relay: RelayConnection) -> Result<()> {
|
|||||||
Message::Frame(_) => {}
|
Message::Frame(_) => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
permit = input.reserve(), if !worker_stopped && pending_input.is_some() => {
|
||||||
|
match permit {
|
||||||
|
Ok(permit) => {
|
||||||
|
permit.send(pending_input.take().expect("pending terminal input"));
|
||||||
|
if close_after_input {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(_) => {
|
||||||
|
worker_stopped = true;
|
||||||
|
pending_input = None;
|
||||||
|
worker_stop_deadline =
|
||||||
|
Some(tokio::time::Instant::now() + PTY_EXIT_DRAIN_TIMEOUT);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
outgoing = output.recv() => {
|
outgoing = output.recv() => {
|
||||||
let Some(message) = outgoing else { break; };
|
let Some(message) = outgoing else { break; };
|
||||||
|
if let Some(deadline) = worker_stop_deadline {
|
||||||
|
tokio::time::timeout_at(deadline, sink.send(message))
|
||||||
|
.await
|
||||||
|
.context("terminal relay output drain timed out")?
|
||||||
|
.context("failed to send terminal relay output")?;
|
||||||
|
} else {
|
||||||
sink.send(message).await.context("failed to send terminal relay output")?;
|
sink.send(message).await.context("failed to send terminal relay output")?;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn send_terminal_output(frame: Message, relay: &mut Option<mpsc::Sender<Message>>) -> bool {
|
async fn reserve_relay_output(
|
||||||
if let Some(tx) = relay.as_ref()
|
sender: Option<mpsc::Sender<Message>>,
|
||||||
&& tx.try_send(frame).is_err()
|
) -> Result<mpsc::OwnedPermit<Message>, ()> {
|
||||||
{
|
match sender {
|
||||||
*relay = None;
|
Some(sender) => sender.reserve_owned().await.map_err(|_| ()),
|
||||||
return false;
|
None => std::future::pending().await,
|
||||||
}
|
}
|
||||||
true
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn send_terminal_snapshot(snapshot: Vec<u8>, relay: &mut Option<mpsc::Sender<Message>>) -> bool {
|
/// Flushes frames the select loop left pending, then sends the exit lifecycle
|
||||||
let Some(tx) = relay.as_ref() else {
|
/// frame behind them, bounded by `deadline`: a relay that stays saturated past
|
||||||
return true;
|
/// it drops the backlog rather than holding worker shutdown open. Queuing the
|
||||||
|
/// exit frame without this would drop it together with the pending buffer when
|
||||||
|
/// the worker returns.
|
||||||
|
async fn flush_relay_output(
|
||||||
|
relay_output: &mut RelayOutputBuffer,
|
||||||
|
exit_frame: Option<Message>,
|
||||||
|
deadline: Duration,
|
||||||
|
) {
|
||||||
|
let _ = tokio::time::timeout(deadline, async {
|
||||||
|
while relay_output.has_pending() {
|
||||||
|
let sender = relay_output.pending_sender();
|
||||||
|
match reserve_relay_output(sender).await {
|
||||||
|
Ok(permit) => relay_output.send_reserved(permit),
|
||||||
|
Err(()) => {
|
||||||
|
relay_output.disconnect();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let Some(frame) = exit_frame else {
|
||||||
|
return;
|
||||||
};
|
};
|
||||||
for chunk in snapshot.chunks(MAX_TERMINAL_FRAME_BYTES) {
|
let sender = relay_output
|
||||||
if tx.try_send(Message::Binary(chunk.to_vec().into())).is_err() {
|
.pending_sender()
|
||||||
*relay = None;
|
.or_else(|| relay_output.live_sender());
|
||||||
return false;
|
match reserve_relay_output(sender).await {
|
||||||
|
Ok(permit) => {
|
||||||
|
permit.send(frame);
|
||||||
}
|
}
|
||||||
|
Err(()) => relay_output.disconnect(),
|
||||||
}
|
}
|
||||||
true
|
})
|
||||||
|
.await;
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(not(unix))]
|
#[cfg(not(unix))]
|
||||||
@@ -716,6 +899,102 @@ mod tests {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn exit_frame_is_delivered_after_saturated_relay_drains() {
|
||||||
|
let (tx, mut rx) = mpsc::channel(RELAY_OUTPUT_QUEUE);
|
||||||
|
let mut relay_output = RelayOutputBuffer::default();
|
||||||
|
relay_output.connect(tx);
|
||||||
|
|
||||||
|
let total = RELAY_OUTPUT_QUEUE * 2;
|
||||||
|
for i in 0..total {
|
||||||
|
relay_output.queue_frame(Message::Text(format!("out-{i}").into()));
|
||||||
|
}
|
||||||
|
assert!(relay_output.has_pending());
|
||||||
|
|
||||||
|
let collected = tokio::spawn(async move {
|
||||||
|
let mut frames = Vec::new();
|
||||||
|
while let Some(frame) = rx.recv().await {
|
||||||
|
frames.push(frame);
|
||||||
|
}
|
||||||
|
frames
|
||||||
|
});
|
||||||
|
|
||||||
|
let exit_frame = serde_json::to_string(&TerminalControl::Exited { exit_code: Some(0) })
|
||||||
|
.expect("exit frame encodes");
|
||||||
|
flush_relay_output(
|
||||||
|
&mut relay_output,
|
||||||
|
Some(Message::Text(exit_frame.clone().into())),
|
||||||
|
PTY_EXIT_DRAIN_TIMEOUT,
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
relay_output.disconnect();
|
||||||
|
|
||||||
|
let mut frames = collected.await.expect("relay reader completes");
|
||||||
|
assert_eq!(frames.len(), total + 1);
|
||||||
|
let Some(Message::Text(last)) = frames.pop() else {
|
||||||
|
panic!("last frame should be the exit control frame");
|
||||||
|
};
|
||||||
|
assert_eq!(last.as_str(), exit_frame);
|
||||||
|
for (i, frame) in frames.iter().enumerate() {
|
||||||
|
let Message::Text(payload) = frame else {
|
||||||
|
panic!("expected text frame at position {i}, got {frame:?}");
|
||||||
|
};
|
||||||
|
assert_eq!(payload.as_str(), format!("out-{i}"));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn relay_drains_final_output_after_worker_stops_during_concurrent_controls() {
|
||||||
|
let (input, input_rx) = mpsc::channel(RELAY_INPUT_QUEUE);
|
||||||
|
drop(input_rx);
|
||||||
|
let (output_tx, output_rx) = mpsc::channel(RELAY_OUTPUT_QUEUE);
|
||||||
|
output_tx
|
||||||
|
.send(Message::Binary(vec![1, 2, 3].into()))
|
||||||
|
.await
|
||||||
|
.expect("live output queues");
|
||||||
|
let exit_frame = serde_json::to_string(&TerminalControl::Exited { exit_code: Some(0) })
|
||||||
|
.expect("exit frame encodes");
|
||||||
|
output_tx
|
||||||
|
.send(Message::Text(exit_frame.clone().into()))
|
||||||
|
.await
|
||||||
|
.expect("exit output queues");
|
||||||
|
drop(output_tx);
|
||||||
|
|
||||||
|
let source = futures_util::stream::iter([
|
||||||
|
Ok(Message::Binary(vec![4].into())),
|
||||||
|
Ok(Message::Text(
|
||||||
|
serde_json::to_string(&TerminalControl::Resize {
|
||||||
|
rows: 30,
|
||||||
|
cols: 100,
|
||||||
|
})
|
||||||
|
.expect("resize encodes")
|
||||||
|
.into(),
|
||||||
|
)),
|
||||||
|
Ok(Message::Text(
|
||||||
|
serde_json::to_string(&TerminalControl::Snapshot)
|
||||||
|
.expect("snapshot encodes")
|
||||||
|
.into(),
|
||||||
|
)),
|
||||||
|
]);
|
||||||
|
let delivered = Arc::new(Mutex::new(Vec::new()));
|
||||||
|
let sink = futures_util::sink::unfold(delivered.clone(), |delivered, frame| async move {
|
||||||
|
delivered.lock().expect("delivered frames").push(frame);
|
||||||
|
Ok::<_, tokio_tungstenite::tungstenite::Error>(delivered)
|
||||||
|
});
|
||||||
|
|
||||||
|
relay_loop(source, Box::pin(sink), input, 7, output_rx)
|
||||||
|
.await
|
||||||
|
.expect("relay drains after worker completion");
|
||||||
|
|
||||||
|
let delivered = delivered.lock().expect("delivered frames");
|
||||||
|
assert_eq!(delivered.len(), 2);
|
||||||
|
assert_eq!(delivered[0].clone().into_data(), vec![1, 2, 3]);
|
||||||
|
let Message::Text(exit) = &delivered[1] else {
|
||||||
|
panic!("exit control should follow final bytes");
|
||||||
|
};
|
||||||
|
assert_eq!(exit.as_str(), exit_frame);
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn open_rejects_invalid_initial_size_before_spawning() {
|
fn open_rejects_invalid_initial_size_before_spawning() {
|
||||||
let mut config = crate::config::DEFAULT_CONFIG.clone();
|
let mut config = crate::config::DEFAULT_CONFIG.clone();
|
||||||
@@ -893,13 +1172,19 @@ mod tests {
|
|||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn snapshot_is_split_at_the_terminal_frame_limit() {
|
async fn snapshot_is_split_at_the_terminal_frame_limit() {
|
||||||
let snapshot = vec![7; MAX_TERMINAL_FRAME_BYTES * 2 + 1];
|
let snapshot = vec![7; MAX_TERMINAL_FRAME_BYTES * 2 + 1];
|
||||||
let (tx, mut rx) = mpsc::channel(3);
|
let (tx, mut rx) = mpsc::channel(1);
|
||||||
let mut relay = Some(tx);
|
let mut relay = RelayOutputBuffer::default();
|
||||||
|
relay.connect(tx);
|
||||||
assert!(send_terminal_snapshot(snapshot.clone(), &mut relay));
|
relay.queue_snapshot(snapshot.clone());
|
||||||
|
|
||||||
let mut restored = Vec::new();
|
let mut restored = Vec::new();
|
||||||
for _ in 0..3 {
|
for _ in 0..3 {
|
||||||
|
if rx.is_empty() {
|
||||||
|
let permit = reserve_relay_output(relay.pending_sender())
|
||||||
|
.await
|
||||||
|
.expect("relay remains connected");
|
||||||
|
relay.send_reserved(permit);
|
||||||
|
}
|
||||||
let Message::Binary(chunk) = rx.recv().await.expect("snapshot chunk") else {
|
let Message::Binary(chunk) = rx.recv().await.expect("snapshot chunk") else {
|
||||||
panic!("snapshot chunks must be binary");
|
panic!("snapshot chunks must be binary");
|
||||||
};
|
};
|
||||||
@@ -907,23 +1192,95 @@ mod tests {
|
|||||||
restored.extend_from_slice(&chunk);
|
restored.extend_from_slice(&chunk);
|
||||||
}
|
}
|
||||||
assert_eq!(restored, snapshot);
|
assert_eq!(restored, snapshot);
|
||||||
|
assert!(!relay.has_pending());
|
||||||
|
assert!(relay.is_connected());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn saturated_relay_output_stays_attached_without_dropping() {
|
||||||
|
let (tx, mut rx) = mpsc::channel(1);
|
||||||
|
let mut relay = RelayOutputBuffer::default();
|
||||||
|
relay.connect(tx);
|
||||||
|
relay.queue_live(Message::Binary(vec![1].into()));
|
||||||
|
// This read discovers saturation. The worker retains it, then pauses
|
||||||
|
// further PTY reads until relay capacity becomes available.
|
||||||
|
relay.queue_live(Message::Binary(vec![2].into()));
|
||||||
|
|
||||||
|
assert!(relay.is_connected());
|
||||||
|
assert!(relay.has_pending());
|
||||||
|
assert!(relay.backpressure_wait.is_some());
|
||||||
|
assert_eq!(
|
||||||
|
rx.recv().await.expect("original queued frame").into_data(),
|
||||||
|
vec![1]
|
||||||
|
);
|
||||||
|
let permit = reserve_relay_output(relay.pending_sender())
|
||||||
|
.await
|
||||||
|
.expect("relay remains connected");
|
||||||
|
relay.send_reserved(permit);
|
||||||
|
assert_eq!(
|
||||||
|
rx.recv().await.expect("backpressured frame").into_data(),
|
||||||
|
vec![2]
|
||||||
|
);
|
||||||
|
assert!(!relay.has_pending());
|
||||||
|
assert!(relay.backpressure_wait.is_none());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn saturated_relay_output_detaches_without_waiting() {
|
fn disconnect_finishes_backpressure_episode() {
|
||||||
let (tx, mut rx) = mpsc::channel(1);
|
let (tx, _rx) = mpsc::channel(1);
|
||||||
tx.try_send(Message::Binary(vec![1].into()))
|
let mut relay = RelayOutputBuffer::default();
|
||||||
.expect("prefill relay queue");
|
relay.connect(tx);
|
||||||
let mut relay = Some(tx);
|
relay.queue_live(Message::Binary(vec![1].into()));
|
||||||
|
relay.queue_live(Message::Binary(vec![2].into()));
|
||||||
|
assert!(relay.backpressure_wait.is_some());
|
||||||
|
relay.disconnect();
|
||||||
|
assert!(relay.backpressure_wait.is_none());
|
||||||
|
assert!(!relay.has_pending());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn snapshot_waits_behind_pending_live_output() {
|
||||||
|
let (tx, mut rx) = mpsc::channel(1);
|
||||||
|
let mut relay = RelayOutputBuffer::default();
|
||||||
|
relay.connect(tx);
|
||||||
|
relay.queue_live(Message::Binary(vec![1].into()));
|
||||||
|
relay.queue_live(Message::Binary(vec![2].into()));
|
||||||
|
relay.queue_snapshot(vec![3]);
|
||||||
|
|
||||||
|
let mut received = Vec::new();
|
||||||
|
for _ in 0..3 {
|
||||||
|
if rx.is_empty() {
|
||||||
|
let permit = reserve_relay_output(relay.pending_sender())
|
||||||
|
.await
|
||||||
|
.expect("relay remains connected");
|
||||||
|
relay.send_reserved(permit);
|
||||||
|
}
|
||||||
|
received.extend(rx.recv().await.expect("ordered relay frame").into_data());
|
||||||
|
}
|
||||||
|
assert_eq!(received, vec![1, 2, 3]);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn replacement_relay_does_not_receive_old_pending_output() {
|
||||||
|
let (old_tx, mut old_rx) = mpsc::channel(1);
|
||||||
|
let (new_tx, mut new_rx) = mpsc::channel(1);
|
||||||
|
let mut relay = RelayOutputBuffer::default();
|
||||||
|
relay.connect(old_tx);
|
||||||
|
relay.queue_live(Message::Binary(vec![1].into()));
|
||||||
|
relay.queue_live(Message::Binary(vec![2].into()));
|
||||||
|
|
||||||
|
relay.connect(new_tx);
|
||||||
|
relay.queue_live(Message::Binary(vec![3].into()));
|
||||||
|
|
||||||
assert!(!send_terminal_output(
|
|
||||||
Message::Binary(vec![2].into()),
|
|
||||||
&mut relay,
|
|
||||||
));
|
|
||||||
assert!(relay.is_none());
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
rx.try_recv().expect("original queued frame").into_data(),
|
old_rx.try_recv().expect("old queued frame").into_data(),
|
||||||
vec![1]
|
vec![1]
|
||||||
);
|
);
|
||||||
|
assert!(old_rx.try_recv().is_err());
|
||||||
|
assert_eq!(
|
||||||
|
new_rx.try_recv().expect("new relay frame").into_data(),
|
||||||
|
vec![3]
|
||||||
|
);
|
||||||
|
assert!(!relay.has_pending());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
+19
-25
@@ -1,28 +1,22 @@
|
|||||||
use time::UtcOffset;
|
use anyhow::Result;
|
||||||
use time::format_description::well_known::Rfc3339;
|
use wakey_core::telemetry::{TelemetryConfig, TelemetryGuard};
|
||||||
use tracing_subscriber::fmt::time::OffsetTime;
|
|
||||||
use tracing_subscriber::{EnvFilter, fmt, layer::SubscriberExt, util::SubscriberInitExt};
|
|
||||||
|
|
||||||
pub fn init(verbose: u8) {
|
pub fn init(
|
||||||
let filter = EnvFilter::try_from_default_env()
|
verbose: u8,
|
||||||
.or_else(|_| EnvFilter::try_new(default_filter(verbose)))
|
config: &TelemetryConfig,
|
||||||
.expect("static tracing filter should parse");
|
agent_id: Option<&str>,
|
||||||
|
) -> Result<TelemetryGuard> {
|
||||||
tracing_subscriber::registry()
|
let guard = wakey_core::telemetry::init(
|
||||||
.with(filter)
|
config,
|
||||||
.with(fmt::layer().with_timer(local_offset_timer()))
|
"wakey-agent",
|
||||||
.init();
|
|
||||||
}
|
|
||||||
|
|
||||||
fn default_filter(verbose: u8) -> &'static str {
|
|
||||||
match verbose {
|
match verbose {
|
||||||
0 => "wakey_agent=info,wakey=info",
|
0 => "wakey_agent=info,wakey=info,wakey_core=info,opentelemetry=warn",
|
||||||
1 => "wakey_agent=debug,wakey=debug",
|
1 => "wakey_agent=debug,wakey=debug,wakey_core=info,opentelemetry=warn",
|
||||||
_ => "wakey_agent=trace,wakey=debug",
|
_ => "wakey_agent=trace,wakey=debug,wakey_core=info,opentelemetry=warn",
|
||||||
}
|
},
|
||||||
}
|
agent_id,
|
||||||
|
)?;
|
||||||
fn local_offset_timer() -> OffsetTime<Rfc3339> {
|
#[cfg(target_os = "linux")]
|
||||||
let offset = UtcOffset::current_local_offset().unwrap_or(UtcOffset::UTC);
|
let guard = guard.with_process_metrics(wakey_linux::telemetry::ProcfsProcessMetrics);
|
||||||
OffsetTime::new(offset, Rfc3339)
|
Ok(guard)
|
||||||
}
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "INSERT INTO audit_events\n (event_key, event_id, ts_unix, actor_type, actor_id, agent_id, request_id,\n event_type, outcome, latency_ms, message, metadata_json)\n VALUES (?1, ?2, ?3, 'test', NULL, NULL, NULL, 'test', 'ok', NULL, 'old', '{}')",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 3
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "6ce45cc94cbfdd948aad5fb986f9b5976c79bb63759bc9ecff1a7873e2d70137"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "UPDATE known_devices SET display_name = ?1, updated_at_unix = ?2 WHERE device_id = ?3",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 3
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "897b794a7f94af42b545d74a5b859850fc9935f40385061b029fab5892a2aded"
|
||||||
|
}
|
||||||
+20
@@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "SELECT COUNT(*) FROM alert_transitions WHERE transition_id = ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [
|
||||||
|
{
|
||||||
|
"name": "COUNT(*)",
|
||||||
|
"ordinal": 0,
|
||||||
|
"type_info": "Integer"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": [
|
||||||
|
false
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"hash": "b324587b94ee2e98dc8eff5f6735303a2784b74f49068ce8e946cedcd8d12c4e"
|
||||||
|
}
|
||||||
+20
@@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "SELECT COUNT(*) FROM audit_events WHERE message = ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [
|
||||||
|
{
|
||||||
|
"name": "COUNT(*)",
|
||||||
|
"ordinal": 0,
|
||||||
|
"type_info": "Integer"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": [
|
||||||
|
false
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"hash": "bd1666cea8840df04ddcdc8a0c42b9b7771f9bf63c45613c4b6f8b86be479d70"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "DELETE FROM audit_events WHERE ts_unix < ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "ca612fee8c80f815b33dbc088cc7e0a7552f1f4e38390959e14ea5cc765bf2c7"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "DELETE FROM agent_devices WHERE last_seen_unix < ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "cc4b885c31f965013ccaa6f8ae33f6c8d68898654fc0873682c8212339088306"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "DELETE FROM alert_transitions WHERE ts_unix < ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "ce850a32c8702595bbc3207bf86d53209c48be145562192b3a8f84903d39eac1"
|
||||||
|
}
|
||||||
+20
@@ -0,0 +1,20 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "SELECT COUNT(*) FROM audit_events WHERE event_id = ?1",
|
||||||
|
"describe": {
|
||||||
|
"columns": [
|
||||||
|
{
|
||||||
|
"name": "COUNT(*)",
|
||||||
|
"ordinal": 0,
|
||||||
|
"type_info": "Integer"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 1
|
||||||
|
},
|
||||||
|
"nullable": [
|
||||||
|
false
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"hash": "dc7056b7421dcc83648cbd360273ebcce1156ce0f7da2d39f054025ab9b66b6b"
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
{
|
||||||
|
"db_name": "SQLite",
|
||||||
|
"query": "INSERT INTO alert_transitions\n (transition_key, transition_id, ts_unix, alert_id, kind, agent_id,\n from_status, to_status, message, metadata_json)\n VALUES (?1, ?2, ?3, 'alert-old', 'test', NULL, NULL, 'active', 'old', '{}')",
|
||||||
|
"describe": {
|
||||||
|
"columns": [],
|
||||||
|
"parameters": {
|
||||||
|
"Right": 3
|
||||||
|
},
|
||||||
|
"nullable": []
|
||||||
|
},
|
||||||
|
"hash": "e83a43217a3bfaacd2f5e76df1fd05447558646fb2b47963751a7ab53b3a99b6"
|
||||||
|
}
|
||||||
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "wakey-control-plane"
|
name = "wakey-control-plane"
|
||||||
version = "0.3.3"
|
version = "0.3.7"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
publish = ["gitea"]
|
publish = ["gitea"]
|
||||||
|
|
||||||
@@ -9,6 +9,9 @@ anyhow = { workspace = true }
|
|||||||
axum = { version = "0.8", features = ["ws", "json"] }
|
axum = { version = "0.8", features = ["ws", "json"] }
|
||||||
clap = { version = "4", features = ["derive"] }
|
clap = { version = "4", features = ["derive"] }
|
||||||
futures-util = "0.3"
|
futures-util = "0.3"
|
||||||
|
humantime-serde = "1"
|
||||||
|
http-cache-semantics = { version = "3", default-features = false, features = ["reqwest"] }
|
||||||
|
jsonwebtoken = { version = "10.3", default-features = false, features = ["aws_lc_rs"] }
|
||||||
macaddr = { workspace = true }
|
macaddr = { workspace = true }
|
||||||
nix = { version = "0.31", default-features = false, features = [
|
nix = { version = "0.31", default-features = false, features = [
|
||||||
"signal",
|
"signal",
|
||||||
@@ -20,6 +23,7 @@ reqwest = { version = "0.13", default-features = false, features = [
|
|||||||
] }
|
] }
|
||||||
serde = { workspace = true }
|
serde = { workspace = true }
|
||||||
serde_json = { workspace = true }
|
serde_json = { workspace = true }
|
||||||
|
thiserror = "2"
|
||||||
|
|
||||||
sqlx = { version = "0.8", features = [
|
sqlx = { version = "0.8", features = [
|
||||||
"runtime-tokio",
|
"runtime-tokio",
|
||||||
@@ -33,17 +37,13 @@ toml = "1.1"
|
|||||||
tokio = { workspace = true }
|
tokio = { workspace = true }
|
||||||
tower-http = { version = "0.6.8", features = ["fs"] }
|
tower-http = { version = "0.6.8", features = ["fs"] }
|
||||||
tracing = { workspace = true }
|
tracing = { workspace = true }
|
||||||
tracing-opentelemetry = "0.32"
|
|
||||||
tracing-subscriber = { version = "0.3", features = [
|
|
||||||
"env-filter",
|
|
||||||
"fmt",
|
|
||||||
"json",
|
|
||||||
"time",
|
|
||||||
] }
|
|
||||||
time = { version = "0.3", features = ["formatting", "local-offset"] }
|
|
||||||
opentelemetry = "0.31"
|
|
||||||
opentelemetry-otlp = { version = "0.31", features = ["grpc-tonic", "trace"] }
|
|
||||||
opentelemetry_sdk = { version = "0.31", features = ["rt-tokio"] }
|
|
||||||
uuid = { workspace = true }
|
uuid = { workspace = true }
|
||||||
wakey-agent = { path = "../wakey-agent", registry = "gitea", version = "0" }
|
wakey-agent = { path = "../wakey-agent", registry = "gitea", version = "0" }
|
||||||
wakey-core = { path = "../wakey-core", registry = "gitea", version = "0" }
|
wakey-core = { path = "../wakey-core", registry = "gitea", version = "0", features = ["telemetry"] }
|
||||||
|
|
||||||
|
[target.'cfg(target_os = "linux")'.dependencies]
|
||||||
|
wakey-linux = { path = "../wakey-linux", registry = "gitea", version = "0", features = ["telemetry"] }
|
||||||
|
|
||||||
|
[dev-dependencies]
|
||||||
|
tower = { version = "0.5", features = ["util"] }
|
||||||
|
jsonwebtoken = { version = "10.3", default-features = false, features = ["use_pem", "aws_lc_rs"] }
|
||||||
@@ -1,4 +1,5 @@
|
|||||||
use std::collections::{BTreeMap, HashSet};
|
use std::collections::{BTreeMap, HashSet};
|
||||||
|
use std::sync::Arc;
|
||||||
use std::time::{SystemTime, UNIX_EPOCH};
|
use std::time::{SystemTime, UNIX_EPOCH};
|
||||||
|
|
||||||
use axum::Json;
|
use axum::Json;
|
||||||
@@ -7,12 +8,14 @@ use axum::extract::{Query, State};
|
|||||||
use axum::http::StatusCode;
|
use axum::http::StatusCode;
|
||||||
use axum::response::IntoResponse;
|
use axum::response::IntoResponse;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
|
use tokio::select;
|
||||||
|
use tokio::sync::{RwLock, broadcast};
|
||||||
use tokio::time::Duration;
|
use tokio::time::Duration;
|
||||||
use tracing::warn;
|
use tracing::warn;
|
||||||
|
|
||||||
use crate::api::ApiError;
|
use crate::api::ApiError;
|
||||||
use crate::runtime::AppState;
|
use crate::runtime::AppState;
|
||||||
use crate::state::{AlertState, AuditEvent, AuditEventFilter};
|
use crate::state::{AlertState, AlertTransition, AuditEvent, AuditEventFilter};
|
||||||
|
|
||||||
struct AlertBuildContext<'a> {
|
struct AlertBuildContext<'a> {
|
||||||
now_unix: u64,
|
now_unix: u64,
|
||||||
@@ -26,21 +29,14 @@ struct AlertBuildContext<'a> {
|
|||||||
enroll_rejected_threshold: u64,
|
enroll_rejected_threshold: u64,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
|
||||||
pub struct ActiveAlertsQuery {
|
|
||||||
pub lookback_seconds: Option<u64>,
|
|
||||||
pub timeout_threshold: Option<u64>,
|
|
||||||
pub auth_rejected_threshold: Option<u64>,
|
|
||||||
pub enroll_rejected_threshold: Option<u64>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize)]
|
#[derive(Debug, Deserialize)]
|
||||||
pub struct AlertHistoryQuery {
|
pub struct AlertHistoryQuery {
|
||||||
pub since_unix: Option<u64>,
|
pub since_unix: Option<u64>,
|
||||||
pub limit: Option<usize>,
|
pub limit: Option<usize>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize, Clone)]
|
#[derive(Debug, Clone, Deserialize)]
|
||||||
|
#[serde(default)]
|
||||||
pub struct AlertRuleConfig {
|
pub struct AlertRuleConfig {
|
||||||
pub lookback_seconds: Option<u64>,
|
pub lookback_seconds: Option<u64>,
|
||||||
pub timeout_threshold: Option<u64>,
|
pub timeout_threshold: Option<u64>,
|
||||||
@@ -48,25 +44,255 @@ pub struct AlertRuleConfig {
|
|||||||
pub enroll_rejected_threshold: Option<u64>,
|
pub enroll_rejected_threshold: Option<u64>,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn active_alerts(
|
impl Default for AlertRuleConfig {
|
||||||
State(state): State<AppState>,
|
fn default() -> Self {
|
||||||
Query(query): Query<ActiveAlertsQuery>,
|
Self {
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
lookback_seconds: Some(900),
|
||||||
let alerts = evaluate_alerts(
|
timeout_threshold: Some(3),
|
||||||
&state,
|
auth_rejected_threshold: Some(3),
|
||||||
AlertRuleConfig {
|
enroll_rejected_threshold: Some(5),
|
||||||
lookback_seconds: query.lookback_seconds,
|
}
|
||||||
timeout_threshold: query.timeout_threshold,
|
}
|
||||||
auth_rejected_threshold: query.auth_rejected_threshold,
|
|
||||||
enroll_rejected_threshold: query.enroll_rejected_threshold,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
if let Err(err) = state.store.sync_alert_transitions(&alerts).await {
|
|
||||||
warn!(error = %err, "failed to sync alert transitions");
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl AlertRuleConfig {
|
||||||
|
fn validate(&self) -> Result<(), ApiError> {
|
||||||
|
let defaults = Self::default();
|
||||||
|
let overrides = [
|
||||||
|
(
|
||||||
|
"lookback_seconds",
|
||||||
|
self.lookback_seconds,
|
||||||
|
defaults.lookback_seconds,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"timeout_threshold",
|
||||||
|
self.timeout_threshold,
|
||||||
|
defaults.timeout_threshold,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"auth_rejected_threshold",
|
||||||
|
self.auth_rejected_threshold,
|
||||||
|
defaults.auth_rejected_threshold,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"enroll_rejected_threshold",
|
||||||
|
self.enroll_rejected_threshold,
|
||||||
|
defaults.enroll_rejected_threshold,
|
||||||
|
),
|
||||||
|
];
|
||||||
|
if let Some((field, _, _)) = overrides
|
||||||
|
.into_iter()
|
||||||
|
.find(|(_, requested, default)| requested.is_some() && requested != default)
|
||||||
|
{
|
||||||
|
return Err(ApiError::new(
|
||||||
|
StatusCode::BAD_REQUEST,
|
||||||
|
"unsupported_alert_override",
|
||||||
|
format!("custom alert {field} is not supported; omit it or use its default value"),
|
||||||
|
));
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Clone, Default)]
|
||||||
|
pub struct AlertsSnapshot {
|
||||||
|
pub ts_unix: u64,
|
||||||
|
pub alerts: Vec<AlertState>,
|
||||||
|
pub recent_transitions: Vec<AlertTransition>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Holds the latest alert snapshot computed by the background evaluator and
|
||||||
|
/// broadcasts each new snapshot to connected `alerts/ws` subscribers.
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct AlertsHub {
|
||||||
|
tx: broadcast::Sender<String>,
|
||||||
|
status: Arc<RwLock<HubStatus>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Default)]
|
||||||
|
struct HubStatus {
|
||||||
|
latest_payload: Option<String>,
|
||||||
|
latest_alerts: Vec<AlertState>,
|
||||||
|
last_success_unix: Option<u64>,
|
||||||
|
failure: Option<String>,
|
||||||
|
last_health_available: Option<bool>,
|
||||||
|
}
|
||||||
|
|
||||||
|
const ALERT_SNAPSHOT_MAX_AGE_SECS: u64 = 15;
|
||||||
|
|
||||||
|
impl Default for AlertsHub {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl AlertsHub {
|
||||||
|
pub fn new() -> Self {
|
||||||
|
let (tx, _) = broadcast::channel(16);
|
||||||
|
Self {
|
||||||
|
tx,
|
||||||
|
status: Arc::new(RwLock::new(HubStatus::default())),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn subscribe(&self) -> broadcast::Receiver<String> {
|
||||||
|
self.tx.subscribe()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn latest_alerts(&self) -> Vec<AlertState> {
|
||||||
|
self.status.read().await.latest_alerts.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn mark_unavailable(&self, reason: impl Into<String>) {
|
||||||
|
let mut status = self.status.write().await;
|
||||||
|
if status.failure.is_some() {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
status.failure = Some(reason.into());
|
||||||
|
let reason = status.failure.as_deref().unwrap_or_default().to_owned();
|
||||||
|
status.last_health_available = Some(false);
|
||||||
|
self.broadcast_health(false, Some(reason));
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn unavailable_reason(&self) -> Option<String> {
|
||||||
|
let status = self.status.read().await;
|
||||||
|
evaluator_unavailable_reason(&status, now_unix())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn resync_payloads(&self) -> (String, Option<String>) {
|
||||||
|
let status = self.status.read().await;
|
||||||
|
(
|
||||||
|
encode_health(evaluator_unavailable_reason(&status, now_unix())),
|
||||||
|
status.latest_payload.clone(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn broadcast_health_if_changed(&self) {
|
||||||
|
let mut status = self.status.write().await;
|
||||||
|
if status.last_success_unix.is_none() || status.failure.is_some() {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
let reason = evaluator_unavailable_reason(&status, now_unix());
|
||||||
|
let available = reason.is_none();
|
||||||
|
if status.last_health_available == Some(available) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
status.last_health_available = Some(available);
|
||||||
|
self.broadcast_health(available, reason);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn broadcast_health(&self, available: bool, reason: Option<String>) {
|
||||||
|
let _ = self.tx.send(encode_health_message(available, reason));
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn publish(&self, snapshot: AlertsSnapshot) {
|
||||||
|
let payload = serde_json::json!({
|
||||||
|
"type": "alerts_snapshot",
|
||||||
|
"ts_unix": snapshot.ts_unix,
|
||||||
|
"alerts": snapshot.alerts,
|
||||||
|
"recent_transitions": snapshot.recent_transitions,
|
||||||
|
});
|
||||||
|
let Ok(encoded) = serde_json::to_string(&payload) else {
|
||||||
|
warn!("failed to encode alerts snapshot payload");
|
||||||
|
return;
|
||||||
|
};
|
||||||
|
let mut status = self.status.write().await;
|
||||||
|
let was_unavailable = evaluator_unavailable_reason(&status, now_unix()).is_some();
|
||||||
|
status.latest_alerts = snapshot.alerts;
|
||||||
|
status.latest_payload = Some(encoded.clone());
|
||||||
|
status.last_success_unix = Some(snapshot.ts_unix);
|
||||||
|
status.failure = None;
|
||||||
|
let available = evaluator_unavailable_reason(&status, now_unix()).is_none();
|
||||||
|
if available {
|
||||||
|
status.last_health_available = Some(true);
|
||||||
|
}
|
||||||
|
if was_unavailable && available {
|
||||||
|
self.broadcast_health(true, None);
|
||||||
|
}
|
||||||
|
let _ = self.tx.send(encoded);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn evaluator_unavailable_reason(status: &HubStatus, now: u64) -> Option<String> {
|
||||||
|
if let Some(failure) = &status.failure {
|
||||||
|
return Some(failure.clone());
|
||||||
|
}
|
||||||
|
match status.last_success_unix {
|
||||||
|
Some(ts) if now.saturating_sub(ts) <= ALERT_SNAPSHOT_MAX_AGE_SECS => None,
|
||||||
|
Some(_) => Some("alert evaluator snapshot is stale".into()),
|
||||||
|
None => Some("alert evaluator has not produced a snapshot yet".into()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn encode_health(reason: Option<String>) -> String {
|
||||||
|
encode_health_message(reason.is_none(), reason)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn encode_health_message(available: bool, reason: Option<String>) -> String {
|
||||||
|
let mut message = serde_json::json!({
|
||||||
|
"type": "alerts_health",
|
||||||
|
"available": available,
|
||||||
|
});
|
||||||
|
if let Some(reason) = reason {
|
||||||
|
message["reason"] = serde_json::Value::String(reason);
|
||||||
|
}
|
||||||
|
message.to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Background task that owns alert evaluation: on a fixed ticker it recomputes
|
||||||
|
/// the snapshot, persists transitions, and publishes to subscribers so neither
|
||||||
|
/// the GET endpoint nor individual WS clients evaluate alerts themselves.
|
||||||
|
pub async fn run_alert_evaluator(state: AppState) {
|
||||||
|
let config = AlertRuleConfig::default();
|
||||||
|
let mut tick = tokio::time::interval(Duration::from_secs(5));
|
||||||
|
tick.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Skip);
|
||||||
|
loop {
|
||||||
|
tick.tick().await;
|
||||||
|
let alerts = match evaluate_alerts(&state, &config).await {
|
||||||
|
Ok(alerts) => alerts,
|
||||||
|
Err(err) => {
|
||||||
|
warn!(code = %err.code, "failed to evaluate alerts in background task");
|
||||||
|
state.alerts.mark_unavailable(err.message.clone()).await;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Err(err) = state.store.sync_alert_transitions(&alerts).await {
|
||||||
|
warn!(error = %err, "failed to sync alert transitions in background task");
|
||||||
|
}
|
||||||
|
|
||||||
|
let recent_transitions = match state.store.list_alert_transitions(None, 20).await {
|
||||||
|
Ok(history) => history,
|
||||||
|
Err(err) => {
|
||||||
|
warn!(error = %err, "failed to load alert transition history for snapshot");
|
||||||
|
Vec::new()
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
state
|
||||||
|
.alerts
|
||||||
|
.publish(AlertsSnapshot {
|
||||||
|
ts_unix: now_unix(),
|
||||||
|
alerts,
|
||||||
|
recent_transitions,
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn active_alerts(
|
||||||
|
State(state): State<AppState>,
|
||||||
|
Query(config): Query<AlertRuleConfig>,
|
||||||
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
|
config.validate()?;
|
||||||
|
if let Some(reason) = state.alerts.unavailable_reason().await {
|
||||||
|
return Err(ApiError::new(
|
||||||
|
StatusCode::SERVICE_UNAVAILABLE,
|
||||||
|
"alerts_unavailable",
|
||||||
|
reason,
|
||||||
|
));
|
||||||
|
}
|
||||||
|
let alerts = state.alerts.latest_alerts().await;
|
||||||
Ok((StatusCode::OK, Json(alerts)))
|
Ok((StatusCode::OK, Json(alerts)))
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -94,64 +320,62 @@ pub async fn alert_history(
|
|||||||
pub async fn alerts_stream(
|
pub async fn alerts_stream(
|
||||||
ws: WebSocketUpgrade,
|
ws: WebSocketUpgrade,
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Query(query): Query<ActiveAlertsQuery>,
|
|
||||||
) -> impl IntoResponse {
|
) -> impl IntoResponse {
|
||||||
let config = AlertRuleConfig {
|
ws.on_upgrade(move |socket| alerts_stream_socket(state, socket))
|
||||||
lookback_seconds: query.lookback_seconds,
|
}
|
||||||
timeout_threshold: query.timeout_threshold,
|
|
||||||
auth_rejected_threshold: query.auth_rejected_threshold,
|
async fn alerts_stream_socket(state: AppState, mut socket: WebSocket) {
|
||||||
enroll_rejected_threshold: query.enroll_rejected_threshold,
|
let mut rx = state.alerts.subscribe();
|
||||||
};
|
let mut freshness_tick = tokio::time::interval(Duration::from_secs(1));
|
||||||
ws.on_upgrade(move |socket| alerts_stream_socket(state, socket, config))
|
freshness_tick.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Skip);
|
||||||
|
|
||||||
|
let (health, snapshot) = state.alerts.resync_payloads().await;
|
||||||
|
if socket.send(Message::Text(health.into())).await.is_err() {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if let Some(encoded) = snapshot
|
||||||
|
&& socket.send(Message::Text(encoded.into())).await.is_err()
|
||||||
|
{
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn alerts_stream_socket(state: AppState, mut socket: WebSocket, config: AlertRuleConfig) {
|
|
||||||
let mut tick = tokio::time::interval(Duration::from_secs(5));
|
|
||||||
loop {
|
loop {
|
||||||
tick.tick().await;
|
select! {
|
||||||
let alerts = match evaluate_alerts(&state, config.clone()).await {
|
incoming = socket.recv() => {
|
||||||
Ok(alerts) => alerts,
|
match incoming {
|
||||||
Err(err) => {
|
Some(Ok(Message::Close(_))) | None | Some(Err(_)) => break,
|
||||||
warn!(code = %err.code, "failed to evaluate alerts for stream");
|
Some(Ok(_)) => {}
|
||||||
continue;
|
|
||||||
}
|
}
|
||||||
};
|
|
||||||
|
|
||||||
if let Err(err) = state.store.sync_alert_transitions(&alerts).await {
|
|
||||||
warn!(error = %err, "failed to sync alert transitions in stream");
|
|
||||||
}
|
}
|
||||||
|
update = rx.recv() => {
|
||||||
let history = match state.store.list_alert_transitions(None, 20).await {
|
match update {
|
||||||
Ok(h) => h,
|
Ok(encoded) => {
|
||||||
Err(err) => {
|
|
||||||
warn!(error = %err, "failed to load alert transition history for stream");
|
|
||||||
Vec::new()
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let payload = serde_json::json!({
|
|
||||||
"type": "alerts_snapshot",
|
|
||||||
"ts_unix": now_unix(),
|
|
||||||
"alerts": alerts,
|
|
||||||
"recent_transitions": history,
|
|
||||||
});
|
|
||||||
let encoded = match serde_json::to_string(&payload) {
|
|
||||||
Ok(s) => s,
|
|
||||||
Err(err) => {
|
|
||||||
warn!(error = %err, "failed to encode alerts stream payload");
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
if socket.send(Message::Text(encoded.into())).await.is_err() {
|
if socket.send(Message::Text(encoded.into())).await.is_err() {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Err(broadcast::error::RecvError::Lagged(_)) => {
|
||||||
|
let (health, snapshot) = state.alerts.resync_payloads().await;
|
||||||
|
if socket.send(Message::Text(health.into())).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if let Some(encoded) = snapshot
|
||||||
|
&& socket.send(Message::Text(encoded.into())).await.is_err()
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(broadcast::error::RecvError::Closed) => break,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ = freshness_tick.tick() => state.alerts.broadcast_health_if_changed().await,
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn evaluate_alerts(
|
async fn evaluate_alerts(
|
||||||
state: &AppState,
|
state: &AppState,
|
||||||
config: AlertRuleConfig,
|
config: &AlertRuleConfig,
|
||||||
) -> Result<Vec<AlertState>, ApiError> {
|
) -> Result<Vec<AlertState>, ApiError> {
|
||||||
let lookback_seconds = config.lookback_seconds.unwrap_or(900).clamp(60, 86_400);
|
let lookback_seconds = config.lookback_seconds.unwrap_or(900).clamp(60, 86_400);
|
||||||
let timeout_threshold = config.timeout_threshold.unwrap_or(3).max(1);
|
let timeout_threshold = config.timeout_threshold.unwrap_or(3).max(1);
|
||||||
@@ -161,7 +385,14 @@ async fn evaluate_alerts(
|
|||||||
let now = now_unix();
|
let now = now_unix();
|
||||||
let since_unix = now.saturating_sub(lookback_seconds);
|
let since_unix = now.saturating_sub(lookback_seconds);
|
||||||
|
|
||||||
let enrolled_agents = state.store.list_agents().await;
|
let enrolled_agents = state.store.list_agents().await.map_err(|err| {
|
||||||
|
warn!(error = %err, "failed listing agents for alert evaluation");
|
||||||
|
ApiError::new(
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR,
|
||||||
|
"alerts_query_failed",
|
||||||
|
err.to_string(),
|
||||||
|
)
|
||||||
|
})?;
|
||||||
let connected_agents = state
|
let connected_agents = state
|
||||||
.sessions
|
.sessions
|
||||||
.read()
|
.read()
|
||||||
@@ -359,9 +590,11 @@ fn now_unix() -> u64 {
|
|||||||
mod tests {
|
mod tests {
|
||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
|
|
||||||
|
use axum::http::StatusCode;
|
||||||
|
|
||||||
use crate::state::AuditEvent;
|
use crate::state::AuditEvent;
|
||||||
|
|
||||||
use super::{AlertBuildContext, build_alerts};
|
use super::{AlertBuildContext, AlertRuleConfig, AlertsHub, AlertsSnapshot, build_alerts};
|
||||||
|
|
||||||
fn event(agent_id: Option<&str>, event_type: &str, outcome: &str, ts_unix: u64) -> AuditEvent {
|
fn event(agent_id: Option<&str>, event_type: &str, outcome: &str, ts_unix: u64) -> AuditEvent {
|
||||||
AuditEvent {
|
AuditEvent {
|
||||||
@@ -450,4 +683,146 @@ mod tests {
|
|||||||
.any(|a| a.kind == "enroll_reject_spike" && a.agent_id.is_none())
|
.any(|a| a.kind == "enroll_reject_spike" && a.agent_id.is_none())
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn alert_rule_config_accepts_omitted_and_default_values() {
|
||||||
|
AlertRuleConfig::default().validate().unwrap();
|
||||||
|
AlertRuleConfig {
|
||||||
|
lookback_seconds: None,
|
||||||
|
timeout_threshold: None,
|
||||||
|
auth_rejected_threshold: None,
|
||||||
|
enroll_rejected_threshold: None,
|
||||||
|
}
|
||||||
|
.validate()
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn alert_rule_config_rejects_custom_overrides() {
|
||||||
|
let configs = [
|
||||||
|
AlertRuleConfig {
|
||||||
|
lookback_seconds: Some(300),
|
||||||
|
..Default::default()
|
||||||
|
},
|
||||||
|
AlertRuleConfig {
|
||||||
|
timeout_threshold: Some(4),
|
||||||
|
..Default::default()
|
||||||
|
},
|
||||||
|
AlertRuleConfig {
|
||||||
|
auth_rejected_threshold: Some(4),
|
||||||
|
..Default::default()
|
||||||
|
},
|
||||||
|
AlertRuleConfig {
|
||||||
|
enroll_rejected_threshold: Some(6),
|
||||||
|
..Default::default()
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
for config in configs {
|
||||||
|
let error = config.validate().unwrap_err();
|
||||||
|
assert_eq!(error.status, StatusCode::BAD_REQUEST);
|
||||||
|
assert_eq!(error.code, "unsupported_alert_override");
|
||||||
|
assert!(error.message.contains("not supported"));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn alert_hub_reports_initial_failure_and_recovers_on_snapshot() {
|
||||||
|
let hub = AlertsHub::new();
|
||||||
|
assert!(hub.unavailable_reason().await.is_some());
|
||||||
|
let mut updates = hub.subscribe();
|
||||||
|
|
||||||
|
hub.mark_unavailable("store query failed").await;
|
||||||
|
assert_eq!(
|
||||||
|
hub.unavailable_reason().await.as_deref(),
|
||||||
|
Some("store query failed")
|
||||||
|
);
|
||||||
|
let failure: serde_json::Value =
|
||||||
|
serde_json::from_str(&updates.try_recv().unwrap()).unwrap();
|
||||||
|
assert_eq!(failure["type"], "alerts_health");
|
||||||
|
assert_eq!(failure["available"], false);
|
||||||
|
assert_eq!(failure["reason"], "store query failed");
|
||||||
|
|
||||||
|
hub.mark_unavailable("same outage, later attempt").await;
|
||||||
|
assert!(
|
||||||
|
updates.try_recv().is_err(),
|
||||||
|
"failure notifications deduplicate"
|
||||||
|
);
|
||||||
|
|
||||||
|
hub.publish(AlertsSnapshot {
|
||||||
|
ts_unix: super::now_unix(),
|
||||||
|
alerts: Vec::new(),
|
||||||
|
recent_transitions: Vec::new(),
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
assert_eq!(hub.unavailable_reason().await, None);
|
||||||
|
let recovery: serde_json::Value =
|
||||||
|
serde_json::from_str(&updates.try_recv().unwrap()).unwrap();
|
||||||
|
assert_eq!(recovery["type"], "alerts_health");
|
||||||
|
assert_eq!(recovery["available"], true);
|
||||||
|
assert_eq!(recovery.get("reason"), None);
|
||||||
|
let snapshot: serde_json::Value =
|
||||||
|
serde_json::from_str(&updates.try_recv().unwrap()).unwrap();
|
||||||
|
assert_eq!(snapshot["type"], "alerts_snapshot");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn alert_hub_rejects_stale_snapshots() {
|
||||||
|
let hub = AlertsHub::new();
|
||||||
|
hub.publish(AlertsSnapshot {
|
||||||
|
ts_unix: 0,
|
||||||
|
alerts: Vec::new(),
|
||||||
|
recent_transitions: Vec::new(),
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
assert_eq!(
|
||||||
|
hub.unavailable_reason().await.as_deref(),
|
||||||
|
Some("alert evaluator snapshot is stale")
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn alert_hub_initial_resync_marks_cached_snapshot_stale() {
|
||||||
|
let hub = AlertsHub::new();
|
||||||
|
let (initial_health, initial_snapshot) = hub.resync_payloads().await;
|
||||||
|
let initial_health: serde_json::Value = serde_json::from_str(&initial_health).unwrap();
|
||||||
|
assert_eq!(initial_health["available"], false);
|
||||||
|
assert_eq!(
|
||||||
|
initial_health["reason"],
|
||||||
|
"alert evaluator has not produced a snapshot yet"
|
||||||
|
);
|
||||||
|
assert!(initial_snapshot.is_none());
|
||||||
|
|
||||||
|
hub.publish(AlertsSnapshot {
|
||||||
|
ts_unix: 0,
|
||||||
|
alerts: Vec::new(),
|
||||||
|
recent_transitions: Vec::new(),
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
let (stale_health, cached_snapshot) = hub.resync_payloads().await;
|
||||||
|
let stale_health: serde_json::Value = serde_json::from_str(&stale_health).unwrap();
|
||||||
|
assert_eq!(stale_health["available"], false);
|
||||||
|
assert_eq!(stale_health["reason"], "alert evaluator snapshot is stale");
|
||||||
|
assert!(cached_snapshot.is_some());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn alert_hub_broadcasts_stale_transition_once() {
|
||||||
|
let hub = AlertsHub::new();
|
||||||
|
let mut updates = hub.subscribe();
|
||||||
|
hub.publish(AlertsSnapshot {
|
||||||
|
ts_unix: 0,
|
||||||
|
alerts: Vec::new(),
|
||||||
|
recent_transitions: Vec::new(),
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
let _snapshot = updates.try_recv().unwrap();
|
||||||
|
|
||||||
|
hub.broadcast_health_if_changed().await;
|
||||||
|
let stale: serde_json::Value = serde_json::from_str(&updates.try_recv().unwrap()).unwrap();
|
||||||
|
assert_eq!(stale["available"], false);
|
||||||
|
assert_eq!(stale["reason"], "alert evaluator snapshot is stale");
|
||||||
|
hub.broadcast_health_if_changed().await;
|
||||||
|
assert!(updates.try_recv().is_err());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -4,7 +4,7 @@ use axum::http::StatusCode;
|
|||||||
use axum::response::IntoResponse;
|
use axum::response::IntoResponse;
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use std::time::Instant;
|
use std::time::Instant;
|
||||||
use tracing::{info, info_span, warn};
|
use tracing::{info, warn};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
use wakey_agent::protocol::{
|
use wakey_agent::protocol::{
|
||||||
AgentCapability, AgentCapabilityOptions, AgentCommand, ErrorPayload, RequestId, ServerMessage,
|
AgentCapability, AgentCapabilityOptions, AgentCommand, ErrorPayload, RequestId, ServerMessage,
|
||||||
@@ -40,7 +40,18 @@ pub struct RelayCommandResponse {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub async fn list_agents(State(state): State<AppState>) -> Result<impl IntoResponse, ApiError> {
|
pub async fn list_agents(State(state): State<AppState>) -> Result<impl IntoResponse, ApiError> {
|
||||||
let enrolled = state.store.list_agents_with_nicknames().await;
|
let enrolled = state
|
||||||
|
.store
|
||||||
|
.list_agents_with_nicknames()
|
||||||
|
.await
|
||||||
|
.map_err(|err| {
|
||||||
|
warn!(error = %err, "failed listing agents");
|
||||||
|
ApiError::new(
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR,
|
||||||
|
"list_agents_failed",
|
||||||
|
err.to_string(),
|
||||||
|
)
|
||||||
|
})?;
|
||||||
let sessions = state.sessions.read().await;
|
let sessions = state.sessions.read().await;
|
||||||
|
|
||||||
let agents = enrolled
|
let agents = enrolled
|
||||||
@@ -74,6 +85,7 @@ pub async fn run_command(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "relay_command", skip_all, fields(agent_id = %agent_id, request_id = tracing::field::Empty, command = tracing::field::Empty))]
|
||||||
pub async fn relay_agent_command(
|
pub async fn relay_agent_command(
|
||||||
state: &AppState,
|
state: &AppState,
|
||||||
agent_id: &str,
|
agent_id: &str,
|
||||||
@@ -82,14 +94,11 @@ pub async fn relay_agent_command(
|
|||||||
) -> Result<RelayCommandResponse, ApiError> {
|
) -> Result<RelayCommandResponse, ApiError> {
|
||||||
let request_id_string = format!("req-{}", Uuid::new_v4());
|
let request_id_string = format!("req-{}", Uuid::new_v4());
|
||||||
let command_kind = command_kind(&command);
|
let command_kind = command_kind(&command);
|
||||||
|
tracing::Span::current()
|
||||||
|
.record("request_id", &request_id_string)
|
||||||
|
.record("command", command_kind);
|
||||||
let started = Instant::now();
|
let started = Instant::now();
|
||||||
let span = info_span!(
|
let mut operation = wakey_core::telemetry::Operation::new(command_kind);
|
||||||
"relay_command",
|
|
||||||
agent_id = %agent_id,
|
|
||||||
request_id = %request_id_string,
|
|
||||||
command = %command_kind,
|
|
||||||
);
|
|
||||||
let _span_guard = span.enter();
|
|
||||||
|
|
||||||
let request_id = RequestId::try_from(request_id_string.clone()).map_err(|err| {
|
let request_id = RequestId::try_from(request_id_string.clone()).map_err(|err| {
|
||||||
ApiError::new(
|
ApiError::new(
|
||||||
@@ -120,9 +129,9 @@ pub async fn relay_agent_command(
|
|||||||
.insert(request_id_string.clone(), pending_tx);
|
.insert(request_id_string.clone(), pending_tx);
|
||||||
|
|
||||||
info!("dispatching command to agent");
|
info!("dispatching command to agent");
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.to_string()),
|
agent_id: Some(agent_id.to_string()),
|
||||||
@@ -133,20 +142,18 @@ pub async fn relay_agent_command(
|
|||||||
message: "dispatched command to connected agent".into(),
|
message: "dispatched command to connected agent".into(),
|
||||||
metadata: serde_json::json!({ "command": command_kind }),
|
metadata: serde_json::json!({ "command": command_kind }),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for command dispatch");
|
|
||||||
}
|
|
||||||
|
|
||||||
if let Err(err) = tx.send(SessionEvent::Message(ServerMessage::Command {
|
if let Err(err) = tx.send(SessionEvent::Message(ServerMessage::Command {
|
||||||
request_id,
|
request_id,
|
||||||
command,
|
command,
|
||||||
|
trace_context: wakey_core::telemetry::current_trace_context(),
|
||||||
})) {
|
})) {
|
||||||
state.pending.lock().await.remove(&request_id_string);
|
state.pending.lock().await.remove(&request_id_string);
|
||||||
warn!(error = %err, "failed sending command to agent session");
|
warn!(error = %err, "failed sending command to agent session");
|
||||||
if let Err(audit_err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.to_string()),
|
agent_id: Some(agent_id.to_string()),
|
||||||
@@ -157,10 +164,7 @@ pub async fn relay_agent_command(
|
|||||||
message: err.to_string(),
|
message: err.to_string(),
|
||||||
metadata: serde_json::json!({ "command": command_kind }),
|
metadata: serde_json::json!({ "command": command_kind }),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %audit_err, "failed to append audit event for command send failure");
|
|
||||||
}
|
|
||||||
return Err(ApiError::new(
|
return Err(ApiError::new(
|
||||||
StatusCode::BAD_GATEWAY,
|
StatusCode::BAD_GATEWAY,
|
||||||
"agent_send_failed",
|
"agent_send_failed",
|
||||||
@@ -168,18 +172,16 @@ pub async fn relay_agent_command(
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
|
|
||||||
let timeout = std::time::Duration::from_millis(
|
let timeout = timeout_ms
|
||||||
timeout_ms
|
.map(|value| std::time::Duration::from_millis(value.max(1)))
|
||||||
.unwrap_or(state.command_timeout.as_millis() as u64)
|
.unwrap_or(state.command_timeout);
|
||||||
.max(1),
|
|
||||||
);
|
|
||||||
let outcome = tokio::time::timeout(timeout, pending_rx).await;
|
let outcome = tokio::time::timeout(timeout, pending_rx).await;
|
||||||
let response = match outcome {
|
let response = match outcome {
|
||||||
Ok(Ok(AgentReply::Result(result))) => {
|
Ok(Ok(AgentReply::Result(result))) => {
|
||||||
info!("agent command completed");
|
info!("agent command completed");
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.to_string()),
|
agent_id: Some(agent_id.to_string()),
|
||||||
@@ -190,10 +192,7 @@ pub async fn relay_agent_command(
|
|||||||
message: "agent command completed".into(),
|
message: "agent command completed".into(),
|
||||||
metadata: serde_json::json!({ "command": command_kind }),
|
metadata: serde_json::json!({ "command": command_kind }),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for command success");
|
|
||||||
}
|
|
||||||
RelayCommandResponse {
|
RelayCommandResponse {
|
||||||
request_id: request_id_string,
|
request_id: request_id_string,
|
||||||
status: "ok".into(),
|
status: "ok".into(),
|
||||||
@@ -203,9 +202,9 @@ pub async fn relay_agent_command(
|
|||||||
}
|
}
|
||||||
Ok(Ok(AgentReply::Error(error))) => {
|
Ok(Ok(AgentReply::Error(error))) => {
|
||||||
warn!(code = %error.code, "agent command returned error");
|
warn!(code = %error.code, "agent command returned error");
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.to_string()),
|
agent_id: Some(agent_id.to_string()),
|
||||||
@@ -216,10 +215,7 @@ pub async fn relay_agent_command(
|
|||||||
message: error.message.clone(),
|
message: error.message.clone(),
|
||||||
metadata: serde_json::json!({ "command": command_kind, "code": error.code }),
|
metadata: serde_json::json!({ "command": command_kind, "code": error.code }),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for command error result");
|
|
||||||
}
|
|
||||||
RelayCommandResponse {
|
RelayCommandResponse {
|
||||||
request_id: request_id_string,
|
request_id: request_id_string,
|
||||||
status: "error".into(),
|
status: "error".into(),
|
||||||
@@ -229,9 +225,9 @@ pub async fn relay_agent_command(
|
|||||||
}
|
}
|
||||||
Ok(Err(_)) => {
|
Ok(Err(_)) => {
|
||||||
warn!("agent response channel dropped");
|
warn!("agent response channel dropped");
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.to_string()),
|
agent_id: Some(agent_id.to_string()),
|
||||||
@@ -242,10 +238,7 @@ pub async fn relay_agent_command(
|
|||||||
message: "agent response channel dropped".into(),
|
message: "agent response channel dropped".into(),
|
||||||
metadata: serde_json::json!({ "command": command_kind }),
|
metadata: serde_json::json!({ "command": command_kind }),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for dropped response");
|
|
||||||
}
|
|
||||||
return Err(ApiError::new(
|
return Err(ApiError::new(
|
||||||
StatusCode::BAD_GATEWAY,
|
StatusCode::BAD_GATEWAY,
|
||||||
"agent_response_dropped",
|
"agent_response_dropped",
|
||||||
@@ -253,14 +246,15 @@ pub async fn relay_agent_command(
|
|||||||
));
|
));
|
||||||
}
|
}
|
||||||
Err(_) => {
|
Err(_) => {
|
||||||
|
operation.outcome("timeout");
|
||||||
state.pending.lock().await.remove(&request_id_string);
|
state.pending.lock().await.remove(&request_id_string);
|
||||||
warn!(
|
warn!(
|
||||||
timeout_ms = timeout.as_millis() as u64,
|
timeout_ms = timeout.as_millis() as u64,
|
||||||
"agent command timed out"
|
"agent command timed out"
|
||||||
);
|
);
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.to_string()),
|
agent_id: Some(agent_id.to_string()),
|
||||||
@@ -271,10 +265,7 @@ pub async fn relay_agent_command(
|
|||||||
message: "agent command timed out".into(),
|
message: "agent command timed out".into(),
|
||||||
metadata: serde_json::json!({ "command": command_kind, "timeout_ms": timeout.as_millis() as u64 }),
|
metadata: serde_json::json!({ "command": command_kind, "timeout_ms": timeout.as_millis() as u64 }),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for timeout");
|
|
||||||
}
|
|
||||||
return Err(ApiError::new(
|
return Err(ApiError::new(
|
||||||
StatusCode::GATEWAY_TIMEOUT,
|
StatusCode::GATEWAY_TIMEOUT,
|
||||||
"agent_timeout",
|
"agent_timeout",
|
||||||
@@ -283,6 +274,11 @@ pub async fn relay_agent_command(
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
operation.outcome(if response.status == "ok" {
|
||||||
|
"ok"
|
||||||
|
} else {
|
||||||
|
"error"
|
||||||
|
});
|
||||||
Ok(response)
|
Ok(response)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -5,12 +5,14 @@ mod stats;
|
|||||||
|
|
||||||
pub use devices::{
|
pub use devices::{
|
||||||
attach_device_identifier, create_known_device, detach_device_identifier, forget_known_device,
|
attach_device_identifier, create_known_device, detach_device_identifier, forget_known_device,
|
||||||
get_known_device, list_known_devices, merge_known_device,
|
get_known_device, list_known_devices, merge_known_device, rename_known_device,
|
||||||
};
|
};
|
||||||
pub use enroll::{
|
pub use enroll::{
|
||||||
EnrollTokenStatus, IssueEnrollTokenResponse, RevokeAgentResponse, RevokeEnrollTokenResponse,
|
EnrollTokenStatus, IssueEnrollTokenResponse, RevokeAgentResponse, RevokeEnrollTokenResponse,
|
||||||
enroll, healthz, issue_enroll_token, list_enroll_tokens, revoke_agent, revoke_enroll_token,
|
enroll, healthz, issue_enroll_token, list_enroll_tokens, revoke_agent, revoke_enroll_token,
|
||||||
set_agent_nickname,
|
set_agent_nickname,
|
||||||
};
|
};
|
||||||
pub use fleet::{list_fleet_devices, refresh_fleet_devices, wake_fleet_device};
|
pub(crate) use fleet::FleetCache;
|
||||||
|
pub(crate) use fleet::sample_fleet_counts;
|
||||||
|
pub use fleet::{get_fleet_device, list_fleet_devices, refresh_fleet_devices, wake_fleet_device};
|
||||||
pub use stats::{StateStatsResponse, state_stats};
|
pub use stats::{StateStatsResponse, state_stats};
|
||||||
@@ -25,6 +25,47 @@ pub struct DeviceIdentifierRequest {
|
|||||||
pub value: String,
|
pub value: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Deserialize)]
|
||||||
|
pub struct RenameKnownDeviceRequest {
|
||||||
|
pub display_name: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn rename_known_device(
|
||||||
|
State(state): State<AppState>,
|
||||||
|
AxumPath(device_id): AxumPath<String>,
|
||||||
|
Json(req): Json<RenameKnownDeviceRequest>,
|
||||||
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
|
// The store validates too, but checking here distinguishes bad input from
|
||||||
|
// storage failures without interpreting anyhow error messages.
|
||||||
|
if req.display_name.trim().is_empty() {
|
||||||
|
return Err(ApiError::new(
|
||||||
|
StatusCode::BAD_REQUEST,
|
||||||
|
"rename_known_device_failed",
|
||||||
|
"display_name must not be empty",
|
||||||
|
));
|
||||||
|
}
|
||||||
|
match state
|
||||||
|
.store
|
||||||
|
.rename_known_device(&device_id, &req.display_name)
|
||||||
|
.await
|
||||||
|
{
|
||||||
|
Ok(Some(device)) => Ok((StatusCode::OK, Json(known_device_response(device)))),
|
||||||
|
Ok(None) => Err(ApiError::new(
|
||||||
|
StatusCode::NOT_FOUND,
|
||||||
|
"known_device_not_found",
|
||||||
|
"known device not found",
|
||||||
|
)),
|
||||||
|
Err(err) => {
|
||||||
|
warn!(error = %err, "failed to rename known device");
|
||||||
|
Err(ApiError::new(
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR,
|
||||||
|
"rename_known_device_failed",
|
||||||
|
err.to_string(),
|
||||||
|
))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
pub struct KnownDeviceResponse {
|
pub struct KnownDeviceResponse {
|
||||||
pub device_id: String,
|
pub device_id: String,
|
||||||
@@ -262,3 +303,112 @@ fn known_device_response(device: crate::state::KnownDevice) -> KnownDeviceRespon
|
|||||||
.collect(),
|
.collect(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use std::time::Duration;
|
||||||
|
use tokio::sync::{Mutex, RwLock};
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn rename_distinguishes_invalid_names_missing_devices_and_storage_failures() {
|
||||||
|
let dir = std::env::temp_dir().join(format!("wakey-rename-api-{}", uuid::Uuid::new_v4()));
|
||||||
|
let store = Arc::new(
|
||||||
|
crate::state::Store::load_or_init(
|
||||||
|
&dir.join("state.sqlite3"),
|
||||||
|
vec![],
|
||||||
|
Duration::from_secs(60),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap(),
|
||||||
|
);
|
||||||
|
let state = AppState {
|
||||||
|
store: store.clone(),
|
||||||
|
sessions: Arc::new(RwLock::new(HashMap::new())),
|
||||||
|
pending: Arc::new(Mutex::new(HashMap::new())),
|
||||||
|
public_url: "http://localhost".into(),
|
||||||
|
command_timeout: Duration::from_secs(30),
|
||||||
|
enroll_token_ttl: Duration::from_secs(60),
|
||||||
|
terminals: crate::runtime::terminals::TerminalRegistry::new(),
|
||||||
|
alerts: crate::api::AlertsHub::new(),
|
||||||
|
fleet: Arc::new(crate::api::FleetCache::default()),
|
||||||
|
};
|
||||||
|
let device = store
|
||||||
|
.create_known_device(KnownDeviceInput {
|
||||||
|
display_name: "before".into(),
|
||||||
|
pinned: false,
|
||||||
|
notes: None,
|
||||||
|
identifiers: vec![],
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let rename = |id: String, name: &str| {
|
||||||
|
rename_known_device(
|
||||||
|
State(state.clone()),
|
||||||
|
AxumPath(id),
|
||||||
|
Json(RenameKnownDeviceRequest {
|
||||||
|
display_name: name.into(),
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
};
|
||||||
|
let response = rename(device.device_id.clone(), " after ")
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.into_response();
|
||||||
|
assert_eq!(response.status(), StatusCode::OK);
|
||||||
|
assert_eq!(
|
||||||
|
store
|
||||||
|
.get_known_device(&device.device_id)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.unwrap()
|
||||||
|
.display_name,
|
||||||
|
"after"
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
rename("missing".into(), "valid")
|
||||||
|
.await
|
||||||
|
.err()
|
||||||
|
.unwrap()
|
||||||
|
.status,
|
||||||
|
StatusCode::NOT_FOUND
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
rename(device.device_id.clone(), " \t\n ")
|
||||||
|
.await
|
||||||
|
.err()
|
||||||
|
.unwrap()
|
||||||
|
.status,
|
||||||
|
StatusCode::BAD_REQUEST
|
||||||
|
);
|
||||||
|
// A deterministic SQLite failure, not a clock-sensitive lock timeout.
|
||||||
|
let mut tx = store.begin_write().await.unwrap();
|
||||||
|
sqlx::query("CREATE TRIGGER reject_rename BEFORE UPDATE ON known_devices BEGIN SELECT RAISE(ABORT, 'test rename failure'); END")
|
||||||
|
.execute(&mut *tx).await.unwrap();
|
||||||
|
tx.commit().await.unwrap();
|
||||||
|
let failure = rename(device.device_id.clone(), "valid")
|
||||||
|
.await
|
||||||
|
.err()
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(failure.status, StatusCode::INTERNAL_SERVER_ERROR);
|
||||||
|
assert_eq!(failure.code, "rename_known_device_failed");
|
||||||
|
assert_eq!(
|
||||||
|
store
|
||||||
|
.get_known_device(&device.device_id)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.unwrap()
|
||||||
|
.display_name,
|
||||||
|
"after"
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
rename(device.device_id, " ").await.err().unwrap().status,
|
||||||
|
StatusCode::BAD_REQUEST
|
||||||
|
);
|
||||||
|
drop(state);
|
||||||
|
drop(store);
|
||||||
|
std::fs::remove_dir_all(dir).unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -74,9 +74,9 @@ pub async fn enroll(
|
|||||||
match state.store.enroll(&req.enroll_token).await {
|
match state.store.enroll(&req.enroll_token).await {
|
||||||
Ok(issued) => {
|
Ok(issued) => {
|
||||||
info!(agent_id = %issued.agent_id, "agent enrollment accepted");
|
info!(agent_id = %issued.agent_id, "agent enrollment accepted");
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "agent".into(),
|
actor_type: "agent".into(),
|
||||||
actor_id: Some(issued.agent_id.clone()),
|
actor_id: Some(issued.agent_id.clone()),
|
||||||
agent_id: Some(issued.agent_id.clone()),
|
agent_id: Some(issued.agent_id.clone()),
|
||||||
@@ -87,10 +87,7 @@ pub async fn enroll(
|
|||||||
message: "agent enrollment accepted".into(),
|
message: "agent enrollment accepted".into(),
|
||||||
metadata: serde_json::json!({}),
|
metadata: serde_json::json!({}),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for enroll success");
|
|
||||||
}
|
|
||||||
Ok((
|
Ok((
|
||||||
StatusCode::OK,
|
StatusCode::OK,
|
||||||
Json(EnrollResponse {
|
Json(EnrollResponse {
|
||||||
@@ -102,9 +99,9 @@ pub async fn enroll(
|
|||||||
}
|
}
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
warn!(error = %err, "agent enrollment rejected");
|
warn!(error = %err, "agent enrollment rejected");
|
||||||
if let Err(audit_err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "agent".into(),
|
actor_type: "agent".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: None,
|
agent_id: None,
|
||||||
@@ -115,10 +112,7 @@ pub async fn enroll(
|
|||||||
message: err.to_string(),
|
message: err.to_string(),
|
||||||
metadata: serde_json::json!({}),
|
metadata: serde_json::json!({}),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %audit_err, "failed to append audit event for enroll rejection");
|
|
||||||
}
|
|
||||||
Err(ApiError::new(
|
Err(ApiError::new(
|
||||||
StatusCode::UNAUTHORIZED,
|
StatusCode::UNAUTHORIZED,
|
||||||
"enrollment_rejected",
|
"enrollment_rejected",
|
||||||
@@ -145,9 +139,9 @@ pub async fn issue_enroll_token(
|
|||||||
expires_at_unix = issued.expires_at_unix,
|
expires_at_unix = issued.expires_at_unix,
|
||||||
"issued enroll token"
|
"issued enroll token"
|
||||||
);
|
);
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: None,
|
agent_id: None,
|
||||||
@@ -161,10 +155,7 @@ pub async fn issue_enroll_token(
|
|||||||
"expires_at_unix": issued.expires_at_unix,
|
"expires_at_unix": issued.expires_at_unix,
|
||||||
}),
|
}),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for token issuance");
|
|
||||||
}
|
|
||||||
Ok((
|
Ok((
|
||||||
StatusCode::OK,
|
StatusCode::OK,
|
||||||
Json(IssueEnrollTokenResponse {
|
Json(IssueEnrollTokenResponse {
|
||||||
@@ -189,9 +180,9 @@ pub async fn list_enroll_tokens(
|
|||||||
) -> Result<impl IntoResponse, ApiError> {
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
match state.store.list_enroll_tokens().await {
|
match state.store.list_enroll_tokens().await {
|
||||||
Ok(tokens) => {
|
Ok(tokens) => {
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: None,
|
agent_id: None,
|
||||||
@@ -204,10 +195,7 @@ pub async fn list_enroll_tokens(
|
|||||||
"count": tokens.len(),
|
"count": tokens.len(),
|
||||||
}),
|
}),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for token listing");
|
|
||||||
}
|
|
||||||
let body = tokens
|
let body = tokens
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|t| EnrollTokenStatus {
|
.map(|t| EnrollTokenStatus {
|
||||||
@@ -235,9 +223,9 @@ pub async fn revoke_enroll_token(
|
|||||||
) -> Result<impl IntoResponse, ApiError> {
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
match state.store.revoke_enroll_token(&token).await {
|
match state.store.revoke_enroll_token(&token).await {
|
||||||
Ok(revoked) => {
|
Ok(revoked) => {
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: None,
|
agent_id: None,
|
||||||
@@ -256,10 +244,7 @@ pub async fn revoke_enroll_token(
|
|||||||
},
|
},
|
||||||
metadata: serde_json::json!({ "token": token }),
|
metadata: serde_json::json!({ "token": token }),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for token revoke");
|
|
||||||
}
|
|
||||||
Ok((
|
Ok((
|
||||||
StatusCode::OK,
|
StatusCode::OK,
|
||||||
Json(RevokeEnrollTokenResponse { token, revoked }),
|
Json(RevokeEnrollTokenResponse { token, revoked }),
|
||||||
@@ -289,9 +274,9 @@ pub async fn revoke_agent(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.clone()),
|
agent_id: Some(agent_id.clone()),
|
||||||
@@ -310,10 +295,7 @@ pub async fn revoke_agent(
|
|||||||
},
|
},
|
||||||
metadata: serde_json::json!({ "agent_id": agent_id }),
|
metadata: serde_json::json!({ "agent_id": agent_id }),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for agent revoke");
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok((
|
Ok((
|
||||||
StatusCode::OK,
|
StatusCode::OK,
|
||||||
@@ -349,9 +331,9 @@ pub async fn set_agent_nickname(
|
|||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(updated) => {
|
Ok(updated) => {
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: "admin_api".into(),
|
actor_type: "admin_api".into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.clone()),
|
agent_id: Some(agent_id.clone()),
|
||||||
@@ -373,10 +355,7 @@ pub async fn set_agent_nickname(
|
|||||||
"nickname": normalized,
|
"nickname": normalized,
|
||||||
}),
|
}),
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(error = %err, "failed to append audit event for nickname set");
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok((
|
Ok((
|
||||||
StatusCode::OK,
|
StatusCode::OK,
|
||||||
|
|||||||
@@ -2,10 +2,15 @@ use std::collections::{BTreeMap, BTreeSet, HashMap};
|
|||||||
use std::net::IpAddr;
|
use std::net::IpAddr;
|
||||||
|
|
||||||
use macaddr::MacAddr;
|
use macaddr::MacAddr;
|
||||||
use wakey_core::{EndpointSource, Presence};
|
use wakey_core::Presence;
|
||||||
|
|
||||||
use crate::state::{AgentDeviceWithChildren, DeviceIdentifier, KnownDevice, KnownDeviceSummary};
|
use crate::state::{AgentDeviceWithChildren, DeviceIdentifier, KnownDevice, KnownDeviceSummary};
|
||||||
|
|
||||||
|
mod addresses;
|
||||||
|
mod routes;
|
||||||
|
use addresses::add_address_evidence;
|
||||||
|
use routes::{add_endpoint_routes, add_legacy_routes, compare_route_quality};
|
||||||
|
|
||||||
use super::types::{
|
use super::types::{
|
||||||
FleetDevice, FleetDeviceAgent, FleetDeviceEndpoint, FleetWakeRoute, ListFleetDevicesQuery,
|
FleetDevice, FleetDeviceAgent, FleetDeviceEndpoint, FleetWakeRoute, ListFleetDevicesQuery,
|
||||||
};
|
};
|
||||||
@@ -26,6 +31,9 @@ pub(crate) struct AgentRuntimeStatus {
|
|||||||
struct FleetAccumulator {
|
struct FleetAccumulator {
|
||||||
device_key: String,
|
device_key: String,
|
||||||
display_name: Option<String>,
|
display_name: Option<String>,
|
||||||
|
/// Naming-only evidence, used when no name, MAC, or summary IP is available.
|
||||||
|
/// Must not contribute to summary IPs or wake routes.
|
||||||
|
fallback_ip: Option<IpAddr>,
|
||||||
known_device: Option<KnownDeviceSummary>,
|
known_device: Option<KnownDeviceSummary>,
|
||||||
pinned: bool,
|
pinned: bool,
|
||||||
ips: BTreeSet<IpAddr>,
|
ips: BTreeSet<IpAddr>,
|
||||||
@@ -45,6 +53,7 @@ impl Default for FleetAccumulator {
|
|||||||
Self {
|
Self {
|
||||||
device_key: String::new(),
|
device_key: String::new(),
|
||||||
display_name: None,
|
display_name: None,
|
||||||
|
fallback_ip: None,
|
||||||
known_device: None,
|
known_device: None,
|
||||||
pinned: false,
|
pinned: false,
|
||||||
ips: BTreeSet::new(),
|
ips: BTreeSet::new(),
|
||||||
@@ -55,12 +64,14 @@ impl Default for FleetAccumulator {
|
|||||||
endpoints: Vec::new(),
|
endpoints: Vec::new(),
|
||||||
first_seen_unix: None,
|
first_seen_unix: None,
|
||||||
last_seen_unix: None,
|
last_seen_unix: None,
|
||||||
presence: Presence::Offline,
|
// Start at the lowest signal so unknown-only evidence stays unknown.
|
||||||
|
presence: Presence::Unknown,
|
||||||
routes: BTreeMap::new(),
|
routes: BTreeMap::new(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "fleet_group_devices", skip_all, fields(known = known_devices.len(), observed = agent_devices.len()))]
|
||||||
pub(crate) fn build_fleet_devices(
|
pub(crate) fn build_fleet_devices(
|
||||||
known_devices: Vec<KnownDevice>,
|
known_devices: Vec<KnownDevice>,
|
||||||
agent_devices: Vec<AgentDeviceWithChildren>,
|
agent_devices: Vec<AgentDeviceWithChildren>,
|
||||||
@@ -193,8 +204,22 @@ fn add_agent_device_to_entry(
|
|||||||
agent_device: AgentDeviceWithChildren,
|
agent_device: AgentDeviceWithChildren,
|
||||||
context: &FleetBuildContext,
|
context: &FleetBuildContext,
|
||||||
) {
|
) {
|
||||||
let device_offline = agent_device.device.presence() == Presence::Offline;
|
add_address_evidence(entry, &agent_device);
|
||||||
if let Some(summary) = device_known_device(&agent_device, context)
|
add_names_and_identity(entry, &agent_device, context);
|
||||||
|
let status = add_agent_summary(entry, &agent_device, context);
|
||||||
|
if agent_device.endpoints.is_empty() {
|
||||||
|
add_legacy_routes(entry, &agent_device, &status);
|
||||||
|
} else {
|
||||||
|
add_endpoint_routes(entry, &agent_device, &status);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn add_names_and_identity(
|
||||||
|
entry: &mut FleetAccumulator,
|
||||||
|
agent_device: &AgentDeviceWithChildren,
|
||||||
|
context: &FleetBuildContext,
|
||||||
|
) {
|
||||||
|
if let Some(summary) = device_known_device(agent_device, context)
|
||||||
&& entry.known_device.is_none()
|
&& entry.known_device.is_none()
|
||||||
{
|
{
|
||||||
if let Some(ref name) = agent_device.device.display_name {
|
if let Some(ref name) = agent_device.device.display_name {
|
||||||
@@ -203,33 +228,19 @@ fn add_agent_device_to_entry(
|
|||||||
entry.pinned = summary.pinned;
|
entry.pinned = summary.pinned;
|
||||||
entry.known_device = Some(summary);
|
entry.known_device = Some(summary);
|
||||||
}
|
}
|
||||||
if agent_device.endpoints.is_empty() {
|
|
||||||
for mac in &agent_device.macs {
|
|
||||||
entry.macs.insert(*mac);
|
|
||||||
}
|
|
||||||
if !device_offline {
|
|
||||||
for ip in &agent_device.ips {
|
|
||||||
entry.ips.insert(*ip);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
for endpoint in &agent_device.endpoints {
|
|
||||||
if let Some(mac) = endpoint.key.mac {
|
|
||||||
entry.macs.insert(mac);
|
|
||||||
}
|
|
||||||
if endpoint_ip_is_summary_eligible(endpoint.key.source, endpoint.presence)
|
|
||||||
&& let Some(ip) = endpoint.key.ip
|
|
||||||
{
|
|
||||||
entry.ips.insert(ip);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for hostname in &agent_device.hostnames {
|
for hostname in &agent_device.hostnames {
|
||||||
if entry.display_name.is_none() {
|
if entry.display_name.is_none() {
|
||||||
entry.display_name = Some(hostname.clone());
|
entry.display_name = Some(hostname.clone());
|
||||||
}
|
}
|
||||||
entry.hostnames.insert(hostname.clone());
|
entry.hostnames.insert(hostname.clone());
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn add_agent_summary(
|
||||||
|
entry: &mut FleetAccumulator,
|
||||||
|
agent_device: &AgentDeviceWithChildren,
|
||||||
|
context: &FleetBuildContext,
|
||||||
|
) -> AgentRuntimeStatus {
|
||||||
let first_seen = agent_device.device.first_seen();
|
let first_seen = agent_device.device.first_seen();
|
||||||
let last_seen = agent_device.device.last_seen();
|
let last_seen = agent_device.device.last_seen();
|
||||||
entry.first_seen_unix = Some(
|
entry.first_seen_unix = Some(
|
||||||
@@ -244,8 +255,7 @@ fn add_agent_device_to_entry(
|
|||||||
.map(|current| current.max(last_seen))
|
.map(|current| current.max(last_seen))
|
||||||
.unwrap_or(last_seen),
|
.unwrap_or(last_seen),
|
||||||
);
|
);
|
||||||
let device_presence = agent_device.device.presence();
|
entry.presence = std::cmp::max(entry.presence, agent_device.presence());
|
||||||
entry.presence = std::cmp::max(entry.presence, device_presence);
|
|
||||||
|
|
||||||
let agent_id = agent_device.device.agent_id.clone();
|
let agent_id = agent_device.device.agent_id.clone();
|
||||||
let status = context
|
let status = context
|
||||||
@@ -270,105 +280,7 @@ fn add_agent_device_to_entry(
|
|||||||
connected: status.connected,
|
connected: status.connected,
|
||||||
last_seen_unix: last_seen,
|
last_seen_unix: last_seen,
|
||||||
});
|
});
|
||||||
|
status
|
||||||
if agent_device.endpoints.is_empty() {
|
|
||||||
entry.sources.insert("device".to_string());
|
|
||||||
}
|
|
||||||
|
|
||||||
for endpoint in &agent_device.endpoints {
|
|
||||||
let source = endpoint_source_label(endpoint.key.source).to_string();
|
|
||||||
entry.sources.insert(source.clone());
|
|
||||||
let endpoint_last_seen = endpoint.last_seen_unix.unwrap_or(last_seen);
|
|
||||||
let endpoint_first_seen = endpoint.first_seen_unix.or(Some(first_seen));
|
|
||||||
entry.endpoints.push(FleetDeviceEndpoint {
|
|
||||||
agent_id: agent_id.clone(),
|
|
||||||
nickname: status.nickname.clone(),
|
|
||||||
connected: status.connected,
|
|
||||||
source: source.clone(),
|
|
||||||
mac: endpoint.key.mac,
|
|
||||||
ip: endpoint.key.ip,
|
|
||||||
hostname: endpoint.hostname.clone(),
|
|
||||||
interface: endpoint.interface.clone(),
|
|
||||||
presence: endpoint.presence,
|
|
||||||
first_seen_unix: endpoint_first_seen,
|
|
||||||
last_seen_unix: Some(endpoint_last_seen),
|
|
||||||
});
|
|
||||||
|
|
||||||
let rid = route_id(
|
|
||||||
&agent_id,
|
|
||||||
endpoint.key.mac.as_ref(),
|
|
||||||
endpoint.key.ip.as_ref(),
|
|
||||||
);
|
|
||||||
let wakeable = status.connected && endpoint.key.mac.is_some();
|
|
||||||
insert_route(
|
|
||||||
entry,
|
|
||||||
FleetWakeRoute {
|
|
||||||
route_id: rid,
|
|
||||||
agent_id: agent_id.clone(),
|
|
||||||
nickname: status.nickname.clone(),
|
|
||||||
connected: status.connected,
|
|
||||||
mac: endpoint.key.mac,
|
|
||||||
ip: endpoint.key.ip,
|
|
||||||
hostname: endpoint.hostname.clone(),
|
|
||||||
interface: endpoint.interface.clone(),
|
|
||||||
source,
|
|
||||||
presence: endpoint.presence,
|
|
||||||
last_seen_unix: endpoint_last_seen,
|
|
||||||
wakeable,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
if agent_device.endpoints.is_empty() && !agent_device.macs.is_empty() {
|
|
||||||
for mac in &agent_device.macs {
|
|
||||||
let ip_for_mac = agent_device.ips.first().copied();
|
|
||||||
let hostname_for_mac = agent_device.hostnames.first().cloned();
|
|
||||||
let rid = route_id(&agent_id, Some(mac), ip_for_mac.as_ref());
|
|
||||||
let wakeable = status.connected;
|
|
||||||
insert_route(
|
|
||||||
entry,
|
|
||||||
FleetWakeRoute {
|
|
||||||
route_id: rid,
|
|
||||||
agent_id: agent_id.clone(),
|
|
||||||
nickname: status.nickname.clone(),
|
|
||||||
connected: status.connected,
|
|
||||||
mac: Some(*mac),
|
|
||||||
ip: ip_for_mac,
|
|
||||||
hostname: hostname_for_mac,
|
|
||||||
interface: None,
|
|
||||||
source: "device".to_string(),
|
|
||||||
presence: device_presence,
|
|
||||||
last_seen_unix: last_seen,
|
|
||||||
wakeable,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if agent_device.endpoints.is_empty()
|
|
||||||
&& agent_device.macs.is_empty()
|
|
||||||
&& let Some(ip) = agent_device.ips.first()
|
|
||||||
{
|
|
||||||
let hostname = agent_device.hostnames.first().cloned();
|
|
||||||
let rid = route_id(&agent_id, None, Some(ip));
|
|
||||||
insert_route(
|
|
||||||
entry,
|
|
||||||
FleetWakeRoute {
|
|
||||||
route_id: rid,
|
|
||||||
agent_id: agent_id.clone(),
|
|
||||||
nickname: status.nickname.clone(),
|
|
||||||
connected: status.connected,
|
|
||||||
mac: None,
|
|
||||||
ip: Some(*ip),
|
|
||||||
hostname,
|
|
||||||
interface: None,
|
|
||||||
source: "device".to_string(),
|
|
||||||
presence: device_presence,
|
|
||||||
last_seen_unix: last_seen,
|
|
||||||
wakeable: false,
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn add_identifier_to_entry(entry: &mut FleetAccumulator, identifier: &DeviceIdentifier) {
|
fn add_identifier_to_entry(entry: &mut FleetAccumulator, identifier: &DeviceIdentifier) {
|
||||||
@@ -421,37 +333,6 @@ fn device_known_device(
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn endpoint_ip_is_summary_eligible(source: EndpointSource, presence: Presence) -> bool {
|
|
||||||
match source {
|
|
||||||
EndpointSource::DhcpLease => true,
|
|
||||||
EndpointSource::Neighbor => presence != Presence::Offline,
|
|
||||||
EndpointSource::HookNeighbor | EndpointSource::HookDhcp => false,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn insert_route(entry: &mut FleetAccumulator, candidate: FleetWakeRoute) {
|
|
||||||
match entry.routes.entry(candidate.route_id.clone()) {
|
|
||||||
std::collections::btree_map::Entry::Vacant(slot) => {
|
|
||||||
slot.insert(candidate);
|
|
||||||
}
|
|
||||||
std::collections::btree_map::Entry::Occupied(mut slot) => {
|
|
||||||
if compare_route_quality(&candidate, slot.get()).is_gt() {
|
|
||||||
slot.insert(candidate);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn compare_route_quality(a: &FleetWakeRoute, b: &FleetWakeRoute) -> std::cmp::Ordering {
|
|
||||||
a.connected
|
|
||||||
.cmp(&b.connected)
|
|
||||||
.then_with(|| a.wakeable.cmp(&b.wakeable))
|
|
||||||
.then_with(|| a.ip.is_some().cmp(&b.ip.is_some()))
|
|
||||||
.then_with(|| a.presence.cmp(&b.presence))
|
|
||||||
.then_with(|| a.last_seen_unix.cmp(&b.last_seen_unix))
|
|
||||||
.then_with(|| source_quality_rank(&a.source).cmp(&source_quality_rank(&b.source)))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) fn known_device_summary(device: &KnownDevice) -> KnownDeviceSummary {
|
pub(crate) fn known_device_summary(device: &KnownDevice) -> KnownDeviceSummary {
|
||||||
KnownDeviceSummary {
|
KnownDeviceSummary {
|
||||||
device_id: device.device_id.clone(),
|
device_id: device.device_id.clone(),
|
||||||
@@ -462,6 +343,7 @@ pub(crate) fn known_device_summary(device: &KnownDevice) -> KnownDeviceSummary {
|
|||||||
|
|
||||||
impl FleetAccumulator {
|
impl FleetAccumulator {
|
||||||
fn into_response(self) -> FleetDevice {
|
fn into_response(self) -> FleetDevice {
|
||||||
|
let display_name = self.display_label();
|
||||||
let mut route_candidates = self.routes.into_values().collect::<Vec<_>>();
|
let mut route_candidates = self.routes.into_values().collect::<Vec<_>>();
|
||||||
route_candidates
|
route_candidates
|
||||||
.sort_by(|a, b| compare_route_quality(b, a).then_with(|| a.agent_id.cmp(&b.agent_id)));
|
.sort_by(|a, b| compare_route_quality(b, a).then_with(|| a.agent_id.cmp(&b.agent_id)));
|
||||||
@@ -469,13 +351,6 @@ impl FleetAccumulator {
|
|||||||
.iter()
|
.iter()
|
||||||
.find(|route| route.wakeable)
|
.find(|route| route.wakeable)
|
||||||
.cloned();
|
.cloned();
|
||||||
let display_name = self
|
|
||||||
.display_name
|
|
||||||
.or_else(|| self.hostnames.iter().next().cloned())
|
|
||||||
.or_else(|| self.macs.iter().next().map(|mac| mac.to_string()))
|
|
||||||
.or_else(|| self.ips.iter().next().map(|ip| ip.to_string()))
|
|
||||||
.unwrap_or_else(|| "(unknown device)".to_string());
|
|
||||||
|
|
||||||
FleetDevice {
|
FleetDevice {
|
||||||
device_key: self.device_key,
|
device_key: self.device_key,
|
||||||
display_name,
|
display_name,
|
||||||
@@ -496,34 +371,6 @@ impl FleetAccumulator {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn endpoint_source_label(source: EndpointSource) -> &'static str {
|
|
||||||
match source {
|
|
||||||
EndpointSource::Neighbor => "neighbor",
|
|
||||||
EndpointSource::DhcpLease => "dhcp_lease",
|
|
||||||
EndpointSource::HookNeighbor => "hook_neighbor",
|
|
||||||
EndpointSource::HookDhcp => "hook_dhcp",
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn source_quality_rank(source: &str) -> u8 {
|
|
||||||
match source {
|
|
||||||
"neighbor" => 4,
|
|
||||||
"dhcp_lease" => 3,
|
|
||||||
"hook_neighbor" => 2,
|
|
||||||
"hook_dhcp" => 1,
|
|
||||||
_ => 0,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn route_id(agent_id: &str, mac: Option<&MacAddr>, ip: Option<&IpAddr>) -> String {
|
|
||||||
format!(
|
|
||||||
"{}|{}|{}",
|
|
||||||
agent_id,
|
|
||||||
mac.map(|m| m.to_string()).unwrap_or_default(),
|
|
||||||
ip.map(|i| i.to_string()).unwrap_or_default()
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn normalize_filter(value: Option<&str>) -> Option<String> {
|
fn normalize_filter(value: Option<&str>) -> Option<String> {
|
||||||
value
|
value
|
||||||
.map(|value| value.trim().to_ascii_lowercase())
|
.map(|value| value.trim().to_ascii_lowercase())
|
||||||
|
|||||||
@@ -0,0 +1,81 @@
|
|||||||
|
//! Current-address summaries and naming-only evidence have different eligibility.
|
||||||
|
|
||||||
|
use super::FleetAccumulator;
|
||||||
|
use crate::state::AgentDeviceWithChildren;
|
||||||
|
use wakey_core::{EndpointSource, Presence};
|
||||||
|
|
||||||
|
/// Keep naming evidence distinct from addresses eligible for the summary.
|
||||||
|
pub(super) fn add_address_evidence(
|
||||||
|
entry: &mut FleetAccumulator,
|
||||||
|
agent_device: &AgentDeviceWithChildren,
|
||||||
|
) {
|
||||||
|
// Retain address evidence for naming only, without promoting stale IPs into
|
||||||
|
// the active summary or wake-route candidates.
|
||||||
|
let fallback_ip = agent_device
|
||||||
|
.ips
|
||||||
|
.iter()
|
||||||
|
.copied()
|
||||||
|
.chain(
|
||||||
|
agent_device
|
||||||
|
.endpoints
|
||||||
|
.iter()
|
||||||
|
.filter_map(|endpoint| endpoint.key.ip),
|
||||||
|
)
|
||||||
|
.min();
|
||||||
|
entry.fallback_ip = entry.fallback_ip.into_iter().chain(fallback_ip).min();
|
||||||
|
if agent_device.endpoints.is_empty() {
|
||||||
|
for mac in &agent_device.macs {
|
||||||
|
entry.macs.insert(*mac);
|
||||||
|
}
|
||||||
|
if agent_device.presence() != Presence::Offline {
|
||||||
|
for ip in &agent_device.ips {
|
||||||
|
entry.ips.insert(*ip);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
for endpoint in &agent_device.endpoints {
|
||||||
|
if let Some(mac) = endpoint.key.mac {
|
||||||
|
entry.macs.insert(mac);
|
||||||
|
}
|
||||||
|
if endpoint_ip_is_summary_eligible(endpoint.key.source, endpoint.presence)
|
||||||
|
&& let Some(ip) = endpoint.key.ip
|
||||||
|
{
|
||||||
|
entry.ips.insert(ip);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn endpoint_ip_is_summary_eligible(source: EndpointSource, presence: Presence) -> bool {
|
||||||
|
match source {
|
||||||
|
EndpointSource::DhcpLease => true,
|
||||||
|
EndpointSource::Neighbor => presence != Presence::Offline,
|
||||||
|
EndpointSource::HookNeighbor | EndpointSource::HookDhcp => false,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl FleetAccumulator {
|
||||||
|
/// Names and MACs outrank address evidence; presence belongs in the badge.
|
||||||
|
/// Summary IPs outrank naming-only fallback IPs, which may be inactive.
|
||||||
|
pub(super) fn display_label(&self) -> String {
|
||||||
|
self.display_name
|
||||||
|
.clone()
|
||||||
|
.or_else(|| self.hostnames.iter().next().cloned())
|
||||||
|
.or_else(|| {
|
||||||
|
self.macs
|
||||||
|
.iter()
|
||||||
|
.next()
|
||||||
|
.map(|mac| mac.to_string())
|
||||||
|
.or_else(|| {
|
||||||
|
self.ips
|
||||||
|
.iter()
|
||||||
|
.next()
|
||||||
|
.copied()
|
||||||
|
.or(self.fallback_ip)
|
||||||
|
.map(|ip| ip.to_string())
|
||||||
|
})
|
||||||
|
.map(|address| format!("Device: {address}"))
|
||||||
|
})
|
||||||
|
.unwrap_or_else(|| "(unknown device)".to_string())
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,173 @@
|
|||||||
|
//! Build and rank agent-specific wake routes without promoting IPs into summaries.
|
||||||
|
|
||||||
|
use super::{AgentRuntimeStatus, FleetAccumulator, FleetDeviceEndpoint, FleetWakeRoute};
|
||||||
|
use crate::state::AgentDeviceWithChildren;
|
||||||
|
use macaddr::MacAddr;
|
||||||
|
use std::net::IpAddr;
|
||||||
|
use wakey_core::EndpointSource;
|
||||||
|
|
||||||
|
/// Endpoint evidence is preserved even when its IP is excluded from the summary.
|
||||||
|
pub(super) fn add_endpoint_routes(
|
||||||
|
entry: &mut FleetAccumulator,
|
||||||
|
agent_device: &AgentDeviceWithChildren,
|
||||||
|
status: &AgentRuntimeStatus,
|
||||||
|
) {
|
||||||
|
let agent_id = &agent_device.device.agent_id;
|
||||||
|
let first_seen = agent_device.device.first_seen();
|
||||||
|
let last_seen = agent_device.device.last_seen();
|
||||||
|
for endpoint in &agent_device.endpoints {
|
||||||
|
let source = endpoint_source_label(endpoint.key.source).to_string();
|
||||||
|
entry.sources.insert(source.clone());
|
||||||
|
let endpoint_last_seen = endpoint.last_seen_unix.unwrap_or(last_seen);
|
||||||
|
let endpoint_first_seen = endpoint.first_seen_unix.or(Some(first_seen));
|
||||||
|
entry.endpoints.push(FleetDeviceEndpoint {
|
||||||
|
agent_id: agent_id.clone(),
|
||||||
|
nickname: status.nickname.clone(),
|
||||||
|
connected: status.connected,
|
||||||
|
source: source.clone(),
|
||||||
|
mac: endpoint.key.mac,
|
||||||
|
ip: endpoint.key.ip,
|
||||||
|
hostname: endpoint.hostname.clone(),
|
||||||
|
interface: endpoint.interface.clone(),
|
||||||
|
presence: endpoint.presence,
|
||||||
|
first_seen_unix: endpoint_first_seen,
|
||||||
|
last_seen_unix: Some(endpoint_last_seen),
|
||||||
|
});
|
||||||
|
|
||||||
|
let rid = route_id(
|
||||||
|
agent_id,
|
||||||
|
endpoint.key.mac.as_ref(),
|
||||||
|
endpoint.key.ip.as_ref(),
|
||||||
|
);
|
||||||
|
let wakeable = status.connected && endpoint.key.mac.is_some();
|
||||||
|
insert_route(
|
||||||
|
entry,
|
||||||
|
FleetWakeRoute {
|
||||||
|
route_id: rid,
|
||||||
|
agent_id: agent_id.clone(),
|
||||||
|
nickname: status.nickname.clone(),
|
||||||
|
connected: status.connected,
|
||||||
|
mac: endpoint.key.mac,
|
||||||
|
ip: endpoint.key.ip,
|
||||||
|
hostname: endpoint.hostname.clone(),
|
||||||
|
interface: endpoint.interface.clone(),
|
||||||
|
source,
|
||||||
|
presence: endpoint.presence,
|
||||||
|
last_seen_unix: endpoint_last_seen,
|
||||||
|
wakeable,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Compatibility for stored rows without authoritative endpoint evidence.
|
||||||
|
pub(super) fn add_legacy_routes(
|
||||||
|
entry: &mut FleetAccumulator,
|
||||||
|
agent_device: &AgentDeviceWithChildren,
|
||||||
|
status: &AgentRuntimeStatus,
|
||||||
|
) {
|
||||||
|
entry.sources.insert("device".to_string());
|
||||||
|
let agent_id = &agent_device.device.agent_id;
|
||||||
|
let device_presence = agent_device.presence();
|
||||||
|
let last_seen = agent_device.device.last_seen();
|
||||||
|
if !agent_device.macs.is_empty() {
|
||||||
|
for mac in &agent_device.macs {
|
||||||
|
let ip_for_mac = agent_device.ips.first().copied();
|
||||||
|
let hostname_for_mac = agent_device.hostnames.first().cloned();
|
||||||
|
let rid = route_id(agent_id, Some(mac), ip_for_mac.as_ref());
|
||||||
|
let wakeable = status.connected;
|
||||||
|
insert_route(
|
||||||
|
entry,
|
||||||
|
FleetWakeRoute {
|
||||||
|
route_id: rid,
|
||||||
|
agent_id: agent_id.clone(),
|
||||||
|
nickname: status.nickname.clone(),
|
||||||
|
connected: status.connected,
|
||||||
|
mac: Some(*mac),
|
||||||
|
ip: ip_for_mac,
|
||||||
|
hostname: hostname_for_mac,
|
||||||
|
interface: None,
|
||||||
|
source: "device".to_string(),
|
||||||
|
presence: device_presence,
|
||||||
|
last_seen_unix: last_seen,
|
||||||
|
wakeable,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if agent_device.macs.is_empty()
|
||||||
|
&& let Some(ip) = agent_device.ips.first()
|
||||||
|
{
|
||||||
|
let hostname = agent_device.hostnames.first().cloned();
|
||||||
|
let rid = route_id(agent_id, None, Some(ip));
|
||||||
|
insert_route(
|
||||||
|
entry,
|
||||||
|
FleetWakeRoute {
|
||||||
|
route_id: rid,
|
||||||
|
agent_id: agent_id.clone(),
|
||||||
|
nickname: status.nickname.clone(),
|
||||||
|
connected: status.connected,
|
||||||
|
mac: None,
|
||||||
|
ip: Some(*ip),
|
||||||
|
hostname,
|
||||||
|
interface: None,
|
||||||
|
source: "device".to_string(),
|
||||||
|
presence: device_presence,
|
||||||
|
last_seen_unix: last_seen,
|
||||||
|
wakeable: false,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn insert_route(entry: &mut FleetAccumulator, candidate: FleetWakeRoute) {
|
||||||
|
match entry.routes.entry(candidate.route_id.clone()) {
|
||||||
|
std::collections::btree_map::Entry::Vacant(slot) => {
|
||||||
|
slot.insert(candidate);
|
||||||
|
}
|
||||||
|
std::collections::btree_map::Entry::Occupied(mut slot) => {
|
||||||
|
if compare_route_quality(&candidate, slot.get()).is_gt() {
|
||||||
|
slot.insert(candidate);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) fn compare_route_quality(a: &FleetWakeRoute, b: &FleetWakeRoute) -> std::cmp::Ordering {
|
||||||
|
a.connected
|
||||||
|
.cmp(&b.connected)
|
||||||
|
.then_with(|| a.wakeable.cmp(&b.wakeable))
|
||||||
|
.then_with(|| a.ip.is_some().cmp(&b.ip.is_some()))
|
||||||
|
.then_with(|| a.presence.cmp(&b.presence))
|
||||||
|
.then_with(|| a.last_seen_unix.cmp(&b.last_seen_unix))
|
||||||
|
.then_with(|| source_quality_rank(&a.source).cmp(&source_quality_rank(&b.source)))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn endpoint_source_label(source: EndpointSource) -> &'static str {
|
||||||
|
match source {
|
||||||
|
EndpointSource::Neighbor => "neighbor",
|
||||||
|
EndpointSource::DhcpLease => "dhcp_lease",
|
||||||
|
EndpointSource::HookNeighbor => "hook_neighbor",
|
||||||
|
EndpointSource::HookDhcp => "hook_dhcp",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn source_quality_rank(source: &str) -> u8 {
|
||||||
|
match source {
|
||||||
|
"neighbor" => 4,
|
||||||
|
"dhcp_lease" => 3,
|
||||||
|
"hook_neighbor" => 2,
|
||||||
|
"hook_dhcp" => 1,
|
||||||
|
_ => 0,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn route_id(agent_id: &str, mac: Option<&MacAddr>, ip: Option<&IpAddr>) -> String {
|
||||||
|
format!(
|
||||||
|
"{}|{}|{}",
|
||||||
|
agent_id,
|
||||||
|
mac.map(|m| m.to_string()).unwrap_or_default(),
|
||||||
|
ip.map(|i| i.to_string()).unwrap_or_default()
|
||||||
|
)
|
||||||
|
}
|
||||||
@@ -1,18 +1,20 @@
|
|||||||
use std::collections::{BTreeSet, HashMap};
|
use std::collections::{BTreeSet, HashMap};
|
||||||
|
|
||||||
use axum::Json;
|
use axum::Json;
|
||||||
use axum::extract::{Query, State};
|
use axum::extract::{Path, Query, State};
|
||||||
use axum::http::StatusCode;
|
use axum::http::StatusCode;
|
||||||
use axum::response::IntoResponse;
|
use axum::response::IntoResponse;
|
||||||
use tracing::warn;
|
use tracing::warn;
|
||||||
use wakey_agent::protocol::{AgentCommand, InventoryRequest, WakeRequest};
|
|
||||||
|
|
||||||
use crate::api::ApiError;
|
use crate::api::ApiError;
|
||||||
use crate::api::commands::relay_agent_command;
|
|
||||||
use crate::runtime::AppState;
|
use crate::runtime::AppState;
|
||||||
|
|
||||||
mod build;
|
mod build;
|
||||||
|
mod refresh;
|
||||||
mod types;
|
mod types;
|
||||||
|
mod wake;
|
||||||
|
pub use refresh::refresh_fleet_devices;
|
||||||
|
pub use wake::wake_fleet_device;
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests;
|
mod tests;
|
||||||
@@ -21,11 +23,75 @@ use build::{
|
|||||||
AgentRuntimeStatus, FleetBuildContext, build_fleet_devices, filter_fleet_devices,
|
AgentRuntimeStatus, FleetBuildContext, build_fleet_devices, filter_fleet_devices,
|
||||||
known_device_summary,
|
known_device_summary,
|
||||||
};
|
};
|
||||||
use types::{
|
use types::{FleetDevice, ListFleetDevicesQuery};
|
||||||
FleetDevice, ListFleetDevicesQuery, RefreshFleetAgentResult, RefreshFleetDevicesRequest,
|
|
||||||
RefreshFleetDevicesResponse, WakeFleetDeviceRequest, WakeFleetDeviceResponse,
|
|
||||||
};
|
|
||||||
|
|
||||||
|
pub(crate) type FleetCache =
|
||||||
|
crate::runtime::view_cache::ViewCache<(u64, BTreeSet<String>), Vec<FleetDevice>>;
|
||||||
|
|
||||||
|
async fn fleet_view(state: &AppState) -> anyhow::Result<std::sync::Arc<Vec<FleetDevice>>> {
|
||||||
|
state
|
||||||
|
.fleet
|
||||||
|
.get_or_try_init(
|
||||||
|
|| async {
|
||||||
|
let connected = state
|
||||||
|
.sessions
|
||||||
|
.read()
|
||||||
|
.await
|
||||||
|
.keys()
|
||||||
|
.cloned()
|
||||||
|
.collect::<BTreeSet<_>>();
|
||||||
|
(state.store.fleet_revision(), connected)
|
||||||
|
},
|
||||||
|
|| build_fleet_view(state),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Count stored rows separately from grouped identities. Reject mixed revisions;
|
||||||
|
/// filtered UI results must never become the source of current-state metrics.
|
||||||
|
pub(crate) async fn sample_fleet_counts(
|
||||||
|
state: &AppState,
|
||||||
|
) -> anyhow::Result<wakey_core::telemetry::FleetCounts> {
|
||||||
|
for _ in 0..3 {
|
||||||
|
let revision = state.store.fleet_revision();
|
||||||
|
let mut counts = wakey_core::telemetry::FleetCounts::default();
|
||||||
|
for (agent, _) in state.store.list_agents_with_nicknames().await? {
|
||||||
|
counts.agents.entry(agent).or_default();
|
||||||
|
}
|
||||||
|
for row in state.store.list_agent_device_rows().await? {
|
||||||
|
let agent = counts.agents.entry(row.device.agent_id).or_default();
|
||||||
|
agent.devices += 1;
|
||||||
|
agent.endpoints += row.endpoints.len() as u64;
|
||||||
|
}
|
||||||
|
let fleet = fleet_view(state).await?;
|
||||||
|
counts.saved = fleet
|
||||||
|
.iter()
|
||||||
|
.filter(|device| device.known_device.is_some())
|
||||||
|
.count() as u64;
|
||||||
|
counts.unsaved = fleet.len() as u64 - counts.saved;
|
||||||
|
if state.store.fleet_revision() == revision {
|
||||||
|
return Ok(counts);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(crate::runtime::view_cache::RebuildLimit.into())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn fleet_error(err: anyhow::Error, fallback_code: &str) -> ApiError {
|
||||||
|
if err.is::<crate::runtime::view_cache::RebuildLimit>() {
|
||||||
|
return ApiError::new(
|
||||||
|
StatusCode::SERVICE_UNAVAILABLE,
|
||||||
|
"fleet_busy",
|
||||||
|
err.to_string(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
ApiError::new(
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR,
|
||||||
|
fallback_code,
|
||||||
|
err.to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "fleet_list", skip_all)]
|
||||||
pub async fn list_fleet_devices(
|
pub async fn list_fleet_devices(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Query(query): Query<ListFleetDevicesQuery>,
|
Query(query): Query<ListFleetDevicesQuery>,
|
||||||
@@ -34,142 +100,25 @@ pub async fn list_fleet_devices(
|
|||||||
Ok(devices) => Ok((StatusCode::OK, Json(devices))),
|
Ok(devices) => Ok((StatusCode::OK, Json(devices))),
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
warn!(error = %err, "failed to list fleet devices");
|
warn!(error = %err, "failed to list fleet devices");
|
||||||
Err(ApiError::new(
|
Err(fleet_error(err, "list_fleet_devices_failed"))
|
||||||
StatusCode::INTERNAL_SERVER_ERROR,
|
|
||||||
"list_fleet_devices_failed",
|
|
||||||
err.to_string(),
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn refresh_fleet_devices(
|
/// Details are an exact identity lookup, not a filtered/paginated list lookup.
|
||||||
|
/// In particular, remembering an unsaved row must remain readable under its new key.
|
||||||
|
#[tracing::instrument(name = "fleet_get", skip_all)]
|
||||||
|
pub async fn get_fleet_device(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Json(req): Json<RefreshFleetDevicesRequest>,
|
Path(device_key): Path<String>,
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
let mut agent_ids = if req.agent_ids.is_empty() {
|
let devices = fleet_view(&state)
|
||||||
let sessions = state.sessions.read().await;
|
|
||||||
sessions.keys().cloned().collect::<Vec<_>>()
|
|
||||||
} else {
|
|
||||||
req.agent_ids.clone()
|
|
||||||
};
|
|
||||||
agent_ids.sort();
|
|
||||||
agent_ids.dedup();
|
|
||||||
|
|
||||||
let mut results = Vec::with_capacity(agent_ids.len());
|
|
||||||
let mut total_accepted = 0usize;
|
|
||||||
for agent_id in agent_ids {
|
|
||||||
let response = relay_agent_command(
|
|
||||||
&state,
|
|
||||||
&agent_id,
|
|
||||||
AgentCommand::Inventory(InventoryRequest {
|
|
||||||
query: None,
|
|
||||||
name: None,
|
|
||||||
ips: Vec::new(),
|
|
||||||
devs: Vec::new(),
|
|
||||||
nuds: Vec::new(),
|
|
||||||
macs: Vec::new(),
|
|
||||||
}),
|
|
||||||
req.timeout_ms,
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
match response {
|
|
||||||
Ok(response) if response.status == "ok" => {
|
|
||||||
let Some(result) = response.result else {
|
|
||||||
results.push(RefreshFleetAgentResult {
|
|
||||||
agent_id,
|
|
||||||
status: "error".into(),
|
|
||||||
accepted: 0,
|
|
||||||
error: Some("inventory command returned no result".into()),
|
|
||||||
});
|
|
||||||
continue;
|
|
||||||
};
|
|
||||||
match serde_json::from_value::<wakey_core::DeviceInventory>(result)
|
|
||||||
.map(|i| i.devices)
|
|
||||||
{
|
|
||||||
Ok(devices) => {
|
|
||||||
match state
|
|
||||||
.store
|
|
||||||
.replace_agent_device_snapshot(&agent_id, &devices)
|
|
||||||
.await
|
.await
|
||||||
{
|
.map_err(|err| fleet_error(err, "get_fleet_device_failed"))?;
|
||||||
Ok(accepted) => {
|
|
||||||
total_accepted = total_accepted.saturating_add(accepted);
|
|
||||||
results.push(RefreshFleetAgentResult {
|
|
||||||
agent_id,
|
|
||||||
status: "ok".into(),
|
|
||||||
accepted,
|
|
||||||
error: None,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
Err(err) => results.push(RefreshFleetAgentResult {
|
|
||||||
agent_id,
|
|
||||||
status: "error".into(),
|
|
||||||
accepted: 0,
|
|
||||||
error: Some(err.to_string()),
|
|
||||||
}),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Err(err) => results.push(RefreshFleetAgentResult {
|
|
||||||
agent_id,
|
|
||||||
status: "error".into(),
|
|
||||||
accepted: 0,
|
|
||||||
error: Some(err.to_string()),
|
|
||||||
}),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Ok(response) => results.push(RefreshFleetAgentResult {
|
|
||||||
agent_id,
|
|
||||||
status: "error".into(),
|
|
||||||
accepted: 0,
|
|
||||||
error: response
|
|
||||||
.error
|
|
||||||
.map(|error| error.message)
|
|
||||||
.or_else(|| Some("inventory command failed".into())),
|
|
||||||
}),
|
|
||||||
Err(err) => results.push(RefreshFleetAgentResult {
|
|
||||||
agent_id,
|
|
||||||
status: "error".into(),
|
|
||||||
accepted: 0,
|
|
||||||
error: Some(format!("{}: {}", err.status, err.message)),
|
|
||||||
}),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok((
|
|
||||||
StatusCode::OK,
|
|
||||||
Json(RefreshFleetDevicesResponse {
|
|
||||||
total_accepted,
|
|
||||||
agents: results,
|
|
||||||
}),
|
|
||||||
))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn wake_fleet_device(
|
|
||||||
State(state): State<AppState>,
|
|
||||||
Json(req): Json<WakeFleetDeviceRequest>,
|
|
||||||
) -> Result<impl IntoResponse, ApiError> {
|
|
||||||
let query = ListFleetDevicesQuery {
|
|
||||||
query: None,
|
|
||||||
presence: None,
|
|
||||||
known: None,
|
|
||||||
agent_id: None,
|
|
||||||
visibility: Some("all".into()),
|
|
||||||
limit: Some(1000),
|
|
||||||
};
|
|
||||||
let devices = load_fleet_devices(&state, &query).await.map_err(|err| {
|
|
||||||
warn!(error = %err, "failed loading fleet devices for wake");
|
|
||||||
ApiError::new(
|
|
||||||
StatusCode::INTERNAL_SERVER_ERROR,
|
|
||||||
"load_fleet_devices_failed",
|
|
||||||
err.to_string(),
|
|
||||||
)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
let device = devices
|
let device = devices
|
||||||
.into_iter()
|
.iter()
|
||||||
.find(|device| device.device_key == req.device_key)
|
.find(|device| device.device_key == device_key)
|
||||||
|
.cloned()
|
||||||
.ok_or_else(|| {
|
.ok_or_else(|| {
|
||||||
ApiError::new(
|
ApiError::new(
|
||||||
StatusCode::NOT_FOUND,
|
StatusCode::NOT_FOUND,
|
||||||
@@ -177,73 +126,36 @@ pub async fn wake_fleet_device(
|
|||||||
"fleet device not found",
|
"fleet device not found",
|
||||||
)
|
)
|
||||||
})?;
|
})?;
|
||||||
|
Ok(Json(device))
|
||||||
let route = match req.route_id.as_deref() {
|
|
||||||
Some(route_id) => device
|
|
||||||
.route_candidates
|
|
||||||
.into_iter()
|
|
||||||
.find(|route| route.route_id == route_id),
|
|
||||||
None => device.recommended_route,
|
|
||||||
}
|
|
||||||
.ok_or_else(|| {
|
|
||||||
ApiError::new(
|
|
||||||
StatusCode::BAD_REQUEST,
|
|
||||||
"wake_route_unavailable",
|
|
||||||
"no wakeable connected MAC-backed route is available",
|
|
||||||
)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
let Some(mac) = route.mac else {
|
|
||||||
return Err(ApiError::new(
|
|
||||||
StatusCode::BAD_REQUEST,
|
|
||||||
"wake_route_unavailable",
|
|
||||||
"selected route does not include a MAC address",
|
|
||||||
));
|
|
||||||
};
|
|
||||||
if !route.connected {
|
|
||||||
return Err(ApiError::new(
|
|
||||||
StatusCode::BAD_REQUEST,
|
|
||||||
"wake_route_unavailable",
|
|
||||||
"selected route agent is not connected",
|
|
||||||
));
|
|
||||||
}
|
|
||||||
if !route.wakeable {
|
|
||||||
return Err(ApiError::new(
|
|
||||||
StatusCode::BAD_REQUEST,
|
|
||||||
"wake_route_unavailable",
|
|
||||||
"selected route is not wakeable",
|
|
||||||
));
|
|
||||||
}
|
|
||||||
|
|
||||||
let command = relay_agent_command(
|
|
||||||
&state,
|
|
||||||
&route.agent_id,
|
|
||||||
AgentCommand::Wake(WakeRequest {
|
|
||||||
query: None,
|
|
||||||
mac: Some(mac),
|
|
||||||
ip: route.ip,
|
|
||||||
}),
|
|
||||||
req.timeout_ms,
|
|
||||||
)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok((
|
|
||||||
StatusCode::OK,
|
|
||||||
Json(WakeFleetDeviceResponse { route, command }),
|
|
||||||
))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "fleet_load", skip_all, fields(grouped = tracing::field::Empty, filtered = tracing::field::Empty, returned = tracing::field::Empty))]
|
||||||
async fn load_fleet_devices(
|
async fn load_fleet_devices(
|
||||||
state: &AppState,
|
state: &AppState,
|
||||||
query: &ListFleetDevicesQuery,
|
query: &ListFleetDevicesQuery,
|
||||||
) -> anyhow::Result<Vec<FleetDevice>> {
|
) -> anyhow::Result<Vec<FleetDevice>> {
|
||||||
|
let cached = fleet_view(state).await?;
|
||||||
|
// Cache the complete view. Query-specific visibility, search and limits
|
||||||
|
// must never affect what another client (or a wake request) can retrieve.
|
||||||
|
let mut devices = cached.as_ref().clone();
|
||||||
|
tracing::Span::current().record("grouped", devices.len());
|
||||||
|
filter_fleet_devices(&mut devices, query);
|
||||||
|
tracing::Span::current().record("filtered", devices.len());
|
||||||
|
let limit = query.limit.unwrap_or(500).clamp(1, 1000);
|
||||||
|
devices.truncate(limit);
|
||||||
|
tracing::Span::current().record("returned", devices.len());
|
||||||
|
Ok(devices)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "fleet_rebuild", skip_all)]
|
||||||
|
async fn build_fleet_view(state: &AppState) -> anyhow::Result<Vec<FleetDevice>> {
|
||||||
let known_devices = state.store.list_known_devices().await?;
|
let known_devices = state.store.list_known_devices().await?;
|
||||||
let agent_devices = state.store.list_agent_device_rows().await?;
|
let agent_devices = state.store.list_agent_device_rows().await?;
|
||||||
let connected = {
|
let connected = {
|
||||||
let sessions = state.sessions.read().await;
|
let sessions = state.sessions.read().await;
|
||||||
sessions.keys().cloned().collect::<BTreeSet<_>>()
|
sessions.keys().cloned().collect::<BTreeSet<_>>()
|
||||||
};
|
};
|
||||||
let agents = state.store.list_agents_with_nicknames().await;
|
let agents = state.store.list_agents_with_nicknames().await?;
|
||||||
let agent_status = agents
|
let agent_status = agents
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.map(|(agent_id, nickname)| {
|
.map(|(agent_id, nickname)| {
|
||||||
@@ -270,10 +182,5 @@ async fn load_fleet_devices(
|
|||||||
agent_status,
|
agent_status,
|
||||||
identifier_map,
|
identifier_map,
|
||||||
};
|
};
|
||||||
let mut devices = build_fleet_devices(known_devices, agent_devices, &context);
|
Ok(build_fleet_devices(known_devices, agent_devices, &context))
|
||||||
filter_fleet_devices(&mut devices, query);
|
|
||||||
let limit = query.limit.unwrap_or(500).clamp(1, 1000);
|
|
||||||
// this is going to be 10 million times harder if i decide to do this lazily
|
|
||||||
devices.truncate(limit);
|
|
||||||
Ok(devices)
|
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,432 @@
|
|||||||
|
//! Refresh agents independently: one failed relay or snapshot does not cancel others.
|
||||||
|
|
||||||
|
use axum::{Json, extract::State, http::StatusCode, response::IntoResponse};
|
||||||
|
use futures_util::StreamExt;
|
||||||
|
use tracing::{info, warn};
|
||||||
|
use wakey_agent::protocol::{AgentCommand, InventoryRequest};
|
||||||
|
|
||||||
|
use super::types::{
|
||||||
|
RefreshFleetAgentResult, RefreshFleetDevicesRequest, RefreshFleetDevicesResponse,
|
||||||
|
};
|
||||||
|
use crate::api::{
|
||||||
|
ApiError,
|
||||||
|
commands::{RelayCommandResponse, relay_agent_command},
|
||||||
|
};
|
||||||
|
use crate::runtime::AppState;
|
||||||
|
use crate::state::Store;
|
||||||
|
|
||||||
|
const REFRESH_CONCURRENCY: usize = 8;
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "fleet_refresh", skip_all, fields(agents = tracing::field::Empty, succeeded = tracing::field::Empty, failed = tracing::field::Empty, accepted = tracing::field::Empty))]
|
||||||
|
pub async fn refresh_fleet_devices(
|
||||||
|
State(state): State<AppState>,
|
||||||
|
Json(req): Json<RefreshFleetDevicesRequest>,
|
||||||
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
|
let agent_ids = select_agents(&state, req.agent_ids).await;
|
||||||
|
tracing::Span::current().record("agents", agent_ids.len());
|
||||||
|
let results = futures_util::stream::iter(
|
||||||
|
agent_ids
|
||||||
|
.into_iter()
|
||||||
|
.map(|agent_id| refresh_agent(&state, agent_id, req.timeout_ms)),
|
||||||
|
)
|
||||||
|
.buffer_unordered(REFRESH_CONCURRENCY)
|
||||||
|
.collect::<Vec<_>>()
|
||||||
|
.await;
|
||||||
|
let response = summarize(results);
|
||||||
|
let succeeded = response
|
||||||
|
.agents
|
||||||
|
.iter()
|
||||||
|
.filter(|result| result.status == "ok")
|
||||||
|
.count();
|
||||||
|
let span = tracing::Span::current();
|
||||||
|
span.record("succeeded", succeeded);
|
||||||
|
span.record("failed", response.agents.len() - succeeded);
|
||||||
|
span.record("accepted", response.total_accepted);
|
||||||
|
info!("fleet refresh completed");
|
||||||
|
Ok((StatusCode::OK, Json(response)))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Default to a point-in-time connected set. Explicit IDs remain eligible so
|
||||||
|
/// the relay can report disconnected agents as per-agent errors, not omit them.
|
||||||
|
async fn select_agents(state: &AppState, requested: Vec<String>) -> Vec<String> {
|
||||||
|
let ids = if requested.is_empty() {
|
||||||
|
state.sessions.read().await.keys().cloned().collect()
|
||||||
|
} else {
|
||||||
|
requested
|
||||||
|
};
|
||||||
|
normalize_refresh_agent_ids(ids)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(super) fn normalize_refresh_agent_ids(mut agent_ids: Vec<String>) -> Vec<String> {
|
||||||
|
agent_ids.sort();
|
||||||
|
agent_ids.dedup();
|
||||||
|
agent_ids
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "fleet_refresh_agent", skip_all, fields(agent_id = %agent_id, outcome = tracing::field::Empty, accepted = tracing::field::Empty))]
|
||||||
|
async fn refresh_agent(
|
||||||
|
state: &AppState,
|
||||||
|
agent_id: String,
|
||||||
|
timeout_ms: Option<u64>,
|
||||||
|
) -> RefreshFleetAgentResult {
|
||||||
|
let response = relay_agent_command(
|
||||||
|
state,
|
||||||
|
&agent_id,
|
||||||
|
AgentCommand::Inventory(InventoryRequest {
|
||||||
|
query: None,
|
||||||
|
name: None,
|
||||||
|
ips: Vec::new(),
|
||||||
|
devs: Vec::new(),
|
||||||
|
nuds: Vec::new(),
|
||||||
|
macs: Vec::new(),
|
||||||
|
}),
|
||||||
|
timeout_ms,
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
let result = store_inventory(&state.store, &agent_id, response).await;
|
||||||
|
tracing::Span::current().record("outcome", result.status.as_str());
|
||||||
|
tracing::Span::current().record("accepted", result.accepted);
|
||||||
|
if result.status != "ok" {
|
||||||
|
// The response retains diagnostic detail; do not export payloads.
|
||||||
|
warn!("fleet agent refresh failed");
|
||||||
|
}
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Validate the relay result before replacing anything. An empty *valid*
|
||||||
|
/// inventory is still a complete snapshot and intentionally clears old rows.
|
||||||
|
fn decode_inventory(
|
||||||
|
response: Result<RelayCommandResponse, ApiError>,
|
||||||
|
) -> Result<wakey_core::DeviceInventory, String> {
|
||||||
|
let response = response.map_err(|err| format!("{}: {}", err.status, err.message))?;
|
||||||
|
if response.status != "ok" {
|
||||||
|
return Err(response
|
||||||
|
.error
|
||||||
|
.map(|error| error.message)
|
||||||
|
.unwrap_or_else(|| "inventory command failed".into()));
|
||||||
|
}
|
||||||
|
let result = response
|
||||||
|
.result
|
||||||
|
.ok_or_else(|| "inventory command returned no result".to_string())?;
|
||||||
|
serde_json::from_value(result).map_err(|err| err.to_string())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn store_inventory(
|
||||||
|
store: &Store,
|
||||||
|
agent_id: &str,
|
||||||
|
response: Result<RelayCommandResponse, ApiError>,
|
||||||
|
) -> RefreshFleetAgentResult {
|
||||||
|
let accepted = async {
|
||||||
|
let inventory = decode_inventory(response)?;
|
||||||
|
store
|
||||||
|
.replace_agent_device_snapshot(agent_id, &inventory.devices)
|
||||||
|
.await
|
||||||
|
.map_err(|err| err.to_string())
|
||||||
|
}
|
||||||
|
.await;
|
||||||
|
match accepted {
|
||||||
|
Ok(accepted) => RefreshFleetAgentResult {
|
||||||
|
agent_id: agent_id.into(),
|
||||||
|
status: "ok".into(),
|
||||||
|
accepted,
|
||||||
|
error: None,
|
||||||
|
},
|
||||||
|
Err(error) => RefreshFleetAgentResult {
|
||||||
|
agent_id: agent_id.into(),
|
||||||
|
status: "error".into(),
|
||||||
|
accepted: 0,
|
||||||
|
error: Some(error),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Concurrent completion order is deliberately not part of the response contract.
|
||||||
|
fn summarize(mut agents: Vec<RefreshFleetAgentResult>) -> RefreshFleetDevicesResponse {
|
||||||
|
agents.sort_by(|a, b| a.agent_id.cmp(&b.agent_id));
|
||||||
|
RefreshFleetDevicesResponse {
|
||||||
|
total_accepted: agents.iter().map(|result| result.accepted).sum(),
|
||||||
|
agents,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use serde_json::json;
|
||||||
|
|
||||||
|
fn response(result: Option<serde_json::Value>) -> RelayCommandResponse {
|
||||||
|
RelayCommandResponse {
|
||||||
|
request_id: "refresh-test".into(),
|
||||||
|
status: "ok".into(),
|
||||||
|
result,
|
||||||
|
error: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn refresh_updates_cached_fleet_despite_a_disconnected_agent() {
|
||||||
|
use crate::runtime::{AgentReply, AgentSession, SessionEvent};
|
||||||
|
use std::{collections::HashMap, sync::Arc, time::Duration};
|
||||||
|
use tokio::sync::{Mutex, RwLock, mpsc};
|
||||||
|
use wakey_agent::protocol::{CommandResult, ServerMessage};
|
||||||
|
|
||||||
|
let dir =
|
||||||
|
std::env::temp_dir().join(format!("wakey-refresh-handler-{}", uuid::Uuid::new_v4()));
|
||||||
|
let state = AppState {
|
||||||
|
store: Arc::new(
|
||||||
|
Store::load_or_init(&dir.join("state.sqlite3"), vec![], Duration::from_secs(60))
|
||||||
|
.await
|
||||||
|
.unwrap(),
|
||||||
|
),
|
||||||
|
sessions: Arc::new(RwLock::new(HashMap::new())),
|
||||||
|
pending: Arc::new(Mutex::new(HashMap::new())),
|
||||||
|
public_url: "http://localhost".into(),
|
||||||
|
command_timeout: Duration::from_secs(30),
|
||||||
|
enroll_token_ttl: Duration::from_secs(60),
|
||||||
|
terminals: crate::runtime::terminals::TerminalRegistry::new(),
|
||||||
|
alerts: crate::api::AlertsHub::new(),
|
||||||
|
fleet: Arc::new(super::super::FleetCache::default()),
|
||||||
|
};
|
||||||
|
let token = state
|
||||||
|
.store
|
||||||
|
.issue_enroll_token(Duration::from_secs(60))
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let agent_id = state
|
||||||
|
.store
|
||||||
|
.enroll(&token.enroll_token)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.agent_id;
|
||||||
|
let device = |ip: &str| {
|
||||||
|
wakey_core::Device::from_parts(
|
||||||
|
vec![wakey_core::NeighborEntry {
|
||||||
|
ip: ip.parse().unwrap(),
|
||||||
|
mac: Some("aa:bb:cc:dd:ee:01".parse().unwrap()),
|
||||||
|
dev: Some("br-lan".into()),
|
||||||
|
state: wakey_core::NeighborState::Reachable,
|
||||||
|
}],
|
||||||
|
vec![],
|
||||||
|
)
|
||||||
|
};
|
||||||
|
state
|
||||||
|
.store
|
||||||
|
.replace_agent_device_snapshot(&agent_id, &[device("192.168.1.10")])
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let (tx, mut rx) = mpsc::unbounded_channel();
|
||||||
|
state.sessions.write().await.insert(
|
||||||
|
agent_id.clone(),
|
||||||
|
AgentSession {
|
||||||
|
connection_id: "test".into(),
|
||||||
|
tx,
|
||||||
|
capabilities: vec![],
|
||||||
|
capability_options: Default::default(),
|
||||||
|
},
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
select_agents(&state, vec![]).await.as_slice(),
|
||||||
|
std::slice::from_ref(&agent_id)
|
||||||
|
);
|
||||||
|
|
||||||
|
// Prime the real cache with old data before refresh touches the store.
|
||||||
|
let query = super::super::types::ListFleetDevicesQuery {
|
||||||
|
visibility: Some("all".into()),
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
let before = super::super::load_fleet_devices(&state, &query)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(before.len(), 1);
|
||||||
|
assert_eq!(
|
||||||
|
before[0].ips,
|
||||||
|
["192.168.1.10".parse::<std::net::IpAddr>().unwrap()]
|
||||||
|
);
|
||||||
|
|
||||||
|
let refresh = refresh_fleet_devices(
|
||||||
|
State(state.clone()),
|
||||||
|
Json(RefreshFleetDevicesRequest {
|
||||||
|
agent_ids: vec!["disconnected".into(), agent_id.clone(), agent_id.clone()],
|
||||||
|
timeout_ms: None,
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
// Answer through the production relay's pending-request channel. No
|
||||||
|
// sleep or polling: receiving the command proves its waiter exists.
|
||||||
|
let reply = async {
|
||||||
|
let SessionEvent::Message(ServerMessage::Command {
|
||||||
|
request_id,
|
||||||
|
command,
|
||||||
|
..
|
||||||
|
}) = rx.recv().await.unwrap()
|
||||||
|
else {
|
||||||
|
panic!("expected inventory command");
|
||||||
|
};
|
||||||
|
let AgentCommand::Inventory(request) = command else {
|
||||||
|
panic!("expected inventory command");
|
||||||
|
};
|
||||||
|
assert!(request.query.is_none() && request.name.is_none());
|
||||||
|
assert!(
|
||||||
|
request.ips.is_empty()
|
||||||
|
&& request.macs.is_empty()
|
||||||
|
&& request.devs.is_empty()
|
||||||
|
&& request.nuds.is_empty()
|
||||||
|
);
|
||||||
|
let result =
|
||||||
|
serde_json::to_value(CommandResult::Inventory(wakey_core::DeviceInventory {
|
||||||
|
devices: vec![device("192.168.1.20")],
|
||||||
|
}))
|
||||||
|
.unwrap();
|
||||||
|
state
|
||||||
|
.pending
|
||||||
|
.lock()
|
||||||
|
.await
|
||||||
|
.remove(request_id.as_str())
|
||||||
|
.unwrap()
|
||||||
|
.send(AgentReply::Result(result))
|
||||||
|
.unwrap_or_else(|_| panic!("relay waiter dropped"));
|
||||||
|
};
|
||||||
|
let (response, ()) = tokio::join!(refresh, reply);
|
||||||
|
let response = response.unwrap().into_response();
|
||||||
|
assert_eq!(response.status(), StatusCode::OK);
|
||||||
|
let body = axum::body::to_bytes(response.into_body(), 4096)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let result: RefreshFleetDevicesResponse = serde_json::from_slice(&body).unwrap();
|
||||||
|
assert_eq!(result.total_accepted, 1);
|
||||||
|
assert_eq!(result.agents.len(), 2);
|
||||||
|
assert!(
|
||||||
|
result
|
||||||
|
.agents
|
||||||
|
.windows(2)
|
||||||
|
.all(|pair| pair[0].agent_id < pair[1].agent_id)
|
||||||
|
);
|
||||||
|
let success = result
|
||||||
|
.agents
|
||||||
|
.iter()
|
||||||
|
.find(|row| row.agent_id == agent_id)
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(success.status, "ok");
|
||||||
|
assert_eq!(success.accepted, 1);
|
||||||
|
assert!(success.error.is_none());
|
||||||
|
let failed = result
|
||||||
|
.agents
|
||||||
|
.iter()
|
||||||
|
.find(|row| row.agent_id == "disconnected")
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(failed.status, "error");
|
||||||
|
assert_eq!(failed.accepted, 0);
|
||||||
|
assert_eq!(
|
||||||
|
failed.error.as_deref(),
|
||||||
|
Some("404 Not Found: agent is not connected")
|
||||||
|
);
|
||||||
|
assert!(state.pending.lock().await.is_empty());
|
||||||
|
assert!(matches!(
|
||||||
|
rx.try_recv(),
|
||||||
|
Err(mpsc::error::TryRecvError::Empty)
|
||||||
|
));
|
||||||
|
|
||||||
|
let after = super::super::load_fleet_devices(&state, &query)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(after.len(), 1);
|
||||||
|
assert_eq!(after[0].device_key, before[0].device_key);
|
||||||
|
assert_eq!(
|
||||||
|
after[0].ips,
|
||||||
|
["192.168.1.20".parse::<std::net::IpAddr>().unwrap()]
|
||||||
|
);
|
||||||
|
let stored = state
|
||||||
|
.store
|
||||||
|
.list_agent_device_rows_for_agent(&agent_id)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(stored.len(), 1);
|
||||||
|
assert_eq!(stored[0].endpoints.len(), 1);
|
||||||
|
assert_eq!(
|
||||||
|
stored[0].endpoints[0].key.ip,
|
||||||
|
Some("192.168.1.20".parse().unwrap())
|
||||||
|
);
|
||||||
|
drop(state);
|
||||||
|
std::fs::remove_dir_all(dir).unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn refresh_stores_snapshots_but_invalid_results_leave_them_untouched() {
|
||||||
|
let dir = std::env::temp_dir().join(format!("wakey-refresh-{}", uuid::Uuid::new_v4()));
|
||||||
|
let store = Store::load_or_init(
|
||||||
|
&dir.join("state.sqlite3"),
|
||||||
|
vec![],
|
||||||
|
std::time::Duration::from_secs(60),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let device = wakey_core::Device::from_parts(
|
||||||
|
vec![wakey_core::NeighborEntry {
|
||||||
|
ip: "192.168.1.10".parse().unwrap(),
|
||||||
|
mac: Some("aa:bb:cc:dd:ee:01".parse().unwrap()),
|
||||||
|
dev: Some("br-lan".into()),
|
||||||
|
state: wakey_core::NeighborState::Reachable,
|
||||||
|
}],
|
||||||
|
vec![],
|
||||||
|
);
|
||||||
|
let inventory = serde_json::to_value(wakey_core::DeviceInventory {
|
||||||
|
devices: vec![device],
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
let stored = store_inventory(&store, "agent", Ok(response(Some(inventory)))).await;
|
||||||
|
assert_eq!(stored.status, "ok");
|
||||||
|
assert_eq!(stored.accepted, 1);
|
||||||
|
let revision = store.fleet_revision();
|
||||||
|
let mut command_failed = response(None);
|
||||||
|
command_failed.status = "error".into();
|
||||||
|
command_failed.error = Some(wakey_agent::protocol::ErrorPayload {
|
||||||
|
code: "inventory_failed".into(),
|
||||||
|
message: "source unavailable".into(),
|
||||||
|
retryable: None,
|
||||||
|
});
|
||||||
|
for failed in [
|
||||||
|
Ok(response(None)),
|
||||||
|
Ok(response(Some(json!({"devices": "invalid"})))),
|
||||||
|
Ok(command_failed),
|
||||||
|
Err(ApiError::new(
|
||||||
|
StatusCode::BAD_GATEWAY,
|
||||||
|
"agent_send_failed",
|
||||||
|
"send failed",
|
||||||
|
)),
|
||||||
|
] {
|
||||||
|
let invalid = store_inventory(&store, "agent", failed).await;
|
||||||
|
assert_eq!(invalid.status, "error");
|
||||||
|
assert_eq!(invalid.accepted, 0);
|
||||||
|
assert!(invalid.error.is_some());
|
||||||
|
assert_eq!(store.fleet_revision(), revision);
|
||||||
|
let rows = store
|
||||||
|
.list_agent_device_rows_for_agent("agent")
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(rows.len(), 1);
|
||||||
|
assert_eq!(rows[0].endpoints.len(), 1);
|
||||||
|
assert_eq!(
|
||||||
|
rows[0].endpoints[0].key.ip,
|
||||||
|
Some("192.168.1.10".parse().unwrap())
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let rows = store
|
||||||
|
.list_agent_device_rows_for_agent("agent")
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(rows.len(), 1);
|
||||||
|
assert_eq!(rows[0].endpoints.len(), 1);
|
||||||
|
let empty =
|
||||||
|
store_inventory(&store, "agent", Ok(response(Some(json!({"devices": []}))))).await;
|
||||||
|
assert_eq!(empty.status, "ok");
|
||||||
|
assert_eq!(empty.accepted, 0);
|
||||||
|
assert!(
|
||||||
|
store
|
||||||
|
.list_agent_device_rows_for_agent("agent")
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.is_empty()
|
||||||
|
);
|
||||||
|
drop(store);
|
||||||
|
std::fs::remove_dir_all(dir).unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -8,9 +8,333 @@ use super::build::{
|
|||||||
AgentRuntimeStatus, FleetBuildContext, build_fleet_devices, filter_fleet_devices,
|
AgentRuntimeStatus, FleetBuildContext, build_fleet_devices, filter_fleet_devices,
|
||||||
known_device_summary,
|
known_device_summary,
|
||||||
};
|
};
|
||||||
|
use super::refresh::normalize_refresh_agent_ids;
|
||||||
use super::types::ListFleetDevicesQuery;
|
use super::types::ListFleetDevicesQuery;
|
||||||
use crate::state::{AgentDeviceRow, AgentDeviceWithChildren, DeviceIdentifier, KnownDevice};
|
use crate::state::{AgentDeviceRow, AgentDeviceWithChildren, DeviceIdentifier, KnownDevice};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn cache_rebuild_limit_is_retryable_but_db_failures_keep_their_error_code() {
|
||||||
|
let busy = super::fleet_error(
|
||||||
|
crate::runtime::view_cache::RebuildLimit.into(),
|
||||||
|
"list_fleet_devices_failed",
|
||||||
|
);
|
||||||
|
assert_eq!(busy.status, axum::http::StatusCode::SERVICE_UNAVAILABLE);
|
||||||
|
assert_eq!(busy.code, "fleet_busy");
|
||||||
|
let failed = super::fleet_error(
|
||||||
|
anyhow::anyhow!("database failed"),
|
||||||
|
"list_fleet_devices_failed",
|
||||||
|
);
|
||||||
|
assert_eq!(failed.status, axum::http::StatusCode::INTERNAL_SERVER_ERROR);
|
||||||
|
assert_eq!(failed.code, "list_fleet_devices_failed");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn fleet_cache_tracks_store_edits_and_connections_without_caching_filters() {
|
||||||
|
use crate::runtime::{AgentSession, AppState};
|
||||||
|
use crate::state::{DeviceIdentifierInput, KnownDeviceInput, Store};
|
||||||
|
use std::sync::Arc;
|
||||||
|
use std::time::Duration;
|
||||||
|
use tokio::sync::{Mutex, RwLock, mpsc};
|
||||||
|
|
||||||
|
let dir = std::env::temp_dir().join(format!("wakey-fleet-cache-{}", uuid::Uuid::new_v4()));
|
||||||
|
let state = AppState {
|
||||||
|
store: Arc::new(
|
||||||
|
Store::load_or_init(&dir.join("state.sqlite3"), vec![], Duration::from_secs(60))
|
||||||
|
.await
|
||||||
|
.unwrap(),
|
||||||
|
),
|
||||||
|
sessions: Arc::new(RwLock::new(HashMap::new())),
|
||||||
|
pending: Arc::new(Mutex::new(HashMap::new())),
|
||||||
|
public_url: "http://localhost".into(),
|
||||||
|
command_timeout: Duration::from_secs(30),
|
||||||
|
enroll_token_ttl: Duration::from_secs(60),
|
||||||
|
terminals: crate::runtime::terminals::TerminalRegistry::new(),
|
||||||
|
alerts: crate::api::AlertsHub::new(),
|
||||||
|
fleet: Arc::new(super::FleetCache::default()),
|
||||||
|
};
|
||||||
|
let store = &state.store;
|
||||||
|
assert_eq!(
|
||||||
|
super::sample_fleet_counts(&state).await.unwrap(),
|
||||||
|
Default::default()
|
||||||
|
);
|
||||||
|
let token = store
|
||||||
|
.issue_enroll_token(Duration::from_secs(60))
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let agent = store.enroll(&token.enroll_token).await.unwrap();
|
||||||
|
let empty = super::sample_fleet_counts(&state).await.unwrap();
|
||||||
|
assert_eq!(empty.agents[&agent.agent_id].devices, 0);
|
||||||
|
assert_eq!(empty.agents[&agent.agent_id].endpoints, 0);
|
||||||
|
let observed = wakey_core::Device::from_parts(
|
||||||
|
vec![wakey_core::NeighborEntry {
|
||||||
|
ip: "192.168.1.10".parse().unwrap(),
|
||||||
|
mac: Some("aa:bb:cc:dd:ee:01".parse().unwrap()),
|
||||||
|
dev: Some("br-lan".into()),
|
||||||
|
state: wakey_core::NeighborState::Reachable,
|
||||||
|
}],
|
||||||
|
vec![],
|
||||||
|
);
|
||||||
|
store
|
||||||
|
.replace_agent_device_snapshot(&agent.agent_id, &[observed])
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let initial = super::fleet_view(&state).await.unwrap();
|
||||||
|
assert_eq!(initial.len(), 1);
|
||||||
|
let counts = super::sample_fleet_counts(&state).await.unwrap();
|
||||||
|
assert_eq!(counts.agents[&agent.agent_id].devices, 1);
|
||||||
|
assert_eq!(counts.agents[&agent.agent_id].endpoints, 1);
|
||||||
|
assert_eq!((counts.saved, counts.unsaved), (0, 1));
|
||||||
|
assert!(Arc::ptr_eq(
|
||||||
|
&initial,
|
||||||
|
&super::fleet_view(&state).await.unwrap()
|
||||||
|
));
|
||||||
|
assert!(initial[0].recommended_route.is_none());
|
||||||
|
|
||||||
|
let (tx, _rx) = mpsc::unbounded_channel();
|
||||||
|
state.sessions.write().await.insert(
|
||||||
|
agent.agent_id.clone(),
|
||||||
|
AgentSession {
|
||||||
|
connection_id: "test".into(),
|
||||||
|
tx,
|
||||||
|
capabilities: vec![],
|
||||||
|
capability_options: Default::default(),
|
||||||
|
},
|
||||||
|
);
|
||||||
|
let connected = super::fleet_view(&state).await.unwrap();
|
||||||
|
assert!(connected[0].recommended_route.is_some());
|
||||||
|
store
|
||||||
|
.set_agent_nickname(&agent.agent_id, Some("router"))
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(
|
||||||
|
super::fleet_view(&state).await.unwrap()[0].agents[0]
|
||||||
|
.nickname
|
||||||
|
.as_deref(),
|
||||||
|
Some("router")
|
||||||
|
);
|
||||||
|
|
||||||
|
let saved = store
|
||||||
|
.create_known_device(KnownDeviceInput {
|
||||||
|
display_name: "my pc".into(),
|
||||||
|
pinned: true,
|
||||||
|
notes: None,
|
||||||
|
identifiers: vec![DeviceIdentifierInput {
|
||||||
|
kind: "mac".into(),
|
||||||
|
value: "aa:bb:cc:dd:ee:01".into(),
|
||||||
|
}],
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(
|
||||||
|
super::fleet_view(&state).await.unwrap()[0].display_name,
|
||||||
|
"my pc"
|
||||||
|
);
|
||||||
|
store
|
||||||
|
.rename_known_device(&saved.device_id, "renamed pc")
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let before_error = super::fleet_view(&state).await.unwrap();
|
||||||
|
assert!(
|
||||||
|
store
|
||||||
|
.rename_known_device(&saved.device_id, " ")
|
||||||
|
.await
|
||||||
|
.is_err()
|
||||||
|
);
|
||||||
|
assert!(Arc::ptr_eq(
|
||||||
|
&before_error,
|
||||||
|
&super::fleet_view(&state).await.unwrap()
|
||||||
|
));
|
||||||
|
assert_eq!(
|
||||||
|
super::fleet_view(&state).await.unwrap()[0].display_name,
|
||||||
|
"renamed pc"
|
||||||
|
);
|
||||||
|
let query = ListFleetDevicesQuery {
|
||||||
|
query: Some("does-not-exist".into()),
|
||||||
|
visibility: Some("all".into()),
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
assert!(
|
||||||
|
super::load_fleet_devices(&state, &query)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.is_empty()
|
||||||
|
);
|
||||||
|
// The filtered list excludes this row, but exact details still follow the
|
||||||
|
// authoritative saved key after remembering and renaming the observation.
|
||||||
|
{
|
||||||
|
use axum::{
|
||||||
|
extract::{Path, State},
|
||||||
|
response::IntoResponse,
|
||||||
|
};
|
||||||
|
let response = super::get_fleet_device(
|
||||||
|
State(state.clone()),
|
||||||
|
Path(format!("known:{}", saved.device_id)),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.into_response();
|
||||||
|
assert_eq!(response.status(), axum::http::StatusCode::OK);
|
||||||
|
let body = axum::body::to_bytes(response.into_body(), usize::MAX)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let details: serde_json::Value = serde_json::from_slice(&body).unwrap();
|
||||||
|
assert_eq!(details["display_name"], "renamed pc");
|
||||||
|
assert_eq!(details["known_device"]["device_id"], saved.device_id);
|
||||||
|
let missing =
|
||||||
|
super::get_fleet_device(State(state.clone()), Path(initial[0].device_key.clone()))
|
||||||
|
.await
|
||||||
|
.err()
|
||||||
|
.expect("old unsaved grouping key no longer exists");
|
||||||
|
assert_eq!(missing.status, axum::http::StatusCode::NOT_FOUND);
|
||||||
|
assert_eq!(missing.code, "fleet_device_not_found");
|
||||||
|
}
|
||||||
|
let counts = super::sample_fleet_counts(&state).await.unwrap();
|
||||||
|
assert_eq!((counts.saved, counts.unsaved), (1, 0));
|
||||||
|
let all = ListFleetDevicesQuery {
|
||||||
|
visibility: Some("all".into()),
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
assert_eq!(
|
||||||
|
super::load_fleet_devices(&state, &all).await.unwrap().len(),
|
||||||
|
1
|
||||||
|
);
|
||||||
|
|
||||||
|
store
|
||||||
|
.detach_device_identifier(&saved.device_id, &saved.identifiers[0].identifier_key)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(super::fleet_view(&state).await.unwrap().len(), 2);
|
||||||
|
let counts = super::sample_fleet_counts(&state).await.unwrap();
|
||||||
|
assert_eq!((counts.saved, counts.unsaved), (1, 1));
|
||||||
|
store
|
||||||
|
.attach_device_identifier(
|
||||||
|
&saved.device_id,
|
||||||
|
DeviceIdentifierInput {
|
||||||
|
kind: "mac".into(),
|
||||||
|
value: "aa:bb:cc:dd:ee:01".into(),
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(super::fleet_view(&state).await.unwrap().len(), 1);
|
||||||
|
let other = store
|
||||||
|
.create_known_device(KnownDeviceInput {
|
||||||
|
display_name: "other".into(),
|
||||||
|
pinned: false,
|
||||||
|
notes: None,
|
||||||
|
identifiers: vec![],
|
||||||
|
})
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(super::fleet_view(&state).await.unwrap().len(), 2);
|
||||||
|
let limited = ListFleetDevicesQuery {
|
||||||
|
visibility: Some("all".into()),
|
||||||
|
limit: Some(1),
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
assert_eq!(
|
||||||
|
super::load_fleet_devices(&state, &limited)
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.len(),
|
||||||
|
1
|
||||||
|
);
|
||||||
|
let counts = super::sample_fleet_counts(&state).await.unwrap();
|
||||||
|
assert_eq!((counts.saved, counts.unsaved), (2, 0));
|
||||||
|
assert_eq!(
|
||||||
|
super::load_fleet_devices(&state, &all).await.unwrap().len(),
|
||||||
|
2
|
||||||
|
);
|
||||||
|
store
|
||||||
|
.merge_known_devices(&saved.device_id, &other.device_id)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(super::fleet_view(&state).await.unwrap().len(), 1);
|
||||||
|
store.forget_known_device(&saved.device_id).await.unwrap();
|
||||||
|
assert!(
|
||||||
|
super::fleet_view(&state).await.unwrap()[0]
|
||||||
|
.known_device
|
||||||
|
.is_none()
|
||||||
|
);
|
||||||
|
|
||||||
|
state.sessions.write().await.clear();
|
||||||
|
let counts = super::sample_fleet_counts(&state).await.unwrap();
|
||||||
|
assert_eq!(counts.agents[&agent.agent_id].devices, 1);
|
||||||
|
assert_eq!((counts.saved, counts.unsaved), (0, 1));
|
||||||
|
assert!(
|
||||||
|
super::fleet_view(&state).await.unwrap()[0]
|
||||||
|
.recommended_route
|
||||||
|
.is_none()
|
||||||
|
);
|
||||||
|
let before_reload = super::fleet_view(&state).await.unwrap();
|
||||||
|
store.reload_from_disk().await.unwrap();
|
||||||
|
assert!(!Arc::ptr_eq(
|
||||||
|
&before_reload,
|
||||||
|
&super::fleet_view(&state).await.unwrap()
|
||||||
|
));
|
||||||
|
store.revoke_agent(&agent.agent_id).await.unwrap();
|
||||||
|
assert!(
|
||||||
|
super::fleet_view(&state).await.unwrap()[0].agents[0]
|
||||||
|
.nickname
|
||||||
|
.is_none()
|
||||||
|
);
|
||||||
|
store
|
||||||
|
.replace_agent_device_snapshot(&agent.agent_id, &[])
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert!(super::fleet_view(&state).await.unwrap().is_empty());
|
||||||
|
assert_eq!(
|
||||||
|
super::sample_fleet_counts(&state).await.unwrap(),
|
||||||
|
Default::default()
|
||||||
|
);
|
||||||
|
// Identical devices on two agents count twice as stored rows, once as a fleet identity.
|
||||||
|
// Offline endpoints are retained too; these metrics are not reachability counters.
|
||||||
|
let offline = wakey_core::Device::from_parts(
|
||||||
|
vec![wakey_core::NeighborEntry {
|
||||||
|
ip: "192.168.1.10".parse().unwrap(),
|
||||||
|
mac: Some("aa:bb:cc:dd:ee:01".parse().unwrap()),
|
||||||
|
dev: Some("br-lan".into()),
|
||||||
|
state: wakey_core::NeighborState::Failed,
|
||||||
|
}],
|
||||||
|
vec![],
|
||||||
|
);
|
||||||
|
for id in ["agent-a", "agent-b"] {
|
||||||
|
store
|
||||||
|
.replace_agent_device_snapshot(id, std::slice::from_ref(&offline))
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
let counts = super::sample_fleet_counts(&state).await.unwrap();
|
||||||
|
assert_eq!(counts.agents.len(), 2);
|
||||||
|
for count in counts.agents.values() {
|
||||||
|
assert_eq!(count.devices, 1);
|
||||||
|
assert_eq!(count.endpoints, 1);
|
||||||
|
}
|
||||||
|
assert_eq!((counts.saved, counts.unsaved), (0, 1));
|
||||||
|
store
|
||||||
|
.replace_agent_device_snapshot("agent-a", &[])
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let counts = super::sample_fleet_counts(&state).await.unwrap();
|
||||||
|
assert!(!counts.agents.contains_key("agent-a"));
|
||||||
|
assert_eq!(counts.agents["agent-b"].devices, 1);
|
||||||
|
drop(state);
|
||||||
|
std::fs::remove_dir_all(dir).unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn refresh_agent_selection_keeps_all_agents_beyond_previous_cap() {
|
||||||
|
let ids = (0..140)
|
||||||
|
.map(|index| format!("agent-{index:03}"))
|
||||||
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
|
let selected = normalize_refresh_agent_ids(ids);
|
||||||
|
|
||||||
|
assert_eq!(selected.len(), 140);
|
||||||
|
assert_eq!(selected.first().map(String::as_str), Some("agent-000"));
|
||||||
|
assert_eq!(selected.last().map(String::as_str), Some("agent-139"));
|
||||||
|
}
|
||||||
|
|
||||||
fn context(connected: &[&str]) -> FleetBuildContext {
|
fn context(connected: &[&str]) -> FleetBuildContext {
|
||||||
FleetBuildContext {
|
FleetBuildContext {
|
||||||
agent_status: connected
|
agent_status: connected
|
||||||
@@ -433,11 +757,153 @@ fn unknown_ip_only_offline_device_is_hidden_by_default() {
|
|||||||
);
|
);
|
||||||
|
|
||||||
assert_eq!(devices.len(), 1);
|
assert_eq!(devices.len(), 1);
|
||||||
assert_eq!(devices[0].display_name, "(unknown device)");
|
assert_eq!(devices[0].display_name, "Device: 192.168.1.2");
|
||||||
filter_fleet_devices(&mut devices, &ListFleetDevicesQuery::default());
|
filter_fleet_devices(&mut devices, &ListFleetDevicesQuery::default());
|
||||||
assert!(devices.is_empty());
|
assert!(devices.is_empty());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn endpoint_presence_overrides_conflicting_cached_summary() {
|
||||||
|
for connected in [false, true] {
|
||||||
|
for (cached, evidence) in [
|
||||||
|
(Presence::Unknown, Presence::Offline),
|
||||||
|
(Presence::Online, Presence::Unknown),
|
||||||
|
(Presence::Offline, Presence::Online),
|
||||||
|
] {
|
||||||
|
let mut row = offline_ip_only_unknown("agent-a", "ip:192.168.1.2", "192.168.1.2", 20);
|
||||||
|
row.device.presence = cached.as_str().into();
|
||||||
|
let mut endpoint = endpoint_with_presence(
|
||||||
|
EndpointSource::Neighbor,
|
||||||
|
None,
|
||||||
|
Some("192.168.1.2"),
|
||||||
|
evidence,
|
||||||
|
20,
|
||||||
|
);
|
||||||
|
endpoint.hostname = None;
|
||||||
|
row.endpoints = vec![endpoint];
|
||||||
|
let ctx = if connected {
|
||||||
|
context(&["agent-a"])
|
||||||
|
} else {
|
||||||
|
context(&[])
|
||||||
|
};
|
||||||
|
let devices = build_fleet_devices(Vec::new(), vec![row], &ctx);
|
||||||
|
assert_eq!(devices[0].presence, evidence);
|
||||||
|
assert_eq!(devices[0].endpoints[0].presence, evidence);
|
||||||
|
assert_eq!(devices[0].agents[0].connected, connected);
|
||||||
|
if evidence == Presence::Offline {
|
||||||
|
assert_eq!(devices[0].display_name, "Device: 192.168.1.2");
|
||||||
|
assert!(devices[0].ips.is_empty());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn unnamed_endpoint_labels_preserve_presence_and_inactive_ips() {
|
||||||
|
for (presence, mac, ip, expected) in [
|
||||||
|
(
|
||||||
|
Presence::Offline,
|
||||||
|
None,
|
||||||
|
Some("192.168.1.2"),
|
||||||
|
"Device: 192.168.1.2",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
Presence::Unknown,
|
||||||
|
None,
|
||||||
|
Some("192.168.1.2"),
|
||||||
|
"Device: 192.168.1.2",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
Presence::Offline,
|
||||||
|
Some("aa:bb:cc:dd:ee:01"),
|
||||||
|
None,
|
||||||
|
"Device: AA:BB:CC:DD:EE:01",
|
||||||
|
),
|
||||||
|
] {
|
||||||
|
let mut row = offline_ip_only_unknown("agent-a", "test", "192.168.1.2", 20);
|
||||||
|
row.ips.clear();
|
||||||
|
row.device.presence = presence.as_str().into();
|
||||||
|
let mut evidence = endpoint_with_presence(EndpointSource::Neighbor, mac, ip, presence, 20);
|
||||||
|
evidence.hostname = None;
|
||||||
|
row.endpoints = vec![evidence];
|
||||||
|
let devices = build_fleet_devices(Vec::new(), vec![row], &context(&["agent-a"]));
|
||||||
|
assert_eq!(devices[0].display_name, expected);
|
||||||
|
assert_eq!(devices[0].presence, presence);
|
||||||
|
if presence == Presence::Offline {
|
||||||
|
assert!(devices[0].ips.is_empty());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn unnamed_mac_device_does_not_use_inactive_ip_or_presence_as_its_name() {
|
||||||
|
for presence in [Presence::Offline, Presence::Unknown, Presence::Online] {
|
||||||
|
let mut row =
|
||||||
|
offline_ip_only_unknown("agent-a", "mac:22:07:53:BF:FC:12", "192.168.100.125", 20);
|
||||||
|
row.macs = vec!["22:07:53:bf:fc:12".parse().unwrap()];
|
||||||
|
row.hostnames.clear();
|
||||||
|
row.endpoints = [EndpointSource::Neighbor, EndpointSource::HookNeighbor]
|
||||||
|
.into_iter()
|
||||||
|
.map(|source| {
|
||||||
|
let mut endpoint = endpoint_with_presence(
|
||||||
|
source,
|
||||||
|
Some("22:07:53:bf:fc:12"),
|
||||||
|
Some("192.168.100.125"),
|
||||||
|
presence,
|
||||||
|
20,
|
||||||
|
);
|
||||||
|
endpoint.hostname = None;
|
||||||
|
endpoint
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
let devices = build_fleet_devices(Vec::new(), vec![row], &context(&["agent-a"]));
|
||||||
|
assert_eq!(devices[0].display_name, "Device: 22:07:53:BF:FC:12");
|
||||||
|
assert_eq!(devices[0].presence, presence);
|
||||||
|
assert_eq!(devices[0].endpoints.len(), 2);
|
||||||
|
if presence == Presence::Offline {
|
||||||
|
assert!(devices[0].ips.is_empty());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn ip_only_label_prefers_summary_ip_and_uses_inactive_evidence_only_as_fallback() {
|
||||||
|
for has_current_ip in [true, false] {
|
||||||
|
let mut row = offline_ip_only_unknown("agent-a", "test", "192.168.1.2", 20);
|
||||||
|
let mut inactive = endpoint_with_presence(
|
||||||
|
EndpointSource::Neighbor,
|
||||||
|
None,
|
||||||
|
Some("192.168.1.2"),
|
||||||
|
Presence::Offline,
|
||||||
|
20,
|
||||||
|
);
|
||||||
|
inactive.hostname = None;
|
||||||
|
row.endpoints = vec![inactive];
|
||||||
|
if has_current_ip {
|
||||||
|
let mut current = endpoint_with_presence(
|
||||||
|
EndpointSource::Neighbor,
|
||||||
|
None,
|
||||||
|
Some("192.168.1.10"),
|
||||||
|
Presence::Online,
|
||||||
|
20,
|
||||||
|
);
|
||||||
|
current.hostname = None;
|
||||||
|
row.endpoints.push(current);
|
||||||
|
}
|
||||||
|
let devices = build_fleet_devices(Vec::new(), vec![row], &context(&["agent-a"]));
|
||||||
|
if has_current_ip {
|
||||||
|
assert_eq!(devices[0].display_name, "Device: 192.168.1.10");
|
||||||
|
assert_eq!(
|
||||||
|
devices[0].ips,
|
||||||
|
vec!["192.168.1.10".parse::<IpAddr>().unwrap()]
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
assert_eq!(devices[0].display_name, "Device: 192.168.1.2");
|
||||||
|
assert!(devices[0].ips.is_empty());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn visibility_all_keeps_unknown_ip_only_offline_device() {
|
fn visibility_all_keeps_unknown_ip_only_offline_device() {
|
||||||
let mut devices = build_fleet_devices(
|
let mut devices = build_fleet_devices(
|
||||||
@@ -459,7 +925,7 @@ fn visibility_all_keeps_unknown_ip_only_offline_device() {
|
|||||||
},
|
},
|
||||||
);
|
);
|
||||||
assert_eq!(devices.len(), 1);
|
assert_eq!(devices.len(), 1);
|
||||||
assert_eq!(devices[0].display_name, "(unknown device)");
|
assert_eq!(devices[0].display_name, "Device: 192.168.1.2");
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
|
|||||||
@@ -0,0 +1,410 @@
|
|||||||
|
use axum::{Json, extract::State, http::StatusCode, response::IntoResponse};
|
||||||
|
use macaddr::MacAddr;
|
||||||
|
use tracing::{Instrument, info, warn};
|
||||||
|
use wakey_agent::protocol::{AgentCommand, WakeRequest};
|
||||||
|
|
||||||
|
use super::types::{FleetDevice, FleetWakeRoute, WakeFleetDeviceRequest, WakeFleetDeviceResponse};
|
||||||
|
use super::{fleet_error, fleet_view};
|
||||||
|
use crate::api::{ApiError, commands::relay_agent_command};
|
||||||
|
use crate::runtime::AppState;
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "fleet_wake", skip_all, fields(explicit_route = req.route_id.is_some(), outcome = tracing::field::Empty))]
|
||||||
|
pub async fn wake_fleet_device(
|
||||||
|
State(state): State<AppState>,
|
||||||
|
Json(req): Json<WakeFleetDeviceRequest>,
|
||||||
|
) -> Result<impl IntoResponse, ApiError> {
|
||||||
|
// Early database/relay errors retain an outcome even without a command result.
|
||||||
|
tracing::Span::current().record("outcome", "error");
|
||||||
|
let selected = load_selected_route(&state, &req.device_key, req.route_id.as_deref()).await?;
|
||||||
|
info!(agent_id = %selected.route.agent_id, has_ip = selected.route.ip.is_some(),
|
||||||
|
explicit_route = req.route_id.is_some(), "fleet wake route selected");
|
||||||
|
let response = dispatch_wake(&state, selected, req.timeout_ms).await?;
|
||||||
|
tracing::Span::current().record("outcome", response.command.status.as_str());
|
||||||
|
Ok((StatusCode::OK, Json(response)))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// List visibility and pagination are presentation policy, not wake eligibility.
|
||||||
|
/// Borrow the complete cached view and clone only the selected route.
|
||||||
|
async fn load_selected_route(
|
||||||
|
state: &AppState,
|
||||||
|
device_key: &str,
|
||||||
|
route_id: Option<&str>,
|
||||||
|
) -> Result<SelectedRoute, ApiError> {
|
||||||
|
let span = tracing::info_span!("fleet_load", grouped = tracing::field::Empty);
|
||||||
|
let devices = fleet_view(state)
|
||||||
|
.instrument(span.clone())
|
||||||
|
.await
|
||||||
|
.map_err(|err| {
|
||||||
|
warn!(error = %err, "failed loading fleet devices for wake");
|
||||||
|
fleet_error(err, "load_fleet_devices_failed")
|
||||||
|
})?;
|
||||||
|
span.record("grouped", devices.len());
|
||||||
|
select_route(&devices, device_key, route_id).map_err(|error| {
|
||||||
|
tracing::Span::current().record("outcome", error.outcome());
|
||||||
|
error.into_api_error()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Selection does not rank routes: the fleet builder already chose the recommendation.
|
||||||
|
/// An explicit route never falls back, including when it is disconnected or IP-only.
|
||||||
|
fn select_route(
|
||||||
|
devices: &[FleetDevice],
|
||||||
|
device_key: &str,
|
||||||
|
route_id: Option<&str>,
|
||||||
|
) -> Result<SelectedRoute, RouteError> {
|
||||||
|
let device = devices
|
||||||
|
.iter()
|
||||||
|
.find(|device| device.device_key == device_key)
|
||||||
|
.ok_or(RouteError::DeviceNotFound)?;
|
||||||
|
let route = match route_id {
|
||||||
|
Some(id) => device
|
||||||
|
.route_candidates
|
||||||
|
.iter()
|
||||||
|
.find(|route| route.route_id == id),
|
||||||
|
None => device.recommended_route.as_ref(),
|
||||||
|
}
|
||||||
|
.ok_or(RouteError::Unavailable)?;
|
||||||
|
let mac = route.mac.ok_or(RouteError::MissingMac)?;
|
||||||
|
if !route.connected {
|
||||||
|
return Err(RouteError::Disconnected);
|
||||||
|
}
|
||||||
|
if !route.wakeable {
|
||||||
|
return Err(RouteError::NotWakeable);
|
||||||
|
}
|
||||||
|
Ok(SelectedRoute {
|
||||||
|
route: route.clone(),
|
||||||
|
mac,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Only the selector constructs this; dispatch cannot accidentally use an IP-only route.
|
||||||
|
#[derive(Debug)]
|
||||||
|
struct SelectedRoute {
|
||||||
|
route: FleetWakeRoute,
|
||||||
|
mac: MacAddr,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl SelectedRoute {
|
||||||
|
fn command(&self) -> AgentCommand {
|
||||||
|
AgentCommand::Wake(WakeRequest {
|
||||||
|
query: None,
|
||||||
|
mac: Some(self.mac),
|
||||||
|
ip: self.route.ip,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn dispatch_wake(
|
||||||
|
state: &AppState,
|
||||||
|
selected: SelectedRoute,
|
||||||
|
timeout_ms: Option<u64>,
|
||||||
|
) -> Result<WakeFleetDeviceResponse, ApiError> {
|
||||||
|
// The relay checks the current agent connection again; cached connectivity
|
||||||
|
// is only selection evidence, not a guarantee that the agent is still there.
|
||||||
|
let command = relay_agent_command(
|
||||||
|
state,
|
||||||
|
&selected.route.agent_id,
|
||||||
|
selected.command(),
|
||||||
|
timeout_ms,
|
||||||
|
)
|
||||||
|
.await?;
|
||||||
|
Ok(WakeFleetDeviceResponse {
|
||||||
|
route: selected.route,
|
||||||
|
command,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq, Eq)]
|
||||||
|
enum RouteError {
|
||||||
|
DeviceNotFound,
|
||||||
|
Unavailable,
|
||||||
|
MissingMac,
|
||||||
|
Disconnected,
|
||||||
|
NotWakeable,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl RouteError {
|
||||||
|
fn outcome(&self) -> &'static str {
|
||||||
|
match self {
|
||||||
|
Self::DeviceNotFound => "device_not_found",
|
||||||
|
Self::Unavailable => "route_unavailable",
|
||||||
|
Self::MissingMac => "route_without_mac",
|
||||||
|
Self::Disconnected => "route_disconnected",
|
||||||
|
Self::NotWakeable => "route_not_wakeable",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn into_api_error(self) -> ApiError {
|
||||||
|
let message = match self {
|
||||||
|
Self::DeviceNotFound => {
|
||||||
|
return ApiError::new(
|
||||||
|
StatusCode::NOT_FOUND,
|
||||||
|
"fleet_device_not_found",
|
||||||
|
"fleet device not found",
|
||||||
|
);
|
||||||
|
}
|
||||||
|
Self::Unavailable => "no wakeable connected MAC-backed route is available",
|
||||||
|
Self::MissingMac => "selected route does not include a MAC address",
|
||||||
|
Self::Disconnected => "selected route agent is not connected",
|
||||||
|
Self::NotWakeable => "selected route is not wakeable",
|
||||||
|
};
|
||||||
|
ApiError::new(StatusCode::BAD_REQUEST, "wake_route_unavailable", message)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::super::{load_fleet_devices, types::ListFleetDevicesQuery};
|
||||||
|
use super::*;
|
||||||
|
use wakey_core::Presence;
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn wake_lookup_can_select_a_device_beyond_the_list_limit() {
|
||||||
|
use std::collections::{BTreeSet, HashMap};
|
||||||
|
use std::sync::Arc;
|
||||||
|
use std::time::Duration;
|
||||||
|
use tokio::sync::{Mutex, RwLock};
|
||||||
|
|
||||||
|
let dir = std::env::temp_dir().join(format!("wakey-wake-limit-{}", uuid::Uuid::new_v4()));
|
||||||
|
let state = AppState {
|
||||||
|
store: Arc::new(
|
||||||
|
crate::state::Store::load_or_init(
|
||||||
|
&dir.join("state.sqlite3"),
|
||||||
|
vec![],
|
||||||
|
Duration::from_secs(60),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap(),
|
||||||
|
),
|
||||||
|
sessions: Arc::new(RwLock::new(HashMap::new())),
|
||||||
|
pending: Arc::new(Mutex::new(HashMap::new())),
|
||||||
|
public_url: "http://localhost".into(),
|
||||||
|
command_timeout: Duration::from_secs(30),
|
||||||
|
enroll_token_ttl: Duration::from_secs(60),
|
||||||
|
terminals: crate::runtime::terminals::TerminalRegistry::new(),
|
||||||
|
alerts: crate::api::AlertsHub::new(),
|
||||||
|
fleet: Arc::new(super::super::FleetCache::default()),
|
||||||
|
};
|
||||||
|
// Seed the complete cached view, as the builder would. Exercise both
|
||||||
|
// production read paths without thousands of unrelated SQL inserts.
|
||||||
|
state
|
||||||
|
.fleet
|
||||||
|
.get_or_try_init(
|
||||||
|
|| async { (state.store.fleet_revision(), BTreeSet::new()) },
|
||||||
|
|| async {
|
||||||
|
Ok((0..1001)
|
||||||
|
.map(|i| {
|
||||||
|
let mut row = device(vec![route("valid")], Some(route("valid")));
|
||||||
|
row.device_key = format!("pc-{i}");
|
||||||
|
row
|
||||||
|
})
|
||||||
|
.collect())
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let listed = load_fleet_devices(
|
||||||
|
&state,
|
||||||
|
&ListFleetDevicesQuery {
|
||||||
|
visibility: Some("all".into()),
|
||||||
|
limit: Some(1000),
|
||||||
|
..Default::default()
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(listed.len(), 1000);
|
||||||
|
assert!(listed.iter().all(|row| row.device_key != "pc-1000"));
|
||||||
|
{
|
||||||
|
use axum::extract::Path;
|
||||||
|
let details =
|
||||||
|
super::super::get_fleet_device(State(state.clone()), Path("pc-1000".into()))
|
||||||
|
.await
|
||||||
|
.unwrap()
|
||||||
|
.into_response();
|
||||||
|
assert_eq!(details.status(), StatusCode::OK);
|
||||||
|
let body = axum::body::to_bytes(details.into_body(), usize::MAX)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
let row: serde_json::Value = serde_json::from_slice(&body).unwrap();
|
||||||
|
assert_eq!(row["device_key"], "pc-1000");
|
||||||
|
}
|
||||||
|
for route_id in [None, Some("valid")] {
|
||||||
|
let selected = load_selected_route(&state, "pc-1000", route_id)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(selected.route.route_id, "valid");
|
||||||
|
}
|
||||||
|
drop(state);
|
||||||
|
std::fs::remove_dir_all(dir).unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
fn route(id: &str) -> FleetWakeRoute {
|
||||||
|
FleetWakeRoute {
|
||||||
|
route_id: id.into(),
|
||||||
|
agent_id: "agent-a".into(),
|
||||||
|
nickname: None,
|
||||||
|
connected: true,
|
||||||
|
mac: Some("aa:bb:cc:dd:ee:01".parse().unwrap()),
|
||||||
|
ip: Some("192.168.1.10".parse().unwrap()),
|
||||||
|
hostname: None,
|
||||||
|
interface: None,
|
||||||
|
source: "neighbor".into(),
|
||||||
|
presence: Presence::Offline,
|
||||||
|
last_seen_unix: 1,
|
||||||
|
wakeable: true,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn device(candidates: Vec<FleetWakeRoute>, recommended: Option<FleetWakeRoute>) -> FleetDevice {
|
||||||
|
FleetDevice {
|
||||||
|
device_key: "pc".into(),
|
||||||
|
display_name: "pc".into(),
|
||||||
|
known_device: None,
|
||||||
|
pinned: false,
|
||||||
|
ips: vec![],
|
||||||
|
macs: vec![],
|
||||||
|
hostnames: vec![],
|
||||||
|
agents: vec![],
|
||||||
|
sources: vec![],
|
||||||
|
endpoints: vec![],
|
||||||
|
first_seen_unix: None,
|
||||||
|
last_seen_unix: None,
|
||||||
|
presence: Presence::Offline,
|
||||||
|
route_candidates: candidates,
|
||||||
|
recommended_route: recommended,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn selection_uses_recommendation_or_exact_override_and_preserves_address_pair() {
|
||||||
|
let first = route("first");
|
||||||
|
let mut recommended = route("recommended");
|
||||||
|
recommended.agent_id = "agent-b".into();
|
||||||
|
recommended.ip = Some("192.168.2.20".parse().unwrap());
|
||||||
|
let devices = [device(vec![first, recommended.clone()], Some(recommended))];
|
||||||
|
for (override_id, route_id, agent, ip) in [
|
||||||
|
(None, "recommended", "agent-b", "192.168.2.20"),
|
||||||
|
(Some("first"), "first", "agent-a", "192.168.1.10"),
|
||||||
|
] {
|
||||||
|
let selected = select_route(&devices, "pc", override_id).unwrap();
|
||||||
|
assert_eq!(selected.route.route_id, route_id);
|
||||||
|
assert_eq!(selected.route.agent_id, agent);
|
||||||
|
let AgentCommand::Wake(command) = selected.command() else {
|
||||||
|
panic!("not a wake command")
|
||||||
|
};
|
||||||
|
assert!(command.query.is_none());
|
||||||
|
assert_eq!(command.mac, selected.route.mac);
|
||||||
|
assert_eq!(command.ip, Some(ip.parse().unwrap()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn missing_device_and_missing_routes_keep_distinct_errors() {
|
||||||
|
let valid = route("valid");
|
||||||
|
let devices = [device(vec![valid.clone()], Some(valid))];
|
||||||
|
assert_eq!(
|
||||||
|
select_route(&devices, "missing", None).unwrap_err(),
|
||||||
|
RouteError::DeviceNotFound
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
select_route(&devices, "pc", Some("missing")).unwrap_err(),
|
||||||
|
RouteError::Unavailable
|
||||||
|
);
|
||||||
|
let devices = [device(vec![route("valid")], None)];
|
||||||
|
assert_eq!(
|
||||||
|
select_route(&devices, "pc", None).unwrap_err(),
|
||||||
|
RouteError::Unavailable
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn invalid_explicit_routes_do_not_fall_back_and_validation_order_is_preserved() {
|
||||||
|
for (mac, connected, wakeable, expected) in [
|
||||||
|
(None, false, false, RouteError::MissingMac),
|
||||||
|
(
|
||||||
|
Some("aa:bb:cc:dd:ee:01"),
|
||||||
|
false,
|
||||||
|
false,
|
||||||
|
RouteError::Disconnected,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
Some("aa:bb:cc:dd:ee:01"),
|
||||||
|
true,
|
||||||
|
false,
|
||||||
|
RouteError::NotWakeable,
|
||||||
|
),
|
||||||
|
] {
|
||||||
|
let mut invalid = route("invalid");
|
||||||
|
invalid.mac = mac.map(|value| value.parse().unwrap());
|
||||||
|
invalid.connected = connected;
|
||||||
|
invalid.wakeable = wakeable;
|
||||||
|
let devices = [device(vec![invalid, route("valid")], Some(route("valid")))];
|
||||||
|
assert_eq!(
|
||||||
|
select_route(&devices, "pc", Some("invalid")).unwrap_err(),
|
||||||
|
expected
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn mac_only_offline_route_is_still_wakeable() {
|
||||||
|
let mut mac_only = route("mac-only");
|
||||||
|
mac_only.ip = None;
|
||||||
|
let devices = [device(vec![mac_only.clone()], Some(mac_only))];
|
||||||
|
let selected = select_route(&devices, "pc", None).unwrap();
|
||||||
|
let AgentCommand::Wake(command) = selected.command() else {
|
||||||
|
panic!("not a wake command")
|
||||||
|
};
|
||||||
|
assert!(command.mac.is_some());
|
||||||
|
assert!(command.ip.is_none());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn route_errors_preserve_http_codes_messages_and_trace_outcomes() {
|
||||||
|
for (error, status, code, message, outcome) in [
|
||||||
|
(
|
||||||
|
RouteError::DeviceNotFound,
|
||||||
|
StatusCode::NOT_FOUND,
|
||||||
|
"fleet_device_not_found",
|
||||||
|
"fleet device not found",
|
||||||
|
"device_not_found",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
RouteError::Unavailable,
|
||||||
|
StatusCode::BAD_REQUEST,
|
||||||
|
"wake_route_unavailable",
|
||||||
|
"no wakeable connected MAC-backed route is available",
|
||||||
|
"route_unavailable",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
RouteError::MissingMac,
|
||||||
|
StatusCode::BAD_REQUEST,
|
||||||
|
"wake_route_unavailable",
|
||||||
|
"selected route does not include a MAC address",
|
||||||
|
"route_without_mac",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
RouteError::Disconnected,
|
||||||
|
StatusCode::BAD_REQUEST,
|
||||||
|
"wake_route_unavailable",
|
||||||
|
"selected route agent is not connected",
|
||||||
|
"route_disconnected",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
RouteError::NotWakeable,
|
||||||
|
StatusCode::BAD_REQUEST,
|
||||||
|
"wake_route_unavailable",
|
||||||
|
"selected route is not wakeable",
|
||||||
|
"route_not_wakeable",
|
||||||
|
),
|
||||||
|
] {
|
||||||
|
assert_eq!(error.outcome(), outcome);
|
||||||
|
let response = error.into_api_error();
|
||||||
|
assert_eq!(response.status, status);
|
||||||
|
assert_eq!(response.code, code);
|
||||||
|
assert_eq!(response.message, message);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -6,20 +6,23 @@ mod audit;
|
|||||||
mod commands;
|
mod commands;
|
||||||
mod control;
|
mod control;
|
||||||
mod terminals;
|
mod terminals;
|
||||||
|
pub(crate) use control::FleetCache;
|
||||||
|
pub(crate) use control::sample_fleet_counts;
|
||||||
|
|
||||||
pub use alerts::{active_alerts, alert_history, alerts_stream};
|
pub use alerts::{AlertsHub, active_alerts, alert_history, alerts_stream, run_alert_evaluator};
|
||||||
pub use audit::list_audit_events;
|
pub use audit::list_audit_events;
|
||||||
pub use commands::{list_agents, run_command};
|
pub use commands::{list_agents, run_command};
|
||||||
pub use control::{
|
pub use control::{
|
||||||
EnrollTokenStatus, IssueEnrollTokenResponse, RevokeAgentResponse, RevokeEnrollTokenResponse,
|
EnrollTokenStatus, IssueEnrollTokenResponse, RevokeAgentResponse, RevokeEnrollTokenResponse,
|
||||||
StateStatsResponse, attach_device_identifier, create_known_device, detach_device_identifier,
|
StateStatsResponse, attach_device_identifier, create_known_device, detach_device_identifier,
|
||||||
enroll, forget_known_device, get_known_device, healthz, issue_enroll_token, list_enroll_tokens,
|
enroll, forget_known_device, get_fleet_device, get_known_device, healthz, issue_enroll_token,
|
||||||
list_fleet_devices, list_known_devices, merge_known_device, refresh_fleet_devices,
|
list_enroll_tokens, list_fleet_devices, list_known_devices, merge_known_device,
|
||||||
revoke_agent, revoke_enroll_token, set_agent_nickname, state_stats, wake_fleet_device,
|
refresh_fleet_devices, rename_known_device, revoke_agent, revoke_enroll_token,
|
||||||
|
set_agent_nickname, state_stats, wake_fleet_device,
|
||||||
};
|
};
|
||||||
pub use terminals::{
|
pub use terminals::{
|
||||||
agent_terminal_ws, attach_terminal, close_terminal, create_terminal, get_terminal,
|
agent_terminal_ws, attach_terminal, close_terminal, create_terminal, get_terminal,
|
||||||
list_terminals, operator_terminal_ws,
|
list_terminals, operator_terminal_ws, terminal_events_ws,
|
||||||
};
|
};
|
||||||
|
|
||||||
use axum::response::{IntoResponse, Response};
|
use axum::response::{IntoResponse, Response};
|
||||||
|
|||||||
@@ -6,7 +6,8 @@ use axum::response::Response;
|
|||||||
use futures_util::{SinkExt, StreamExt};
|
use futures_util::{SinkExt, StreamExt};
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
use tracing::{info, warn};
|
use tokio::sync::broadcast;
|
||||||
|
use tracing::{Instrument, info, warn};
|
||||||
use wakey_agent::protocol::{
|
use wakey_agent::protocol::{
|
||||||
AgentCapability, DEFAULT_TERMINAL_MAX_SESSIONS, DEFAULT_TERMINAL_SESSION_TTL_SECONDS,
|
AgentCapability, DEFAULT_TERMINAL_MAX_SESSIONS, DEFAULT_TERMINAL_SESSION_TTL_SECONDS,
|
||||||
ServerMessage, TerminalAgentHandshake, TerminalControl, TerminalId, TerminalOperatorHandshake,
|
ServerMessage, TerminalAgentHandshake, TerminalControl, TerminalId, TerminalOperatorHandshake,
|
||||||
@@ -48,6 +49,7 @@ pub struct TerminalSessionResponse {
|
|||||||
pub attachment_token: Option<String>,
|
pub attachment_token: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "terminal_create", skip_all)]
|
||||||
pub async fn create_terminal(
|
pub async fn create_terminal(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Json(request): Json<CreateTerminalRequest>,
|
Json(request): Json<CreateTerminalRequest>,
|
||||||
@@ -182,6 +184,7 @@ pub async fn list_terminals(State(state): State<AppState>) -> Json<Vec<TerminalS
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "terminal_attach", skip_all)]
|
||||||
pub async fn attach_terminal(
|
pub async fn attach_terminal(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Path(terminal_id): Path<String>,
|
Path(terminal_id): Path<String>,
|
||||||
@@ -209,6 +212,7 @@ pub async fn attach_terminal(
|
|||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tracing::instrument(name = "terminal_close", skip_all)]
|
||||||
pub async fn close_terminal(
|
pub async fn close_terminal(
|
||||||
State(state): State<AppState>,
|
State(state): State<AppState>,
|
||||||
Path(terminal_id): Path<String>,
|
Path(terminal_id): Path<String>,
|
||||||
@@ -244,6 +248,49 @@ pub async fn operator_terminal_ws(
|
|||||||
.on_upgrade(move |socket| handle_operator_terminal_socket(state, terminal_id, socket))
|
.on_upgrade(move |socket| handle_operator_terminal_socket(state, terminal_id, socket))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn terminal_events_ws(ws: WebSocketUpgrade, State(state): State<AppState>) -> Response {
|
||||||
|
ws.on_upgrade(move |socket| handle_terminal_events_socket(state, socket))
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle_terminal_events_socket(state: AppState, socket: WebSocket) {
|
||||||
|
let mut events = state.terminals.subscribe();
|
||||||
|
let (mut write, mut read) = socket.split();
|
||||||
|
if write
|
||||||
|
.send(Message::Text(r#"{"type":"sessions_changed"}"#.into()))
|
||||||
|
.await
|
||||||
|
.is_err()
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
loop {
|
||||||
|
tokio::select! {
|
||||||
|
event = events.recv() => {
|
||||||
|
match event {
|
||||||
|
Ok(()) | Err(broadcast::error::RecvError::Lagged(_)) => {
|
||||||
|
if write.send(Message::Text(r#"{"type":"sessions_changed"}"#.into())).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(broadcast::error::RecvError::Closed) => break,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
incoming = read.next() => {
|
||||||
|
match incoming {
|
||||||
|
Some(Ok(Message::Ping(payload))) => {
|
||||||
|
if write.send(Message::Pong(payload)).await.is_err() {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Some(Ok(Message::Pong(_))) => {}
|
||||||
|
Some(Ok(Message::Close(_))) | None | Some(Err(_)) => break,
|
||||||
|
Some(Ok(Message::Text(_))) | Some(Ok(Message::Binary(_))) => {}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async fn handle_agent_terminal_socket(state: AppState, terminal_id: String, mut socket: WebSocket) {
|
async fn handle_agent_terminal_socket(state: AppState, terminal_id: String, mut socket: WebSocket) {
|
||||||
let auth = match receive_text_handshake(&mut socket).await.and_then(|text| {
|
let auth = match receive_text_handshake(&mut socket).await.and_then(|text| {
|
||||||
serde_json::from_str::<TerminalAgentHandshake>(&text).map_err(|_| "invalid handshake")
|
serde_json::from_str::<TerminalAgentHandshake>(&text).map_err(|_| "invalid handshake")
|
||||||
@@ -260,10 +307,17 @@ async fn handle_agent_terminal_socket(state: AppState, terminal_id: String, mut
|
|||||||
let (mut outbound, pending) = match state
|
let (mut outbound, pending) = match state
|
||||||
.terminals
|
.terminals
|
||||||
.attach_agent(&terminal_id, &auth.0, &auth.1)
|
.attach_agent(&terminal_id, &auth.0, &auth.1)
|
||||||
|
.instrument(
|
||||||
|
tracing::info_span!("terminal_agent_transport_attach", terminal_id = %terminal_id),
|
||||||
|
)
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(outbound) => outbound,
|
Ok(outbound) => outbound,
|
||||||
Err(code) => {
|
Err(code) => {
|
||||||
|
warn!(
|
||||||
|
terminal_id,
|
||||||
|
code, "agent terminal transport attachment rejected"
|
||||||
|
);
|
||||||
close_socket(&mut socket, code).await;
|
close_socket(&mut socket, code).await;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@@ -346,10 +400,17 @@ async fn handle_operator_terminal_socket(
|
|||||||
let mut outbound = match state
|
let mut outbound = match state
|
||||||
.terminals
|
.terminals
|
||||||
.attach_operator(&terminal_id, &auth.0, &auth.1)
|
.attach_operator(&terminal_id, &auth.0, &auth.1)
|
||||||
|
.instrument(
|
||||||
|
tracing::info_span!("terminal_operator_transport_attach", terminal_id = %terminal_id),
|
||||||
|
)
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
Ok(attached) => attached,
|
Ok(attached) => attached,
|
||||||
Err(code) => {
|
Err(code) => {
|
||||||
|
warn!(
|
||||||
|
terminal_id,
|
||||||
|
code, "operator terminal transport attachment rejected"
|
||||||
|
);
|
||||||
close_socket(&mut socket, code).await;
|
close_socket(&mut socket, code).await;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@@ -360,6 +421,7 @@ async fn handle_operator_terminal_socket(
|
|||||||
if let Err(code) = state
|
if let Err(code) = state
|
||||||
.terminals
|
.terminals
|
||||||
.relay_to_agent(&terminal_id, TerminalRelayFrame::Text(snapshot))
|
.relay_to_agent(&terminal_id, TerminalRelayFrame::Text(snapshot))
|
||||||
|
.instrument(tracing::info_span!("terminal_snapshot_request", terminal_id = %terminal_id))
|
||||||
.await
|
.await
|
||||||
{
|
{
|
||||||
warn!(terminal_id, code, "failed to request terminal snapshot");
|
warn!(terminal_id, code, "failed to request terminal snapshot");
|
||||||
@@ -649,9 +711,9 @@ async fn append_terminal_audit(
|
|||||||
terminal_id: &str,
|
terminal_id: &str,
|
||||||
audit: TerminalAudit<'_>,
|
audit: TerminalAudit<'_>,
|
||||||
) {
|
) {
|
||||||
if let Err(err) = state
|
state
|
||||||
.store
|
.store
|
||||||
.append_audit_event(AuditEventInput {
|
.audit(AuditEventInput {
|
||||||
actor_type: audit.actor_type.into(),
|
actor_type: audit.actor_type.into(),
|
||||||
actor_id: None,
|
actor_id: None,
|
||||||
agent_id: Some(agent_id.to_string()),
|
agent_id: Some(agent_id.to_string()),
|
||||||
@@ -662,10 +724,7 @@ async fn append_terminal_audit(
|
|||||||
message: audit.message.into(),
|
message: audit.message.into(),
|
||||||
metadata: audit.metadata,
|
metadata: audit.metadata,
|
||||||
})
|
})
|
||||||
.await
|
.await;
|
||||||
{
|
|
||||||
warn!(terminal_id, event_type = audit.event_type, error = %err, "failed to append terminal audit event");
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn validate_size(rows: u16, cols: u16) -> Result<(), ApiError> {
|
fn validate_size(rows: u16, cols: u16) -> Result<(), ApiError> {
|
||||||
|
|||||||
@@ -72,6 +72,7 @@ pub struct ServeArgs {
|
|||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
pub enroll_token_ttl_seconds: Option<u64>,
|
pub enroll_token_ttl_seconds: Option<u64>,
|
||||||
|
|
||||||
|
/// Audit/alert history retention in seconds (legacy CLI name; not device retention).
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
pub observation_retention_seconds: Option<u64>,
|
pub observation_retention_seconds: Option<u64>,
|
||||||
|
|
||||||
@@ -129,6 +130,7 @@ pub struct InitConfigArgs {
|
|||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
pub enroll_token_ttl_seconds: Option<u64>,
|
pub enroll_token_ttl_seconds: Option<u64>,
|
||||||
|
|
||||||
|
/// Audit/alert history retention in seconds (legacy CLI name; not device retention).
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
pub observation_retention_seconds: Option<u64>,
|
pub observation_retention_seconds: Option<u64>,
|
||||||
|
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ use anyhow::{Context, Result};
|
|||||||
|
|
||||||
use crate::cli::{InitConfigArgs, ServeArgs};
|
use crate::cli::{InitConfigArgs, ServeArgs};
|
||||||
use crate::config::resolve::{load_file_config, normalize_public_url, resolve_path};
|
use crate::config::resolve::{load_file_config, normalize_public_url, resolve_path};
|
||||||
use crate::config::types::{DEFAULT_CONFIG, WritableConfig, WritableTelemetry};
|
use crate::config::types::{DEFAULT_CONFIG, FileConfig, FileTelemetryConfig};
|
||||||
|
|
||||||
pub fn write_init_config(args: &InitConfigArgs) -> Result<Option<PathBuf>> {
|
pub fn write_init_config(args: &InitConfigArgs) -> Result<Option<PathBuf>> {
|
||||||
if args.stdout && args.config_file.is_some() {
|
if args.stdout && args.config_file.is_some() {
|
||||||
@@ -44,10 +44,7 @@ pub fn write_init_config(args: &InitConfigArgs) -> Result<Option<PathBuf>> {
|
|||||||
.unwrap_or_else(|| "<defaults>".to_string())
|
.unwrap_or_else(|| "<defaults>".to_string())
|
||||||
)
|
)
|
||||||
})?,
|
})?,
|
||||||
None => defaults
|
None => defaults.bind,
|
||||||
.bind
|
|
||||||
.parse()
|
|
||||||
.expect("static default bind should parse"),
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
let public_url = normalize_public_url(
|
let public_url = normalize_public_url(
|
||||||
@@ -90,44 +87,71 @@ pub fn write_init_config(args: &InitConfigArgs) -> Result<Option<PathBuf>> {
|
|||||||
args.bootstrap_enroll_tokens.clone()
|
args.bootstrap_enroll_tokens.clone()
|
||||||
};
|
};
|
||||||
|
|
||||||
let body = WritableConfig {
|
let command_timeout = args
|
||||||
data_dir,
|
|
||||||
bind: bind.to_string(),
|
|
||||||
public_url,
|
|
||||||
state_file,
|
|
||||||
command_timeout_ms: args
|
|
||||||
.command_timeout_ms
|
.command_timeout_ms
|
||||||
.or(base.command_timeout_ms)
|
.map(std::time::Duration::from_millis)
|
||||||
.unwrap_or(defaults.command_timeout_ms)
|
.or(base.command_timeout)
|
||||||
.max(1),
|
.unwrap_or(defaults.command_timeout);
|
||||||
enroll_token_ttl_seconds: args
|
let enroll_token_ttl = args
|
||||||
.enroll_token_ttl_seconds
|
.enroll_token_ttl_seconds
|
||||||
.or(base.enroll_token_ttl_seconds)
|
.map(std::time::Duration::from_secs)
|
||||||
.unwrap_or(defaults.enroll_token_ttl_seconds)
|
.or(base.enroll_token_ttl)
|
||||||
.max(1),
|
.unwrap_or(defaults.enroll_token_ttl);
|
||||||
observation_retention_seconds: args
|
let history_retention = args
|
||||||
.observation_retention_seconds
|
.observation_retention_seconds
|
||||||
.or(base.observation_retention_seconds)
|
.map(std::time::Duration::from_secs)
|
||||||
.unwrap_or(defaults.observation_retention_seconds),
|
.or(base.history_retention)
|
||||||
pid_file,
|
.unwrap_or(defaults.history_retention);
|
||||||
ui_dist_dir,
|
super::resolve::validate_durations(command_timeout, enroll_token_ttl, history_retention)?;
|
||||||
bootstrap_enroll_tokens,
|
let device_retention = base.device_retention.unwrap_or(defaults.device_retention);
|
||||||
telemetry: WritableTelemetry {
|
super::resolve::validate_device_retention(device_retention)?;
|
||||||
|
|
||||||
|
let body = FileConfig {
|
||||||
|
access: base.access,
|
||||||
|
data_dir: Some(data_dir),
|
||||||
|
bind: Some(bind.to_string()),
|
||||||
|
public_url: Some(public_url),
|
||||||
|
state_file: Some(state_file),
|
||||||
|
command_timeout: Some(command_timeout),
|
||||||
|
enroll_token_ttl: Some(enroll_token_ttl),
|
||||||
|
history_retention: Some(history_retention),
|
||||||
|
device_retention: Some(device_retention),
|
||||||
|
pid_file: Some(pid_file),
|
||||||
|
ui_dist_dir: Some(ui_dist_dir),
|
||||||
|
// Preserve the previous generated template's omission of an empty list.
|
||||||
|
// Explicit empty lists still round-trip through FileConfig unchanged.
|
||||||
|
bootstrap_enroll_tokens: (!bootstrap_enroll_tokens.is_empty())
|
||||||
|
.then_some(bootstrap_enroll_tokens),
|
||||||
|
telemetry: Some(FileTelemetryConfig {
|
||||||
|
console_filter: base_telemetry
|
||||||
|
.console_filter
|
||||||
|
.or_else(|| defaults.telemetry.console_filter.clone()),
|
||||||
|
otlp_log_filter: base_telemetry
|
||||||
|
.otlp_log_filter
|
||||||
|
.or_else(|| defaults.telemetry.otlp_log_filter.clone()),
|
||||||
|
trace_filter: base_telemetry
|
||||||
|
.trace_filter
|
||||||
|
.or_else(|| defaults.telemetry.trace_filter.clone()),
|
||||||
|
authorization_env: base_telemetry
|
||||||
|
.authorization_env
|
||||||
|
.or_else(|| defaults.telemetry.authorization_env.clone()),
|
||||||
otlp_endpoint: args
|
otlp_endpoint: args
|
||||||
.telemetry_otlp_endpoint
|
.telemetry_otlp_endpoint
|
||||||
.clone()
|
.clone()
|
||||||
.or(base_telemetry.otlp_endpoint)
|
.or(base_telemetry.otlp_endpoint)
|
||||||
.or_else(|| defaults.telemetry.otlp_endpoint.clone()),
|
.or_else(|| defaults.telemetry.otlp_endpoint.clone()),
|
||||||
service_name: args
|
service_name: Some(
|
||||||
.telemetry_service_name
|
args.telemetry_service_name
|
||||||
.clone()
|
.clone()
|
||||||
.or(base_telemetry.service_name)
|
.or(base_telemetry.service_name)
|
||||||
.unwrap_or_else(|| defaults.telemetry.service_name.clone()),
|
.unwrap_or_else(|| defaults.telemetry.service_name.clone()),
|
||||||
json_logs: args
|
),
|
||||||
.telemetry_json_logs
|
json_logs: Some(
|
||||||
|
args.telemetry_json_logs
|
||||||
.or(base_telemetry.json_logs)
|
.or(base_telemetry.json_logs)
|
||||||
.unwrap_or(defaults.telemetry.json_logs),
|
.unwrap_or(defaults.telemetry.json_logs),
|
||||||
},
|
),
|
||||||
|
}),
|
||||||
};
|
};
|
||||||
|
|
||||||
let rendered = toml::to_string_pretty(&body).context("failed to render config template")?;
|
let rendered = toml::to_string_pretty(&body).context("failed to render config template")?;
|
||||||
@@ -217,12 +241,22 @@ public_url = "https://old.example/"
|
|||||||
state_file = "old.sqlite3"
|
state_file = "old.sqlite3"
|
||||||
command_timeout_ms = 1234
|
command_timeout_ms = 1234
|
||||||
enroll_token_ttl_seconds = 5678
|
enroll_token_ttl_seconds = 5678
|
||||||
|
device_retention = "0d"
|
||||||
pid_file = "old.pid"
|
pid_file = "old.pid"
|
||||||
ui_dist_dir = "/opt/wakey/ui"
|
ui_dist_dir = "/opt/wakey/ui"
|
||||||
bootstrap_enroll_tokens = ["old-token"]
|
bootstrap_enroll_tokens = ["old-token"]
|
||||||
|
|
||||||
|
[access]
|
||||||
|
issuer = "https://access.example"
|
||||||
|
audience = "protected-app"
|
||||||
|
jwks_url = "https://keys.example/custom"
|
||||||
|
|
||||||
[telemetry]
|
[telemetry]
|
||||||
otlp_endpoint = "http://otel.example"
|
otlp_endpoint = "http://otel.example"
|
||||||
|
authorization_env = "WAKEY_OTLP_AUTHORIZATION"
|
||||||
|
console_filter = "info"
|
||||||
|
otlp_log_filter = "debug"
|
||||||
|
trace_filter = "info"
|
||||||
service_name = "old-service"
|
service_name = "old-service"
|
||||||
json_logs = true
|
json_logs = true
|
||||||
"#,
|
"#,
|
||||||
@@ -254,13 +288,41 @@ json_logs = true
|
|||||||
let parsed: toml::Value = toml::from_str(&rendered).expect("parse derived config");
|
let parsed: toml::Value = toml::from_str(&rendered).expect("parse derived config");
|
||||||
assert_eq!(parsed["public_url"].as_str(), Some("https://new.example"));
|
assert_eq!(parsed["public_url"].as_str(), Some("https://new.example"));
|
||||||
assert_eq!(parsed["bind"].as_str(), Some("127.0.0.1:9090"));
|
assert_eq!(parsed["bind"].as_str(), Some("127.0.0.1:9090"));
|
||||||
assert_eq!(parsed["command_timeout_ms"].as_integer(), Some(999));
|
let converted: super::FileConfig = toml::from_str(&rendered).unwrap();
|
||||||
assert_eq!(parsed["enroll_token_ttl_seconds"].as_integer(), Some(5678));
|
assert_eq!(
|
||||||
|
converted.command_timeout,
|
||||||
|
Some(std::time::Duration::from_millis(999))
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
converted.enroll_token_ttl,
|
||||||
|
Some(std::time::Duration::from_secs(5678))
|
||||||
|
);
|
||||||
|
assert!(parsed.get("command_timeout_ms").is_none());
|
||||||
|
assert_eq!(converted.device_retention, Some(std::time::Duration::ZERO));
|
||||||
|
assert_eq!(
|
||||||
|
converted.access,
|
||||||
|
Some(crate::config::AccessConfig {
|
||||||
|
issuer: "https://access.example".into(),
|
||||||
|
audience: "protected-app".into(),
|
||||||
|
jwks_url: Some("https://keys.example/custom".into()),
|
||||||
|
})
|
||||||
|
);
|
||||||
|
assert!(parsed.get("enroll_token_ttl_seconds").is_none());
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
parsed["state_file"].as_str(),
|
parsed["state_file"].as_str(),
|
||||||
Some("/tmp/wakey-base/old.sqlite3")
|
Some("/tmp/wakey-base/old.sqlite3")
|
||||||
);
|
);
|
||||||
assert_eq!(parsed["telemetry"]["json_logs"].as_bool(), Some(true));
|
assert_eq!(parsed["telemetry"]["json_logs"].as_bool(), Some(true));
|
||||||
|
assert_eq!(parsed["telemetry"]["console_filter"].as_str(), Some("info"));
|
||||||
|
assert_eq!(
|
||||||
|
parsed["telemetry"]["otlp_log_filter"].as_str(),
|
||||||
|
Some("debug")
|
||||||
|
);
|
||||||
|
assert_eq!(parsed["telemetry"]["trace_filter"].as_str(), Some("info"));
|
||||||
|
assert_eq!(
|
||||||
|
parsed["telemetry"]["authorization_env"].as_str(),
|
||||||
|
Some("WAKEY_OTLP_AUTHORIZATION")
|
||||||
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
parsed["telemetry"]["service_name"].as_str(),
|
parsed["telemetry"]["service_name"].as_str(),
|
||||||
Some("old-service")
|
Some("old-service")
|
||||||
@@ -304,10 +366,9 @@ json_logs = true
|
|||||||
parsed["state_file"].as_str(),
|
parsed["state_file"].as_str(),
|
||||||
defaults.data_dir.join(&defaults.state_file).to_str()
|
defaults.data_dir.join(&defaults.state_file).to_str()
|
||||||
);
|
);
|
||||||
assert_eq!(
|
let converted: super::FileConfig = toml::from_str(&rendered).unwrap();
|
||||||
parsed["command_timeout_ms"].as_integer(),
|
assert_eq!(converted.command_timeout, Some(defaults.command_timeout));
|
||||||
Some(defaults.command_timeout_ms as i64)
|
assert_eq!(converted.device_retention, Some(defaults.device_retention));
|
||||||
);
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
parsed["telemetry"]["service_name"].as_str(),
|
parsed["telemetry"]["service_name"].as_str(),
|
||||||
Some(defaults.telemetry.service_name.as_str())
|
Some(defaults.telemetry.service_name.as_str())
|
||||||
|
|||||||
@@ -8,4 +8,4 @@ pub use resolve::{
|
|||||||
resolve_revoke_agent_settings, resolve_revoke_enroll_token_settings,
|
resolve_revoke_agent_settings, resolve_revoke_enroll_token_settings,
|
||||||
resolve_state_stats_settings,
|
resolve_state_stats_settings,
|
||||||
};
|
};
|
||||||
pub use types::{DaemonConfig, TelemetryConfig};
|
pub use types::{AccessConfig, DaemonConfig, TelemetryConfig};
|
||||||
@@ -33,10 +33,7 @@ impl DaemonConfig {
|
|||||||
args.config_file.display()
|
args.config_file.display()
|
||||||
)
|
)
|
||||||
})?,
|
})?,
|
||||||
None => defaults
|
None => defaults.bind,
|
||||||
.bind
|
|
||||||
.parse()
|
|
||||||
.expect("static default bind should parse"),
|
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -54,26 +51,24 @@ impl DaemonConfig {
|
|||||||
.unwrap_or_else(|| defaults.state_file.clone());
|
.unwrap_or_else(|| defaults.state_file.clone());
|
||||||
let state_file = resolve_path(&data_dir, state_file_raw);
|
let state_file = resolve_path(&data_dir, state_file_raw);
|
||||||
|
|
||||||
let command_timeout = Duration::from_millis(
|
let command_timeout = args
|
||||||
args.command_timeout_ms
|
.command_timeout_ms
|
||||||
.or(file.command_timeout_ms)
|
.map(Duration::from_millis)
|
||||||
.unwrap_or(defaults.command_timeout_ms)
|
.or(file.command_timeout)
|
||||||
.max(1),
|
.unwrap_or(defaults.command_timeout);
|
||||||
);
|
let enroll_token_ttl = args
|
||||||
|
.enroll_token_ttl_seconds
|
||||||
let enroll_token_ttl = Duration::from_secs(
|
.map(Duration::from_secs)
|
||||||
args.enroll_token_ttl_seconds
|
.or(file.enroll_token_ttl)
|
||||||
.or(file.enroll_token_ttl_seconds)
|
.unwrap_or(defaults.enroll_token_ttl);
|
||||||
.unwrap_or(defaults.enroll_token_ttl_seconds)
|
let history_retention = args
|
||||||
.max(1),
|
.observation_retention_seconds
|
||||||
);
|
.map(Duration::from_secs)
|
||||||
|
.or(file.history_retention)
|
||||||
let observation_retention = Duration::from_secs(
|
.unwrap_or(defaults.history_retention);
|
||||||
args.observation_retention_seconds
|
validate_durations(command_timeout, enroll_token_ttl, history_retention)?;
|
||||||
.or(file.observation_retention_seconds)
|
let device_retention = file.device_retention.unwrap_or(defaults.device_retention);
|
||||||
.unwrap_or(defaults.observation_retention_seconds)
|
validate_device_retention(device_retention)?;
|
||||||
.max(1),
|
|
||||||
);
|
|
||||||
|
|
||||||
let pid_file_raw = args
|
let pid_file_raw = args
|
||||||
.pid_file
|
.pid_file
|
||||||
@@ -104,11 +99,13 @@ impl DaemonConfig {
|
|||||||
state_file,
|
state_file,
|
||||||
command_timeout,
|
command_timeout,
|
||||||
enroll_token_ttl,
|
enroll_token_ttl,
|
||||||
observation_retention,
|
history_retention,
|
||||||
|
device_retention,
|
||||||
pid_file,
|
pid_file,
|
||||||
ui_dist_dir,
|
ui_dist_dir,
|
||||||
bootstrap_enroll_tokens,
|
bootstrap_enroll_tokens,
|
||||||
telemetry,
|
telemetry,
|
||||||
|
access: file.access,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -143,12 +140,12 @@ pub fn resolve_issue_token_settings(args: &IssueEnrollTokenArgs) -> Result<Issue
|
|||||||
&args.target,
|
&args.target,
|
||||||
)?;
|
)?;
|
||||||
|
|
||||||
let ttl = Duration::from_secs(
|
let ttl = args
|
||||||
args.ttl_seconds
|
.ttl_seconds
|
||||||
.or(file.enroll_token_ttl_seconds)
|
.map(Duration::from_secs)
|
||||||
.unwrap_or(defaults.enroll_token_ttl_seconds)
|
.or(file.enroll_token_ttl)
|
||||||
.max(1),
|
.unwrap_or(defaults.enroll_token_ttl);
|
||||||
);
|
validate_durations(defaults.command_timeout, ttl, defaults.history_retention)?;
|
||||||
|
|
||||||
Ok(IssueTokenSettings {
|
Ok(IssueTokenSettings {
|
||||||
data_dir: state.data_dir,
|
data_dir: state.data_dir,
|
||||||
@@ -209,8 +206,66 @@ pub(crate) fn load_file_config(path: &Path) -> Result<FileConfig> {
|
|||||||
|
|
||||||
let raw = std::fs::read_to_string(path)
|
let raw = std::fs::read_to_string(path)
|
||||||
.with_context(|| format!("failed to read config file {}", path.display()))?;
|
.with_context(|| format!("failed to read config file {}", path.display()))?;
|
||||||
toml::from_str::<FileConfig>(&raw)
|
let raw: toml::Value = toml::from_str(&raw)
|
||||||
.with_context(|| format!("failed to parse config file {}", path.display()))
|
.with_context(|| format!("failed to parse config file {}", path.display()))?;
|
||||||
|
let mut value = serde_json::to_value(raw)?;
|
||||||
|
let deprecated =
|
||||||
|
wakey_core::config_duration::normalize(&mut value, super::types::DURATION_KEYS)
|
||||||
|
.map_err(anyhow::Error::msg)?;
|
||||||
|
let config = serde_json::from_value(value)
|
||||||
|
.with_context(|| format!("failed to parse config file {}", path.display()))?;
|
||||||
|
for key in deprecated {
|
||||||
|
// Daemon config is loaded before telemetry initialization. Keep these
|
||||||
|
// warnings on stderr, including when init-config emits TOML on stdout.
|
||||||
|
eprintln!(
|
||||||
|
"warning: {}: `{}` is deprecated; use `{}` with a duration string",
|
||||||
|
path.display(),
|
||||||
|
key.old,
|
||||||
|
key.new
|
||||||
|
);
|
||||||
|
}
|
||||||
|
Ok(config)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Device expiry uses whole-second snapshot timestamps, not an active timer.
|
||||||
|
/// Unlike history retention, zero is a supported opt-out.
|
||||||
|
pub(crate) fn validate_device_retention(retention: Duration) -> Result<()> {
|
||||||
|
anyhow::ensure!(
|
||||||
|
retention.subsec_nanos() == 0,
|
||||||
|
"device_retention must use whole seconds"
|
||||||
|
);
|
||||||
|
anyhow::ensure!(
|
||||||
|
i64::try_from(retention.as_secs()).is_ok(),
|
||||||
|
"device_retention is too large"
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn validate_durations(
|
||||||
|
command_timeout: Duration,
|
||||||
|
enroll_token_ttl: Duration,
|
||||||
|
history_retention: Duration,
|
||||||
|
) -> Result<()> {
|
||||||
|
for (name, value) in [
|
||||||
|
("command_timeout", command_timeout),
|
||||||
|
("enroll_token_ttl", enroll_token_ttl),
|
||||||
|
("history_retention", history_retention),
|
||||||
|
] {
|
||||||
|
anyhow::ensure!(!value.is_zero(), "{name} must be positive");
|
||||||
|
anyhow::ensure!(
|
||||||
|
std::time::Instant::now().checked_add(value).is_some(),
|
||||||
|
"{name} exceeds the platform timer range"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
anyhow::ensure!(
|
||||||
|
enroll_token_ttl.subsec_nanos() == 0,
|
||||||
|
"enroll_token_ttl must use whole seconds"
|
||||||
|
);
|
||||||
|
anyhow::ensure!(
|
||||||
|
history_retention.subsec_nanos() == 0,
|
||||||
|
"history_retention must use whole seconds"
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn resolve_admin_public_url(
|
fn resolve_admin_public_url(
|
||||||
@@ -229,8 +284,13 @@ fn resolve_admin_public_url(
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn resolve_telemetry(file: Option<FileTelemetryConfig>) -> TelemetryConfig {
|
fn resolve_telemetry(file: Option<FileTelemetryConfig>) -> TelemetryConfig {
|
||||||
let mut out = TelemetryConfig::default();
|
// Service-specific defaults live in the CC template, not the shared type.
|
||||||
|
let mut out = DEFAULT_CONFIG.telemetry.clone();
|
||||||
if let Some(file) = file {
|
if let Some(file) = file {
|
||||||
|
out.authorization_env = file.authorization_env;
|
||||||
|
out.console_filter = file.console_filter;
|
||||||
|
out.otlp_log_filter = file.otlp_log_filter;
|
||||||
|
out.trace_filter = file.trace_filter;
|
||||||
if let Some(endpoint) = file.otlp_endpoint {
|
if let Some(endpoint) = file.otlp_endpoint {
|
||||||
let trimmed = endpoint.trim().to_string();
|
let trimmed = endpoint.trim().to_string();
|
||||||
if !trimmed.is_empty() {
|
if !trimmed.is_empty() {
|
||||||
@@ -285,11 +345,12 @@ fn resolve_state_access(
|
|||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use super::{DaemonConfig, normalize_public_url, resolve_path};
|
use super::{DaemonConfig, normalize_public_url, resolve_path, validate_device_retention};
|
||||||
use crate::cli::ServeArgs;
|
use crate::cli::ServeArgs;
|
||||||
use crate::config::types::DEFAULT_CONFIG;
|
use crate::config::types::DEFAULT_CONFIG;
|
||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
|
use std::time::Duration;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn normalize_public_url_trims_trailing_slash() {
|
fn normalize_public_url_trims_trailing_slash() {
|
||||||
@@ -340,13 +401,57 @@ mod tests {
|
|||||||
resolved.state_file,
|
resolved.state_file,
|
||||||
defaults.data_dir.join(&defaults.state_file)
|
defaults.data_dir.join(&defaults.state_file)
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(resolved.command_timeout, defaults.command_timeout);
|
||||||
resolved.command_timeout,
|
assert_eq!(resolved.device_retention, defaults.device_retention);
|
||||||
std::time::Duration::from_millis(defaults.command_timeout_ms)
|
|
||||||
);
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
resolved.telemetry.service_name,
|
resolved.telemetry.service_name,
|
||||||
defaults.telemetry.service_name
|
defaults.telemetry.service_name
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn device_retention_accepts_disabled_and_whole_seconds_only() {
|
||||||
|
assert!(validate_device_retention(Duration::ZERO).is_ok());
|
||||||
|
assert!(validate_device_retention(DEFAULT_CONFIG.device_retention).is_ok());
|
||||||
|
assert!(validate_device_retention(Duration::from_millis(1_500)).is_err());
|
||||||
|
assert!(validate_device_retention(Duration::from_secs(u64::MAX)).is_err());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn invalid_duration_policies_are_rejected_without_clamping() {
|
||||||
|
use std::time::Duration;
|
||||||
|
let defaults = &*DEFAULT_CONFIG;
|
||||||
|
assert!(
|
||||||
|
super::validate_durations(
|
||||||
|
Duration::ZERO,
|
||||||
|
defaults.enroll_token_ttl,
|
||||||
|
defaults.history_retention
|
||||||
|
)
|
||||||
|
.is_err()
|
||||||
|
);
|
||||||
|
assert!(
|
||||||
|
super::validate_durations(
|
||||||
|
defaults.command_timeout,
|
||||||
|
Duration::from_millis(500),
|
||||||
|
defaults.history_retention
|
||||||
|
)
|
||||||
|
.is_err()
|
||||||
|
);
|
||||||
|
assert!(
|
||||||
|
super::validate_durations(
|
||||||
|
defaults.command_timeout,
|
||||||
|
defaults.enroll_token_ttl,
|
||||||
|
Duration::ZERO
|
||||||
|
)
|
||||||
|
.is_err()
|
||||||
|
);
|
||||||
|
assert!(
|
||||||
|
super::validate_durations(
|
||||||
|
Duration::from_secs(u64::MAX),
|
||||||
|
defaults.enroll_token_ttl,
|
||||||
|
defaults.history_retention
|
||||||
|
)
|
||||||
|
.is_err()
|
||||||
|
);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -4,6 +4,8 @@ use std::sync::LazyLock;
|
|||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
|
use wakey_core::config_duration::{self, LegacyDuration};
|
||||||
|
pub use wakey_core::telemetry::TelemetryConfig;
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct DaemonConfig {
|
pub struct DaemonConfig {
|
||||||
@@ -13,92 +15,113 @@ pub struct DaemonConfig {
|
|||||||
pub state_file: PathBuf,
|
pub state_file: PathBuf,
|
||||||
pub command_timeout: Duration,
|
pub command_timeout: Duration,
|
||||||
pub enroll_token_ttl: Duration,
|
pub enroll_token_ttl: Duration,
|
||||||
#[allow(dead_code)]
|
pub history_retention: Duration,
|
||||||
pub observation_retention: Duration,
|
/// Maximum age since an agent snapshot included a device; zero disables expiry.
|
||||||
|
pub device_retention: Duration,
|
||||||
pub pid_file: PathBuf,
|
pub pid_file: PathBuf,
|
||||||
pub ui_dist_dir: PathBuf,
|
pub ui_dist_dir: PathBuf,
|
||||||
pub bootstrap_enroll_tokens: Vec<String>,
|
pub bootstrap_enroll_tokens: Vec<String>,
|
||||||
pub telemetry: TelemetryConfig,
|
pub telemetry: TelemetryConfig,
|
||||||
|
pub access: Option<AccessConfig>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
/// Trusted origin-verifier settings; no URL is taken from a JWT header or claim.
|
||||||
pub struct TelemetryConfig {
|
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
|
||||||
pub otlp_endpoint: Option<String>,
|
#[serde(deny_unknown_fields)]
|
||||||
pub service_name: String,
|
pub struct AccessConfig {
|
||||||
pub json_logs: bool,
|
pub issuer: String,
|
||||||
|
pub audience: String,
|
||||||
|
/// Omission uses the Cloudflare Access certs path beneath the issuer.
|
||||||
|
pub jwks_url: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for TelemetryConfig {
|
/// TOML DTO for both loading partial files and generating complete files.
|
||||||
fn default() -> Self {
|
/// Options preserve omission for precedence; runtime conversion stays separate.
|
||||||
let defaults = &DEFAULT_CONFIG.telemetry;
|
#[derive(Debug, Clone, Serialize, Deserialize, Default, PartialEq, Eq)]
|
||||||
Self {
|
#[serde(remote = "Self")]
|
||||||
otlp_endpoint: defaults.otlp_endpoint.clone(),
|
|
||||||
service_name: defaults.service_name.clone(),
|
|
||||||
json_logs: defaults.json_logs,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Deserialize, Default)]
|
|
||||||
pub(crate) struct FileConfig {
|
pub(crate) struct FileConfig {
|
||||||
pub(crate) data_dir: Option<PathBuf>,
|
pub(crate) data_dir: Option<PathBuf>,
|
||||||
pub(crate) bind: Option<String>,
|
pub(crate) bind: Option<String>,
|
||||||
pub(crate) public_url: Option<String>,
|
pub(crate) public_url: Option<String>,
|
||||||
pub(crate) state_file: Option<PathBuf>,
|
pub(crate) state_file: Option<PathBuf>,
|
||||||
pub(crate) command_timeout_ms: Option<u64>,
|
#[serde(default, with = "humantime_serde::option")]
|
||||||
pub(crate) enroll_token_ttl_seconds: Option<u64>,
|
pub(crate) command_timeout: Option<Duration>,
|
||||||
pub(crate) observation_retention_seconds: Option<u64>,
|
#[serde(default, with = "humantime_serde::option")]
|
||||||
|
pub(crate) enroll_token_ttl: Option<Duration>,
|
||||||
|
#[serde(default, with = "humantime_serde::option")]
|
||||||
|
pub(crate) history_retention: Option<Duration>,
|
||||||
|
#[serde(default, with = "humantime_serde::option")]
|
||||||
|
pub(crate) device_retention: Option<Duration>,
|
||||||
pub(crate) pid_file: Option<PathBuf>,
|
pub(crate) pid_file: Option<PathBuf>,
|
||||||
pub(crate) ui_dist_dir: Option<PathBuf>,
|
pub(crate) ui_dist_dir: Option<PathBuf>,
|
||||||
#[serde(alias = "enroll_tokens")]
|
#[serde(alias = "enroll_tokens")]
|
||||||
pub(crate) bootstrap_enroll_tokens: Option<Vec<String>>,
|
pub(crate) bootstrap_enroll_tokens: Option<Vec<String>>,
|
||||||
pub(crate) telemetry: Option<FileTelemetryConfig>,
|
pub(crate) telemetry: Option<FileTelemetryConfig>,
|
||||||
|
pub(crate) access: Option<AccessConfig>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Deserialize, Default)]
|
pub(crate) const DURATION_KEYS: &[LegacyDuration] = &[
|
||||||
|
LegacyDuration {
|
||||||
|
old: "command_timeout_ms",
|
||||||
|
new: "command_timeout",
|
||||||
|
unit: Duration::from_millis(1),
|
||||||
|
},
|
||||||
|
LegacyDuration {
|
||||||
|
old: "enroll_token_ttl_seconds",
|
||||||
|
new: "enroll_token_ttl",
|
||||||
|
unit: Duration::from_secs(1),
|
||||||
|
},
|
||||||
|
LegacyDuration {
|
||||||
|
old: "observation_retention_seconds",
|
||||||
|
new: "history_retention",
|
||||||
|
unit: Duration::from_secs(1),
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
impl Serialize for FileConfig {
|
||||||
|
fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
|
||||||
|
Self::serialize(self, serializer)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
impl<'de> Deserialize<'de> for FileConfig {
|
||||||
|
fn deserialize<D: serde::Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
|
||||||
|
config_duration::deserialize_legacy(deserializer, DURATION_KEYS, Self::deserialize)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Serialize, Deserialize, Default, PartialEq, Eq)]
|
||||||
pub(crate) struct FileTelemetryConfig {
|
pub(crate) struct FileTelemetryConfig {
|
||||||
|
// Missing fields must remain distinguishable from explicit values when
|
||||||
|
// init-config merges an existing file with CLI overrides and defaults.
|
||||||
|
pub(crate) console_filter: Option<String>,
|
||||||
|
pub(crate) otlp_log_filter: Option<String>,
|
||||||
|
pub(crate) trace_filter: Option<String>,
|
||||||
pub(crate) otlp_endpoint: Option<String>,
|
pub(crate) otlp_endpoint: Option<String>,
|
||||||
|
pub(crate) authorization_env: Option<String>,
|
||||||
pub(crate) service_name: Option<String>,
|
pub(crate) service_name: Option<String>,
|
||||||
pub(crate) json_logs: Option<bool>,
|
pub(crate) json_logs: Option<bool>,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize)]
|
// Relative template paths are resolved under data_dir by the runtime loader.
|
||||||
pub(crate) struct WritableConfig {
|
pub(crate) static DEFAULT_CONFIG: LazyLock<DaemonConfig> = LazyLock::new(|| DaemonConfig {
|
||||||
pub(crate) data_dir: PathBuf,
|
|
||||||
pub(crate) bind: String,
|
|
||||||
pub(crate) public_url: String,
|
|
||||||
pub(crate) state_file: PathBuf,
|
|
||||||
pub(crate) command_timeout_ms: u64,
|
|
||||||
pub(crate) enroll_token_ttl_seconds: u64,
|
|
||||||
pub(crate) observation_retention_seconds: u64,
|
|
||||||
pub(crate) pid_file: PathBuf,
|
|
||||||
pub(crate) ui_dist_dir: PathBuf,
|
|
||||||
#[serde(skip_serializing_if = "Vec::is_empty")]
|
|
||||||
pub(crate) bootstrap_enroll_tokens: Vec<String>,
|
|
||||||
pub(crate) telemetry: WritableTelemetry,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize)]
|
|
||||||
pub(crate) struct WritableTelemetry {
|
|
||||||
#[serde(skip_serializing_if = "Option::is_none")]
|
|
||||||
pub(crate) otlp_endpoint: Option<String>,
|
|
||||||
pub(crate) service_name: String,
|
|
||||||
pub(crate) json_logs: bool,
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) static DEFAULT_CONFIG: LazyLock<WritableConfig> = LazyLock::new(|| WritableConfig {
|
|
||||||
data_dir: crate::cli::DEFAULT_DATA_DIR.into(),
|
data_dir: crate::cli::DEFAULT_DATA_DIR.into(),
|
||||||
bind: "0.0.0.0:8080".to_string(),
|
bind: SocketAddr::from(([0, 0, 0, 0], 8080)),
|
||||||
public_url: "http://127.0.0.1:8080".to_string(),
|
public_url: "http://127.0.0.1:8080".to_string(),
|
||||||
state_file: "state.sqlite3".into(),
|
state_file: "state.sqlite3".into(),
|
||||||
command_timeout_ms: 30_000,
|
command_timeout: Duration::from_millis(30_000),
|
||||||
enroll_token_ttl_seconds: 86_400,
|
enroll_token_ttl: Duration::from_secs(86_400),
|
||||||
observation_retention_seconds: 2_592_000,
|
history_retention: Duration::from_secs(2_592_000),
|
||||||
|
device_retention: Duration::from_secs(60 * 86_400),
|
||||||
pid_file: "wakey-control-plane.pid".into(),
|
pid_file: "wakey-control-plane.pid".into(),
|
||||||
ui_dist_dir: "ui/dist".into(),
|
ui_dist_dir: "ui/dist".into(),
|
||||||
bootstrap_enroll_tokens: Vec::new(),
|
bootstrap_enroll_tokens: Vec::new(),
|
||||||
telemetry: WritableTelemetry {
|
access: None,
|
||||||
|
telemetry: TelemetryConfig {
|
||||||
|
console_filter: None,
|
||||||
|
otlp_log_filter: None,
|
||||||
|
trace_filter: None,
|
||||||
otlp_endpoint: None,
|
otlp_endpoint: None,
|
||||||
|
authorization_env: None,
|
||||||
service_name: "wakey-control-plane".to_string(),
|
service_name: "wakey-control-plane".to_string(),
|
||||||
json_logs: false,
|
json_logs: false,
|
||||||
},
|
},
|
||||||
@@ -116,3 +139,85 @@ pub struct StateAccessSettings {
|
|||||||
pub state_file: PathBuf,
|
pub state_file: PathBuf,
|
||||||
pub public_url: Option<String>,
|
pub public_url: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::{FileConfig, FileTelemetryConfig};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn file_dto_round_trip_preserves_omission_and_explicit_values() {
|
||||||
|
let omitted: FileConfig = toml::from_str("").unwrap();
|
||||||
|
assert_eq!(omitted, FileConfig::default());
|
||||||
|
assert_eq!(toml::to_string(&omitted).unwrap(), "");
|
||||||
|
|
||||||
|
let explicit = FileConfig {
|
||||||
|
bootstrap_enroll_tokens: Some(Vec::new()),
|
||||||
|
history_retention: Some(std::time::Duration::ZERO),
|
||||||
|
telemetry: Some(FileTelemetryConfig {
|
||||||
|
json_logs: Some(false),
|
||||||
|
console_filter: Some("info".into()),
|
||||||
|
otlp_log_filter: Some("debug".into()),
|
||||||
|
trace_filter: Some("info".into()),
|
||||||
|
..Default::default()
|
||||||
|
}),
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
let rendered = toml::to_string_pretty(&explicit).unwrap();
|
||||||
|
let restored: FileConfig = toml::from_str(&rendered).unwrap();
|
||||||
|
assert_eq!(restored, explicit);
|
||||||
|
assert_ne!(restored, omitted);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn legacy_token_alias_serializes_with_canonical_name() {
|
||||||
|
let file: FileConfig = toml::from_str("enroll_tokens = []").unwrap();
|
||||||
|
assert_eq!(file.bootstrap_enroll_tokens, Some(Vec::new()));
|
||||||
|
let rendered = toml::to_string(&file).unwrap();
|
||||||
|
assert!(rendered.contains("bootstrap_enroll_tokens = []"));
|
||||||
|
assert!(!rendered.starts_with("enroll_tokens"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn legacy_durations_match_new_keys_and_emit_only_strings() {
|
||||||
|
let legacy: FileConfig = toml::from_str("command_timeout_ms = 1234\nenroll_token_ttl_seconds = 3600\nobservation_retention_seconds = 2592000").unwrap();
|
||||||
|
let canonical: FileConfig = toml::from_str(
|
||||||
|
"command_timeout = '1234ms'\nenroll_token_ttl = '1h'\nhistory_retention = '30d'",
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
assert_eq!(legacy, canonical);
|
||||||
|
let output = toml::to_string(&legacy).unwrap();
|
||||||
|
for key in super::DURATION_KEYS {
|
||||||
|
assert!(!output.contains(key.old));
|
||||||
|
assert!(output.contains(key.new));
|
||||||
|
}
|
||||||
|
assert_eq!(toml::from_str::<FileConfig>(&output).unwrap(), canonical);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn file_durations_reject_duplicates_and_bare_numbers() {
|
||||||
|
for key in super::DURATION_KEYS {
|
||||||
|
let both = format!("{} = '1s'\n{} = 1", key.new, key.old);
|
||||||
|
assert!(toml::from_str::<FileConfig>(&both).is_err());
|
||||||
|
let number = format!("{} = 1", key.new);
|
||||||
|
assert!(toml::from_str::<FileConfig>(&number).is_err());
|
||||||
|
let negative = format!("{} = -1", key.old);
|
||||||
|
assert!(toml::from_str::<FileConfig>(&negative).is_err());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn device_retention_round_trips_and_requires_units() {
|
||||||
|
for text in ["60d", "0d"] {
|
||||||
|
let file: FileConfig = toml::from_str(&format!("device_retention = '{text}'")).unwrap();
|
||||||
|
let output = toml::to_string(&file).unwrap();
|
||||||
|
assert_eq!(toml::from_str::<FileConfig>(&output).unwrap(), file);
|
||||||
|
}
|
||||||
|
for input in [
|
||||||
|
"device_retention = 60",
|
||||||
|
"device_retention = 1.5",
|
||||||
|
"device_retention = '-1d'",
|
||||||
|
] {
|
||||||
|
assert!(toml::from_str::<FileConfig>(input).is_err());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -28,8 +28,10 @@ async fn main() -> Result<()> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
let daemon = config::DaemonConfig::from_serve_args(&args)?;
|
let daemon = config::DaemonConfig::from_serve_args(&args)?;
|
||||||
tracing::init(cli.verbose, &daemon.telemetry)?;
|
let telemetry = tracing::init(cli.verbose, &daemon.telemetry)?;
|
||||||
runtime::serve(daemon).await
|
let result = runtime::serve(daemon).await;
|
||||||
|
tokio::task::spawn_blocking(move || telemetry.shutdown()).await?;
|
||||||
|
result
|
||||||
}
|
}
|
||||||
Command::InitConfig(args) => {
|
Command::InitConfig(args) => {
|
||||||
if let Some(path) = config::write_init_config(&args)? {
|
if let Some(path) = config::write_init_config(&args)? {
|
||||||
|
|||||||
@@ -0,0 +1,337 @@
|
|||||||
|
//! Origin authentication oracle for Caddy, not middleware on the control API.
|
||||||
|
//! Enforcement belongs to the proxy; the daemon must remain loopback-only.
|
||||||
|
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::net::SocketAddr;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use std::time::{Duration, SystemTime};
|
||||||
|
|
||||||
|
use anyhow::{Context, Result, ensure};
|
||||||
|
use axum::{
|
||||||
|
Extension, Router,
|
||||||
|
http::{HeaderMap, StatusCode},
|
||||||
|
routing::get,
|
||||||
|
};
|
||||||
|
use http_cache_semantics::{AfterResponse, BeforeRequest, CacheOptions, CachePolicy};
|
||||||
|
use jsonwebtoken::jwk::{AlgorithmParameters, JwkSet, KeyAlgorithm, KeyOperations, PublicKeyUse};
|
||||||
|
use jsonwebtoken::{Algorithm, DecodingKey, Validation, decode, decode_header};
|
||||||
|
use reqwest::{Client, Url};
|
||||||
|
use tokio::sync::Mutex;
|
||||||
|
use tokio::time::Instant;
|
||||||
|
|
||||||
|
use super::AppState;
|
||||||
|
use crate::config::AccessConfig;
|
||||||
|
|
||||||
|
const REFRESH_COOLDOWN: Duration = Duration::from_secs(60);
|
||||||
|
const FETCH_TIMEOUT: Duration = Duration::from_secs(5);
|
||||||
|
const MAX_JWKS_BYTES: usize = 256 * 1024;
|
||||||
|
const MAX_TOKEN_BYTES: usize = 16 * 1024;
|
||||||
|
const MAX_KEYS: usize = 64;
|
||||||
|
|
||||||
|
// This is a private, single-resource cache. Do not invent a freshness lifetime
|
||||||
|
// when the issuer supplies none, or override its no-cache/no-store directives.
|
||||||
|
const CACHE_OPTIONS: CacheOptions = CacheOptions {
|
||||||
|
shared: false,
|
||||||
|
cache_heuristic: 0.0,
|
||||||
|
immutable_min_time_to_live: Duration::ZERO,
|
||||||
|
ignore_cargo_cult: false,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// Absent configuration deliberately leaves the oracle denying every request.
|
||||||
|
pub(super) fn routes(config: Option<&AccessConfig>, bind: SocketAddr) -> Result<Router<AppState>> {
|
||||||
|
let verifier = config
|
||||||
|
.map(|config| {
|
||||||
|
ensure!(
|
||||||
|
bind.ip().is_loopback(),
|
||||||
|
"Access verification requires a loopback bind; Caddy owns enforcement"
|
||||||
|
);
|
||||||
|
Verifier::new(config).map(Arc::new)
|
||||||
|
})
|
||||||
|
.transpose()?;
|
||||||
|
Ok(verifier_routes(verifier))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn verifier_routes<S: Clone + Send + Sync + 'static>(verifier: Option<Arc<Verifier>>) -> Router<S> {
|
||||||
|
Router::new()
|
||||||
|
.route("/internal/auth/verify", get(verify))
|
||||||
|
.layer(Extension(verifier))
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn verify(
|
||||||
|
Extension(verifier): Extension<Option<Arc<Verifier>>>,
|
||||||
|
headers: HeaderMap,
|
||||||
|
) -> StatusCode {
|
||||||
|
let Some(verifier) = verifier else {
|
||||||
|
return StatusCode::UNAUTHORIZED;
|
||||||
|
};
|
||||||
|
let mut values = headers.get_all("cf-access-jwt-assertion").iter();
|
||||||
|
let Some(token) = values.next().and_then(|value| value.to_str().ok()) else {
|
||||||
|
return StatusCode::UNAUTHORIZED;
|
||||||
|
};
|
||||||
|
if values.next().is_some() {
|
||||||
|
return StatusCode::UNAUTHORIZED;
|
||||||
|
}
|
||||||
|
match verifier.validate(token).await {
|
||||||
|
Ok(()) => StatusCode::NO_CONTENT,
|
||||||
|
Err(VerifyError::Invalid) => StatusCode::UNAUTHORIZED,
|
||||||
|
Err(VerifyError::Unavailable) => StatusCode::SERVICE_UNAVAILABLE,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Verifier {
|
||||||
|
client: Client,
|
||||||
|
jwks_url: Url,
|
||||||
|
validation: Validation,
|
||||||
|
cache: Mutex<KeyCache>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Default)]
|
||||||
|
struct KeyCache {
|
||||||
|
keys: HashMap<String, DecodingKey>,
|
||||||
|
policy: Option<CachePolicy>,
|
||||||
|
attempted_at: Option<Instant>,
|
||||||
|
last_fetch_failed: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, thiserror::Error, PartialEq, Eq)]
|
||||||
|
enum VerifyError {
|
||||||
|
#[error("invalid Access assertion")]
|
||||||
|
Invalid,
|
||||||
|
#[error("Access signing keys unavailable")]
|
||||||
|
Unavailable,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Verifier {
|
||||||
|
fn new(config: &AccessConfig) -> Result<Self> {
|
||||||
|
let issuer = trusted_url(&config.issuer)?;
|
||||||
|
ensure!(
|
||||||
|
issuer.path() == "/" && !config.issuer.ends_with('/'),
|
||||||
|
"Access issuer must be an HTTPS origin without a trailing slash"
|
||||||
|
);
|
||||||
|
ensure!(
|
||||||
|
!config.audience.trim().is_empty(),
|
||||||
|
"Access audience must not be empty"
|
||||||
|
);
|
||||||
|
let jwks_url = match &config.jwks_url {
|
||||||
|
Some(url) => trusted_url(url)?,
|
||||||
|
None => issuer.join("/cdn-cgi/access/certs")?,
|
||||||
|
};
|
||||||
|
let mut validation = Validation::new(Algorithm::RS256);
|
||||||
|
validation.set_issuer(&[&config.issuer]);
|
||||||
|
validation.set_audience(&[&config.audience]);
|
||||||
|
validation.set_required_spec_claims(&["exp", "iss", "aud"]);
|
||||||
|
validation.validate_nbf = true;
|
||||||
|
validation.leeway = 0;
|
||||||
|
Ok(Self {
|
||||||
|
client: Client::builder()
|
||||||
|
.timeout(FETCH_TIMEOUT)
|
||||||
|
.redirect(reqwest::redirect::Policy::none())
|
||||||
|
.build()?,
|
||||||
|
jwks_url,
|
||||||
|
validation,
|
||||||
|
cache: Mutex::new(KeyCache::default()),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn validate(&self, token: &str) -> Result<(), VerifyError> {
|
||||||
|
if token.len() > MAX_TOKEN_BYTES {
|
||||||
|
return Err(VerifyError::Invalid);
|
||||||
|
}
|
||||||
|
// Only kid selects a cached key. Token-supplied URLs never drive I/O.
|
||||||
|
let header = decode_header(token).map_err(|_| VerifyError::Invalid)?;
|
||||||
|
if header.alg != Algorithm::RS256 {
|
||||||
|
return Err(VerifyError::Invalid);
|
||||||
|
}
|
||||||
|
let kid = header
|
||||||
|
.kid
|
||||||
|
.filter(|kid| !kid.is_empty() && kid.len() <= 256)
|
||||||
|
.ok_or(VerifyError::Invalid)?;
|
||||||
|
let key = self.key(&kid).await?;
|
||||||
|
decode::<serde_json::Value>(token, &key, &self.validation)
|
||||||
|
.map(|_| ())
|
||||||
|
.map_err(|_| VerifyError::Invalid)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn key(&self, kid: &str) -> Result<DecodingKey, VerifyError> {
|
||||||
|
// Single-flight refresh with a bounded HTTP fetch. The cooldown is
|
||||||
|
// global, not an attacker-controlled per-kid cache, including outages.
|
||||||
|
let mut cache = self.cache.lock().await;
|
||||||
|
let mut request = self
|
||||||
|
.client
|
||||||
|
.get(self.jwks_url.clone())
|
||||||
|
.build()
|
||||||
|
.map_err(|_| VerifyError::Unavailable)?;
|
||||||
|
let now = Instant::now();
|
||||||
|
let fresh = cache.policy.as_ref().is_some_and(|policy| {
|
||||||
|
matches!(
|
||||||
|
policy.before_request(&request, SystemTime::now()),
|
||||||
|
BeforeRequest::Fresh(_)
|
||||||
|
)
|
||||||
|
});
|
||||||
|
if fresh && let Some(key) = cache.keys.get(kid) {
|
||||||
|
return Ok(key.clone());
|
||||||
|
}
|
||||||
|
if cache
|
||||||
|
.attempted_at
|
||||||
|
.is_none_or(|at| now.duration_since(at) >= REFRESH_COOLDOWN)
|
||||||
|
{
|
||||||
|
cache.attempted_at = Some(now);
|
||||||
|
// A missing kid must bypass a still-fresh response cache to notice
|
||||||
|
// rotation. The HTTP library adds ETag/Last-Modified validators.
|
||||||
|
request.headers_mut().insert(
|
||||||
|
reqwest::header::CACHE_CONTROL,
|
||||||
|
reqwest::header::HeaderValue::from_static("no-cache"),
|
||||||
|
);
|
||||||
|
if let Some(policy) = &cache.policy
|
||||||
|
&& let BeforeRequest::Stale {
|
||||||
|
request: conditional,
|
||||||
|
..
|
||||||
|
} = policy.before_request(&request, SystemTime::now())
|
||||||
|
{
|
||||||
|
*request.headers_mut() = conditional.headers;
|
||||||
|
}
|
||||||
|
match self.fetch_keys(&request, &cache).await {
|
||||||
|
Ok((keys, policy)) => {
|
||||||
|
let selected = keys.get(kid).cloned().ok_or(VerifyError::Invalid);
|
||||||
|
if policy.is_storable() {
|
||||||
|
cache.keys = keys;
|
||||||
|
cache.policy = Some(policy);
|
||||||
|
} else {
|
||||||
|
// no-store permits this response's use for this request,
|
||||||
|
// but neither keys nor HTTP metadata may be retained.
|
||||||
|
cache.keys.clear();
|
||||||
|
cache.policy = None;
|
||||||
|
}
|
||||||
|
cache.last_fetch_failed = false;
|
||||||
|
return selected;
|
||||||
|
}
|
||||||
|
Err(_) => {
|
||||||
|
// Never include upstream bodies, tokens or claims in logs.
|
||||||
|
tracing::warn!(
|
||||||
|
"Access signing-key refresh failed; verification fails closed when no fresh matching key is available"
|
||||||
|
);
|
||||||
|
cache.last_fetch_failed = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !fresh || cache.last_fetch_failed {
|
||||||
|
Err(VerifyError::Unavailable)
|
||||||
|
} else {
|
||||||
|
Err(VerifyError::Invalid)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn fetch_keys(
|
||||||
|
&self,
|
||||||
|
request: &reqwest::Request,
|
||||||
|
cached: &KeyCache,
|
||||||
|
) -> Result<(HashMap<String, DecodingKey>, CachePolicy)> {
|
||||||
|
let mut response = self
|
||||||
|
.client
|
||||||
|
.execute(
|
||||||
|
request
|
||||||
|
.try_clone()
|
||||||
|
.context("JWKS request cannot be cloned")?,
|
||||||
|
)
|
||||||
|
.await?
|
||||||
|
.error_for_status()?;
|
||||||
|
let received_at = SystemTime::now();
|
||||||
|
if response.status() == StatusCode::NOT_MODIFIED {
|
||||||
|
ensure!(
|
||||||
|
request
|
||||||
|
.headers()
|
||||||
|
.contains_key(reqwest::header::IF_NONE_MATCH)
|
||||||
|
|| request
|
||||||
|
.headers()
|
||||||
|
.contains_key(reqwest::header::IF_MODIFIED_SINCE),
|
||||||
|
"unsolicited JWKS 304"
|
||||||
|
);
|
||||||
|
let policy = cached
|
||||||
|
.policy
|
||||||
|
.as_ref()
|
||||||
|
.context("JWKS 304 without stored keys")?;
|
||||||
|
return match policy.after_response(request, &response, received_at) {
|
||||||
|
AfterResponse::NotModified(policy, _) => Ok((cached.keys.clone(), policy)),
|
||||||
|
AfterResponse::Modified(_, _) => {
|
||||||
|
anyhow::bail!("JWKS 304 did not match stored validators")
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
ensure!(
|
||||||
|
response.status() == StatusCode::OK,
|
||||||
|
"JWKS must return 200 or a matching 304"
|
||||||
|
);
|
||||||
|
let policy = CachePolicy::new_options(request, &response, received_at, CACHE_OPTIONS);
|
||||||
|
ensure!(
|
||||||
|
response
|
||||||
|
.content_length()
|
||||||
|
.is_none_or(|len| len <= MAX_JWKS_BYTES as u64),
|
||||||
|
"JWKS too large"
|
||||||
|
);
|
||||||
|
let mut body = Vec::new();
|
||||||
|
while let Some(chunk) = response.chunk().await? {
|
||||||
|
ensure!(chunk.len() <= MAX_JWKS_BYTES - body.len(), "JWKS too large");
|
||||||
|
body.extend_from_slice(&chunk);
|
||||||
|
}
|
||||||
|
Ok((parse_keys(&body)?, policy))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn trusted_url(value: &str) -> Result<Url> {
|
||||||
|
let url = Url::parse(value).context("invalid Access URL")?;
|
||||||
|
ensure!(
|
||||||
|
url.scheme() == "https"
|
||||||
|
&& url.host_str().is_some()
|
||||||
|
&& url.username().is_empty()
|
||||||
|
&& url.password().is_none()
|
||||||
|
&& url.query().is_none()
|
||||||
|
&& url.fragment().is_none(),
|
||||||
|
"Access URLs require HTTPS without credentials, query or fragment"
|
||||||
|
);
|
||||||
|
Ok(url)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn parse_keys(body: &[u8]) -> Result<HashMap<String, DecodingKey>> {
|
||||||
|
let jwks: JwkSet = serde_json::from_slice(body)?;
|
||||||
|
ensure!(jwks.keys.len() <= MAX_KEYS, "too many signing keys");
|
||||||
|
let mut keys = HashMap::new();
|
||||||
|
for jwk in jwks.keys {
|
||||||
|
if !matches!(jwk.algorithm, AlgorithmParameters::RSA(_))
|
||||||
|
|| jwk
|
||||||
|
.common
|
||||||
|
.key_algorithm
|
||||||
|
.is_some_and(|alg| alg != KeyAlgorithm::RS256)
|
||||||
|
|| jwk
|
||||||
|
.common
|
||||||
|
.public_key_use
|
||||||
|
.as_ref()
|
||||||
|
.is_some_and(|usage| *usage != PublicKeyUse::Signature)
|
||||||
|
|| jwk
|
||||||
|
.common
|
||||||
|
.key_operations
|
||||||
|
.as_ref()
|
||||||
|
.is_some_and(|ops| !ops.contains(&KeyOperations::Verify))
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
let Some(kid) = jwk
|
||||||
|
.common
|
||||||
|
.key_id
|
||||||
|
.as_ref()
|
||||||
|
.filter(|kid| !kid.is_empty() && kid.len() <= 256)
|
||||||
|
else {
|
||||||
|
continue;
|
||||||
|
};
|
||||||
|
let key = DecodingKey::from_jwk(&jwk)?;
|
||||||
|
ensure!(
|
||||||
|
keys.insert(kid.clone(), key).is_none(),
|
||||||
|
"duplicate signing key ID"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
// A valid empty set revokes every cached key. Do not treat it as a fetch
|
||||||
|
// failure and preserve keys that the issuer has deliberately removed.
|
||||||
|
Ok(keys)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests;
|
||||||
Loaded 100 of 131 files, more files were not shown because too many files have changed in this diff.
Show more
Reference in new issue
Block a user