dig_node_control_interface/traits.rs
1//! The two contract traits: the client-facing call builder/parser and the node-facing handler.
2//!
3//! * [`ControlCall`] binds a typed params struct to its [`ControlMethod`] and its typed result — so
4//! a caller writes `client.request(&SetCapParams { cap_bytes })` and gets back a `SetCapResult`,
5//! never a stringly-typed `Value`.
6//! * [`ControlClient`] is what a CLIENT depends on: build a JSON-RPC request from a typed call, and
7//! parse a response back into the typed result (or a [`ControlError`]). Pure — no transport; the
8//! consumer carries the bytes over dig-ipc / loopback-mTLS itself.
9//! * [`ControlHandler`] is what a NODE implements to SERVE the surface: one typed method per control
10//! method, plus a provided [`dispatch`](ControlHandler::dispatch) that routes a raw request to the
11//! right method — the single anti-drift seam the conformance KATs exercise.
12
13use async_trait::async_trait;
14use serde::de::DeserializeOwned;
15use serde::Serialize;
16use serde_json::Value;
17
18use crate::envelope::{JsonRpcRequest, JsonRpcResponse, RequestId};
19use crate::error::{ControlError, ControlErrorCode};
20use crate::method::ControlMethod;
21use crate::params;
22use crate::results;
23
24/// A typed control call: a params struct that knows its [`ControlMethod`] and its result type.
25///
26/// Implemented by every struct in [`crate::params`]; this is what makes
27/// [`ControlClient::parse_response`] return the right typed result for each method at compile time.
28pub trait ControlCall: Serialize {
29 /// The wire method this call invokes.
30 const METHOD: ControlMethod;
31 /// The typed result this call returns on success.
32 type Output: DeserializeOwned;
33}
34
35/// Serialize a typed call's params into a JSON object (`{}` for a no-param call, never `null`).
36fn params_value<C: ControlCall>(call: &C) -> Value {
37 match serde_json::to_value(call) {
38 Ok(Value::Null) => Value::Object(Default::default()),
39 Ok(v) => v,
40 Err(_) => Value::Object(Default::default()),
41 }
42}
43
44/// Build the JSON-RPC request envelope for a typed control call. Pure.
45pub fn build_request<C: ControlCall>(id: RequestId, call: &C) -> JsonRpcRequest {
46 JsonRpcRequest::new(id, C::METHOD.name(), params_value(call))
47}
48
49/// Parse a JSON-RPC response into a typed result, or the [`ControlError`] it carried. Pure.
50pub fn parse_response<C: ControlCall>(
51 response: JsonRpcResponse,
52) -> Result<C::Output, ControlError> {
53 let value = response.into_result()?;
54 serde_json::from_value(value).map_err(|e| {
55 ControlError::of(
56 ControlErrorCode::ControlError,
57 format!("failed to parse {} result: {e}", C::METHOD.name()),
58 )
59 })
60}
61
62/// The client-facing half of the contract: turn typed calls into requests and responses back into
63/// typed results.
64///
65/// The default implementations cover every client; a consumer implements this trait only to
66/// customise request construction (e.g. attaching the control token in a bespoke way). The blanket
67/// [`DefaultControlClient`] gives callers the standard behaviour for free.
68pub trait ControlClient {
69 /// Build the request envelope for a typed call with the given request `id`.
70 fn build_request<C: ControlCall>(&self, id: RequestId, call: &C) -> JsonRpcRequest {
71 build_request(id, call)
72 }
73
74 /// Parse a response envelope into the typed result for call type `C`.
75 fn parse_response<C: ControlCall>(
76 &self,
77 response: JsonRpcResponse,
78 ) -> Result<C::Output, ControlError> {
79 parse_response::<C>(response)
80 }
81}
82
83/// The standard, zero-configuration [`ControlClient`] using the default request/response behaviour.
84#[derive(Debug, Clone, Copy, Default)]
85pub struct DefaultControlClient;
86
87impl ControlClient for DefaultControlClient {}
88
89/// The node-facing half of the contract: a running node implements this to SERVE the control
90/// surface. Each method is typed to the catalog's params/results; the provided
91/// [`dispatch`](ControlHandler::dispatch) routes a raw [`JsonRpcRequest`] to the right method so a
92/// server needs only one entry point and can never mis-route.
93///
94/// Open/proxied shapes (the updater beacon status, the pairing list, the peer-pool snapshot) return
95/// [`Value`] rather than a frozen struct, matching the catalog's [`ControlCall::Output`] for those
96/// methods.
97#[async_trait]
98pub trait ControlHandler: Sync {
99 /// `control.status`
100 async fn status(&self) -> Result<results::StatusResult, ControlError>;
101 /// `control.config.get`
102 async fn config_get(&self) -> Result<results::ConfigResult, ControlError>;
103 /// `control.config.setUpstream`
104 async fn config_set_upstream(
105 &self,
106 params: params::SetUpstreamParams,
107 ) -> Result<results::SetUpstreamResult, ControlError>;
108 /// `control.log.setLevel`
109 async fn log_set_level(
110 &self,
111 params: params::SetLevelParams,
112 ) -> Result<results::SetLevelResult, ControlError>;
113 /// `control.cache.get`
114 async fn cache_get(&self) -> Result<results::CacheView, ControlError>;
115 /// `control.cache.setCap`
116 async fn cache_set_cap(
117 &self,
118 params: params::SetCapParams,
119 ) -> Result<results::SetCapResult, ControlError>;
120 /// `control.cache.clear`
121 async fn cache_clear(&self) -> Result<results::CacheClearResult, ControlError>;
122 /// `control.hostedStores.list`
123 async fn hosted_stores_list(&self) -> Result<results::HostedStoresListResult, ControlError>;
124 /// `control.hostedStores.pin`
125 async fn hosted_stores_pin(
126 &self,
127 params: params::PinParams,
128 ) -> Result<results::PinResult, ControlError>;
129 /// `control.hostedStores.unpin`
130 async fn hosted_stores_unpin(
131 &self,
132 params: params::UnpinParams,
133 ) -> Result<results::UnpinResult, ControlError>;
134 /// `control.hostedStores.status`
135 async fn hosted_stores_status(
136 &self,
137 params: params::HostedStoreStatusParams,
138 ) -> Result<results::HostedStoreStatusResult, ControlError>;
139 /// `control.capsule.fetch`
140 async fn capsule_fetch(
141 &self,
142 params: params::CapsuleFetchParams,
143 ) -> Result<results::CapsuleFetchResult, ControlError>;
144 /// `control.sync.status`
145 async fn sync_status(&self) -> Result<results::SyncStatusResult, ControlError>;
146 /// `control.sync.trigger`
147 async fn sync_trigger(
148 &self,
149 params: params::SyncTriggerParams,
150 ) -> Result<results::SyncTriggerResult, ControlError>;
151 /// `control.updater.status`
152 async fn updater_status(&self) -> Result<Value, ControlError>;
153 /// `control.updater.setChannel`
154 async fn updater_set_channel(
155 &self,
156 params: params::SetChannelParams,
157 ) -> Result<Value, ControlError>;
158 /// `control.updater.pause`
159 async fn updater_pause(&self, params: params::PauseParams) -> Result<Value, ControlError>;
160 /// `control.updater.resume`
161 async fn updater_resume(&self) -> Result<Value, ControlError>;
162 /// `control.updater.checkNow`
163 async fn updater_check_now(&self) -> Result<Value, ControlError>;
164 /// `control.pairing.list`
165 async fn pairing_list(&self) -> Result<Value, ControlError>;
166 /// `control.pairing.approve`
167 async fn pairing_approve(
168 &self,
169 params: params::ApproveParams,
170 ) -> Result<results::PairingApproveResult, ControlError>;
171 /// `control.pairing.revoke`
172 async fn pairing_revoke(
173 &self,
174 params: params::RevokeParams,
175 ) -> Result<results::PairingRevokeResult, ControlError>;
176 /// `control.peerStatus`
177 async fn peer_status(&self) -> Result<Value, ControlError>;
178 /// `control.peers.connect`
179 async fn peers_connect(
180 &self,
181 params: params::PeersConnectParams,
182 ) -> Result<results::PeersConnectResult, ControlError>;
183 /// `control.peers.disconnect`
184 async fn peers_disconnect(
185 &self,
186 params: params::PeersDisconnectParams,
187 ) -> Result<results::PeersDisconnectResult, ControlError>;
188 /// `control.chiaPeers.add` — start trusting a Chia full node the operator RUNS.
189 ///
190 /// An implementation MUST write through to the ONE peer store its wallet replica reads; a
191 /// second peer list is a drift bug waiting to happen.
192 ///
193 /// Three obligations the shell cannot infer from the types:
194 ///
195 /// - it requires the MASTER token ([`ControlMethod::requires_master_token`]). The entry it
196 /// writes carries authority that outlives the calling token, and `control.pairing.revoke`
197 /// does not remove it;
198 /// - `params.ip` is canonicalised with [`crate::params::canonical_peer_ip`] and STORED in that
199 /// form, so `remove` and `list` can match what `add` wrote;
200 /// - `corroboration_bypassed` reports the RESULTING trust state and `notice` carries the
201 /// node's own warning verbatim. Reporting a bypass that did not happen tells an operator
202 /// they configured a node they did not.
203 async fn chia_peers_add(
204 &self,
205 params: params::ChiaPeersAddParams,
206 ) -> Result<results::ChiaPeersAddResult, ControlError>;
207 /// `control.chiaPeers.list` — the tracked Chia full-node peers, banned ones included.
208 ///
209 /// Ordinary token tier: a read that confers nothing, and a paired client that cannot show the
210 /// operator this list cannot show them the trust state they are subject to.
211 async fn chia_peers_list(&self) -> Result<results::ChiaPeersListResult, ControlError>;
212 /// `control.chiaPeers.remove` — stop trusting a Chia full node.
213 ///
214 /// MASTER token, like `add`. This is the ONLY un-trust remedy, so an implementation MUST
215 /// return [`results::ChiaPeerRemovalOutcome::NoSuchPeer`] when nothing matched rather than
216 /// reporting a removal it did not perform.
217 async fn chia_peers_remove(
218 &self,
219 params: params::ChiaPeersRemoveParams,
220 ) -> Result<results::ChiaPeersRemoveResult, ControlError>;
221 /// `control.subscribe`
222 ///
223 /// `params.kind` is OPTIONAL on the wire and absent means
224 /// [`SubscriptionKind::Capsule`](params::SubscriptionKind::Capsule). An implementation MUST
225 /// persist untagged rows it already holds as capsules rather than discarding them: a node that
226 /// refuses to read its own pre-existing `subscriptions.json` starts with an empty one, and the
227 /// upgrade silently unsubscribes the user from everything.
228 async fn subscribe(
229 &self,
230 params: params::SubscribeParams,
231 ) -> Result<results::SubscribeResult, ControlError>;
232 /// `control.unsubscribe`
233 async fn unsubscribe(
234 &self,
235 params: params::UnsubscribeParams,
236 ) -> Result<results::UnsubscribeResult, ControlError>;
237 /// `control.listSubscriptions`
238 async fn list_subscriptions(&self) -> Result<results::ListSubscriptionsResult, ControlError>;
239 /// `control.wallet.balance` (READ-only)
240 async fn wallet_balance(
241 &self,
242 params: params::WalletBalanceParams,
243 ) -> Result<results::WalletBalanceResult, ControlError>;
244 /// `control.wallet.coins` (READ-only, OPEN)
245 ///
246 /// An empty `coins` list MUST mean "a chain was consulted and this address holds nothing".
247 /// A read that could not consult a chain MUST return the matching catalogued error instead.
248 async fn wallet_coins(
249 &self,
250 params: params::WalletCoinsParams,
251 ) -> Result<results::WalletCoinsResult, ControlError>;
252 /// `control.wallet.coinById` (READ-only, OPEN)
253 ///
254 /// `Ok(coin: None)` MUST mean "a chain was consulted and holds no such coin". A read that could
255 /// not consult a chain MUST return the matching catalogued error instead — a caller that cannot
256 /// tell those apart reports a spent mint as pending forever.
257 ///
258 /// The params are validated at DESERIALIZATION (lowercase 64-hex, `0x` stripped), so any path
259 /// that decodes `WalletCoinByIdParams` refuses malformed ids as `INVALID_PARAMS` before this
260 /// method is called.
261 async fn wallet_coin_by_id(
262 &self,
263 params: params::WalletCoinByIdParams,
264 ) -> Result<results::WalletCoinByIdResult, ControlError>;
265 /// `control.wallet.coinSpend` (READ-only, OPEN)
266 ///
267 /// `Ok(spend: None)` MUST mean "a chain was consulted and holds no spend of that coin" — the
268 /// coin is unspent, or unknown. A read that could not consult a chain MUST return the matching
269 /// catalogued error instead: a caller following a singleton forward reads "no spend" as *this is
270 /// the tip* and stops walking, so a failure disguised as absence produces a spend built against
271 /// a superseded singleton.
272 ///
273 /// A returned spend's `puzzle_reveal` MUST tree-hash to the spent coin's own `puzzle_hash`, and
274 /// the implementation MUST fail closed — an error, never an unverified reveal — when it does not
275 /// or when the reveal will not parse. The reveal comes from a peer, and a peer can lie.
276 ///
277 /// The params are validated at DESERIALIZATION (lowercase 64-hex, `0x` stripped), so any path
278 /// that decodes `WalletCoinSpendParams` refuses malformed ids as `INVALID_PARAMS` before this
279 /// method is called.
280 async fn wallet_coin_spend(
281 &self,
282 params: params::WalletCoinSpendParams,
283 ) -> Result<results::WalletCoinSpendResult, ControlError>;
284 /// `control.wallet.coinsByParent` (READ-only, OPEN)
285 ///
286 /// Returns the parent's DIRECT children and nothing further. An implementation MUST NOT recurse:
287 /// a transitive walk over caller-supplied input is unbounded work the caller cannot bound, and a
288 /// partial walk returned as a complete one is a lineage with a silent hole in it.
289 ///
290 /// An empty list MUST mean "a chain was consulted and this parent created no known children".
291 /// A read that could not consult a chain MUST return the matching catalogued error instead.
292 ///
293 /// The answer is ONE PAGE. An implementation MUST return at most
294 /// `params.effective_limit()` records, in ASCENDING `coin_id` order, starting strictly after
295 /// `params.after_coin_id` when one is given; it MUST set `complete` to whether the page carries
296 /// the last child; and it MUST set `cursor` to the last record it actually returned (`None` for
297 /// an empty page). It MUST NOT report `complete: true` on a page it truncated — a caller reads
298 /// that as the end of a lineage branch. The params are validated at DESERIALIZATION, so an
299 /// out-of-range page size is refused as `INVALID_PARAMS` before this method is called.
300 ///
301 /// Every record MUST report `asset: None`: naming a coin by its parent classifies nothing, and
302 /// asserting a class this read never verified is a claim a caller would then spend against.
303 async fn wallet_coins_by_parent(
304 &self,
305 params: params::WalletCoinsByParentParams,
306 ) -> Result<results::WalletCoinsByParentResult, ControlError>;
307 /// `control.wallet.arrivals` (READ-only, TOKEN-GATED)
308 ///
309 /// Gated although it is a read: the caller supplies only a cursor, so the answer names this
310 /// node's OWN watched puzzle hashes and the receive history behind them.
311 ///
312 /// Every returned row MUST be a CONFIRMED arrival that the node itself judged: above its arrival
313 /// baseline, not previously reported, and not the wallet's own change. An implementation MUST NOT
314 /// emit a mempool sighting here, and MUST answer an empty page rather than an error when it has
315 /// no baseline — "nothing arrived" is the honest answer from a wallet that cannot yet tell
316 /// history from news.
317 ///
318 /// `cursor` MUST be the position of the last row actually returned (or the caller's `after_seq`
319 /// for an empty page) and MUST NOT be `latest`; see
320 /// [`WalletArrivalsResult::latest`](results::WalletArrivalsResult::latest).
321 async fn wallet_arrivals(
322 &self,
323 params: params::WalletArrivalsParams,
324 ) -> Result<results::WalletArrivalsResult, ControlError>;
325 /// `control.wallet.peak` (READ-only, OPEN)
326 async fn wallet_peak(&self) -> Result<results::WalletPeakResult, ControlError>;
327 /// `control.peerCounts` (READ-only, OPEN)
328 ///
329 /// `dig_peer_count` MUST be dig-node-core's `connected_peers` — the same figure
330 /// `control.peerStatus` reports — and `chia_peer_count` MUST be the SAME observation
331 /// `wallet_sync_status` reports, served from ONE source so the two answers agree. `None` means
332 /// the count cannot be observed; a network that is not running is UNKNOWN, never `Some(0)`.
333 async fn peer_counts(&self) -> Result<results::PeerCountsResult, ControlError>;
334 /// `control.wallet.syncStatus` (READ-only, OPEN)
335 ///
336 /// `WalletSyncPhase::Synced` MUST require BOTH that the initial catch-up completed and that at
337 /// least one Chia peer connection is live now, which makes it strictly stronger than
338 /// `WalletPeakResult::synced`. `peak_height` MUST be the node's OWN replica's height or `None`,
339 /// never an oracle's, and `chia_peer_count` counts CHIA full-node peers -- never DIG peers.
340 async fn wallet_sync_status(&self) -> Result<results::WalletSyncStatusResult, ControlError>;
341 /// `control.wallet.broadcast` (TOKEN-GATED)
342 ///
343 /// Pushes an ALREADY-SIGNED bundle: the implementation never signs, and never receives anything
344 /// it could sign with (§908). A mempool refusal is `Ok` with `accepted: false`; failing to
345 /// reach a mempool is `Err`.
346 async fn wallet_broadcast(
347 &self,
348 params: params::WalletBroadcastParams,
349 ) -> Result<results::WalletBroadcastResult, ControlError>;
350 /// `control.wallet.watch` (TOKEN-GATED)
351 ///
352 /// Enrols PUBLIC keys for the node's chain replica to follow. The implementation MUST derive the
353 /// addresses itself, from the SAME derivation it applies to the keys already in its own custody
354 /// — a second derivation is a second opinion about which addresses a key covers, and the client
355 /// would read the difference as missing money.
356 ///
357 /// It MUST be IDEMPOTENT: keys already enrolled are reported as `added: 0` and the call
358 /// succeeds. It MUST persist the enrolment across restarts — a set that evaporates on restart
359 /// makes a node that syncs today and reports a zero balance tomorrow.
360 ///
361 /// The params are validated at DESERIALIZATION (lowercase 96-hex, `0x` stripped), so any path
362 /// that decodes `WalletWatchParams` refuses a malformed key as `INVALID_PARAMS` — for the WHOLE
363 /// request — before this method is called.
364 async fn wallet_watch(
365 &self,
366 params: params::WalletWatchParams,
367 ) -> Result<results::WalletWatchResult, ControlError>;
368 /// `control.wallet.unwatch` (TOKEN-GATED)
369 ///
370 /// Deregisters keys, and the following MUST actually stop: the addresses leave the replica's
371 /// watched set, not merely the list this node reports. A registry that keeps syncing what it
372 /// says it forgot is the failure this method exists to make impossible.
373 ///
374 /// Deregistering a key that was never enrolled is a success reporting `removed: 0`.
375 async fn wallet_unwatch(
376 &self,
377 params: params::WalletUnwatchParams,
378 ) -> Result<results::WalletUnwatchResult, ControlError>;
379 /// `control.wallet.watched` (READ-only, TOKEN-GATED)
380 ///
381 /// Gated although it is a read: the caller supplies nothing, so the answer names this node's OWN
382 /// enrolled keys.
383 ///
384 /// MUST return exactly the keys enrolment added, in the wire form they were accepted in, so a
385 /// client can compare its own set against the node's by value. MUST NOT include the node's own
386 /// custody keys: this method reports what was ENROLLED through it, and a caller reconciling
387 /// against a superset would unwatch keys it never watched.
388 async fn wallet_watched(&self) -> Result<results::WalletWatchedResult, ControlError>;
389 /// `control.wallet.reservations.held` (READ-only, TOKEN-GATED)
390 ///
391 /// Gated although it is a read: the caller supplies nothing, so the answer names this node's OWN
392 /// in-flight commitments.
393 ///
394 /// An implementation MUST read its own clock rather than accept one, MUST omit every reservation
395 /// that has already lapsed at that instant, and MUST report the clock it used as `as_of_unix`.
396 ///
397 /// An implementation that cannot read its reservation set MUST return
398 /// [`ControlErrorCode::WalletReservationsUnavailable`](crate::error::ControlErrorCode::WalletReservationsUnavailable)
399 /// and MUST NOT return an empty list. "Nothing is held" permits a caller to spend; "I cannot
400 /// tell" must stop it, and the two are indistinguishable once collapsed.
401 async fn wallet_reservations_held(
402 &self,
403 ) -> Result<results::WalletReservationsHeldResult, ControlError>;
404 /// `control.wallet.reservations.reserve` (TOKEN-GATED)
405 ///
406 /// Acquisition MUST be atomic across concurrent callers: take EVERY coin in `coin_ids` or take
407 /// none. On a clash an implementation MUST have written nothing and MUST answer
408 /// [`ControlErrorCode::WalletCoinsReserved`](crate::error::ControlErrorCode::WalletCoinsReserved)
409 /// — never a shortfall code, because the user has the money and is waiting on a settlement, and
410 /// never a partial success, because a caller believing it holds inputs it does not is the state
411 /// all-or-none exists to make unreachable.
412 ///
413 /// An empty `coin_ids` MUST succeed, returning a handle that releases nothing.
414 ///
415 /// The hold MUST expire on its own. An implementation clamps the requested `ttl_secs` to its own
416 /// maximum, applies its default when none is given, and MUST report the lifetime it actually
417 /// applied — a caller told nothing would release on a schedule the node does not keep.
418 ///
419 /// An implementation MUST NOT require, accept or store key material here (§908). A reservation
420 /// is bookkeeping: it narrows what a selector will choose and authorizes nothing.
421 async fn wallet_reservations_reserve(
422 &self,
423 params: params::WalletReservationsReserveParams,
424 ) -> Result<results::WalletReservationsReserveResult, ControlError>;
425 /// `control.wallet.reservations.release` (TOKEN-GATED)
426 ///
427 /// Releasing a handle that names no live reservation — lapsed, or already released — MUST be a
428 /// SUCCESS reporting `released: false`, never an error. A caller releasing on confirmation
429 /// cannot know whether the TTL got there first, and an error there teaches callers to discard
430 /// the result, which is how the release path stops being called at all.
431 ///
432 /// An implementation MUST free every coin the handle holds, or none of them, for the same reason
433 /// acquisition is all-or-none: a half-freed reservation leaves coins held by a handle the caller
434 /// has thrown away, and only the TTL would ever recover them.
435 async fn wallet_reservations_release(
436 &self,
437 params: params::WalletReservationsReleaseParams,
438 ) -> Result<results::WalletReservationsReleaseResult, ControlError>;
439 /// `control.spends.list` (READ-only, TOKEN-GATED)
440 ///
441 /// One page of the automated-spend audit record — the spends this node made WITHOUT
442 /// per-transaction approval. Gated although it is a read: the caller names no identifier, so the
443 /// answer is this node's OWN spending history.
444 ///
445 /// An implementation MUST NOT let this call initiate, sign, retry, cancel or amend a spend, and
446 /// MUST NOT expose any control method that edits or deletes an entry. The record replaces
447 /// authorization with accountability, and an editable record accounts for nothing.
448 ///
449 /// Four obligations, each of which a plausible implementation gets wrong:
450 ///
451 /// - **Report the failure STAGE, never a bare "failed."** Only
452 /// [`SpendFailureStage::Signing`](results::SpendFailureStage::Signing) means the money
453 /// definitely did not move; a broadcast or confirmation failure is an unknown outcome. An
454 /// implementation that flattens the stage makes every client structurally unable to tell a
455 /// person the truth about their money.
456 /// - **Keep [`Unresolved`](results::SpendOutcome::Unresolved) distinct from `Failed`.** It means
457 /// the node signed and does not know how it ended.
458 /// - **State completeness explicitly.** `complete` MUST be `false` whenever a matching row was
459 /// withheld, and `cursor` MUST be the id of the last row actually returned.
460 /// - **Report unreadable entries.** `unreadable_lines` MUST count entries the node could not
461 /// parse; a trail that lost rows must never read as a tidy shorter one. A record that could not
462 /// be read AT ALL is
463 /// [`SpendAuditUnreadable`](crate::error::ControlErrorCode::SpendAuditUnreadable), never an
464 /// empty page — while a record that was never written IS an empty page, because a node that has
465 /// never spent automatically is the ordinary case.
466 async fn spends_list(
467 &self,
468 params: params::SpendsListParams,
469 ) -> Result<results::SpendsListResult, ControlError>;
470 /// `control.profile.putBody` (TOKEN-GATED)
471 ///
472 /// An implementation MUST independently resolve the profile's root ON CHAIN, recompute the root
473 /// of the supplied body, and REFUSE the call unless the two agree and that root is confirmed.
474 /// The caller's `root` is a claim to be checked — never a fact to be trusted — and dig-app gets
475 /// no exemption: it holds the key and signs the root (§908), but the bytes reach the node the
476 /// same way a peer's bytes do, and the same check binds both. An implementation that stores
477 /// what it is handed makes this node serve arbitrary bytes under someone else's profile id.
478 ///
479 /// A body whose DECODED length exceeds [`MAX_BODY_BYTES`](params::MAX_BODY_BYTES) (4 MiB) MUST
480 /// be refused as `INVALID_PARAMS`, before it is persisted: a body larger than that cannot be
481 /// served to a peer inside dig-gossip's frame ceiling, so accepting it would store something
482 /// permanently unsyncable.
483 ///
484 /// Returning `Ok` therefore asserts BOTH that the root was confirmed on chain and that the body
485 /// is persisted and servable. A refusal is an error, never an `Ok` carrying `stored: false`.
486 async fn profile_put_body(
487 &self,
488 params: params::ProfilePutBodyParams,
489 ) -> Result<results::ProfilePutBodyResult, ControlError>;
490 /// `control.profile.getBody` (READ-only, TOKEN-GATED)
491 ///
492 /// `Ok(body_b64: None)` MUST mean "this node was consulted and holds no body at that root". A
493 /// read that FAILED MUST return a catalogued error instead — a caller that reads a failure as
494 /// absence renders an existing profile as an empty one.
495 ///
496 /// The returned `root` MUST be the root the caller asked for; a node MUST NOT substitute a
497 /// newer body it happens to hold.
498 async fn profile_get_body(
499 &self,
500 params: params::ProfileGetBodyParams,
501 ) -> Result<results::ProfileGetBodyResult, ControlError>;
502 /// `pairing.request` (OPEN)
503 async fn pairing_request(
504 &self,
505 params: params::RequestParams,
506 ) -> Result<results::PairingRequestResult, ControlError>;
507 /// `pairing.poll` (OPEN)
508 async fn pairing_poll(
509 &self,
510 params: params::PollParams,
511 ) -> Result<results::PairingPollResult, ControlError>;
512
513 /// Route a raw JSON-RPC request to the right typed method and build the response envelope.
514 ///
515 /// Deserializes the params for methods that take them, calls the handler, and serializes the
516 /// typed result. An unknown method → `METHOD_NOT_FOUND`; malformed params → `INVALID_PARAMS`.
517 /// This is the single seam a server dispatches through — the KATs exercise it end-to-end.
518 async fn dispatch(&self, request: JsonRpcRequest) -> JsonRpcResponse {
519 let id = request.id.clone();
520 let Some(method) = ControlMethod::from_name(&request.method) else {
521 return JsonRpcResponse::error(
522 id,
523 ControlError::of(
524 ControlErrorCode::MethodNotFound,
525 format!("unknown control method: {}", request.method),
526 ),
527 );
528 };
529 match self.dispatch_method(method, request.params).await {
530 Ok(result) => JsonRpcResponse::success(id, result),
531 Err(err) => JsonRpcResponse::error(id, err),
532 }
533 }
534
535 /// Route to the typed method by [`ControlMethod`], returning the result as a [`Value`]. Split
536 /// from [`dispatch`](ControlHandler::dispatch) so the envelope wrapping stays in one place.
537 #[doc(hidden)]
538 async fn dispatch_method(
539 &self,
540 method: ControlMethod,
541 params: Value,
542 ) -> Result<Value, ControlError> {
543 /// Deserialize a method's params, mapping a shape error to `INVALID_PARAMS`.
544 fn decode<T: DeserializeOwned>(params: Value) -> Result<T, ControlError> {
545 serde_json::from_value(params)
546 .map_err(|e| ControlError::of(ControlErrorCode::InvalidParams, e.to_string()))
547 }
548 /// Serialize a typed result to a `Value` (infallible for our derive-Serialize results).
549 fn encode<T: Serialize>(value: T) -> Result<Value, ControlError> {
550 serde_json::to_value(value)
551 .map_err(|e| ControlError::of(ControlErrorCode::ControlError, e.to_string()))
552 }
553 match method {
554 ControlMethod::Status => encode(self.status().await?),
555 ControlMethod::ConfigGet => encode(self.config_get().await?),
556 ControlMethod::ConfigSetUpstream => {
557 encode(self.config_set_upstream(decode(params)?).await?)
558 }
559 ControlMethod::LogSetLevel => encode(self.log_set_level(decode(params)?).await?),
560 ControlMethod::CacheGet => encode(self.cache_get().await?),
561 ControlMethod::CacheSetCap => encode(self.cache_set_cap(decode(params)?).await?),
562 ControlMethod::CacheClear => encode(self.cache_clear().await?),
563 ControlMethod::HostedStoresList => encode(self.hosted_stores_list().await?),
564 ControlMethod::HostedStoresPin => {
565 encode(self.hosted_stores_pin(decode(params)?).await?)
566 }
567 ControlMethod::HostedStoresUnpin => {
568 encode(self.hosted_stores_unpin(decode(params)?).await?)
569 }
570 ControlMethod::HostedStoresStatus => {
571 encode(self.hosted_stores_status(decode(params)?).await?)
572 }
573 ControlMethod::CapsuleFetch => encode(self.capsule_fetch(decode(params)?).await?),
574 ControlMethod::SyncStatus => encode(self.sync_status().await?),
575 ControlMethod::SyncTrigger => encode(self.sync_trigger(decode(params)?).await?),
576 ControlMethod::UpdaterStatus => self.updater_status().await,
577 ControlMethod::UpdaterSetChannel => self.updater_set_channel(decode(params)?).await,
578 ControlMethod::UpdaterPause => self.updater_pause(decode(params)?).await,
579 ControlMethod::UpdaterResume => self.updater_resume().await,
580 ControlMethod::UpdaterCheckNow => self.updater_check_now().await,
581 ControlMethod::PairingList => self.pairing_list().await,
582 ControlMethod::PairingApprove => encode(self.pairing_approve(decode(params)?).await?),
583 ControlMethod::PairingRevoke => encode(self.pairing_revoke(decode(params)?).await?),
584 ControlMethod::PeerStatus => self.peer_status().await,
585 ControlMethod::PeerCounts => encode(self.peer_counts().await?),
586 ControlMethod::PeersConnect => encode(self.peers_connect(decode(params)?).await?),
587 ControlMethod::PeersDisconnect => encode(self.peers_disconnect(decode(params)?).await?),
588 ControlMethod::ChiaPeersAdd => encode(self.chia_peers_add(decode(params)?).await?),
589 ControlMethod::ChiaPeersList => encode(self.chia_peers_list().await?),
590 ControlMethod::ChiaPeersRemove => {
591 encode(self.chia_peers_remove(decode(params)?).await?)
592 }
593 ControlMethod::Subscribe => encode(self.subscribe(decode(params)?).await?),
594 ControlMethod::Unsubscribe => encode(self.unsubscribe(decode(params)?).await?),
595 ControlMethod::ListSubscriptions => encode(self.list_subscriptions().await?),
596 ControlMethod::WalletBalance => encode(self.wallet_balance(decode(params)?).await?),
597 ControlMethod::WalletCoins => encode(self.wallet_coins(decode(params)?).await?),
598 // Re-validated here idempotently; deserialization already enforced the same rule.
599 ControlMethod::WalletCoinById => {
600 let params: params::WalletCoinByIdParams = decode(params)?;
601 encode(self.wallet_coin_by_id(params.validated()?).await?)
602 }
603 // Re-validated here idempotently; deserialization already enforced the same rule.
604 ControlMethod::WalletCoinSpend => {
605 let params: params::WalletCoinSpendParams = decode(params)?;
606 encode(self.wallet_coin_spend(params.validated()?).await?)
607 }
608 ControlMethod::WalletCoinsByParent => {
609 let params: params::WalletCoinsByParentParams = decode(params)?;
610 encode(self.wallet_coins_by_parent(params.validated()?).await?)
611 }
612 ControlMethod::WalletArrivals => encode(self.wallet_arrivals(decode(params)?).await?),
613 ControlMethod::WalletPeak => encode(self.wallet_peak().await?),
614 ControlMethod::WalletSyncStatus => encode(self.wallet_sync_status().await?),
615 ControlMethod::WalletBroadcast => encode(self.wallet_broadcast(decode(params)?).await?),
616 // Re-validated here idempotently; deserialization already enforced the same rule.
617 ControlMethod::WalletWatch => {
618 let params: params::WalletWatchParams = decode(params)?;
619 encode(self.wallet_watch(params.validated()?).await?)
620 }
621 // Re-validated here idempotently; deserialization already enforced the same rule.
622 ControlMethod::WalletUnwatch => {
623 let params: params::WalletUnwatchParams = decode(params)?;
624 encode(self.wallet_unwatch(params.validated()?).await?)
625 }
626 ControlMethod::WalletWatched => encode(self.wallet_watched().await?),
627 ControlMethod::WalletReservationsHeld => encode(self.wallet_reservations_held().await?),
628 ControlMethod::WalletReservationsReserve => {
629 encode(self.wallet_reservations_reserve(decode(params)?).await?)
630 }
631 ControlMethod::WalletReservationsRelease => {
632 encode(self.wallet_reservations_release(decode(params)?).await?)
633 }
634 // Re-validated here idempotently; deserialization already enforced the same rule.
635 ControlMethod::SpendsList => {
636 let params: params::SpendsListParams = decode(params)?;
637 encode(self.spends_list(params.validated()?).await?)
638 }
639 ControlMethod::ProfilePutBody => encode(self.profile_put_body(decode(params)?).await?),
640 ControlMethod::ProfileGetBody => encode(self.profile_get_body(decode(params)?).await?),
641 ControlMethod::PairingRequest => encode(self.pairing_request(decode(params)?).await?),
642 ControlMethod::PairingPoll => encode(self.pairing_poll(decode(params)?).await?),
643 }
644 }
645}