peat_ffi/lib.rs
1//! Peat FFI - Foreign Function Interface for Kotlin/Swift
2//!
3//! This crate provides UniFFI bindings to expose Peat functionality
4//! to Kotlin (Android) and Swift (iOS) consumer applications.
5//!
6//! ## Features
7//!
8//! - **CoT Encoding**: Convert track data to Cursor-on-Target XML
9//! - **Sync** (optional): P2P document sync via AutomergeIroh backend
10//!
11//! Uses proc-macro only UniFFI approach (no UDL file).
12//!
13//! ## Android JNI Support
14//!
15//! This crate also provides direct JNI bindings that bypass JNA's symbol lookup
16//! issues on Android. The JNI functions are exported with standard naming
17//! (Java_package_Class_method) and can be called directly via Android's NDK.
18
19// Allow pre-existing warnings in FFI code - will clean up incrementally
20#![allow(unused_variables)]
21#![allow(unused_mut)]
22#![allow(dead_code)]
23#![allow(clippy::incompatible_msrv)]
24#![allow(clippy::unnecessary_cast)]
25#![allow(clippy::single_match)]
26#![allow(clippy::items_after_test_module)]
27
28use std::collections::HashMap;
29use std::sync::Arc;
30
31// JNI support for Android
32use jni::objects::{GlobalRef, JByteArray, JClass, JString, JValue};
33use jni::sys::{jboolean, jint, jstring, JavaVM, JNI_VERSION_1_6};
34use jni::JNIEnv;
35use std::os::raw::c_void;
36use std::sync::{LazyLock, Mutex};
37
38// Global JavaVM reference for JNI callbacks from any thread
39static JAVA_VM: LazyLock<Mutex<Option<jni::JavaVM>>> = LazyLock::new(|| Mutex::new(None));
40
41// Global reference to PeerEventManager class
42static PEER_EVENT_MANAGER_CLASS: LazyLock<Mutex<Option<GlobalRef>>> =
43 LazyLock::new(|| Mutex::new(None));
44
45// Global reference to the currently-registered DocumentChangeListener instance.
46// Only one subscription is supported at a time (mirrors UniFFI's
47// PeatNode::subscribe constraint). Held as a GlobalRef so it survives across
48// JNI thread attaches.
49#[cfg(feature = "sync")]
50static DOCUMENT_CHANGE_LISTENER: LazyLock<Mutex<Option<GlobalRef>>> =
51 LazyLock::new(|| Mutex::new(None));
52
53// Flag controlling the lifetime of the document-change subscription task.
54// Set to true by subscribeDocumentChangesJni, false by
55// unsubscribeDocumentChangesJni. The spawned tokio task polls this on each recv
56// to know whether to exit.
57#[cfg(feature = "sync")]
58static DOCUMENT_SUBSCRIPTION_ACTIVE: LazyLock<std::sync::atomic::AtomicBool> =
59 LazyLock::new(|| std::sync::atomic::AtomicBool::new(false));
60
61// peat#885 fault-injection flag, test-only. When armed via
62// `forceStoreErrorForTestingJni`, the next `getDocumentJni` call
63// short-circuits to the Err branch (throws RuntimeException) without
64// touching the underlying store. Self-clears on consumption — one
65// trigger per arm. Process-wide rather than per-handle because tests
66// typically run sequentially on a single instrumented runner; if
67// concurrent multi-handle fault injection is ever needed, promote
68// to a `HashMap<handle, AtomicBool>` keyed by node handle.
69//
70// Always present in non-test builds (the function name's "ForTesting"
71// suffix is the API marker; calling it from production code does no
72// harm beyond setting a flag that a non-test code path never reads
73// because production never calls forceStoreErrorForTestingJni).
74#[cfg(feature = "sync")]
75static FORCE_STORE_ERROR_FOR_TESTING: std::sync::atomic::AtomicBool =
76 std::sync::atomic::AtomicBool::new(false);
77
78// ADR-059 Slice 1.b.2: outbound BLE frame callback. The Kotlin listener
79// receives `onFrame(transportId, collection, bytes)` for every encoded
80// document the BLE translator produces in `TransportManager`'s fan-out.
81// Replaceable: a second subscribe swaps the GlobalRef without re-registering
82// the underlying translator/sink. Cleared on unsubscribe.
83#[cfg(all(feature = "sync", feature = "bluetooth"))]
84static OUTBOUND_FRAME_LISTENER: LazyLock<Mutex<Option<GlobalRef>>> =
85 LazyLock::new(|| Mutex::new(None));
86
87// FanoutHandle held alive across the subscription lifetime. Drop cancels
88// the observer tasks; `unsubscribeOutboundFramesJni` takes the value and
89// drops it explicitly. Wrapped in a feature-gated alias because the type
90// only exists when peat-mesh is in scope.
91#[cfg(all(feature = "sync", feature = "bluetooth"))]
92static OUTBOUND_FRAME_FANOUT: LazyLock<Mutex<Option<peat_mesh::transport::FanoutHandle>>> =
93 LazyLock::new(|| Mutex::new(None));
94
95// Global Peat node handle that survives APK replacement
96// This allows Kotlin code to recover the node connection after plugin hot-swap
97#[cfg(feature = "sync")]
98static GLOBAL_NODE_HANDLE: LazyLock<Mutex<i64>> = LazyLock::new(|| Mutex::new(0));
99
100/// Store an **owning** reference to `node` in [`GLOBAL_NODE_HANDLE`], dropping
101/// any previously-stored owning reference.
102///
103/// Invariant: the slot holds either `0` or a pointer produced here by
104/// `Arc::into_raw(Arc::clone(..))` — i.e. always an owning reference,
105/// independent of how the *originating* handle is released (UniFFI/Dart GC for
106/// `create_node`, or `freeNodeJni` for the JNI create paths). This is what
107/// makes `getGlobalNodeHandleJni` safe for the BLE bridge: the node can't be
108/// freed out from under a consumer while the global still references it.
109/// [`clearGlobalNodeHandleJni`] releases this reference. Centralising the
110/// write here keeps every create path consistent so the clear path can always
111/// safely `Arc::from_raw` + drop without risking a double-free or UAF.
112#[cfg(feature = "sync")]
113fn set_global_node_handle(node: &Arc<PeatNode>) {
114 store_owning_node_in_slot(&GLOBAL_NODE_HANDLE, node);
115}
116
117/// Store an owning `Arc<PeatNode>` pointer in `slot`, dropping any previously
118/// stored owning pointer. Factored out of [`set_global_node_handle`] /
119/// [`clearGlobalNodeHandleJni`] so the store/clear ownership semantics can be
120/// unit-tested against a *local* slot (the real `GLOBAL_NODE_HANDLE` is
121/// process-global shared state that parallel tests would race on).
122#[cfg(feature = "sync")]
123fn store_owning_node_in_slot(slot: &Mutex<i64>, node: &Arc<PeatNode>) {
124 if let Ok(mut g) = slot.lock() {
125 let prev = std::mem::replace(&mut *g, Arc::into_raw(Arc::clone(node)) as i64);
126 if prev != 0 {
127 // SAFETY: `prev` was produced by `Arc::into_raw(Arc::clone(..))`
128 // in a prior call to this helper; reclaim it to drop that ref.
129 unsafe { drop(Arc::from_raw(prev as *const PeatNode)) };
130 }
131 }
132}
133
134/// Zero `slot`, dropping the owning `Arc<PeatNode>` pointer it held (if any).
135/// Backs [`clearGlobalNodeHandleJni`].
136#[cfg(feature = "sync")]
137fn clear_owning_node_slot(slot: &Mutex<i64>) {
138 if let Ok(mut g) = slot.lock() {
139 let prev = std::mem::replace(&mut *g, 0);
140 if prev != 0 {
141 // SAFETY: the slot only ever holds `0` or an owning pointer from
142 // `store_owning_node_in_slot`; reclaim + drop to release the ref.
143 unsafe { drop(Arc::from_raw(prev as *const PeatNode)) };
144 }
145 }
146}
147
148// Global BLE transport reference for Android JNI access
149// Kotlin signals BLE state (started/stopped, peer discovery) into this
150// transport which makes TransportManager aware of BLE availability for PACE
151// routing.
152#[cfg(all(feature = "bluetooth", target_os = "android"))]
153static ANDROID_BLE_TRANSPORT: LazyLock<
154 Mutex<Option<Arc<PeatBleTransport<peat_btle::platform::android::AndroidAdapter>>>>,
155> = LazyLock::new(|| Mutex::new(None));
156
157use peat_protocol::cot::{
158 CotEncoder, Position as CotPosition, TrackUpdate, Velocity as CotVelocity,
159};
160
161#[cfg(feature = "sync")]
162use peat_protocol::network::{IrohTransport, PeerInfo as PeatPeerInfo, TransportPeerEvent};
163#[cfg(feature = "sync")]
164use peat_protocol::storage::{AutomergeBackend, AutomergeStore, StorageBackend, SyncCapable};
165#[cfg(feature = "sync")]
166use peat_protocol::sync::automerge::AutomergeIrohBackend;
167#[cfg(feature = "sync")]
168use peat_protocol::sync::{BackendConfig, DataSyncBackend, TransportConfig};
169// Blob transfer via peat-mesh NetworkedIrohBlobStore (ADR-060).
170// Parallel endpoint model — blob store runs its own iroh Router/Endpoint
171// separate from PeatNode.iroh_transport's sync endpoint.
172use peat_mesh::storage::automerge_store::{
173 ChangeOrigin as _PeatMeshChangeOrigin, DocChange as _PeatMeshDocChange,
174};
175#[cfg(feature = "sync")]
176use peat_mesh::storage::{
177 BlobMetadata, BlobStore, BlobStoreExt, BlobToken, NetworkedIrohBlobStore,
178};
179#[cfg(feature = "sync")]
180use peat_mesh::IrohConfig as PeatMeshIrohConfig;
181#[cfg(all(feature = "sync", feature = "bluetooth"))]
182use peat_protocol::transport::btle::PeatBleTransport;
183#[cfg(feature = "sync")]
184use peat_protocol::transport::{
185 CollectionRouteTable, IrohMeshTransport, Transport, TransportCapabilities, TransportInstance,
186 TransportManager, TransportManagerConfig, TransportPolicy, TransportType,
187};
188#[cfg(feature = "sync")]
189use std::net::SocketAddr;
190#[cfg(feature = "sync")]
191use std::path::PathBuf;
192#[cfg(feature = "sync")]
193use std::sync::atomic::{AtomicBool, Ordering};
194
195// Setup UniFFI scaffolding
196uniffi::setup_scaffolding!();
197
198// FFIBuffer wrappers for Dart FFI bindings
199pub mod dart_ffi;
200
201// Shared water-supply Counter — a self-contained Automerge CRDT doc carried
202// over BLE (CRDT-over-Automerge-over-BLE). See docs/crdt-counter-over-ble.md.
203#[cfg(feature = "sync")]
204mod water_counter;
205
206// Generic CRDT key-value documents (nodes/commands/cells/mission) — same
207// Automerge-over-BLE pattern as the counter, for record collections.
208#[cfg(feature = "sync")]
209mod crdt_kv;
210
211// Persistent roster of known group peers — the reconnect foundation. Remembers
212// the group a node joined so it can re-dial members after a restart, network
213// change, or transport switch. See roster.rs.
214#[cfg(feature = "sync")]
215mod roster;
216
217// Per-peer reconnect supervisor — the dial state machine + backoff policy that
218// walks the roster and keeps a live path up to each member over whatever
219// transport is currently reachable. Pure policy; orchestration is in lib.rs.
220// See supervisor.rs.
221#[cfg(feature = "sync")]
222mod supervisor;
223
224/// Get the Peat library version
225#[uniffi::export]
226pub fn peat_version() -> String {
227 env!("CARGO_PKG_VERSION").to_string()
228}
229
230/// Geographic position for FFI
231#[derive(Debug, Clone, uniffi::Record)]
232pub struct Position {
233 /// Latitude in degrees (WGS84)
234 pub lat: f64,
235 /// Longitude in degrees (WGS84)
236 pub lon: f64,
237 /// Height Above Ellipsoid in meters (optional)
238 pub hae: Option<f64>,
239}
240
241/// Velocity vector for FFI
242#[derive(Debug, Clone, uniffi::Record)]
243pub struct Velocity {
244 /// Bearing in degrees (0 = North, clockwise)
245 pub bearing: f64,
246 /// Speed in meters per second
247 pub speed_mps: f64,
248}
249
250/// Track data for CoT encoding
251#[derive(Debug, Clone, uniffi::Record)]
252pub struct TrackData {
253 /// Unique track identifier
254 pub track_id: String,
255 /// Source node ID
256 pub source_node: String,
257 /// Geographic position
258 pub position: Position,
259 /// Optional velocity
260 pub velocity: Option<Velocity>,
261 /// MIL-STD-2525 classification (e.g., "a-f-G-U-C")
262 pub classification: String,
263 /// Detection confidence (0.0 - 1.0)
264 pub confidence: f64,
265 /// Optional cell ID (for squad-level tracks)
266 pub cell_id: Option<String>,
267 /// Optional formation ID
268 pub formation_id: Option<String>,
269}
270
271/// FFI Error type
272#[derive(Debug, thiserror::Error, uniffi::Error)]
273pub enum PeatError {
274 #[error("Encoding error: {msg}")]
275 EncodingError { msg: String },
276 #[error("Invalid input: {msg}")]
277 InvalidInput { msg: String },
278 #[error("Storage error: {msg}")]
279 StorageError { msg: String },
280 #[error("Connection error: {msg}")]
281 ConnectionError { msg: String },
282 #[error("Sync error: {msg}")]
283 SyncError { msg: String },
284}
285
286/// Encode a track to CoT XML string
287#[uniffi::export]
288pub fn encode_track_to_cot(track: TrackData) -> Result<String, PeatError> {
289 // Validate input
290 if track.track_id.is_empty() {
291 return Err(PeatError::InvalidInput {
292 msg: "track_id cannot be empty".to_string(),
293 });
294 }
295
296 // Convert FFI types to internal types
297 let position = CotPosition {
298 lat: track.position.lat,
299 lon: track.position.lon,
300 cep_m: None,
301 hae: track.position.hae,
302 };
303
304 let velocity = track.velocity.map(|v| CotVelocity {
305 bearing: v.bearing,
306 speed_mps: v.speed_mps,
307 });
308
309 let track_update = TrackUpdate {
310 track_id: track.track_id,
311 source_node: track.source_node,
312 source_model: "peat-ffi".to_string(),
313 model_version: peat_version(),
314 cell_id: track.cell_id,
315 formation_id: track.formation_id,
316 timestamp: chrono::Utc::now(),
317 position,
318 velocity,
319 classification: track.classification,
320 confidence: track.confidence,
321 attributes: HashMap::new(),
322 };
323
324 // Encode to CoT
325 let encoder = CotEncoder::new();
326 let event = encoder
327 .track_update_to_event(&track_update)
328 .map_err(|e| PeatError::EncodingError { msg: e.to_string() })?;
329
330 event
331 .to_xml()
332 .map_err(|e| PeatError::EncodingError { msg: e.to_string() })
333}
334
335/// Create a position from coordinates
336#[uniffi::export]
337pub fn create_position(lat: f64, lon: f64, hae: Option<f64>) -> Position {
338 Position { lat, lon, hae }
339}
340
341/// Create a velocity from bearing and speed
342#[uniffi::export]
343pub fn create_velocity(bearing: f64, speed_mps: f64) -> Velocity {
344 Velocity { bearing, speed_mps }
345}
346
347// =============================================================================
348// PeatNode - P2P Sync Support (requires "sync" feature)
349// =============================================================================
350
351/// Transport configuration for BLE and other transports (ADR-039, #556)
352///
353/// Controls which transports are enabled and their settings.
354/// Used by `NodeConfig` to configure multi-transport support.
355#[cfg(feature = "sync")]
356#[derive(Debug, Clone, Default, uniffi::Record)]
357pub struct TransportConfigFFI {
358 /// Enable Bluetooth LE transport (requires "bluetooth" feature)
359 /// When enabled, BLE mesh networking is available alongside Iroh/QUIC
360 pub enable_ble: bool,
361 /// BLE mesh ID (optional, defaults to app_id if not specified)
362 /// Used to identify the BLE mesh network for peer discovery
363 pub ble_mesh_id: Option<String>,
364 /// BLE power profile: "aggressive", "balanced", or "low_power"
365 /// - aggressive: Maximum range/speed, higher battery usage
366 /// - balanced: Default, moderate power usage
367 /// - low_power: Minimal battery impact, reduced range/speed
368 pub ble_power_profile: Option<String>,
369 /// Transport preference order (optional)
370 /// List of transport names in order of preference, e.g., ["iroh", "ble",
371 /// "lora"] Used by TransportManager's PACE policy for transport
372 /// selection
373 pub transport_preference: Option<Vec<String>>,
374 /// Per-collection transport routing (optional)
375 /// JSON-encoded CollectionRouteTable for explicit collection->transport
376 /// bindings. Collections not listed fall through to PACE/legacy
377 /// scoring.
378 pub collection_routes_json: Option<String>,
379 /// Enable iroh's n0 hosted public relay pool + DNS discovery at runtime
380 /// (peat-flutter relay toggle). Default `false` keeps the local-only,
381 /// no-phone-home posture (`presets::Empty`). When `true`, the iroh
382 /// endpoint is built with `presets::N0`, routing traffic through n0's
383 /// PUBLIC relay infrastructure (`*.iroh.network`) so internet-connected
384 /// devices can sync without a shared LAN. Opt-in only.
385 ///
386 /// MUST remain the last field: the hand-maintained Dart FFI codec in
387 /// peat-flutter (`peat_ffi.dart`) reads/writes record fields in
388 /// declaration order, and the field was appended there too.
389 pub enable_n0_relay: bool,
390}
391
392/// Configuration for creating a PeatNode
393#[cfg(feature = "sync")]
394#[derive(Debug, Clone, uniffi::Record)]
395pub struct NodeConfig {
396 /// Application/Formation ID (used for peer discovery and authentication)
397 /// This identifies which "formation" or "swarm" this node belongs to.
398 pub app_id: String,
399 /// Shared secret key (base64-encoded 32 bytes) for peer authentication
400 /// Only peers with matching app_id AND shared_key can connect.
401 /// Generate with: `openssl rand -base64 32`
402 pub shared_key: String,
403 /// Bind address for P2P connections (e.g., "0.0.0.0:0" for auto-assign)
404 pub bind_address: Option<String>,
405 /// Storage path for Automerge documents
406 pub storage_path: String,
407 /// Transport configuration (optional, defaults to Iroh-only)
408 /// Use this to enable BLE and configure multi-transport behavior
409 pub transport: Option<TransportConfigFFI>,
410}
411
412/// Information about a peer node for connection
413#[cfg(feature = "sync")]
414#[derive(Debug, Clone, uniffi::Record)]
415pub struct PeerInfo {
416 /// Human-readable peer name
417 pub name: String,
418 /// Hex-encoded node ID (Iroh endpoint ID)
419 pub node_id: String,
420 /// List of addresses (e.g., "127.0.0.1:19001")
421 pub addresses: Vec<String>,
422 /// Optional relay URL
423 pub relay_url: Option<String>,
424}
425
426/// Sync statistics
427#[cfg(feature = "sync")]
428#[derive(Debug, Clone, uniffi::Record)]
429pub struct SyncStats {
430 /// Whether sync is currently active
431 pub sync_active: bool,
432 /// Number of connected peers
433 pub connected_peers: u32,
434 /// Total bytes sent
435 pub bytes_sent: u64,
436 /// Total bytes received
437 pub bytes_received: u64,
438}
439
440// =============================================================================
441// ADR-032 §Amendment A — Per-Peer Transport State (UniFFI surface)
442// =============================================================================
443//
444// Mirror types over `peat_mesh::transport::LinkState` family. The
445// peat-mesh types aren't UniFFI-decorated (they live in the transport
446// layer, not the binding layer), so we re-shape them into peat-ffi
447// `Record`s/`Enum`s with `From<peat_mesh::...>` conversions. Kotlin
448// plugin consumers render directly off these.
449//
450// Per ADR-032 §Amendment A's host-rendering rule, peat-ffi is the
451// *single source of truth* for transport-state queries in the UI; the
452// plugin MUST NOT reach into peat-btle's UniFFI directly for this
453// purpose. The unified loop walks `TransportManager`, calls
454// `peer_link_state` on each registered transport, and overlays
455// `transport_id` from the registered id (interface overlay is a
456// follow-up — `TransportManager` doesn't yet expose a public
457// instance-metadata accessor).
458
459/// Per-peer transport state across all registered transports.
460///
461/// Returned by [`PeatNode::peer_transport_state`] and contained in the
462/// list returned by [`PeatNode::all_peer_transport_states`]. An empty
463/// `links` vec is a valid state and means "this peer is not currently
464/// reachable via any registered transport" — visualization should
465/// render the peer with no transport badges, not as an error.
466#[cfg(feature = "sync")]
467#[derive(Debug, Clone, uniffi::Record)]
468pub struct PeerTransportState {
469 /// Hex-encoded peer node identifier (matches the form produced by
470 /// `PeatNode::node_id` and `PeatNode::connected_peers`).
471 pub peer_id: String,
472 /// Links for each transport that currently has a record of this
473 /// peer. Order is implementation-defined (usually
474 /// `TransportManager`'s registration order). An empty list is
475 /// valid — see struct docs.
476 pub links: Vec<TransportLink>,
477}
478
479/// One transport's link state for a peer (FFI mirror of
480/// `peat_mesh::transport::LinkState`).
481#[cfg(feature = "sync")]
482#[derive(Debug, Clone, uniffi::Record)]
483pub struct TransportLink {
484 /// Identifies the registered transport instance, e.g. `"ble-hci0"`,
485 /// `"iroh-wlan0"`. Per ADR-032 §Amendment A, peat-ffi overlays this
486 /// from the `TransportManager`-registered id at synthesis time.
487 pub transport_id: String,
488 /// Transport family, lowercase string for cross-language
489 /// portability (`"ble"` / `"iroh"` / `"lora"` / `"satellite"` / …).
490 pub transport_type: String,
491 /// Physical interface name where applicable (`eth0`, `wlan0`,
492 /// `p2p-wlan0`). `None` for transports that don't expose a NIC
493 /// concept (e.g. BLE, LoRa).
494 pub interface: Option<String>,
495 /// Bucketed quality. Each transport defines its own thresholds.
496 pub quality: TransportLinkQuality,
497 /// Round-trip-time estimate in milliseconds, where the transport
498 /// can measure or estimate it.
499 pub rtt_ms: Option<u32>,
500 /// Received signal strength in dBm, populated by transports that
501 /// expose it (BLE, LoRa, tactical radio). `None` for IP transports.
502 pub rssi_dbm: Option<i8>,
503 /// Path classification for IP-style transports with a relay
504 /// concept (iroh's `PathInfo::is_relay()`). `None` where the
505 /// concept doesn't apply (BLE).
506 pub path_kind: Option<TransportPathKind>,
507}
508
509/// Bucketed link quality for UI tier indicators.
510#[cfg(feature = "sync")]
511#[derive(Debug, Clone, Copy, uniffi::Enum)]
512pub enum TransportLinkQuality {
513 Excellent,
514 Good,
515 Fair,
516 Weak,
517 Unknown,
518}
519
520/// Connection path classification.
521///
522/// `Mixed` (multi-path concurrent) was considered during ADR-032
523/// §Amendment A and intentionally deferred until a real emitter exists.
524#[cfg(feature = "sync")]
525#[derive(Debug, Clone, Copy, uniffi::Enum)]
526pub enum TransportPathKind {
527 Direct,
528 Relay,
529}
530
531#[cfg(feature = "sync")]
532impl From<peat_mesh::transport::LinkQuality> for TransportLinkQuality {
533 fn from(q: peat_mesh::transport::LinkQuality) -> Self {
534 match q {
535 peat_mesh::transport::LinkQuality::Excellent => TransportLinkQuality::Excellent,
536 peat_mesh::transport::LinkQuality::Good => TransportLinkQuality::Good,
537 peat_mesh::transport::LinkQuality::Fair => TransportLinkQuality::Fair,
538 peat_mesh::transport::LinkQuality::Weak => TransportLinkQuality::Weak,
539 peat_mesh::transport::LinkQuality::Unknown => TransportLinkQuality::Unknown,
540 }
541 }
542}
543
544#[cfg(feature = "sync")]
545impl From<peat_mesh::transport::PathKind> for TransportPathKind {
546 fn from(p: peat_mesh::transport::PathKind) -> Self {
547 match p {
548 peat_mesh::transport::PathKind::Direct => TransportPathKind::Direct,
549 peat_mesh::transport::PathKind::Relay => TransportPathKind::Relay,
550 }
551 }
552}
553
554#[cfg(feature = "sync")]
555impl From<peat_mesh::transport::LinkState> for TransportLink {
556 fn from(s: peat_mesh::transport::LinkState) -> Self {
557 // `transport_type` to lowercase string — the ADR's enum names
558 // (BluetoothLE, Quic, etc.) are descriptive but don't match the
559 // string form callers tend to use ("ble", "iroh"). Map
560 // explicitly so a future enum-variant addition is a compile-
561 // time prompt to extend this map rather than silently emitting
562 // a Debug-formatted string.
563 let transport_type = match s.transport_type {
564 peat_mesh::transport::TransportType::BluetoothLE => "ble".to_string(),
565 peat_mesh::transport::TransportType::Quic => "iroh".to_string(),
566 peat_mesh::transport::TransportType::LoRa => "lora".to_string(),
567 peat_mesh::transport::TransportType::WifiDirect => "wifi-direct".to_string(),
568 peat_mesh::transport::TransportType::TacticalRadio => "tactical-radio".to_string(),
569 peat_mesh::transport::TransportType::Satellite => "satellite".to_string(),
570 peat_mesh::transport::TransportType::BluetoothClassic => {
571 "bluetooth-classic".to_string()
572 }
573 peat_mesh::transport::TransportType::Custom(n) => format!("custom-{n}"),
574 };
575 TransportLink {
576 transport_id: s.transport_id,
577 transport_type,
578 interface: s.interface,
579 quality: s.quality.into(),
580 rtt_ms: s.rtt_ms,
581 rssi_dbm: s.rssi_dbm,
582 path_kind: s.path_kind.map(Into::into),
583 }
584 }
585}
586
587/// Type of document change event
588#[cfg(feature = "sync")]
589#[derive(Debug, Clone, uniffi::Enum)]
590pub enum ChangeType {
591 /// Document was created or updated
592 Upsert,
593 /// Document was deleted
594 Delete,
595}
596
597/// Where a document change came from — a local write vs. a remote peer's sync.
598///
599/// This is the substrate-level signal a consumer needs to decide whether a
600/// change is worth surfacing to the user (e.g. a local notification). A
601/// *remote* change is something another node did and may warrant alerting;
602/// a *local* change is the user's own edit and usually is not. peat stays
603/// domain-agnostic: it reports origin, collection, and doc id — the consumer
604/// owns the notability policy and the notification text. Mirrors peat-mesh's
605/// internal `ChangeOrigin` (which isn't UniFFI-decorated) at the FFI boundary.
606#[cfg(feature = "sync")]
607#[derive(Debug, Clone, uniffi::Enum)]
608pub enum ChangeOrigin {
609 /// The change originated from a local write on this node.
610 Local,
611 /// The change arrived from a remote peer via sync. `peer_id` is that peer's
612 /// stable identifier (hex node id for the iroh transport; the store keeps it
613 /// transport-agnostic so BLE/Lite peers populate the same field).
614 Remote { peer_id: String },
615}
616
617#[cfg(feature = "sync")]
618impl From<_PeatMeshChangeOrigin> for ChangeOrigin {
619 fn from(o: _PeatMeshChangeOrigin) -> Self {
620 match o {
621 _PeatMeshChangeOrigin::Local => ChangeOrigin::Local,
622 _PeatMeshChangeOrigin::Remote(peer_id) => ChangeOrigin::Remote { peer_id },
623 }
624 }
625}
626
627#[cfg(all(test, feature = "sync"))]
628mod change_event_tests {
629 use super::*;
630
631 #[test]
632 fn remote_change_splits_key_and_carries_peer() {
633 let dc = _PeatMeshDocChange {
634 key: "tracks:abc-123".to_string(),
635 origin: _PeatMeshChangeOrigin::Remote("peerhex".to_string()),
636 };
637 let out = document_change_from(dc);
638 assert_eq!(out.collection, "tracks");
639 assert_eq!(out.doc_id, "abc-123");
640 assert!(matches!(out.change_type, ChangeType::Upsert));
641 match out.origin {
642 ChangeOrigin::Remote { peer_id } => assert_eq!(peer_id, "peerhex"),
643 other => panic!("expected Remote origin, got {other:?}"),
644 }
645 }
646
647 #[test]
648 fn local_change_without_separator_uses_default_collection() {
649 let dc = _PeatMeshDocChange {
650 key: "loosekey".to_string(),
651 origin: _PeatMeshChangeOrigin::Local,
652 };
653 let out = document_change_from(dc);
654 assert_eq!(out.collection, "default");
655 assert_eq!(out.doc_id, "loosekey");
656 assert!(matches!(out.origin, ChangeOrigin::Local));
657 }
658
659 #[test]
660 fn key_with_multiple_colons_splits_on_first() {
661 // split_once(':') keeps everything after the first colon as the id.
662 let dc = _PeatMeshDocChange {
663 key: "ns:weird:id".to_string(),
664 origin: _PeatMeshChangeOrigin::Local,
665 };
666 let out = document_change_from(dc);
667 assert_eq!(out.collection, "ns");
668 assert_eq!(out.doc_id, "weird:id");
669 }
670}
671
672/// Document change event for subscriptions
673#[cfg(feature = "sync")]
674#[derive(Debug, Clone, uniffi::Record)]
675pub struct DocumentChange {
676 /// Collection name
677 pub collection: String,
678 /// Document ID
679 pub doc_id: String,
680 /// Type of change
681 pub change_type: ChangeType,
682 /// Where the change came from — local edit vs. remote peer sync. Lets a
683 /// consumer notify on remote changes without alerting on its own edits.
684 pub origin: ChangeOrigin,
685}
686
687/// Build the FFI [`DocumentChange`] from peat-mesh's origin-tagged `DocChange`.
688///
689/// Splits the `"collection:doc_id"` key (falling back to the `"default"`
690/// collection when the key has no separator) and carries the [`ChangeOrigin`]
691/// through so consumers can distinguish a remote peer's change from a local
692/// edit. `change_type` is always `Upsert`: the change stream is key-granular
693/// and does not distinguish tombstone deletes at this layer (a delete arrives
694/// as an upsert of a tombstoned document), so detecting deletes would require
695/// reading the doc body — a domain concern left to the consumer.
696#[cfg(feature = "sync")]
697fn document_change_from(doc_change: _PeatMeshDocChange) -> DocumentChange {
698 let origin = ChangeOrigin::from(doc_change.origin);
699 match doc_change.key.split_once(':') {
700 Some((collection, doc_id)) => DocumentChange {
701 collection: collection.to_string(),
702 doc_id: doc_id.to_string(),
703 change_type: ChangeType::Upsert,
704 origin,
705 },
706 None => DocumentChange {
707 collection: "default".to_string(),
708 doc_id: doc_change.key,
709 change_type: ChangeType::Upsert,
710 origin,
711 },
712 }
713}
714
715/// Encoded BLE outbound frame produced by the `BleTranslator` fan-out.
716///
717/// Received by calling [`PeatNode::poll_outbound_frames`] on the host side.
718/// The host is responsible for the final transport-specific framing (GATT
719/// write, encryption envelope) before putting `bytes` on the radio.
720#[cfg(all(feature = "sync", feature = "bluetooth"))]
721#[derive(Debug, Clone, uniffi::Record)]
722pub struct OutboundFrame {
723 /// Transport identifier — `"ble"` for typed 0xB6 frames, `"ble-lite"`
724 /// for universal-Document (peat-lite) frames.
725 pub transport_id: String,
726 /// Collection the document belongs to (e.g. `"tracks"`, `"platforms"`).
727 pub collection: String,
728 /// postcard-encoded typed BLE struct ready for the radio.
729 pub bytes: Vec<u8>,
730}
731
732/// Callback interface for document change notifications
733///
734/// Implement this interface in Kotlin/Swift to receive document updates.
735#[cfg(feature = "sync")]
736#[uniffi::export(callback_interface)]
737pub trait DocumentCallback: Send + Sync {
738 /// Called when a document changes
739 fn on_change(&self, change: DocumentChange);
740
741 /// Called when an error occurs in the subscription
742 fn on_error(&self, message: String);
743}
744
745/// Outbound transport-frame callback for non-Android platforms (iOS via
746/// UniFFI). Mirrors the Android `OutboundFrameListener` JNI surface
747/// (`subscribeOutboundFramesJni`); the trait method receives the same
748/// `(transport_id, collection, bytes)` triple per encoded document.
749///
750/// On Android the JNI path is used directly because UniFFI 0.28's Kotlin
751/// backend wraps callback interfaces in `com.sun.jna.Callback`, which
752/// fails under Android plugin-host classloader isolation. Implementations
753/// on non-Android platforms should expect any-thread invocation from the
754/// `peat-mesh` runtime.
755///
756/// The `register_outbound_frame_callback` method on [`PeatNode`] that
757/// would consume this trait is deferred to a follow-up: the
758/// `Drop`-vs-async `unregister_translator` interaction needs an
759/// `Arc<TransportManager>` refactor of `PeatNode` to be done cleanly
760/// (current `TransportManager` field is owned, not Arc-wrapped, so a
761/// subscription handle has no clean way to drive teardown on drop).
762/// The trait declaration here serves as documentation of the iOS-side
763/// shape so the follow-up can land without an FFI break.
764#[cfg(all(feature = "sync", feature = "bluetooth"))]
765#[uniffi::export(callback_interface)]
766pub trait OutboundFrameCallback: Send + Sync {
767 fn on_frame(&self, transport_id: String, collection: String, bytes: Vec<u8>);
768}
769
770/// Handle for an active document subscription
771///
772/// Drop this handle to unsubscribe from document changes.
773#[cfg(feature = "sync")]
774#[derive(uniffi::Object)]
775pub struct SubscriptionHandle {
776 active: Arc<AtomicBool>,
777 /// Queued changes for polling consumers (populated by `subscribe_poll`).
778 pending: Arc<std::sync::Mutex<std::collections::VecDeque<DocumentChange>>>,
779}
780
781#[cfg(feature = "sync")]
782impl SubscriptionHandle {
783 fn new(active: Arc<AtomicBool>) -> Self {
784 Self {
785 active,
786 pending: Arc::new(std::sync::Mutex::new(std::collections::VecDeque::new())),
787 }
788 }
789
790 fn new_with_queue(
791 active: Arc<AtomicBool>,
792 pending: Arc<std::sync::Mutex<std::collections::VecDeque<DocumentChange>>>,
793 ) -> Self {
794 Self { active, pending }
795 }
796}
797
798#[cfg(feature = "sync")]
799#[uniffi::export]
800impl SubscriptionHandle {
801 /// Check if the subscription is still active
802 pub fn is_active(&self) -> bool {
803 self.active.load(Ordering::SeqCst)
804 }
805
806 /// Cancel the subscription
807 pub fn cancel(&self) {
808 self.active.store(false, Ordering::SeqCst);
809 }
810
811 /// Drain all pending document changes. Non-blocking.
812 ///
813 /// Only populated when the subscription was opened via
814 /// [`PeatNode::subscribe_poll`]. Always returns an empty Vec for
815 /// subscriptions opened via [`PeatNode::subscribe`] (callback path).
816 pub fn poll_changes(&self) -> Vec<DocumentChange> {
817 self.pending
818 .lock()
819 .map(|mut q| q.drain(..).collect())
820 .unwrap_or_default()
821 }
822}
823
824#[cfg(feature = "sync")]
825impl Drop for SubscriptionHandle {
826 fn drop(&mut self) {
827 self.active.store(false, Ordering::SeqCst);
828 }
829}
830
831/// A Peat network node with P2P sync capabilities
832///
833/// Max concurrent in-flight reconnect dials (see
834/// [`PeatNode::reconnect_dial_semaphore`]). Mobile group sizes are small; this
835/// only bounds the cold-start volley when a large saved roster (or
836/// `wake_reconnect` clearing all backoffs) would otherwise dial every peer at
837/// once. Subsequent rounds are decorrelated by per-peer backoff jitter.
838#[cfg(feature = "sync")]
839const MAX_CONCURRENT_RECONNECT_DIALS: usize = 8;
840
841/// Wraps AutomergeIrohBackend for authenticated document sync.
842/// Requires matching app_id and shared_key for peer connections.
843#[cfg(feature = "sync")]
844#[derive(uniffi::Object)]
845pub struct PeatNode {
846 /// The sync backend with FormationKey authentication
847 sync_backend: Arc<AutomergeIrohBackend>,
848 /// Storage backend for document operations (shared with sync_backend)
849 /// Note: This is the SAME backend instance used by sync_backend to ensure
850 /// sync coordinator state is shared. Do NOT create a separate backend.
851 storage_backend: Arc<AutomergeBackend>,
852 /// Generic application-level mesh document layer wrapping `sync_backend`.
853 /// Composed alongside the existing typed surface (nodes, cells,
854 /// tracks, …) so callers can reach generic publish/get/query/observe
855 /// without going through type-specific JNI methods. Foundation step 3 of
856 /// the peat-mesh-completion / peat-btle-reduction work — see
857 /// `PEAT-MESH-COMPLETION-0.9.0.md`.
858 #[cfg(feature = "sync")]
859 node: Arc<peat_mesh::Node>,
860 /// peat-protocol's [`BleTranslator`] (ADR-041) used by the `ingest*Jni`
861 /// family of methods. Translates typed BLE structs to Automerge
862 /// documents; the result is published into [`Self::node`] with
863 /// `Some("ble")` origin so ADR-059's same-node echo suppression keeps
864 /// the doc from being re-encoded back out to BLE. The earlier
865 /// `BleGateway` wrapper composing translator + node was removed in
866 /// Slice 1.b.2.2 — composition happens inline in the JNI helpers
867 /// because peat-ffi owns both halves anyway, so the wrapper added no
868 /// boundary worth defending.
869 ///
870 /// [`BleTranslator`]: peat_protocol::sync::ble_translation::BleTranslator
871 #[cfg(all(feature = "sync", feature = "bluetooth"))]
872 ble_translator: Arc<peat_protocol::sync::ble_translation::BleTranslator>,
873 /// Transport manager for multi-transport coordination (ADR-032)
874 /// Enables PACE policy-based transport selection and future BLE integration
875 transport_manager: TransportManager,
876 /// Direct reference to Iroh transport for backward-compatible methods
877 /// (peer_count, connected_peers, etc.)
878 iroh_transport: Arc<IrohTransport>,
879 /// Store reference for subscriptions
880 store: Arc<AutomergeStore>,
881 #[allow(dead_code)] // Kept for potential future use (e.g., storage cleanup)
882 storage_path: PathBuf,
883 /// Persistent roster of known group peers, for reconnection. Keyed by
884 /// node_id, scoped by group; holds only non-secret reachability hints.
885 /// Foundation for the per-peer reconnect supervisor (walks this roster to
886 /// keep a live path up to each known member over whatever transport is
887 /// currently reachable). See `roster.rs`.
888 #[cfg(feature = "sync")]
889 roster: Arc<roster::RosterStore>,
890 /// Per-peer reconnect supervisor: the dial state machine + backoff policy
891 /// driven by a periodic tick (and, later, external event hooks) to keep a
892 /// live path up to each roster member. See `supervisor.rs`.
893 #[cfg(feature = "sync")]
894 supervisor: Arc<supervisor::Supervisor>,
895 /// Caps concurrent in-flight reconnect dials so a cold start with a large
896 /// saved roster (or a `wake_reconnect` that clears every backoff) doesn't
897 /// fan out one simultaneous iroh dial + handshake per peer. See
898 /// [`MAX_CONCURRENT_RECONNECT_DIALS`].
899 #[cfg(feature = "sync")]
900 reconnect_dial_semaphore: Arc<tokio::sync::Semaphore>,
901 /// Tokio runtime for async operations
902 runtime: Arc<tokio::runtime::Runtime>,
903 /// Flag to stop cleanup task on drop (used by background task)
904 #[allow(dead_code)]
905 cleanup_running: Arc<AtomicBool>,
906 /// Optional blob store running on a parallel iroh endpoint (ADR-060).
907 /// None when blob transfer is disabled — this is the common case for
908 /// sim nodes that don't need to serve or fetch binary payloads.
909 /// Constructed via PeatNode::enable_blob_transfer() after node creation.
910 #[cfg(feature = "sync")]
911 blob_store: std::sync::RwLock<Option<Arc<NetworkedIrohBlobStore>>>,
912 /// Queue of outbound BLE frames produced by the `BleTranslator` fan-out.
913 /// Populated by `QueueOutboundSink::send_outbound`; drained by
914 /// `poll_outbound_frames`. None when the `bluetooth` feature is off.
915 #[cfg(all(feature = "sync", feature = "bluetooth"))]
916 outbound_queue: Arc<std::sync::Mutex<std::collections::VecDeque<OutboundFrame>>>,
917 /// `FanoutHandle` for the active outbound subscription, if any.
918 /// Held alive between `start_outbound_frames` and `stop_outbound_frames`.
919 #[cfg(all(feature = "sync", feature = "bluetooth"))]
920 outbound_fanout: std::sync::Mutex<Option<peat_mesh::transport::FanoutHandle>>,
921 /// Dedup set for BLE multi-hop relay: frame-hash -> last-relayed instant.
922 ///
923 /// peat-mesh's fan-out re-fans an ingested frame to OTHER transports but
924 /// SUPPRESSES same-transport (BLE->BLE) re-emit to avoid a broadcast loop
925 /// (ADR-059 echo-suppression). That suppression also blocks legitimate
926 /// multi-hop relay in an all-BLE topology (A -> B -> C): B applies A's
927 /// frame but never forwards it to C, so C can stay permanently stale.
928 /// We re-emit each freshly-ingested frame onto the BLE outbound queue
929 /// so B relays it to C. The dedup (bounded, TTL-swept) throttles
930 /// identical re-advertises so a relayed frame isn't re-broadcast in a
931 /// loop — a NEW value (different bytes) always relays immediately;
932 /// redundant re-adverts within the TTL are dropped. See
933 /// peat#978-adjacent relay gap.
934 #[cfg(all(feature = "sync", feature = "bluetooth"))]
935 relay_seen: std::sync::Mutex<std::collections::HashMap<u64, std::time::Instant>>,
936 /// Shared water-supply Counter (CRDT-over-Automerge-over-BLE).
937 /// Self-contained Automerge doc; its save() bytes ride the BLE frame
938 /// bus and merge natively.
939 #[cfg(feature = "sync")]
940 water_counter: water_counter::WaterCounter,
941 /// Generic CRDT KV documents (nodes/commands/cells/mission), Automerge over
942 /// the same crdt frame as the counter — mesh-wide convergence, no
943 /// lite-bridge.
944 #[cfg(feature = "sync")]
945 crdt_kv: crdt_kv::CrdtKvDocs,
946}
947
948/// Shared connect → formation-handshake → sync-trigger pipeline backing both
949/// [`PeatNode::connect_peer`] (blocking) and [`PeatNode::connect_peer_nowait`]
950/// (fire-and-forget).
951///
952/// Resolves once the dial and optional formation handshake complete: on success
953/// the peer is emitted as Connected and a delayed document-sync sweep is
954/// spawned. The error is returned to the caller, which decides whether to
955/// propagate it (blocking variant) or log-and-drop it (fire-and-forget
956/// variant). Keeping the body here means the two entry points cannot drift.
957#[cfg(feature = "sync")]
958async fn connect_peer_inner(
959 iroh_transport: &IrohTransport,
960 sync_backend: &AutomergeIrohBackend,
961 storage_backend: &AutomergeBackend,
962 peat_peer: PeatPeerInfo,
963) -> Result<(), PeatError> {
964 let conn_opt = iroh_transport
965 .connect_peer(&peat_peer)
966 .await
967 .map_err(|e| PeatError::ConnectionError { msg: e.to_string() })?;
968
969 // None: no new connection — the accept path is handling it.
970 let Some(conn) = conn_opt else {
971 return Ok(());
972 };
973 let peer_id = conn.remote_id();
974
975 let Some(formation_key) = sync_backend.formation_key() else {
976 // No formation key — emit Connected without handshake (backward compat).
977 iroh_transport.emit_peer_connected(peer_id);
978 return Ok(());
979 };
980
981 use peat_protocol::network::perform_initiator_handshake;
982 if let Err(e) = perform_initiator_handshake(&conn, &formation_key).await {
983 conn.close(1u32.into(), b"authentication failed");
984 iroh_transport.disconnect(&peer_id).ok();
985 return Err(PeatError::ConnectionError {
986 msg: format!("Formation handshake failed: {}", e),
987 });
988 }
989
990 // Emit Connected to trigger immediate sync handler spawning.
991 iroh_transport.emit_peer_connected(peer_id);
992
993 // Explicitly trigger document sync with the new peer. The event-based sync
994 // handler spawner should also handle this, but we trigger directly to
995 // ensure documents flow.
996 if let Some(coordinator) = storage_backend.sync_coordinator() {
997 let coord = Arc::clone(coordinator);
998 tokio::spawn(async move {
999 // Brief delay for the connection to stabilize.
1000 tokio::time::sleep(tokio::time::Duration::from_millis(500)).await;
1001 match coord.sync_all_documents_with_peer(peer_id).await {
1002 Ok(()) => {
1003 tracing::debug!(peer = ?peer_id, "sync_all_documents_with_peer succeeded");
1004 #[cfg(target_os = "android")]
1005 android_log("sync_all_documents_with_peer: SUCCESS");
1006 }
1007 Err(e) => {
1008 tracing::warn!(peer = ?peer_id, error = %e, "sync_all_documents_with_peer failed");
1009 #[cfg(target_os = "android")]
1010 android_log(&format!("sync_all_documents_with_peer: FAILED - {}", e));
1011 }
1012 }
1013 });
1014 }
1015
1016 Ok(())
1017}
1018
1019#[cfg(feature = "sync")]
1020#[uniffi::export]
1021impl PeatNode {
1022 // ── Shared water-supply Counter (CRDT-over-Automerge-over-BLE) ──────────
1023 // The doc's save() bytes are carried over the BLE frame bus; merge is
1024 // commutative/idempotent, so the caller can broadcast/relay freely.
1025
1026 // The Automerge doc bytes cross the FFI as a HEX string (the well-trodden
1027 // String marshalling path; the doc is tiny so 2x size is irrelevant). The
1028 // caller broadcasts the hex over the BLE bridge and feeds inbound hex to
1029 // `crdt_counter_merge`.
1030
1031 /// Current merged value of the shared water-supply Counter.
1032 pub fn crdt_counter_value(&self) -> i64 {
1033 self.water_counter.value()
1034 }
1035
1036 /// Apply `delta` liters to the shared Counter; returns the doc's save()
1037 /// bytes (hex) for the caller to broadcast to peers.
1038 pub fn crdt_counter_increment(&self, delta: i64) -> String {
1039 hex::encode(self.water_counter.increment(delta))
1040 }
1041
1042 /// Merge an inbound peer doc (hex of its save() bytes); returns the new
1043 /// value. Safe with duplicate / stale / relayed / out-of-order input.
1044 pub fn crdt_counter_merge(&self, hex_doc: String) -> i64 {
1045 match hex::decode(hex_doc.trim()) {
1046 Ok(bytes) => self.water_counter.merge(&bytes),
1047 Err(_) => self.water_counter.value(),
1048 }
1049 }
1050
1051 /// Current save() bytes (hex), for periodic re-broadcast (catch-up).
1052 pub fn crdt_counter_snapshot(&self) -> String {
1053 hex::encode(self.water_counter.snapshot())
1054 }
1055
1056 // ── Generic CRDT KV documents (nodes/commands/cells/mission) ────────────
1057 // Records are key -> JSON-string in a per-collection Automerge doc; merge is
1058 // set-union across keys (LWW per key). Same crdt-frame transport as the
1059 // counter; doc bytes cross the FFI as hex.
1060
1061 /// Upsert `key = value_json` in `collection`; returns the doc's save()
1062 /// bytes (hex) to broadcast.
1063 pub fn crdt_kv_put(&self, collection: String, key: String, value_json: String) -> String {
1064 hex::encode(self.crdt_kv.put(&collection, &key, &value_json))
1065 }
1066
1067 /// All records in `collection` as a JSON object `{key: value}`.
1068 pub fn crdt_kv_all(&self, collection: String) -> String {
1069 self.crdt_kv.all_json(&collection)
1070 }
1071
1072 /// Merge an inbound peer doc (hex) into `collection`.
1073 pub fn crdt_kv_merge(&self, collection: String, hex_doc: String) {
1074 if let Ok(bytes) = hex::decode(hex_doc.trim()) {
1075 self.crdt_kv.merge(&collection, &bytes);
1076 }
1077 }
1078
1079 /// Current save() bytes (hex) of `collection`, for periodic re-broadcast.
1080 pub fn crdt_kv_snapshot(&self, collection: String) -> String {
1081 hex::encode(self.crdt_kv.snapshot(&collection))
1082 }
1083
1084 /// Get this node's unique identifier (hex-encoded)
1085 pub fn node_id(&self) -> String {
1086 hex::encode(self.iroh_transport.endpoint_id().as_bytes())
1087 }
1088
1089 /// Get this node's endpoint address for peer connections
1090 pub fn endpoint_addr(&self) -> String {
1091 format!("{:?}", self.iroh_transport.endpoint_addr())
1092 }
1093
1094 /// Get the number of connected peers
1095 pub fn peer_count(&self) -> u32 {
1096 self.iroh_transport.peer_count() as u32
1097 }
1098
1099 /// Get list of connected peer IDs
1100 pub fn connected_peers(&self) -> Vec<String> {
1101 self.iroh_transport
1102 .connected_peers()
1103 .iter()
1104 .map(|id| hex::encode(id.as_bytes()))
1105 .collect()
1106 }
1107
1108 /// Return this node's iroh-endpoint first IP listening address
1109 /// as an `"ip:port"` string, or `None` if no socket has been
1110 /// bound yet.
1111 ///
1112 /// Intended for two-instance instrumented tests where two nodes
1113 /// in the same process need to dial each other on loopback —
1114 /// neither has the other's address from discovery, so the test
1115 /// harness fetches it here and passes it to `connectPeerJni` on
1116 /// the dialing side. peat-mesh#138 M4.
1117 pub fn endpoint_socket_addr(&self) -> Option<String> {
1118 self.iroh_transport.bound_socket_addr_string()
1119 }
1120
1121 /// Start sync operations
1122 ///
1123 /// The authenticated accept loop (with formation handshake) is already
1124 /// running from sync_backend.initialize() in create_node(). This method
1125 /// starts the sync coordination layer: event-based and polling-based
1126 /// sync handlers.
1127 pub fn start_sync(&self) -> Result<(), PeatError> {
1128 #[cfg(target_os = "android")]
1129 android_log("start_sync: called");
1130
1131 // IMPORTANT: Use runtime.enter() to ensure tokio::spawn() inside start_sync()
1132 // can find the runtime context. block_on() alone doesn't guarantee this on
1133 // all platforms (especially Android where the JNI thread may not have proper
1134 // thread-local storage for the Tokio runtime handle).
1135 let _guard = self.runtime.enter();
1136
1137 #[cfg(target_os = "android")]
1138 android_log("start_sync: runtime entered");
1139
1140 // Must run inside Tokio runtime because start_sync() calls tokio::spawn()
1141 let result = self.runtime.block_on(async {
1142 #[cfg(target_os = "android")]
1143 android_log("start_sync: inside block_on");
1144
1145 // CRITICAL: Call start_sync() on the ACTUAL storage_backend instance,
1146 // NOT on sync_backend.sync_engine() which returns a CLONED instance
1147 // that doesn't have the transport event subscriptions set up!
1148 //
1149 // Note: The authenticated accept loop (with formation handshake and
1150 // Connected event emission) is already running — it was started by
1151 // sync_backend.initialize() in create_node(). The storage_backend's
1152 // start_sync() will see the accept loop as already running and skip
1153 // starting the plain (unauthenticated) accept loop.
1154 self.storage_backend
1155 .start_sync()
1156 .map_err(|e| PeatError::SyncError { msg: e.to_string() })
1157 });
1158
1159 #[cfg(target_os = "android")]
1160 match &result {
1161 Ok(_) => android_log("start_sync: SUCCESS - sync handlers spawned"),
1162 Err(e) => android_log(&format!("start_sync: FAILED - {}", e)),
1163 }
1164
1165 result
1166 }
1167
1168 /// Stop sync operations
1169 pub fn stop_sync(&self) -> Result<(), PeatError> {
1170 // Must run inside Tokio runtime for consistency with start_sync()
1171 self.runtime.block_on(async {
1172 self.storage_backend
1173 .stop_sync()
1174 .map_err(|e| PeatError::SyncError { msg: e.to_string() })
1175 })
1176 }
1177
1178 /// Get sync statistics
1179 pub fn sync_stats(&self) -> Result<SyncStats, PeatError> {
1180 let stats = self
1181 .storage_backend
1182 .sync_stats()
1183 .map_err(|e| PeatError::SyncError { msg: e.to_string() })?;
1184
1185 Ok(SyncStats {
1186 sync_active: stats.peer_count > 0, // Infer from peer count
1187 connected_peers: self.iroh_transport.peer_count() as u32,
1188 bytes_sent: stats.bytes_sent,
1189 bytes_received: stats.bytes_received,
1190 })
1191 }
1192
1193 /// ADR-032 §Amendment A — unified per-peer transport state.
1194 ///
1195 /// Walks `TransportManager` for the given peer, calls
1196 /// `peer_link_state` on each registered transport that can reach
1197 /// it, and overlays the registered `TransportInstance.id` onto the
1198 /// returned `LinkState.transport_id` (per the host-rendering rule:
1199 /// the producer doesn't know its own registered id, the consumer
1200 /// fills it). Returns `Ok(PeerTransportState { peer_id, links: vec![] })`
1201 /// for peers no transport reports — "absence is a valid state."
1202 ///
1203 /// Hex-encoded `peer_id` matches the form `connected_peers()`
1204 /// returns. Invalid hex is propagated as-is to peat-mesh's
1205 /// `NodeId::new`, which is also a `String` wrapper — invalid input
1206 /// surfaces as an empty `links` vec rather than an error, matching
1207 /// the absence contract.
1208 pub fn peer_transport_state(&self, peer_id: String) -> Result<PeerTransportState, PeatError> {
1209 let mesh_peer = peat_mesh::NodeId::new(peer_id.clone());
1210 let links = self
1211 .transport_manager
1212 .available_instances_for_peer(&mesh_peer)
1213 .into_iter()
1214 .filter_map(|transport_id| {
1215 let transport = self.transport_manager.get_instance(&transport_id)?;
1216 let mut state = transport.peer_link_state(&mesh_peer)?;
1217 // Host-rendering rule: overlay the registered id onto
1218 // the producer's placeholder. See
1219 // `peat_mesh::transport::btle::BLE_TRANSPORT_ID_PLACEHOLDER`.
1220 state.transport_id = transport_id;
1221 Some(TransportLink::from(state))
1222 })
1223 .collect();
1224 Ok(PeerTransportState { peer_id, links })
1225 }
1226
1227 /// ADR-032 §Amendment A — transport state for the peer set this
1228 /// `peat-ffi` instance currently enumerates from iroh.
1229 ///
1230 /// Designed for the plugin's periodic poll (~2 s) — the
1231 /// implementation walks transport state in a single pass without
1232 /// per-peer recursion.
1233 ///
1234 /// **Coverage caveat (Slice-4.d-interim — not the final SSOT
1235 /// shape).** This method enumerates peers exclusively from
1236 /// `self.iroh_transport.connected_peers()`. BLE-only peers
1237 /// (peers reachable via peat-btle but not currently visible to
1238 /// iroh) are **not** included. Plugin authors must continue to
1239 /// merge BLE-only peers from peat-btle's UniFFI surface
1240 /// directly until the single-source-of-truth migration
1241 /// completes. The Amendment A SSOT promise — "peat-ffi is the
1242 /// single source of truth, the plugin MUST NOT reach into
1243 /// peat-btle's UniFFI directly" — is the destination, not the
1244 /// current implementation; this method's coverage is a strict
1245 /// subset of that destination. Treat the cross-FFI peat-btle
1246 /// reach as a documented interim, not an idiom to standardize on.
1247 /// Tracked under defenseunicorns/peat#828.
1248 pub fn all_peer_transport_states(&self) -> Result<Vec<PeerTransportState>, PeatError> {
1249 // Collect a deduped peer set across registered transports.
1250 // peat-mesh's TransportManager doesn't expose a single
1251 // "all known peers" iterator, so we union over registered
1252 // instance peers via `iroh_transport.connected_peers()` for
1253 // the iroh side (the only transport peat-ffi currently
1254 // surfaces directly). BLE-side peers come through the
1255 // bluetooth feature's transport registration; their
1256 // connected_peers are surfaced through the same walk on
1257 // peer_transport_state once the caller knows their id from
1258 // the BLE-side UniFFI lookup. For now this method covers
1259 // peers visible to iroh; the plugin merges BLE-only peers
1260 // from its peat-btle UniFFI consumer separately while the
1261 // single-source-of-truth migration completes.
1262 let mut peer_ids: Vec<String> = self
1263 .iroh_transport
1264 .connected_peers()
1265 .iter()
1266 .map(|id| hex::encode(id.as_bytes()))
1267 .collect();
1268 peer_ids.sort();
1269 peer_ids.dedup();
1270
1271 let mut out = Vec::with_capacity(peer_ids.len());
1272 for peer_id in peer_ids {
1273 out.push(self.peer_transport_state(peer_id)?);
1274 }
1275 Ok(out)
1276 }
1277
1278 /// Request a full document sync with all connected peers.
1279 /// This pushes all local documents to each peer and pulls any documents
1280 /// they have. Useful for ensuring newly created documents propagate
1281 /// after the initial connection.
1282 pub fn request_sync(&self) -> Result<(), PeatError> {
1283 if let Some(coordinator) = self.storage_backend.sync_coordinator() {
1284 let peers = self.iroh_transport.connected_peers();
1285 let peer_count = peers.len();
1286 // Logcat-visible signal of every request_sync invocation:
1287 // peer count + each push's success/failure. peat-protocol's
1288 // internal `tracing::info!` doesn't reach logcat because no
1289 // tracing-subscriber is installed on Android, so the only
1290 // way to observe whether `sync_all_documents_with_peer`
1291 // actually ran is to surface it here at the FFI boundary
1292 // where `android_log` works.
1293 #[cfg(target_os = "android")]
1294 android_log(&format!(
1295 "request_sync: starting with {} connected peer(s)",
1296 peer_count
1297 ));
1298 let coord = Arc::clone(coordinator);
1299 self.runtime.block_on(async {
1300 for peer_id in peers {
1301 match coord.sync_all_documents_with_peer(peer_id).await {
1302 Ok(()) => {
1303 #[cfg(target_os = "android")]
1304 {
1305 let peer_hex = hex::encode(peer_id.as_bytes());
1306 android_log(&format!(
1307 "request_sync: pushed to peer {}",
1308 &peer_hex[..16]
1309 ));
1310 }
1311 }
1312 Err(_e) => {
1313 #[cfg(target_os = "android")]
1314 {
1315 let peer_hex = hex::encode(peer_id.as_bytes());
1316 android_log(&format!(
1317 "request_sync: FAILED for peer {}: {}",
1318 &peer_hex[..16],
1319 _e
1320 ));
1321 }
1322 }
1323 }
1324 }
1325 });
1326 #[cfg(target_os = "android")]
1327 android_log(&format!(
1328 "request_sync: complete ({} peer(s) attempted)",
1329 peer_count
1330 ));
1331 }
1332 Ok(())
1333 }
1334
1335 /// Connect to a peer node with formation handshake
1336 ///
1337 /// Establishes a QUIC connection, performs formation-key authentication,
1338 /// and emits a Connected event to trigger immediate sync handler spawning.
1339 pub fn connect_peer(&self, peer: PeerInfo) -> Result<(), PeatError> {
1340 let peat_peer = PeatPeerInfo {
1341 name: peer.name,
1342 node_id: peer.node_id,
1343 addresses: peer.addresses,
1344 relay_url: peer.relay_url,
1345 };
1346
1347 let _guard = self.runtime.enter();
1348 self.runtime.block_on(connect_peer_inner(
1349 &self.iroh_transport,
1350 &self.sync_backend,
1351 &self.storage_backend,
1352 peat_peer,
1353 ))
1354 }
1355
1356 /// Connect to a peer WITHOUT blocking the caller.
1357 ///
1358 /// Same connect + formation-handshake + sync-trigger as [`connect_peer`]
1359 /// (they share [`connect_peer_inner`]), but spawned on the runtime so the
1360 /// FFI call returns immediately. The dial completes in the background; on
1361 /// success the peer appears in `connected_peers` and a Connected event
1362 /// fires. Intended for UI callers: a Dart isolate blocks on a synchronous
1363 /// FFI call, so `connect_peer`'s `block_on` freezes the UI for the whole
1364 /// dial (~seconds for an unreachable peer). There is no synchronous caller
1365 /// to hand a background failure to, so errors are surfaced via `tracing`
1366 /// (and `android_log` on Android) and otherwise dropped.
1367 pub fn connect_peer_nowait(&self, peer: PeerInfo) -> Result<(), PeatError> {
1368 let peat_peer = PeatPeerInfo {
1369 name: peer.name,
1370 node_id: peer.node_id,
1371 addresses: peer.addresses,
1372 relay_url: peer.relay_url,
1373 };
1374 let iroh_transport = Arc::clone(&self.iroh_transport);
1375 let sync_backend = Arc::clone(&self.sync_backend);
1376 let storage_backend = Arc::clone(&self.storage_backend);
1377
1378 self.runtime.spawn(async move {
1379 if let Err(e) =
1380 connect_peer_inner(&iroh_transport, &sync_backend, &storage_backend, peat_peer)
1381 .await
1382 {
1383 tracing::warn!(error = %e, "connect_peer_nowait: background dial failed");
1384 #[cfg(target_os = "android")]
1385 android_log(&format!("connect_peer_nowait: {}", e));
1386 }
1387 });
1388
1389 Ok(())
1390 }
1391
1392 /// Disconnect from a peer by node ID
1393 ///
1394 /// Note: Currently disconnects matching peer from internal connection map.
1395 pub fn disconnect_peer(&self, node_id: &str) -> Result<(), PeatError> {
1396 // Find the matching endpoint ID from connected peers
1397 let connected = self.iroh_transport.connected_peers();
1398 for endpoint_id in connected {
1399 if hex::encode(endpoint_id.as_bytes()) == node_id {
1400 return self
1401 .iroh_transport
1402 .disconnect(&endpoint_id)
1403 .map_err(|e| PeatError::ConnectionError { msg: e.to_string() });
1404 }
1405 }
1406
1407 Err(PeatError::ConnectionError {
1408 msg: format!("Peer {} not found in connected peers", node_id),
1409 })
1410 }
1411
1412 /// Store a JSON document in a collection
1413 pub fn put_document(
1414 &self,
1415 collection: &str,
1416 doc_id: &str,
1417 json_data: &str,
1418 ) -> Result<(), PeatError> {
1419 // Parse JSON to validate it
1420 let _: serde_json::Value =
1421 serde_json::from_str(json_data).map_err(|e| PeatError::InvalidInput {
1422 msg: format!("Invalid JSON: {}", e),
1423 })?;
1424
1425 self.runtime.block_on(async {
1426 let backend = &self.storage_backend;
1427 let coll = backend.collection(collection);
1428
1429 coll.upsert(doc_id, json_data.as_bytes().to_vec())
1430 .map_err(|e| PeatError::StorageError { msg: e.to_string() })
1431 })
1432 }
1433
1434 /// Retrieve a document from the **raw-bytes store** as JSON.
1435 ///
1436 /// # Storage path
1437 ///
1438 /// This reads from `storage_backend.collection()` — the raw
1439 /// key-value store. It will NOT see documents that were:
1440 ///
1441 /// - Published via `publishDocumentJni` (which goes through
1442 /// `peat_mesh::Node::publish`, the document layer)
1443 /// - Received from a peer via Automerge sync (which writes into the
1444 /// document layer's CRDT, not the raw store)
1445 ///
1446 /// The JNI counterpart `getDocumentJni` deliberately uses
1447 /// `peat_mesh::Node::get()` instead so it round-trips with
1448 /// `publishDocumentJni`. If you're writing a new JNI method
1449 /// that reads documents published or synced via the document
1450 /// layer, follow `getDocumentJni`'s pattern, not this method's.
1451 pub fn get_document(
1452 &self,
1453 collection: &str,
1454 doc_id: &str,
1455 ) -> Result<Option<String>, PeatError> {
1456 self.runtime.block_on(async {
1457 let backend = &self.storage_backend;
1458 let coll = backend.collection(collection);
1459
1460 match coll.get(doc_id) {
1461 Ok(Some(bytes)) => {
1462 let json = String::from_utf8(bytes).map_err(|e| PeatError::StorageError {
1463 msg: format!("Invalid UTF-8: {}", e),
1464 })?;
1465 Ok(Some(json))
1466 }
1467 Ok(None) => Ok(None),
1468 Err(e) => Err(PeatError::StorageError { msg: e.to_string() }),
1469 }
1470 })
1471 }
1472
1473 /// Delete a document from a collection
1474 pub fn delete_document(&self, collection: &str, doc_id: &str) -> Result<(), PeatError> {
1475 self.runtime.block_on(async {
1476 let backend = &self.storage_backend;
1477 let coll = backend.collection(collection);
1478
1479 coll.delete(doc_id)
1480 .map_err(|e| PeatError::StorageError { msg: e.to_string() })
1481 })
1482 }
1483
1484 /// List all document IDs in a collection
1485 pub fn list_documents(&self, collection: &str) -> Result<Vec<String>, PeatError> {
1486 self.runtime.block_on(async {
1487 let backend = &self.storage_backend;
1488 let coll = backend.collection(collection);
1489
1490 let docs = coll
1491 .scan()
1492 .map_err(|e| PeatError::StorageError { msg: e.to_string() })?;
1493
1494 Ok(docs.into_iter().map(|(id, _)| id).collect())
1495 })
1496 }
1497
1498 /// Manually trigger sync for a specific document
1499 pub fn sync_document(&self, collection: &str, doc_id: &str) -> Result<(), PeatError> {
1500 let doc_key = format!("{}:{}", collection, doc_id);
1501
1502 self.runtime.block_on(async {
1503 let backend = &self.storage_backend;
1504
1505 backend
1506 .sync_document(&doc_key)
1507 .await
1508 .map_err(|e| PeatError::SyncError { msg: e.to_string() })
1509 })
1510 }
1511
1512 /// Subscribe to document changes
1513 ///
1514 /// Returns a SubscriptionHandle that must be kept alive to receive
1515 /// callbacks. When the handle is dropped or cancel() is called, the
1516 /// subscription stops.
1517 ///
1518 /// The callback will receive DocumentChange events for all documents.
1519 /// Filter by collection in your callback implementation if needed.
1520 ///
1521 /// Note: Only one subscription per node is supported. Calling subscribe
1522 /// again will fail if a subscription is already active.
1523 pub fn subscribe(
1524 &self,
1525 callback: Box<dyn DocumentCallback>,
1526 ) -> Result<Arc<SubscriptionHandle>, PeatError> {
1527 // Subscribe to ALL changes (local + peer-synced). Same origin-based dedup
1528 // as subscribe_poll: Remote events only fire the first time a doc_key is seen.
1529 let change_rx = self.store.subscribe_to_changes_with_origin();
1530
1531 // Create active flag for the subscription
1532 let active = Arc::new(AtomicBool::new(true));
1533 let active_clone = Arc::clone(&active);
1534 // Spawn a task to listen for changes and call the callback.
1535 // Dedup is handled at the Dart layer via content hashing — emit all
1536 // events here so cross-device updates are never silently dropped.
1537 let callback = Arc::new(callback);
1538 self.runtime.spawn(async move {
1539 let mut rx = change_rx;
1540
1541 while active_clone.load(Ordering::SeqCst) {
1542 tokio::select! {
1543 result = rx.recv() => {
1544 match result {
1545 Ok(doc_change) => {
1546 // Generic, origin-tagged change event — consumer
1547 // decides notability (e.g. notify on Remote).
1548 callback.on_change(document_change_from(doc_change));
1549 }
1550 Err(tokio::sync::broadcast::error::RecvError::Lagged(n)) => {
1551 // Some messages were skipped due to slow receiver
1552 callback.on_error(format!("Lagged {} messages", n));
1553 }
1554 Err(tokio::sync::broadcast::error::RecvError::Closed) => {
1555 // Channel closed
1556 callback.on_error("Document change channel closed".to_string());
1557 break;
1558 }
1559 }
1560 }
1561 _ = tokio::time::sleep(tokio::time::Duration::from_millis(100)) => {
1562 // Periodic check if we should stop
1563 if !active_clone.load(Ordering::SeqCst) {
1564 break;
1565 }
1566 }
1567 }
1568 }
1569 });
1570
1571 Ok(Arc::new(SubscriptionHandle::new(active)))
1572 }
1573
1574 /// Subscribe to document changes using a poll-based model.
1575 ///
1576 /// Returns a [`SubscriptionHandle`] whose
1577 /// [`SubscriptionHandle::poll_changes`] method drains buffered
1578 /// [`DocumentChange`] events. Callers drive delivery by periodically
1579 /// calling `poll_changes` (e.g. from a Dart isolate loop or
1580 /// `Timer.periodic`) — no foreign callback interface is required.
1581 ///
1582 /// Drop or call [`SubscriptionHandle::cancel`] on the handle to stop.
1583 ///
1584 /// # Broadcast lag
1585 ///
1586 /// The underlying channel has a bounded capacity. If `poll_changes` is not
1587 /// called frequently enough relative to the document-change rate, the
1588 /// broadcast channel will lag and silently drop events — `poll_changes`
1589 /// returns a partial set with no indication that events were missed.
1590 /// Callers should treat a long gap between `poll_changes` calls (e.g. the
1591 /// app was backgrounded) as a signal to trigger a full collection resync
1592 /// rather than relying on the change stream alone.
1593 pub fn subscribe_poll(&self) -> Result<Arc<SubscriptionHandle>, PeatError> {
1594 // Subscribe to ALL changes (local + peer-synced) via the origin-tagged channel.
1595 //
1596 // The gossip channel fires on every Automerge sync protocol exchange, including
1597 // redundant re-syncs of unchanged documents. To prevent a sync loop (periodic
1598 // requestSync re-fires Remote events for every already-known doc), we apply
1599 // origin-based deduplication:
1600 // Emit all events — dedup is handled in the Dart layer via content
1601 // hashing so cross-device updates (including repeated increments)
1602 // are never silently dropped by the Rust subscription.
1603 let change_rx = self.store.subscribe_to_changes_with_origin();
1604 let active = Arc::new(AtomicBool::new(true));
1605 let active_clone = Arc::clone(&active);
1606 let pending = Arc::new(std::sync::Mutex::new(std::collections::VecDeque::<
1607 DocumentChange,
1608 >::new()));
1609 let pending_clone = Arc::clone(&pending);
1610
1611 self.runtime.spawn(async move {
1612 let mut rx = change_rx;
1613 while active_clone.load(Ordering::SeqCst) {
1614 tokio::select! {
1615 result = rx.recv() => {
1616 match result {
1617 Ok(doc_change) => {
1618 if let Ok(mut q) = pending_clone.lock() {
1619 q.push_back(document_change_from(doc_change));
1620 }
1621 }
1622 Err(tokio::sync::broadcast::error::RecvError::Closed) => break,
1623 Err(tokio::sync::broadcast::error::RecvError::Lagged(_)) => {}
1624 }
1625 }
1626 _ = tokio::time::sleep(tokio::time::Duration::from_millis(100)) => {
1627 if !active_clone.load(Ordering::SeqCst) {
1628 break;
1629 }
1630 }
1631 }
1632 }
1633 });
1634
1635 Ok(Arc::new(SubscriptionHandle::new_with_queue(
1636 active, pending,
1637 )))
1638 }
1639}
1640
1641/// Create a new PeatNode with FormationKey authentication
1642///
1643/// Requires `app_id` and `shared_key` for peer authentication.
1644/// Only peers with matching credentials can connect and sync.
1645///
1646/// # Arguments
1647///
1648/// * `config` - Node configuration including:
1649/// - `app_id`: Formation/application identifier (use same value for all nodes
1650/// in your swarm)
1651/// - `shared_key`: Base64-encoded 32-byte secret key (generate with `openssl
1652/// rand -base64 32`)
1653/// - `bind_address`: Optional address to bind (default: "0.0.0.0:0")
1654/// - `storage_path`: Directory for persistent storage
1655///
1656/// Note: This function is NOT async because we manage our own Tokio runtime
1657/// to ensure proper context for Iroh transport operations.
1658#[cfg(feature = "sync")]
1659#[uniffi::export]
1660pub fn create_node(config: NodeConfig) -> Result<Arc<PeatNode>, PeatError> {
1661 use std::time::Instant;
1662 let total_start = Instant::now();
1663
1664 // Validate credentials
1665 if config.app_id.is_empty() {
1666 return Err(PeatError::InvalidInput {
1667 msg: "app_id cannot be empty".to_string(),
1668 });
1669 }
1670 if config.shared_key.is_empty() {
1671 return Err(PeatError::InvalidInput {
1672 msg: "shared_key cannot be empty".to_string(),
1673 });
1674 }
1675
1676 // Helper: read RSS from /proc/self/status
1677 fn get_rss_kb() -> u64 {
1678 std::fs::read_to_string("/proc/self/status")
1679 .ok()
1680 .and_then(|s| {
1681 s.lines()
1682 .find(|l| l.starts_with("VmRSS:"))
1683 .and_then(|l| l.split_whitespace().nth(1))
1684 .and_then(|v| v.parse().ok())
1685 })
1686 .unwrap_or(0)
1687 }
1688
1689 #[cfg(target_os = "android")]
1690 android_log(&format!("[MEM] Before runtime: {} kB", get_rss_kb()));
1691
1692 // TIMING: Create runtime
1693 let phase_start = Instant::now();
1694
1695 // Create a dedicated Tokio runtime for this node
1696 // Use 4 worker threads to avoid starving BLE D-Bus tasks when Iroh
1697 // background tasks (discovery, relay, pkarr) are running concurrently.
1698 let runtime = tokio::runtime::Builder::new_multi_thread()
1699 .worker_threads(4)
1700 .enable_all()
1701 .build()
1702 .map_err(|e| PeatError::SyncError {
1703 msg: format!("Failed to create runtime: {}", e),
1704 })?;
1705
1706 let runtime_ms = phase_start.elapsed().as_millis();
1707 #[cfg(target_os = "android")]
1708 android_log(&format!("[TIMING] Runtime creation: {}ms", runtime_ms));
1709 #[cfg(target_os = "android")]
1710 android_log(&format!("[MEM] After runtime: {} kB", get_rss_kb()));
1711 #[cfg(not(target_os = "android"))]
1712 eprintln!("[Peat TIMING] Runtime creation: {}ms", runtime_ms);
1713
1714 // Parse bind address
1715 let bind_addr: SocketAddr = config
1716 .bind_address
1717 .as_deref()
1718 .unwrap_or("0.0.0.0:0")
1719 .parse()
1720 .map_err(|e| PeatError::InvalidInput {
1721 msg: format!("Invalid bind address: {}", e),
1722 })?;
1723
1724 // Create storage path
1725 let storage_path = PathBuf::from(&config.storage_path);
1726 std::fs::create_dir_all(&storage_path).map_err(|e| PeatError::StorageError {
1727 msg: format!("Failed to create storage directory: {}", e),
1728 })?;
1729
1730 // TIMING: Parallel store + transport initialization
1731 let phase_start = Instant::now();
1732
1733 // OPTIMIZATION: Run store opening and transport creation in parallel
1734 // These are independent operations that can overlap to reduce startup time.
1735 // - AutomergeStore::open() is blocking I/O (redb database)
1736 // - IrohTransport creation is async (QUIC endpoint binding)
1737 //
1738 // OPTIMIZATION: Use fast constructor WITHOUT mDNS discovery for faster startup.
1739 // mDNS discovery is deferred until after the sync backend is initialized.
1740 // This reduces "startup intensity" that was causing Docker API timeouts
1741 // in large-scale deployments (see 384-node hierarchical simulations).
1742 let seed = format!("{}/{}", config.app_id, config.storage_path);
1743 let storage_path_for_store = storage_path.clone();
1744 // Runtime relay posture (peat-flutter relay toggle): opt into n0's hosted
1745 // public relay pool only when the caller asked for it. Defaults to the
1746 // local-only posture so unconfigured callers don't phone home.
1747 let enable_n0_relay = config
1748 .transport
1749 .as_ref()
1750 .map(|t| t.enable_n0_relay)
1751 .unwrap_or(false);
1752
1753 let (store, transport, store_ms, transport_ms) = runtime.block_on(async {
1754 let store_start = Instant::now();
1755 let transport_start = Instant::now();
1756
1757 // Spawn store opening on blocking thread pool (it does sync I/O).
1758 // Retry up to 10 times with 200 ms delays — the previous node's redb
1759 // file lock may not be released immediately when the user stops and
1760 // immediately restarts the node (background Arcs are still alive).
1761 let store_handle = tokio::task::spawn_blocking(move || {
1762 let mut last_err = None;
1763 // Retry for up to ~30 s — iOS background tasks can hold the redb
1764 // lock for longer than macOS before their Arcs are fully released.
1765 for _ in 0..60u32 {
1766 match AutomergeStore::open(&storage_path_for_store) {
1767 Ok(s) => return (Ok(s), store_start.elapsed().as_millis()),
1768 Err(e) => {
1769 last_err = Some(e);
1770 std::thread::sleep(std::time::Duration::from_millis(500));
1771 }
1772 }
1773 }
1774 (Err(last_err.unwrap()), store_start.elapsed().as_millis())
1775 });
1776
1777 // Create transport WITH mDNS discovery wired into the endpoint
1778 let transport_future = async {
1779 let result =
1780 IrohTransport::from_seed_with_discovery_at_addr(&seed, bind_addr, enable_n0_relay)
1781 .await;
1782 (result, transport_start.elapsed().as_millis())
1783 };
1784
1785 // Wait for both to complete
1786 let (store_result, transport_result) = tokio::join!(store_handle, transport_future);
1787
1788 // Unwrap the JoinHandle result first, then the actual result
1789 let (store_inner, store_elapsed) = store_result.map_err(|e| PeatError::StorageError {
1790 msg: format!("Store task panicked: {}", e),
1791 })?;
1792 let store = store_inner.map_err(|e| PeatError::StorageError {
1793 msg: format!("Failed to open store: {}", e),
1794 })?;
1795
1796 #[cfg(target_os = "android")]
1797 android_log(&format!(
1798 "[MEM] After store open: {} kB (store {}ms)",
1799 get_rss_kb(),
1800 store_elapsed
1801 ));
1802
1803 let (transport_inner, transport_elapsed) = transport_result;
1804 let transport = transport_inner.map_err(|e| PeatError::ConnectionError {
1805 msg: format!("Failed to create transport with mDNS: {}", e),
1806 })?;
1807
1808 #[cfg(target_os = "android")]
1809 android_log(&format!(
1810 "[MEM] After iroh transport: {} kB (transport {}ms)",
1811 get_rss_kb(),
1812 transport_elapsed
1813 ));
1814
1815 Ok::<_, PeatError>((
1816 Arc::new(store),
1817 Arc::new(transport),
1818 store_elapsed,
1819 transport_elapsed,
1820 ))
1821 })?;
1822
1823 let parallel_total_ms = phase_start.elapsed().as_millis();
1824 #[cfg(target_os = "android")]
1825 {
1826 android_log(&format!("[TIMING] Store open: {}ms", store_ms));
1827 android_log(&format!(
1828 "[TIMING] Transport create (with mDNS): {}ms",
1829 transport_ms
1830 ));
1831 android_log(&format!(
1832 "[TIMING] Parallel total (max of above): {}ms",
1833 parallel_total_ms
1834 ));
1835 }
1836 #[cfg(not(target_os = "android"))]
1837 {
1838 eprintln!("[Peat TIMING] Store open: {}ms", store_ms);
1839 eprintln!(
1840 "[Peat TIMING] Transport create (with mDNS): {}ms",
1841 transport_ms
1842 );
1843 eprintln!(
1844 "[Peat TIMING] Parallel total (max of above): {}ms",
1845 parallel_total_ms
1846 );
1847 }
1848
1849 // Create storage backend with transport
1850 let storage_backend = Arc::new(AutomergeBackend::with_transport(
1851 Arc::clone(&store),
1852 Arc::clone(&transport),
1853 ));
1854
1855 // Create sync backend (AutomergeIrohBackend) for authenticated P2P sync
1856 // Note: AutomergeIrohBackend wraps storage::AutomergeBackend for the
1857 // DataSyncBackend trait
1858 let sync_backend = Arc::new(AutomergeIrohBackend::new(
1859 Arc::clone(&storage_backend),
1860 Arc::clone(&transport),
1861 ));
1862
1863 // IMPORTANT (Issue #275): Subscribe to peer events BEFORE initializing sync
1864 // backend. The initialize() call spawns the accept loop, so we need to
1865 // subscribe first to catch all connection events including the initial
1866 // ones.
1867 let mut event_rx = transport.subscribe_peer_events();
1868
1869 // TIMING: Sync backend initialization
1870 let phase_start = Instant::now();
1871
1872 // Initialize sync backend with credentials for FormationKey authentication
1873 let backend_config = BackendConfig {
1874 app_id: config.app_id.clone(),
1875 persistence_dir: storage_path.clone(),
1876 shared_key: Some(config.shared_key.clone()),
1877 transport: TransportConfig::default(),
1878 extra: std::collections::HashMap::new(),
1879 };
1880
1881 runtime.block_on(async {
1882 sync_backend
1883 .initialize(backend_config)
1884 .await
1885 .map_err(|e| PeatError::SyncError {
1886 msg: format!("Failed to initialize sync backend: {}", e),
1887 })
1888 })?;
1889
1890 let sync_init_ms = phase_start.elapsed().as_millis();
1891 #[cfg(target_os = "android")]
1892 {
1893 android_log(&format!("[TIMING] Sync backend init: {}ms", sync_init_ms));
1894 android_log("=== sync_backend.initialize() completed successfully ===");
1895 }
1896 #[cfg(not(target_os = "android"))]
1897 eprintln!("[Peat TIMING] Sync backend init: {}ms", sync_init_ms);
1898
1899 // Start background task to listen for peer events and forward to Java (Issue
1900 // #275)
1901 let cleanup_running = Arc::new(AtomicBool::new(true));
1902 let cleanup_flag = Arc::clone(&cleanup_running);
1903 let runtime_arc = Arc::new(runtime);
1904
1905 // Clone transport for the cleanup task
1906 let transport_for_cleanup = Arc::clone(&transport);
1907
1908 // Log that we're starting the peer event listener
1909 #[cfg(target_os = "android")]
1910 android_log("Starting peer event listener task (Issue #275)");
1911
1912 runtime_arc.spawn(async move {
1913 #[cfg(target_os = "android")]
1914 android_log("Peer event listener task running");
1915
1916 while cleanup_flag.load(Ordering::Relaxed) {
1917 tokio::select! {
1918 event_result = event_rx.recv() => {
1919 match event_result {
1920 Some(event) => {
1921 #[cfg(target_os = "android")]
1922 android_log(&format!("Received transport peer event: {:?}", event));
1923
1924 match event {
1925 TransportPeerEvent::Connected { endpoint_id, .. } => {
1926 let peer_id = hex::encode(endpoint_id.as_bytes());
1927 #[cfg(target_os = "android")]
1928 android_log(&format!("Processing Connected event for peer: {}", peer_id));
1929 notify_peer_connected(&peer_id);
1930 }
1931 TransportPeerEvent::Disconnected { endpoint_id, reason } => {
1932 let peer_id = hex::encode(endpoint_id.as_bytes());
1933 #[cfg(target_os = "android")]
1934 android_log(&format!("Processing Disconnected event for peer: {} reason: {}", peer_id, reason));
1935 notify_peer_disconnected(&peer_id, &reason);
1936 }
1937 }
1938 }
1939 None => {
1940 #[cfg(target_os = "android")]
1941 android_log("Event channel closed, exiting peer event listener");
1942 break;
1943 }
1944 }
1945 }
1946 _ = tokio::time::sleep(std::time::Duration::from_secs(5)) => {
1947 // Periodically call peer_count() to trigger cleanup_closed_connections()
1948 // This detects dead connections and emits Disconnected events
1949 let count = transport_for_cleanup.peer_count();
1950 #[cfg(target_os = "android")]
1951 android_log(&format!("Periodic cleanup tick - peer count: {}", count));
1952 }
1953 }
1954 }
1955
1956 #[cfg(target_os = "android")]
1957 android_log("Peer event listener task exiting");
1958 });
1959
1960 // IMPORTANT (Issue #378): Use the storage_backend from sync_backend, NOT a new
1961 // one! Creating a separate AutomergeBackend would cause sync coordinator
1962 // state to be split, resulting in data not being received from peers.
1963 let storage_backend = sync_backend.storage_backend();
1964
1965 // Create TransportManager for multi-transport coordination (ADR-032, #555)
1966 // Build TransportManagerConfig from FFI config (PACE policy + collection
1967 // routes)
1968 let mut tm_config = TransportManagerConfig::default();
1969
1970 if let Some(ref transport_config) = config.transport {
1971 // Build PACE policy from transport_preference
1972 if let Some(ref prefs) = transport_config.transport_preference {
1973 let policy = TransportPolicy::new("ffi-config").primary(prefs.clone());
1974 tm_config.default_policy = Some(policy);
1975 }
1976
1977 // Parse collection routes from JSON
1978 if let Some(ref routes_json) = transport_config.collection_routes_json {
1979 match serde_json::from_str::<CollectionRouteTable>(routes_json) {
1980 Ok(table) => {
1981 tm_config.collection_routes = table;
1982 }
1983 Err(e) => {
1984 eprintln!("[Peat] Failed to parse collection_routes_json: {}", e);
1985 }
1986 }
1987 }
1988 }
1989
1990 let mut transport_manager = TransportManager::new(tm_config);
1991
1992 // Create IrohMeshTransport wrapper and register with TransportManager.
1993 // This allows the transport to be selected via PACE policy alongside
1994 // future transports.
1995 //
1996 // ADR-062 Phase 2 (peat#926): peat-mesh's IrohMeshTransport takes
1997 // `Vec<PeerInfo>` directly instead of `Arc<RwLock<PeerConfig>>` — the
1998 // `formation` and `local` fields of PeerConfig were never used by the
1999 // transport itself; they remain in peat-protocol's security layer.
2000 // peat-ffi starts with an empty static-peer list; runtime peer
2001 // additions go through `iroh_mesh_transport.set_static_peers(...)`.
2002 let iroh_mesh_transport = Arc::new(IrohMeshTransport::new(Arc::clone(&transport), Vec::new()));
2003 let iroh_as_transport: Arc<dyn Transport> = iroh_mesh_transport.clone();
2004 transport_manager.register(iroh_as_transport.clone());
2005
2006 // Register as PACE instance for collection routing
2007 let iroh_instance = TransportInstance::new(
2008 "iroh-primary",
2009 TransportType::Quic,
2010 TransportCapabilities::quic(),
2011 )
2012 .with_description("Primary Iroh/QUIC transport");
2013 transport_manager.register_instance(iroh_instance, iroh_as_transport);
2014
2015 // Initialize BLE transport if enabled (ADR-039, #556)
2016 #[cfg(feature = "bluetooth")]
2017 if let Some(ref transport_config) = config.transport {
2018 if transport_config.enable_ble {
2019 #[cfg(target_os = "android")]
2020 {
2021 use peat_btle::platform::android::AndroidAdapter;
2022 use peat_btle::{BleConfig, BluetoothLETransport};
2023
2024 android_log("BLE transport requested - initializing AndroidAdapter stub");
2025
2026 // Derive BLE node ID from Iroh endpoint key (same as Linux path)
2027 let iroh_endpoint_id = transport.endpoint_id();
2028 let iroh_key_bytes = iroh_endpoint_id.as_bytes();
2029 let ble_node_id = peat_btle::NodeId::new(u32::from_be_bytes([
2030 iroh_key_bytes[28],
2031 iroh_key_bytes[29],
2032 iroh_key_bytes[30],
2033 iroh_key_bytes[31],
2034 ]));
2035 let ble_config = BleConfig::new(ble_node_id);
2036 let adapter = AndroidAdapter::new_stub();
2037 let btle = BluetoothLETransport::new(ble_config, adapter);
2038 let ble_transport = Arc::new(PeatBleTransport::new(btle));
2039 let ble_as_transport: Arc<dyn Transport> = ble_transport.clone();
2040 transport_manager.register(ble_as_transport.clone());
2041
2042 // Register as PACE instance for collection routing
2043 let ble_instance = TransportInstance::new(
2044 "ble-primary",
2045 TransportType::BluetoothLE,
2046 TransportCapabilities::bluetooth_le(),
2047 )
2048 .with_description("Primary BLE transport (Android)");
2049 transport_manager.register_instance(ble_instance, ble_as_transport);
2050
2051 // Store in global for JNI access
2052 *ANDROID_BLE_TRANSPORT.lock().unwrap() = Some(ble_transport);
2053
2054 android_log("BLE transport registered as PACE instance 'ble-primary'");
2055 }
2056
2057 #[cfg(not(target_os = "android"))]
2058 {
2059 // On non-Android platforms, we can initialize BLE directly
2060 // Linux uses BluerAdapter, macOS uses CoreBluetoothAdapter
2061 #[cfg(target_os = "linux")]
2062 {
2063 use peat_btle::platform::linux::BluerAdapter;
2064 use peat_btle::{BleAdapter, BleConfig, BluetoothLETransport, PowerProfile};
2065
2066 // Parse power profile from config
2067 let power_profile = match transport_config.ble_power_profile.as_deref() {
2068 Some("aggressive") => PowerProfile::Aggressive,
2069 Some("low_power") => PowerProfile::LowPower,
2070 _ => PowerProfile::Balanced,
2071 };
2072
2073 // Derive a 32-bit BLE node ID from the Iroh endpoint's public key
2074 // Use last 4 bytes of the 32-byte key for a unique-enough identifier
2075 let iroh_endpoint_id = transport.endpoint_id();
2076 let iroh_key_bytes = iroh_endpoint_id.as_bytes();
2077 let ble_node_id = peat_btle::NodeId::new(u32::from_be_bytes([
2078 iroh_key_bytes[28],
2079 iroh_key_bytes[29],
2080 iroh_key_bytes[30],
2081 iroh_key_bytes[31],
2082 ]));
2083
2084 // Create BLE config with node ID, power profile, and mesh ID
2085 let mut ble_config = BleConfig::new(ble_node_id);
2086 ble_config.power_profile = power_profile;
2087 if let Some(ref mesh_id) = transport_config.ble_mesh_id {
2088 ble_config.mesh.mesh_id = mesh_id.clone();
2089 }
2090
2091 // Create BLE transport with BluerAdapter
2092 // IMPORTANT: All async BLE operations (create adapter, init, register
2093 // GATT, start advertising/scanning) MUST happen in a single block_on().
2094 // Splitting into two block_on() calls suspends the tokio runtime between
2095 // them, which can cause the GATT ApplicationHandle's D-Bus registration
2096 // to be dropped before advertising starts — making the GATT service
2097 // intermittently invisible to remote devices.
2098 //
2099 // Brings `MeshTransport` into scope so `ble_transport.start()` resolves;
2100 // mirrors the import at the other start() call site (line ~3259).
2101 use peat_protocol::transport::MeshTransport;
2102 match runtime_arc.block_on(async {
2103 let mut adapter = BluerAdapter::new().await?;
2104
2105 // Initialize adapter with config (stores node ID, mesh ID, etc.)
2106 adapter.init(&ble_config).await?;
2107
2108 // Register GATT service with BlueZ so peers can connect
2109 adapter.register_gatt_service().await?;
2110
2111 // Wrap in transport layers
2112 let btle = BluetoothLETransport::new(ble_config, adapter);
2113 let ble_transport = Arc::new(PeatBleTransport::new(btle));
2114
2115 // Start advertising and scanning in the same async context
2116 ble_transport.start().await.map_err(|e| {
2117 peat_btle::BleError::PlatformError(format!(
2118 "Failed to start BLE transport: {}",
2119 e
2120 ))
2121 })?;
2122
2123 Ok::<_, peat_btle::BleError>(ble_transport)
2124 }) {
2125 Ok(ble_transport) => {
2126 let ble_as_transport: Arc<dyn Transport> = ble_transport.clone();
2127 transport_manager.register(ble_as_transport.clone());
2128
2129 // Register as PACE instance for collection routing
2130 let ble_instance = TransportInstance::new(
2131 "ble-primary",
2132 TransportType::BluetoothLE,
2133 TransportCapabilities::bluetooth_le(),
2134 )
2135 .with_description("Primary BLE transport");
2136 transport_manager.register_instance(ble_instance, ble_as_transport);
2137 eprintln!(
2138 "[Peat] BLE transport registered as PACE instance 'ble-primary'"
2139 );
2140 }
2141 Err(e) => {
2142 eprintln!("[Peat] Failed to initialize BLE adapter: {} (continuing without BLE)", e);
2143 }
2144 }
2145 }
2146
2147 #[cfg(not(target_os = "linux"))]
2148 eprintln!(
2149 "[Peat] BLE transport requested but not yet implemented for this platform"
2150 );
2151 }
2152 }
2153 }
2154
2155 // TIMING: Total startup time
2156 let total_ms = total_start.elapsed().as_millis();
2157 #[cfg(target_os = "android")]
2158 android_log(&format!(
2159 "[TIMING] === TOTAL create_node: {}ms ===",
2160 total_ms
2161 ));
2162 #[cfg(not(target_os = "android"))]
2163 eprintln!("[Peat TIMING] === TOTAL create_node: {}ms ===", total_ms);
2164
2165 // Compose `peat_mesh::Node` over the same `AutomergeIrohBackend` the
2166 // existing typed surface uses. Both layers see the same underlying
2167 // doc store; the Node adds a generic publish/observe surface for
2168 // doc-type-agnostic callers (the `ingest*Jni` family, future
2169 // per-doc-type typed wrappers).
2170 #[cfg(feature = "sync")]
2171 let node = {
2172 use peat_mesh::sync::traits::DataSyncBackend;
2173 let backend_dyn: Arc<dyn DataSyncBackend> = sync_backend.clone();
2174 Arc::new(peat_mesh::Node::new(backend_dyn))
2175 };
2176
2177 // BleTranslator: BLE-typed structs ↔ Automerge documents (ADR-041).
2178 // Built only when the bluetooth feature is enabled. Used by the
2179 // `ingest*Jni` family of methods + (Slice 1.b.2.2) the
2180 // `OutboundFrameCallback` JNI surface.
2181 #[cfg(all(feature = "sync", feature = "bluetooth"))]
2182 let ble_translator = {
2183 use peat_protocol::sync::ble_translation::BleTranslator;
2184 Arc::new(BleTranslator::with_defaults())
2185 };
2186
2187 let node_arc = Arc::new(PeatNode {
2188 sync_backend,
2189 storage_backend,
2190 #[cfg(feature = "sync")]
2191 node,
2192 #[cfg(all(feature = "sync", feature = "bluetooth"))]
2193 ble_translator,
2194 transport_manager,
2195 iroh_transport: transport,
2196 store,
2197 #[cfg(feature = "sync")]
2198 water_counter: water_counter::WaterCounter::load_or_init(
2199 storage_path.join("water.automerge"),
2200 ),
2201 #[cfg(feature = "sync")]
2202 crdt_kv: crdt_kv::CrdtKvDocs::new(storage_path.clone()),
2203 // Load the persisted roster (or start empty) before storage_path is
2204 // moved into the struct below.
2205 #[cfg(feature = "sync")]
2206 roster: Arc::new(roster::RosterStore::load(&storage_path)),
2207 #[cfg(feature = "sync")]
2208 supervisor: Arc::new(supervisor::Supervisor::new()),
2209 #[cfg(feature = "sync")]
2210 reconnect_dial_semaphore: Arc::new(tokio::sync::Semaphore::new(
2211 MAX_CONCURRENT_RECONNECT_DIALS,
2212 )),
2213 storage_path,
2214 runtime: runtime_arc,
2215 cleanup_running,
2216 #[cfg(feature = "sync")]
2217 blob_store: std::sync::RwLock::new(None),
2218 #[cfg(all(feature = "sync", feature = "bluetooth"))]
2219 outbound_queue: Arc::new(std::sync::Mutex::new(std::collections::VecDeque::new())),
2220 #[cfg(all(feature = "sync", feature = "bluetooth"))]
2221 outbound_fanout: std::sync::Mutex::new(None),
2222 #[cfg(all(feature = "sync", feature = "bluetooth"))]
2223 relay_seen: std::sync::Mutex::new(std::collections::HashMap::new()),
2224 });
2225
2226 // Publish an OWNING reference to the JNI-visible global so a Kotlin bridge
2227 // (e.g. the BLE pipe) can reach a node created via the Dart/UniFFI path
2228 // without risking use-after-free: the prior code stashed a non-owning
2229 // alias whose sole owner was the Dart handle, so Dart's GC finalizer could
2230 // free the node out from under a `getGlobalNodeHandleJni` consumer.
2231 //
2232 // Android-only: the global is consumed solely by the JNI bridges (BLE /
2233 // Wi-Fi Direct). iOS reaches BLE via the independent UniFFI poll bridge and
2234 // never reads it, so storing an owning Arc there would only leak — the
2235 // node's sole owner on iOS must be the Dart UniFFI handle so `close()`/
2236 // dispose actually drops it and releases the redb file lock. Without this
2237 // gate, an in-app Stop on iOS left the node (and its redb store) alive, so
2238 // the next Start hit "Failed to open redb database" on the still-locked
2239 // file. iOS has no `clearGlobalNodeHandleJni` counterpart to release it.
2240 // Reconnect supervisor: periodic background tick. Holds a Weak ref so it
2241 // never keeps the node alive on its own — when the node is dropped (e.g. an
2242 // iOS Stop releasing the redb lock), the next upgrade() fails and the task
2243 // exits. The `cleanup_running` flag gives an explicit early stop too.
2244 #[cfg(feature = "sync")]
2245 {
2246 let weak = Arc::downgrade(&node_arc);
2247 let stop = Arc::clone(&node_arc.cleanup_running);
2248 node_arc.runtime.spawn(async move {
2249 // Initial delay so inbound accepts / first handshakes settle before
2250 // we start proactively dialing.
2251 tokio::time::sleep(std::time::Duration::from_secs(2)).await;
2252 while stop.load(Ordering::Relaxed) {
2253 match weak.upgrade() {
2254 Some(node) => node.run_supervisor_tick(now_unix_ms()),
2255 None => break, // node dropped — nothing left to supervise
2256 }
2257 tokio::time::sleep(std::time::Duration::from_secs(3)).await;
2258 }
2259 });
2260 }
2261
2262 #[cfg(target_os = "android")]
2263 set_global_node_handle(&node_arc);
2264 Ok(node_arc)
2265}
2266
2267// =============================================================================
2268// Shared dial path + reconnect supervisor orchestration
2269// =============================================================================
2270
2271/// Unix-epoch milliseconds, or 0 if the clock is before the epoch. Used to
2272/// drive the supervisor's backoff timing; monotonicity matters more than
2273/// wall-clock accuracy, and a one-off 0 just makes a peer eligible sooner.
2274#[cfg(feature = "sync")]
2275fn now_unix_ms() -> u64 {
2276 use std::time::{SystemTime, UNIX_EPOCH};
2277 SystemTime::now()
2278 .duration_since(UNIX_EPOCH)
2279 .map(|d| d.as_millis() as u64)
2280 .unwrap_or(0)
2281}
2282
2283/// Connect to a peer, run the formation handshake, and trigger document sync.
2284///
2285/// The single dial path shared by `connect_peer_nowait` (one-shot UI dial) and
2286/// the reconnect supervisor. Returns `true` if the peer is up afterward — a new
2287/// connection that passed the handshake, OR a peer the transport reports was
2288/// already connected (the accept/existing path owns it). Returns `false` on a
2289/// connect, handshake, or transport failure, which the supervisor turns into a
2290/// backoff.
2291#[cfg(feature = "sync")]
2292async fn dial_peer_and_sync(
2293 iroh_transport: &Arc<IrohTransport>,
2294 sync_backend: &Arc<AutomergeIrohBackend>,
2295 storage_backend: &Arc<AutomergeBackend>,
2296 peer: &PeatPeerInfo,
2297) -> bool {
2298 let conn_opt = match iroh_transport.connect_peer(peer).await {
2299 Ok(c) => c,
2300 Err(e) => {
2301 // Unconditional `tracing` so iOS/desktop consumers capture the dial
2302 // failure (e.g. via OSLog), not just Android's native log.
2303 tracing::warn!(error = %e, "dial_peer_and_sync: connect failed");
2304 #[cfg(target_os = "android")]
2305 android_log(&format!("dial_peer_and_sync: connect failed - {}", e));
2306 return false;
2307 }
2308 };
2309 // None means the transport already holds a connection to this peer; the
2310 // accept path is handling it, so report it up.
2311 let conn = match conn_opt {
2312 Some(c) => c,
2313 None => return true,
2314 };
2315 let peer_id = conn.remote_id();
2316
2317 let Some(formation_key) = sync_backend.formation_key() else {
2318 // No formation key — emit Connected without handshake (backward compat).
2319 iroh_transport.emit_peer_connected(peer_id);
2320 return true;
2321 };
2322
2323 use peat_protocol::network::perform_initiator_handshake;
2324 match perform_initiator_handshake(&conn, &formation_key).await {
2325 Ok(()) => {
2326 iroh_transport.emit_peer_connected(peer_id);
2327 if let Some(coordinator) = storage_backend.sync_coordinator() {
2328 let coord = Arc::clone(coordinator);
2329 let sync_peer = peer_id;
2330 tokio::spawn(async move {
2331 // Brief delay for the connection to stabilize before sync.
2332 tokio::time::sleep(tokio::time::Duration::from_millis(500)).await;
2333 let _ = coord.sync_all_documents_with_peer(sync_peer).await;
2334 });
2335 }
2336 true
2337 }
2338 Err(e) => {
2339 conn.close(1u32.into(), b"authentication failed");
2340 iroh_transport.disconnect(&peer_id).ok();
2341 tracing::warn!(error = %e, "dial_peer_and_sync: handshake failed");
2342 #[cfg(target_os = "android")]
2343 android_log(&format!("dial_peer_and_sync: handshake failed - {}", e));
2344 false
2345 }
2346 }
2347}
2348
2349#[cfg(feature = "sync")]
2350impl PeatNode {
2351 /// Whether `node_id` has a live link on ANY registered transport (iroh,
2352 /// BLE, …). This is how the supervisor dedups across transports: a peer
2353 /// already reachable over BLE is treated as connected, so we don't also
2354 /// dial it over iroh/relay. peat-mesh exposes no global cross-transport
2355 /// connected iterator (peat#828), but it does answer this per-peer via the
2356 /// transport manager, which is all the supervisor needs.
2357 fn peer_connected_any_transport(&self, node_id: &str) -> bool {
2358 let mesh_peer = peat_mesh::NodeId::new(node_id.to_string());
2359 self.transport_manager
2360 .available_instances_for_peer(&mesh_peer)
2361 .into_iter()
2362 .any(|tid| {
2363 self.transport_manager
2364 .get_instance(&tid)
2365 .and_then(|t| t.peer_link_state(&mesh_peer))
2366 // Count a transport as connected only when it reports a real,
2367 // *measured* link quality. `peer_link_state` returns `Some`
2368 // whenever the transport has a record for the peer — which can
2369 // include a stale/failed record, surfaced as
2370 // `LinkQuality::Unknown`. Treating Unknown as connected would
2371 // make this dedup silently suppress a legitimate reconnect
2372 // ("the node never reconnects because the supervisor thinks it
2373 // already is"). Erring toward a cheap, idempotent re-dial on
2374 // Unknown is the safer failure mode.
2375 .is_some_and(|s| {
2376 !matches!(s.quality, peat_mesh::transport::LinkQuality::Unknown)
2377 })
2378 })
2379 }
2380
2381 /// Spawn a single dial for a roster member, marking it `Connecting` and
2382 /// recording the outcome (connected + `last_seen`, or a backoff). The caller
2383 /// must have already checked the peer is eligible and not connected.
2384 /// Non-blocking: the dial runs on the runtime.
2385 fn try_dial_roster_peer(&self, entry: roster::RosterEntry, now_ms: u64) {
2386 self.supervisor.mark_connecting(&entry.node_id, now_ms);
2387
2388 let iroh_transport = Arc::clone(&self.iroh_transport);
2389 let sync_backend = Arc::clone(&self.sync_backend);
2390 let storage_backend = Arc::clone(&self.storage_backend);
2391 let supervisor = Arc::clone(&self.supervisor);
2392 let roster = Arc::clone(&self.roster);
2393 let peer = PeatPeerInfo {
2394 name: entry.name,
2395 node_id: entry.node_id.clone(),
2396 addresses: entry.addresses,
2397 relay_url: entry.relay_url,
2398 };
2399 let node_id = entry.node_id;
2400 let dial_permits = Arc::clone(&self.reconnect_dial_semaphore);
2401
2402 self.runtime.spawn(async move {
2403 // Throttle the actual dial (peers are already marked Connecting, so
2404 // the tick won't re-spawn them while they wait for a permit). Caps
2405 // the cold-start fan-out; `acquire` only errors if the semaphore is
2406 // closed (shutdown), in which case we just proceed.
2407 let _permit = dial_permits.acquire_owned().await.ok();
2408 let ok =
2409 dial_peer_and_sync(&iroh_transport, &sync_backend, &storage_backend, &peer).await;
2410 if ok {
2411 supervisor.mark_connected(&node_id);
2412 roster.mark_seen(&node_id, now_unix_ms());
2413 } else {
2414 supervisor.mark_failed(&node_id, now_unix_ms());
2415 }
2416 });
2417 }
2418
2419 /// One reconcile-and-dial pass of the reconnect supervisor.
2420 ///
2421 /// Builds the cross-transport connected set (iroh's connected peers, plus
2422 /// any roster member with a live link on another transport such as BLE),
2423 /// reconciles it against supervisor state, stamps `last_seen` for peers that
2424 /// just came up, prunes tracking for peers no longer in the roster, then
2425 /// spawns a dial for every roster member that is disconnected and eligible.
2426 fn run_supervisor_tick(&self, now_ms: u64) {
2427 let entries = self.roster.list();
2428
2429 // "Connected" means connected over ANY transport. Start from iroh's set
2430 // (which also covers inbound peers not in the roster), then fold in
2431 // roster members reachable on a non-iroh transport so we don't
2432 // redundantly dial a peer that's already up over BLE.
2433 let mut connected: std::collections::HashSet<String> = self
2434 .iroh_transport
2435 .connected_peers()
2436 .into_iter()
2437 .map(|e| hex::encode(e.as_bytes()))
2438 .collect();
2439 for e in &entries {
2440 if !connected.contains(&e.node_id) && self.peer_connected_any_transport(&e.node_id) {
2441 connected.insert(e.node_id.clone());
2442 }
2443 }
2444
2445 // Ground supervisor state in reality; stamp freshly-connected members.
2446 for id in self.supervisor.reconcile(&connected) {
2447 if self.roster.get(&id).is_some() {
2448 self.roster.mark_seen(&id, now_ms);
2449 }
2450 }
2451
2452 // Keep the supervisor map bounded to roster ∪ connected.
2453 let mut keep = connected.clone();
2454 keep.extend(entries.iter().map(|e| e.node_id.clone()));
2455 self.supervisor.retain(&keep);
2456
2457 for entry in entries {
2458 if connected.contains(&entry.node_id) {
2459 continue; // already up over some transport
2460 }
2461 if !self.supervisor.eligible(&entry.node_id, now_ms) {
2462 continue; // dialing, or backing off
2463 }
2464 self.try_dial_roster_peer(entry, now_ms);
2465 }
2466 }
2467}
2468
2469#[cfg(feature = "sync")]
2470#[uniffi::export]
2471impl PeatNode {
2472 /// Run one reconnect pass immediately, dialing any disconnected, eligible
2473 /// roster member. Safe to call repeatedly — peers already connected or
2474 /// mid-dial are skipped, and failures are rate-limited by backoff. A
2475 /// "gentle" trigger: it does NOT clear backoffs (use [`Self::wake_reconnect`]
2476 /// for that). Sits on top of the periodic background tick.
2477 pub fn reconnect_known_peers(&self) {
2478 self.run_supervisor_tick(now_unix_ms());
2479 }
2480
2481 /// React to a hint that a *specific* roster member is reachable right now —
2482 /// e.g. a BLE neighbour advertisement, or a relay "peer online" signal.
2483 ///
2484 /// If the peer is known, not already connected over any transport, and not
2485 /// mid-dial/backoff, it is dialed immediately — bypassing the periodic tick
2486 /// so the attempt lands inside a tight mobile background-execution budget.
2487 /// If the peer is already reachable over some transport (e.g. it just
2488 /// connected over BLE), this records that instead of dialing. Unknown peers
2489 /// are a no-op.
2490 pub fn on_peer_observed(&self, node_id: String) {
2491 let Some(entry) = self.roster.get(&node_id) else {
2492 return; // not a group member we track
2493 };
2494 let now = now_unix_ms();
2495 if self.peer_connected_any_transport(&node_id) {
2496 // Already up over some transport — record it, don't redial.
2497 self.supervisor.mark_connected(&node_id);
2498 self.roster.mark_seen(&node_id, now);
2499 return;
2500 }
2501 if self.supervisor.eligible(&node_id, now) {
2502 self.try_dial_roster_peer(entry, now);
2503 }
2504 }
2505
2506 /// React to a change that may have broadly restored connectivity — the
2507 /// network came up, or the app returned to foreground. Clears all backoffs
2508 /// so every known peer is immediately eligible, then runs one reconnect
2509 /// pass. Use [`Self::on_peer_observed`] when you know which peer is
2510 /// reachable; use this when you don't.
2511 pub fn wake_reconnect(&self) {
2512 self.supervisor.reset_backoff_all();
2513 self.run_supervisor_tick(now_unix_ms());
2514 }
2515}
2516
2517// =============================================================================
2518// Persistent roster (reconnect foundation)
2519// =============================================================================
2520//
2521// Exposes the on-disk roster of known group peers so a consumer can remember
2522// the group it joined and re-dial members after a restart, network change, or
2523// transport switch. The roster stores only non-secret reachability data; it is
2524// NOT a substitute for the formation key, which authenticates each connection.
2525// This is the foundation slice — the per-peer reconnect supervisor (backoff +
2526// cross-transport dedup) and the event hooks that drive it land on top.
2527#[cfg(feature = "sync")]
2528#[uniffi::export]
2529impl PeatNode {
2530 /// Insert or update a known peer in the roster (keyed by `node_id`) and
2531 /// persist it. Idempotent — re-upserting refreshes addresses/relay/name and
2532 /// never moves `last_seen_ms` backwards.
2533 pub fn roster_upsert(&self, entry: roster::RosterEntry) {
2534 self.roster.upsert(entry);
2535 }
2536
2537 /// Convenience: remember a `PeerInfo` (the same struct handed to
2538 /// `connect_peer`) under a group, stamping last-seen as "never" (0). This is
2539 /// the call a consumer makes for each member when joining a group (e.g. from
2540 /// a scanned join token), so the reconnect supervisor can re-dial them.
2541 /// Idempotent; re-remembering refreshes addresses/relay/name.
2542 pub fn roster_remember(&self, group_id: String, peer: PeerInfo) {
2543 self.roster.upsert(roster::RosterEntry {
2544 node_id: peer.node_id,
2545 group_id,
2546 name: peer.name,
2547 addresses: peer.addresses,
2548 relay_url: peer.relay_url,
2549 last_seen_ms: 0,
2550 });
2551 }
2552
2553 /// Remove a peer from the roster. Returns true if it was present.
2554 pub fn roster_remove(&self, node_id: String) -> bool {
2555 self.roster.remove(&node_id)
2556 }
2557
2558 /// Fetch a single roster entry by `node_id`.
2559 pub fn roster_get(&self, node_id: String) -> Option<roster::RosterEntry> {
2560 self.roster.get(&node_id)
2561 }
2562
2563 /// All roster entries, sorted by `node_id`.
2564 pub fn roster_list(&self) -> Vec<roster::RosterEntry> {
2565 self.roster.list()
2566 }
2567
2568 /// Roster entries for a single group.
2569 pub fn roster_list_by_group(&self, group_id: String) -> Vec<roster::RosterEntry> {
2570 self.roster.list_by_group(&group_id)
2571 }
2572}
2573
2574// Add new error variants for sync operations
2575#[cfg(feature = "sync")]
2576impl From<anyhow::Error> for PeatError {
2577 fn from(e: anyhow::Error) -> Self {
2578 PeatError::SyncError { msg: e.to_string() }
2579 }
2580}
2581
2582// =============================================================================
2583// Peat Data Types for Consumer Integration
2584// =============================================================================
2585//
2586// These types represent Peat entities that can be synced and displayed by
2587// consumer plugins. They use well-known collection names for document storage.
2588
2589/// Well-known collection names for Peat data
2590pub mod collections {
2591 /// Collection for Peat cells (teams/squads)
2592 pub const CELLS: &str = "cells";
2593 /// Collection for detected tracks (entities being tracked)
2594 pub const TRACKS: &str = "tracks";
2595 /// Collection for nodes (robots, drones, sensors)
2596 pub const NODES: &str = "nodes";
2597 /// Collection for capability advertisements
2598 pub const CAPABILITIES: &str = "capabilities";
2599 /// Collection for commands (C2 messages)
2600 pub const COMMANDS: &str = "commands";
2601 /// Collection for operator-placed map markers (CoT pins synced
2602 /// across the mesh via the universal-Document transport,
2603 /// ADR-035). Receiver renders consistently regardless of which
2604 /// peer originated the marker — the doc store is the source of
2605 /// truth, transport is invisible to consumers.
2606 pub const MARKERS: &str = "markers";
2607}
2608
2609/// CoT 2525 placeholder type that
2610/// [`parse_marker_publish_json`] substitutes when a tombstone body
2611/// arrives without an explicit `type` field. Tombstones intentionally
2612/// omit geo + type to keep the BLE frame tight (~40 bytes vs ~120
2613/// for a full marker); receivers filter `_deleted: true` entries out
2614/// of "current markers" views before the placeholder is rendered, so
2615/// the value never reaches a UI. Lifted to a named constant so a
2616/// future change to the placeholder shape (e.g., shifting to a
2617/// neutral "unknown" or an empty string) lands in one place rather
2618/// than being scattered through the parser.
2619const TOMBSTONE_PLACEHOLDER_TYPE: &str = "a-u-G";
2620
2621/// Cell status enumeration
2622#[derive(Debug, Clone, Copy, PartialEq, Eq, uniffi::Enum)]
2623pub enum CellStatus {
2624 /// Cell is active and operational
2625 Active,
2626 /// Cell is forming (members joining)
2627 Forming,
2628 /// Cell has degraded capability
2629 Degraded,
2630 /// Cell is offline
2631 Offline,
2632}
2633
2634impl CellStatus {
2635 fn from_str(s: &str) -> Self {
2636 match s.to_uppercase().as_str() {
2637 "ACTIVE" => Self::Active,
2638 "FORMING" => Self::Forming,
2639 "DEGRADED" => Self::Degraded,
2640 "OFFLINE" => Self::Offline,
2641 _ => Self::Offline,
2642 }
2643 }
2644
2645 fn as_str(&self) -> &'static str {
2646 match self {
2647 Self::Active => "ACTIVE",
2648 Self::Forming => "FORMING",
2649 Self::Degraded => "DEGRADED",
2650 Self::Offline => "OFFLINE",
2651 }
2652 }
2653}
2654
2655/// Peat Cell information for display
2656#[derive(Debug, Clone, uniffi::Record)]
2657pub struct CellInfo {
2658 /// Unique cell identifier
2659 pub id: String,
2660 /// Human-readable cell name (e.g., "Alpha Team")
2661 pub name: String,
2662 /// Cell status
2663 pub status: CellStatus,
2664 /// Number of nodes in this cell
2665 pub node_count: u32,
2666 /// Center latitude (WGS84)
2667 pub center_lat: f64,
2668 /// Center longitude (WGS84)
2669 pub center_lon: f64,
2670 /// List of capabilities (e.g., ["OBJECT_TRACKING", "COMMUNICATION"])
2671 pub capabilities: Vec<String>,
2672 /// Parent formation ID (if any)
2673 pub formation_id: Option<String>,
2674 /// Cell leader node ID (if any)
2675 pub leader_id: Option<String>,
2676 /// Last update timestamp (Unix millis)
2677 pub last_update: i64,
2678 /// Optional scenario command piggybacked on cell (e.g., "START_SCENARIO",
2679 /// "STOP_SCENARIO")
2680 pub scenario_command: Option<String>,
2681}
2682
2683/// Track category enumeration
2684#[derive(Debug, Clone, Copy, PartialEq, Eq, uniffi::Enum)]
2685pub enum TrackCategory {
2686 Person,
2687 Vehicle,
2688 Aircraft,
2689 Vessel,
2690 Installation,
2691 Unknown,
2692}
2693
2694impl TrackCategory {
2695 fn from_str(s: &str) -> Self {
2696 match s.to_uppercase().as_str() {
2697 "PERSON" => Self::Person,
2698 "VEHICLE" => Self::Vehicle,
2699 "AIRCRAFT" => Self::Aircraft,
2700 "VESSEL" => Self::Vessel,
2701 "INSTALLATION" => Self::Installation,
2702 _ => Self::Unknown,
2703 }
2704 }
2705
2706 fn as_str(&self) -> &'static str {
2707 match self {
2708 Self::Person => "PERSON",
2709 Self::Vehicle => "VEHICLE",
2710 Self::Aircraft => "AIRCRAFT",
2711 Self::Vessel => "VESSEL",
2712 Self::Installation => "INSTALLATION",
2713 Self::Unknown => "UNKNOWN",
2714 }
2715 }
2716}
2717
2718/// Track information for display
2719#[derive(Debug, Clone, uniffi::Record)]
2720pub struct TrackInfo {
2721 /// Unique track identifier
2722 pub id: String,
2723 /// Source node that detected this track
2724 pub source_node: String,
2725 /// Cell ID that owns this track (if any)
2726 pub cell_id: Option<String>,
2727 /// Formation ID (if any)
2728 pub formation_id: Option<String>,
2729 /// Track latitude (WGS84)
2730 pub lat: f64,
2731 /// Track longitude (WGS84)
2732 pub lon: f64,
2733 /// Height above ellipsoid (meters, optional)
2734 pub hae: Option<f64>,
2735 /// Circular error probable (meters, optional)
2736 pub cep: Option<f64>,
2737 /// Heading in degrees (0 = North, optional)
2738 pub heading: Option<f64>,
2739 /// Speed in m/s (optional)
2740 pub speed: Option<f64>,
2741 /// MIL-STD-2525 classification or category
2742 pub classification: String,
2743 /// Detection confidence (0.0 - 1.0)
2744 pub confidence: f64,
2745 /// Track category
2746 pub category: TrackCategory,
2747 /// Created timestamp (Unix millis)
2748 pub created_at: i64,
2749 /// Last update timestamp (Unix millis)
2750 pub last_update: i64,
2751 /// Additional key-value attributes (callsign, image chip data, etc.)
2752 pub attributes: HashMap<String, String>,
2753}
2754
2755/// Node status enumeration
2756#[derive(Debug, Clone, Copy, PartialEq, Eq, uniffi::Enum)]
2757pub enum NodeStatus {
2758 /// Node is ready
2759 Ready,
2760 /// Node is active
2761 Active,
2762 /// Node has degraded capability
2763 Degraded,
2764 /// Node is offline
2765 Offline,
2766 /// Node is loading/initializing
2767 Loading,
2768}
2769
2770impl NodeStatus {
2771 fn from_str(s: &str) -> Self {
2772 match s.to_uppercase().as_str() {
2773 "READY" => Self::Ready,
2774 "ACTIVE" => Self::Active,
2775 "DEGRADED" => Self::Degraded,
2776 "OFFLINE" => Self::Offline,
2777 "LOADING" => Self::Loading,
2778 _ => Self::Offline,
2779 }
2780 }
2781
2782 pub fn as_str(&self) -> &'static str {
2783 match self {
2784 Self::Ready => "READY",
2785 Self::Active => "ACTIVE",
2786 Self::Degraded => "DEGRADED",
2787 Self::Offline => "OFFLINE",
2788 Self::Loading => "LOADING",
2789 }
2790 }
2791}
2792
2793/// Node information for display
2794#[derive(Debug, Clone, uniffi::Record)]
2795pub struct NodeInfo {
2796 /// Unique node identifier
2797 pub id: String,
2798 /// Node type (e.g., "UGV", "UAV", "Soldier System")
2799 pub node_type: String,
2800 /// Node name/callsign
2801 pub name: String,
2802 /// Node status
2803 pub status: NodeStatus,
2804 /// Node latitude (WGS84)
2805 pub lat: f64,
2806 /// Node longitude (WGS84)
2807 pub lon: f64,
2808 /// Height above ellipsoid (meters, optional)
2809 pub hae: Option<f64>,
2810 /// Readiness level (0.0 - 1.0)
2811 pub readiness: f64,
2812 /// List of capabilities
2813 pub capabilities: Vec<String>,
2814 /// Cell membership (if any)
2815 pub cell_id: Option<String>,
2816 /// Battery / fuel percentage (0–100). Optional because not every
2817 /// node has a measurable battery (fixed sensors, pre-lock
2818 /// watches), and legacy publishes from pre-2026-05-08 hosts didn't
2819 /// carry the field. Wire key: `battery_percent`. See
2820 /// [`parse_battery_percent`] for the clamp + None semantics.
2821 pub battery_percent: Option<i32>,
2822 /// Heart rate in BPM, sourced from wearable sensors (WearOS watch,
2823 /// M5Stack health). Wire key: `heart_rate`. Required to surface a
2824 /// vitals indicator on the operator card; absent on node types
2825 /// that don't carry a wearable. See [`parse_heart_rate`] for the
2826 /// clamp + None semantics.
2827 pub heart_rate: Option<i32>,
2828 /// Last heartbeat timestamp (Unix millis). Defaults to `0` when
2829 /// the publisher omits the field, surfaced to the UI as
2830 /// "1970-01-01 stale" — different intent from `battery_percent`'s
2831 /// `None` ("unknown sensor state"). Don't fold this into the same
2832 /// `Option<T>` shape: a missing heartbeat *is* a stale-record
2833 /// signal, not absence-of-data, and the node-overlay code uses
2834 /// the time delta directly without a None-check branch.
2835 pub last_heartbeat: i64,
2836}
2837
2838/// Operator-placed map marker — the typed shape every peer renders
2839/// in the Peat Markers panel and on the MapView (ADR-035 Universal
2840/// Document transport, "markers" collection).
2841///
2842/// Origin-agnostic: this struct is what the local doc store holds,
2843/// independent of which peer published it. The plugin's mental model
2844/// is "created somewhere, synced everywhere, displayed consistently"
2845/// — `MarkerInfo` is the synced shape, the wire transport is
2846/// invisible above this surface.
2847///
2848/// Wire-key parity with the JSON the prior raw-JSON publish path
2849/// produced (uid, type, lat, lon, hae, ts, callsign, color), so the
2850/// migration to the typed API is wire-compatible: docs published by
2851/// the old raw-JSON path round-trip cleanly into `MarkerInfo`.
2852#[derive(Debug, Clone, uniffi::Record)]
2853pub struct MarkerInfo {
2854 /// Unique marker identifier — the operator-placed UID, typically
2855 /// UUID-shaped (e.g. `4ae7b0a0-1995-447c-...`).
2856 pub uid: String,
2857 /// CoT 2525-style type code (e.g. `"a-f-G-U-C"` for friendly
2858 /// ground unit combat, `"b-m-p-w"` for waypoint).
2859 pub marker_type: String,
2860 /// Latitude (WGS84).
2861 pub lat: f64,
2862 /// Longitude (WGS84).
2863 pub lon: f64,
2864 /// Height above ellipsoid (meters). `None` when the publisher
2865 /// had no altitude fix; receivers render at ground level.
2866 pub hae: Option<f64>,
2867 /// Unix epoch milliseconds — the publisher's clock at marker
2868 /// drop time. Receivers DON'T treat this as a presence-staleness
2869 /// timestamp (markers persist until deleted, unlike nodes);
2870 /// it's purely "when did the operator drop this pin."
2871 pub ts: i64,
2872 /// Operator callsign of the publisher. `None` when the publisher
2873 /// didn't stamp it.
2874 pub callsign: Option<String>,
2875 /// Marker color (consumer-defined encoding — commonly a 32-bit
2876 /// ARGB integer, sign-extended). `None` when default coloring
2877 /// applies.
2878 pub color: Option<i32>,
2879 /// Cell membership (organizational unit within mesh), if scoped.
2880 /// `None` for cell-agnostic markers.
2881 pub cell_id: Option<String>,
2882 /// Soft-delete sentinel. When `true`, the marker is a tombstone
2883 /// — peers sync the deletion (CRDT keeps the entry so concurrent
2884 /// edits resolve consistently) but consumer UIs filter it out
2885 /// of "current markers" views. peat-mesh's fan-out today does
2886 /// NOT propagate `ChangeEvent::Removed` (Slice 2 work), so the
2887 /// soft-delete-sentinel pattern is the only way to communicate
2888 /// deletions across the mesh until that lands. Wire key: `_deleted`
2889 /// (matches the peat-mesh `transport::document_codec` synthesis
2890 /// convention from PR #103).
2891 pub deleted: bool,
2892}
2893
2894// Wire-shape contract for `Option<T>` fields on `NodeInfo`
2895// (Rust-side emit/parse only; downstream consumers in other repos
2896// have their own contracts).
2897//
2898// - **Emit:** `serialize_node_json` and `serialize_nodes_get_json` both render
2899// `Option::None` as JSON `null` via `serde_json::json!` macro semantics.
2900// There is no second emit shape from this codec.
2901//
2902// - **Parse:** `parse_node_json` and `parse_node_publish_json` both treat JSON
2903// `null` AND a missing key the same way — both yield `None`.
2904// `serde_json::Value` indexing returns `Value::Null` for missing keys, and
2905// the typed accessors (`as_i64`, `as_str`, …) return `None` on a null
2906// variant. So receivers don't need to distinguish "absent" from "explicit
2907// null" — they're equivalent on the read side. Locked in by
2908// `legacy_json_without_battery_or_heart_parses_with_none` (absent) and
2909// `battery_and_heart_reject_non_numeric` (explicit null).
2910//
2911// - **Forward-compat:** parsers ignore unknown keys. Any wire shape a
2912// future-version peer adds passes through unchanged.
2913
2914/// Command status enumeration
2915#[derive(Debug, Clone, Copy, PartialEq, Eq, uniffi::Enum)]
2916pub enum CommandStatus {
2917 /// Command is pending execution
2918 Pending,
2919 /// Command is being executed
2920 Executing,
2921 /// Command completed successfully
2922 Completed,
2923 /// Command failed
2924 Failed,
2925 /// Command was cancelled
2926 Cancelled,
2927}
2928
2929impl CommandStatus {
2930 fn from_str(s: &str) -> Self {
2931 match s.to_uppercase().as_str() {
2932 "PENDING" => Self::Pending,
2933 "EXECUTING" => Self::Executing,
2934 "COMPLETED" => Self::Completed,
2935 "FAILED" => Self::Failed,
2936 "CANCELLED" => Self::Cancelled,
2937 _ => Self::Pending,
2938 }
2939 }
2940
2941 fn as_str(&self) -> &'static str {
2942 match self {
2943 Self::Pending => "PENDING",
2944 Self::Executing => "EXECUTING",
2945 Self::Completed => "COMPLETED",
2946 Self::Failed => "FAILED",
2947 Self::Cancelled => "CANCELLED",
2948 }
2949 }
2950}
2951
2952/// Command information for C2
2953#[derive(Debug, Clone, uniffi::Record)]
2954pub struct CommandInfo {
2955 /// Unique command identifier
2956 pub id: String,
2957 /// Command type (e.g., "TRACK_TARGET", "MOVE", "ABORT")
2958 pub command_type: String,
2959 /// Target cell or node ID
2960 pub target_id: String,
2961 /// Command parameters as JSON string
2962 pub parameters: String,
2963 /// Command priority (1-5, 1 = highest)
2964 pub priority: u8,
2965 /// Command status
2966 pub status: CommandStatus,
2967 /// Originator ID
2968 pub originator: String,
2969 /// Created timestamp (Unix millis)
2970 pub created_at: i64,
2971 /// Last update timestamp (Unix millis)
2972 pub last_update: i64,
2973}
2974
2975// =============================================================================
2976// PeatNode Extensions for Typed Data Access
2977// =============================================================================
2978
2979#[cfg(feature = "sync")]
2980#[uniffi::export]
2981impl PeatNode {
2982 // -------------------------------------------------------------------------
2983 // Cell Operations
2984 // -------------------------------------------------------------------------
2985
2986 /// Get all cells from the sync document
2987 pub fn get_cells(&self) -> Result<Vec<CellInfo>, PeatError> {
2988 self.runtime.block_on(async {
2989 let backend = &self.storage_backend;
2990 let coll = backend.collection(collections::CELLS);
2991
2992 let docs = coll
2993 .scan()
2994 .map_err(|e| PeatError::StorageError { msg: e.to_string() })?;
2995
2996 let mut cells = Vec::new();
2997 for (id, data) in docs {
2998 if let Ok(json) = String::from_utf8(data) {
2999 if let Ok(cell) = parse_cell_json(&id, &json) {
3000 cells.push(cell);
3001 }
3002 }
3003 }
3004 Ok(cells)
3005 })
3006 }
3007
3008 /// Get a specific cell by ID
3009 pub fn get_cell(&self, cell_id: &str) -> Result<Option<CellInfo>, PeatError> {
3010 self.runtime.block_on(async {
3011 let backend = &self.storage_backend;
3012 let coll = backend.collection(collections::CELLS);
3013
3014 match coll.get(cell_id) {
3015 Ok(Some(data)) => {
3016 let json = String::from_utf8(data).map_err(|e| PeatError::StorageError {
3017 msg: format!("Invalid UTF-8: {}", e),
3018 })?;
3019 let cell = parse_cell_json(cell_id, &json)?;
3020 Ok(Some(cell))
3021 }
3022 Ok(None) => Ok(None),
3023 Err(e) => Err(PeatError::StorageError { msg: e.to_string() }),
3024 }
3025 })
3026 }
3027
3028 /// Store a cell
3029 pub fn put_cell(&self, cell: CellInfo) -> Result<(), PeatError> {
3030 let json = serialize_cell_json(&cell)?;
3031 self.runtime.block_on(async {
3032 let backend = &self.storage_backend;
3033 let coll = backend.collection(collections::CELLS);
3034 coll.upsert(&cell.id, json.into_bytes())
3035 .map_err(|e| PeatError::StorageError { msg: e.to_string() })
3036 })
3037 }
3038
3039 // -------------------------------------------------------------------------
3040 // Track Operations
3041 // -------------------------------------------------------------------------
3042
3043 /// Get all tracks from the sync document.
3044 ///
3045 /// Reads via `peat_mesh::Node::query(...)` so the writer/reader API
3046 /// stays consistent with `ingest_position_via_translator`'s
3047 /// `Node::publish_with_origin` path. The earlier implementation
3048 /// scanned `AutomergeBackend::collection(...).scan()` directly,
3049 /// expecting the bytes to be flat JSON of the original body — but
3050 /// `publish_with_origin` writes a Document whose Automerge map
3051 /// shape doesn't match that expectation, so every body field came
3052 /// back at `parse_track_json`'s `unwrap_or` defaults (peat#832).
3053 /// Going through `Node::query` decodes the Document fields
3054 /// properly and the read result matches what the writer published.
3055 /// The `track_tests::ingest_position_via_translator_then_get_tracks_preserves_body`
3056 /// test locks this in.
3057 pub fn get_tracks(&self) -> Result<Vec<TrackInfo>, PeatError> {
3058 use peat_mesh::sync::types::Query;
3059 self.runtime.block_on(async {
3060 let docs = self
3061 .node
3062 .query(collections::TRACKS, &Query::All)
3063 .await
3064 .map_err(|e| PeatError::StorageError { msg: e.to_string() })?;
3065
3066 let mut tracks = Vec::with_capacity(docs.len());
3067 for doc in docs {
3068 if let Some(id) = doc.id.clone() {
3069 if let Ok(track) = track_from_document(&id, &doc) {
3070 tracks.push(track);
3071 }
3072 }
3073 }
3074 Ok(tracks)
3075 })
3076 }
3077
3078 /// Get a specific track by ID. Routes through `Node::get` for the
3079 /// same writer/reader symmetry reason as `get_tracks` (peat#832).
3080 pub fn get_track(&self, track_id: &str) -> Result<Option<TrackInfo>, PeatError> {
3081 self.runtime.block_on(async {
3082 let id = track_id.to_string();
3083 match self.node.get(collections::TRACKS, &id).await {
3084 Ok(Some(doc)) => Ok(Some(track_from_document(track_id, &doc)?)),
3085 Ok(None) => Ok(None),
3086 Err(e) => Err(PeatError::StorageError { msg: e.to_string() }),
3087 }
3088 })
3089 }
3090
3091 /// Store a track. Publishes through `Node::publish` so the
3092 /// resulting Document lives in the same storage namespace
3093 /// `Node::query` / `Node::get` read from — the BLE-bridged
3094 /// `ingest_position_via_translator` path already publishes this
3095 /// way, so unifying the typed `put_track` path keeps writer/reader
3096 /// symmetric for both publish surfaces (peat#832).
3097 ///
3098 /// Behavioral change vs pre-#836: this now fires through
3099 /// `TransportManager` fan-out (the `Node::publish` path emits a
3100 /// `ChangeEvent` that BLE / iroh transport drains observe), where
3101 /// the pre-fix `coll.upsert(json_bytes)` only emitted the
3102 /// in-process observer broadcast. No production caller exists
3103 /// today (production tracks come in via `ingestPositionJni`), so
3104 /// the change is observable only via UniFFI Kotlin / Swift
3105 /// consumers if any appear later. Documented here so the next
3106 /// reader doesn't have to re-trace the change to find out.
3107 pub fn put_track(&self, track: TrackInfo) -> Result<(), PeatError> {
3108 let doc = track_to_document(&track)?;
3109 self.runtime.block_on(async {
3110 self.node
3111 .publish(collections::TRACKS, doc)
3112 .await
3113 .map(|_id| ())
3114 .map_err(|e| PeatError::StorageError { msg: e.to_string() })
3115 })
3116 }
3117
3118 // -------------------------------------------------------------------------
3119 // Node Operations
3120 // -------------------------------------------------------------------------
3121
3122 /// Get all nodes from the sync document
3123 pub fn get_nodes(&self) -> Result<Vec<NodeInfo>, PeatError> {
3124 self.runtime.block_on(async {
3125 let backend = &self.storage_backend;
3126 let coll = backend.collection(collections::NODES);
3127
3128 let docs = coll
3129 .scan()
3130 .map_err(|e| PeatError::StorageError { msg: e.to_string() })?;
3131
3132 let mut nodes = Vec::new();
3133 for (id, data) in docs {
3134 if let Ok(json) = String::from_utf8(data) {
3135 if let Ok(node) = parse_node_json(&id, &json) {
3136 nodes.push(node);
3137 }
3138 }
3139 }
3140 Ok(nodes)
3141 })
3142 }
3143
3144 /// Store a node
3145 pub fn put_node(&self, node: NodeInfo) -> Result<(), PeatError> {
3146 let json = serialize_node_json(&node)?;
3147 self.runtime.block_on(async {
3148 let backend = &self.storage_backend;
3149 let coll = backend.collection(collections::NODES);
3150 coll.upsert(&node.id, json.into_bytes())
3151 .map_err(|e| PeatError::StorageError { msg: e.to_string() })
3152 })
3153 }
3154
3155 // -------------------------------------------------------------------------
3156 // Marker Operations (operator-placed map pins, synced via ADR-035
3157 // Universal Document transport)
3158 // -------------------------------------------------------------------------
3159
3160 /// Get all markers from the sync document.
3161 ///
3162 /// Returns the canonical typed list of operator-placed pins
3163 /// across the mesh. Origin-agnostic — locally-created and
3164 /// peer-synced markers are indistinguishable in the result.
3165 /// Plugin consumers (PeatMapComponent's periodic refresh, the
3166 /// Peat Markers panel readout) call this and render every entry
3167 /// with the same code path.
3168 pub fn get_markers(&self) -> Result<Vec<MarkerInfo>, PeatError> {
3169 self.runtime.block_on(async {
3170 let backend = &self.storage_backend;
3171 let coll = backend.collection(collections::MARKERS);
3172
3173 let docs = coll
3174 .scan()
3175 .map_err(|e| PeatError::StorageError { msg: e.to_string() })?;
3176
3177 let mut markers = Vec::new();
3178 for (id, data) in docs {
3179 let json_str = String::from_utf8_lossy(&data);
3180 match parse_marker_publish_json(&id, &json_str) {
3181 Ok(m) => markers.push(m),
3182 Err(_) => {
3183 // Malformed entry — skip silently. Same shape
3184 // as get_nodes / get_commands handle parse
3185 // errors: don't poison the whole list with one
3186 // bad doc.
3187 }
3188 }
3189 }
3190 Ok(markers)
3191 })
3192 }
3193
3194 /// Store a marker.
3195 ///
3196 /// Persists into the `markers` collection. peat-mesh's fan-out
3197 /// observes the change and routes via the registered transports
3198 /// (universal-Document path on BLE via LiteBridgeTranslator,
3199 /// iroh sync for cross-mesh peers). Receivers see the same
3200 /// `MarkerInfo` shape on their side.
3201 pub fn put_marker(&self, marker: MarkerInfo) -> Result<(), PeatError> {
3202 let json = serialize_marker_json(&marker)?;
3203 let uid = marker.uid.clone();
3204 self.runtime.block_on(async {
3205 let backend = &self.storage_backend;
3206 let coll = backend.collection(collections::MARKERS);
3207 coll.upsert(&uid, json.into_bytes())
3208 .map_err(|e| PeatError::StorageError { msg: e.to_string() })
3209 })
3210 }
3211
3212 // -------------------------------------------------------------------------
3213 // Command Operations (C2)
3214 // -------------------------------------------------------------------------
3215
3216 /// Get all pending commands
3217 pub fn get_commands(&self) -> Result<Vec<CommandInfo>, PeatError> {
3218 self.runtime.block_on(async {
3219 let backend = &self.storage_backend;
3220 let coll = backend.collection(collections::COMMANDS);
3221
3222 let docs = coll
3223 .scan()
3224 .map_err(|e| PeatError::StorageError { msg: e.to_string() })?;
3225
3226 let mut commands = Vec::new();
3227 for (id, data) in docs {
3228 if let Ok(json) = String::from_utf8(data) {
3229 if let Ok(cmd) = parse_command_json(&id, &json) {
3230 commands.push(cmd);
3231 }
3232 }
3233 }
3234 Ok(commands)
3235 })
3236 }
3237
3238 /// Store a command (for C2 issuance)
3239 pub fn put_command(&self, command: CommandInfo) -> Result<(), PeatError> {
3240 let json = serialize_command_json(&command)?;
3241 self.runtime.block_on(async {
3242 let backend = &self.storage_backend;
3243 let coll = backend.collection(collections::COMMANDS);
3244 coll.upsert(&command.id, json.into_bytes())
3245 .map_err(|e| PeatError::StorageError { msg: e.to_string() })
3246 })
3247 }
3248}
3249
3250// =============================================================================
3251// Blob Transfer (ADR-060) — not UniFFI-exported; reached via direct JNI only
3252// =============================================================================
3253
3254#[cfg(feature = "sync")]
3255impl PeatNode {
3256 /// Enable the parallel blob-transfer endpoint.
3257 ///
3258 /// Constructs a `NetworkedIrohBlobStore` on the tokio runtime owned by
3259 /// this node and stores it for later use via `blob_put` / `blob_get`.
3260 /// Bind address defaults to `0.0.0.0:0` (ephemeral) when None.
3261 pub fn enable_blob_transfer(
3262 &self,
3263 bind_addr: Option<std::net::SocketAddr>,
3264 ) -> Result<(), PeatError> {
3265 let blob_dir = self.storage_path.join("blobs");
3266 std::fs::create_dir_all(&blob_dir).map_err(|e| PeatError::StorageError {
3267 msg: format!("Failed to create blob dir {:?}: {}", blob_dir, e),
3268 })?;
3269
3270 let config = PeatMeshIrohConfig {
3271 bind_addr,
3272 ..Default::default()
3273 };
3274
3275 let store = self
3276 .runtime
3277 .block_on(NetworkedIrohBlobStore::from_config(blob_dir, &config))
3278 .map_err(|e| PeatError::SyncError {
3279 msg: format!("Failed to create blob store: {}", e),
3280 })?;
3281
3282 #[cfg(target_os = "android")]
3283 android_log(&format!(
3284 "Blob transfer enabled. EndpointId={}",
3285 store.endpoint_id().fmt_short()
3286 ));
3287
3288 let mut slot = self.blob_store.write().map_err(|_| PeatError::SyncError {
3289 msg: "blob_store lock poisoned".to_string(),
3290 })?;
3291 *slot = Some(store);
3292 Ok(())
3293 }
3294
3295 /// Add a known blob peer by hex EndpointId and socket address.
3296 /// Uses peat-mesh's `add_peer_from_hex` so no iroh types cross into
3297 /// peat-ffi.
3298 pub fn blob_add_peer(&self, peer_id_hex: &str, address: &str) -> Result<(), PeatError> {
3299 let store_guard = self.blob_store.read().map_err(|_| PeatError::SyncError {
3300 msg: "blob_store lock poisoned".to_string(),
3301 })?;
3302 let store = store_guard.as_ref().ok_or(PeatError::SyncError {
3303 msg: "blob transfer not enabled".to_string(),
3304 })?;
3305
3306 let store_clone = Arc::clone(store);
3307 let hex = peer_id_hex.to_string();
3308 let addr = address.to_string();
3309 self.runtime
3310 .block_on(async move { store_clone.add_peer_from_hex(&hex, &addr).await })
3311 .map_err(|e| PeatError::SyncError {
3312 msg: format!("blob_add_peer: {}", e),
3313 })?;
3314
3315 #[cfg(target_os = "android")]
3316 android_log(&format!(
3317 "Blob peer added: {} at {}",
3318 &peer_id_hex[..16.min(peer_id_hex.len())],
3319 address
3320 ));
3321
3322 Ok(())
3323 }
3324
3325 /// Store bytes in the local blob store. Returns the content hash as hex.
3326 pub fn blob_put(&self, data: &[u8], content_type: &str) -> Result<String, PeatError> {
3327 let store_guard = self.blob_store.read().map_err(|_| PeatError::SyncError {
3328 msg: "blob_store lock poisoned".to_string(),
3329 })?;
3330 let store = store_guard.as_ref().ok_or(PeatError::SyncError {
3331 msg: "blob transfer not enabled".to_string(),
3332 })?;
3333
3334 let metadata = BlobMetadata {
3335 content_type: Some(content_type.to_string()),
3336 name: None,
3337 custom: Default::default(),
3338 };
3339
3340 let store_clone = Arc::clone(store);
3341 let data_vec = data.to_vec();
3342 let token = self
3343 .runtime
3344 .block_on(async move {
3345 store_clone
3346 .create_blob_from_bytes(&data_vec, metadata)
3347 .await
3348 })
3349 .map_err(|e| PeatError::StorageError {
3350 msg: format!("blob put failed: {}", e),
3351 })?;
3352
3353 Ok(token.hash.as_hex().to_string())
3354 }
3355
3356 /// Fetch blob bytes by content hash (hex). Tries local first, then
3357 /// known peers. Returns the bytes or an error.
3358 pub fn blob_get(&self, hash_hex: &str) -> Result<Vec<u8>, PeatError> {
3359 let store_guard = self.blob_store.read().map_err(|_| PeatError::SyncError {
3360 msg: "blob_store lock poisoned".to_string(),
3361 })?;
3362 let store = store_guard.as_ref().ok_or(PeatError::SyncError {
3363 msg: "blob transfer not enabled".to_string(),
3364 })?;
3365
3366 let token = BlobToken {
3367 hash: peat_mesh::storage::BlobHash(hash_hex.to_string()),
3368 size_bytes: 0, // unknown; fetch_blob doesn't use this for lookup
3369 metadata: BlobMetadata {
3370 content_type: None,
3371 name: None,
3372 custom: Default::default(),
3373 },
3374 };
3375
3376 let store_clone = Arc::clone(store);
3377 let handle = self
3378 .runtime
3379 .block_on(async move { store_clone.fetch_blob_simple(&token).await })
3380 .map_err(|e| PeatError::StorageError {
3381 msg: format!("blob fetch failed: {}", e),
3382 })?;
3383
3384 std::fs::read(&handle.path).map_err(|e| PeatError::StorageError {
3385 msg: format!("blob read failed: {}", e),
3386 })
3387 }
3388
3389 /// Check if a blob exists locally without network fetch.
3390 pub fn blob_exists_locally(&self, hash_hex: &str) -> bool {
3391 let store_guard = match self.blob_store.read() {
3392 Ok(g) => g,
3393 Err(_) => return false,
3394 };
3395 let store = match store_guard.as_ref() {
3396 Some(s) => s,
3397 None => return false,
3398 };
3399 let hash = peat_mesh::storage::BlobHash(hash_hex.to_string());
3400 store.blob_exists_locally(&hash)
3401 }
3402
3403 /// Get the blob endpoint ID as hex (returns None if blob transfer is
3404 /// disabled).
3405 pub fn blob_endpoint_id(&self) -> Option<String> {
3406 let store_guard = self.blob_store.read().ok()?;
3407 let store = store_guard.as_ref()?;
3408 Some(hex::encode(store.endpoint_id().as_bytes()))
3409 }
3410
3411 /// Get the blob endpoint's bound socket address as "ip:port".
3412 /// Useful for configuring remote peers and for tests.
3413 pub fn blob_bound_addr(&self) -> Option<String> {
3414 let store_guard = self.blob_store.read().ok()?;
3415 let store = store_guard.as_ref()?;
3416 store.bound_addr_string()
3417 }
3418}
3419
3420// =============================================================================
3421// JSON Serialization Helpers
3422// =============================================================================
3423
3424fn parse_cell_json(id: &str, json: &str) -> Result<CellInfo, PeatError> {
3425 let root: serde_json::Value =
3426 serde_json::from_str(json).map_err(|e| PeatError::InvalidInput {
3427 msg: format!("Invalid JSON: {}", e),
3428 })?;
3429 // Docs published through the node layer are wrapped as {id, fields:{..}};
3430 // flat (legacy) writes keep fields at the root. Read from `fields` if present.
3431 let v = match root.get("fields") {
3432 Some(f) if f.is_object() => f,
3433 _ => &root,
3434 };
3435
3436 Ok(CellInfo {
3437 id: id.to_string(),
3438 name: v["name"].as_str().unwrap_or(id).to_string(),
3439 status: CellStatus::from_str(v["status"].as_str().unwrap_or("OFFLINE")),
3440 node_count: v["node_count"].as_u64().unwrap_or(0) as u32,
3441 center_lat: v["center_lat"].as_f64().unwrap_or(0.0),
3442 center_lon: v["center_lon"].as_f64().unwrap_or(0.0),
3443 capabilities: v["capabilities"]
3444 .as_array()
3445 .map(|arr| {
3446 arr.iter()
3447 .filter_map(|v| v.as_str().map(|s| s.to_string()))
3448 .collect()
3449 })
3450 .unwrap_or_default(),
3451 formation_id: v["formation_id"].as_str().map(|s| s.to_string()),
3452 leader_id: v["leader_id"].as_str().map(|s| s.to_string()),
3453 last_update: v["last_update"].as_i64().unwrap_or(0),
3454 scenario_command: v["scenario_command"].as_str().map(|s| s.to_string()),
3455 })
3456}
3457
3458fn serialize_cell_json(cell: &CellInfo) -> Result<String, PeatError> {
3459 let v = serde_json::json!({
3460 "name": cell.name,
3461 "status": cell.status.as_str(),
3462 "node_count": cell.node_count,
3463 "center_lat": cell.center_lat,
3464 "center_lon": cell.center_lon,
3465 "capabilities": cell.capabilities,
3466 "formation_id": cell.formation_id,
3467 "leader_id": cell.leader_id,
3468 "last_update": cell.last_update,
3469 "scenario_command": cell.scenario_command,
3470 });
3471 serde_json::to_string(&v).map_err(|e| PeatError::EncodingError { msg: e.to_string() })
3472}
3473
3474/// Adapt a `TrackInfo` into a `peat_mesh::Document` for publishing.
3475///
3476/// Routes through the existing `serialize_track_json` so the body-field
3477/// encoding rules stay in one place — re-deserializing the JSON into a
3478/// `Map<String, Value>` and stuffing into `Document.fields` is the same
3479/// shape `peat_protocol::sync::ble_translation::value_to_mesh_document`
3480/// produces from the translator path. One extra serde round-trip per
3481/// `put_track`; acceptable for the consumer counts the plugin handles.
3482fn track_to_document(track: &TrackInfo) -> Result<peat_mesh::sync::types::Document, PeatError> {
3483 let json = serialize_track_json(track)?;
3484 let value: serde_json::Value =
3485 serde_json::from_str(&json).map_err(|e| PeatError::EncodingError {
3486 msg: format!("track_to_document: re-parse failed: {}", e),
3487 })?;
3488 let fields: std::collections::HashMap<String, serde_json::Value> = match value {
3489 serde_json::Value::Object(map) => map.into_iter().collect(),
3490 _ => std::collections::HashMap::new(),
3491 };
3492 Ok(peat_mesh::sync::types::Document {
3493 id: Some(track.id.clone()),
3494 fields,
3495 updated_at: std::time::SystemTime::now(),
3496 })
3497}
3498
3499/// Adapt a `peat_mesh::Document` into a `TrackInfo`.
3500///
3501/// Routes through the existing `parse_track_json` so the body-field
3502/// mapping rules stay in one place — `Document.fields` is a flat
3503/// `HashMap<String, Value>`, so re-emitting them as a JSON object is
3504/// a one-step adapter rather than a full reimplementation. The cost
3505/// is one extra serde_json round-trip per track on read; acceptable
3506/// for the consumer counts the plugin handles (single-digit
3507/// nodes × tens of tracks).
3508fn track_from_document(
3509 id: &str,
3510 doc: &peat_mesh::sync::types::Document,
3511) -> Result<TrackInfo, PeatError> {
3512 let body: serde_json::Map<String, serde_json::Value> = doc
3513 .fields
3514 .iter()
3515 .map(|(k, v)| (k.clone(), v.clone()))
3516 .collect();
3517 let json = serde_json::to_string(&serde_json::Value::Object(body))
3518 .map_err(|e| PeatError::EncodingError { msg: e.to_string() })?;
3519 parse_track_json(id, &json)
3520}
3521
3522fn parse_track_json(id: &str, json: &str) -> Result<TrackInfo, PeatError> {
3523 let v: serde_json::Value = serde_json::from_str(json).map_err(|e| PeatError::InvalidInput {
3524 msg: format!("Invalid JSON: {}", e),
3525 })?;
3526
3527 Ok(TrackInfo {
3528 id: id.to_string(),
3529 source_node: v["source_node"].as_str().unwrap_or("unknown").to_string(),
3530 cell_id: v["cell_id"].as_str().map(|s| s.to_string()),
3531 formation_id: v["formation_id"].as_str().map(|s| s.to_string()),
3532 lat: v["lat"].as_f64().unwrap_or(0.0),
3533 lon: v["lon"].as_f64().unwrap_or(0.0),
3534 hae: v["hae"].as_f64(),
3535 cep: v["cep"].as_f64(),
3536 heading: v["heading"].as_f64(),
3537 speed: v["speed"].as_f64(),
3538 classification: v["classification"].as_str().unwrap_or("a-u-G").to_string(),
3539 confidence: v["confidence"].as_f64().unwrap_or(0.5),
3540 category: TrackCategory::from_str(v["category"].as_str().unwrap_or("UNKNOWN")),
3541 created_at: v["created_at"].as_i64().unwrap_or(0),
3542 last_update: v["last_update"].as_i64().unwrap_or(0),
3543 attributes: v["attributes"]
3544 .as_object()
3545 .map(|obj| {
3546 obj.iter()
3547 .map(|(k, v)| (k.clone(), v.as_str().unwrap_or("").to_string()))
3548 .collect()
3549 })
3550 .unwrap_or_default(),
3551 })
3552}
3553
3554fn serialize_track_json(track: &TrackInfo) -> Result<String, PeatError> {
3555 let v = serde_json::json!({
3556 "source_node": track.source_node,
3557 "cell_id": track.cell_id,
3558 "formation_id": track.formation_id,
3559 "lat": track.lat,
3560 "lon": track.lon,
3561 "hae": track.hae,
3562 "cep": track.cep,
3563 "heading": track.heading,
3564 "speed": track.speed,
3565 "classification": track.classification,
3566 "confidence": track.confidence,
3567 "category": track.category.as_str(),
3568 "created_at": track.created_at,
3569 "last_update": track.last_update,
3570 "attributes": track.attributes,
3571 });
3572 serde_json::to_string(&v).map_err(|e| PeatError::EncodingError { msg: e.to_string() })
3573}
3574
3575fn parse_node_json(id: &str, json: &str) -> Result<NodeInfo, PeatError> {
3576 let root: serde_json::Value =
3577 serde_json::from_str(json).map_err(|e| PeatError::InvalidInput {
3578 msg: format!("Invalid JSON: {}", e),
3579 })?;
3580
3581 // Node docs published through the node layer are wrapped as
3582 // `{id, fields:{..}, updated_at}`; flat (legacy storage_backend) writes
3583 // keep the fields at the root. Read from `fields` when it's an object.
3584 let v = match root.get("fields") {
3585 Some(f) if f.is_object() => f,
3586 _ => &root,
3587 };
3588
3589 Ok(NodeInfo {
3590 id: id.to_string(),
3591 node_type: v["node_type"].as_str().unwrap_or("unknown").to_string(),
3592 name: v["name"].as_str().unwrap_or(id).to_string(),
3593 status: NodeStatus::from_str(v["status"].as_str().unwrap_or("OFFLINE")),
3594 lat: v["lat"].as_f64().unwrap_or(0.0),
3595 lon: v["lon"].as_f64().unwrap_or(0.0),
3596 hae: v["hae"].as_f64(),
3597 readiness: v["readiness"].as_f64().unwrap_or(0.0),
3598 capabilities: v["capabilities"]
3599 .as_array()
3600 .map(|arr| {
3601 arr.iter()
3602 .filter_map(|v| v.as_str().map(|s| s.to_string()))
3603 .collect()
3604 })
3605 .unwrap_or_default(),
3606 cell_id: v["cell_id"].as_str().map(|s| s.to_string()),
3607 battery_percent: parse_battery_percent(&v["battery_percent"]),
3608 heart_rate: parse_heart_rate(&v["heart_rate"]),
3609 last_heartbeat: v["last_heartbeat"].as_i64().unwrap_or(0),
3610 })
3611}
3612
3613/// Parse a Kotlin-side `publishNodeJni` payload into a
3614/// `NodeInfo`.
3615///
3616/// Distinct from `parse_node_json` because the JNI publish path
3617/// supplies a few different defaults: `node_type` defaults to
3618/// `"SOLDIER"` here vs `"unknown"` in the storage parser; `status`
3619/// defaults to `"ACTIVE"` here vs `"OFFLINE"` for storage; `readiness`
3620/// defaults to `1.0` here vs `0.0`. The `last_heartbeat` field is
3621/// honored from the wire when present (with a `now() + 60s` clock-skew
3622/// clamp via `parse_publish_last_heartbeat`); falls back to local
3623/// `Utc::now()` only when the publisher omits it. See
3624/// [`parse_publish_last_heartbeat`] for the full semantics.
3625///
3626/// Centralizing this in a free function makes it directly
3627/// unit-testable and means the inline JNI path and the test suite
3628/// share the exact codec implementation — the duplication that hid
3629/// peat#835.
3630///
3631/// Errors:
3632/// - `InvalidInput` if the JSON is malformed or `id` is missing/empty (consumed
3633/// as the storage key downstream; an empty id would collide with
3634/// `getNodesJni`'s scan results).
3635fn parse_node_publish_json(json_str: &str) -> Result<NodeInfo, PeatError> {
3636 let v: serde_json::Value =
3637 serde_json::from_str(json_str).map_err(|e| PeatError::InvalidInput {
3638 msg: format!("publishNode: invalid JSON: {}", e),
3639 })?;
3640
3641 let id = match v["id"].as_str() {
3642 Some(id) if !id.is_empty() => id.to_string(),
3643 _ => {
3644 return Err(PeatError::InvalidInput {
3645 msg: "publishNode: missing or empty 'id' field".to_string(),
3646 });
3647 }
3648 };
3649
3650 Ok(NodeInfo {
3651 id,
3652 node_type: v["node_type"].as_str().unwrap_or("SOLDIER").to_string(),
3653 name: v["name"].as_str().unwrap_or("Unknown").to_string(),
3654 status: NodeStatus::from_str(v["status"].as_str().unwrap_or("ACTIVE")),
3655 lat: v["lat"].as_f64().unwrap_or(0.0),
3656 lon: v["lon"].as_f64().unwrap_or(0.0),
3657 hae: v["hae"].as_f64(),
3658 readiness: v["readiness"].as_f64().unwrap_or(1.0),
3659 capabilities: v["capabilities"]
3660 .as_array()
3661 .map(|arr| {
3662 arr.iter()
3663 .filter_map(|v| v.as_str().map(|s| s.to_string()))
3664 .collect()
3665 })
3666 .unwrap_or_else(|| vec!["PLI".to_string()]),
3667 cell_id: v["cell_id"].as_str().map(|s| s.to_string()),
3668 battery_percent: parse_battery_percent(&v["battery_percent"]),
3669 heart_rate: parse_heart_rate(&v["heart_rate"]),
3670 last_heartbeat: parse_publish_last_heartbeat(&v["last_heartbeat"]),
3671 })
3672}
3673
3674/// Parse the `last_heartbeat` field on a publish-side JSON envelope.
3675///
3676/// Three intents we must honor faithfully:
3677/// 1. **Wire absent → stamp `now()`.** Real publishers (Kotlin self-PLI,
3678/// BLE-bridged peripheral relay) don't carry a timestamp; the JNI surface
3679/// always meant "this publish is fresh."
3680/// 2. **Wire `0` → preserve `0`.** Per `NodeInfo`'s field doc, `last_heartbeat
3681/// = 0` is the documented stale-record sentinel ("1970-01-01 stale"). The
3682/// earlier `> 0` filter silently overrode this — a publisher sending the
3683/// documented stale marker got `Utc::now()` back, the *opposite* signal.
3684/// That was a writer/reader-asymmetry regression of the same class peat#835
3685/// was opened to fix; round-4 drops the filter.
3686/// 3. **Wire absurdly far in the future → clamp to `now()`.** A peer with a
3687/// future-skewed clock can publish `i64::MAX` or any timestamp ahead of
3688/// local time; downstream Kotlin staleness UI consumes the value raw via
3689/// `getStalenessString` and would show the node as "always fresh." Cap
3690/// acceptance at `now() + 60_000ms` (60 s grace for legitimate clock drift
3691/// in distributed systems); beyond that, treat as adversarial /
3692/// misconfigured and stamp local `now()`.
3693///
3694/// 4. **Wire negative → collapse to the stale-marker (`0`).** Round-4 let
3695/// negatives pass through with a doc-comment claiming downstream time-delta
3696/// arithmetic still produced a sensible age; that's wrong: `now - i64::MIN`
3697/// overflows i64, and Kotlin `Long` subtraction silently wraps, producing
3698/// nonsense staleness output (or panic in Rust debug builds). Negative
3699/// timestamps are pathological — pre-epoch publish makes no sense in this
3700/// product — and collapsing them onto the documented stale-marker (`0`)
3701/// keeps the UI's arithmetic safe while preserving the "very stale" intent.
3702fn parse_publish_last_heartbeat(v: &serde_json::Value) -> i64 {
3703 let now_ms = chrono::Utc::now().timestamp_millis();
3704 // 60 s grace covers normal NTP drift between mobile devices on
3705 // unrelated networks; beyond that, the value is broken.
3706 const FUTURE_GRACE_MS: i64 = 60_000;
3707 let max_acceptable = now_ms.saturating_add(FUTURE_GRACE_MS);
3708 match v.as_i64() {
3709 Some(n) if n > max_acceptable => now_ms,
3710 // Collapse negatives to the documented stale-marker — both
3711 // bound the downstream Long-subtraction and preserve the
3712 // publisher's "very stale" intent unambiguously.
3713 Some(n) if n < 0 => 0,
3714 Some(n) => n,
3715 None => now_ms,
3716 }
3717}
3718
3719/// Serialize a slice of `NodeInfo` into the JSON-array shape
3720/// `getNodesJni` returns to Kotlin.
3721///
3722/// Mirror of [`parse_node_publish_json`] for the read-back path.
3723/// Pre-round-3 this was inlined inside the JNI function — that's the
3724/// duplicated-codec class peat#835 was opened to lock; extracting it
3725/// here makes the emit-side schema directly testable and keeps
3726/// writer/reader symmetry single-sourced.
3727///
3728/// Falls through to `"[]"` on serializer failure (the JNI surface
3729/// returned the same string on `get_nodes` errors before the
3730/// extraction; preserving that for back-compat).
3731///
3732/// Not gated on `feature = "sync"` even though the only caller
3733/// (`getNodesJni`) is — the body operates on `NodeInfo` and
3734/// `serde_json` only, and the mirror parser `serialize_node_json`
3735/// is unconditional. Asymmetric gating between the pair would be
3736/// confusing to maintainers and `cargo check --no-default-features`
3737/// wouldn't catch the inconsistency.
3738fn serialize_nodes_get_json(nodes: &[NodeInfo]) -> String {
3739 let json_array: Vec<serde_json::Value> = nodes
3740 .iter()
3741 .map(|p| {
3742 serde_json::json!({
3743 "id": p.id,
3744 "node_type": p.node_type,
3745 "name": p.name,
3746 "status": p.status.as_str(),
3747 "lat": p.lat,
3748 "lon": p.lon,
3749 "hae": p.hae,
3750 "readiness": p.readiness,
3751 "capabilities": p.capabilities,
3752 "cell_id": p.cell_id,
3753 "battery_percent": p.battery_percent,
3754 "heart_rate": p.heart_rate,
3755 "last_heartbeat": p.last_heartbeat,
3756 })
3757 })
3758 .collect();
3759 serde_json::to_string(&json_array).unwrap_or_else(|_| "[]".to_string())
3760}
3761
3762/// Coerce a JSON `Value` into a numeric value as i64.
3763///
3764/// Accepts both integer (`85`) and float (`85.0`, `85.5`) JSON
3765/// numbers; floats round half-away-from-zero per `f64::round()`.
3766/// Returns `None` for any other variant (string, null, array, object,
3767/// missing key).
3768///
3769/// Why both forms: serde_json maps JSON numbers into one of three
3770/// internal representations (i64 / u64 / f64), and `Value::as_i64`
3771/// only matches the first. A Kotlin publisher serializing
3772/// `Int.toDouble().toString()` (i.e. `"85.0"` reaches the parser as
3773/// the float variant), or any node whose JSON serializer renders
3774/// integers with a trailing `.0`, would silently drop the field
3775/// through the int-only path. That's the **same data-loss bug class
3776/// peat#835 was opened to lock**: a publisher writes a value and the
3777/// receiver decodes `None`, indistinguishable from "no sensor."
3778/// Empirically `serde_json::json!(85.0).as_i64() == None`; the float
3779/// fallback closes the gap.
3780///
3781/// **Precision contract — important for callers reusing this helper
3782/// outside of `parse_battery_percent` / `parse_heart_rate`**:
3783///
3784/// JSON Numbers above `i64::MAX` (i.e. stored as `u64` in serde_json,
3785/// 9.22e18..1.84e19) are unreachable by `as_i64()` and traverse the
3786/// `as_f64()` fallback. f64 has only 53 bits of mantissa, so values
3787/// above 2⁵³ (≈ 9.0e15) lose integer precision via that path —
3788/// e.g. `9_007_199_254_740_993_u64` round-trips through f64 as
3789/// `9_007_199_254_740_992`.
3790///
3791/// For `battery_percent` (0..=100) and `heart_rate` (0..=250) this is
3792/// inconsequential: the subsequent `clamp` truncates any
3793/// astronomically-large value to the same range end. Callers operating
3794/// on a wider range or needing exact integer fidelity above 2⁵³ should
3795/// pre-validate the wire shape (e.g. reject non-i64 Numbers explicitly)
3796/// rather than reuse this helper.
3797///
3798/// **Rounding mode**: `f64::round()` rounds half-away-from-zero
3799/// (`85.5 → 86`, `-85.5 → -86`). If a future caller depends on
3800/// banker's-rounding or half-to-even semantics, switch to
3801/// `f.round_ties_even()` (Rust 1.77+) and update tests accordingly.
3802fn coerce_json_number_to_i64(v: &serde_json::Value) -> Option<i64> {
3803 if let Some(n) = v.as_i64() {
3804 return Some(n);
3805 }
3806 // `f64::round() as i64` is saturating in current Rust (1.45+):
3807 // `f64::INFINITY as i64 == i64::MAX`, NaN as i64 == 0. Both
3808 // outcomes get clamped by the caller into the logical range, so
3809 // pathological floats fail-safe rather than panic.
3810 v.as_f64().map(|f| f.round() as i64)
3811}
3812
3813/// Parse a JSON `Value` into a battery percentage, clamping into the
3814/// physical 0..=100 range.
3815///
3816/// - Accepts integer or float JSON numbers (`85`, `85.0`, `85.5` → `85`). See
3817/// [`coerce_json_number_to_i64`] for why both forms.
3818/// - Numeric values clamp on out-of-range. The silent-`None`-on- overflow shape
3819/// `as_i64().and_then(|n| i32::try_from(n).ok())` produced was the same bug
3820/// class peat#835 was opened to prevent: a pathological 2³² `battery_percent`
3821/// becomes "no battery sensor," visually identical to the legitimate `None`
3822/// case. Clamp fails-safe to 0 or 100 instead.
3823/// - Non-numeric (string, object, missing key, JSON null) returns `None`. We
3824/// accept "no battery sensor" but reject silent type coercion — a `"85"`
3825/// *string* wire payload is a publisher bug, not a value to interpret.
3826///
3827/// Wire form: number in 0–100 (integer or float), or `null` / absent
3828/// for "unknown."
3829fn parse_battery_percent(v: &serde_json::Value) -> Option<i32> {
3830 let n = coerce_json_number_to_i64(v)?;
3831 Some(n.clamp(0, 100) as i32)
3832}
3833
3834/// Parse a JSON `Value` into a heart rate (BPM), clamping into the
3835/// 0..=250 range.
3836///
3837/// - Accepts integer or float JSON numbers; floats round.
3838/// - Lower bound is **0**, not 30: athletic resting bradycardia can dip into
3839/// the 20s, and a sensor reporting 0/asystole is a real emergency signal that
3840/// the UI should surface, not silently round up. The earlier 30 floor masked
3841/// these. Upper bound stays 250 (well above maximal exertion ~220−age) to
3842/// catch overflow payloads.
3843/// - Non-numeric returns `None` ("no wearable sensor present").
3844///
3845/// Wire form: number in 0–250 (integer or float), or `null` / absent
3846/// for "unknown."
3847fn parse_heart_rate(v: &serde_json::Value) -> Option<i32> {
3848 let n = coerce_json_number_to_i64(v)?;
3849 Some(n.clamp(0, 250) as i32)
3850}
3851
3852/// Parse a `MarkerInfo` from the wire JSON (publish-side), with
3853/// graceful field absence: missing optional fields → `None`, missing
3854/// required geo (`uid`/`type`/`lat`/`lon`) → `InvalidInput`.
3855///
3856/// The parser is wire-compatible with the JSON the prior raw-JSON
3857/// publish path produced — see the field comments on `MarkerInfo`
3858/// for key-by-key parity. The `id` argument lets the scan-side
3859/// caller supply the doc id (the doc store's key) when it's not in
3860/// the body; we accept either source as the `uid`.
3861fn parse_marker_publish_json(id: &str, json_str: &str) -> Result<MarkerInfo, PeatError> {
3862 let v: serde_json::Value =
3863 serde_json::from_str(json_str).map_err(|e| PeatError::InvalidInput {
3864 msg: format!("marker JSON: {}", e),
3865 })?;
3866
3867 let uid = v["uid"]
3868 .as_str()
3869 .map(|s| s.to_string())
3870 .filter(|s| !s.is_empty())
3871 .unwrap_or_else(|| id.to_string());
3872 if uid.is_empty() {
3873 return Err(PeatError::InvalidInput {
3874 msg: "marker missing uid (and no doc-store id supplied)".to_string(),
3875 });
3876 }
3877
3878 // Deletion-sentinel detection. A tombstone marker is just
3879 // `{uid, _deleted: true}` — type/lat/lon optional. Receivers
3880 // know to filter the entry out of "current markers" views. We
3881 // need the deletion to ride the same wire envelope as a normal
3882 // marker (peat-mesh fan-out doesn't propagate Removed events
3883 // today), so the doc-store retains the tombstone for CRDT
3884 // consistency.
3885 let deleted = v["_deleted"].as_bool().unwrap_or(false);
3886
3887 let marker_type = if deleted {
3888 v["type"]
3889 .as_str()
3890 .unwrap_or(TOMBSTONE_PLACEHOLDER_TYPE)
3891 .to_string()
3892 } else {
3893 v["type"]
3894 .as_str()
3895 .ok_or_else(|| PeatError::InvalidInput {
3896 msg: format!("marker {uid} missing CoT type"),
3897 })?
3898 .to_string()
3899 };
3900 let lat = if deleted {
3901 v["lat"].as_f64().unwrap_or(0.0)
3902 } else {
3903 v["lat"].as_f64().ok_or_else(|| PeatError::InvalidInput {
3904 msg: format!("marker {uid} missing lat"),
3905 })?
3906 };
3907 let lon = if deleted {
3908 v["lon"].as_f64().unwrap_or(0.0)
3909 } else {
3910 v["lon"].as_f64().ok_or_else(|| PeatError::InvalidInput {
3911 msg: format!("marker {uid} missing lon"),
3912 })?
3913 };
3914 let hae = v["hae"].as_f64();
3915 let ts = v["ts"].as_i64().unwrap_or(0);
3916 let callsign = v["callsign"]
3917 .as_str()
3918 .filter(|s| !s.is_empty())
3919 .map(|s| s.to_string());
3920 let color = coerce_json_number_to_i64(&v["color"]).map(|n| n as i32);
3921 let cell_id = v["cell_id"]
3922 .as_str()
3923 .filter(|s| !s.is_empty())
3924 .map(|s| s.to_string());
3925
3926 Ok(MarkerInfo {
3927 uid,
3928 marker_type,
3929 lat,
3930 lon,
3931 hae,
3932 ts,
3933 callsign,
3934 color,
3935 cell_id,
3936 deleted,
3937 })
3938}
3939
3940/// Serialize the typed list to the JSON shape `getMarkersJni`
3941/// returns. Wire-key parity with `serialize_marker_json` so a doc
3942/// round-trips through the get path identically to the put path.
3943fn serialize_markers_get_json(markers: &[MarkerInfo]) -> String {
3944 let json_array: Vec<serde_json::Value> = markers
3945 .iter()
3946 .map(|m| {
3947 let mut obj = serde_json::json!({
3948 "uid": m.uid,
3949 "type": m.marker_type,
3950 "lat": m.lat,
3951 "lon": m.lon,
3952 "hae": m.hae,
3953 "ts": m.ts,
3954 "callsign": m.callsign,
3955 "color": m.color,
3956 "cell_id": m.cell_id,
3957 });
3958 if m.deleted {
3959 obj["_deleted"] = serde_json::Value::Bool(true);
3960 }
3961 obj
3962 })
3963 .collect();
3964 // `serde_json::to_string` on a `Vec<serde_json::Value>` composed
3965 // entirely of primitives, booleans, strings, and JSON objects we
3966 // just constructed is infallible — the failure modes are
3967 // I/O on `to_writer`, non-string map keys, or NaN floats without
3968 // the `arbitrary_precision` feature. None of those can arise
3969 // from this shape, so the unwrap-to-`"[]"` fallback is dead code
3970 // that exists only because the signature returns `String` (not
3971 // `Result<String, _>`) for symmetry with the JNI consumers'
3972 // `Ok("[]")` semantics on storage error. If a future field type
3973 // change introduces a fallible shape (e.g., `f64::NAN` for a
3974 // missing-altitude sentinel), promote this to `Result` and
3975 // surface the error to the caller.
3976 serde_json::to_string(&json_array).unwrap_or_else(|_| "[]".to_string())
3977}
3978
3979/// Serialize a single marker for `put_marker` storage. Wire-key
3980/// parity with `serialize_markers_get_json` (single object instead
3981/// of array — same key set, same shapes) so a doc written via
3982/// `put_marker` reads identically through `get_markers`.
3983fn serialize_marker_json(marker: &MarkerInfo) -> Result<String, PeatError> {
3984 let mut v = serde_json::json!({
3985 "uid": marker.uid,
3986 "type": marker.marker_type,
3987 "lat": marker.lat,
3988 "lon": marker.lon,
3989 "hae": marker.hae,
3990 "ts": marker.ts,
3991 "callsign": marker.callsign,
3992 "color": marker.color,
3993 "cell_id": marker.cell_id,
3994 });
3995 if marker.deleted {
3996 v["_deleted"] = serde_json::Value::Bool(true);
3997 }
3998 serde_json::to_string(&v).map_err(|e| PeatError::EncodingError { msg: e.to_string() })
3999}
4000
4001fn serialize_node_json(node: &NodeInfo) -> Result<String, PeatError> {
4002 let v = serde_json::json!({
4003 "node_type": node.node_type,
4004 "name": node.name,
4005 "status": node.status.as_str(),
4006 "lat": node.lat,
4007 "lon": node.lon,
4008 "hae": node.hae,
4009 "readiness": node.readiness,
4010 "capabilities": node.capabilities,
4011 "cell_id": node.cell_id,
4012 "battery_percent": node.battery_percent,
4013 "heart_rate": node.heart_rate,
4014 "last_heartbeat": node.last_heartbeat,
4015 });
4016 serde_json::to_string(&v).map_err(|e| PeatError::EncodingError { msg: e.to_string() })
4017}
4018
4019fn parse_command_json(id: &str, json: &str) -> Result<CommandInfo, PeatError> {
4020 let root: serde_json::Value =
4021 serde_json::from_str(json).map_err(|e| PeatError::InvalidInput {
4022 msg: format!("Invalid JSON: {}", e),
4023 })?;
4024 // Docs published through the node layer are wrapped as {id, fields:{..}};
4025 // flat (legacy) writes keep fields at the root. Read from `fields` if present.
4026 let v = match root.get("fields") {
4027 Some(f) if f.is_object() => f,
4028 _ => &root,
4029 };
4030
4031 Ok(CommandInfo {
4032 id: id.to_string(),
4033 command_type: v["command_type"].as_str().unwrap_or("UNKNOWN").to_string(),
4034 target_id: v["target_id"].as_str().unwrap_or("").to_string(),
4035 parameters: v["parameters"].to_string(),
4036 priority: v["priority"].as_u64().unwrap_or(3) as u8,
4037 status: CommandStatus::from_str(v["status"].as_str().unwrap_or("PENDING")),
4038 originator: v["originator"].as_str().unwrap_or("").to_string(),
4039 created_at: v["created_at"].as_i64().unwrap_or(0),
4040 last_update: v["last_update"].as_i64().unwrap_or(0),
4041 })
4042}
4043
4044fn serialize_command_json(command: &CommandInfo) -> Result<String, PeatError> {
4045 // Parse parameters as JSON or use empty object
4046 let params: serde_json::Value =
4047 serde_json::from_str(&command.parameters).unwrap_or(serde_json::json!({}));
4048
4049 let v = serde_json::json!({
4050 "command_type": command.command_type,
4051 "target_id": command.target_id,
4052 "parameters": params,
4053 "priority": command.priority,
4054 "status": command.status.as_str(),
4055 "originator": command.originator,
4056 "created_at": command.created_at,
4057 "last_update": command.last_update,
4058 });
4059 serde_json::to_string(&v).map_err(|e| PeatError::EncodingError { msg: e.to_string() })
4060}
4061
4062#[cfg(test)]
4063mod tests {
4064 use super::*;
4065
4066 #[test]
4067 fn test_peat_version() {
4068 let version = peat_version();
4069 assert!(!version.is_empty());
4070 assert!(version.contains('.'));
4071 }
4072
4073 /// `create_node` must honor `TransportConfigFFI.enable_n0_relay` in BOTH
4074 /// postures, proving the runtime relay flag flows the whole stack:
4075 /// peat-ffi `NodeConfig` -> `IrohTransport::from_seed_with_discovery_at_addr`
4076 /// -> `relay_policy_builder` (presets::N0 vs Empty). Both must construct and
4077 /// bind a working endpoint. Binding does not require reaching n0, so this
4078 /// runs offline; the live relay path is covered by peat-mesh's `#[ignore]`d
4079 /// `relay_n0_sync_e2e` and the manual two-device test.
4080 #[cfg(feature = "sync")]
4081 #[test]
4082 fn create_node_honors_enable_n0_relay_in_both_postures() {
4083 fn make(suffix: &str, enable_n0_relay: bool) -> Arc<PeatNode> {
4084 let storage = std::env::temp_dir().join(format!(
4085 "peat-ffi-relay-test-{}-{}",
4086 std::process::id(),
4087 suffix
4088 ));
4089 let _ = std::fs::remove_dir_all(&storage);
4090 let node = create_node(NodeConfig {
4091 app_id: "relay-toggle-ffi-test".to_string(),
4092 shared_key: "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=".to_string(),
4093 bind_address: Some("127.0.0.1:0".to_string()),
4094 storage_path: storage.to_string_lossy().into_owned(),
4095 transport: Some(TransportConfigFFI {
4096 enable_ble: false,
4097 ble_mesh_id: None,
4098 ble_power_profile: None,
4099 transport_preference: None,
4100 collection_routes_json: None,
4101 enable_n0_relay,
4102 }),
4103 })
4104 .unwrap_or_else(|e| panic!("create_node (relay={enable_n0_relay}) failed: {e:?}"));
4105 // Endpoint must be bound either way.
4106 assert!(
4107 !node.endpoint_addr().is_empty(),
4108 "bound endpoint must report an address (relay={enable_n0_relay})"
4109 );
4110 node
4111 }
4112
4113 let _local = make("off", false);
4114 let _relayed = make("on", true);
4115 }
4116
4117 /// Wrapper-tier coverage for the `connect_peer_nowait` UniFFI export added
4118 /// for non-blocking UI callers. These exercise the public `PeatNode`
4119 /// surface through `create_node` — the same entry point Dart/Swift
4120 /// consumers hit — so a signature/argument-slot drift between the method
4121 /// and its hand-written Dart FFI shim, or a panic on the synchronous
4122 /// return path, fails here rather than at downstream link time.
4123 #[cfg(feature = "sync")]
4124 mod connect_peer_nowait_wrapper_tests {
4125 use super::*;
4126
4127 fn test_cfg(storage_path: &str) -> NodeConfig {
4128 NodeConfig {
4129 app_id: "connect-nowait-wrapper-test".to_string(),
4130 shared_key: "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=".to_string(),
4131 bind_address: Some("127.0.0.1:0".to_string()),
4132 storage_path: storage_path.to_string(),
4133 transport: None,
4134 }
4135 }
4136
4137 /// Surface-tier round-trip for the reconnect-supervisor ROSTER methods,
4138 /// driven through a real `PeatNode` (the `#[uniffi::export]` surface a
4139 /// consumer calls) rather than the internal `RosterStore`. A delegation
4140 /// or arg-marshalling bug in `roster_remember` / `roster_list` /
4141 /// `roster_get` / `roster_remove` / `roster_list_by_group` fails here,
4142 /// not at downstream link time. (The `RosterStore` internals have their
4143 /// own unit tests in `roster.rs`; this proves the node methods wire up.)
4144 #[test]
4145 fn roster_remember_list_get_remove_round_trip() {
4146 let tmp = tempfile::tempdir().unwrap();
4147 let node = create_node(test_cfg(tmp.path().to_str().unwrap())).expect("create_node");
4148
4149 // A well-formed node_id from a throwaway node (the id is the roster
4150 // key); roster_remember is the per-member call a consumer makes from
4151 // a scanned join token.
4152 let donor_tmp = tempfile::tempdir().unwrap();
4153 let donor = create_node(test_cfg(donor_tmp.path().to_str().unwrap())).expect("donor");
4154 let peer_id = donor.node_id();
4155 drop(donor);
4156
4157 let peer = PeerInfo {
4158 name: "bravo".to_string(),
4159 node_id: peer_id.clone(),
4160 addresses: vec!["127.0.0.1:19001".to_string()],
4161 relay_url: Some("https://relay.example".to_string()),
4162 };
4163
4164 // remember -> list/get see it, under the right group, fields intact.
4165 node.roster_remember("group-1".to_string(), peer);
4166 assert_eq!(
4167 node.roster_list().len(),
4168 1,
4169 "remembered peer in roster_list"
4170 );
4171 let got = node
4172 .roster_get(peer_id.clone())
4173 .expect("roster_get finds the remembered peer");
4174 assert_eq!(got.node_id, peer_id);
4175 assert_eq!(got.group_id, "group-1");
4176 assert_eq!(got.name, "bravo");
4177 assert_eq!(got.addresses, vec!["127.0.0.1:19001".to_string()]);
4178 assert_eq!(got.relay_url.as_deref(), Some("https://relay.example"));
4179
4180 // list_by_group is scoped to the group.
4181 assert_eq!(node.roster_list_by_group("group-1".to_string()).len(), 1);
4182 assert!(
4183 node.roster_list_by_group("other".to_string()).is_empty(),
4184 "roster_list_by_group is scoped to the group"
4185 );
4186
4187 // remove -> reports presence, then get is None and the roster empties.
4188 assert!(
4189 node.roster_remove(peer_id.clone()),
4190 "roster_remove returns true for a present peer"
4191 );
4192 assert!(
4193 node.roster_get(peer_id.clone()).is_none(),
4194 "roster_get is None after remove"
4195 );
4196 assert!(node.roster_list().is_empty(), "roster empty after remove");
4197 assert!(
4198 !node.roster_remove(peer_id),
4199 "roster_remove returns false for an absent peer"
4200 );
4201 }
4202
4203 /// Surface-tier check that `DocumentChange.origin` is carried through
4204 /// the `subscribe_poll` -> `poll_changes` path on a real `PeatNode`: a
4205 /// locally published document surfaces as a `DocumentChange` tagged
4206 /// `ChangeOrigin::Local`. This proves the wrapped Record actually
4207 /// carries the new origin field out through the subscribe surface (not
4208 /// just that `document_change_from` maps it — that conversion, including
4209 /// the `Remote { peer_id }` variant, is covered by `change_event_tests`).
4210 /// The full cross-node Remote round-trip is a sync-tier e2e gate
4211 /// (peat-mesh), which doesn't substitute for surface coverage but is the
4212 /// right tier for driving an actual peer-synced change.
4213 #[test]
4214 fn subscribe_poll_surfaces_document_change_origin() {
4215 let tmp = tempfile::tempdir().unwrap();
4216 let node = create_node(test_cfg(tmp.path().to_str().unwrap())).expect("create_node");
4217 let sub = node.subscribe_poll().expect("subscribe_poll");
4218
4219 // Publish through the mesh document layer — the same path the
4220 // existing subscribe_poll_drain_and_cancel test uses to feed
4221 // subscribe_to_changes(); a local publish is tagged Local.
4222 let mesh_node = Arc::clone(&node.node);
4223 node.runtime
4224 .block_on(publish_document_into_node(
4225 &mesh_node,
4226 "test",
4227 r#"{"id":"doc-001","x":1}"#,
4228 ))
4229 .expect("publish_document_into_node");
4230
4231 let mut origin: Option<ChangeOrigin> = None;
4232 for _ in 0..40 {
4233 if let Some(ch) = sub
4234 .poll_changes()
4235 .into_iter()
4236 .find(|c| c.collection == "test")
4237 {
4238 origin = Some(ch.origin);
4239 break;
4240 }
4241 std::thread::sleep(std::time::Duration::from_millis(50));
4242 }
4243 assert!(
4244 matches!(origin, Some(ChangeOrigin::Local)),
4245 "a locally published document must surface through poll_changes \
4246 tagged ChangeOrigin::Local; got {origin:?}"
4247 );
4248 }
4249
4250 /// Surface-tier coverage for the `ChangeOrigin::Remote { peer_id }`
4251 /// variant on the subscribe exit path (the sibling test covers `Local`).
4252 /// A remote-sync receive tags the origin-broadcast change `Remote(peer)`;
4253 /// `AutomergeStore::put_with_origin` is exactly the entry point the sync
4254 /// coordinator's receive path uses, so injecting through it fires the
4255 /// same `subscribe_to_changes_with_origin` broadcast `subscribe_poll`
4256 /// reads — deterministically, with no flaky two-node network sync. This
4257 /// asserts the wrapped `DocumentChange` surfaces `Remote { peer_id }`
4258 /// with the id intact: the marshalling regression the internal
4259 /// `document_change_from` test can't see, and the whole point of the
4260 /// field (consumers notify only on remote changes).
4261 #[test]
4262 fn subscribe_poll_surfaces_remote_origin() {
4263 let tmp = tempfile::tempdir().unwrap();
4264 let node = create_node(test_cfg(tmp.path().to_str().unwrap())).expect("create_node");
4265 let sub = node.subscribe_poll().expect("subscribe_poll");
4266
4267 // Inject a Remote-attributed change straight into the origin-tagged
4268 // broadcast via the same store call the receive-from-peer path uses.
4269 // Key is "collection:doc_id" (document_change_from splits on ':').
4270 let doc = automerge::Automerge::new();
4271 node.store
4272 .put_with_origin(
4273 "test:doc-r1",
4274 &doc,
4275 _PeatMeshChangeOrigin::Remote("peerhex".to_string()),
4276 )
4277 .expect("put_with_origin");
4278
4279 let mut origin: Option<ChangeOrigin> = None;
4280 for _ in 0..40 {
4281 if let Some(ch) = sub
4282 .poll_changes()
4283 .into_iter()
4284 .find(|c| c.collection == "test")
4285 {
4286 origin = Some(ch.origin);
4287 break;
4288 }
4289 std::thread::sleep(std::time::Duration::from_millis(50));
4290 }
4291 match origin {
4292 Some(ChangeOrigin::Remote { peer_id }) => assert_eq!(
4293 peer_id, "peerhex",
4294 "Remote origin must carry the publishing origin id"
4295 ),
4296 other => {
4297 panic!("expected ChangeOrigin::Remote {{ peer_id: peerhex }}, got {other:?}")
4298 }
4299 }
4300 }
4301
4302 /// Surface-tier round-trip for `roster_upsert`, which takes a
4303 /// `RosterEntry` *by argument* — the encode direction the
4304 /// `roster_remember` round-trip (a `PeerInfo` arg) never exercises.
4305 /// Proves the `RosterEntry` marshalling on the way in, and that
4306 /// re-upserting the same node id refreshes in place.
4307 #[test]
4308 fn roster_upsert_round_trip_through_the_node() {
4309 let tmp = tempfile::tempdir().unwrap();
4310 let node = create_node(test_cfg(tmp.path().to_str().unwrap())).expect("create_node");
4311
4312 let donor_tmp = tempfile::tempdir().unwrap();
4313 let donor = create_node(test_cfg(donor_tmp.path().to_str().unwrap())).expect("donor");
4314 let peer_id = donor.node_id();
4315 drop(donor);
4316
4317 node.roster_upsert(roster::RosterEntry {
4318 node_id: peer_id.clone(),
4319 group_id: "g1".to_string(),
4320 name: "charlie".to_string(),
4321 addresses: vec!["127.0.0.1:7000".to_string()],
4322 relay_url: None,
4323 last_seen_ms: 42,
4324 });
4325 let got = node
4326 .roster_get(peer_id.clone())
4327 .expect("roster_get after upsert");
4328 assert_eq!(got.group_id, "g1");
4329 assert_eq!(got.name, "charlie");
4330 assert_eq!(got.addresses, vec!["127.0.0.1:7000".to_string()]);
4331 assert_eq!(got.last_seen_ms, 42);
4332
4333 // Re-upsert on the same node id refreshes in place (no duplicate).
4334 node.roster_upsert(roster::RosterEntry {
4335 node_id: peer_id.clone(),
4336 group_id: "g1".to_string(),
4337 name: "charlie-2".to_string(),
4338 addresses: vec![],
4339 relay_url: Some("https://relay.example".to_string()),
4340 last_seen_ms: 99,
4341 });
4342 assert_eq!(
4343 node.roster_list().len(),
4344 1,
4345 "re-upsert must refresh, not duplicate"
4346 );
4347 let got2 = node
4348 .roster_get(peer_id)
4349 .expect("roster_get after re-upsert");
4350 assert_eq!(got2.name, "charlie-2");
4351 assert_eq!(got2.relay_url.as_deref(), Some("https://relay.example"));
4352 }
4353
4354 /// Surface-tier smoke for the three reconnect triggers
4355 /// (`reconnect_known_peers`, `wake_reconnect`, `on_peer_observed`) — they
4356 /// are fire-and-forget (return void), so this drives them through the
4357 /// `PeatNode` UniFFI surface and asserts they neither panic nor corrupt
4358 /// roster state, both with an empty roster and with a remembered offline
4359 /// peer (which makes them actually run a reconnect pass). Catches an
4360 /// arg-marshalling or wiring break that an internal supervisor unit test
4361 /// can't see.
4362 #[test]
4363 fn reconnect_triggers_callable_through_the_node() {
4364 let tmp = tempfile::tempdir().unwrap();
4365 let node = create_node(test_cfg(tmp.path().to_str().unwrap())).expect("create_node");
4366
4367 // Empty roster: all three are safe no-ops, including on_peer_observed
4368 // for an unknown id.
4369 node.reconnect_known_peers();
4370 node.wake_reconnect();
4371 node.on_peer_observed("not-a-known-peer".to_string());
4372 assert!(
4373 node.roster_list().is_empty(),
4374 "no-op triggers must not invent roster entries"
4375 );
4376
4377 // With a remembered (unreachable) peer, the same calls drive a real
4378 // reconnect pass without panicking and leave the roster intact.
4379 let donor_tmp = tempfile::tempdir().unwrap();
4380 let donor = create_node(test_cfg(donor_tmp.path().to_str().unwrap())).expect("donor");
4381 let peer_id = donor.node_id();
4382 drop(donor);
4383 node.roster_remember(
4384 "g1".to_string(),
4385 PeerInfo {
4386 name: "bravo".to_string(),
4387 node_id: peer_id.clone(),
4388 addresses: vec!["127.0.0.1:65000".to_string()], // unreachable
4389 relay_url: None,
4390 },
4391 );
4392
4393 node.reconnect_known_peers();
4394 node.on_peer_observed(peer_id.clone());
4395 node.wake_reconnect();
4396
4397 assert_eq!(node.roster_list().len(), 1);
4398 assert!(
4399 node.roster_get(peer_id).is_some(),
4400 "remembered peer survives the reconnect passes"
4401 );
4402 }
4403
4404 /// Happy path: a fire-and-forget dial returns synchronously and the
4405 /// background task converges the connection. Two in-process nodes share
4406 /// `app_id`/`shared_key`; node A dials node B's bound loopback socket
4407 /// via `connect_peer_nowait` and B must show up in A's
4408 /// `connected_peers` shortly after — proving the spawned dial +
4409 /// formation handshake + `emit_peer_connected` path runs end to end.
4410 #[test]
4411 fn nowait_connects_two_loopback_nodes() {
4412 let tmp_a = tempfile::tempdir().unwrap();
4413 let tmp_b = tempfile::tempdir().unwrap();
4414 let node_a = create_node(test_cfg(tmp_a.path().to_str().unwrap())).expect("node_a");
4415 let node_b = create_node(test_cfg(tmp_b.path().to_str().unwrap())).expect("node_b");
4416
4417 let addr_b = node_b
4418 .endpoint_socket_addr()
4419 .expect("node_b must report a bound loopback socket addr");
4420 let peer_b = PeerInfo {
4421 name: "node-b".to_string(),
4422 node_id: node_b.node_id(),
4423 addresses: vec![addr_b],
4424 relay_url: None,
4425 };
4426
4427 // The whole point of the API: the call returns without blocking on
4428 // the dial.
4429 node_a
4430 .connect_peer_nowait(peer_b)
4431 .expect("connect_peer_nowait must return Ok immediately");
4432
4433 // The background dial + handshake converges shortly after; poll with
4434 // a bounded timeout rather than a fixed sleep.
4435 let target = node_b.node_id();
4436 let mut connected = false;
4437 for _ in 0..120 {
4438 if node_a.connected_peers().contains(&target) {
4439 connected = true;
4440 break;
4441 }
4442 std::thread::sleep(std::time::Duration::from_millis(50));
4443 }
4444 assert!(
4445 connected,
4446 "node_b must join node_a's connected_peers after a background connect_peer_nowait dial"
4447 );
4448 }
4449
4450 /// Contract + robustness: even for an unreachable peer the call returns
4451 /// `Ok(())` immediately (the blocking `connect_peer` would stall on the
4452 /// dial), and the doomed background task neither panics nor wedges the
4453 /// node. Uses a valid node_id whose owner has been dropped, pointed at a
4454 /// dead loopback port, so neither the direct address nor mDNS can
4455 /// resolve it.
4456 #[test]
4457 fn nowait_returns_immediately_for_unreachable_peer_without_panic() {
4458 let tmp = tempfile::tempdir().unwrap();
4459 let node = create_node(test_cfg(tmp.path().to_str().unwrap())).expect("create_node");
4460
4461 let donor_tmp = tempfile::tempdir().unwrap();
4462 let donor = create_node(test_cfg(donor_tmp.path().to_str().unwrap())).expect("donor");
4463 let dead_id = donor.node_id();
4464 drop(donor); // owner gone — the id is well-formed but unroutable.
4465
4466 let peer = PeerInfo {
4467 name: "unreachable".to_string(),
4468 node_id: dead_id.clone(),
4469 addresses: vec!["127.0.0.1:1".to_string()],
4470 relay_url: None,
4471 };
4472 node.connect_peer_nowait(peer).expect(
4473 "connect_peer_nowait must return Ok immediately, even for an unreachable peer",
4474 );
4475
4476 // Let the doomed dial run; the node must stay responsive and the
4477 // peer must never join.
4478 for _ in 0..10 {
4479 std::thread::sleep(std::time::Duration::from_millis(50));
4480 }
4481 assert!(
4482 !node.connected_peers().contains(&dead_id),
4483 "an unreachable peer must never appear in connected_peers"
4484 );
4485 }
4486 }
4487
4488 #[test]
4489 fn test_encode_track() {
4490 let track = TrackData {
4491 track_id: "track-001".to_string(),
4492 source_node: "node-1".to_string(),
4493 position: Position {
4494 lat: 34.0522,
4495 lon: -118.2437,
4496 hae: Some(100.0),
4497 },
4498 velocity: Some(Velocity {
4499 bearing: 90.0,
4500 speed_mps: 10.0,
4501 }),
4502 classification: "a-f-G-U-C".to_string(),
4503 confidence: 0.95,
4504 cell_id: Some("cell-1".to_string()),
4505 formation_id: None,
4506 };
4507
4508 let result = encode_track_to_cot(track);
4509 assert!(result.is_ok());
4510
4511 let xml = result.unwrap();
4512 assert!(xml.contains("<event"));
4513 assert!(xml.contains("track-001"));
4514 }
4515
4516 #[test]
4517 fn test_encode_minimal_track() {
4518 let track = TrackData {
4519 track_id: "t1".to_string(),
4520 source_node: "p1".to_string(),
4521 position: Position {
4522 lat: 0.0,
4523 lon: 0.0,
4524 hae: None,
4525 },
4526 velocity: None,
4527 classification: "a-u-G".to_string(),
4528 confidence: 0.5,
4529 cell_id: None,
4530 formation_id: None,
4531 };
4532
4533 let result = encode_track_to_cot(track);
4534 assert!(result.is_ok());
4535 }
4536
4537 #[test]
4538 fn test_invalid_track_id() {
4539 let track = TrackData {
4540 track_id: "".to_string(), // Empty - should fail
4541 source_node: "p1".to_string(),
4542 position: Position {
4543 lat: 0.0,
4544 lon: 0.0,
4545 hae: None,
4546 },
4547 velocity: None,
4548 classification: "a-u-G".to_string(),
4549 confidence: 0.5,
4550 cell_id: None,
4551 formation_id: None,
4552 };
4553
4554 let result = encode_track_to_cot(track);
4555 assert!(result.is_err());
4556 }
4557
4558 #[test]
4559 fn test_helper_functions() {
4560 let pos = create_position(34.0, -118.0, Some(50.0));
4561 assert_eq!(pos.lat, 34.0);
4562 assert_eq!(pos.lon, -118.0);
4563 assert_eq!(pos.hae, Some(50.0));
4564
4565 let vel = create_velocity(45.0, 15.0);
4566 assert_eq!(vel.bearing, 45.0);
4567 assert_eq!(vel.speed_mps, 15.0);
4568 }
4569
4570 /// Tests for the generic `publish_document_into_node` helper that backs
4571 /// `Java_..._publishDocumentJni`. Foundation step 3 of the
4572 /// peat-mesh-completion / peat-btle-reduction work — see
4573 /// `PEAT-MESH-COMPLETION-0.9.0.md`.
4574 ///
4575 /// Running through `tokio::runtime::Runtime::block_on` rather than a
4576 /// `#[tokio::test]` attribute matches the rest of peat-ffi (which doesn't
4577 /// pull tokio macros into dev-dependencies just for tests) and exercises
4578 /// the same `runtime.block_on(...)` shape the JNI wrapper itself uses.
4579 #[cfg(feature = "sync")]
4580 mod publish_document_tests {
4581 use super::*;
4582 use peat_mesh::sync::traits::DataSyncBackend;
4583 use peat_mesh::sync::InMemoryBackend;
4584
4585 fn fresh_node() -> peat_mesh::Node {
4586 let backend: Arc<dyn DataSyncBackend> = Arc::new(InMemoryBackend::new_initialized());
4587 peat_mesh::Node::new(backend)
4588 }
4589
4590 fn rt() -> tokio::runtime::Runtime {
4591 tokio::runtime::Builder::new_current_thread()
4592 .enable_all()
4593 .build()
4594 .expect("runtime")
4595 }
4596
4597 /// Publishing a JSON object with an explicit `"id"` field round-trips
4598 /// through the node: the returned id matches, and `node.get(...)`
4599 /// yields a Document carrying the body fields verbatim.
4600 #[test]
4601 fn round_trip_with_explicit_id() {
4602 let rt = rt();
4603 rt.block_on(async {
4604 let node = fresh_node();
4605 let json = r#"{
4606 "id": "chat-001",
4607 "sender": "ALPHA-1",
4608 "text": "hello",
4609 "timestamp": 1700000000000
4610 }"#;
4611 let id = publish_document_into_node(&node, "chats", json)
4612 .await
4613 .expect("publish");
4614 assert_eq!(id, "chat-001");
4615
4616 let got = node
4617 .get("chats", &"chat-001".to_string())
4618 .await
4619 .expect("get")
4620 .expect("found");
4621 assert_eq!(
4622 got.fields.get("sender").and_then(|v| v.as_str()),
4623 Some("ALPHA-1")
4624 );
4625 assert_eq!(
4626 got.fields.get("text").and_then(|v| v.as_str()),
4627 Some("hello")
4628 );
4629 assert!(
4630 !got.fields.contains_key("id"),
4631 "id is hoisted to Document::id, not duplicated in fields"
4632 );
4633 });
4634 }
4635
4636 /// JSON without an `"id"` field still publishes; the backend assigns
4637 /// one (UUID under `InMemoryBackend`). The returned id is non-empty
4638 /// and the doc is retrievable by it.
4639 #[test]
4640 fn id_assignment_when_absent() {
4641 let rt = rt();
4642 rt.block_on(async {
4643 let node = fresh_node();
4644 let json = r#"{"text":"orphan","sender":"BRAVO-2"}"#;
4645 let id = publish_document_into_node(&node, "chats", json)
4646 .await
4647 .expect("publish");
4648 assert!(!id.is_empty(), "backend must assign an id");
4649
4650 let got = node.get("chats", &id).await.expect("get").expect("found");
4651 assert_eq!(
4652 got.fields.get("text").and_then(|v| v.as_str()),
4653 Some("orphan")
4654 );
4655 });
4656 }
4657
4658 /// Malformed JSON returns Err — the JNI wrapper translates this into
4659 /// an empty-string return to the Java caller.
4660 #[test]
4661 fn malformed_json_errors() {
4662 let rt = rt();
4663 rt.block_on(async {
4664 let node = fresh_node();
4665 let result = publish_document_into_node(&node, "chats", "not-json").await;
4666 assert!(result.is_err());
4667 });
4668 }
4669
4670 /// Non-object JSON (array, string, number) returns Err — the
4671 /// document model requires an object at the top level.
4672 #[test]
4673 fn non_object_json_errors() {
4674 let rt = rt();
4675 rt.block_on(async {
4676 let node = fresh_node();
4677 let result = publish_document_into_node(&node, "chats", "[1, 2, 3]").await;
4678 assert!(result.is_err());
4679 });
4680 }
4681
4682 /// Non-string id (e.g. integer) is treated as id-absent — the backend
4683 /// assigns one rather than coercing the integer. Aligns with
4684 /// peat-protocol's `value_to_mesh_document`, which made the same
4685 /// decision in PR #802 round-1 review.
4686 #[test]
4687 fn non_string_id_falls_back_to_assigned() {
4688 let rt = rt();
4689 rt.block_on(async {
4690 let node = fresh_node();
4691 let json = r#"{"id":42,"text":"weird"}"#;
4692 let id = publish_document_into_node(&node, "chats", json)
4693 .await
4694 .expect("publish");
4695 assert_ne!(id, "42", "non-string id must be discarded, not coerced");
4696 assert!(!id.is_empty());
4697 });
4698 }
4699
4700 /// Origin-aware variant publishes successfully and threads the
4701 /// origin string through to peat-mesh. ADR-059 Amendment 2 Slice
4702 /// 1.b.4 requires this so the plugin's `BleDecodedDocumentBridge`
4703 /// can ingest 0xB6 frames into the doc store without re-emitting
4704 /// them back out to BLE — `Some("ble")` triggers the same
4705 /// loop-prevention fan-out skip the existing `ingestPositionJni`
4706 /// path uses.
4707 #[test]
4708 fn origin_variant_publishes_with_explicit_id() {
4709 let rt = rt();
4710 rt.block_on(async {
4711 let node = fresh_node();
4712 let json = r#"{"id":"ble-decoded-001","sender":"OBS-1","text":"x"}"#;
4713 let id = publish_document_into_node_with_origin(
4714 &node,
4715 "chats",
4716 json,
4717 Some("ble".to_string()),
4718 )
4719 .await
4720 .expect("publish_with_origin");
4721 assert_eq!(id, "ble-decoded-001");
4722
4723 let got = node
4724 .get("chats", &"ble-decoded-001".to_string())
4725 .await
4726 .expect("get")
4727 .expect("found");
4728 assert_eq!(
4729 got.fields.get("sender").and_then(|v| v.as_str()),
4730 Some("OBS-1")
4731 );
4732 });
4733 }
4734
4735 /// `None` origin makes the helper behave identically to the plain
4736 /// publish path — locks the back-compat invariant the wrapper
4737 /// `publish_document_into_node` relies on.
4738 #[test]
4739 fn origin_variant_with_none_matches_plain_publish() {
4740 let rt = rt();
4741 rt.block_on(async {
4742 let node = fresh_node();
4743 let json = r#"{"id":"plain-001","text":"plain"}"#;
4744 let id = publish_document_into_node_with_origin(&node, "chats", json, None)
4745 .await
4746 .expect("publish_with_origin(None)");
4747 assert_eq!(id, "plain-001");
4748
4749 let got = node
4750 .get("chats", &"plain-001".to_string())
4751 .await
4752 .expect("get")
4753 .expect("found");
4754 assert_eq!(
4755 got.fields.get("text").and_then(|v| v.as_str()),
4756 Some("plain")
4757 );
4758 });
4759 }
4760 }
4761
4762 /// Tests for the BLE-translator helpers backing the `ingest*Jni`
4763 /// family. Slice 1.b.2.2 of ADR-059 — the inbound BLE→Node→iroh path
4764 /// now goes directly through `BleTranslator` + `Node::publish_with_origin`
4765 /// (the legacy `BleGateway` wrapper was deleted; its responsibilities
4766 /// composed in-line here).
4767 #[cfg(all(feature = "sync", feature = "bluetooth"))]
4768 mod ingest_position_tests {
4769 use super::*;
4770 use peat_mesh::sync::traits::DataSyncBackend;
4771 use peat_mesh::sync::InMemoryBackend;
4772 use peat_protocol::sync::ble_translation::BleTranslator;
4773
4774 struct Fixture {
4775 translator: BleTranslator,
4776 node: peat_mesh::Node,
4777 }
4778
4779 fn fresh_fixture() -> Fixture {
4780 let backend: Arc<dyn DataSyncBackend> = Arc::new(InMemoryBackend::new_initialized());
4781 Fixture {
4782 translator: BleTranslator::with_defaults(),
4783 node: peat_mesh::Node::new(backend),
4784 }
4785 }
4786
4787 fn rt() -> tokio::runtime::Runtime {
4788 tokio::runtime::Builder::new_current_thread()
4789 .enable_all()
4790 .build()
4791 .expect("runtime")
4792 }
4793
4794 /// Happy path: a fully-populated JSON envelope ingests into the
4795 /// tracks collection, the returned id is the translator's
4796 /// BLE-prefixed track id (`ble-` + uppercase 8-hex peripheral id),
4797 /// and the resulting Document carries the position fields plus
4798 /// `ble_origin: true` so any outbound BLE re-encoder filtering
4799 /// on that marker breaks the loop.
4800 #[test]
4801 fn round_trip_full_envelope() {
4802 let rt = rt();
4803 rt.block_on(async {
4804 let fx = fresh_fixture();
4805 // peripheral_id 0xCAFE0001 = 3_405_643_777 — sanity-check the
4806 // hex form by using a constant rather than hand-converting.
4807 const PERIPHERAL: u32 = 0xCAFE_0001;
4808 let json = format!(
4809 r#"{{
4810 "lat": 40.7128,
4811 "lon": -74.0060,
4812 "altitude": 100.0,
4813 "accuracy": 5.0,
4814 "peripheral_id": {},
4815 "callsign": "SCOUT-CAFE",
4816 "mesh_id": "29C916FA"
4817 }}"#,
4818 PERIPHERAL
4819 );
4820 let id = ingest_position_via_translator(&fx.translator, &fx.node, &json)
4821 .await
4822 .expect("ingest");
4823 // Translator format: ble_id_prefix ("ble-") + uppercase 8-hex.
4824 assert_eq!(id, format!("ble-{:08X}", PERIPHERAL));
4825
4826 let doc = fx
4827 .node
4828 .get(fx.translator.tracks_collection(), &id)
4829 .await
4830 .expect("get")
4831 .expect("found");
4832 assert_eq!(
4833 doc.fields.get("ble_origin"),
4834 Some(&serde_json::Value::Bool(true)),
4835 "ble_origin marker required for outbound loop suppression"
4836 );
4837 });
4838 }
4839
4840 /// Optional fields can be omitted: altitude, accuracy, callsign,
4841 /// mesh_id all default to None and the ingest still succeeds.
4842 #[test]
4843 fn omits_optional_fields() {
4844 let rt = rt();
4845 rt.block_on(async {
4846 let fx = fresh_fixture();
4847 let json = r#"{
4848 "lat": 40.7128,
4849 "lon": -74.0060,
4850 "peripheral_id": 1
4851 }"#;
4852 let id = ingest_position_via_translator(&fx.translator, &fx.node, json)
4853 .await
4854 .expect("ingest");
4855 assert_eq!(id, "ble-00000001");
4856 });
4857 }
4858
4859 /// Missing required fields (lat/lon/peripheral_id) error rather
4860 /// than silently defaulting. The JNI wrapper translates the Err
4861 /// into an empty-string Java return.
4862 #[test]
4863 fn missing_required_fields_errors() {
4864 let rt = rt();
4865 rt.block_on(async {
4866 let fx = fresh_fixture();
4867 let json_no_lat = r#"{"lon": -74.0, "peripheral_id": 1}"#;
4868 assert!(
4869 ingest_position_via_translator(&fx.translator, &fx.node, json_no_lat)
4870 .await
4871 .is_err()
4872 );
4873
4874 let json_no_id = r#"{"lat": 40.0, "lon": -74.0}"#;
4875 assert!(
4876 ingest_position_via_translator(&fx.translator, &fx.node, json_no_id)
4877 .await
4878 .is_err()
4879 );
4880 });
4881 }
4882
4883 /// Malformed JSON errors (matches the contract of the JNI wrapper).
4884 #[test]
4885 fn malformed_json_errors() {
4886 let rt = rt();
4887 rt.block_on(async {
4888 let fx = fresh_fixture();
4889 let result =
4890 ingest_position_via_translator(&fx.translator, &fx.node, "not-json").await;
4891 assert!(result.is_err());
4892 });
4893 }
4894
4895 /// Regression for PR #804 round-1 [WARNING]: a Kotlin caller that
4896 /// serializes peripheral_id from a signed `Int` field (rather than
4897 /// `Long`/`UInt`) emits a negative JSON literal for any u32 with
4898 /// the high bit set. The parser must reinterpret-cast through i32
4899 /// to recover the original u32; the resulting track id must match
4900 /// what the same u32 written as a positive literal produced.
4901 #[test]
4902 fn peripheral_id_negative_int_form_recovers_to_same_u32() {
4903 let rt = rt();
4904 rt.block_on(async {
4905 let fx = fresh_fixture();
4906 // 0xCAFE_0001 = 3_405_643_777 as u32; -889_323_519 is the
4907 // sign-extended Int form (verified: (3_405_643_777_i64 -
4908 // 4_294_967_296) == -889_323_519).
4909 const POSITIVE: i64 = 3_405_643_777;
4910 const NEGATIVE: i64 = -889_323_519;
4911 let expected_id = "ble-CAFE0001";
4912
4913 let positive_json = format!(
4914 r#"{{ "lat": 40.0, "lon": -74.0, "peripheral_id": {} }}"#,
4915 POSITIVE
4916 );
4917 let negative_json = format!(
4918 r#"{{ "lat": 40.0, "lon": -74.0, "peripheral_id": {} }}"#,
4919 NEGATIVE
4920 );
4921
4922 let id_pos =
4923 ingest_position_via_translator(&fx.translator, &fx.node, &positive_json)
4924 .await
4925 .expect("positive form ingests");
4926 assert_eq!(id_pos, expected_id);
4927
4928 let id_neg =
4929 ingest_position_via_translator(&fx.translator, &fx.node, &negative_json)
4930 .await
4931 .expect("negative (Kotlin Int) form ingests");
4932 assert_eq!(
4933 id_neg, expected_id,
4934 "both forms must yield the same track id"
4935 );
4936 });
4937 }
4938
4939 /// Out-of-range values reject rather than silently truncate.
4940 /// Without bounds-checking, a >u32::MAX value would `as u32`
4941 /// truncate and collide distinct logical IDs onto the same
4942 /// translator-emitted track id, mis-attributing positions.
4943 #[test]
4944 fn peripheral_id_out_of_range_errors() {
4945 let rt = rt();
4946 rt.block_on(async {
4947 let fx = fresh_fixture();
4948
4949 // u32::MAX + 1
4950 let too_big = r#"{ "lat": 40.0, "lon": -74.0, "peripheral_id": 4294967296 }"#;
4951 assert!(
4952 ingest_position_via_translator(&fx.translator, &fx.node, too_big)
4953 .await
4954 .is_err()
4955 );
4956
4957 // i32::MIN - 1
4958 let too_small = r#"{ "lat": 40.0, "lon": -74.0, "peripheral_id": -2147483649 }"#;
4959 assert!(
4960 ingest_position_via_translator(&fx.translator, &fx.node, too_small)
4961 .await
4962 .is_err()
4963 );
4964 });
4965 }
4966
4967 /// u32::MAX and i32::MIN are valid boundaries. u32::MAX exercises
4968 /// the top of the positive form; i32::MIN exercises the top of the
4969 /// negative-Int form (a u32 with `high_bit=1, rest=0` =
4970 /// `0x8000_0000` = `-2_147_483_648` as Int).
4971 #[test]
4972 fn peripheral_id_boundaries_accepted() {
4973 let rt = rt();
4974 rt.block_on(async {
4975 let fx = fresh_fixture();
4976
4977 let max_json = r#"{ "lat": 40.0, "lon": -74.0, "peripheral_id": 4294967295 }"#;
4978 let id = ingest_position_via_translator(&fx.translator, &fx.node, max_json)
4979 .await
4980 .expect("u32::MAX");
4981 assert_eq!(id, "ble-FFFFFFFF");
4982
4983 let min_int_json = r#"{ "lat": 40.0, "lon": -74.0, "peripheral_id": -2147483648 }"#;
4984 let id = ingest_position_via_translator(&fx.translator, &fx.node, min_int_json)
4985 .await
4986 .expect("i32::MIN as Int form");
4987 assert_eq!(id, "ble-80000000");
4988 });
4989 }
4990
4991 /// Slice 1.b.2.2: the rewire publishes through
4992 /// `Node::publish_with_origin(.., Some("ble"))`, so the resulting
4993 /// `ChangeEvent::Updated` must carry `origin = Some("ble")`. This
4994 /// is the load-bearing assertion that `TransportManager` fan-out
4995 /// can suppress the BLE→Node→observer→BLE same-node echo without
4996 /// it, the loop-break invariant is gone.
4997 #[tokio::test]
4998 async fn ingest_emits_observer_event_with_ble_origin() {
4999 use peat_mesh::sync::types::{ChangeEvent, Query};
5000 let fx = fresh_fixture();
5001 let mut tracks = fx
5002 .node
5003 .observe(fx.translator.tracks_collection(), &Query::All)
5004 .expect("observe");
5005
5006 let json = r#"{
5007 "lat": 40.7,
5008 "lon": -74.0,
5009 "peripheral_id": 1,
5010 "callsign": "SCOUT-1"
5011 }"#;
5012 let _ = ingest_position_via_translator(&fx.translator, &fx.node, json)
5013 .await
5014 .expect("ingest");
5015
5016 // Skip the Initial snapshot, then assert the Updated event's origin.
5017 loop {
5018 let ev = tracks.receiver.recv().await.expect("event");
5019 if let ChangeEvent::Updated { origin, .. } = ev {
5020 assert_eq!(
5021 origin,
5022 Some("ble".to_string()),
5023 "ingestPositionJni must publish with Some(\"ble\") origin per ADR-059"
5024 );
5025 break;
5026 }
5027 }
5028 }
5029 }
5030
5031 /// Tests for the outbound BLE-frame fan-out path (ADR-059 Slice 1.b.2).
5032 /// The JNI surface itself can't be exercised without a JVM, but the
5033 /// underlying mechanism — `TransportManager` registers a translator + sink,
5034 /// observer pushes through encode_outbound, sink receives bytes — is fully
5035 /// exercisable with a recording sink standing in for `JniOutboundSink`.
5036 #[cfg(all(feature = "sync", feature = "bluetooth"))]
5037 mod outbound_frame_tests {
5038 use super::*;
5039 use peat_mesh::sync::traits::DataSyncBackend;
5040 use peat_mesh::sync::InMemoryBackend;
5041 use peat_mesh::transport::{
5042 FanoutHandle, OutboundSink, TranslationContext, Translator,
5043 TranslatorRegistrationConfig,
5044 };
5045 use peat_protocol::sync::ble_translation::BleTranslator;
5046 use peat_protocol::transport::{TransportManager, TransportManagerConfig};
5047 use std::sync::Mutex as StdMutex;
5048 use tokio::time::{timeout, Duration};
5049
5050 /// Records `(transport_id, collection, bytes)` triples each time
5051 /// `send_outbound` fires. Stand-in for the JNI dispatcher in unit
5052 /// tests — we assert against the recorded frames rather than calling
5053 /// into a JVM.
5054 #[derive(Default)]
5055 struct RecordingSink {
5056 frames: StdMutex<Vec<(String, String, Vec<u8>)>>,
5057 }
5058
5059 #[async_trait::async_trait]
5060 impl OutboundSink for RecordingSink {
5061 async fn send_outbound(
5062 &self,
5063 bytes: Vec<u8>,
5064 ctx: &TranslationContext,
5065 ) -> anyhow::Result<()> {
5066 let collection = ctx.collection.clone().unwrap_or_default();
5067 self.frames
5068 .lock()
5069 .unwrap()
5070 .push(("ble".to_string(), collection, bytes));
5071 Ok(())
5072 }
5073 }
5074
5075 impl RecordingSink {
5076 fn snapshot(&self) -> Vec<(String, String, Vec<u8>)> {
5077 self.frames.lock().unwrap().clone()
5078 }
5079 }
5080
5081 struct Fixture {
5082 node: Arc<peat_mesh::Node>,
5083 translator: Arc<BleTranslator>,
5084 transport_manager: TransportManager,
5085 sink: Arc<RecordingSink>,
5086 }
5087
5088 fn fixture() -> Fixture {
5089 let backend: Arc<dyn DataSyncBackend> = Arc::new(InMemoryBackend::new_initialized());
5090 Fixture {
5091 node: Arc::new(peat_mesh::Node::new(backend)),
5092 translator: Arc::new(BleTranslator::with_defaults()),
5093 transport_manager: TransportManager::new(TransportManagerConfig::default()),
5094 sink: Arc::new(RecordingSink::default()),
5095 }
5096 }
5097
5098 async fn register_and_start(fx: &Fixture) -> anyhow::Result<FanoutHandle> {
5099 let translator_dyn: Arc<dyn Translator> = fx.translator.clone();
5100 let sink_dyn: Arc<dyn OutboundSink> = fx.sink.clone();
5101 fx.transport_manager
5102 .register_translator(
5103 translator_dyn,
5104 sink_dyn,
5105 TranslatorRegistrationConfig::ble(),
5106 )
5107 .await?;
5108 fx.transport_manager.start_fanout(
5109 Arc::clone(&fx.node),
5110 vec![fx.translator.tracks_collection().to_string()],
5111 )
5112 }
5113
5114 /// Wait up to 1s for the recording sink to receive at least
5115 /// `expected_count` frames. The fan-out is asynchronous (observer
5116 /// task → channel → drain task → sink), so a brief poll loop is
5117 /// the right shape — fixed sleeps would be flaky.
5118 async fn wait_for_frames(sink: &RecordingSink, expected: usize) {
5119 let _ = timeout(Duration::from_secs(1), async {
5120 loop {
5121 if sink.snapshot().len() >= expected {
5122 return;
5123 }
5124 tokio::time::sleep(Duration::from_millis(20)).await;
5125 }
5126 })
5127 .await;
5128 }
5129
5130 /// Baseline: a doc published via the iroh-side bridge (no
5131 /// `Some("ble")` origin) reaches the BLE sink — the
5132 /// translator-encode + drain-task path is wired correctly.
5133 #[tokio::test]
5134 async fn iroh_origin_doc_reaches_ble_sink() {
5135 let fx = fixture();
5136 let _h = register_and_start(&fx).await.expect("register");
5137
5138 // No origin = "iroh-side" doc. The fan-out should encode + deliver.
5139 let doc = peat_mesh::sync::types::Document::with_id("ble-00000001".to_string(), {
5140 let mut f = std::collections::HashMap::new();
5141 f.insert("lat".to_string(), serde_json::json!(40.0));
5142 f.insert("lon".to_string(), serde_json::json!(-74.0));
5143 f.insert(
5144 "source_node".to_string(),
5145 serde_json::json!("iroh-00000001"),
5146 );
5147 f.insert("hae".to_string(), serde_json::json!(100.0));
5148 f.insert("cep".to_string(), serde_json::json!(5.0));
5149 f.insert("classification".to_string(), serde_json::json!("a-f-G-U-C"));
5150 f.insert("confidence".to_string(), serde_json::json!(0.9));
5151 f.insert("category".to_string(), serde_json::json!("friendly"));
5152 f.insert("callsign".to_string(), serde_json::json!("ALPHA-1"));
5153 f.insert(
5154 "created_at".to_string(),
5155 serde_json::json!(1_700_000_000_000_i64),
5156 );
5157 f.insert(
5158 "last_update".to_string(),
5159 serde_json::json!(1_700_000_000_000_i64),
5160 );
5161 f
5162 });
5163 fx.node.publish("tracks", doc).await.expect("publish");
5164
5165 wait_for_frames(&fx.sink, 1).await;
5166 let frames = fx.sink.snapshot();
5167 assert!(
5168 !frames.is_empty(),
5169 "iroh-origin track must reach ble sink; got 0 frames"
5170 );
5171 let (transport, collection, bytes) = &frames[0];
5172 assert_eq!(transport, "ble");
5173 assert_eq!(collection, "tracks");
5174 assert!(!bytes.is_empty(), "encoded bytes must be non-empty");
5175 }
5176
5177 /// Loop suppression: a doc with `origin = Some("ble")` (i.e.
5178 /// ingestPositionJni's output) MUST NOT be re-encoded back out the
5179 /// BLE sink. This is the same-node echo-loop break ADR-059 §
5180 /// "Origin propagation" requires.
5181 #[tokio::test]
5182 async fn ble_origin_doc_does_not_re_encode_to_ble_sink() {
5183 let fx = fixture();
5184 let _h = register_and_start(&fx).await.expect("register");
5185
5186 let doc = peat_mesh::sync::types::Document::with_id("ble-CAFE0001".to_string(), {
5187 let mut f = std::collections::HashMap::new();
5188 f.insert("lat".to_string(), serde_json::json!(40.0));
5189 f.insert("lon".to_string(), serde_json::json!(-74.0));
5190 f.insert("ble_origin".to_string(), serde_json::json!(true));
5191 f
5192 });
5193
5194 fx.node
5195 .publish_with_origin("tracks", doc, Some("ble".to_string()))
5196 .await
5197 .expect("publish");
5198
5199 // Hold the awaited window slightly past the steady-state
5200 // observer fan-out latency; if loop suppression is broken,
5201 // the sink would have received the encoded frame by now.
5202 tokio::time::sleep(Duration::from_millis(150)).await;
5203
5204 let frames = fx.sink.snapshot();
5205 assert!(
5206 frames.is_empty(),
5207 "ble-origin doc must be suppressed from outbound BLE \
5208 (ADR-059 same-node echo break); got {} frames",
5209 frames.len()
5210 );
5211 }
5212
5213 /// Dropping the `FanoutHandle` (mirroring
5214 /// `unsubscribeOutboundFramesJni`'s teardown) stops further
5215 /// frames from reaching the sink.
5216 #[tokio::test]
5217 async fn drop_handle_stops_subsequent_delivery() {
5218 let fx = fixture();
5219 let h = register_and_start(&fx).await.expect("register");
5220
5221 // Sanity: first publish reaches sink.
5222 fx.node
5223 .publish(
5224 "tracks",
5225 peat_mesh::sync::types::Document::with_id("ble-00000001".to_string(), {
5226 let mut f = std::collections::HashMap::new();
5227 f.insert("lat".to_string(), serde_json::json!(40.0));
5228 f.insert("lon".to_string(), serde_json::json!(-74.0));
5229 f.insert("source_node".to_string(), serde_json::json!("iroh-1"));
5230 f.insert("callsign".to_string(), serde_json::json!("A"));
5231 f.insert("hae".to_string(), serde_json::json!(0.0));
5232 f.insert("cep".to_string(), serde_json::json!(0.0));
5233 f.insert("classification".to_string(), serde_json::json!("a-f-G-U-C"));
5234 f.insert("confidence".to_string(), serde_json::json!(0.5));
5235 f.insert("category".to_string(), serde_json::json!("friendly"));
5236 f.insert(
5237 "created_at".to_string(),
5238 serde_json::json!(1_700_000_000_000_i64),
5239 );
5240 f.insert(
5241 "last_update".to_string(),
5242 serde_json::json!(1_700_000_000_000_i64),
5243 );
5244 f
5245 }),
5246 )
5247 .await
5248 .expect("publish-1");
5249 wait_for_frames(&fx.sink, 1).await;
5250 let pre_drop_count = fx.sink.snapshot().len();
5251 assert!(pre_drop_count >= 1);
5252
5253 // Drop the handle — observer tasks for this fan-out cancel.
5254 // The cancellation token is set synchronously on drop, but the
5255 // observer task only notices on its next `select!` poll, so we
5256 // yield+sleep briefly to let the runtime actually cancel the
5257 // task before producing the new broadcast. Without this gap,
5258 // tokio::select!'s non-biased polling may race the new event
5259 // ahead of the cancellation arm. (peat-mesh's observer_task
5260 // would benefit from `biased;` to make this deterministic;
5261 // tracked as a Slice 2 hardening item.)
5262 drop(h);
5263 tokio::time::sleep(Duration::from_millis(50)).await;
5264
5265 // Publish AFTER cancellation has settled. Use a distinct doc
5266 // id so any leaked frame would be visibly separate from
5267 // pre-drop traffic.
5268 fx.node
5269 .publish(
5270 "tracks",
5271 peat_mesh::sync::types::Document::with_id("ble-00000002".to_string(), {
5272 let mut f = std::collections::HashMap::new();
5273 f.insert("lat".to_string(), serde_json::json!(41.0));
5274 f.insert("lon".to_string(), serde_json::json!(-75.0));
5275 f.insert("source_node".to_string(), serde_json::json!("iroh-2"));
5276 f.insert("callsign".to_string(), serde_json::json!("B"));
5277 f.insert("hae".to_string(), serde_json::json!(0.0));
5278 f.insert("cep".to_string(), serde_json::json!(0.0));
5279 f.insert("classification".to_string(), serde_json::json!("a-f-G-U-C"));
5280 f.insert("confidence".to_string(), serde_json::json!(0.5));
5281 f.insert("category".to_string(), serde_json::json!("friendly"));
5282 f.insert(
5283 "created_at".to_string(),
5284 serde_json::json!(1_700_000_000_001_i64),
5285 );
5286 f.insert(
5287 "last_update".to_string(),
5288 serde_json::json!(1_700_000_000_001_i64),
5289 );
5290 f
5291 }),
5292 )
5293 .await
5294 .expect("publish-2");
5295
5296 tokio::time::sleep(Duration::from_millis(200)).await;
5297
5298 let post_drop_count = fx.sink.snapshot().len();
5299 assert_eq!(
5300 post_drop_count, pre_drop_count,
5301 "no frames must arrive after FanoutHandle drop"
5302 );
5303 }
5304
5305 /// Re-register after teardown succeeds — the unsubscribe path is
5306 /// exercised against a clean slate. Mirrors the
5307 /// `unsubscribeOutboundFramesJni` → `subscribeOutboundFramesJni` JNI
5308 /// flow.
5309 #[tokio::test]
5310 async fn re_register_after_unregister_succeeds() {
5311 let fx = fixture();
5312 let h = register_and_start(&fx).await.expect("register-1");
5313 drop(h);
5314 fx.transport_manager
5315 .unregister_translator("ble")
5316 .await
5317 .expect("unregister");
5318
5319 // Second register must succeed (no transport_id collision).
5320 let _h2 = register_and_start(&fx).await.expect("register-2");
5321 }
5322
5323 /// Double-register on the same `transport_id` rejects with the
5324 /// ADR-059 §"Transport ID uniqueness" invariant. The JNI
5325 /// `subscribeOutboundFramesJni` defends against this by checking
5326 /// the FanoutHandle slot before re-registering — this test guards
5327 /// the underlying invariant the JNI relies on.
5328 #[tokio::test]
5329 async fn double_register_rejects() {
5330 let fx = fixture();
5331 let _h = register_and_start(&fx).await.expect("register-1");
5332 let result = register_and_start(&fx).await;
5333 assert!(
5334 result.is_err(),
5335 "second register on same transport_id must error"
5336 );
5337 }
5338
5339 // ----- Poll-API unit tests -----
5340
5341 /// `QueueOutboundSink::send_outbound` enqueues frames that can be
5342 /// drained via the queue directly — mirrors what `poll_outbound_frames`
5343 /// does at the `PeatNode` level.
5344 #[tokio::test]
5345 async fn queue_sink_enqueues_frames() {
5346 let queue = Arc::new(std::sync::Mutex::new(std::collections::VecDeque::<
5347 OutboundFrame,
5348 >::new()));
5349 let sink = QueueOutboundSink {
5350 transport_id: "ble",
5351 queue: Arc::clone(&queue),
5352 };
5353 let ctx = TranslationContext::inbound("ble").with_collection("tracks");
5354 sink.send_outbound(vec![0xAA, 0xBB], &ctx).await.unwrap();
5355 sink.send_outbound(vec![0xCC], &ctx).await.unwrap();
5356
5357 let frames: Vec<OutboundFrame> = queue.lock().unwrap().drain(..).collect();
5358 assert_eq!(frames.len(), 2);
5359 assert_eq!(frames[0].transport_id, "ble");
5360 assert_eq!(frames[0].collection, "tracks");
5361 assert_eq!(frames[0].bytes, vec![0xAA, 0xBB]);
5362 assert_eq!(frames[1].bytes, vec![0xCC]);
5363 }
5364
5365 /// A document published via the fan-out path reaches the
5366 /// `QueueOutboundSink`, confirming the poll-API wiring matches the
5367 /// existing `RecordingSink`-based path. Mirrors
5368 /// `iroh_origin_doc_reaches_ble_sink`.
5369 #[tokio::test]
5370 async fn queue_sink_receives_fanned_out_doc() {
5371 let backend: Arc<dyn DataSyncBackend> = Arc::new(InMemoryBackend::new_initialized());
5372 let node = Arc::new(peat_mesh::Node::new(backend));
5373 let translator = Arc::new(BleTranslator::with_defaults());
5374 let tm = TransportManager::new(TransportManagerConfig::default());
5375 let queue = Arc::new(std::sync::Mutex::new(std::collections::VecDeque::<
5376 OutboundFrame,
5377 >::new()));
5378 let sink: Arc<dyn OutboundSink> = Arc::new(QueueOutboundSink {
5379 transport_id: "ble",
5380 queue: Arc::clone(&queue),
5381 });
5382 let translator_dyn: Arc<dyn Translator> = translator.clone();
5383 tm.register_translator(translator_dyn, sink, TranslatorRegistrationConfig::ble())
5384 .await
5385 .expect("register");
5386 let _h = tm
5387 .start_fanout(
5388 Arc::clone(&node),
5389 vec![translator.tracks_collection().to_string()],
5390 )
5391 .expect("start_fanout");
5392
5393 let doc = peat_mesh::sync::types::Document::with_id("q-00000001".to_string(), {
5394 let mut f = std::collections::HashMap::new();
5395 f.insert("lat".to_string(), serde_json::json!(51.5));
5396 f.insert("lon".to_string(), serde_json::json!(-0.1));
5397 f.insert("source_platform".to_string(), serde_json::json!("iroh-q01"));
5398 f.insert("hae".to_string(), serde_json::json!(10.0));
5399 f.insert("cep".to_string(), serde_json::json!(2.0));
5400 f.insert("classification".to_string(), serde_json::json!("a-f-G-U-C"));
5401 f.insert("confidence".to_string(), serde_json::json!(0.8));
5402 f.insert("category".to_string(), serde_json::json!("friendly"));
5403 f.insert("callsign".to_string(), serde_json::json!("BRAVO-1"));
5404 f.insert(
5405 "created_at".to_string(),
5406 serde_json::json!(1_700_000_001_000_i64),
5407 );
5408 f
5409 });
5410 node.publish(translator.tracks_collection(), doc)
5411 .await
5412 .expect("publish");
5413
5414 let _ = timeout(Duration::from_secs(1), async {
5415 loop {
5416 if !queue.lock().unwrap().is_empty() {
5417 return;
5418 }
5419 tokio::time::sleep(Duration::from_millis(20)).await;
5420 }
5421 })
5422 .await;
5423
5424 let frames: Vec<OutboundFrame> = queue.lock().unwrap().drain(..).collect();
5425 assert!(
5426 !frames.is_empty(),
5427 "queue sink must receive at least one frame"
5428 );
5429 assert_eq!(frames[0].transport_id, "ble");
5430 assert_eq!(frames[0].collection, translator.tracks_collection());
5431 }
5432
5433 /// `ingest_inbound_frame` round-trips: produce postcard bytes via
5434 /// `BleTranslator::encode_outbound` (the same path the real fan-out
5435 /// uses), then decode them back through `decode_inbound` and publish
5436 /// with `Some("ble")` origin (ADR-059 echo-suppression invariant).
5437 /// Tests the same primitives that `PeatNode::ingest_inbound_frame`
5438 /// uses.
5439 #[tokio::test]
5440 async fn ingest_inbound_frame_roundtrip_publishes_with_ble_origin() {
5441 let backend: Arc<dyn DataSyncBackend> = Arc::new(InMemoryBackend::new_initialized());
5442 let node = Arc::new(peat_mesh::Node::new(backend));
5443 let translator = Arc::new(BleTranslator::with_defaults());
5444
5445 // Build a minimal tracks document and encode it to postcard bytes.
5446 let outbound_doc =
5447 peat_mesh::sync::types::Document::with_id("enc-00000001".to_string(), {
5448 let mut f = std::collections::HashMap::new();
5449 f.insert("lat".to_string(), serde_json::json!(48.858));
5450 f.insert("lon".to_string(), serde_json::json!(2.294));
5451 f.insert(
5452 "source_platform".to_string(),
5453 serde_json::json!("iroh-enc01"),
5454 );
5455 f.insert("hae".to_string(), serde_json::json!(50.0));
5456 f.insert("cep".to_string(), serde_json::json!(3.0));
5457 f.insert("classification".to_string(), serde_json::json!("a-f-G-U-C"));
5458 f.insert("confidence".to_string(), serde_json::json!(0.9));
5459 f.insert("category".to_string(), serde_json::json!("friendly"));
5460 f.insert("callsign".to_string(), serde_json::json!("DELTA-1"));
5461 f.insert(
5462 "created_at".to_string(),
5463 serde_json::json!(1_700_000_002_000_i64),
5464 );
5465 f
5466 });
5467 let encode_ctx = TranslationContext::inbound("ble")
5468 .with_collection(translator.tracks_collection().to_string());
5469 let postcard_bytes = translator
5470 .encode_outbound(&outbound_doc, &encode_ctx)
5471 .await
5472 .expect("encode_outbound should produce Some bytes for a tracks doc");
5473
5474 // Decode — mirrors what `ingest_inbound_frame` does.
5475 let decode_ctx = TranslationContext::inbound("ble")
5476 .with_collection(translator.tracks_collection().to_string());
5477 let decoded = translator
5478 .decode_inbound(&postcard_bytes, &decode_ctx)
5479 .await
5480 .expect("decode_inbound")
5481 .expect("should produce a document for tracks");
5482
5483 // Publish with ble origin so echo-suppression fires correctly.
5484 let id = node
5485 .publish_with_origin(
5486 translator.tracks_collection(),
5487 decoded,
5488 Some("ble".to_string()),
5489 )
5490 .await
5491 .expect("publish");
5492
5493 // Verify the doc landed in the store.
5494 let stored = node
5495 .get(translator.tracks_collection(), &id)
5496 .await
5497 .expect("get")
5498 .expect("doc must be present after ingest");
5499 assert!(
5500 stored.fields.contains_key("lat"),
5501 "decoded document must contain lat field"
5502 );
5503 }
5504 }
5505
5506 /// Universal-Document path coexistence with the typed BLE path.
5507 /// Locks the load-bearing invariant for ADR-035 / ADR-059 Slice 1.b
5508 /// "scope #3": both translators register on the same physical wire
5509 /// under distinct transport_ids, the catch-all `LiteBridgeTranslator`
5510 /// is gated by `CollectionGatedLiteBridge` so it doesn't double-emit
5511 /// on the typed BleTranslator's collections, and origin-skip
5512 /// disambiguates each codec's emission independently.
5513 #[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
5514 mod lite_bridge_outbound_frame_tests {
5515 use super::*;
5516 use peat_mesh::sync::traits::DataSyncBackend;
5517 use peat_mesh::sync::InMemoryBackend;
5518 use peat_mesh::transport::{
5519 FanoutHandle, OutboundSink, TranslationContext, Translator,
5520 TranslatorRegistrationConfig, BLE_LITE_BRIDGE,
5521 };
5522 use peat_protocol::sync::ble_translation::BleTranslator;
5523 use peat_protocol::transport::{TransportManager, TransportManagerConfig};
5524 use std::sync::Mutex as StdMutex;
5525 use tokio::time::{timeout, Duration};
5526
5527 /// Like the typed-BLE `RecordingSink`, but stores its own
5528 /// transport_id so two parallel sinks can be told apart.
5529 struct TaggedRecordingSink {
5530 transport_id: &'static str,
5531 frames: StdMutex<Vec<(String, String, Vec<u8>)>>,
5532 }
5533
5534 #[async_trait::async_trait]
5535 impl OutboundSink for TaggedRecordingSink {
5536 async fn send_outbound(
5537 &self,
5538 bytes: Vec<u8>,
5539 ctx: &TranslationContext,
5540 ) -> anyhow::Result<()> {
5541 let collection = ctx.collection.clone().unwrap_or_default();
5542 self.frames.lock().unwrap().push((
5543 self.transport_id.to_string(),
5544 collection,
5545 bytes,
5546 ));
5547 Ok(())
5548 }
5549 }
5550
5551 impl TaggedRecordingSink {
5552 fn new(transport_id: &'static str) -> Arc<Self> {
5553 Arc::new(Self {
5554 transport_id,
5555 frames: StdMutex::new(Vec::new()),
5556 })
5557 }
5558
5559 fn snapshot(&self) -> Vec<(String, String, Vec<u8>)> {
5560 self.frames.lock().unwrap().clone()
5561 }
5562 }
5563
5564 async fn wait_for_any(sinks: &[&Arc<TaggedRecordingSink>], min_total: usize) {
5565 let _ = timeout(Duration::from_secs(1), async {
5566 loop {
5567 let total: usize = sinks.iter().map(|s| s.snapshot().len()).sum();
5568 if total >= min_total {
5569 return;
5570 }
5571 tokio::time::sleep(Duration::from_millis(20)).await;
5572 }
5573 })
5574 .await;
5575 }
5576
5577 struct CoexistenceFixture {
5578 node: Arc<peat_mesh::Node>,
5579 transport_manager: TransportManager,
5580 ble_sink: Arc<TaggedRecordingSink>,
5581 lite_sink: Arc<TaggedRecordingSink>,
5582 }
5583
5584 async fn coexistence_fixture() -> (CoexistenceFixture, FanoutHandle) {
5585 let backend: Arc<dyn DataSyncBackend> = Arc::new(InMemoryBackend::new_initialized());
5586 let node = Arc::new(peat_mesh::Node::new(backend));
5587 let mgr = TransportManager::new(TransportManagerConfig::default());
5588
5589 let ble_translator = Arc::new(BleTranslator::with_defaults());
5590 let ble_sink = TaggedRecordingSink::new("ble");
5591 let ble_translator_dyn: Arc<dyn Translator> = ble_translator.clone();
5592 let ble_sink_dyn: Arc<dyn OutboundSink> = ble_sink.clone();
5593 mgr.register_translator(
5594 ble_translator_dyn,
5595 ble_sink_dyn,
5596 TranslatorRegistrationConfig::ble(),
5597 )
5598 .await
5599 .expect("register typed BLE");
5600
5601 let lite_translator: Arc<dyn Translator> = Arc::new(
5602 CollectionGatedLiteBridge::for_ble_with_collections(LITE_BRIDGE_COLLECTIONS),
5603 );
5604 let lite_sink = TaggedRecordingSink::new(BLE_LITE_BRIDGE);
5605 let lite_sink_dyn: Arc<dyn OutboundSink> = lite_sink.clone();
5606 mgr.register_translator(
5607 lite_translator,
5608 lite_sink_dyn,
5609 TranslatorRegistrationConfig::ble(),
5610 )
5611 .await
5612 .expect("register lite-bridge");
5613
5614 // Observe both typed and universal-Document collections —
5615 // matches the production `subscribeOutboundFramesJni` shape.
5616 let mut collections = vec![
5617 ble_translator.tracks_collection().to_string(),
5618 ble_translator.nodes_collection().to_string(),
5619 ];
5620 for c in LITE_BRIDGE_COLLECTIONS {
5621 collections.push((*c).to_string());
5622 }
5623
5624 let handle = mgr
5625 .start_fanout(Arc::clone(&node), collections)
5626 .expect("start_fanout");
5627
5628 (
5629 CoexistenceFixture {
5630 node,
5631 transport_manager: mgr,
5632 ble_sink,
5633 lite_sink,
5634 },
5635 handle,
5636 )
5637 }
5638
5639 fn marker_doc(uuid: &str) -> peat_mesh::sync::types::Document {
5640 let mut fields = std::collections::HashMap::new();
5641 fields.insert("type".to_string(), serde_json::json!("a-f-G-U-C"));
5642 fields.insert("lat".to_string(), serde_json::json!(33.71));
5643 fields.insert("lon".to_string(), serde_json::json!(-84.41));
5644 peat_mesh::sync::types::Document::with_id(uuid.to_string(), fields)
5645 }
5646
5647 fn track_doc(uuid: &str) -> peat_mesh::sync::types::Document {
5648 // Minimum field set BleTranslator's track-encode requires.
5649 let mut f = std::collections::HashMap::new();
5650 f.insert("lat".to_string(), serde_json::json!(40.0));
5651 f.insert("lon".to_string(), serde_json::json!(-74.0));
5652 f.insert("source_node".to_string(), serde_json::json!("iroh-1"));
5653 f.insert("hae".to_string(), serde_json::json!(0.0));
5654 f.insert("cep".to_string(), serde_json::json!(0.0));
5655 f.insert("classification".to_string(), serde_json::json!("a-f-G-U-C"));
5656 f.insert("confidence".to_string(), serde_json::json!(0.5));
5657 f.insert("category".to_string(), serde_json::json!("friendly"));
5658 f.insert("callsign".to_string(), serde_json::json!("ALPHA-1"));
5659 f.insert(
5660 "created_at".to_string(),
5661 serde_json::json!(1_700_000_000_000_i64),
5662 );
5663 f.insert(
5664 "last_update".to_string(),
5665 serde_json::json!(1_700_000_000_000_i64),
5666 );
5667 peat_mesh::sync::types::Document::with_id(uuid.to_string(), f)
5668 }
5669
5670 /// A doc on `"markers"` (universal-Document collection) reaches
5671 /// the lite-bridge sink only — the typed BleTranslator declines
5672 /// the unknown collection silently, so the typed sink stays
5673 /// empty. The lite-bridge sink's bytes round-trip back through
5674 /// the codec to the original Document fields.
5675 #[tokio::test]
5676 async fn marker_publish_reaches_only_lite_bridge_sink() {
5677 let (fx, _h) = coexistence_fixture().await;
5678
5679 let doc = marker_doc("marker-uuid-001");
5680 let original_fields = doc.fields.clone();
5681 fx.node
5682 .publish_with_origin("markers", doc, Some("self".to_string()))
5683 .await
5684 .expect("publish marker");
5685
5686 wait_for_any(&[&fx.ble_sink, &fx.lite_sink], 1).await;
5687
5688 let ble_frames = fx.ble_sink.snapshot();
5689 let lite_frames = fx.lite_sink.snapshot();
5690
5691 assert!(
5692 ble_frames.is_empty(),
5693 "typed BLE sink MUST decline 'markers' (unknown collection); \
5694 got {} frames",
5695 ble_frames.len()
5696 );
5697 assert_eq!(
5698 lite_frames.len(),
5699 1,
5700 "lite-bridge sink should see exactly one envelope for the marker"
5701 );
5702 let (transport_id, collection, bytes) = &lite_frames[0];
5703 assert_eq!(transport_id, BLE_LITE_BRIDGE);
5704 assert_eq!(collection, "markers");
5705
5706 // Round-trip the bytes back through the codec — proves the
5707 // wire frame is well-formed and reconstructs the original
5708 // Document fields.
5709 let (envelope_collection, decoded) =
5710 peat_mesh::transport::document_codec::decode_document(bytes)
5711 .expect("decode envelope");
5712 assert_eq!(envelope_collection, "markers");
5713 assert_eq!(decoded.id.as_deref(), Some("marker-uuid-001"));
5714 assert_eq!(decoded.fields, original_fields);
5715 }
5716
5717 /// Tombstone variant of the markers-collection fanout path.
5718 /// A doc carrying `_deleted: true` on the `"markers"`
5719 /// collection must reach the lite-bridge sink with the
5720 /// sentinel preserved end-to-end. peat-mesh's fan-out skips
5721 /// `ChangeEvent::Removed` today (Slice-2 work); the soft-
5722 /// delete sentinel rides the Updated channel via this same
5723 /// path. If the codec drops the `_deleted` key in either
5724 /// direction, deletions never propagate and markers reappear
5725 /// on peers after every refresh — the failure mode that
5726 /// motivated this PR. Re-decoding the envelope bytes confirms
5727 /// the wire shape carries the flag.
5728 #[tokio::test]
5729 async fn marker_tombstone_publish_reaches_lite_bridge_sink_with_deleted_flag() {
5730 let (fx, _h) = coexistence_fixture().await;
5731
5732 let mut fields = std::collections::HashMap::new();
5733 fields.insert("_deleted".to_string(), serde_json::json!(true));
5734 fields.insert("ts".to_string(), serde_json::json!(1_700_000_000_000_i64));
5735 let doc = peat_mesh::sync::types::Document::with_id(
5736 "marker-tombstone-001".to_string(),
5737 fields.clone(),
5738 );
5739
5740 fx.node
5741 .publish_with_origin("markers", doc, Some("self".to_string()))
5742 .await
5743 .expect("publish tombstone");
5744
5745 wait_for_any(&[&fx.ble_sink, &fx.lite_sink], 1).await;
5746
5747 let ble_frames = fx.ble_sink.snapshot();
5748 let lite_frames = fx.lite_sink.snapshot();
5749 assert!(
5750 ble_frames.is_empty(),
5751 "typed BLE sink MUST decline 'markers' tombstone (unknown collection)"
5752 );
5753 assert_eq!(
5754 lite_frames.len(),
5755 1,
5756 "lite-bridge sink should see exactly one envelope for the tombstone"
5757 );
5758 let (_, collection, bytes) = &lite_frames[0];
5759 assert_eq!(collection, "markers");
5760
5761 let (envelope_collection, decoded) =
5762 peat_mesh::transport::document_codec::decode_document(bytes)
5763 .expect("decode tombstone envelope");
5764 assert_eq!(envelope_collection, "markers");
5765 assert_eq!(decoded.id.as_deref(), Some("marker-tombstone-001"));
5766 assert_eq!(
5767 decoded.fields.get("_deleted"),
5768 Some(&serde_json::json!(true)),
5769 "tombstone _deleted: true must survive the BLE wire round-trip"
5770 );
5771 }
5772
5773 /// A doc on `"tracks"` (typed BLE collection) reaches the typed
5774 /// BLE sink only — the gating wrapper declines the
5775 /// non-allow-list collection, so the lite-bridge sink stays
5776 /// empty. This is the load-bearing assertion that the gate
5777 /// prevents double emission on typed-BLE collections.
5778 #[tokio::test]
5779 async fn track_publish_reaches_only_typed_ble_sink() {
5780 let (fx, _h) = coexistence_fixture().await;
5781
5782 let doc = track_doc("ble-CAFE0001");
5783 fx.node.publish("tracks", doc).await.expect("publish track");
5784
5785 wait_for_any(&[&fx.ble_sink, &fx.lite_sink], 1).await;
5786
5787 let ble_frames = fx.ble_sink.snapshot();
5788 let lite_frames = fx.lite_sink.snapshot();
5789
5790 assert_eq!(
5791 ble_frames.len(),
5792 1,
5793 "typed BLE sink should see the track frame"
5794 );
5795 assert!(
5796 lite_frames.is_empty(),
5797 "lite-bridge sink MUST decline 'tracks' (not in \
5798 LITE_BRIDGE_COLLECTIONS allow-list); got {} frames",
5799 lite_frames.len()
5800 );
5801 }
5802
5803 /// Origin-skip is independent per codec: a marker published
5804 /// with `origin = Some(BLE_LITE_BRIDGE)` (i.e. just received
5805 /// from BLE via the universal-Document path) must NOT
5806 /// re-emit through the lite-bridge sink. The typed BLE sink is
5807 /// unaffected — it would have declined the unknown collection
5808 /// regardless.
5809 #[tokio::test]
5810 async fn ble_lite_origin_marker_does_not_re_emit_to_lite_bridge() {
5811 let (fx, _h) = coexistence_fixture().await;
5812
5813 // Skip-origin doc.
5814 let skip_doc = marker_doc("marker-skip");
5815 fx.node
5816 .publish_with_origin("markers", skip_doc, Some(BLE_LITE_BRIDGE.to_string()))
5817 .await
5818 .expect("publish skip");
5819
5820 // Barrier doc with non-skip origin — when this lands at the
5821 // lite-bridge sink we know the prior skip-origin doc was
5822 // already processed (and correctly suppressed) by the
5823 // FIFO observer.
5824 let barrier_doc = marker_doc("marker-barrier");
5825 fx.node
5826 .publish_with_origin("markers", barrier_doc, Some("self".to_string()))
5827 .await
5828 .expect("publish barrier");
5829
5830 wait_for_any(&[&fx.lite_sink], 1).await;
5831
5832 let lite_frames = fx.lite_sink.snapshot();
5833 assert_eq!(
5834 lite_frames.len(),
5835 1,
5836 "lite-bridge sink MUST receive only the barrier doc; \
5837 the BLE_LITE_BRIDGE-origin doc must be suppressed by \
5838 origin-skip (echo-loop break)"
5839 );
5840 // Confirm the captured doc is the barrier, not the
5841 // skip-origin one — defends against an inverted-skip bug.
5842 let bytes = &lite_frames[0].2;
5843 let (_collection, decoded) =
5844 peat_mesh::transport::document_codec::decode_document(bytes)
5845 .expect("decode envelope");
5846 assert_eq!(decoded.id.as_deref(), Some("marker-barrier"));
5847 }
5848
5849 /// Re-register after teardown succeeds — both translators get
5850 /// torn down + re-registered cleanly. Mirrors the
5851 /// unsubscribe → subscribe JNI flow with the lite-bridge
5852 /// branch active.
5853 #[tokio::test]
5854 async fn re_register_with_lite_bridge_after_unregister_succeeds() {
5855 let (fx, h1) = coexistence_fixture().await;
5856 drop(h1);
5857 fx.transport_manager
5858 .unregister_translator(BLE_LITE_BRIDGE)
5859 .await
5860 .expect("unregister lite-bridge");
5861 fx.transport_manager
5862 .unregister_translator("ble")
5863 .await
5864 .expect("unregister typed BLE");
5865
5866 // Second register pass on the same TransportManager must
5867 // succeed (no transport_id collision left over).
5868 let ble_translator = Arc::new(BleTranslator::with_defaults());
5869 let ble_sink = TaggedRecordingSink::new("ble");
5870 let ble_translator_dyn: Arc<dyn Translator> = ble_translator.clone();
5871 let ble_sink_dyn: Arc<dyn OutboundSink> = ble_sink.clone();
5872 fx.transport_manager
5873 .register_translator(
5874 ble_translator_dyn,
5875 ble_sink_dyn,
5876 TranslatorRegistrationConfig::ble(),
5877 )
5878 .await
5879 .expect("re-register typed BLE");
5880
5881 let lite_translator: Arc<dyn Translator> = Arc::new(
5882 CollectionGatedLiteBridge::for_ble_with_collections(LITE_BRIDGE_COLLECTIONS),
5883 );
5884 let lite_sink = TaggedRecordingSink::new(BLE_LITE_BRIDGE);
5885 let lite_sink_dyn: Arc<dyn OutboundSink> = lite_sink.clone();
5886 fx.transport_manager
5887 .register_translator(
5888 lite_translator,
5889 lite_sink_dyn,
5890 TranslatorRegistrationConfig::ble(),
5891 )
5892 .await
5893 .expect("re-register lite-bridge");
5894 }
5895 }
5896
5897 /// Wrapper-tier E2E tests for the poll API added for Dart/Flutter
5898 /// consumers.
5899 ///
5900 /// These tests exercise the full path through the `PeatNode` wrapper —
5901 /// `subscribe_poll` / `poll_changes`, `start_outbound_frames` /
5902 /// `poll_outbound_frames` / `stop_outbound_frames`, and
5903 /// `ingest_inbound_frame` — using `create_node` as the entry point, the
5904 /// same way Flutter consumers do. Each test is intentionally independent
5905 /// (separate temp dirs, separate nodes) so failures are local.
5906 #[cfg(all(feature = "sync", feature = "bluetooth"))]
5907 mod poll_api_wrapper_tests {
5908 use super::*;
5909
5910 fn test_cfg(storage_path: &str) -> NodeConfig {
5911 NodeConfig {
5912 app_id: "poll-wrapper-test".to_string(),
5913 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
5914 bind_address: Some("127.0.0.1:0".to_string()),
5915 storage_path: storage_path.to_string(),
5916 transport: None,
5917 }
5918 }
5919
5920 /// `subscribe_poll` + `poll_changes` + `cancel` through the `PeatNode`
5921 /// wrapper.
5922 ///
5923 /// Creates a real node via `create_node`, subscribes with
5924 /// `subscribe_poll`, publishes a document via the mesh document
5925 /// layer (the path that actually
5926 /// triggers `subscribe_to_changes`), and verifies the change arrives
5927 /// through `poll_changes`. Also confirms the drain is
5928 /// idempotent and that `cancel` is safe to call multiple times.
5929 #[test]
5930 fn subscribe_poll_drain_and_cancel() {
5931 let tmp = tempfile::tempdir().unwrap();
5932 let node = create_node(test_cfg(tmp.path().to_str().unwrap())).expect("create_node");
5933
5934 let handle = node.subscribe_poll().expect("subscribe_poll");
5935
5936 // Publish through the mesh document layer — this feeds subscribe_to_changes().
5937 let mesh_node = Arc::clone(&node.node);
5938 node.runtime
5939 .block_on(publish_document_into_node(
5940 &mesh_node,
5941 "test",
5942 r#"{"id":"doc-001","x":1}"#,
5943 ))
5944 .expect("publish_document_into_node");
5945
5946 // Give the spawned Tokio task time to pick up the broadcast.
5947 std::thread::sleep(std::time::Duration::from_millis(100));
5948
5949 let changes = handle.poll_changes();
5950 assert!(
5951 !changes.is_empty(),
5952 "poll_changes must return changes after publish_document_into_node"
5953 );
5954 assert!(
5955 changes.iter().any(|c| c.collection == "test"),
5956 "change must be for the 'test' collection; got: {changes:?}"
5957 );
5958
5959 // Drain is idempotent — second call returns nothing.
5960 assert!(
5961 handle.poll_changes().is_empty(),
5962 "second poll must be empty after drain"
5963 );
5964
5965 // cancel is safe to call repeatedly.
5966 handle.cancel();
5967 handle.cancel();
5968 }
5969
5970 /// `start_outbound_frames` → publish → `poll_outbound_frames` →
5971 /// `ingest_inbound_frame` → `stop_outbound_frames` → idempotent
5972 /// re-start.
5973 ///
5974 /// Covers the full wrapper path for the BLE poll API:
5975 /// - `start_outbound_frames` idempotency (second call is a no-op, not
5976 /// an error)
5977 /// - A document published to "tracks" via the mesh layer produces an
5978 /// outbound BLE frame visible through `poll_outbound_frames`
5979 /// - The polled frame can be fed into a second node via
5980 /// `ingest_inbound_frame` and the decoded document appears in that
5981 /// node's mesh store
5982 /// - `stop_outbound_frames` + `start_outbound_frames` re-registers the
5983 /// translator without a duplicate-id collision
5984 #[test]
5985 fn outbound_frames_start_poll_ingest_stop_restart() {
5986 let tmp_a = tempfile::tempdir().unwrap();
5987 let tmp_b = tempfile::tempdir().unwrap();
5988 let node_a = create_node(test_cfg(tmp_a.path().to_str().unwrap())).expect("node_a");
5989 let node_b = create_node(test_cfg(tmp_b.path().to_str().unwrap())).expect("node_b");
5990
5991 // start is idempotent — second call must succeed, not error.
5992 node_a.start_outbound_frames().expect("start 1");
5993 node_a
5994 .start_outbound_frames()
5995 .expect("start 2 (idempotent no-op)");
5996
5997 // Publish a properly-structured tracks doc so BleTranslator can encode it.
5998 let tracks_json = r#"{
5999 "id": "track-wrap-001",
6000 "lat": 51.5, "lon": -0.1,
6001 "source_platform": "test-01",
6002 "hae": 10.0, "cep": 2.0,
6003 "classification": "a-f-G-U-C",
6004 "confidence": 0.9,
6005 "category": "friendly",
6006 "callsign": "ALPHA-1",
6007 "created_at": 1700000001000
6008 }"#;
6009 let mesh_a = Arc::clone(&node_a.node);
6010 node_a
6011 .runtime
6012 .block_on(publish_document_into_node(&mesh_a, "tracks", tracks_json))
6013 .expect("publish tracks");
6014
6015 // Poll with retries to allow the async fan-out observer to fire.
6016 let mut frames = Vec::new();
6017 for _ in 0..40 {
6018 frames = node_a.poll_outbound_frames();
6019 if !frames.is_empty() {
6020 break;
6021 }
6022 std::thread::sleep(std::time::Duration::from_millis(25));
6023 }
6024 assert!(
6025 !frames.is_empty(),
6026 "outbound frames must appear after publishing to 'tracks'"
6027 );
6028 assert_eq!(frames[0].transport_id, "ble");
6029 assert_eq!(frames[0].collection, "tracks");
6030
6031 // Ingest on node_b — exercising the ingest_inbound_frame wrapper path.
6032 let doc_id = node_b
6033 .ingest_inbound_frame("tracks".to_string(), frames[0].bytes.clone())
6034 .expect("ingest_inbound_frame must not error")
6035 .expect("must return a doc_id for a valid tracks frame");
6036 assert!(!doc_id.is_empty(), "ingested doc_id must be non-empty");
6037
6038 // Document must be in node_b's mesh store.
6039 let stored = node_b
6040 .runtime
6041 .block_on(Arc::clone(&node_b.node).get("tracks", &doc_id))
6042 .expect("get must not error")
6043 .expect("ingested document must be in node_b's store");
6044 assert!(
6045 stored.fields.contains_key("lat"),
6046 "decoded track must carry lat field"
6047 );
6048
6049 // stop → re-start: translator must re-register without duplicate-id error.
6050 node_a.stop_outbound_frames();
6051 node_a
6052 .start_outbound_frames()
6053 .expect("re-start after stop must succeed");
6054 node_a.stop_outbound_frames(); // cleanup
6055 }
6056
6057 /// Receive-side counterpart for the universal-Document (`ble-lite`)
6058 /// codec — the path the production BLE pipe uses and that the typed
6059 /// `ingest_inbound_frame` test above does not exercise.
6060 ///
6061 /// Publishes to a `LITE_BRIDGE_COLLECTIONS` member the typed
6062 /// translator declines (`demo`), so it fans out solely as a `ble-lite`
6063 /// frame; captures that frame; ingests it on a second node via
6064 /// `PeatNode::ingest_inbound_lite_frame`; then asserts:
6065 /// (a) it converges into the receiver's store with the payload intact,
6066 /// (b) echo-suppression holds — the receiver does NOT re-emit it on
6067 /// `ble-lite` (origin = `Some("ble-lite")` → fan-out skips the
6068 /// originating transport). A regression here is the BLE echo storm.
6069 #[cfg(feature = "lite-bridge")]
6070 #[test]
6071 fn lite_outbound_poll_ingest_converges_without_echo() {
6072 let tmp_a = tempfile::tempdir().unwrap();
6073 let tmp_b = tempfile::tempdir().unwrap();
6074 let node_a = create_node(test_cfg(tmp_a.path().to_str().unwrap())).expect("node_a");
6075 let node_b = create_node(test_cfg(tmp_b.path().to_str().unwrap())).expect("node_b");
6076
6077 node_a.start_outbound_frames().expect("start a");
6078 node_b.start_outbound_frames().expect("start b");
6079
6080 // "demo" is on the lite-bridge allow-list AND declined by the typed
6081 // BleTranslator, so it fans out solely as a ble-lite frame.
6082 let demo_json = r#"{"id":"counter-demo-lite","inc":3,"dec":1,"by":"BRAVO"}"#;
6083 let mesh_a = Arc::clone(&node_a.node);
6084 node_a
6085 .runtime
6086 .block_on(publish_document_into_node(&mesh_a, "demo", demo_json))
6087 .expect("publish demo");
6088
6089 // Capture the ble-lite frame for the demo doc.
6090 let mut lite = None;
6091 for _ in 0..40 {
6092 if let Some(f) = node_a
6093 .poll_outbound_frames()
6094 .into_iter()
6095 .find(|f| f.transport_id == "ble-lite" && f.collection == "demo")
6096 {
6097 lite = Some(f);
6098 break;
6099 }
6100 std::thread::sleep(std::time::Duration::from_millis(25));
6101 }
6102 let lite = lite.expect("a ble-lite frame must appear for the 'demo' doc");
6103
6104 // Drain anything node_b emitted before the ingest (expected: none).
6105 let _ = node_b.poll_outbound_frames();
6106
6107 // Ingest via the lite wrapper path on node_b.
6108 let doc_id = node_b
6109 .ingest_inbound_lite_frame("demo".to_string(), lite.bytes.clone())
6110 .expect("ingest_inbound_lite_frame must not error")
6111 .expect("must return a doc_id for a valid demo lite frame");
6112 assert!(!doc_id.is_empty(), "ingested doc_id must be non-empty");
6113
6114 // (a) Converged into node_b's store with the payload intact.
6115 let stored = node_b
6116 .runtime
6117 .block_on(Arc::clone(&node_b.node).get("demo", &doc_id))
6118 .expect("get must not error")
6119 .expect("ingested demo doc must be in node_b's store");
6120 assert_eq!(
6121 stored.fields.get("inc").and_then(|v| v.as_i64()),
6122 Some(3),
6123 "decoded demo doc must carry inc=3"
6124 );
6125 assert_eq!(
6126 stored.fields.get("by").and_then(|v| v.as_str()),
6127 Some("BRAVO"),
6128 "decoded demo doc must carry the 'by' field"
6129 );
6130
6131 // (b) Echo-suppression: node_b must NOT re-emit the just-ingested
6132 // doc on ble-lite. Any such frame in this window is the echo storm.
6133 let mut echoed = false;
6134 for _ in 0..16 {
6135 if node_b
6136 .poll_outbound_frames()
6137 .iter()
6138 .any(|f| f.transport_id == "ble-lite" && f.collection == "demo")
6139 {
6140 echoed = true;
6141 break;
6142 }
6143 std::thread::sleep(std::time::Duration::from_millis(25));
6144 }
6145 assert!(
6146 !echoed,
6147 "ingested ble-lite doc must NOT be re-emitted on ble-lite \
6148 (origin-skip / echo-suppression)"
6149 );
6150
6151 node_a.stop_outbound_frames();
6152 node_b.stop_outbound_frames();
6153 }
6154
6155 /// Direct coverage for the owning-handle store/clear semantics behind
6156 /// `set_global_node_handle` / `clearGlobalNodeHandleJni` (peat#978 UAF
6157 /// fix). Exercised against a LOCAL slot so it can't race the
6158 /// process-global `GLOBAL_NODE_HANDLE` other create-path tests touch.
6159 /// Asserts: store stashes a non-zero owning pointer (+1 strong ref);
6160 /// clear zeros the slot and drops exactly that one ref (no leak, no
6161 /// double-free).
6162 #[test]
6163 fn owning_node_slot_store_then_clear_drops_exactly_one_ref() {
6164 let tmp = tempfile::tempdir().unwrap();
6165 let node = create_node(test_cfg(tmp.path().to_str().unwrap())).expect("node");
6166 let slot = std::sync::Mutex::new(0i64);
6167
6168 let before = Arc::strong_count(&node);
6169 store_owning_node_in_slot(&slot, &node);
6170 assert_ne!(
6171 *slot.lock().unwrap(),
6172 0,
6173 "store must stash a non-zero owning pointer"
6174 );
6175 assert_eq!(
6176 Arc::strong_count(&node),
6177 before + 1,
6178 "store must add exactly one owning reference"
6179 );
6180
6181 clear_owning_node_slot(&slot);
6182 assert_eq!(*slot.lock().unwrap(), 0, "clear must zero the slot");
6183 assert_eq!(
6184 Arc::strong_count(&node),
6185 before,
6186 "clear must drop exactly the one stored reference (no leak/double-free)"
6187 );
6188 }
6189 }
6190
6191 /// Wrapped-vs-flat document-shape parsing (peat#978). Docs published
6192 /// through the node layer arrive wrapped as `{id, fields:{..},
6193 /// updated_at}`; legacy `storage_backend` writes are flat.
6194 /// `parse_node/cell/command_json` must read both shapes identically —
6195 /// the contract `LITE_BRIDGE_COLLECTIONS` now depends on for
6196 /// nodes/cells/commands to round-trip over BLE. The lite-bridge E2E
6197 /// test uses the flat `demo` shape, so it exercised only the
6198 /// fallback-to-root branch; these lock in the wrapped-`fields` branch.
6199 mod doc_shape_parse_tests {
6200 use super::*;
6201
6202 fn wrap(fields_json: &str) -> String {
6203 String::from(r#"{"id":"x","fields":"#)
6204 + fields_json
6205 + r#","updated_at":{"secs_since_epoch":1730000000,"nanos_since_epoch":0}}"#
6206 }
6207
6208 #[test]
6209 fn parse_node_json_wrapped_equals_flat() {
6210 let flat = r#"{"node_type":"peat-flutter","name":"Kilo","status":"ACTIVE","readiness":1.0,"capabilities":["comms","leader"],"last_heartbeat":1730000000000}"#;
6211 let a = parse_node_json("n1", flat).expect("flat parse");
6212 let b = parse_node_json("n1", &wrap(flat)).expect("wrapped parse");
6213 assert_eq!(
6214 b.name, "Kilo",
6215 "wrapped name must come from fields, not the id"
6216 );
6217 assert_eq!(b.name, a.name);
6218 assert_eq!(b.node_type, a.node_type);
6219 assert_eq!(b.capabilities, a.capabilities);
6220 assert_eq!(
6221 b.capabilities,
6222 vec!["comms".to_string(), "leader".to_string()]
6223 );
6224 assert_eq!(b.last_heartbeat, a.last_heartbeat);
6225 assert_eq!(b.last_heartbeat, 1730000000000);
6226 }
6227
6228 #[test]
6229 fn parse_cell_json_wrapped_equals_flat() {
6230 let flat = r#"{"name":"Alpha Cell","status":"ACTIVE","node_count":2,"capabilities":["comms"],"leader_id":"n1","last_update":1730000000000}"#;
6231 let a = parse_cell_json("alpha", flat).expect("flat parse");
6232 let b = parse_cell_json("alpha", &wrap(flat)).expect("wrapped parse");
6233 assert_eq!(b.name, "Alpha Cell");
6234 assert_eq!(b.node_count, 2);
6235 assert_eq!(b.node_count, a.node_count);
6236 assert_eq!(b.leader_id, a.leader_id);
6237 assert_eq!(b.capabilities, a.capabilities);
6238 }
6239
6240 #[test]
6241 fn parse_command_json_wrapped_equals_flat() {
6242 let flat = r#"{"command_type":"WATER_REQUEST","target_id":"leader","parameters":{"quantity":5,"from":"Kilo"},"priority":1,"status":"PENDING","originator":"n1","created_at":1730000000000,"last_update":1730000000000}"#;
6243 let a = parse_command_json("req-1", flat).expect("flat parse");
6244 let b = parse_command_json("req-1", &wrap(flat)).expect("wrapped parse");
6245 assert_eq!(b.command_type, "WATER_REQUEST");
6246 assert_eq!(b.command_type, a.command_type);
6247 assert_eq!(b.originator, a.originator);
6248 assert_eq!(b.target_id, a.target_id);
6249 // parameters round-trips as the same JSON-object string in both shapes.
6250 assert_eq!(b.parameters, a.parameters);
6251 }
6252 }
6253
6254 #[cfg(feature = "sync")]
6255 mod blob_tests {
6256 use super::*;
6257
6258 /// Generate a synthetic test JPEG with a color gradient and a label.
6259 /// Synthetic "JPEG-like" payload for blob-transfer tests. Starts with
6260 /// the SOI marker (FF D8) and ends with EOI (FF D9) so the test
6261 /// assertions (`bytes[0]==0xFF`, `bytes[1]==0xD8`, `len > 100`,
6262 /// `len < 80_000`) all hold; the bytes in between are deterministic
6263 /// per (label, hue_shift) so each call produces a distinct blob
6264 /// hash. The blob-transfer path under test is byte-agnostic — using
6265 /// real JPEG encoding would pull the `image` crate's ~40 transitive
6266 /// dependencies into the workspace just for a synthetic test
6267 /// payload, which trips cargo-vet for no functional benefit.
6268 fn generate_test_image(label: &str, width: u32, height: u32, hue_shift: u8) -> Vec<u8> {
6269 let body_len = (width as usize * height as usize) / 4;
6270 let mut buf = Vec::with_capacity(body_len + label.len() + 8);
6271 buf.extend_from_slice(&[0xFF, 0xD8]); // SOI
6272 buf.extend_from_slice(label.as_bytes());
6273 buf.push(hue_shift);
6274 buf.extend(std::iter::repeat(hue_shift.wrapping_mul(3)).take(body_len));
6275 buf.extend_from_slice(&[0xFF, 0xD9]); // EOI
6276 buf
6277 }
6278
6279 fn test_node_config(storage_path: &str) -> NodeConfig {
6280 NodeConfig {
6281 app_id: "blob-test".to_string(),
6282 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6283 bind_address: Some("127.0.0.1:0".to_string()),
6284 storage_path: storage_path.to_string(),
6285 transport: None,
6286 }
6287 }
6288
6289 #[test]
6290 fn test_blob_put_get_local_roundtrip() {
6291 let tmp = tempfile::tempdir().unwrap();
6292 let node = create_node(test_node_config(tmp.path().to_str().unwrap()))
6293 .expect("create_node failed");
6294
6295 node.enable_blob_transfer(None)
6296 .expect("enable_blob_transfer failed");
6297
6298 assert!(
6299 node.blob_endpoint_id().is_some(),
6300 "blob endpoint should be initialized"
6301 );
6302
6303 let test_data = b"SKUNK-1 image chip placeholder";
6304 let hash = node
6305 .blob_put(test_data, "image/jpeg")
6306 .expect("blob_put failed");
6307 assert!(!hash.is_empty(), "hash should be non-empty");
6308
6309 assert!(
6310 node.blob_exists_locally(&hash),
6311 "blob should exist locally after put"
6312 );
6313
6314 let retrieved = node.blob_get(&hash).expect("blob_get failed");
6315 assert_eq!(retrieved, test_data, "retrieved bytes must match original");
6316 }
6317
6318 #[test]
6319 fn test_blob_get_nonexistent_returns_error() {
6320 let tmp = tempfile::tempdir().unwrap();
6321 let node = create_node(test_node_config(tmp.path().to_str().unwrap()))
6322 .expect("create_node failed");
6323
6324 node.enable_blob_transfer(None)
6325 .expect("enable_blob_transfer failed");
6326
6327 let fake_hash = "0000000000000000000000000000000000000000000000000000000000000000";
6328 assert!(
6329 !node.blob_exists_locally(fake_hash),
6330 "nonexistent hash should not be local"
6331 );
6332
6333 let result = node.blob_get(fake_hash);
6334 assert!(result.is_err(), "fetching nonexistent blob should error");
6335 }
6336
6337 #[test]
6338 fn test_blob_transfer_disabled_errors() {
6339 let tmp = tempfile::tempdir().unwrap();
6340 let node = create_node(test_node_config(tmp.path().to_str().unwrap()))
6341 .expect("create_node failed");
6342
6343 // Don't call enable_blob_transfer — methods should return errors
6344 assert!(node.blob_endpoint_id().is_none());
6345 assert!(node.blob_put(b"data", "text/plain").is_err());
6346 assert!(node.blob_get("abc").is_err());
6347 assert!(!node.blob_exists_locally("abc"));
6348 }
6349
6350 #[test]
6351 fn test_blob_cross_node_transfer() {
6352 let tmp_a = tempfile::tempdir().unwrap();
6353 let tmp_b = tempfile::tempdir().unwrap();
6354
6355 let node_a = create_node(NodeConfig {
6356 app_id: "blob-xfer-test".to_string(),
6357 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6358 bind_address: Some("127.0.0.1:0".to_string()),
6359 storage_path: tmp_a.path().to_str().unwrap().to_string(),
6360 transport: None,
6361 })
6362 .expect("create node A");
6363
6364 let node_b = create_node(NodeConfig {
6365 app_id: "blob-xfer-test".to_string(),
6366 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6367 bind_address: Some("127.0.0.1:0".to_string()),
6368 storage_path: tmp_b.path().to_str().unwrap().to_string(),
6369 transport: None,
6370 })
6371 .expect("create node B");
6372
6373 // Enable blob transfer on both with ephemeral ports
6374 node_a
6375 .enable_blob_transfer(Some("127.0.0.1:0".parse().unwrap()))
6376 .expect("enable blob A");
6377 node_b
6378 .enable_blob_transfer(Some("127.0.0.1:0".parse().unwrap()))
6379 .expect("enable blob B");
6380
6381 let a_endpoint_id = node_a.blob_endpoint_id().expect("A blob endpoint");
6382 let a_addr = node_a.blob_bound_addr().expect("A bound addr");
6383
6384 // Register A as a blob peer on B
6385 node_b
6386 .blob_add_peer(&a_endpoint_id, &a_addr)
6387 .expect("add peer");
6388
6389 // Put blob on A
6390 let test_data = b"cross-node image chip test payload 1234567890";
6391 let hash = node_a.blob_put(test_data, "image/jpeg").expect("put on A");
6392
6393 // Fetch from B — should pull from A via iroh-blobs downloader
6394 let retrieved = node_b.blob_get(&hash).expect("get from B");
6395 assert_eq!(
6396 retrieved, test_data,
6397 "cross-node blob transfer: bytes must match"
6398 );
6399 }
6400
6401 #[test]
6402 fn test_e2e_contact_report_with_image_chip() {
6403 // End-to-end: sim node publishes a contact report (TrackUpdate)
6404 // with an embedded image chip blob hash. Tablet node syncs the
6405 // document and fetches the blob by hash. Validates the full
6406 // demo chain: mesh-leader → Iroh doc sync → tablet receives
6407 // track → tablet fetches image via blob transfer.
6408
6409 let tmp_sim = tempfile::tempdir().unwrap();
6410 let tmp_tablet = tempfile::tempdir().unwrap();
6411
6412 // Create sim node (mesh-leader stand-in)
6413 let sim = create_node(NodeConfig {
6414 app_id: "e2e-contact-test".to_string(),
6415 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6416 bind_address: Some("127.0.0.1:0".to_string()),
6417 storage_path: tmp_sim.path().to_str().unwrap().to_string(),
6418 transport: None,
6419 })
6420 .expect("create sim node");
6421
6422 // Create tablet node
6423 let tablet = create_node(NodeConfig {
6424 app_id: "e2e-contact-test".to_string(),
6425 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6426 bind_address: Some("127.0.0.1:0".to_string()),
6427 storage_path: tmp_tablet.path().to_str().unwrap().to_string(),
6428 transport: None,
6429 })
6430 .expect("create tablet node");
6431
6432 // Enable blob transfer on both
6433 sim.enable_blob_transfer(Some("127.0.0.1:0".parse().unwrap()))
6434 .expect("sim blob");
6435 tablet
6436 .enable_blob_transfer(Some("127.0.0.1:0".parse().unwrap()))
6437 .expect("tablet blob");
6438
6439 // Wire blob peers
6440 let sim_blob_id = sim.blob_endpoint_id().unwrap();
6441 let sim_blob_addr = sim.blob_bound_addr().unwrap();
6442 tablet
6443 .blob_add_peer(&sim_blob_id, &sim_blob_addr)
6444 .expect("tablet add sim as blob peer");
6445
6446 // Connect doc-sync peers so the track document propagates
6447 let sim_sync_id = sim.node_id();
6448 let sim_sync_addr = format!("{:?}", sim.iroh_transport.endpoint_addr());
6449 // For doc sync, connect tablet → sim via Iroh transport
6450 let sim_peer = PeerInfo {
6451 name: "sim".to_string(),
6452 node_id: sim_sync_id.clone(),
6453 addresses: vec![],
6454 relay_url: None,
6455 };
6456 // Use the runtime to connect
6457 let sim_clone = Arc::clone(&sim);
6458 let tablet_clone = Arc::clone(&tablet);
6459 tablet.runtime.block_on(async {
6460 tablet_clone
6461 .iroh_transport
6462 .connect_peer(&peat_protocol::network::PeerInfo {
6463 name: "sim".to_string(),
6464 node_id: sim_sync_id,
6465 addresses: vec![sim_clone
6466 .iroh_transport
6467 .endpoint_addr()
6468 .addrs
6469 .iter()
6470 .next()
6471 .map(|a| format!("{}", a))
6472 .unwrap_or_default()],
6473 relay_url: None,
6474 })
6475 .await
6476 .ok();
6477 });
6478
6479 // 1. Sim creates an image chip blob
6480 let fake_jpeg = b"\xFF\xD8\xFF\xE0fake-jpeg-contact-report-image-chip-data";
6481 let image_hash = sim.blob_put(fake_jpeg, "image/jpeg").expect("sim blob put");
6482
6483 // 2. Sim publishes a contact report (TrackUpdate) to the tracks collection
6484 let track_json = serde_json::json!({
6485 "id": "red-track-1",
6486 "source_node": "sensor-node-3",
6487 "source_model": "FLIR Vue Pro R 640",
6488 "model_version": "1.0",
6489 "cell_id": "company-CHARLIE",
6490 "lat": 32.655,
6491 "lon": -117.245,
6492 "heading": 0.0,
6493 "speed": 7.7,
6494 "classification": "a-h-S",
6495 "confidence": 0.82,
6496 "category": "VESSEL",
6497 "attributes": {
6498 "callsign": "SKUNK-1",
6499 "speed_kts": "15",
6500 "vehicle_class": "fast attack craft",
6501 "reporter": "sensor-node-3",
6502 "distance_to_reporter_m": "800",
6503 "image_chip_hash": &image_hash,
6504 },
6505 "last_update": std::time::SystemTime::now()
6506 .duration_since(std::time::UNIX_EPOCH).unwrap().as_millis() as i64,
6507 });
6508
6509 // Write to the tracks collection on the sim node
6510 let sim_backend = &sim.storage_backend;
6511 let tracks_coll = sim_backend.collection("tracks");
6512 tracks_coll
6513 .upsert("red-track-1", track_json.to_string().into_bytes())
6514 .expect("sim upsert track");
6515
6516 // 3. Wait for doc sync (give Iroh a moment to propagate)
6517 std::thread::sleep(std::time::Duration::from_secs(2));
6518
6519 // 4. Tablet reads the tracks collection
6520 let tablet_tracks = tablet_clone.storage_backend.collection("tracks");
6521 let track_doc = tablet_tracks.scan().expect("tablet scan tracks");
6522
6523 // The track may or may not have synced in 2s — this is the
6524 // realistic case. If it synced, validate the full chain.
6525 // If not, the blob transfer tests above already prove the
6526 // primitive works; this test extends coverage to the doc layer.
6527 if let Some((_id, data)) = track_doc.into_iter().find(|(id, _)| id == "red-track-1") {
6528 let parsed: serde_json::Value = serde_json::from_slice(&data).expect("valid JSON");
6529 assert_eq!(parsed["source_node"], "sensor-node-3");
6530 assert_eq!(parsed["classification"], "a-h-S");
6531 assert_eq!(parsed["attributes"]["callsign"], "SKUNK-1");
6532 assert_eq!(parsed["attributes"]["image_chip_hash"], image_hash);
6533
6534 // 5. Tablet fetches the image chip blob by hash
6535 let chip_hash = parsed["attributes"]["image_chip_hash"]
6536 .as_str()
6537 .expect("hash is string");
6538 let chip_bytes = tablet.blob_get(chip_hash).expect("tablet blob get");
6539 assert_eq!(
6540 chip_bytes, fake_jpeg,
6541 "image chip bytes must match across mesh"
6542 );
6543
6544 eprintln!("E2E PASS: contact report + image chip transferred through mesh");
6545 } else {
6546 // Doc sync didn't complete in 2s — not a failure of our code,
6547 // just Iroh mesh formation timing. The blob tests above prove
6548 // the primitive. Log and pass.
6549 eprintln!(
6550 "E2E SKIP: doc sync didn't complete in 2s (blob transfer \
6551 validated separately). Re-run if you want full chain coverage."
6552 );
6553 }
6554 }
6555
6556 #[test]
6557 fn test_blob_transfer_with_synthetic_image() {
6558 let tmp_a = tempfile::tempdir().unwrap();
6559 let tmp_b = tempfile::tempdir().unwrap();
6560
6561 let node_a = create_node(NodeConfig {
6562 app_id: "img-xfer-test".to_string(),
6563 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6564 bind_address: Some("127.0.0.1:0".to_string()),
6565 storage_path: tmp_a.path().to_str().unwrap().to_string(),
6566 transport: None,
6567 })
6568 .expect("create node A");
6569
6570 let node_b = create_node(NodeConfig {
6571 app_id: "img-xfer-test".to_string(),
6572 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6573 bind_address: Some("127.0.0.1:0".to_string()),
6574 storage_path: tmp_b.path().to_str().unwrap().to_string(),
6575 transport: None,
6576 })
6577 .expect("create node B");
6578
6579 node_a
6580 .enable_blob_transfer(Some("127.0.0.1:0".parse().unwrap()))
6581 .expect("enable A");
6582 node_b
6583 .enable_blob_transfer(Some("127.0.0.1:0".parse().unwrap()))
6584 .expect("enable B");
6585
6586 let a_id = node_a.blob_endpoint_id().unwrap();
6587 let a_addr = node_a.blob_bound_addr().unwrap();
6588 node_b.blob_add_peer(&a_id, &a_addr).expect("add peer");
6589
6590 // Generate 4 keyframe images (matching the demo's progression stages)
6591 let images = vec![
6592 (
6593 "distant",
6594 generate_test_image("SKUNK-1 DISTANT", 160, 120, 40),
6595 ),
6596 (
6597 "approach",
6598 generate_test_image("SKUNK-1 APPROACH", 160, 120, 80),
6599 ),
6600 ("close", generate_test_image("SKUNK-1 CLOSE", 160, 120, 160)),
6601 ("id", generate_test_image("SKUNK-1 ID", 160, 120, 220)),
6602 ];
6603
6604 for (label, jpeg_bytes) in &images {
6605 assert!(jpeg_bytes.len() > 100, "{} should be a real JPEG", label);
6606 assert!(
6607 jpeg_bytes.len() < 80_000,
6608 "{} should be under 80KB (got {})",
6609 label,
6610 jpeg_bytes.len()
6611 );
6612 // JPEG magic bytes
6613 assert_eq!(jpeg_bytes[0], 0xFF);
6614 assert_eq!(jpeg_bytes[1], 0xD8);
6615 }
6616
6617 // Put all 4 on node A, fetch from node B
6618 let mut hashes = Vec::new();
6619 for (label, jpeg_bytes) in &images {
6620 let hash = node_a
6621 .blob_put(jpeg_bytes, "image/jpeg")
6622 .unwrap_or_else(|e| panic!("put {label}: {e}"));
6623 hashes.push((label.to_string(), hash));
6624 }
6625
6626 for (label, hash) in &hashes {
6627 let fetched = node_b
6628 .blob_get(hash)
6629 .unwrap_or_else(|e| panic!("get {label}: {e}"));
6630 let original = &images.iter().find(|(l, _)| l == label).unwrap().1;
6631 assert_eq!(
6632 fetched.len(),
6633 original.len(),
6634 "{}: fetched size must match",
6635 label
6636 );
6637 assert_eq!(
6638 fetched, *original,
6639 "{}: fetched bytes must match original",
6640 label
6641 );
6642 }
6643
6644 eprintln!(
6645 "IMAGE TRANSFER PASS: 4 synthetic JPEGs transferred cross-node ({} total bytes)",
6646 images.iter().map(|(_, b)| b.len()).sum::<usize>()
6647 );
6648 }
6649 }
6650
6651 /// Surface-tier tests for the two new public entry points added
6652 /// for peat-mesh#138 M4 (peat#879): `PeatNode::endpoint_socket_addr`
6653 /// and `PeatNode::get_document`. Both are wrapped by JNI symbols
6654 /// (`endpointSocketAddrJni`, `getDocumentJni`) that the two-
6655 /// instance instrumented test suite in peat-mesh/android-tests
6656 /// will consume in M4b. Per the surface-tier E2E rule these need
6657 /// in-crate tests independent of that downstream consumer.
6658 #[cfg(feature = "sync")]
6659 mod m4_endpoint_and_get_document_tests {
6660 use super::*;
6661
6662 fn test_node_config(storage_path: &str) -> NodeConfig {
6663 NodeConfig {
6664 app_id: "m4-test".to_string(),
6665 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6666 bind_address: Some("127.0.0.1:0".to_string()),
6667 storage_path: storage_path.to_string(),
6668 transport: None,
6669 }
6670 }
6671
6672 /// `endpoint_socket_addr` on a freshly-bound node returns a
6673 /// string that round-trips through `SocketAddr::parse` and
6674 /// carries a non-zero port. This is the contract M4b's
6675 /// instrumented test relies on when it feeds the returned
6676 /// string back into `connectPeerJni` on the other instance.
6677 #[test]
6678 fn endpoint_socket_addr_returns_parseable_loopback_addr() {
6679 let tmp = tempfile::tempdir().unwrap();
6680 let node = create_node(test_node_config(tmp.path().to_str().unwrap()))
6681 .expect("create_node failed");
6682
6683 let addr_str = node
6684 .endpoint_socket_addr()
6685 .expect("a bound node must report at least one IP address");
6686
6687 let parsed: std::net::SocketAddr = addr_str.parse().unwrap_or_else(|e| {
6688 panic!("endpoint_socket_addr returned '{addr_str}' which doesn't parse as SocketAddr: {e}")
6689 });
6690 assert!(
6691 parsed.port() > 0,
6692 "port must be nonzero for a bound socket, got {parsed}"
6693 );
6694 }
6695
6696 /// Publish a doc through the document layer, then read it
6697 /// back through the same layer. Locks in the round-trip
6698 /// contract that `publishDocumentJni` + `getDocumentJni`
6699 /// expose: both go through `peat_mesh::Node`'s document API,
6700 /// not the older raw-bytes Collection path used by typed
6701 /// helpers like `publish_node`.
6702 ///
6703 /// The in-process variant locks in the publish+get half on a
6704 /// single instance; cross-node sync is exercised by M4b on
6705 /// real devices in peat-mesh/android-tests.
6706 #[test]
6707 fn document_layer_round_trip_publish_then_get() {
6708 let tmp = tempfile::tempdir().unwrap();
6709 let node = create_node(test_node_config(tmp.path().to_str().unwrap()))
6710 .expect("create_node failed");
6711
6712 let collection = "markers";
6713 let doc_id = "M-RT-1";
6714 let body = format!(r#"{{"id":"{doc_id}","name":"alpha","severity":3}}"#);
6715
6716 let mesh_node = Arc::clone(&node.node);
6717 let returned_id = node
6718 .runtime
6719 .block_on(publish_document_into_node(&mesh_node, collection, &body))
6720 .expect("publish_document_into_node");
6721 assert_eq!(returned_id, doc_id);
6722
6723 let fetched = node
6724 .runtime
6725 .block_on(mesh_node.get(collection, &doc_id.to_string()))
6726 .expect("get must not Err")
6727 .expect("doc must be present on the publishing node");
6728
6729 // Body content must round-trip; assert on the two fields
6730 // M4b's Kotlin test pins. The published id is hoisted to
6731 // Document::id; assert separately.
6732 assert_eq!(
6733 fetched.id.as_deref(),
6734 Some(doc_id),
6735 "published id must round-trip through Document::id"
6736 );
6737 assert_eq!(
6738 fetched.fields.get("name").and_then(|v| v.as_str()),
6739 Some("alpha")
6740 );
6741 assert_eq!(
6742 fetched.fields.get("severity").and_then(|v| v.as_i64()),
6743 Some(3)
6744 );
6745 }
6746
6747 /// Surface-tier coverage for `getDocumentJni`'s JSON
6748 /// serialization path (peat#879 QA round 2). The struct-
6749 /// level round-trip test above exercises storage; this one
6750 /// exercises the extracted `serialize_document_for_get_jni`
6751 /// helper that produces the exact bytes the JNI returns —
6752 /// covering the id-reinsertion, field-iteration, and
6753 /// `to_string()` encoding the QA reviewer flagged as
6754 /// untested.
6755 #[test]
6756 fn jni_serializer_reinserts_id_alongside_fields() {
6757 // Publish through the same path the JNI consumer takes,
6758 // read back via Node::get, then run the JNI's serializer
6759 // and assert on the JSON the consumer would actually see.
6760 let tmp = tempfile::tempdir().unwrap();
6761 let node = create_node(test_node_config(tmp.path().to_str().unwrap()))
6762 .expect("create_node failed");
6763
6764 let collection = "markers";
6765 let doc_id = "M-RT-1";
6766 let body = format!(r#"{{"id":"{doc_id}","name":"alpha","severity":3}}"#);
6767
6768 let mesh_node = Arc::clone(&node.node);
6769 let _ = node
6770 .runtime
6771 .block_on(publish_document_into_node(&mesh_node, collection, &body))
6772 .expect("publish");
6773
6774 let fetched = node
6775 .runtime
6776 .block_on(mesh_node.get(collection, &doc_id.to_string()))
6777 .expect("get must not Err")
6778 .expect("doc must be present");
6779
6780 // Serialize via the exact helper getDocumentJni uses.
6781 let json = serialize_document_for_get_jni(&fetched);
6782 let parsed: serde_json::Value =
6783 serde_json::from_str(&json).expect("JNI output must parse as JSON");
6784
6785 // The Kotlin consumer expects: a plain object with id +
6786 // every other field. Pin each field shape including the
6787 // reinserted id (the QA-flagged regression surface).
6788 assert!(
6789 parsed.is_object(),
6790 "output must be a JSON object, got {parsed:?}"
6791 );
6792 assert_eq!(parsed["id"], doc_id, "id must be reinserted");
6793 assert_eq!(parsed["name"], "alpha");
6794 assert_eq!(parsed["severity"], 3);
6795 // Field count: id + name + severity — no extras.
6796 assert_eq!(
6797 parsed.as_object().unwrap().len(),
6798 3,
6799 "unexpected extra fields in JNI serialization: {parsed}"
6800 );
6801 }
6802
6803 /// Boundary: a Document with no `id` (a write path that
6804 /// didn't go through publish-with-explicit-id) serializes
6805 /// without an `"id"` key — never as `"id": null`. This
6806 /// matches the consumer contract that `id` is present iff
6807 /// the document had one assigned.
6808 #[test]
6809 fn jni_serializer_omits_id_when_none() {
6810 let doc = peat_mesh::sync::Document {
6811 id: None,
6812 fields: {
6813 let mut m = std::collections::HashMap::new();
6814 m.insert("k".to_string(), serde_json::Value::String("v".into()));
6815 m
6816 },
6817 updated_at: std::time::SystemTime::now(),
6818 };
6819
6820 let json = serialize_document_for_get_jni(&doc);
6821 let parsed: serde_json::Value = serde_json::from_str(&json).expect("parseable JSON");
6822
6823 assert!(
6824 parsed.get("id").is_none(),
6825 "expected id absent (not null) when Document::id is None, got {json}"
6826 );
6827 assert_eq!(parsed["k"], "v");
6828 }
6829
6830 /// `peat_mesh::Node::get` on a never-published key returns
6831 /// `Ok(None)`. The `getDocumentJni` wrapper maps this to a
6832 /// null jstring — test-readable as "not yet converged"
6833 /// rather than "store failed". Symmetry with
6834 /// `document_layer_round_trip_publish_then_get`.
6835 #[test]
6836 fn document_layer_get_returns_none_for_missing_doc() {
6837 let tmp = tempfile::tempdir().unwrap();
6838 let node = create_node(test_node_config(tmp.path().to_str().unwrap()))
6839 .expect("create_node failed");
6840
6841 let mesh_node = Arc::clone(&node.node);
6842 let result = node
6843 .runtime
6844 .block_on(mesh_node.get("markers", &"never-published".to_string()))
6845 .expect("get must not Err");
6846 assert!(
6847 result.is_none(),
6848 "expected None for a never-published doc, got {result:?}"
6849 );
6850 }
6851 }
6852
6853 /// Round-trip tests for the `NodeInfo` JSON wire schema.
6854 ///
6855 /// Locks in the symmetry contract between `parse_node_json`
6856 /// (storage → struct) and `serialize_node_json` (struct →
6857 /// storage), and the parallel JNI inline encode/decode in
6858 /// `Java_..._publishNodeJni` / `Java_..._getNodesJni`. The
6859 /// pre-2026-05-08 schema dropped `battery_percent` and `heart_rate`
6860 /// silently across the FFI boundary: Kotlin published them, Rust
6861 /// didn't extract them, the receiver's `getNodesJni` didn't
6862 /// emit them, the Kotlin parser saw them as `null`, and operator
6863 /// cards on remote peers showed no battery/heart indicators.
6864 /// Without a Rust-side test the bug compile-cleaned and only
6865 /// surfaced via three-device on-hardware UAT. Each assertion below
6866 /// corresponds to one optional field; future schema additions
6867 /// should add a parallel assertion + bump
6868 /// `every_optional_field_round_trips_through_storage` so the
6869 /// matrix stays exhaustive.
6870 #[cfg(feature = "sync")]
6871 mod node_tests {
6872 use super::*;
6873
6874 fn fixture(battery: Option<i32>, heart: Option<i32>) -> NodeInfo {
6875 NodeInfo {
6876 id: "ANDROID-fixture".to_string(),
6877 node_type: "SOLDIER".to_string(),
6878 name: "HOBO".to_string(),
6879 status: NodeStatus::Active,
6880 lat: 33.71576,
6881 lon: -84.41152,
6882 hae: Some(305.0),
6883 readiness: 1.0,
6884 capabilities: vec!["PLI".to_string()],
6885 cell_id: Some("BRAVO".to_string()),
6886 battery_percent: battery,
6887 heart_rate: heart,
6888 last_heartbeat: 1_700_000_000_000,
6889 }
6890 }
6891
6892 /// `serialize_node_json` → `parse_node_json` is the
6893 /// path `put_node` / `get_nodes` traverse via the
6894 /// AutomergeBackend storage. Every field a `NodeInfo`
6895 /// carries today must round-trip; if a future field is added
6896 /// to the struct without being added to either codec function,
6897 /// this assertion catches it before the FFI consumer does.
6898 #[test]
6899 fn every_optional_field_round_trips_through_storage_codec() {
6900 let original = fixture(Some(85), Some(72));
6901 let json = serialize_node_json(&original).expect("serialize");
6902 let parsed = parse_node_json(&original.id, &json).expect("parse");
6903
6904 assert_eq!(parsed.id, original.id);
6905 assert_eq!(parsed.node_type, original.node_type);
6906 assert_eq!(parsed.name, original.name);
6907 assert_eq!(parsed.lat, original.lat);
6908 assert_eq!(parsed.lon, original.lon);
6909 assert_eq!(parsed.hae, original.hae);
6910 assert_eq!(parsed.readiness, original.readiness);
6911 assert_eq!(parsed.capabilities, original.capabilities);
6912 assert_eq!(parsed.cell_id, original.cell_id);
6913 assert_eq!(parsed.battery_percent, original.battery_percent);
6914 assert_eq!(parsed.heart_rate, original.heart_rate);
6915 assert_eq!(parsed.last_heartbeat, original.last_heartbeat);
6916 }
6917
6918 /// `battery_percent: None` must serialize to a JSON `null` (or
6919 /// absent) and parse back to `None` — not silently fill 0,
6920 /// which the dropdown UI would render as "battery dead" on
6921 /// nodes that simply have no battery sensor (fixed
6922 /// sensors, demo nodes).
6923 #[test]
6924 fn battery_none_round_trips_as_none() {
6925 let original = fixture(None, None);
6926 let json = serialize_node_json(&original).expect("serialize");
6927 let parsed = parse_node_json(&original.id, &json).expect("parse");
6928
6929 assert!(parsed.battery_percent.is_none());
6930 assert!(parsed.heart_rate.is_none());
6931 }
6932
6933 /// Schema is forward-compatible: a JSON written by a newer
6934 /// peer that adds a field we don't know yet must still parse,
6935 /// dropping the unknown key. Conversely, a JSON written by an
6936 /// older peer that lacks `battery_percent` / `heart_rate`
6937 /// must parse with those fields as `None` rather than failing.
6938 #[test]
6939 fn legacy_json_without_battery_or_heart_parses_with_none() {
6940 let legacy_json = serde_json::json!({
6941 "node_type": "SOLDIER",
6942 "name": "LEGACY-PEER",
6943 "status": "ACTIVE",
6944 "lat": 33.71,
6945 "lon": -84.41,
6946 "hae": null,
6947 "readiness": 1.0,
6948 "capabilities": ["PLI"],
6949 "cell_id": "BRAVO",
6950 "last_heartbeat": 1_700_000_000_000_i64,
6951 })
6952 .to_string();
6953
6954 let parsed =
6955 parse_node_json("LEGACY-PEER", &legacy_json).expect("legacy json must parse");
6956
6957 assert!(parsed.battery_percent.is_none());
6958 assert!(parsed.heart_rate.is_none());
6959 assert_eq!(parsed.cell_id.as_deref(), Some("BRAVO"));
6960 }
6961
6962 /// `put_node` → `get_nodes` is the actual storage
6963 /// path the JNI layer exposes. Bypassing the codec helpers
6964 /// and going through `node.put_node(...)` exercises the
6965 /// AutomergeBackend serialize/scan/deserialize loop end-to-end
6966 /// — which is exactly where peat#832 (BLE-bridged tracks
6967 /// losing body fields) demonstrated the codec helpers can
6968 /// look correct in isolation while still dropping data
6969 /// across the storage round-trip.
6970 #[test]
6971 fn put_node_get_nodes_preserves_battery_and_heart() {
6972 let tmp = tempfile::tempdir().unwrap();
6973 let node = create_node(NodeConfig {
6974 app_id: "node-rt-test".to_string(),
6975 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
6976 bind_address: Some("127.0.0.1:0".to_string()),
6977 storage_path: tmp.path().to_str().unwrap().to_string(),
6978 transport: None,
6979 })
6980 .expect("create_node");
6981
6982 let original = fixture(Some(85), Some(72));
6983 node.put_node(original.clone()).expect("put_node");
6984
6985 let listed = node.get_nodes().expect("get_nodes");
6986 let found = listed
6987 .iter()
6988 .find(|p| p.id == original.id)
6989 .expect("published node must appear in get_nodes");
6990
6991 assert_eq!(
6992 found.battery_percent,
6993 Some(85),
6994 "battery_percent dropped between put_node and get_nodes"
6995 );
6996 assert_eq!(
6997 found.heart_rate,
6998 Some(72),
6999 "heart_rate dropped between put_node and get_nodes"
7000 );
7001 assert_eq!(found.cell_id.as_deref(), Some("BRAVO"));
7002 }
7003
7004 /// JNI inline-parser path: the publish surface consumers
7005 /// actually hit. Builds a JSON envelope shaped exactly like
7006 /// a typical self-position broadcaster would publish, runs
7007 /// it through the same `parse_node_publish_json` helper
7008 /// `publishNodeJni` invokes, and verifies battery + heart
7009 /// land in the resulting `NodeInfo`. Locks the duplicated
7010 /// codec — pre-2026-05-08 this was inlined inside the JNI
7011 /// function and unit tests couldn't reach it, which is how
7012 /// peat#835's bug class (silent field drop on the publish
7013 /// path) shipped without a CI signal.
7014 #[test]
7015 fn publish_json_inline_parser_extracts_battery_and_heart() {
7016 let json = r#"{
7017 "id": "ANDROID-abc123",
7018 "name": "HOBO",
7019 "node_type": "SOLDIER",
7020 "lat": 33.71576,
7021 "lon": -84.41152,
7022 "hae": 305.0,
7023 "status": "ACTIVE",
7024 "capabilities": ["PLI"],
7025 "readiness": 1.0,
7026 "cell_id": "BRAVO",
7027 "battery_percent": 85,
7028 "heart_rate": 72
7029 }"#;
7030
7031 let parsed = parse_node_publish_json(json).expect("parse");
7032
7033 assert_eq!(parsed.id, "ANDROID-abc123");
7034 assert_eq!(parsed.battery_percent, Some(85));
7035 assert_eq!(parsed.heart_rate, Some(72));
7036 assert_eq!(parsed.cell_id.as_deref(), Some("BRAVO"));
7037 assert!(parsed.capabilities.contains(&"PLI".to_string()));
7038 }
7039
7040 /// Reject an empty `id` at the publish boundary — the id is
7041 /// the storage key downstream. The pre-extraction inline code
7042 /// returned 0/JNI_FALSE on this case; the test pins the
7043 /// equivalent error contract.
7044 #[test]
7045 fn publish_json_rejects_missing_id() {
7046 let json = r#"{"name":"HOBO","node_type":"SOLDIER","lat":33.7,"lon":-84.4}"#;
7047 assert!(parse_node_publish_json(json).is_err());
7048
7049 let empty_id = r#"{"id":"","name":"HOBO","lat":33.7,"lon":-84.4}"#;
7050 assert!(parse_node_publish_json(empty_id).is_err());
7051 }
7052
7053 /// Out-of-range numeric values clamp to the logical end of
7054 /// the range rather than silently dropping to `None`. The
7055 /// silent-`None`-on-overflow shape is the same bug class
7056 /// peat#835 exists to lock — a pathological 2³² battery
7057 /// becoming "no sensor" is visually identical to the
7058 /// legitimate None case, which is exactly the data-loss
7059 /// failure mode the PR exists to prevent.
7060 #[test]
7061 fn battery_and_heart_clamp_out_of_range_numbers() {
7062 // Battery above 100 clamps to 100.
7063 let high = serde_json::json!(9999);
7064 assert_eq!(parse_battery_percent(&high), Some(100));
7065
7066 // Negative battery clamps to 0.
7067 let neg = serde_json::json!(-50);
7068 assert_eq!(parse_battery_percent(&neg), Some(0));
7069
7070 // i64::MAX clamps to 100 — the silent-None-on-overflow
7071 // case the pre-clamp `as_i64().and_then(i32::try_from)`
7072 // chain produced None for. After clamp, fail-safe.
7073 let huge = serde_json::json!(i64::MAX);
7074 assert_eq!(parse_battery_percent(&huge), Some(100));
7075
7076 // Heart rate above 250 clamps to 250 (max plausible BPM).
7077 let bpm_high = serde_json::json!(500);
7078 assert_eq!(parse_heart_rate(&bpm_high), Some(250));
7079
7080 // Heart rate below 0 clamps to 0; legitimate low BPM
7081 // (bradycardia, asystole) passes through unchanged. The
7082 // 30-floor was lowered in round-3 — see
7083 // `heart_rate_preserves_bradycardia_below_30`.
7084 let bpm_neg = serde_json::json!(-50);
7085 assert_eq!(parse_heart_rate(&bpm_neg), Some(0));
7086 let bpm_low_real = serde_json::json!(10);
7087 assert_eq!(parse_heart_rate(&bpm_low_real), Some(10));
7088 }
7089
7090 /// Non-numeric values (publisher serialization bug, hostile
7091 /// peer, schema drift) parse as `None` rather than coercing.
7092 /// We accept "no sensor" but reject silent type coercion —
7093 /// `"85"` as a JSON string is a publisher bug, not a value
7094 /// to interpret.
7095 #[test]
7096 fn battery_and_heart_reject_non_numeric() {
7097 let s = serde_json::json!("85");
7098 assert!(parse_battery_percent(&s).is_none());
7099 assert!(parse_heart_rate(&s).is_none());
7100
7101 let null = serde_json::Value::Null;
7102 assert!(parse_battery_percent(&null).is_none());
7103 assert!(parse_heart_rate(&null).is_none());
7104
7105 let arr = serde_json::json!([85]);
7106 assert!(parse_battery_percent(&arr).is_none());
7107 }
7108
7109 /// Forward-compat: a peer running a future schema that adds
7110 /// fields we don't know about must still parse cleanly,
7111 /// silently dropping the unknowns. Locks the existing
7112 /// `unwrap_or` / `optional`-style behavior so a future
7113 /// stricter parser doesn't regress this on accident.
7114 #[test]
7115 fn parse_silently_drops_unknown_future_fields() {
7116 let json = r#"{
7117 "node_type": "SOLDIER",
7118 "name": "FUTURE-PEER",
7119 "status": "ACTIVE",
7120 "lat": 33.71,
7121 "lon": -84.41,
7122 "readiness": 1.0,
7123 "capabilities": ["PLI"],
7124 "cell_id": "BRAVO",
7125 "battery_percent": 90,
7126 "last_heartbeat": 1700000000000,
7127
7128 "future_v2_field_one": "should be ignored",
7129 "future_v2_struct": { "nested": 42 },
7130 "future_v2_array": [1, 2, 3]
7131 }"#;
7132
7133 let parsed =
7134 parse_node_json("FUTURE-PEER", json).expect("future-shaped json must parse");
7135 assert_eq!(parsed.battery_percent, Some(90));
7136 assert_eq!(parsed.cell_id.as_deref(), Some("BRAVO"));
7137 // No assertion about the unknown fields — they're
7138 // intentionally dropped on the floor. The test exists to
7139 // keep us honest if anyone tries to switch to a stricter
7140 // `serde_json::from_str::<TypedStruct>` shape.
7141 }
7142
7143 /// **Round-3 / peat#835 review item P2-1**: float-typed
7144 /// numeric wire payloads must not silently drop. The
7145 /// pre-round-3 implementation used `as_i64()?` which returns
7146 /// `None` for any JSON Number stored as float — a Kotlin
7147 /// publisher serializing `battery_percent` as `Double`
7148 /// (`85.0`), or any node whose JSON serializer renders
7149 /// integers with a trailing `.0`, would silently lose the
7150 /// field. That's the same data-loss bug class peat#835 was
7151 /// opened to lock in the first place.
7152 #[test]
7153 fn battery_accepts_float_form() {
7154 assert_eq!(parse_battery_percent(&serde_json::json!(85.0)), Some(85));
7155 // Fractional rounds to nearest.
7156 assert_eq!(parse_battery_percent(&serde_json::json!(85.7)), Some(86));
7157 assert_eq!(parse_battery_percent(&serde_json::json!(85.4)), Some(85));
7158 // Float still clamps.
7159 assert_eq!(parse_battery_percent(&serde_json::json!(150.0)), Some(100));
7160 assert_eq!(parse_battery_percent(&serde_json::json!(-10.5)), Some(0));
7161 }
7162
7163 #[test]
7164 fn heart_rate_accepts_float_form() {
7165 assert_eq!(parse_heart_rate(&serde_json::json!(72.0)), Some(72));
7166 assert_eq!(parse_heart_rate(&serde_json::json!(72.6)), Some(73));
7167 assert_eq!(parse_heart_rate(&serde_json::json!(300.0)), Some(250));
7168 }
7169
7170 /// Bradycardia: athletic resting HR can dip into the 20s,
7171 /// asystole reads as 0. Round-3 lowered the floor from 30 to
7172 /// 0 so the UI gets the truth and can decide what to flag.
7173 /// The pre-round-3 floor of 30 silently rounded these up,
7174 /// hiding the very signal a heart-rate indicator should
7175 /// surface.
7176 #[test]
7177 fn heart_rate_preserves_bradycardia_below_30() {
7178 assert_eq!(parse_heart_rate(&serde_json::json!(25)), Some(25));
7179 assert_eq!(parse_heart_rate(&serde_json::json!(0)), Some(0));
7180 // Negative still clamps to 0 — sensor noise / signed-int
7181 // serialization bug.
7182 assert_eq!(parse_heart_rate(&serde_json::json!(-5)), Some(0));
7183 }
7184
7185 /// **Round-3**: extracted emit-side codec
7186 /// `serialize_nodes_get_json` mirrors the parse-side
7187 /// extraction (`parse_node_publish_json`). Without the
7188 /// extraction, the inline `getNodesJni` json! macro was a
7189 /// duplicated codec the test suite couldn't reach — same
7190 /// drift class peat#835 originally exposed on the parse side.
7191 /// This test pins the emit shape end-to-end.
7192 #[test]
7193 fn serialize_nodes_get_json_round_trips_through_parser() {
7194 let original = NodeInfo {
7195 id: "ANDROID-emit".to_string(),
7196 node_type: "SOLDIER".to_string(),
7197 name: "EMIT-TEST".to_string(),
7198 status: NodeStatus::Active,
7199 lat: 33.71576,
7200 lon: -84.41152,
7201 hae: Some(305.0),
7202 readiness: 1.0,
7203 capabilities: vec!["PLI".to_string()],
7204 cell_id: Some("BRAVO".to_string()),
7205 battery_percent: Some(85),
7206 heart_rate: Some(72),
7207 last_heartbeat: 1_700_000_000_000,
7208 };
7209
7210 let emitted = serialize_nodes_get_json(std::slice::from_ref(&original));
7211 let arr: Vec<serde_json::Value> = serde_json::from_str(&emitted).expect("array");
7212 assert_eq!(arr.len(), 1);
7213
7214 // Parse the emitted JSON back through the storage parser
7215 // (the path `getNodes` consumers' downstream Kotlin
7216 // parsers mirror) and assert symmetry.
7217 let obj_str = serde_json::to_string(&arr[0]).expect("obj");
7218 let parsed = parse_node_json(&original.id, &obj_str).expect("parse");
7219 assert_eq!(parsed.battery_percent, Some(85));
7220 assert_eq!(parsed.heart_rate, Some(72));
7221 assert_eq!(parsed.cell_id.as_deref(), Some("BRAVO"));
7222 assert_eq!(parsed.last_heartbeat, 1_700_000_000_000);
7223 }
7224
7225 /// **Round-3 P3-1**: when a publisher provides a
7226 /// `last_heartbeat` on the wire, the publish-path parser
7227 /// honors it instead of stamping `Utc::now()`. Resolves the
7228 /// doc-comment-vs-behavior tension: the field doc-comment
7229 /// describes a "0 means stale" convention that the publish
7230 /// path was actively preventing from ever shipping.
7231 #[test]
7232 fn publish_json_honors_wire_last_heartbeat() {
7233 let supplied: i64 = 1_700_000_123_456;
7234 let json = format!(
7235 r#"{{
7236 "id": "ANDROID-replay",
7237 "name": "REPLAY",
7238 "node_type": "SOLDIER",
7239 "lat": 0.0, "lon": 0.0,
7240 "status": "ACTIVE",
7241 "last_heartbeat": {}
7242 }}"#,
7243 supplied
7244 );
7245 let parsed = parse_node_publish_json(&json).expect("parse");
7246 assert_eq!(parsed.last_heartbeat, supplied);
7247 }
7248
7249 /// And: when the wire omits `last_heartbeat`, fall back to
7250 /// `now()` (preserving back-compat with publishers that don't
7251 /// stamp the field).
7252 #[test]
7253 fn publish_json_stamps_now_when_last_heartbeat_absent() {
7254 let before = chrono::Utc::now().timestamp_millis();
7255 let json = r#"{
7256 "id": "ANDROID-no-stamp",
7257 "name": "FRESH",
7258 "node_type": "SOLDIER",
7259 "lat": 0.0, "lon": 0.0,
7260 "status": "ACTIVE"
7261 }"#;
7262 let parsed = parse_node_publish_json(json).expect("parse");
7263 let after = chrono::Utc::now().timestamp_millis();
7264 assert!(
7265 parsed.last_heartbeat >= before && parsed.last_heartbeat <= after,
7266 "last_heartbeat ({}) should be in [{}, {}]",
7267 parsed.last_heartbeat,
7268 before,
7269 after
7270 );
7271 }
7272
7273 /// **Round-4 P1**: wire `last_heartbeat: 0` is the documented
7274 /// stale-record sentinel per the `NodeInfo` field doc;
7275 /// must round-trip unchanged. Round-3's `> 0` filter
7276 /// inverted this contract, silently replacing the
7277 /// stale-marker with `Utc::now()`. Test pins the corrected
7278 /// behavior so the regression can't recur.
7279 #[test]
7280 fn publish_json_preserves_wire_last_heartbeat_zero_as_stale_marker() {
7281 let json = r#"{
7282 "id": "ANDROID-stale",
7283 "name": "STALE",
7284 "node_type": "SOLDIER",
7285 "lat": 0.0, "lon": 0.0,
7286 "status": "ACTIVE",
7287 "last_heartbeat": 0
7288 }"#;
7289 let parsed = parse_node_publish_json(json).expect("parse");
7290 assert_eq!(
7291 parsed.last_heartbeat, 0,
7292 "wire `last_heartbeat: 0` must pass through as the stale-record sentinel"
7293 );
7294 }
7295
7296 /// **Round-4 P1 / P2**: smallest non-zero positive timestamp
7297 /// (`1`) and a small value (`12345`) both pass through as-is.
7298 /// These are the boundary values around the prior `> 0`
7299 /// filter; round-4 dropped the filter, so all positive values
7300 /// short of the future-skew clamp must round-trip.
7301 #[test]
7302 fn publish_json_preserves_small_positive_last_heartbeat() {
7303 for wire in [1_i64, 12_345, 1_700_000_000_000] {
7304 let json = format!(
7305 r#"{{"id":"ANDROID-{w}","name":"X","node_type":"SOLDIER","lat":0.0,"lon":0.0,"status":"ACTIVE","last_heartbeat":{w}}}"#,
7306 w = wire,
7307 );
7308 let parsed = parse_node_publish_json(&json).expect("parse");
7309 assert_eq!(
7310 parsed.last_heartbeat, wire,
7311 "wire `{}` must round-trip",
7312 wire
7313 );
7314 }
7315 }
7316
7317 /// **Round-4 P2 #4**: clock-skew injection guard. A peer with
7318 /// a far-future-skewed clock can publish `i64::MAX` (or any
7319 /// timestamp beyond `now() + 60s` grace); the parser caps to
7320 /// `now()` so downstream staleness UI can't be gamed into
7321 /// "always fresh." Negative values pass through (very stale,
7322 /// but not absurd).
7323 #[test]
7324 fn publish_json_clamps_far_future_last_heartbeat_to_now() {
7325 let json = r#"{
7326 "id": "ANDROID-malicious",
7327 "name": "MALICIOUS",
7328 "node_type": "SOLDIER",
7329 "lat": 0.0, "lon": 0.0,
7330 "status": "ACTIVE",
7331 "last_heartbeat": 9223372036854775807
7332 }"#;
7333 let before = chrono::Utc::now().timestamp_millis();
7334 let parsed = parse_node_publish_json(json).expect("parse");
7335 let after = chrono::Utc::now().timestamp_millis();
7336 assert!(
7337 parsed.last_heartbeat >= before && parsed.last_heartbeat <= after,
7338 "i64::MAX must clamp to now(), got {}",
7339 parsed.last_heartbeat
7340 );
7341 }
7342
7343 /// **Round-5**: negative `last_heartbeat` collapses to the
7344 /// stale-marker (`0`) rather than passing through. Round-4
7345 /// let negatives through with a doc-comment claim that
7346 /// downstream Long arithmetic produced a "sensible large
7347 /// positive age" — that was wrong: `now - i64::MIN`
7348 /// overflows, and the Kotlin `Long` subtraction silently
7349 /// wraps. Pin the corrected behavior so a malicious peer
7350 /// publishing `last_heartbeat: i64::MIN` can't game the
7351 /// staleness UI in the opposite direction from the
7352 /// `i64::MAX` case.
7353 #[test]
7354 fn publish_json_clamps_negative_last_heartbeat_to_zero() {
7355 for wire in [-1_i64, -1_700_000_000_000, i64::MIN] {
7356 let json = format!(
7357 r#"{{"id":"ANDROID-neg-{w}","name":"NEG","node_type":"SOLDIER","lat":0.0,"lon":0.0,"status":"ACTIVE","last_heartbeat":{w}}}"#,
7358 w = wire,
7359 );
7360 let parsed = parse_node_publish_json(&json)
7361 .unwrap_or_else(|e| panic!("wire {} must parse: {:?}", wire, e));
7362 assert_eq!(
7363 parsed.last_heartbeat, 0,
7364 "negative wire `{}` must collapse to stale-marker `0`",
7365 wire
7366 );
7367 }
7368 }
7369
7370 /// Wire timestamp within the 60-second future-grace window
7371 /// passes through (legitimate clock drift between mobile
7372 /// devices on unrelated networks). Beyond grace, clamp.
7373 #[test]
7374 fn publish_json_within_grace_window_passes_through_then_clamps_beyond() {
7375 let now = chrono::Utc::now().timestamp_millis();
7376 // 30 s in the future — within grace.
7377 let in_grace = now + 30_000;
7378 let json = format!(
7379 r#"{{"id":"ANDROID-grace","name":"G","node_type":"SOLDIER","lat":0.0,"lon":0.0,"status":"ACTIVE","last_heartbeat":{}}}"#,
7380 in_grace
7381 );
7382 let parsed = parse_node_publish_json(&json).expect("parse");
7383 assert_eq!(parsed.last_heartbeat, in_grace);
7384
7385 // 5 minutes in the future — beyond 60 s grace, clamp.
7386 let beyond = chrono::Utc::now().timestamp_millis() + 5 * 60 * 1000;
7387 let json2 = format!(
7388 r#"{{"id":"ANDROID-skew","name":"S","node_type":"SOLDIER","lat":0.0,"lon":0.0,"status":"ACTIVE","last_heartbeat":{}}}"#,
7389 beyond
7390 );
7391 let parsed2 = parse_node_publish_json(&json2).expect("parse");
7392 assert!(
7393 parsed2.last_heartbeat < beyond,
7394 "5min-future must clamp ({} should be << {})",
7395 parsed2.last_heartbeat,
7396 beyond
7397 );
7398 }
7399
7400 /// **Round-4 P3 #7**: float rounding mode is half-away-from-zero
7401 /// per `f64::round()`. Pin the contract so a future refactor to
7402 /// `round_ties_even` (banker's) doesn't silently change the
7403 /// emitted i32 by ±1 for half-values.
7404 #[test]
7405 fn battery_percent_rounds_halves_away_from_zero() {
7406 assert_eq!(parse_battery_percent(&serde_json::json!(85.5)), Some(86));
7407 assert_eq!(parse_battery_percent(&serde_json::json!(84.5)), Some(85));
7408 // 0.5 rounds to 1, not 0 (half-away-from-zero, not
7409 // banker's-rounding).
7410 assert_eq!(parse_battery_percent(&serde_json::json!(0.5)), Some(1));
7411 }
7412
7413 /// **Round-4 P3 #9**: forward-compat for the publish parser.
7414 /// Mirror of `parse_silently_drops_unknown_future_fields`
7415 /// for the storage parser; both share the
7416 /// `serde_json::Value`-indexing pattern but the contract
7417 /// should be locked separately so a future refactor of
7418 /// either to a typed `serde::Deserialize` doesn't regress
7419 /// half the surface unnoticed.
7420 #[test]
7421 fn publish_json_silently_drops_unknown_future_fields() {
7422 let json = r#"{
7423 "id": "ANDROID-future",
7424 "name": "FUTURE",
7425 "node_type": "SOLDIER",
7426 "lat": 33.71, "lon": -84.41,
7427 "status": "ACTIVE",
7428 "battery_percent": 90,
7429
7430 "future_v2_field_one": "should be ignored",
7431 "future_v2_struct": { "nested": 42 },
7432 "future_v2_array": [1, 2, 3]
7433 }"#;
7434 let parsed = parse_node_publish_json(json).expect("future-shaped publish must parse");
7435 assert_eq!(parsed.battery_percent, Some(90));
7436 assert_eq!(parsed.id, "ANDROID-future");
7437 }
7438 }
7439
7440 /// End-to-end round-trip tests for the track storage path that
7441 /// `Java_..._ingestPositionJni` and `Java_..._getTracksJni` expose
7442 /// to consumer plugins.
7443 ///
7444 /// peat#832 (open as of 2026-05-08) reports the BLE-bridged tracks
7445 /// surface every body field at `parse_track_json`'s `unwrap_or`
7446 /// default (lat/lon=0.0, classification="a-u-G", confidence=0.5,
7447 /// source_node="unknown") even though `ingest_position_via_translator`
7448 /// publishes valid coordinates. The hypothesis the issue records:
7449 /// the writer publishes via `peat_mesh::Node::publish_with_origin`
7450 /// (Document API → Automerge map storage), but the reader uses
7451 /// `AutomergeBackend::collection().scan()` which returns bytes the
7452 /// reader assumes are flat JSON. The two APIs disagree on the
7453 /// on-disk shape, so body fields don't survive the round-trip.
7454 ///
7455 /// Existing `ingest_position_tests` (line ~2520) wires
7456 /// `peat_mesh::Node` against an `InMemoryBackend` from peat-mesh —
7457 /// that backend doesn't carry the AutomergeBackend / Collection
7458 /// scan asymmetry, so it has no way to reproduce the bug. The
7459 /// tests below use `create_node()` (the same factory the JNI
7460 /// surface uses) so the AutomergeBackend disagreement is in scope.
7461 ///
7462 /// `ingest_position_via_translator_then_get_tracks_preserves_body`
7463 /// is the regression gate: pre-fix it failed deterministically,
7464 /// post-fix it locks the symmetry. The dev-team-owns-validation
7465 /// memory captures the broader pattern.
7466 #[cfg(all(feature = "sync", feature = "bluetooth"))]
7467 mod track_tests {
7468 use super::*;
7469 use peat_protocol::sync::ble_translation::{
7470 value_to_mesh_document, BlePosition, BleTranslator,
7471 };
7472
7473 /// Test fixture that holds both the constructed node and the
7474 /// tempdir backing its storage. Bind both via `let _node_fx =
7475 /// ingest_position_test_node();` and let the drop order do the
7476 /// right thing — `Drop for PeatNode` (and its inner
7477 /// `AutomergeStore`) runs first, then the tempdir's
7478 /// `Drop for TempDir` removes the on-disk directory.
7479 ///
7480 /// Earlier this fixture used `std::mem::forget(tmp)` on the
7481 /// `TempDir` with a comment claiming "Tempdirs are nuked at
7482 /// process exit anyway" — that's wrong: `tempfile::TempDir`
7483 /// cleanup runs in its `Drop` impl, which `mem::forget` skips,
7484 /// and process exit doesn't trigger OS-level `/tmp` cleanup.
7485 /// Re-running `cargo test track_tests` locally accumulated
7486 /// `/tmp/.tmpXXXXXX` directories until reboot.
7487 struct TrackFixture {
7488 node: Arc<PeatNode>,
7489 // Field is read via the binding lifetime (Drop runs after
7490 // `node`), not by the test body. `dead_code` would lint
7491 // otherwise — `_tmp` makes the role explicit.
7492 #[allow(dead_code)]
7493 _tmp: tempfile::TempDir,
7494 }
7495
7496 fn ingest_position_test_node() -> TrackFixture {
7497 let tmp = tempfile::tempdir().expect("tempdir");
7498 let path = tmp.path().to_str().expect("tempdir path utf-8").to_string();
7499
7500 let node = create_node(NodeConfig {
7501 app_id: "track-rt-test".to_string(),
7502 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
7503 bind_address: Some("127.0.0.1:0".to_string()),
7504 storage_path: path,
7505 transport: None,
7506 })
7507 .expect("create_node");
7508
7509 TrackFixture { node, _tmp: tmp }
7510 }
7511
7512 /// Sanity check the **flat-JSON** path: `put_track` →
7513 /// `serialize_track_json` → `coll.upsert(json_bytes)` → `coll.scan()`
7514 /// → `parse_track_json` → `get_tracks`. Both writer and reader
7515 /// use the same flat-JSON shape, so this should round-trip
7516 /// today. If this ever fails, the asymmetry has spread to
7517 /// even the typed-API path.
7518 #[test]
7519 fn put_track_get_tracks_preserves_body() {
7520 let fx = ingest_position_test_node();
7521 let pn = &fx.node;
7522
7523 let original = TrackInfo {
7524 id: "manual-001".to_string(),
7525 source_node: "ANDROID-tablet".to_string(),
7526 cell_id: Some("BRAVO".to_string()),
7527 formation_id: None,
7528 lat: 33.71576,
7529 lon: -84.41152,
7530 hae: Some(305.0),
7531 cep: Some(5.0),
7532 heading: Some(87.5),
7533 speed: Some(1.2),
7534 classification: "a-f-G-U-C-I".to_string(),
7535 confidence: 0.9,
7536 category: TrackCategory::Person,
7537 created_at: 1_700_000_000_000,
7538 last_update: 1_700_000_000_000,
7539 attributes: std::collections::HashMap::new(),
7540 };
7541
7542 pn.put_track(original.clone()).expect("put_track");
7543 let listed = pn.get_tracks().expect("get_tracks");
7544 let found = listed
7545 .iter()
7546 .find(|t| t.id == "manual-001")
7547 .expect("track must appear");
7548
7549 assert!(
7550 (found.lat - original.lat).abs() < 1e-9,
7551 "lat dropped via put_track/get_tracks: got {}",
7552 found.lat
7553 );
7554 assert!(
7555 (found.lon - original.lon).abs() < 1e-9,
7556 "lon dropped via put_track/get_tracks: got {}",
7557 found.lon
7558 );
7559 assert_eq!(found.cell_id.as_deref(), Some("BRAVO"));
7560 assert_eq!(found.source_node, original.source_node);
7561 assert_eq!(found.classification, original.classification);
7562 }
7563
7564 /// peat#832 regression gate: the **BLE-bridged path** that
7565 /// `ingestPositionJni` exercises on every BLE peer's position
7566 /// advert. Writer goes through `Node::publish_with_origin`
7567 /// (Document API); the original reader went through
7568 /// `AutomergeBackend::collection().scan()` (flat-JSON API),
7569 /// and the two storage-API namespaces disagreed — every body
7570 /// field came back as a `parse_track_json` `unwrap_or`
7571 /// default (lat/lon=0.0, source_node="unknown",
7572 /// classification="a-u-G"). Fix routes `get_tracks` through
7573 /// `Node::query` so writer and reader share the Document API,
7574 /// and `put_track` was migrated to `Node::publish` to keep
7575 /// the typed-API path consistent. If either path breaks, this
7576 /// test catches it before on-device UAT does.
7577 #[test]
7578 fn ingest_position_via_translator_then_get_tracks_preserves_body() {
7579 let fx = ingest_position_test_node();
7580 let pn = &fx.node;
7581 let translator = BleTranslator::with_defaults();
7582
7583 const PERIPHERAL: u32 = 0xCAFE_0001;
7584 let position = BlePosition {
7585 latitude: 33.71576,
7586 longitude: -84.41152,
7587 altitude: Some(305.0),
7588 accuracy: Some(5.0),
7589 };
7590 let value = translator.position_to_track_in_cell(
7591 &position,
7592 PERIPHERAL,
7593 Some("SCOUT-CAFE"),
7594 Some("BRAVO"),
7595 );
7596 let doc = value_to_mesh_document(value);
7597
7598 pn.runtime.block_on(async {
7599 pn.node
7600 .publish_with_origin(
7601 translator.tracks_collection(),
7602 doc,
7603 Some("ble".to_string()),
7604 )
7605 .await
7606 .expect("publish_with_origin");
7607 });
7608
7609 let tracks = pn.get_tracks().expect("get_tracks");
7610 let found = tracks
7611 .iter()
7612 .find(|t| t.id.contains("CAFE0001"))
7613 .expect("BLE-bridged track must appear in get_tracks output");
7614
7615 assert!(
7616 (found.lat - 33.71576).abs() < 1e-4,
7617 "peat#832: lat dropped — got {} (expected ~33.71576)",
7618 found.lat
7619 );
7620 assert!(
7621 (found.lon - (-84.41152)).abs() < 1e-4,
7622 "peat#832: lon dropped — got {} (expected ~-84.41152)",
7623 found.lon
7624 );
7625 assert_eq!(
7626 found.cell_id.as_deref(),
7627 Some("BRAVO"),
7628 "peat#832: cell_id dropped"
7629 );
7630 assert!(
7631 !found.source_node.is_empty() && found.source_node != "unknown",
7632 "peat#832: source_node reverted to default — got {:?}",
7633 found.source_node
7634 );
7635 assert_ne!(
7636 found.classification, "a-u-G",
7637 "peat#832: classification reverted to default a-u-G"
7638 );
7639 }
7640
7641 /// Single-id read path: `get_track(id)` migrated to
7642 /// `Node::get` along with `get_tracks` (PR #836). Without
7643 /// this test the per-id path was silent in the regression
7644 /// suite — same bug class could re-emerge on it without a
7645 /// signal.
7646 #[test]
7647 fn ingest_position_then_get_track_single_id_preserves_body() {
7648 let fx = ingest_position_test_node();
7649 let pn = &fx.node;
7650 let translator = BleTranslator::with_defaults();
7651
7652 const PERIPHERAL: u32 = 0xCAFE_0002;
7653 let position = BlePosition {
7654 latitude: 33.71576,
7655 longitude: -84.41152,
7656 altitude: Some(305.0),
7657 accuracy: Some(5.0),
7658 };
7659 let value = translator.position_to_track_in_cell(
7660 &position,
7661 PERIPHERAL,
7662 Some("SCOUT-ID-2"),
7663 Some("BRAVO"),
7664 );
7665 let track_id = value
7666 .get("id")
7667 .and_then(|v| v.as_str())
7668 .expect("translator stamps id")
7669 .to_string();
7670 let doc = value_to_mesh_document(value);
7671
7672 pn.runtime.block_on(async {
7673 pn.node
7674 .publish_with_origin(
7675 translator.tracks_collection(),
7676 doc,
7677 Some("ble".to_string()),
7678 )
7679 .await
7680 .expect("publish_with_origin");
7681 });
7682
7683 let single = pn
7684 .get_track(&track_id)
7685 .expect("get_track")
7686 .expect("track must exist for known id");
7687
7688 assert!((single.lat - 33.71576).abs() < 1e-4);
7689 assert!((single.lon - (-84.41152)).abs() < 1e-4);
7690 assert_eq!(single.cell_id.as_deref(), Some("BRAVO"));
7691 assert_eq!(single.id, track_id);
7692 }
7693
7694 /// Pre-fix-shape entries (written via `coll.upsert(json_bytes)`
7695 /// before this PR) won't decode through `Node::query`'s
7696 /// `serde_json::from_slice::<Document>` reader and are silently
7697 /// dropped. Codifies the migration story: devices upgrading to
7698 /// a new `libpeat_ffi.so` will *not* see pre-fix tracks until
7699 /// the BLE peer republishes (every ~5 s in normal operation),
7700 /// but they also won't crash on the stale bytes.
7701 ///
7702 /// Test writes a fake old-shape entry directly through the
7703 /// untyped Collection surface, then calls `get_tracks` and
7704 /// asserts (a) it doesn't error, (b) the legacy entry is
7705 /// invisible. `put_track` itself can't be used here because
7706 /// PR #836 migrated it to `Node::publish` (correctly), so
7707 /// reaching the old shape requires going through
7708 /// `storage_backend.collection().upsert(...)` directly.
7709 #[test]
7710 fn pre_fix_flat_json_entries_are_silently_dropped_not_crashed() {
7711 let fx = ingest_position_test_node();
7712 let pn = &fx.node;
7713
7714 // Old-shape: flat JSON of the body, written via the
7715 // untyped Collection upsert (the pre-#836 `put_track`
7716 // codepath). Bytes are intentionally well-formed JSON so
7717 // any *parse* error that fires would be in the Document
7718 // deserialization step, not in JSON tokenization.
7719 let legacy = serde_json::json!({
7720 "source_node": "ble-DEAD0001",
7721 "lat": 33.0,
7722 "lon": -84.0,
7723 "classification": "a-f-G-U-C-I",
7724 "confidence": 0.9,
7725 "category": "PERSON",
7726 "created_at": 1_700_000_000_000_i64,
7727 "last_update": 1_700_000_000_000_i64,
7728 })
7729 .to_string()
7730 .into_bytes();
7731
7732 // `pn.storage_backend` is `Arc<AutomergeBackend>` from
7733 // `peat_protocol::storage`; its `StorageBackend::collection`
7734 // returns the untyped `Arc<dyn Collection>` whose
7735 // `upsert(doc_id, Vec<u8>)` is the pre-#836 write path the
7736 // bug originally lived in.
7737 let coll = pn.storage_backend.collection(collections::TRACKS);
7738 coll.upsert("legacy-track-DEAD0001", legacy)
7739 .expect("legacy upsert must succeed");
7740
7741 // get_tracks must not error.
7742 let listed = pn.get_tracks().expect("get_tracks must not panic");
7743
7744 // The legacy entry must NOT appear via the Node::query
7745 // path — its bytes don't decode as a Document, so it's
7746 // silently dropped per the documented migration semantics.
7747 assert!(
7748 listed.iter().all(|t| t.id != "legacy-track-DEAD0001"),
7749 "pre-fix legacy entry must be silently invisible after migration: {:?}",
7750 listed.iter().map(|t| &t.id).collect::<Vec<_>>()
7751 );
7752 }
7753 }
7754
7755 /// Marker tombstone schema. peat-mesh's fan-out skips
7756 /// `ChangeEvent::Removed` today (Slice-2 work), so deletion of
7757 /// a synced marker is communicated via a `_deleted: true`
7758 /// sentinel ridden on the Updated channel. Consumers publish a
7759 /// tombstone on deletion and filter `_deleted: true` entries out
7760 /// of "current markers" views on render. These tests pin the
7761 /// wire shape so a future schema change has to pass through the
7762 /// test gate first.
7763 mod marker_tombstone {
7764 use super::*;
7765
7766 /// A minimum-viable tombstone publish carries `uid` +
7767 /// `_deleted: true` only — the publisher omits type/lat/lon
7768 /// to keep the BLE frame small. The parser must accept this
7769 /// shape (placeholders for the absent geo fields), set
7770 /// `deleted = true`, and round-trip cleanly.
7771 #[test]
7772 fn parse_minimal_tombstone() {
7773 let json = r#"{"uid":"abc-123","_deleted":true,"ts":1700000000000}"#;
7774 let m = parse_marker_publish_json("", json).expect("minimal tombstone parses");
7775 assert!(m.deleted, "deleted flag set");
7776 assert_eq!(m.uid, "abc-123");
7777 assert_eq!(m.ts, 1700000000000);
7778 }
7779
7780 /// A live (non-tombstone) marker still requires type/lat/lon.
7781 /// Drops `_deleted` from the body — the parser must default
7782 /// `deleted = false` and enforce the required-fields contract
7783 /// it enforced before the tombstone shape was added.
7784 #[test]
7785 fn parse_live_marker_requires_geo() {
7786 let no_type = r#"{"uid":"x","lat":1.0,"lon":2.0}"#;
7787 assert!(parse_marker_publish_json("", no_type).is_err());
7788
7789 let no_lat = r#"{"uid":"x","type":"a-f-G","lon":2.0}"#;
7790 assert!(parse_marker_publish_json("", no_lat).is_err());
7791
7792 let no_lon = r#"{"uid":"x","type":"a-f-G","lat":1.0}"#;
7793 assert!(parse_marker_publish_json("", no_lon).is_err());
7794
7795 let ok = r#"{"uid":"x","type":"a-f-G","lat":1.0,"lon":2.0}"#;
7796 let m = parse_marker_publish_json("", ok).expect("live marker parses");
7797 assert!(!m.deleted);
7798 }
7799
7800 /// `serialize_marker_json` round-trips a tombstone. The
7801 /// `_deleted: true` key MUST appear in the output (otherwise
7802 /// peers receiving the doc see a normal-looking marker and
7803 /// re-render it after a refresh tick — the deletion would
7804 /// "un-do" itself).
7805 #[test]
7806 fn serialize_tombstone_includes_deleted_key() {
7807 let m = MarkerInfo {
7808 uid: "abc-123".to_string(),
7809 marker_type: "a-u-G".to_string(),
7810 lat: 0.0,
7811 lon: 0.0,
7812 hae: None,
7813 ts: 1700000000000,
7814 callsign: None,
7815 color: None,
7816 cell_id: None,
7817 deleted: true,
7818 };
7819 let json = serialize_marker_json(&m).expect("serializes");
7820 assert!(
7821 json.contains("\"_deleted\":true"),
7822 "tombstone serialization must include _deleted key, got: {json}"
7823 );
7824 }
7825
7826 /// A live marker's serialization MUST NOT include `_deleted`
7827 /// (saves bytes on the wire AND avoids ambiguity for
7828 /// receivers running an older parser that does a strict
7829 /// `_deleted == true` check).
7830 #[test]
7831 fn serialize_live_marker_omits_deleted_key() {
7832 let m = MarkerInfo {
7833 uid: "abc-123".to_string(),
7834 marker_type: "a-f-G-U-C".to_string(),
7835 lat: 33.71,
7836 lon: -84.41,
7837 hae: Some(312.4),
7838 ts: 1700000000000,
7839 callsign: Some("ALPHA-1".to_string()),
7840 color: Some(-65536),
7841 cell_id: None,
7842 deleted: false,
7843 };
7844 let json = serialize_marker_json(&m).expect("serializes");
7845 assert!(
7846 !json.contains("_deleted"),
7847 "live marker must not emit _deleted key, got: {json}"
7848 );
7849 }
7850
7851 /// `serialize_markers_get_json` (the get_markers / scan-side
7852 /// shape, an array) preserves the tombstone flag when the
7853 /// doc store contains both live and deleted entries. The
7854 /// plugin's `renderAllMarkersFromDocStore` reads this output
7855 /// and must be able to identify which entries are tombstones.
7856 #[test]
7857 fn scan_serializes_tombstones_in_array() {
7858 let live = MarkerInfo {
7859 uid: "live".to_string(),
7860 marker_type: "a-f-G".to_string(),
7861 lat: 1.0,
7862 lon: 2.0,
7863 hae: None,
7864 ts: 1,
7865 callsign: None,
7866 color: None,
7867 cell_id: None,
7868 deleted: false,
7869 };
7870 let dead = MarkerInfo {
7871 deleted: true,
7872 ..live.clone()
7873 };
7874 let mut dead = dead;
7875 dead.uid = "dead".to_string();
7876
7877 let json = serialize_markers_get_json(&[live, dead]);
7878 let arr: serde_json::Value = serde_json::from_str(&json).unwrap();
7879 let arr = arr.as_array().unwrap();
7880 assert_eq!(arr.len(), 2);
7881 // Find by uid; can't rely on order.
7882 let live_obj = arr.iter().find(|v| v["uid"] == "live").unwrap();
7883 let dead_obj = arr.iter().find(|v| v["uid"] == "dead").unwrap();
7884 assert!(
7885 live_obj.get("_deleted").is_none(),
7886 "live entry has no _deleted"
7887 );
7888 assert_eq!(
7889 dead_obj["_deleted"].as_bool(),
7890 Some(true),
7891 "dead entry has _deleted: true"
7892 );
7893 }
7894
7895 /// Round-trip: serialize → parse → serialize. The two
7896 /// serialized strings must be byte-identical. Catches
7897 /// codec drift (e.g., one side adds a field the other
7898 /// drops, or `Option<i64> 0` vs absent disagreements).
7899 #[test]
7900 fn tombstone_round_trip_is_stable() {
7901 let m = MarkerInfo {
7902 uid: "round-trip-uid".to_string(),
7903 marker_type: "a-u-G".to_string(),
7904 lat: 0.0,
7905 lon: 0.0,
7906 hae: None,
7907 ts: 1700000000000,
7908 callsign: None,
7909 color: None,
7910 cell_id: None,
7911 deleted: true,
7912 };
7913 let s1 = serialize_marker_json(&m).unwrap();
7914 let parsed = parse_marker_publish_json("", &s1).expect("parses tombstone");
7915 assert!(parsed.deleted, "deleted flag preserved through round-trip");
7916 assert_eq!(parsed.uid, m.uid);
7917 let s2 = serialize_marker_json(&parsed).unwrap();
7918 assert_eq!(s1, s2, "round-trip must produce byte-identical output");
7919 }
7920 }
7921
7922 /// Surface-tier round-trips for the marker API the plugin
7923 /// actually consumes: the UniFFI `PeatNode::put_marker` /
7924 /// `PeatNode::get_markers` path (typed-record wrapper, doc-store
7925 /// persistence, `MARKERS` collection wiring) and the JNI
7926 /// `publishMarkerJni` / `getMarkersJni` path (inline parser +
7927 /// `serialize_markers_get_json`). These tests are the bidirectional
7928 /// E2E coverage the QA review on PR #845 required — internal
7929 /// codec tests in [`marker_tombstone`] don't catch wrapper-vs-
7930 /// internal drift (renamed UniFFI field, doc-store key mismatch,
7931 /// JNI handle lifecycle regression). Storage-side tests follow
7932 /// the `put_node_get_nodes_preserves_battery_and_heart`
7933 /// pattern in [`node_tests`]: `create_node` against
7934 /// `AutomergeBackend` (not `InMemoryBackend`, which silently
7935 /// papers over the publish-vs-scan storage-API asymmetry — see
7936 /// the InMemoryBackend test gap memory).
7937 #[cfg(feature = "sync")]
7938 mod marker_tests {
7939 use super::*;
7940
7941 fn live_marker(uid: &str) -> MarkerInfo {
7942 MarkerInfo {
7943 uid: uid.to_string(),
7944 marker_type: "a-f-G-U-C".to_string(),
7945 lat: 33.71576,
7946 lon: -84.41152,
7947 hae: Some(312.4),
7948 ts: 1_700_000_000_000,
7949 callsign: Some("ALPHA-1".to_string()),
7950 color: Some(-65536),
7951 cell_id: Some("BRAVO".to_string()),
7952 deleted: false,
7953 }
7954 }
7955
7956 fn tombstone_marker(uid: &str) -> MarkerInfo {
7957 MarkerInfo {
7958 uid: uid.to_string(),
7959 marker_type: TOMBSTONE_PLACEHOLDER_TYPE.to_string(),
7960 lat: 0.0,
7961 lon: 0.0,
7962 hae: None,
7963 ts: 1_700_000_000_000,
7964 callsign: None,
7965 color: None,
7966 cell_id: None,
7967 deleted: true,
7968 }
7969 }
7970
7971 fn make_node(label: &str) -> Arc<PeatNode> {
7972 let tmp = tempfile::tempdir().expect("tempdir");
7973 create_node(NodeConfig {
7974 app_id: format!("marker-rt-{label}"),
7975 shared_key: "dGVzdC1rZXktMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0".to_string(),
7976 bind_address: Some("127.0.0.1:0".to_string()),
7977 storage_path: tmp.path().to_str().unwrap().to_string(),
7978 transport: None,
7979 })
7980 .expect("create_node")
7981 }
7982
7983 // ----- UniFFI tier -------------------------------------------------
7984
7985 /// Live marker survives the full UniFFI surface round-trip.
7986 /// Drift point this catches: a future field added to
7987 /// `MarkerInfo` but dropped in `serialize_marker_json` or
7988 /// `parse_marker_publish_json` (the very bug pattern
7989 /// peat#835 / peat#832 sat behind). Every optional field
7990 /// must round-trip; new fields require a parallel assertion
7991 /// below so this matrix stays exhaustive.
7992 #[test]
7993 fn put_marker_get_markers_preserves_live_fields() {
7994 let node = make_node("live");
7995 let original = live_marker("marker-live-001");
7996 node.put_marker(original.clone()).expect("put_marker");
7997
7998 let listed = node.get_markers().expect("get_markers");
7999 let found = listed
8000 .iter()
8001 .find(|m| m.uid == original.uid)
8002 .expect("published marker must appear in get_markers");
8003
8004 assert_eq!(found.marker_type, original.marker_type);
8005 assert_eq!(found.lat, original.lat);
8006 assert_eq!(found.lon, original.lon);
8007 assert_eq!(found.hae, original.hae);
8008 assert_eq!(found.ts, original.ts);
8009 assert_eq!(found.callsign, original.callsign);
8010 assert_eq!(found.color, original.color);
8011 assert_eq!(found.cell_id, original.cell_id);
8012 assert!(!found.deleted, "live marker must not arrive deleted");
8013 }
8014
8015 /// Tombstone survives the UniFFI surface round-trip with the
8016 /// `deleted` flag preserved. Without this assertion a future
8017 /// schema refactor could silently drop `_deleted: true` on
8018 /// store-and-scan — receivers would render the marker as
8019 /// live, the deletion would never propagate, and the only
8020 /// signal would be on-device UAT (the exact bug class the
8021 /// dev-team-owns-validation rule exists to lock in CI).
8022 #[test]
8023 fn put_marker_get_markers_preserves_tombstone() {
8024 let node = make_node("tomb");
8025 let original = tombstone_marker("marker-tomb-001");
8026 node.put_marker(original.clone()).expect("put_marker");
8027
8028 let listed = node.get_markers().expect("get_markers");
8029 let found = listed
8030 .iter()
8031 .find(|m| m.uid == original.uid)
8032 .expect("published tombstone must appear in get_markers");
8033
8034 assert!(found.deleted, "tombstone must round-trip with deleted=true");
8035 assert_eq!(found.uid, original.uid);
8036 assert_eq!(found.ts, original.ts);
8037 }
8038
8039 /// Tombstone overwriting a live marker for the same UID:
8040 /// `put_marker` is upsert, the second write replaces the
8041 /// first. `get_markers` returns the tombstone (deleted=true),
8042 /// not the prior live shape. Locks the CRDT semantics the
8043 /// consumer's deletion flow depends on — without upsert,
8044 /// "delete a marker I just placed" would produce two
8045 /// doc-store entries and ambiguous resolution.
8046 #[test]
8047 fn tombstone_upserts_over_live_marker() {
8048 let node = make_node("upsert");
8049 let uid = "marker-upsert-001";
8050 node.put_marker(live_marker(uid)).expect("put live");
8051 node.put_marker(tombstone_marker(uid)).expect("put tomb");
8052
8053 let listed = node.get_markers().expect("get_markers");
8054 let matching: Vec<_> = listed.iter().filter(|m| m.uid == uid).collect();
8055 assert_eq!(
8056 matching.len(),
8057 1,
8058 "upsert must produce exactly one entry per uid, got {}",
8059 matching.len()
8060 );
8061 assert!(matching[0].deleted, "tombstone must win over prior live");
8062 }
8063
8064 // ----- JNI tier ----------------------------------------------------
8065
8066 /// JNI inline-parser path: `publishMarkerJni` decodes a
8067 /// JString into the same `parse_marker_publish_json` helper
8068 /// the typed UniFFI path skips. Builds a JSON envelope shaped
8069 /// exactly like the consumer's marker serializer produces on
8070 /// the wire and verifies every field lands in the resulting
8071 /// `MarkerInfo`. Locks the duplicated codec — same pattern as
8072 /// `publish_json_inline_parser_extracts_battery_and_heart` in
8073 /// [`node_tests`], same rationale (silent field drop on
8074 /// the publish path).
8075 #[test]
8076 fn publish_json_inline_parser_extracts_live_marker_fields() {
8077 let json = r#"{
8078 "uid": "marker-jni-001",
8079 "type": "a-f-G-U-C",
8080 "lat": 33.71576,
8081 "lon": -84.41152,
8082 "hae": 312.4,
8083 "ts": 1700000000000,
8084 "callsign": "ALPHA-1",
8085 "color": -65536,
8086 "cell_id": "BRAVO"
8087 }"#;
8088
8089 let parsed = parse_marker_publish_json("", json).expect("parse");
8090
8091 assert_eq!(parsed.uid, "marker-jni-001");
8092 assert_eq!(parsed.marker_type, "a-f-G-U-C");
8093 assert_eq!(parsed.lat, 33.71576);
8094 assert_eq!(parsed.lon, -84.41152);
8095 assert_eq!(parsed.hae, Some(312.4));
8096 assert_eq!(parsed.callsign.as_deref(), Some("ALPHA-1"));
8097 assert_eq!(parsed.color, Some(-65536));
8098 assert_eq!(parsed.cell_id.as_deref(), Some("BRAVO"));
8099 assert!(!parsed.deleted);
8100 }
8101
8102 /// JNI tombstone inline-parser path: `publishMarkerJni` must
8103 /// accept the stripped tombstone body the consumer's deletion
8104 /// serializer produces (uid + `_deleted: true` + ts, no
8105 /// geo/type/callsign). Catches a regression where the parser
8106 /// tightens up its required-fields validation in a way that
8107 /// breaks the deletion path silently.
8108 #[test]
8109 fn publish_json_inline_parser_accepts_stripped_tombstone() {
8110 let json = r#"{"uid":"marker-jni-tomb-001","_deleted":true,"ts":1700000000000}"#;
8111 let parsed = parse_marker_publish_json("", json).expect("parse stripped tombstone");
8112 assert!(parsed.deleted);
8113 assert_eq!(parsed.uid, "marker-jni-tomb-001");
8114 assert_eq!(parsed.ts, 1_700_000_000_000);
8115 assert_eq!(
8116 parsed.marker_type, TOMBSTONE_PLACEHOLDER_TYPE,
8117 "absent type must resolve to the named placeholder, not a magic literal"
8118 );
8119 }
8120
8121 // ----- JNI + UniFFI: storage round-trip via the get-side serializer
8122 // (the shape getMarkersJni hands to consumers) -------------
8123
8124 /// `getMarkersJni` serializes `Vec<MarkerInfo>` via
8125 /// `serialize_markers_get_json` — the JSON shape consumers
8126 /// parse. A round-trip test pins that the wire shape
8127 /// `get_markers` emits is one a subsequent
8128 /// `parse_marker_publish_json` accepts, ensuring no
8129 /// asymmetric-codec regression slips through.
8130 #[test]
8131 fn get_markers_jni_serialized_shape_re_parses_cleanly() {
8132 let node = make_node("getjni");
8133 node.put_marker(live_marker("marker-getjni-001"))
8134 .expect("put live");
8135 node.put_marker(tombstone_marker("marker-getjni-002"))
8136 .expect("put tomb");
8137
8138 let listed = node.get_markers().expect("get_markers");
8139 let json = serialize_markers_get_json(&listed);
8140
8141 // Decode every entry through the same inline parser the
8142 // publish path uses. If the get-side shape ever diverges
8143 // from the publish-side shape, this fails before it
8144 // reaches a consumer.
8145 let arr: serde_json::Value = serde_json::from_str(&json).expect("valid JSON");
8146 for obj in arr.as_array().expect("array").iter() {
8147 let body = serde_json::to_string(obj).unwrap();
8148 let parsed = parse_marker_publish_json("", &body).expect("get-side body re-parses");
8149 if parsed.uid == "marker-getjni-002" {
8150 assert!(parsed.deleted, "tombstone preserved in scan output");
8151 } else {
8152 assert!(!parsed.deleted, "live preserved in scan output");
8153 }
8154 }
8155 }
8156 }
8157}
8158
8159// =============================================================================
8160// JNI Bindings - Direct Android native method support
8161// =============================================================================
8162//
8163// These functions provide a direct JNI interface that bypasses JNA's symbol
8164// lookup issues on Android. When System.loadLibrary() is called, these
8165// functions are registered via JNI's standard naming convention.
8166//
8167// Usage in Kotlin:
8168// ```kotlin
8169// class PeatJni {
8170// companion object {
8171// init {
8172// System.loadLibrary("peat_ffi")
8173// }
8174// }
8175// external fun peatVersion(): String
8176// external fun testJni(): String
8177// }
8178// ```
8179
8180/// JNI: Get Peat library version
8181///
8182/// Kotlin signature: external fun peatVersion(): String
8183#[no_mangle]
8184pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_peatVersion(
8185 mut env: JNIEnv,
8186 _class: JClass,
8187) -> jstring {
8188 let version = peat_version();
8189 env.new_string(&version)
8190 .expect("Failed to create Java string")
8191 .into_raw()
8192}
8193
8194/// Pinned `GlobalRef` to the Android Context jobject that
8195/// `setAndroidContextJni` last received. The raw pointer we hand to
8196/// `ndk_context::initialize_android_context` is the jobject handle
8197/// inside this GlobalRef; the JVM guarantees the handle remains
8198/// valid (and pointing at the same Java object even if the GC moves
8199/// the underlying heap object) until the GlobalRef is dropped.
8200///
8201/// Storing the GlobalRef in a `Mutex<Option<GlobalRef>>` (rather
8202/// than a `OnceLock`) supports the documented call pattern: the
8203/// surface admits multiple `setAndroidContextJni` invocations, but
8204/// **only before `createNodeJni` starts iroh** (see that fn's
8205/// docstring). The mutex serializes concurrent
8206/// `setAndroidContextJni` callers; it does NOT block readers of
8207/// `ndk_context::android_context()`. Between the
8208/// `release_android_context()` and `initialize_android_context()`
8209/// calls inside `setAndroidContextJni` there is a brief window where
8210/// the global cell is empty — any iroh worker thread that hits
8211/// `android_context()` during that window panics. The pre-iroh-start
8212/// constraint makes the window structurally unreachable in
8213/// practice (no iroh worker exists yet) but a re-init after
8214/// `createNodeJni` is unsafe.
8215#[cfg(target_os = "android")]
8216static ANDROID_CONTEXT_GLOBAL_REF: std::sync::Mutex<Option<jni::objects::GlobalRef>> =
8217 std::sync::Mutex::new(None);
8218
8219/// Set to `true` by `createNodeJni` (and `createNodeWithConfigJni`)
8220/// on first successful node construction; checked by
8221/// `setAndroidContextJni` to reject post-iroh-start invocations.
8222///
8223/// Why this exists: `setAndroidContextJni` must release and
8224/// reinitialize `ndk-context`'s global cell, and there is a brief
8225/// window between the two calls where any iroh worker thread
8226/// reaching `ndk_context::android_context()` panics. The
8227/// `Application.onCreate`-before-`createNodeJni` call pattern keeps
8228/// the window structurally unreachable (no iroh worker exists yet),
8229/// but the Kotlin/Rust doc could be ignored by a consumer that
8230/// re-acquires the Application Context in `onActivityResult` or
8231/// similar. This flag turns that misuse into a logged-and-ignored
8232/// no-op rather than a SIGABRT.
8233///
8234/// One-way: once set, never cleared. Re-init is unsafe by design;
8235/// there is no recovery path. Set via `Release` to publish all
8236/// prior writes (iroh handle install, tokio runtime startup) to any
8237/// `Acquire` reader; checked via `Acquire` to see them. peat#924 QA
8238/// WARNING-2 round 2.
8239#[cfg(target_os = "android")]
8240static IROH_STARTED: std::sync::atomic::AtomicBool = std::sync::atomic::AtomicBool::new(false);
8241
8242/// JNI: Plumb the Android `Context` jobject into `ndk-context`'s
8243/// global cell.
8244///
8245/// Kotlin signature: `external fun setAndroidContextJni(context: Any)`
8246///
8247/// Why this exists: `JNI_OnLoad` initializes `ndk-context` with the
8248/// `JavaVM*` it receives as an argument, but passes `null` for the
8249/// Android `Context` because no `Context` exists yet — `JNI_OnLoad`
8250/// runs before any `Application` has been instantiated by the
8251/// framework. That's enough for the iroh discovery subtree
8252/// (swarm-discovery / mDNS) which only needs the JVM for thread
8253/// attachment. It is NOT enough for code that needs the
8254/// `Context` itself — `hickory-resolver`'s Android `ConnectivityManager`
8255/// probe (transitively reachable via iroh-dns), NDK asset-manager
8256/// access, app-private file path resolution, etc. Those paths panic
8257/// with `android context was not initialized` on first call.
8258///
8259/// Consumers using iroh DNS-based discovery (relay, pkarr,
8260/// non-mDNS peer lookups) MUST call this from
8261/// `Application.onCreate()` passing the application Context BEFORE
8262/// the first `createNodeJni`. Consumers using only mDNS local-link
8263/// discovery (peat-ffi's own surface tests, the QUICKSTART
8264/// scenarios 1–3) can skip it.
8265///
8266/// Multiple calls are allowed, but **only before `createNodeJni`**
8267/// is invoked. Calling this after iroh has started creates a brief
8268/// window between `release_android_context()` and
8269/// `initialize_android_context()` where any concurrent
8270/// `ndk_context::android_context()` reader — iroh-dns
8271/// `hickory-resolver`'s ConnectivityManager probe, the mDNS
8272/// multicast worker, etc. — sees the cell empty and panics with
8273/// "android context was not initialized". The mutex protecting
8274/// `ANDROID_CONTEXT_GLOBAL_REF` serializes concurrent
8275/// `setAndroidContextJni` writers but does NOT block readers
8276/// reaching into `ndk-context`'s own global cell. The
8277/// Application.onCreate-before-createNodeJni call pattern makes
8278/// the window structurally unreachable (no iroh worker exists
8279/// yet); a re-init after iroh starts is unsafe.
8280///
8281/// The JVM pointer remains the same one JNI_OnLoad stored on every
8282/// call; only the Context changes. peat#925 QA WARNING follow-up.
8283#[cfg(target_os = "android")]
8284#[no_mangle]
8285pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_setAndroidContextJni(
8286 env: JNIEnv,
8287 _class: JClass,
8288 context: jni::objects::JObject,
8289) {
8290 // Reject post-iroh-start invocations. The release+reinit pair
8291 // below has a brief window where `ndk_context::android_context()`
8292 // returns the empty cell — once any iroh worker is alive (i.e.
8293 // `createNodeJni` has returned successfully), one of them
8294 // resolving the cell during that window panics. The documented
8295 // call pattern (Application.onCreate before any createNodeJni)
8296 // makes the window unreachable; this `Acquire` load is the
8297 // runtime guardrail for misuse that ignores the doc. peat#924 QA
8298 // WARNING-2 round 2.
8299 use std::sync::atomic::Ordering;
8300 if IROH_STARTED.load(Ordering::Acquire) {
8301 android_log(
8302 "setAndroidContextJni: ignoring — iroh already started; \
8303 call this from Application.onCreate BEFORE createNodeJni. \
8304 See PeatJni.kt KDoc.",
8305 );
8306 return;
8307 }
8308
8309 // JNI delivers `context` as a **local reference** — valid only
8310 // for the duration of this native method call. After we return,
8311 // the JVM is free to recycle the local-ref table slot, and a
8312 // raw pointer to it would alias the wrong (or no) object on the
8313 // next `ndk_context::android_context().context()` lookup.
8314 // Promote to a process-lifetime global reference first, then
8315 // hand `ndk_context` the jobject handle from inside the
8316 // GlobalRef. peat#925 QA WARNING-2.
8317 let global_ref = match env.new_global_ref(&context) {
8318 Ok(gref) => gref,
8319 Err(e) => {
8320 android_log(&format!(
8321 "setAndroidContextJni: env.new_global_ref(context) failed: {}",
8322 e
8323 ));
8324 return;
8325 }
8326 };
8327 let vm_ptr = match env.get_java_vm() {
8328 Ok(vm) => vm.get_java_vm_pointer() as *mut c_void,
8329 Err(_) => {
8330 android_log("setAndroidContextJni: env.get_java_vm() failed");
8331 return;
8332 }
8333 };
8334
8335 // SAFETY: `JNI_OnLoad` cached the JavaVM and called
8336 // `ndk_context::initialize_android_context(vm, null)` exactly
8337 // once at library-load time. `ndk-context 0.1.1` is one-shot —
8338 // calling `initialize_android_context` a second time asserts on
8339 // `previous.is_none()` and SIGABRT's the process (peat#925 QA
8340 // d2d01b23 surface-test surfaced this). The documented re-init
8341 // pattern is `release_android_context()` followed by a fresh
8342 // `initialize_android_context(...)`. We do exactly that here,
8343 // holding the `ANDROID_CONTEXT_GLOBAL_REF` mutex across the pair
8344 // so concurrent `setAndroidContextJni` callers serialize and
8345 // neither sees the cell in a released-but-not-yet-reinitialized
8346 // state. The JavaVM pointer remains the same one JNI_OnLoad
8347 // stored; only the Context changes (from `null` to the
8348 // GlobalRef'd jobject handle on first call; from the previous
8349 // GlobalRef to the new one on subsequent calls).
8350 //
8351 // The jobject handle is pulled from `global_ref.as_raw()` — the
8352 // JVM guarantees this remains valid until the GlobalRef is
8353 // dropped, which we prevent by stashing the GlobalRef in
8354 // `ANDROID_CONTEXT_GLOBAL_REF` below before releasing the lock.
8355 let ctx_ptr = global_ref.as_raw() as *mut c_void;
8356 let mut slot = ANDROID_CONTEXT_GLOBAL_REF.lock().unwrap();
8357 unsafe {
8358 // `release_android_context()` asserts `previous.is_some()`
8359 // — safe because JNI_OnLoad installed the `(vm, null)` entry
8360 // exactly once and this critical section is the only place
8361 // in peat-ffi that releases. If we ever surface a
8362 // `clear_android_context_jni`, it would also need the same
8363 // mutex.
8364 ndk_context::release_android_context();
8365 ndk_context::initialize_android_context(vm_ptr, ctx_ptr);
8366 }
8367 // Replace the cell *after* the ndk_context swap. The drop of
8368 // the previous GlobalRef happens here (out of the Option). The
8369 // new GlobalRef is now the one keeping `ctx_ptr` live.
8370 *slot = Some(global_ref);
8371 drop(slot);
8372
8373 android_log(
8374 "setAndroidContextJni: ndk_context re-initialized with non-null Context (GlobalRef pinned)",
8375 );
8376}
8377
8378/// JNI: Returns whether `ndk-context`'s stored Context is non-null
8379/// — i.e., whether a prior `setAndroidContextJni` call has wired a
8380/// real Application Context into the global cell.
8381///
8382/// Kotlin signature: `external fun verifyAndroidContextJni(): Boolean`
8383///
8384/// Surface-tier test hook (peat#925 QA BLOCKER). Lets an
8385/// instrumented Android test assert end-to-end that
8386/// Kotlin → JNI → Rust → `ndk_context` actually wired the Context
8387/// through, without having to drive a downstream consumer (e.g.,
8388/// hickory-resolver's Android system-DNS probe) just to verify
8389/// the plumbing. Production code should not call this — the
8390/// information is internal to the wiring contract.
8391#[cfg(target_os = "android")]
8392#[no_mangle]
8393pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_verifyAndroidContextJni(
8394 _env: JNIEnv,
8395 _class: JClass,
8396) -> jni::sys::jboolean {
8397 let stored = ndk_context::android_context().context();
8398 if stored.is_null() {
8399 jni::sys::JNI_FALSE
8400 } else {
8401 jni::sys::JNI_TRUE
8402 }
8403}
8404
8405/// JNI: Test that JNI bindings work
8406///
8407/// Kotlin signature: external fun testJni(): String
8408#[no_mangle]
8409pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_testJni(
8410 mut env: JNIEnv,
8411 _class: JClass,
8412) -> jstring {
8413 let msg = "JNI bindings working! Peat FFI loaded successfully.";
8414 env.new_string(msg)
8415 .expect("Failed to create Java string")
8416 .into_raw()
8417}
8418
8419/// JNI: Create a Peat node (simplified for testing)
8420///
8421/// Kotlin signature: external fun createNodeJni(appId: String, sharedKey:
8422/// String, storagePath: String): Long
8423#[cfg(feature = "sync")]
8424#[no_mangle]
8425pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_createNodeJni(
8426 mut env: JNIEnv,
8427 _class: JClass,
8428 app_id: JString,
8429 shared_key: JString,
8430 storage_path: JString,
8431) -> i64 {
8432 let app_id: String = match env.get_string(&app_id) {
8433 Ok(s) => s.into(),
8434 Err(_) => return 0,
8435 };
8436 let shared_key: String = match env.get_string(&shared_key) {
8437 Ok(s) => s.into(),
8438 Err(_) => return 0,
8439 };
8440 let storage_path: String = match env.get_string(&storage_path) {
8441 Ok(s) => s.into(),
8442 Err(_) => return 0,
8443 };
8444
8445 #[cfg(target_os = "android")]
8446 android_log(&format!(
8447 "createNodeJni: app_id={}, storage_path={}",
8448 app_id, storage_path
8449 ));
8450
8451 let config = NodeConfig {
8452 app_id,
8453 shared_key,
8454 bind_address: None,
8455 storage_path,
8456 transport: None,
8457 };
8458
8459 match create_node(config) {
8460 Ok(node) => {
8461 #[cfg(target_os = "android")]
8462 android_log("createNodeJni: Node created successfully");
8463 // Publish "iroh has started" to any future
8464 // `setAndroidContextJni` reader BEFORE handing the
8465 // handle back to Kotlin. `Release` here pairs with
8466 // `Acquire` in setAndroidContextJni — guarantees all
8467 // writes leading up to this point (iroh handle install,
8468 // tokio runtime startup, iroh worker spawn) are visible
8469 // to a setAndroidContextJni call that observes the flag
8470 // set. One-way: never cleared, even on `freeNodeJni` —
8471 // re-issuing setAndroidContextJni after a node lifecycle
8472 // is still unsafe because iroh tokio workers may
8473 // outlive the node handle.
8474 #[cfg(target_os = "android")]
8475 IROH_STARTED.store(true, std::sync::atomic::Ordering::Release);
8476 // Return the Arc pointer as a handle
8477 // Store an OWNING reference in the global (survives APK
8478 // replacement) BEFORE consuming `node` into the JNI handle, so the
8479 // global owns its own ref rather than aliasing the handle. Released
8480 // by clearGlobalNodeHandleJni, independent of this handle's
8481 // freeNodeJni. See set_global_node_handle.
8482 set_global_node_handle(&node);
8483 let handle = Arc::into_raw(node) as i64;
8484 #[cfg(target_os = "android")]
8485 android_log(&format!("createNodeJni: Stored global handle: {}", handle));
8486 handle
8487 }
8488 Err(e) => {
8489 #[cfg(target_os = "android")]
8490 android_log(&format!("createNodeJni: Error creating node: {:?}", e));
8491 0
8492 }
8493 }
8494}
8495
8496/// JNI: Create a PeatNode with transport configuration (ADR-039, #558)
8497///
8498/// This extended version supports BLE transport configuration for unified
8499/// multi-transport operation. When enable_ble is true, the node will attempt
8500/// to initialize BLE transport alongside the default Iroh transport.
8501///
8502/// Note: On Android, BLE transport requires the Android BLE adapter to be
8503/// initialized via JNI callbacks. Full BLE support is pending Android adapter
8504/// integration in peat-btle.
8505///
8506/// Kotlin signature:
8507/// ```kotlin
8508/// external fun createNodeWithConfigJni(
8509/// appId: String,
8510/// sharedKey: String,
8511/// storagePath: String,
8512/// enableBle: Boolean,
8513/// blePowerProfile: String? // "aggressive", "balanced", or "low_power"
8514/// ): Long
8515/// ```
8516#[cfg(feature = "sync")]
8517#[no_mangle]
8518pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_createNodeWithConfigJni(
8519 mut env: JNIEnv,
8520 _class: JClass,
8521 app_id: JString,
8522 shared_key: JString,
8523 storage_path: JString,
8524 enable_ble: jboolean,
8525 ble_power_profile: JString,
8526) -> i64 {
8527 let app_id: String = match env.get_string(&app_id) {
8528 Ok(s) => s.into(),
8529 Err(_) => return 0,
8530 };
8531 let shared_key: String = match env.get_string(&shared_key) {
8532 Ok(s) => s.into(),
8533 Err(_) => return 0,
8534 };
8535 let storage_path: String = match env.get_string(&storage_path) {
8536 Ok(s) => s.into(),
8537 Err(_) => return 0,
8538 };
8539
8540 // Parse BLE power profile (null/empty string means use default)
8541 let power_profile: Option<String> = env.get_string(&ble_power_profile).ok().and_then(|s| {
8542 let s: String = s.into();
8543 if s.is_empty() {
8544 None
8545 } else {
8546 Some(s)
8547 }
8548 });
8549
8550 #[cfg(target_os = "android")]
8551 android_log(&format!(
8552 "createNodeWithConfigJni: app_id={}, storage_path={}, enable_ble={}, power_profile={:?}",
8553 app_id,
8554 storage_path,
8555 enable_ble != 0,
8556 power_profile
8557 ));
8558
8559 // Build transport configuration
8560 let transport_config = if enable_ble != 0 {
8561 Some(TransportConfigFFI {
8562 enable_ble: true,
8563 ble_mesh_id: None, // Use app_id as mesh ID
8564 ble_power_profile: power_profile,
8565 transport_preference: None,
8566 collection_routes_json: None,
8567 // This legacy/convenience entry doesn't expose the relay toggle;
8568 // keep the local-only posture. The Dart `create_node` path sets
8569 // this from the About-tab toggle.
8570 enable_n0_relay: false,
8571 })
8572 } else {
8573 None
8574 };
8575
8576 let config = NodeConfig {
8577 app_id,
8578 shared_key,
8579 bind_address: None,
8580 storage_path,
8581 transport: transport_config,
8582 };
8583
8584 match create_node(config) {
8585 Ok(node) => {
8586 #[cfg(target_os = "android")]
8587 android_log("createNodeWithConfigJni: Node created successfully");
8588 // Publish iroh-started — same Release/Acquire pairing
8589 // with setAndroidContextJni as in createNodeJni above.
8590 // peat#924 QA WARNING-2.
8591 #[cfg(target_os = "android")]
8592 IROH_STARTED.store(true, std::sync::atomic::Ordering::Release);
8593 // Owning global ref before consuming `node` (see set_global_node_handle).
8594 set_global_node_handle(&node);
8595 let handle = Arc::into_raw(node) as i64;
8596 #[cfg(target_os = "android")]
8597 android_log(&format!(
8598 "createNodeWithConfigJni: Stored global handle: {}",
8599 handle
8600 ));
8601 handle
8602 }
8603 Err(e) => {
8604 #[cfg(target_os = "android")]
8605 android_log(&format!(
8606 "createNodeWithConfigJni: Error creating node: {:?}",
8607 e
8608 ));
8609 0
8610 }
8611 }
8612}
8613
8614/// JNI: Get the global node handle (survives APK replacement)
8615///
8616/// Kotlin signature: external fun getGlobalNodeHandleJni(): Long
8617#[cfg(feature = "sync")]
8618#[no_mangle]
8619pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_getGlobalNodeHandleJni(
8620 _env: JNIEnv,
8621 _class: JClass,
8622) -> i64 {
8623 match GLOBAL_NODE_HANDLE.lock() {
8624 Ok(handle) => {
8625 let h = *handle;
8626 #[cfg(target_os = "android")]
8627 android_log(&format!("getGlobalNodeHandleJni: Returning handle: {}", h));
8628 h
8629 }
8630 Err(_) => 0,
8631 }
8632}
8633
8634/// JNI: Release the owning reference stored in [`GLOBAL_NODE_HANDLE`].
8635///
8636/// Counterpart to the `set_global_node_handle` write performed by every
8637/// node-create path. The bridge that consumes `getGlobalNodeHandleJni` (e.g.
8638/// the BLE pipe) calls this on teardown so the node can actually be freed
8639/// once its originating handle is also released. Safe to call repeatedly and
8640/// when no handle is stored (no-op on `0`).
8641///
8642/// Kotlin signature: `external fun clearGlobalNodeHandleJni()`
8643#[cfg(feature = "sync")]
8644#[no_mangle]
8645pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_clearGlobalNodeHandleJni(
8646 _env: JNIEnv,
8647 _class: JClass,
8648) {
8649 clear_owning_node_slot(&GLOBAL_NODE_HANDLE);
8650}
8651
8652/// JNI: Get node ID from a PeatNode handle
8653///
8654/// Kotlin signature: external fun nodeIdJni(handle: Long): String
8655#[cfg(feature = "sync")]
8656#[no_mangle]
8657pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_nodeIdJni(
8658 mut env: JNIEnv,
8659 _class: JClass,
8660 handle: i64,
8661) -> jstring {
8662 if handle == 0 {
8663 return env
8664 .new_string("")
8665 .expect("Failed to create Java string")
8666 .into_raw();
8667 }
8668
8669 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
8670 let node_id = node.node_id();
8671
8672 // Don't drop the Arc - we're just borrowing
8673 std::mem::forget(node);
8674
8675 env.new_string(&node_id)
8676 .expect("Failed to create Java string")
8677 .into_raw()
8678}
8679
8680/// JNI: Get peer count from a PeatNode handle
8681///
8682/// Kotlin signature: external fun peerCountJni(handle: Long): Int
8683#[cfg(feature = "sync")]
8684#[no_mangle]
8685pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_peerCountJni(
8686 _env: JNIEnv,
8687 _class: JClass,
8688 handle: i64,
8689) -> i32 {
8690 if handle == 0 {
8691 return 0;
8692 }
8693
8694 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
8695 let count = node.peer_count() as i32;
8696
8697 // Don't drop the Arc - we're just borrowing
8698 std::mem::forget(node);
8699
8700 count
8701}
8702
8703/// JNI: Request full document sync with all connected peers
8704///
8705/// Kotlin signature: external fun requestSyncJni(handle: Long): Boolean
8706#[cfg(feature = "sync")]
8707#[no_mangle]
8708pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_requestSyncJni(
8709 _env: JNIEnv,
8710 _class: JClass,
8711 handle: i64,
8712) -> jboolean {
8713 if handle == 0 {
8714 return 0;
8715 }
8716 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
8717 let result = node.request_sync().is_ok();
8718 std::mem::forget(node);
8719 result as jboolean
8720}
8721
8722/// JNI: Get this node's iroh-endpoint first IP socket address as an
8723/// `"ip:port"` string, or null if no socket is bound. The result is
8724/// what `connectPeerJni` expects as its `address` argument when one
8725/// in-process instance dials another on loopback (no discovery layer
8726/// to populate it). peat-mesh#138 M4.
8727///
8728/// Kotlin signature: external fun endpointSocketAddrJni(handle: Long): String?
8729#[cfg(feature = "sync")]
8730#[no_mangle]
8731pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_endpointSocketAddrJni(
8732 env: JNIEnv,
8733 _class: JClass,
8734 handle: i64,
8735) -> jstring {
8736 if handle == 0 {
8737 return std::ptr::null_mut();
8738 }
8739 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
8740 let addr = node.endpoint_socket_addr();
8741 std::mem::forget(node);
8742 match addr {
8743 Some(s) => env
8744 .new_string(s)
8745 .map(|js| js.into_raw())
8746 .unwrap_or(std::ptr::null_mut()),
8747 None => std::ptr::null_mut(),
8748 }
8749}
8750
8751/// Serialize a `peat_mesh::Document` back into the JSON-object shape
8752/// the consumer originally posted via `publishDocumentJni`. The
8753/// publish path hoists an `"id"` field to `Document::id`; this
8754/// helper reinserts it so the round-trip preserves the consumer's
8755/// input shape. Extracted from `getDocumentJni`'s body so the
8756/// serialization can be exercised by an in-crate test independent
8757/// of a JVM (peat#879 QA round 2 — surface-tier coverage for the
8758/// JSON output path).
8759#[cfg(feature = "sync")]
8760fn serialize_document_for_get_jni(doc: &peat_mesh::sync::Document) -> String {
8761 let mut obj = serde_json::Map::new();
8762 for (k, v) in &doc.fields {
8763 obj.insert(k.clone(), v.clone());
8764 }
8765 if let Some(id) = &doc.id {
8766 obj.insert("id".to_string(), serde_json::Value::String(id.clone()));
8767 }
8768 serde_json::Value::Object(obj).to_string()
8769}
8770
8771/// JNI: Read a document back from the local store as JSON, or null
8772/// if the document doesn't exist locally. Complements
8773/// `publishDocumentJni` — needed by instrumented tests that verify
8774/// sync convergence by reading on the receiver side. peat-mesh#138 M4.
8775///
8776/// Kotlin signature: external fun getDocumentJni(handle: Long, collection:
8777/// String, docId: String): String?
8778#[cfg(feature = "sync")]
8779#[no_mangle]
8780pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_getDocumentJni(
8781 mut env: JNIEnv,
8782 _class: JClass,
8783 handle: i64,
8784 collection: JString,
8785 doc_id: JString,
8786) -> jstring {
8787 if handle == 0 {
8788 return std::ptr::null_mut();
8789 }
8790 // peat#885 fault-injection short-circuit, consumed before any
8791 // store interaction. `swap(false, ...)` is one-shot — the next
8792 // call returns to the normal read path. Test-only by API
8793 // contract; production callers never arm the flag.
8794 if FORCE_STORE_ERROR_FOR_TESTING.swap(false, std::sync::atomic::Ordering::SeqCst) {
8795 let _ = env.throw_new(
8796 "java/lang/RuntimeException",
8797 "getDocumentJni: forced store error (test fault injection)",
8798 );
8799 return std::ptr::null_mut();
8800 }
8801 let collection_str: String = match env.get_string(&collection) {
8802 Ok(s) => s.into(),
8803 Err(_) => return std::ptr::null_mut(),
8804 };
8805 let doc_id_str: String = match env.get_string(&doc_id) {
8806 Ok(s) => s.into(),
8807 Err(_) => return std::ptr::null_mut(),
8808 };
8809 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
8810 let mesh_node = Arc::clone(&node_owner.node);
8811 let runtime = Arc::clone(&node_owner.runtime);
8812 std::mem::forget(node_owner);
8813
8814 // Read through the same `peat_mesh::Node` document layer that
8815 // `publishDocumentJni` writes to. The older raw-bytes
8816 // `PeatNode::get_document` reads from a different storage path
8817 // (`storage_backend.collection(...)`) and won't see docs that
8818 // arrived via the document layer's publish or that sync replicas
8819 // applied as Automerge ops. peat-mesh#138 M4 / peat#879 QA.
8820 let result = runtime.block_on(mesh_node.get(&collection_str, &doc_id_str));
8821 match result {
8822 Ok(Some(doc)) => {
8823 let json = serialize_document_for_get_jni(&doc);
8824 env.new_string(json)
8825 .map(|js| js.into_raw())
8826 .unwrap_or(std::ptr::null_mut())
8827 }
8828 Ok(None) => std::ptr::null_mut(),
8829 Err(e) => {
8830 // Distinguish "store read failed" from "not present"
8831 // (peat#879 QA WARNING) — silent null on Err would mask
8832 // hard storage errors as ongoing sync-not-converged, and
8833 // the consumer would spin until timeout. Throw across the
8834 // JNI boundary so the caller sees a fail-fast exception
8835 // with the underlying cause.
8836 let msg = format!("getDocumentJni: store read failed: {e}");
8837 let _ = env.throw_new("java/lang/RuntimeException", &msg);
8838 std::ptr::null_mut()
8839 }
8840 }
8841}
8842
8843/// JNI: Test-only fault injection. Arms a one-shot flag so the next
8844/// `getDocumentJni` call short-circuits to the Err branch (throws
8845/// RuntimeException) without touching the underlying store. Self-
8846/// clears on consumption.
8847///
8848/// Exists so consumers can write a deterministic regression test for
8849/// the `getDocumentJni` `Err(_) → env.throw_new` contract without
8850/// depending on Automerge LRU eviction behavior. See peat#885 /
8851/// peat-mesh#138 M4b carryover.
8852///
8853/// Returns 1 on success, 0 if the handle is invalid.
8854///
8855/// Kotlin signature: external fun forceStoreErrorForTestingJni(handle: Long):
8856/// Boolean
8857#[cfg(feature = "sync")]
8858#[no_mangle]
8859pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_forceStoreErrorForTestingJni(
8860 _env: JNIEnv,
8861 _class: JClass,
8862 handle: i64,
8863) -> jboolean {
8864 if handle == 0 {
8865 return 0;
8866 }
8867 FORCE_STORE_ERROR_FOR_TESTING.store(true, std::sync::atomic::Ordering::SeqCst);
8868 1
8869}
8870
8871/// JNI: Get connected peer IDs as a JSON array
8872///
8873/// Kotlin signature: external fun connectedPeersJni(handle: Long): String
8874/// Returns JSON array of hex-encoded peer IDs, or "[]" on error
8875#[cfg(feature = "sync")]
8876#[no_mangle]
8877pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_connectedPeersJni(
8878 mut env: JNIEnv,
8879 _class: JClass,
8880 handle: i64,
8881) -> jstring {
8882 if handle == 0 {
8883 return env
8884 .new_string("[]")
8885 .expect("Failed to create Java string")
8886 .into_raw();
8887 }
8888
8889 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
8890 let peers = node.connected_peers();
8891 let result = serde_json::to_string(&peers).unwrap_or_else(|_| "[]".to_string());
8892
8893 // Don't drop the Arc - we're just borrowing
8894 std::mem::forget(node);
8895
8896 env.new_string(&result)
8897 .expect("Failed to create Java string")
8898 .into_raw()
8899}
8900
8901/// JNI: Start sync on a PeatNode
8902///
8903/// Kotlin signature: external fun startSyncJni(handle: Long): Boolean
8904#[cfg(feature = "sync")]
8905#[no_mangle]
8906pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_startSyncJni(
8907 _env: JNIEnv,
8908 _class: JClass,
8909 handle: i64,
8910) -> bool {
8911 // CRITICAL DEBUG: Log unconditionally to verify this function is called
8912 eprintln!("startSyncJni: CALLED with handle={}", handle);
8913 #[cfg(target_os = "android")]
8914 android_log(&format!("startSyncJni: ENTERED with handle={}", handle));
8915
8916 if handle == 0 {
8917 #[cfg(target_os = "android")]
8918 android_log("startSyncJni: handle is 0, returning false");
8919 return false;
8920 }
8921
8922 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
8923
8924 #[cfg(target_os = "android")]
8925 android_log("startSyncJni: calling node.start_sync()");
8926
8927 let result = match node.start_sync() {
8928 Ok(()) => {
8929 #[cfg(target_os = "android")]
8930 android_log("startSyncJni: start_sync succeeded");
8931 true
8932 }
8933 Err(e) => {
8934 #[cfg(target_os = "android")]
8935 android_log(&format!("startSyncJni: start_sync failed: {}", e));
8936 false
8937 }
8938 };
8939
8940 // Don't drop the Arc - we're just borrowing
8941 std::mem::forget(node);
8942
8943 result
8944}
8945
8946/// JNI: Free a PeatNode handle
8947///
8948/// Kotlin signature: external fun freeNodeJni(handle: Long)
8949#[cfg(feature = "sync")]
8950#[no_mangle]
8951pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_freeNodeJni(
8952 _env: JNIEnv,
8953 _class: JClass,
8954 handle: i64,
8955) {
8956 if handle != 0 {
8957 #[cfg(target_os = "android")]
8958 android_log(&format!("freeNodeJni: Freeing node handle {}", handle));
8959
8960 // Reconstruct the Arc to drop it
8961 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
8962
8963 // Signal the cleanup task to stop
8964 node.cleanup_running.store(false, Ordering::SeqCst);
8965
8966 #[cfg(target_os = "android")]
8967 android_log("freeNodeJni: Signaled cleanup task to stop");
8968
8969 // Give the background task a moment to exit
8970 std::thread::sleep(std::time::Duration::from_millis(100));
8971
8972 // Clear Android BLE transport global to prevent dangling refs
8973 #[cfg(all(feature = "bluetooth", target_os = "android"))]
8974 {
8975 *ANDROID_BLE_TRANSPORT.lock().unwrap() = None;
8976 android_log("freeNodeJni: Cleared ANDROID_BLE_TRANSPORT");
8977 }
8978
8979 // Drop the node - this should release the database
8980 drop(node);
8981
8982 #[cfg(target_os = "android")]
8983 android_log("freeNodeJni: Node dropped");
8984 }
8985}
8986
8987// =============================================================================
8988// BLE Transport JNI Methods (Android)
8989// =============================================================================
8990
8991/// JNI: Signal BLE transport started/stopped
8992///
8993/// Called by Kotlin when the Android BLE stack is ready or shutting down.
8994/// This makes `is_available()` return true/false for PACE routing.
8995///
8996/// Kotlin signature: external fun bleSetStartedJni(handle: Long, started:
8997/// Boolean)
8998#[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
8999#[no_mangle]
9000pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_bleSetStartedJni(
9001 _env: JNIEnv,
9002 _class: JClass,
9003 handle: i64,
9004 started: jboolean,
9005) {
9006 if handle == 0 {
9007 android_log("bleSetStartedJni: Invalid handle (0)");
9008 return;
9009 }
9010
9011 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
9012
9013 use peat_protocol::transport::MeshTransport;
9014
9015 let guard = ANDROID_BLE_TRANSPORT.lock().unwrap();
9016 if let Some(ref ble_transport) = *guard {
9017 if started != 0 {
9018 match node.runtime.block_on(ble_transport.start()) {
9019 Ok(()) => android_log("bleSetStartedJni: BLE transport started"),
9020 Err(e) => android_log(&format!("bleSetStartedJni: start failed: {}", e)),
9021 }
9022 } else {
9023 match node.runtime.block_on(ble_transport.stop()) {
9024 Ok(()) => android_log("bleSetStartedJni: BLE transport stopped"),
9025 Err(e) => android_log(&format!("bleSetStartedJni: stop failed: {}", e)),
9026 }
9027 }
9028 } else {
9029 android_log("bleSetStartedJni: No BLE transport registered");
9030 }
9031 drop(guard);
9032
9033 // Don't drop the Arc - we're just borrowing
9034 std::mem::forget(node);
9035}
9036
9037/// JNI: Add a reachable BLE peer
9038///
9039/// Called by Kotlin when a BLE peer is discovered/connected.
9040/// This makes `can_reach(peer)` return true for PACE routing.
9041///
9042/// Kotlin signature: external fun bleAddPeerJni(handle: Long, peerId: String)
9043#[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
9044#[no_mangle]
9045pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_bleAddPeerJni(
9046 mut env: JNIEnv,
9047 _class: JClass,
9048 handle: i64,
9049 peer_id: JString,
9050) {
9051 if handle == 0 {
9052 android_log("bleAddPeerJni: Invalid handle (0)");
9053 return;
9054 }
9055
9056 let peer_id_str: String = match env.get_string(&peer_id) {
9057 Ok(s) => s.into(),
9058 Err(_) => {
9059 android_log("bleAddPeerJni: Failed to get peer_id string");
9060 return;
9061 }
9062 };
9063
9064 android_log(&format!("bleAddPeerJni: Adding peer {}", peer_id_str));
9065
9066 let guard = ANDROID_BLE_TRANSPORT.lock().unwrap();
9067 if let Some(ref ble_transport) = *guard {
9068 use peat_protocol::transport::NodeId;
9069 ble_transport.add_reachable_peer(NodeId::new(peer_id_str));
9070 } else {
9071 android_log("bleAddPeerJni: No BLE transport registered");
9072 }
9073}
9074
9075/// JNI: Remove a reachable BLE peer
9076///
9077/// Called by Kotlin when a BLE peer is disconnected/lost.
9078/// This makes `can_reach(peer)` return false for PACE routing.
9079///
9080/// Kotlin signature: external fun bleRemovePeerJni(handle: Long, peerId:
9081/// String)
9082#[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
9083#[no_mangle]
9084pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_bleRemovePeerJni(
9085 mut env: JNIEnv,
9086 _class: JClass,
9087 handle: i64,
9088 peer_id: JString,
9089) {
9090 if handle == 0 {
9091 android_log("bleRemovePeerJni: Invalid handle (0)");
9092 return;
9093 }
9094
9095 let peer_id_str: String = match env.get_string(&peer_id) {
9096 Ok(s) => s.into(),
9097 Err(_) => {
9098 android_log("bleRemovePeerJni: Failed to get peer_id string");
9099 return;
9100 }
9101 };
9102
9103 android_log(&format!("bleRemovePeerJni: Removing peer {}", peer_id_str));
9104
9105 let guard = ANDROID_BLE_TRANSPORT.lock().unwrap();
9106 if let Some(ref ble_transport) = *guard {
9107 use peat_protocol::transport::NodeId;
9108 ble_transport.remove_reachable_peer(&NodeId::new(peer_id_str));
9109 } else {
9110 android_log("bleRemovePeerJni: No BLE transport registered");
9111 }
9112}
9113
9114/// JNI: Query whether BLE transport is available (started)
9115///
9116/// Called by Kotlin to check if BLE transport is active for UI display.
9117/// Returns true if BLE transport has been started via bleSetStartedJni.
9118///
9119/// Kotlin signature: external fun bleIsAvailableJni(handle: Long): Boolean
9120#[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
9121#[no_mangle]
9122pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_bleIsAvailableJni(
9123 _env: JNIEnv,
9124 _class: JClass,
9125 handle: i64,
9126) -> jboolean {
9127 if handle == 0 {
9128 android_log("bleIsAvailableJni: Invalid handle (0)");
9129 return 0;
9130 }
9131
9132 use peat_protocol::transport::Transport;
9133
9134 let guard = ANDROID_BLE_TRANSPORT.lock().unwrap();
9135 let result = match guard.as_ref() {
9136 Some(t) => {
9137 if t.is_available() {
9138 1
9139 } else {
9140 0
9141 }
9142 }
9143 None => 0,
9144 };
9145
9146 android_log(&format!("bleIsAvailableJni: {}", result != 0));
9147 result
9148}
9149
9150/// JNI: Get the number of reachable BLE peers
9151///
9152/// Called by Kotlin to get BLE peer count for unified UI display.
9153/// Returns the number of peers added via bleAddPeerJni.
9154///
9155/// Kotlin signature: external fun blePeerCountJni(handle: Long): Int
9156#[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
9157#[no_mangle]
9158pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_blePeerCountJni(
9159 _env: JNIEnv,
9160 _class: JClass,
9161 handle: i64,
9162) -> jint {
9163 if handle == 0 {
9164 android_log("blePeerCountJni: Invalid handle (0)");
9165 return 0;
9166 }
9167
9168 let guard = ANDROID_BLE_TRANSPORT.lock().unwrap();
9169 let count = match guard.as_ref() {
9170 Some(t) => t.reachable_peer_count() as jint,
9171 None => 0,
9172 };
9173
9174 android_log(&format!("blePeerCountJni: {}", count));
9175 count
9176}
9177
9178/// JNI: Get all cells as JSON array string
9179///
9180/// Kotlin signature: external fun getCellsJni(handle: Long): String
9181/// Returns JSON array of cell objects, or "[]" on error
9182#[cfg(feature = "sync")]
9183#[no_mangle]
9184pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_getCellsJni(
9185 mut env: JNIEnv,
9186 _class: JClass,
9187 handle: i64,
9188) -> jstring {
9189 if handle == 0 {
9190 return env
9191 .new_string("[]")
9192 .expect("Failed to create Java string")
9193 .into_raw();
9194 }
9195
9196 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
9197 let result = match node.get_cells() {
9198 Ok(cells) => {
9199 let json_array: Vec<serde_json::Value> = cells
9200 .iter()
9201 .map(|c| {
9202 serde_json::json!({
9203 "id": c.id,
9204 "name": c.name,
9205 "status": c.status.as_str(),
9206 "node_count": c.node_count,
9207 "center_lat": c.center_lat,
9208 "center_lon": c.center_lon,
9209 "capabilities": c.capabilities,
9210 "formation_id": c.formation_id,
9211 "leader_id": c.leader_id,
9212 "last_update": c.last_update,
9213 "scenario_command": c.scenario_command,
9214 })
9215 })
9216 .collect();
9217 serde_json::to_string(&json_array).unwrap_or_else(|_| "[]".to_string())
9218 }
9219 Err(_) => "[]".to_string(),
9220 };
9221
9222 // Don't drop the Arc - we're just borrowing
9223 std::mem::forget(node);
9224
9225 env.new_string(&result)
9226 .expect("Failed to create Java string")
9227 .into_raw()
9228}
9229
9230/// JNI: Get all tracks as JSON array string
9231///
9232/// Kotlin signature: external fun getTracksJni(handle: Long): String
9233/// Returns JSON array of track objects, or "[]" on error
9234#[cfg(feature = "sync")]
9235#[no_mangle]
9236pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_getTracksJni(
9237 mut env: JNIEnv,
9238 _class: JClass,
9239 handle: i64,
9240) -> jstring {
9241 if handle == 0 {
9242 return env
9243 .new_string("[]")
9244 .expect("Failed to create Java string")
9245 .into_raw();
9246 }
9247
9248 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
9249 let result = match node.get_tracks() {
9250 Ok(tracks) => {
9251 let json_array: Vec<serde_json::Value> = tracks
9252 .iter()
9253 .map(|t| {
9254 serde_json::json!({
9255 "id": t.id,
9256 "source_node": t.source_node,
9257 "cell_id": t.cell_id,
9258 "formation_id": t.formation_id,
9259 "lat": t.lat,
9260 "lon": t.lon,
9261 "hae": t.hae,
9262 "cep": t.cep,
9263 "heading": t.heading,
9264 "speed": t.speed,
9265 "classification": t.classification,
9266 "confidence": t.confidence,
9267 "category": t.category.as_str(),
9268 "created_at": t.created_at,
9269 "last_update": t.last_update,
9270 "attributes": t.attributes,
9271 })
9272 })
9273 .collect();
9274 serde_json::to_string(&json_array).unwrap_or_else(|_| "[]".to_string())
9275 }
9276 Err(_) => "[]".to_string(),
9277 };
9278
9279 // Don't drop the Arc - we're just borrowing
9280 std::mem::forget(node);
9281
9282 env.new_string(&result)
9283 .expect("Failed to create Java string")
9284 .into_raw()
9285}
9286
9287/// JNI: Get all nodes as JSON array string
9288///
9289/// Kotlin signature: external fun getNodesJni(handle: Long): String
9290/// Returns JSON array of node objects, or "[]" on error
9291#[cfg(feature = "sync")]
9292#[no_mangle]
9293pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_getNodesJni(
9294 mut env: JNIEnv,
9295 _class: JClass,
9296 handle: i64,
9297) -> jstring {
9298 if handle == 0 {
9299 return env
9300 .new_string("[]")
9301 .expect("Failed to create Java string")
9302 .into_raw();
9303 }
9304
9305 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
9306 let result = match node.get_nodes() {
9307 Ok(nodes) => serialize_nodes_get_json(&nodes),
9308 Err(_) => "[]".to_string(),
9309 };
9310
9311 // Don't drop the Arc - we're just borrowing
9312 std::mem::forget(node);
9313
9314 env.new_string(&result)
9315 .expect("Failed to create Java string")
9316 .into_raw()
9317}
9318
9319/// JNI: Get all commands as JSON array string
9320///
9321/// Kotlin signature: external fun getCommandsJni(handle: Long): String
9322/// Returns JSON array of command objects, or "[]" on error
9323#[cfg(feature = "sync")]
9324#[no_mangle]
9325pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_getCommandsJni(
9326 mut env: JNIEnv,
9327 _class: JClass,
9328 handle: i64,
9329) -> jstring {
9330 if handle == 0 {
9331 return env
9332 .new_string("[]")
9333 .expect("Failed to create Java string")
9334 .into_raw();
9335 }
9336
9337 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
9338 let result = match node.get_commands() {
9339 Ok(commands) => {
9340 let json_array: Vec<serde_json::Value> = commands
9341 .iter()
9342 .map(|c| {
9343 serde_json::json!({
9344 "id": c.id,
9345 "command_type": c.command_type,
9346 "target_id": c.target_id,
9347 "parameters": c.parameters,
9348 "priority": c.priority,
9349 "status": c.status.as_str(),
9350 "originator": c.originator,
9351 "created_at": c.created_at,
9352 "last_update": c.last_update,
9353 })
9354 })
9355 .collect();
9356 serde_json::to_string(&json_array).unwrap_or_else(|_| "[]".to_string())
9357 }
9358 Err(_) => "[]".to_string(),
9359 };
9360
9361 // Don't drop the Arc - we're just borrowing
9362 std::mem::forget(node);
9363
9364 env.new_string(&result)
9365 .expect("Failed to create Java string")
9366 .into_raw()
9367}
9368
9369/// JNI: Publish a node (self-position/PLI) to the Peat network
9370///
9371/// Kotlin signature: external fun publishNodeJni(handle: Long, nodeJson:
9372/// String): Boolean Stores the node in the "nodes" collection for sync to
9373/// peers.
9374///
9375/// Expected JSON format:
9376/// ```json
9377/// {
9378/// "id": "consumer-device-uid",
9379/// "name": "CALLSIGN",
9380/// "node_type": "SOLDIER",
9381/// "lat": 33.7490,
9382/// "lon": -84.3880,
9383/// "hae": 320.0,
9384/// "heading": 45.0,
9385/// "speed": 1.5,
9386/// "status": "ACTIVE",
9387/// "capabilities": ["PLI"],
9388/// "cell_id": null,
9389/// "readiness": 1.0
9390/// }
9391/// ```
9392#[cfg(feature = "sync")]
9393#[no_mangle]
9394pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_publishNodeJni(
9395 mut env: JNIEnv,
9396 _class: JClass,
9397 handle: i64,
9398 node_json: JString,
9399) -> jboolean {
9400 if handle == 0 {
9401 #[cfg(target_os = "android")]
9402 android_log("publishNodeJni: Invalid handle (0)");
9403 return 0; // JNI_FALSE
9404 }
9405
9406 // Get node JSON string from Java
9407 let json_str: String = match env.get_string(&node_json) {
9408 Ok(s) => s.into(),
9409 Err(e) => {
9410 #[cfg(target_os = "android")]
9411 android_log(&format!(
9412 "publishNodeJni: Failed to get JSON string: {:?}",
9413 e
9414 ));
9415 return 0; // JNI_FALSE
9416 }
9417 };
9418
9419 #[cfg(target_os = "android")]
9420 android_log(&format!("publishNodeJni: Received JSON: {}", json_str));
9421
9422 // Parse JSON via the shared helper so the test suite exercises the
9423 // same code the JNI surface does. Pre-2026-05-08 this was inlined
9424 // here, which made it a duplicated codec the unit tests didn't
9425 // reach — the silent-field-drop bug class peat#835 exists to lock
9426 // in came in through this exact site.
9427 let node: NodeInfo = match parse_node_publish_json(&json_str) {
9428 Ok(p) => p,
9429 Err(e) => {
9430 #[cfg(target_os = "android")]
9431 android_log(&format!("publishNodeJni: {}", e));
9432 return 0; // JNI_FALSE
9433 }
9434 };
9435
9436 #[cfg(target_os = "android")]
9437 android_log(&format!(
9438 "publishNodeJni: Publishing node id={}, name={}, lat={}, lon={}",
9439 node.id, node.name, node.lat, node.lon
9440 ));
9441
9442 // Get node from handle and store node
9443 let peat_node = unsafe { Arc::from_raw(handle as *const PeatNode) };
9444 let result = match peat_node.put_node(node) {
9445 Ok(_) => {
9446 #[cfg(target_os = "android")]
9447 android_log("publishNodeJni: Node published successfully");
9448 1 // JNI_TRUE
9449 }
9450 Err(e) => {
9451 #[cfg(target_os = "android")]
9452 android_log(&format!("publishNodeJni: Failed to publish: {:?}", e));
9453 0 // JNI_FALSE
9454 }
9455 };
9456
9457 // Don't drop the Arc - we're just borrowing
9458 std::mem::forget(peat_node);
9459
9460 result
9461}
9462
9463/// JNI: Get all markers as JSON array string
9464///
9465/// Kotlin signature: `external fun getMarkersJni(handle: Long): String`
9466/// Returns JSON array of marker objects, or `"[]"` on error.
9467#[cfg(feature = "sync")]
9468#[no_mangle]
9469pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_getMarkersJni(
9470 mut env: JNIEnv,
9471 _class: JClass,
9472 handle: i64,
9473) -> jstring {
9474 if handle == 0 {
9475 return env
9476 .new_string("[]")
9477 .expect("Failed to create Java string")
9478 .into_raw();
9479 }
9480
9481 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
9482 let result = match node.get_markers() {
9483 Ok(markers) => serialize_markers_get_json(&markers),
9484 Err(e) => {
9485 // Surface storage failures the same way the publish
9486 // side does — otherwise Kotlin sees `"[]"` and can't
9487 // tell "no markers" from "storage error retrieving
9488 // markers." Triage on a tablet starts with the
9489 // PeatFFI logcat tag; this line is what makes "marker
9490 // didn't sync" reports actionable.
9491 #[cfg(target_os = "android")]
9492 android_log(&format!("getMarkersJni: get_markers failed: {:?}", e));
9493 let _ = e;
9494 "[]".to_string()
9495 }
9496 };
9497
9498 // Don't drop the Arc - we're just borrowing
9499 std::mem::forget(node);
9500
9501 env.new_string(&result)
9502 .expect("Failed to create Java string")
9503 .into_raw()
9504}
9505
9506/// JNI: Publish a marker into the doc store. Routes through the
9507/// universal-Document transport on every registered radio
9508/// (LiteBridgeTranslator on BLE, iroh sync for cross-mesh peers).
9509///
9510/// Kotlin signature: `external fun publishMarkerJni(handle: Long, markerJson:
9511/// String): Boolean` Returns `1` (JNI_TRUE) on success, `0` (JNI_FALSE) on
9512/// failure (invalid handle, malformed JSON, missing required fields, storage
9513/// error). The Kotlin caller maps the boolean return back to a
9514/// success / "publish failed" log path — same shape as
9515/// `publishNodeJni`.
9516#[cfg(feature = "sync")]
9517#[no_mangle]
9518pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_publishMarkerJni(
9519 mut env: JNIEnv,
9520 _class: JClass,
9521 handle: i64,
9522 marker_json: JString,
9523) -> jboolean {
9524 if handle == 0 {
9525 #[cfg(target_os = "android")]
9526 android_log("publishMarkerJni: Invalid handle (0)");
9527 return 0;
9528 }
9529
9530 let json_str: String = match env.get_string(&marker_json) {
9531 Ok(s) => s.into(),
9532 Err(e) => {
9533 #[cfg(target_os = "android")]
9534 android_log(&format!(
9535 "publishMarkerJni: Failed to get JSON string: {:?}",
9536 e
9537 ));
9538 let _ = e;
9539 return 0;
9540 }
9541 };
9542
9543 #[cfg(target_os = "android")]
9544 android_log(&format!("publishMarkerJni: Received JSON: {}", json_str));
9545
9546 // Parse — uid is read from the body (no doc-store id available
9547 // pre-storage). parse_marker_publish_json's `id` parameter is
9548 // accepted for the scan-side path; on publish we pass the
9549 // body's uid and reject if absent.
9550 let marker: MarkerInfo = match parse_marker_publish_json("", &json_str) {
9551 Ok(m) => m,
9552 Err(e) => {
9553 #[cfg(target_os = "android")]
9554 android_log(&format!("publishMarkerJni: parse error: {:?}", e));
9555 let _ = e;
9556 return 0;
9557 }
9558 };
9559
9560 #[cfg(target_os = "android")]
9561 if marker.deleted {
9562 android_log(&format!(
9563 "publishMarkerJni: Publishing TOMBSTONE for uid={}",
9564 marker.uid
9565 ));
9566 } else {
9567 android_log(&format!(
9568 "publishMarkerJni: Publishing marker uid={}, type={}, lat={}, lon={}",
9569 marker.uid, marker.marker_type, marker.lat, marker.lon
9570 ));
9571 }
9572
9573 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
9574 let result = match node.put_marker(marker) {
9575 Ok(_) => {
9576 #[cfg(target_os = "android")]
9577 android_log("publishMarkerJni: Marker published successfully");
9578 1
9579 }
9580 Err(e) => {
9581 #[cfg(target_os = "android")]
9582 android_log(&format!("publishMarkerJni: Failed to publish: {:?}", e));
9583 let _ = e;
9584 0
9585 }
9586 };
9587
9588 std::mem::forget(node);
9589 result
9590}
9591
9592/// Publish a generic document into a named collection via `peat_mesh::Node`.
9593///
9594/// JNI wrapper around [`publish_document_into_node`]. The Kotlin caller passes
9595/// a JSON object; top-level keys become the document body. The `"id"` field
9596/// is optional — when present and a string, it becomes the document's id;
9597/// when absent or non-string, the backend assigns one (and returns it). The
9598/// returned Java string is the id that was actually used (caller-supplied or
9599/// backend-assigned), so callers needing a stable id must capture the return
9600/// value rather than assuming the input `"id"` won.
9601///
9602/// Returns an empty Java string on failure: handle invalid, JSON malformed,
9603/// JSON not an object, or backend publish error. Foundation step 3 of the
9604/// peat-mesh-completion work.
9605///
9606/// Kotlin signature: `external fun publishDocumentJni(handle: Long, collection:
9607/// String, json: String): String`
9608#[cfg(feature = "sync")]
9609#[no_mangle]
9610pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_publishDocumentJni(
9611 mut env: JNIEnv,
9612 _class: JClass,
9613 handle: i64,
9614 collection: JString,
9615 json: JString,
9616) -> jstring {
9617 // Track the result string we want to return; build the jstring at the
9618 // single env.new_string() call site at the end. Avoids the tangle of
9619 // borrowing `env` multiple times across short-circuit error returns.
9620 let result_str: String = if handle == 0 {
9621 #[cfg(target_os = "android")]
9622 android_log("publishDocumentJni: Invalid handle (0)");
9623 String::new()
9624 } else {
9625 match (env.get_string(&collection), env.get_string(&json)) {
9626 (Ok(c), Ok(j)) => {
9627 let collection_str: String = c.into();
9628 let json_str: String = j.into();
9629 // Borrow the node Arc without taking ownership — same
9630 // pattern as the other ..._Jni functions in this file.
9631 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
9632 let mesh_node = Arc::clone(&node_owner.node);
9633 let runtime = Arc::clone(&node_owner.runtime);
9634 std::mem::forget(node_owner);
9635
9636 // clippy suggests `.unwrap_or_default()` but the Err
9637 // arm has a real side effect (android_log call) that
9638 // would be lost.
9639 #[allow(clippy::manual_unwrap_or_default)]
9640 match runtime.block_on(publish_document_into_node(
9641 &mesh_node,
9642 &collection_str,
9643 &json_str,
9644 )) {
9645 Ok(id) => id,
9646 Err(_e) => {
9647 #[cfg(target_os = "android")]
9648 android_log(&format!("publishDocumentJni: publish failed: {}", _e));
9649 String::new()
9650 }
9651 }
9652 }
9653 (Err(_e), _) | (_, Err(_e)) => {
9654 #[cfg(target_os = "android")]
9655 android_log(&format!(
9656 "publishDocumentJni: failed to read args: {:?}",
9657 _e
9658 ));
9659 String::new()
9660 }
9661 }
9662 };
9663
9664 env.new_string(result_str)
9665 .map(|s| s.into_raw())
9666 .unwrap_or(std::ptr::null_mut())
9667}
9668
9669/// Origin-aware sibling of [`Java_..._publishDocumentJni`]
9670/// (ADR-059 Amendment 2 — Slice 1.b.4 host-side wiring).
9671///
9672/// Same body as `publishDocumentJni` plus an `origin` parameter that
9673/// flows through to [`peat_mesh::Node::publish_with_origin`]. The
9674/// plugin's `BleDecodedDocumentBridge` calls this with `origin="ble"`
9675/// after decoding a 0xB6 translator frame, so cross-transport fan-out's
9676/// loop-prevention skips the BLE channel on this node and the doc
9677/// doesn't re-emit back out the way it came.
9678///
9679/// Empty `origin` is treated as `None` (equivalent to plain
9680/// `publishDocumentJni`); any non-empty string is passed through
9681/// verbatim. peat-mesh validates the origin against the registered
9682/// transport set; an unknown origin produces a publish-time error
9683/// (logged + empty return string).
9684///
9685/// Kotlin signature: `external fun publishDocumentWithOriginJni(handle: Long,
9686/// collection: String, json: String, origin: String): String`
9687#[cfg(feature = "sync")]
9688#[no_mangle]
9689pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_publishDocumentWithOriginJni(
9690 mut env: JNIEnv,
9691 _class: JClass,
9692 handle: i64,
9693 collection: JString,
9694 json: JString,
9695 origin: JString,
9696) -> jstring {
9697 let result_str: String = if handle == 0 {
9698 #[cfg(target_os = "android")]
9699 android_log("publishDocumentWithOriginJni: Invalid handle (0)");
9700 String::new()
9701 } else {
9702 match (
9703 env.get_string(&collection),
9704 env.get_string(&json),
9705 env.get_string(&origin),
9706 ) {
9707 (Ok(c), Ok(j), Ok(o)) => {
9708 let collection_str: String = c.into();
9709 let json_str: String = j.into();
9710 let origin_str: String = o.into();
9711 let origin_opt = if origin_str.is_empty() {
9712 None
9713 } else {
9714 Some(origin_str)
9715 };
9716 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
9717 let mesh_node = Arc::clone(&node_owner.node);
9718 let runtime = Arc::clone(&node_owner.runtime);
9719 std::mem::forget(node_owner);
9720
9721 #[allow(clippy::manual_unwrap_or_default)]
9722 match runtime.block_on(publish_document_into_node_with_origin(
9723 &mesh_node,
9724 &collection_str,
9725 &json_str,
9726 origin_opt,
9727 )) {
9728 Ok(id) => id,
9729 Err(_e) => {
9730 #[cfg(target_os = "android")]
9731 android_log(&format!(
9732 "publishDocumentWithOriginJni: publish failed: {}",
9733 _e
9734 ));
9735 String::new()
9736 }
9737 }
9738 }
9739 // Per-position match preserves the underlying JNI error in
9740 // the diagnostic, matching `publishDocumentJni`'s shape. A
9741 // wildcard arm would drop `_e` and obscure plugin-side
9742 // debugging when one of the three string args is malformed.
9743 (Err(_e), _, _) | (_, Err(_e), _) | (_, _, Err(_e)) => {
9744 #[cfg(target_os = "android")]
9745 android_log(&format!(
9746 "publishDocumentWithOriginJni: failed to read args: {:?}",
9747 _e
9748 ));
9749 String::new()
9750 }
9751 }
9752 };
9753
9754 env.new_string(result_str)
9755 .map(|s| s.into_raw())
9756 .unwrap_or(std::ptr::null_mut())
9757}
9758
9759/// Pure-Rust helper backing [`Java_..._publishDocumentJni`]. Parses a JSON
9760/// object into a [`peat_mesh::sync::types::Document`] (the `"id"` string
9761/// field, if present, becomes [`Document::id`]; remaining keys land in
9762/// [`Document::fields`]) and publishes it into the given collection on the
9763/// node. Exposed for unit tests so the conversion + publish path can be
9764/// exercised without spinning up a JVM.
9765#[cfg(feature = "sync")]
9766async fn publish_document_into_node(
9767 node: &peat_mesh::Node,
9768 collection: &str,
9769 json: &str,
9770) -> anyhow::Result<String> {
9771 publish_document_into_node_with_origin(node, collection, json, None).await
9772}
9773
9774/// Origin-aware sibling of [`publish_document_into_node`], backing
9775/// [`Java_..._publishDocumentWithOriginJni`] (ADR-059 Amendment 2 Slice
9776/// 1.b.4). When `origin` is `Some(_)`, publishes via
9777/// [`peat_mesh::Node::publish_with_origin`] so cross-transport fan-out's
9778/// loop-prevention skips the named origin transport — required for the
9779/// plugin's `BleDecodedDocumentBridge` to ingest 0xB6 frames into the
9780/// doc store without re-emitting them back out to BLE. With `None` this
9781/// behaves identically to a plain `publish`. Exposed for unit tests so
9782/// the parse + publish-with-origin path can be exercised without a JVM.
9783#[cfg(feature = "sync")]
9784async fn publish_document_into_node_with_origin(
9785 node: &peat_mesh::Node,
9786 collection: &str,
9787 json: &str,
9788 origin: Option<String>,
9789) -> anyhow::Result<String> {
9790 use peat_mesh::sync::types::Document;
9791 use serde_json::Value;
9792
9793 let value: Value =
9794 serde_json::from_str(json).map_err(|e| anyhow::anyhow!("invalid JSON: {}", e))?;
9795
9796 let mut obj = match value {
9797 Value::Object(map) => map,
9798 other => {
9799 return Err(anyhow::anyhow!(
9800 "document JSON must be an object, got {:?}",
9801 other
9802 ))
9803 }
9804 };
9805
9806 let id = obj.remove("id").and_then(|v| match v {
9807 Value::String(s) => Some(s),
9808 _ => None,
9809 });
9810
9811 let fields = obj.into_iter().collect();
9812 let document = match id {
9813 Some(id) => Document::with_id(id, fields),
9814 None => Document::new(fields),
9815 };
9816
9817 match origin {
9818 Some(o) => {
9819 node.publish_with_origin(collection, document, Some(o))
9820 .await
9821 }
9822 None => node.publish(collection, document).await,
9823 }
9824}
9825
9826/// Ingest a peat-btle [`BlePosition`]-shaped JSON envelope: translate it
9827/// to an Automerge track document via [`BleTranslator`] and publish into
9828/// [`peat_mesh::Node`] with `Some("ble")` origin (ADR-059). From there
9829/// iroh-bound peers receive the doc through Automerge sync; the origin
9830/// rides on the resulting `ChangeEvent` so `TransportManager`'s fan-out
9831/// suppresses the same-node `BLE → Node → observer → BLE` echo.
9832///
9833/// JSON envelope (matches the `BlePosition` field shape plus the surrounding
9834/// metadata the translator needs):
9835/// ```json
9836/// {
9837/// "lat": 40.7,
9838/// "lon": -74.0,
9839/// "altitude": 100.0, // optional
9840/// "accuracy": 5.0, // optional
9841/// "peripheral_id": 3405643777,
9842/// "callsign": "SCOUT-CAFE", // optional
9843/// "mesh_id": "29C916FA" // optional
9844/// }
9845/// ```
9846///
9847/// `peripheral_id` accepts the full u32 range expressed two ways: as a
9848/// non-negative integer (Kotlin `Long`/`UInt` paths) or as a sign-extended
9849/// negative integer (Kotlin `Int.toLong()` of a u32 with the high bit set —
9850/// e.g. `0xCAFE_0001` reads as `-889323519` through a signed Int). Both forms
9851/// recover the same u32 internally; values above `u32::MAX` or below
9852/// `i32::MIN` are rejected rather than silently truncated. See
9853/// [`parse_peripheral_id`].
9854///
9855/// Kotlin signature: `external fun ingestPositionJni(handle: Long, json:
9856/// String): String`
9857///
9858/// Returns the assigned track-document id on success, or empty string on any
9859/// failure (handle invalid, bluetooth feature not built, JSON malformed,
9860/// missing required fields, peripheral_id out of range, publish error).
9861///
9862/// [`BleTranslator`]: peat_protocol::sync::ble_translation::BleTranslator
9863#[cfg(all(feature = "sync", feature = "bluetooth"))]
9864#[no_mangle]
9865pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_ingestPositionJni(
9866 mut env: JNIEnv,
9867 _class: JClass,
9868 handle: i64,
9869 json: JString,
9870) -> jstring {
9871 let result_str: String = if handle == 0 {
9872 #[cfg(target_os = "android")]
9873 android_log("ingestPositionJni: Invalid handle (0)");
9874 String::new()
9875 } else {
9876 match env.get_string(&json) {
9877 Ok(j) => {
9878 let json_str: String = j.into();
9879 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
9880 let translator = Arc::clone(&node_owner.ble_translator);
9881 let node = Arc::clone(&node_owner.node);
9882 let runtime = Arc::clone(&node_owner.runtime);
9883 std::mem::forget(node_owner);
9884
9885 // The Err arm has a side effect (android_log) that
9886 // `unwrap_or_default()` cannot preserve, so the `match`
9887 // is intentional. Keeping the lint silenced explicitly
9888 // mirrors the same decision in pre-Slice-1.b.2.2 code.
9889 #[allow(clippy::manual_unwrap_or_default)]
9890 match runtime.block_on(ingest_position_via_translator(
9891 &translator,
9892 &node,
9893 &json_str,
9894 )) {
9895 Ok(id) => id,
9896 Err(_e) => {
9897 #[cfg(target_os = "android")]
9898 android_log(&format!("ingestPositionJni: ingest failed: {}", _e));
9899 String::new()
9900 }
9901 }
9902 }
9903 Err(_e) => {
9904 #[cfg(target_os = "android")]
9905 android_log(&format!("ingestPositionJni: failed to read json: {:?}", _e));
9906 String::new()
9907 }
9908 }
9909 };
9910
9911 env.new_string(result_str)
9912 .map(|s| s.into_raw())
9913 .unwrap_or(std::ptr::null_mut())
9914}
9915
9916/// JNI: Ingest an inbound frame received over BLE into the mesh.
9917///
9918/// Kotlin signature:
9919/// `external fun ingestInboundFrameJni(handle: Long, collection: String,
9920/// postcardBytes: ByteArray): String?`
9921///
9922/// Thin wrapper over [`PeatNode::ingest_inbound_frame`], which decodes the
9923/// frame via the `BleTranslator` and publishes it into the mesh tagged with
9924/// `Some("ble")` origin — so `TransportManager`'s per-transport fan-out
9925/// re-emits it to the OTHER transports (iroh / Wi-Fi) without looping back
9926/// to BLE (ADR-059). This is the inbound counterpart of
9927/// `subscribeOutboundFramesJni`: a Kotlin BLE manager calls this with each
9928/// decrypted frame it receives over the radio.
9929///
9930/// Returns the published document id, or null on failure / no-op (invalid
9931/// handle, byte/string marshaling error, or the translator produced no
9932/// document).
9933#[cfg(all(feature = "sync", feature = "bluetooth"))]
9934#[no_mangle]
9935pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_ingestInboundFrameJni(
9936 mut env: JNIEnv,
9937 _class: JClass,
9938 handle: i64,
9939 collection: JString,
9940 postcard_bytes: JByteArray,
9941) -> jstring {
9942 if handle == 0 {
9943 #[cfg(target_os = "android")]
9944 android_log("ingestInboundFrameJni: Invalid handle (0)");
9945 return std::ptr::null_mut();
9946 }
9947 let collection_str: String = match env.get_string(&collection) {
9948 Ok(s) => s.into(),
9949 Err(_e) => {
9950 #[cfg(target_os = "android")]
9951 android_log(&format!(
9952 "ingestInboundFrameJni: failed to read collection: {:?}",
9953 _e
9954 ));
9955 return std::ptr::null_mut();
9956 }
9957 };
9958 let bytes: Vec<u8> = match env.convert_byte_array(&postcard_bytes) {
9959 Ok(b) => b,
9960 Err(_e) => {
9961 #[cfg(target_os = "android")]
9962 android_log(&format!(
9963 "ingestInboundFrameJni: failed to read bytes: {:?}",
9964 _e
9965 ));
9966 return std::ptr::null_mut();
9967 }
9968 };
9969
9970 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
9971 let result = node_owner.ingest_inbound_frame(collection_str, bytes);
9972 std::mem::forget(node_owner);
9973
9974 match result {
9975 Ok(Some(id)) => env
9976 .new_string(id)
9977 .map(|s| s.into_raw())
9978 .unwrap_or(std::ptr::null_mut()),
9979 Ok(None) => std::ptr::null_mut(),
9980 Err(_e) => {
9981 #[cfg(target_os = "android")]
9982 android_log(&format!("ingestInboundFrameJni: ingest failed: {}", _e));
9983 std::ptr::null_mut()
9984 }
9985 }
9986}
9987
9988/// JNI: Ingest an inbound BLE frame on the universal-Document (peat-lite /
9989/// `ble-lite`) codec — the counterpart of `ingestInboundFrameJni` for raw
9990/// collections the typed translator declines (e.g. the `demo` counter).
9991///
9992/// Kotlin signature:
9993/// `external fun ingestInboundLiteFrameJni(handle: Long, collection: String,
9994/// envelopeBytes: ByteArray): String?`
9995#[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
9996#[no_mangle]
9997pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_ingestInboundLiteFrameJni(
9998 mut env: JNIEnv,
9999 _class: JClass,
10000 handle: i64,
10001 collection: JString,
10002 envelope_bytes: JByteArray,
10003) -> jstring {
10004 if handle == 0 {
10005 #[cfg(target_os = "android")]
10006 android_log("ingestInboundLiteFrameJni: Invalid handle (0)");
10007 return std::ptr::null_mut();
10008 }
10009 let collection_str: String = match env.get_string(&collection) {
10010 Ok(s) => s.into(),
10011 Err(_e) => {
10012 #[cfg(target_os = "android")]
10013 android_log(&format!(
10014 "ingestInboundLiteFrameJni: failed to read collection: {:?}",
10015 _e
10016 ));
10017 return std::ptr::null_mut();
10018 }
10019 };
10020 let bytes: Vec<u8> = match env.convert_byte_array(&envelope_bytes) {
10021 Ok(b) => b,
10022 Err(_e) => {
10023 #[cfg(target_os = "android")]
10024 android_log(&format!(
10025 "ingestInboundLiteFrameJni: failed to read bytes: {:?}",
10026 _e
10027 ));
10028 return std::ptr::null_mut();
10029 }
10030 };
10031
10032 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
10033 let result = node_owner.ingest_inbound_lite_frame(collection_str, bytes);
10034 std::mem::forget(node_owner);
10035
10036 match result {
10037 Ok(Some(id)) => env
10038 .new_string(id)
10039 .map(|s| s.into_raw())
10040 .unwrap_or(std::ptr::null_mut()),
10041 Ok(None) => std::ptr::null_mut(),
10042 Err(_e) => {
10043 #[cfg(target_os = "android")]
10044 android_log(&format!("ingestInboundLiteFrameJni: ingest failed: {}", _e));
10045 std::ptr::null_mut()
10046 }
10047 }
10048}
10049
10050/// JNI: ingest an inbound CRDT-counter frame (CRDT-over-Automerge-over-BLE).
10051///
10052/// `hex_bytes` is the UTF-8 hex of the shared Automerge doc's `save()` bytes —
10053/// the payload of a `0xAF` frame whose transport byte is `2` (crdt). Merges it
10054/// into the shared counter (idempotent/commutative) and returns the new value,
10055/// or -1 on error. Operates on the SAME `PeatNode` Dart created (the global
10056/// handle is an owning alias), so Dart's `crdtCounterValue()` sees the result.
10057///
10058/// Routes by `collection`: `"supply"` merges the Counter (returns the new
10059/// value); any other collection merges the generic CRDT KV doc (returns 0).
10060/// Returns -1 on error.
10061///
10062/// Kotlin: `external fun ingestCrdtFrameJni(handle: Long, collection: String,
10063/// hexBytes: ByteArray): Long`
10064#[cfg(all(feature = "sync", feature = "bluetooth"))]
10065#[no_mangle]
10066pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_ingestCrdtFrameJni(
10067 mut env: JNIEnv,
10068 _class: JClass,
10069 handle: i64,
10070 collection: JString,
10071 hex_bytes: JByteArray,
10072) -> i64 {
10073 if handle == 0 {
10074 return -1;
10075 }
10076 let collection_str: String = match env.get_string(&collection) {
10077 Ok(s) => s.into(),
10078 Err(_) => return -1,
10079 };
10080 let bytes: Vec<u8> = match env.convert_byte_array(&hex_bytes) {
10081 Ok(b) => b,
10082 Err(_) => return -1,
10083 };
10084 let hex = match String::from_utf8(bytes) {
10085 Ok(s) => s,
10086 Err(_) => return -1,
10087 };
10088 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
10089 let v = if collection_str == "supply" {
10090 node_owner.crdt_counter_merge(hex)
10091 } else {
10092 node_owner.crdt_kv_merge(collection_str, hex);
10093 0
10094 };
10095 std::mem::forget(node_owner);
10096 v
10097}
10098
10099/// Pure-Rust helper backing [`Java_..._ingestPositionJni`]. Parses the JSON
10100/// envelope into a [`BlePosition`] plus the surrounding ingest metadata,
10101/// translates to an Automerge document via [`BleTranslator`], and publishes
10102/// into [`peat_mesh::Node`] with `Some("ble")` origin per ADR-059. Exposed
10103/// for unit tests so the parse + translate + publish path can be exercised
10104/// without spinning up a JVM.
10105///
10106/// Hand-rolled JSON parsing rather than `#[derive(Deserialize)]` because
10107/// peat-ffi does not currently depend on `serde` directly (only
10108/// `serde_json`); adding it just for one private marshaling struct isn't
10109/// worth a Cargo.toml change and a fresh transitive footprint.
10110///
10111/// [`BlePosition`]: peat_protocol::sync::ble_translation::BlePosition
10112/// [`BleTranslator`]: peat_protocol::sync::ble_translation::BleTranslator
10113#[cfg(all(feature = "sync", feature = "bluetooth"))]
10114async fn ingest_position_via_translator(
10115 translator: &peat_protocol::sync::ble_translation::BleTranslator,
10116 node: &peat_mesh::Node,
10117 json: &str,
10118) -> anyhow::Result<String> {
10119 use peat_protocol::sync::ble_translation::{value_to_mesh_document, BlePosition};
10120 use serde_json::Value;
10121
10122 let value: Value = serde_json::from_str(json)
10123 .map_err(|e| anyhow::anyhow!("invalid ingest-position JSON: {}", e))?;
10124 let obj = value
10125 .as_object()
10126 .ok_or_else(|| anyhow::anyhow!("ingest-position JSON must be an object"))?;
10127
10128 let lat = obj
10129 .get("lat")
10130 .and_then(Value::as_f64)
10131 .ok_or_else(|| anyhow::anyhow!("ingest-position: missing or non-numeric `lat`"))?
10132 as f32;
10133 let lon = obj
10134 .get("lon")
10135 .and_then(Value::as_f64)
10136 .ok_or_else(|| anyhow::anyhow!("ingest-position: missing or non-numeric `lon`"))?
10137 as f32;
10138 let peripheral_id = parse_peripheral_id(obj.get("peripheral_id"))?;
10139
10140 let altitude = obj
10141 .get("altitude")
10142 .and_then(Value::as_f64)
10143 .map(|v| v as f32);
10144 let accuracy = obj
10145 .get("accuracy")
10146 .and_then(Value::as_f64)
10147 .map(|v| v as f32);
10148 let callsign = obj
10149 .get("callsign")
10150 .and_then(Value::as_str)
10151 .map(str::to_string);
10152 let mesh_id = obj
10153 .get("mesh_id")
10154 .and_then(Value::as_str)
10155 .map(str::to_string);
10156
10157 let position = BlePosition {
10158 latitude: lat,
10159 longitude: lon,
10160 altitude,
10161 accuracy,
10162 };
10163
10164 // Translate, then publish through Node::publish_with_origin so the
10165 // `Some("ble")` origin rides on the resulting ChangeEvent — without
10166 // it, TransportManager fan-out cannot break the BLE-loop on this
10167 // node (ADR-059 §"Origin propagation through async observer
10168 // pipelines").
10169 let value = translator.position_to_track_in_cell(
10170 &position,
10171 peripheral_id,
10172 callsign.as_deref(),
10173 mesh_id.as_deref(),
10174 );
10175 let doc = value_to_mesh_document(value);
10176 node.publish_with_origin(translator.tracks_collection(), doc, Some("ble".to_string()))
10177 .await
10178}
10179
10180/// Parse a `peripheral_id` JSON value into a `u32`, accepting both the
10181/// positive form (Kotlin `Long` / `UInt`) and the sign-extended-Int form
10182/// (Kotlin `Int.toLong()` of a value with the high bit set, which serializes
10183/// as a negative JSON literal). Reinterprets the bits via `i32 as u32` for
10184/// the negative case so a watch with peripheral_id `0xCAFE_0001` round-trips
10185/// the same regardless of which Kotlin numeric type the caller used.
10186///
10187/// Rejects missing values, non-integer values, and values outside
10188/// `[i32::MIN, u32::MAX]` (above-u32::MAX would otherwise silently truncate
10189/// and collide distinct logical IDs onto the same translator-emitted track
10190/// id `ble-XXXXXXXX`, mis-attributing positions to peers — caught by PR
10191/// #804 round-1 review).
10192#[cfg(all(feature = "sync", feature = "bluetooth"))]
10193fn parse_peripheral_id(value: Option<&serde_json::Value>) -> anyhow::Result<u32> {
10194 let raw = value.and_then(serde_json::Value::as_i64).ok_or_else(|| {
10195 anyhow::anyhow!("ingest-position: missing or non-integer `peripheral_id`")
10196 })?;
10197
10198 if (0..=u32::MAX as i64).contains(&raw) {
10199 // Positive: Kotlin Long, UInt, or any numeric type that produced a
10200 // non-negative JSON literal. Direct cast — no truncation since we
10201 // bounded above.
10202 Ok(raw as u32)
10203 } else if (i32::MIN as i64..=-1).contains(&raw) {
10204 // Negative: Kotlin Int.toLong() of a u32 with the high bit set
10205 // (e.g. 0xCAFE_0001 = 3_405_643_777 stored in a signed Int reads as
10206 // -889_323_519). `as i32` preserves the bit pattern, then
10207 // `as u32` reinterprets — so the recovered u32 matches what the
10208 // caller's u32 originally was, before Kotlin's signed-Int coercion.
10209 Ok((raw as i32) as u32)
10210 } else {
10211 Err(anyhow::anyhow!(
10212 "ingest-position: `peripheral_id` {} out of u32 range \
10213 (accepts [i32::MIN, u32::MAX] to handle both Kotlin Int and Long callers)",
10214 raw
10215 ))
10216 }
10217}
10218
10219/// Connect to a known peer by node ID and address (bypasses mDNS).
10220///
10221/// Kotlin signature: external fun connectPeerJni(handle: Long, nodeId: String,
10222/// address: String): Boolean Used by the dual-transport test to connect Android
10223/// to rpi-ci2 over QUIC when mDNS is unreliable.
10224#[cfg(feature = "sync")]
10225#[no_mangle]
10226pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_connectPeerJni(
10227 mut env: JNIEnv,
10228 _class: JClass,
10229 handle: i64,
10230 node_id: JString,
10231 address: JString,
10232) -> jboolean {
10233 if handle == 0 {
10234 #[cfg(target_os = "android")]
10235 android_log("connectPeerJni: Invalid handle (0)");
10236 return 0;
10237 }
10238
10239 let node_id_str: String = match env.get_string(&node_id) {
10240 Ok(s) => s.into(),
10241 Err(e) => {
10242 #[cfg(target_os = "android")]
10243 android_log(&format!("connectPeerJni: Failed to get nodeId: {:?}", e));
10244 return 0;
10245 }
10246 };
10247
10248 let addr_str: String = match env.get_string(&address) {
10249 Ok(s) => s.into(),
10250 Err(e) => {
10251 #[cfg(target_os = "android")]
10252 android_log(&format!("connectPeerJni: Failed to get address: {:?}", e));
10253 return 0;
10254 }
10255 };
10256
10257 #[cfg(target_os = "android")]
10258 android_log(&format!(
10259 "connectPeerJni: Connecting to node={}, addr={}",
10260 node_id_str, addr_str
10261 ));
10262
10263 let peer_info = PeerInfo {
10264 name: "quic-peer".to_string(),
10265 node_id: node_id_str,
10266 addresses: vec![addr_str],
10267 relay_url: None,
10268 };
10269
10270 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
10271 let result = match node.connect_peer(peer_info) {
10272 Ok(()) => {
10273 #[cfg(target_os = "android")]
10274 android_log("connectPeerJni: Connected successfully");
10275 1
10276 }
10277 Err(e) => {
10278 #[cfg(target_os = "android")]
10279 android_log(&format!("connectPeerJni: Failed to connect: {:?}", e));
10280 0
10281 }
10282 };
10283
10284 std::mem::forget(node);
10285 result
10286}
10287
10288// =============================================================================
10289// Document Change Subscription (direct JNI path)
10290// =============================================================================
10291//
10292// This is the push-based equivalent of the UniFFI PeatNode::subscribe() API.
10293// We can't use UniFFI's version from Android plugin consumers because UniFFI
10294// 0.28's Kotlin backend generates callback interfaces that inherit from
10295// com.sun.jna.Callback, and JNA's function-pointer resolution fails under
10296// Android plugin-host linker namespace isolation (see the comment block at
10297// the top of the JNI Bindings section and ADR-059 for full context).
10298//
10299// The direct-JNI path uses the same JAVA_VM + GlobalRef + attach_current_thread
10300// pattern that notify_peer_event already uses for peer connectivity events.
10301// Only one subscription is supported at a time.
10302
10303/// JNI: Subscribe to document change notifications
10304///
10305/// Kotlin signature:
10306/// `external fun subscribeDocumentChangesJni(handle: Long, listener:
10307/// DocumentChangeListener): Boolean`
10308///
10309/// The listener receives `onChange(collection, docId)` for every document
10310/// upsert and `onError(message)` if the underlying broadcast channel lags or
10311/// closes. Calls from the Rust side happen on the tokio runtime thread owned by
10312/// the PeatNode; the listener must be safe to invoke from any thread (consumers
10313/// typically post back to a main-thread Handler before touching UI state).
10314///
10315/// Replacing an existing subscription is allowed: the previous listener's
10316/// GlobalRef is dropped and the new one takes over on the next event.
10317#[cfg(feature = "sync")]
10318#[no_mangle]
10319pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_subscribeDocumentChangesJni(
10320 mut env: JNIEnv,
10321 _class: JClass,
10322 handle: i64,
10323 listener: jni::objects::JObject,
10324) -> jboolean {
10325 use std::sync::atomic::Ordering;
10326
10327 if handle == 0 {
10328 #[cfg(target_os = "android")]
10329 android_log("subscribeDocumentChangesJni: Invalid handle (0)");
10330 return 0;
10331 }
10332
10333 // Stash the listener as a global reference so it survives across JNI
10334 // thread attaches and isn't GC'd out from under us.
10335 let listener_global = match env.new_global_ref(&listener) {
10336 Ok(g) => g,
10337 Err(e) => {
10338 #[cfg(target_os = "android")]
10339 android_log(&format!(
10340 "subscribeDocumentChangesJni: new_global_ref failed: {:?}",
10341 e
10342 ));
10343 return 0;
10344 }
10345 };
10346
10347 // Swap the listener in; drop any previous one.
10348 {
10349 let mut slot = DOCUMENT_CHANGE_LISTENER.lock().unwrap();
10350 *slot = Some(listener_global);
10351 }
10352
10353 // Signal the previous subscription task (if any) to exit before we start
10354 // a new one, then mark the new subscription active.
10355 DOCUMENT_SUBSCRIPTION_ACTIVE.store(false, Ordering::SeqCst);
10356 DOCUMENT_SUBSCRIPTION_ACTIVE.store(true, Ordering::SeqCst);
10357
10358 // Borrow the node without taking ownership of its Arc.
10359 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
10360 let store = Arc::clone(&node.store);
10361 let runtime = Arc::clone(&node.runtime);
10362 std::mem::forget(node);
10363
10364 runtime.spawn(async move {
10365 let mut rx = store.subscribe_to_changes();
10366 while DOCUMENT_SUBSCRIPTION_ACTIVE.load(Ordering::SeqCst) {
10367 tokio::select! {
10368 result = rx.recv() => {
10369 match result {
10370 Ok(doc_key) => {
10371 let (collection, doc_id) = doc_key
10372 .split_once(':')
10373 .map(|(c, d)| (c.to_string(), d.to_string()))
10374 .unwrap_or_else(|| ("default".to_string(), doc_key.clone()));
10375 dispatch_document_change(&collection, &doc_id);
10376 }
10377 Err(tokio::sync::broadcast::error::RecvError::Lagged(n)) => {
10378 dispatch_document_error(&format!("lagged {} messages", n));
10379 }
10380 Err(tokio::sync::broadcast::error::RecvError::Closed) => {
10381 dispatch_document_error("change channel closed");
10382 break;
10383 }
10384 }
10385 }
10386 _ = tokio::time::sleep(tokio::time::Duration::from_millis(200)) => {
10387 // Periodic wake so we notice unsubscribe requests even
10388 // when the broadcast channel is quiet.
10389 }
10390 }
10391 }
10392
10393 // On exit, drop the listener ref if we were the owning subscription.
10394 if !DOCUMENT_SUBSCRIPTION_ACTIVE.load(Ordering::SeqCst) {
10395 let mut slot = DOCUMENT_CHANGE_LISTENER.lock().unwrap();
10396 *slot = None;
10397 }
10398 });
10399
10400 1 // JNI_TRUE
10401}
10402
10403/// JNI: Unsubscribe from document change notifications
10404///
10405/// Kotlin signature: `external fun unsubscribeDocumentChangesJni()`
10406///
10407/// Signals the background subscription task to exit on its next iteration.
10408/// The listener GlobalRef is dropped by the task on exit (not here) to avoid
10409/// a race between unsubscribe and an in-flight dispatch.
10410#[cfg(feature = "sync")]
10411#[no_mangle]
10412pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_unsubscribeDocumentChangesJni(
10413 _env: JNIEnv,
10414 _class: JClass,
10415) {
10416 use std::sync::atomic::Ordering;
10417 DOCUMENT_SUBSCRIPTION_ACTIVE.store(false, Ordering::SeqCst);
10418 #[cfg(target_os = "android")]
10419 android_log("unsubscribeDocumentChangesJni: subscription marked inactive");
10420}
10421
10422/// Snapshot the listener `GlobalRef` from a static slot under its mutex,
10423/// returning a clone that the caller can use without holding the lock.
10424///
10425/// Pulling the lock-acquire/clone/drop dance into a helper keeps every
10426/// dispatch helper above honest about not holding a listener lock across a
10427/// re-entrant JNI `call_method` (QA #808 IDIOM).
10428#[cfg(feature = "sync")]
10429fn clone_listener(slot: &Mutex<Option<GlobalRef>>) -> Option<GlobalRef> {
10430 slot.lock().ok()?.as_ref().cloned()
10431}
10432
10433/// Reconstruct a process-global `JavaVM` from `JAVA_VM` without holding the
10434/// mutex past the read. The underlying pointer is stable for the JVM
10435/// lifetime, so dropping the lock and re-wrapping is safe — and it lets
10436/// JNI calls in dispatch helpers proceed without serializing on `JAVA_VM`.
10437#[cfg(feature = "sync")]
10438fn clone_java_vm() -> Option<jni::JavaVM> {
10439 let raw_ptr = {
10440 let guard = JAVA_VM.lock().ok()?;
10441 guard.as_ref()?.get_java_vm_pointer()
10442 };
10443 // SAFETY: JNI_OnLoad seeded JAVA_VM via `JavaVM::from_raw`, so the
10444 // pointer points at a live `sys::JavaVM` for the rest of the process.
10445 // `JavaVM` has no `Drop` impl — wrapping the same pointer twice does
10446 // not double-free.
10447 unsafe { jni::JavaVM::from_raw(raw_ptr) }.ok()
10448}
10449
10450/// Dispatch a document-change event to the registered Kotlin listener.
10451/// Attaches the current tokio worker thread to the JVM if needed.
10452#[cfg(feature = "sync")]
10453fn dispatch_document_change(collection: &str, doc_id: &str) {
10454 // Snapshot the listener and JavaVM pointer under their locks, then drop
10455 // the guards BEFORE the unbounded JNI `call_method` (QA #808 IDIOM).
10456 // Kotlin's `onChange` may re-enter Rust JNI; holding either lock across
10457 // the call would deadlock the listener slot (re-entrant lock) or
10458 // serialize every translator's dispatch through a single JVM call.
10459 // GlobalRef is Arc-shaped so cloning is cheap; JavaVM is process-stable
10460 // so reconstructing from the raw pointer is sound.
10461 let Some(listener) = clone_listener(&DOCUMENT_CHANGE_LISTENER) else {
10462 return;
10463 };
10464 let Some(java_vm) = clone_java_vm() else {
10465 return;
10466 };
10467
10468 let mut env = match java_vm.attach_current_thread() {
10469 Ok(e) => e,
10470 Err(e) => {
10471 #[cfg(target_os = "android")]
10472 android_log(&format!("dispatch_document_change: attach failed: {:?}", e));
10473 let _ = e;
10474 return;
10475 }
10476 };
10477
10478 let collection_jstr = match env.new_string(collection) {
10479 Ok(s) => s,
10480 Err(_) => return,
10481 };
10482 let doc_id_jstr = match env.new_string(doc_id) {
10483 Ok(s) => s,
10484 Err(_) => return,
10485 };
10486
10487 if let Err(e) = env.call_method(
10488 &listener,
10489 "onChange",
10490 "(Ljava/lang/String;Ljava/lang/String;)V",
10491 &[
10492 JValue::Object(&collection_jstr),
10493 JValue::Object(&doc_id_jstr),
10494 ],
10495 ) {
10496 #[cfg(target_os = "android")]
10497 android_log(&format!(
10498 "dispatch_document_change: call_method failed: {:?}",
10499 e
10500 ));
10501 let _ = e;
10502 let _ = env.exception_describe();
10503 let _ = env.exception_clear();
10504 }
10505}
10506
10507/// Dispatch an error message to the registered Kotlin listener.
10508#[cfg(feature = "sync")]
10509fn dispatch_document_error(message: &str) {
10510 // Snapshot then drop locks before JNI work — see dispatch_document_change.
10511 let Some(listener) = clone_listener(&DOCUMENT_CHANGE_LISTENER) else {
10512 return;
10513 };
10514 let Some(java_vm) = clone_java_vm() else {
10515 return;
10516 };
10517
10518 let mut env = match java_vm.attach_current_thread() {
10519 Ok(e) => e,
10520 Err(_) => return,
10521 };
10522
10523 let msg_jstr = match env.new_string(message) {
10524 Ok(s) => s,
10525 Err(_) => return,
10526 };
10527
10528 if let Err(e) = env.call_method(
10529 &listener,
10530 "onError",
10531 "(Ljava/lang/String;)V",
10532 &[JValue::Object(&msg_jstr)],
10533 ) {
10534 #[cfg(target_os = "android")]
10535 android_log(&format!(
10536 "dispatch_document_error: call_method failed: {:?}",
10537 e
10538 ));
10539 let _ = e;
10540 let _ = env.exception_describe();
10541 let _ = env.exception_clear();
10542 }
10543}
10544
10545// =============================================================================
10546// Outbound-frame poll API — dart:ffi / non-JNI consumers (ADR-059 Slice 1.b)
10547// =============================================================================
10548//
10549// Exposes the same BLE translator fan-out as `subscribeOutboundFramesJni` but
10550// via a queue-drain pattern instead of a foreign callback. The host calls
10551// `start_outbound_frames` once, then polls `poll_outbound_frames` at its own
10552// pace (e.g. from a Dart isolate loop), and calls `stop_outbound_frames` on
10553// teardown. Explicit stop avoids the Drop-drives-async problem that deferred
10554// the original `OutboundFrameCallback` UniFFI trait registration.
10555//
10556// The inbound direction (`ingest_inbound_frame`) accepts postcard-encoded
10557// typed BLE structs (i.e. the bytes *after* peat-btle has stripped the GATT
10558// framing and decrypted the envelope) and publishes the resulting document
10559// with `Some("ble")` origin so ADR-059 echo-suppression fires correctly.
10560
10561/// `OutboundSink` that appends encoded frames to an in-process queue instead
10562/// of dispatching to a JNI callback. Used by `start_outbound_frames`.
10563#[cfg(all(feature = "sync", feature = "bluetooth"))]
10564struct QueueOutboundSink {
10565 transport_id: &'static str,
10566 queue: Arc<std::sync::Mutex<std::collections::VecDeque<OutboundFrame>>>,
10567}
10568
10569#[cfg(all(feature = "sync", feature = "bluetooth"))]
10570#[async_trait::async_trait]
10571impl peat_mesh::transport::OutboundSink for QueueOutboundSink {
10572 async fn send_outbound(
10573 &self,
10574 bytes: Vec<u8>,
10575 ctx: &peat_mesh::transport::TranslationContext,
10576 ) -> anyhow::Result<()> {
10577 let collection = ctx.collection.clone().unwrap_or_default();
10578 self.queue
10579 .lock()
10580 .map_err(|e| anyhow::anyhow!("outbound_queue poisoned: {e}"))?
10581 .push_back(OutboundFrame {
10582 transport_id: self.transport_id.to_string(),
10583 collection,
10584 bytes,
10585 });
10586 Ok(())
10587 }
10588}
10589
10590/// Internal helper: registers the ble (and optionally ble-lite) translator +
10591/// sink pair with `TransportManager`, starts the fan-out, and returns the
10592/// `FanoutHandle`. On any failure, already-registered translators are rolled
10593/// back before the error propagates.
10594///
10595/// `sink_factory` is a closure that receives the `transport_id` string and
10596/// returns the `Arc<dyn OutboundSink>` to wire for that transport. Called
10597/// once for `"ble"` and, with `lite-bridge` on, once for `"ble-lite"`.
10598#[cfg(all(feature = "sync", feature = "bluetooth"))]
10599impl PeatNode {
10600 fn register_ble_fanout(
10601 &self,
10602 sink_factory: impl Fn(&'static str) -> Arc<dyn peat_mesh::transport::OutboundSink>,
10603 ) -> anyhow::Result<peat_mesh::transport::FanoutHandle> {
10604 let translator_dyn: Arc<dyn peat_mesh::transport::Translator> = self.ble_translator.clone();
10605 let ble_sink = sink_factory("ble");
10606
10607 let collections = vec![
10608 self.ble_translator.tracks_collection().to_string(),
10609 self.ble_translator.nodes_collection().to_string(),
10610 self.ble_translator.alerts_collection().to_string(),
10611 self.ble_translator.canned_messages_collection().to_string(),
10612 ];
10613
10614 #[cfg(feature = "lite-bridge")]
10615 let lite_bridge_translator_id = peat_mesh::transport::BLE_LITE_BRIDGE;
10616 #[cfg(feature = "lite-bridge")]
10617 let mut collections = collections;
10618 #[cfg(feature = "lite-bridge")]
10619 for c in LITE_BRIDGE_COLLECTIONS {
10620 // Dedup: `nodes` is already in the base list above. Pushing it again
10621 // would spawn a SECOND observer task for the same collection, and the
10622 // two race on the single-pop `pending_origins` map — one pops the
10623 // ble-lite origin (skips), the other pops `None` and re-fans the
10624 // ingested doc back out → the roster fan-out storm. One observer per
10625 // collection keeps ADR-059 echo-suppression intact.
10626 if !collections.iter().any(|existing| existing == c) {
10627 collections.push((*c).to_string());
10628 }
10629 }
10630 let collections = collections;
10631
10632 self.runtime.block_on(async {
10633 self.transport_manager
10634 .register_translator(
10635 translator_dyn,
10636 ble_sink,
10637 peat_mesh::transport::TranslatorRegistrationConfig::ble(),
10638 )
10639 .await?;
10640
10641 #[cfg(feature = "lite-bridge")]
10642 {
10643 let lite_translator: Arc<dyn peat_mesh::transport::Translator> = Arc::new(
10644 CollectionGatedLiteBridge::for_ble_with_collections(LITE_BRIDGE_COLLECTIONS),
10645 );
10646 let lite_sink = sink_factory(lite_bridge_translator_id);
10647 if let Err(e) = self
10648 .transport_manager
10649 .register_translator(
10650 lite_translator,
10651 lite_sink,
10652 peat_mesh::transport::TranslatorRegistrationConfig::ble(),
10653 )
10654 .await
10655 {
10656 let _ = self.transport_manager.unregister_translator("ble").await;
10657 return Err(e);
10658 }
10659 }
10660
10661 match self
10662 .transport_manager
10663 .start_fanout(Arc::clone(&self.node), collections)
10664 {
10665 Ok(handle) => Ok(handle),
10666 Err(e) => {
10667 #[cfg(feature = "lite-bridge")]
10668 {
10669 let _ = self
10670 .transport_manager
10671 .unregister_translator(lite_bridge_translator_id)
10672 .await;
10673 }
10674 let _ = self.transport_manager.unregister_translator("ble").await;
10675 Err(e)
10676 }
10677 }
10678 })
10679 }
10680
10681 /// Re-emit a freshly-ingested BLE frame onto the outbound queue so this
10682 /// node RELAYS it to its other BLE peers — multi-hop A->B->C.
10683 /// peat-mesh's fan-out already re-fans an ingested frame to OTHER
10684 /// transports (BLE->Wi-Fi/iroh) but suppresses same-transport BLE->BLE
10685 /// re-emit to avoid a broadcast loop; that suppression is exactly what
10686 /// strands an all-BLE follower. Re-emitting here closes that hop.
10687 /// Deduped by frame content with a short TTL so a relayed frame isn't
10688 /// re-broadcast in a loop: a NEW value (different bytes)
10689 /// relays immediately, while identical re-advertises inside the TTL window
10690 /// are dropped (this is what keeps it from recreating the storm the
10691 /// suppression was guarding against). No-op unless an outbound subscription
10692 /// is actively draining the queue, so an idle node doesn't grow it.
10693 fn relay_ble_frame(&self, transport_id: &str, collection: &str, bytes: &[u8]) {
10694 use std::collections::hash_map::DefaultHasher;
10695 use std::hash::{Hash, Hasher};
10696 use std::time::{Duration, Instant};
10697
10698 const RELAY_DEDUP_TTL: Duration = Duration::from_secs(5);
10699 const RELAY_SEEN_CAP: usize = 2048;
10700
10701 // Do NOT relay presence ("nodes"): its heartbeat timestamp changes every
10702 // beat, so every frame is unique and escapes the content dedup — relaying
10703 // it ~Nx-amplifies BLE traffic (congestion → missed heartbeats → roster
10704 // liveness flapping) AND re-broadcasts stale node-identity docs from
10705 // peers' stores (resurfacing zombie identities, so the roster flips
10706 // between a node's old id and its callsign). Presence reaches direct
10707 // neighbours via each node's own advertise; only app STATE needs
10708 // multi-hop relay (counter "demo", "cells", "mission", "commands",
10709 // "markers"), and those re-advertise identical bytes so the dedup
10710 // throttles them to one relay per change.
10711 if collection == "nodes" {
10712 return;
10713 }
10714
10715 let active = match self.outbound_fanout.lock() {
10716 Ok(g) => g.is_some(),
10717 Err(e) => e.into_inner().is_some(),
10718 };
10719 if !active {
10720 return;
10721 }
10722
10723 let mut h = DefaultHasher::new();
10724 transport_id.hash(&mut h);
10725 collection.hash(&mut h);
10726 bytes.hash(&mut h);
10727 let key = h.finish();
10728
10729 let now = Instant::now();
10730 {
10731 let mut seen = self.relay_seen.lock().unwrap_or_else(|e| e.into_inner());
10732 seen.retain(|_, t| now.duration_since(*t) < RELAY_DEDUP_TTL);
10733 if seen.contains_key(&key) {
10734 return; // identical frame relayed recently — drop to break
10735 // loops
10736 }
10737 if seen.len() >= RELAY_SEEN_CAP {
10738 seen.clear();
10739 }
10740 seen.insert(key, now);
10741 }
10742
10743 self.outbound_queue
10744 .lock()
10745 .unwrap_or_else(|e| e.into_inner())
10746 .push_back(OutboundFrame {
10747 transport_id: transport_id.to_string(),
10748 collection: collection.to_string(),
10749 bytes: bytes.to_vec(),
10750 });
10751 }
10752}
10753
10754#[cfg(all(feature = "sync", feature = "bluetooth"))]
10755#[uniffi::export]
10756impl PeatNode {
10757 /// Subscribe to outbound BLE frames via a poll queue.
10758 ///
10759 /// After calling this, encoded frames produced by the `BleTranslator`
10760 /// fan-out accumulate in an internal unbounded queue. Call
10761 /// [`poll_outbound_frames`] frequently to drain it — if the consumer
10762 /// pauses polling the queue will grow without bound, one `Vec<u8>`
10763 /// payload per BLE frame.
10764 ///
10765 /// Idempotent — a second call while already subscribed is a no-op
10766 /// (returns `Ok`).
10767 ///
10768 /// Call [`stop_outbound_frames`] to unsubscribe, tear down the fan-out,
10769 /// and clear any residual frames from the queue.
10770 pub fn start_outbound_frames(&self) -> Result<(), PeatError> {
10771 {
10772 let guard = self
10773 .outbound_fanout
10774 .lock()
10775 .map_err(|_| PeatError::SyncError {
10776 msg: "outbound_fanout poisoned".to_string(),
10777 })?;
10778 if guard.is_some() {
10779 return Ok(()); // already running
10780 }
10781 }
10782 let queue = Arc::clone(&self.outbound_queue);
10783 let handle = self
10784 .register_ble_fanout(move |tid| {
10785 Arc::new(QueueOutboundSink {
10786 transport_id: tid,
10787 queue: Arc::clone(&queue),
10788 })
10789 })
10790 .map_err(|e| PeatError::SyncError { msg: e.to_string() })?;
10791 *self
10792 .outbound_fanout
10793 .lock()
10794 .map_err(|_| PeatError::SyncError {
10795 msg: "outbound_fanout poisoned".to_string(),
10796 })? = Some(handle);
10797 Ok(())
10798 }
10799
10800 /// Drain all queued outbound frames produced since the last call.
10801 ///
10802 /// Returns an empty `Vec` when no frames are pending or when
10803 /// [`start_outbound_frames`] has not been called. Non-blocking.
10804 pub fn poll_outbound_frames(&self) -> Vec<OutboundFrame> {
10805 // If the Mutex is poisoned (a thread panicked while holding it) we
10806 // recover the inner value rather than propagating a panic — the
10807 // VecDeque state is consistent enough to drain safely.
10808 let mut q = self
10809 .outbound_queue
10810 .lock()
10811 .unwrap_or_else(|e| e.into_inner());
10812 q.drain(..).collect()
10813 }
10814
10815 /// Stop outbound-frame delivery and tear down the BLE fan-out.
10816 ///
10817 /// Drops the `FanoutHandle` (cancels observer tasks), unregisters the BLE
10818 /// translator(s), and clears the outbound queue so that stale frames are
10819 /// not delivered after a subsequent [`start_outbound_frames`].
10820 ///
10821 /// Idempotent — safe to call when not subscribed.
10822 pub fn stop_outbound_frames(&self) {
10823 let handle = self
10824 .outbound_fanout
10825 .lock()
10826 .unwrap_or_else(|e| e.into_inner())
10827 .take();
10828 drop(handle); // cancels fan-out observer tasks
10829
10830 // Clear residual frames so a subsequent start_outbound_frames sees a
10831 // clean queue rather than frames from the previous subscription window.
10832 self.outbound_queue
10833 .lock()
10834 .unwrap_or_else(|e| e.into_inner())
10835 .clear();
10836
10837 // Unregister the translator(s) so a future start_outbound_frames
10838 // can re-register without hitting the duplicate-id rejection.
10839 self.runtime.block_on(async {
10840 #[cfg(feature = "lite-bridge")]
10841 {
10842 let _ = self
10843 .transport_manager
10844 .unregister_translator(peat_mesh::transport::BLE_LITE_BRIDGE)
10845 .await;
10846 }
10847 let _ = self.transport_manager.unregister_translator("ble").await;
10848 });
10849 }
10850
10851 /// Feed a BLE inbound frame into the mesh.
10852 ///
10853 /// `postcard_bytes` must be the postcard-encoded typed BLE struct
10854 /// produced by `peat-btle` *after* it has stripped the GATT framing and
10855 /// decrypted the envelope (i.e. the bytes `peat-btle` would pass to its
10856 /// internal `Translator::decode_inbound`).
10857 ///
10858 /// `collection` must name the document collection the bytes belong to
10859 /// (e.g. `"tracks"`, `"platforms"`) — peat-btle knows this from the GATT
10860 /// characteristic or frame type and should pass it through unchanged.
10861 ///
10862 /// On success returns the newly-published document ID. Returns `Ok(None)`
10863 /// if the bytes are addressed to an unknown collection (graceful decline).
10864 pub fn ingest_inbound_frame(
10865 &self,
10866 collection: String,
10867 postcard_bytes: Vec<u8>,
10868 ) -> Result<Option<String>, PeatError> {
10869 use peat_mesh::transport::{TranslationContext, Translator};
10870 let ctx = TranslationContext::inbound("ble").with_collection(collection);
10871 let doc = self
10872 .runtime
10873 .block_on(self.ble_translator.decode_inbound(&postcard_bytes, &ctx))
10874 .map_err(|e| PeatError::SyncError { msg: e.to_string() })?;
10875 let Some(mesh_doc) = doc else {
10876 return Ok(None);
10877 };
10878 let collection_name = ctx.collection.unwrap_or_default();
10879 let id = self
10880 .runtime
10881 .block_on(self.node.publish_with_origin(
10882 &collection_name,
10883 mesh_doc,
10884 Some("ble".to_string()),
10885 ))
10886 .map_err(|e| PeatError::SyncError { msg: e.to_string() })?;
10887 // Multi-hop: relay this frame to our other BLE peers (deduped).
10888 self.relay_ble_frame("ble", &collection_name, &postcard_bytes);
10889 Ok(Some(id.to_string()))
10890 }
10891
10892 /// Publish a JSON document through the **node layer** — the same path the
10893 /// Android `publishDocumentJni` uses — so the write reaches the ADR-059
10894 /// fan-out and is emitted over the bridged transports (BLE/Wi-Fi). The
10895 /// `id` field in the JSON, when present, becomes the document id
10896 /// (returned).
10897 ///
10898 /// Use this instead of `put_document` when the write must propagate to
10899 /// peers via the bridged radios: `put_document`/`put_node` write straight
10900 /// to `storage_backend`, which the fan-out does not observe, so those never
10901 /// emit a BLE frame. Needed by the iOS bridge (which drives the poll API
10902 /// from Dart and has no JNI `publishDocumentJni`).
10903 #[cfg(feature = "sync")]
10904 pub fn publish_document(&self, collection: String, json: String) -> Result<String, PeatError> {
10905 self.runtime
10906 .block_on(publish_document_into_node(&self.node, &collection, &json))
10907 .map_err(|e| PeatError::SyncError { msg: e.to_string() })
10908 }
10909}
10910
10911// `ingest_inbound_lite_frame` lives in its OWN cfg-gated `#[uniffi::export]`
10912// block (not the `all(sync, bluetooth)` block above) so that under
10913// `sync,bluetooth` WITHOUT `lite-bridge` the whole export — including the
10914// generated scaffolding's call to the method — is stripped before the macro
10915// runs. With a per-method `#[cfg(lite-bridge)]` inside the broader block, the
10916// export macro still emitted a call to the cfg'd-out method (E0599). See peat#986.
10917#[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
10918#[uniffi::export]
10919impl PeatNode {
10920 /// Ingest an inbound BLE frame that arrived on the universal-Document
10921 /// (peat-lite / `ble-lite`) codec, as opposed to the typed 0xB6 path in
10922 /// [`ingest_inbound_frame`]. Decodes via the `CollectionGatedLiteBridge`
10923 /// and republishes with `Some("ble-lite")` origin so the mesh re-fans it
10924 /// to the other transports without looping back to BLE. Used for raw
10925 /// collections (e.g. the `demo` counter) that the typed translator
10926 /// declines.
10927 pub fn ingest_inbound_lite_frame(
10928 &self,
10929 collection: String,
10930 envelope_bytes: Vec<u8>,
10931 ) -> Result<Option<String>, PeatError> {
10932 use peat_mesh::transport::{TranslationContext, Translator, BLE_LITE_BRIDGE};
10933 let bridge = CollectionGatedLiteBridge::for_ble_with_collections(LITE_BRIDGE_COLLECTIONS);
10934 let ctx = TranslationContext::inbound(BLE_LITE_BRIDGE).with_collection(collection);
10935 let doc = self
10936 .runtime
10937 .block_on(bridge.decode_inbound(&envelope_bytes, &ctx))
10938 .map_err(|e| PeatError::SyncError { msg: e.to_string() })?;
10939 let Some(mesh_doc) = doc else {
10940 return Ok(None);
10941 };
10942 let collection_name = ctx.collection.unwrap_or_default();
10943 let id = self
10944 .runtime
10945 .block_on(self.node.publish_with_origin(
10946 &collection_name,
10947 mesh_doc,
10948 Some(BLE_LITE_BRIDGE.to_string()),
10949 ))
10950 .map_err(|e| PeatError::SyncError { msg: e.to_string() })?;
10951 // Multi-hop: relay this lite frame to our other BLE peers (deduped).
10952 self.relay_ble_frame(BLE_LITE_BRIDGE, &collection_name, &envelope_bytes);
10953 Ok(Some(id.to_string()))
10954 }
10955}
10956
10957// =============================================================================
10958// OutboundFrameCallback JNI (ADR-059 Slice 1.b)
10959// =============================================================================
10960//
10961// Bridges `TransportManager`'s per-transport fan-out (peat-mesh) into a
10962// Kotlin callback so a consumer plugin's BLE manager can deliver encoded
10963// frames over the radio. The JNI shape mirrors `subscribeDocumentChangesJni`
10964// — a single GlobalRef in a static slot, replaceable on re-subscribe — so
10965// the same patterns audited on PR #803 carry over.
10966
10967/// `OutboundSink` implementation that forwards encoded bytes into the
10968/// registered Kotlin listener. One instance is registered with
10969/// `TransportManager` per `transport_id` we want to fan out — currently
10970/// `"ble"` for typed 0xB6 frames and (with `lite-bridge` on) `"ble-lite"`
10971/// for universal Document envelopes. The structure generalizes to
10972/// LoRa/SBD/etc.
10973#[cfg(all(feature = "sync", feature = "bluetooth"))]
10974struct JniOutboundSink {
10975 transport_id: &'static str,
10976}
10977
10978/// `Translator` wrapper that gates `encode_outbound` by collection.
10979/// Wraps a [`peat_mesh::transport::LiteBridgeTranslator`] (catch-all
10980/// codec — encodes any collection it's handed) with a peat-ffi-policy
10981/// allow-list, so the universal-Document fan-out only fires for
10982/// collections explicitly opted in.
10983///
10984/// Without this wrapper, registering both the typed `BleTranslator`
10985/// (which encodes `"tracks"`/`"nodes"`/`"alerts"`/`"canned_messages"`
10986/// to compact 0xB6 frames) AND the catch-all `LiteBridgeTranslator` on
10987/// the same `TransportManager` would cause **double emission** for the
10988/// typed collections — both translators would encode the same doc and
10989/// dispatch separate frames to Kotlin. The plugin would receive
10990/// duplicate copies, and BLE-link bandwidth doubles for no gain. The
10991/// gate stays in peat-ffi (the consumer that owns the policy decision)
10992/// rather than in `LiteBridgeTranslator` itself, matching ADR-059's
10993/// "policy lives at the consumer, codec is generic" direction.
10994///
10995/// Slice 2's per-doc `allowed_transports` will eventually replace this
10996/// with a runtime annotation on each Document; until then, the
10997/// peat-ffi-static allow-list is the right shape.
10998#[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
10999struct CollectionGatedLiteBridge {
11000 inner: peat_mesh::transport::LiteBridgeTranslator,
11001 allowed: std::collections::HashSet<&'static str>,
11002}
11003
11004#[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
11005impl CollectionGatedLiteBridge {
11006 fn for_ble_with_collections(collections: &'static [&'static str]) -> Self {
11007 Self {
11008 inner: peat_mesh::transport::LiteBridgeTranslator::for_ble(),
11009 allowed: collections.iter().copied().collect(),
11010 }
11011 }
11012}
11013
11014#[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
11015#[async_trait::async_trait]
11016impl peat_mesh::transport::Translator for CollectionGatedLiteBridge {
11017 fn transport_id(&self) -> &'static str {
11018 self.inner.transport_id()
11019 }
11020
11021 async fn encode_outbound(
11022 &self,
11023 doc: &peat_mesh::sync::types::Document,
11024 ctx: &peat_mesh::transport::TranslationContext,
11025 ) -> Option<Vec<u8>> {
11026 // Decline silently for collections outside the allow-list.
11027 // This is the policy filter, not a codec error — matches the
11028 // BleTranslator decline behaviour for unknown collections.
11029 let collection = ctx.collection.as_deref()?;
11030 if !self.allowed.contains(collection) {
11031 return None;
11032 }
11033 self.inner.encode_outbound(doc, ctx).await
11034 }
11035
11036 async fn decode_inbound(
11037 &self,
11038 bytes: &[u8],
11039 ctx: &peat_mesh::transport::TranslationContext,
11040 ) -> anyhow::Result<Option<peat_mesh::sync::types::Document>> {
11041 // Inbound is collection-agnostic at this codec level (the
11042 // envelope carries the collection). The receive-side policy
11043 // decision (which collections to publish_with_origin) lives
11044 // in the consumer (plugin Kotlin), so the gate doesn't apply
11045 // here.
11046 self.inner.decode_inbound(bytes, ctx).await
11047 }
11048}
11049
11050/// Universal-Document collections that ride the `"ble-lite"` codec
11051/// instead of the typed 0xB6 path. Add new entries here when a new
11052/// collection joins the universal transport (chats, alerts-v2, etc.).
11053/// Keep the list tight — every entry is one more codec the universal
11054/// path encodes for, and double-emission with the typed BleTranslator
11055/// would result if both lists overlap.
11056// `nodes` (capabilities/roster) rides the universal codec — the typed
11057// BleTranslator declines it (it only encodes tracks/platforms/alerts/
11058// canned_messages), so without this entry capabilities never reach a BLE
11059// frame and remote rosters stay empty. Safe to carry here now that
11060// `put_node` publishes through the node layer (same wrapped representation
11061// as the ingest), so the two sides converge instead of re-syncing forever.
11062#[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
11063const LITE_BRIDGE_COLLECTIONS: &[&str] =
11064 &["markers", "demo", "nodes", "mission", "cells", "commands"];
11065
11066#[cfg(all(feature = "sync", feature = "bluetooth"))]
11067#[async_trait::async_trait]
11068impl peat_mesh::transport::OutboundSink for JniOutboundSink {
11069 async fn send_outbound(
11070 &self,
11071 bytes: Vec<u8>,
11072 ctx: &peat_mesh::transport::TranslationContext,
11073 ) -> anyhow::Result<()> {
11074 let collection = ctx.collection.as_deref().unwrap_or("");
11075 dispatch_outbound_frame(self.transport_id, collection, &bytes);
11076 Ok(())
11077 }
11078}
11079
11080/// JNI: Subscribe to outbound BLE-encoded frames produced by the
11081/// `BleTranslator` in `TransportManager`'s fan-out.
11082///
11083/// Kotlin signature:
11084/// `external fun subscribeOutboundFramesJni(handle: Long, listener:
11085/// OutboundFrameListener): Boolean`
11086///
11087/// The listener receives `onFrame(transportId, collection, bytes)` for
11088/// each encoded document the translator produces. Calls fire on the
11089/// tokio runtime thread; the listener must tolerate any-thread invocation
11090/// (the plugin posts to its own executor before touching radio state).
11091///
11092/// **Idempotent**: a second call replaces the listener `GlobalRef`; the
11093/// underlying translator + sink registration and observer fan-out tasks
11094/// are kept alive across the swap so no frames are lost between the two
11095/// listeners. Use `unsubscribeOutboundFramesJni` to fully tear down.
11096#[cfg(all(feature = "sync", feature = "bluetooth"))]
11097#[no_mangle]
11098pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_subscribeOutboundFramesJni(
11099 mut env: JNIEnv,
11100 _class: JClass,
11101 handle: i64,
11102 listener: jni::objects::JObject,
11103) -> jboolean {
11104 if handle == 0 {
11105 #[cfg(target_os = "android")]
11106 android_log("subscribeOutboundFramesJni: Invalid handle (0)");
11107 return 0;
11108 }
11109
11110 let listener_global = match env.new_global_ref(&listener) {
11111 Ok(g) => g,
11112 Err(e) => {
11113 #[cfg(target_os = "android")]
11114 android_log(&format!(
11115 "subscribeOutboundFramesJni: new_global_ref failed: {:?}",
11116 e
11117 ));
11118 let _ = e;
11119 return 0;
11120 }
11121 };
11122
11123 // Listener swap is unconditional — second-subscribe just rebinds.
11124 *OUTBOUND_FRAME_LISTENER.lock().unwrap() = Some(listener_global);
11125
11126 // If a fan-out is already running, the swap above is sufficient — the
11127 // existing JniOutboundSink reads the listener slot dynamically.
11128 {
11129 let handle_slot = OUTBOUND_FRAME_FANOUT.lock().unwrap();
11130 if handle_slot.is_some() {
11131 return 1;
11132 }
11133 }
11134
11135 // First subscribe: register translator + sink and start fan-out.
11136 // `TransportManager` is not Clone, so we hold the `node_owner` Arc by
11137 // borrow (not by taking ownership) for the duration of the call;
11138 // forget happens after the registration block completes.
11139 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
11140
11141 // Delegate to the shared registration helper so the JNI and the
11142 // poll-API paths stay aligned. The factory produces a `JniOutboundSink`
11143 // whose `send_outbound` dispatches to the registered Kotlin GlobalRef.
11144 let final_result =
11145 node_owner.register_ble_fanout(|tid| Arc::new(JniOutboundSink { transport_id: tid }));
11146
11147 std::mem::forget(node_owner);
11148
11149 match final_result {
11150 Ok(fanout_handle) => {
11151 *OUTBOUND_FRAME_FANOUT.lock().unwrap() = Some(fanout_handle);
11152 1
11153 }
11154 Err(_e) => {
11155 // Roll back the listener stash so a future retry isn't observed
11156 // as "already subscribed".
11157 *OUTBOUND_FRAME_LISTENER.lock().unwrap() = None;
11158 #[cfg(target_os = "android")]
11159 android_log(&format!(
11160 "subscribeOutboundFramesJni: register/start_fanout failed: {}",
11161 _e
11162 ));
11163 0
11164 }
11165 }
11166}
11167
11168/// JNI: Unsubscribe from outbound frame delivery.
11169///
11170/// Kotlin signature: `external fun unsubscribeOutboundFramesJni(handle: Long)`
11171///
11172/// Drops the `FanoutHandle` (cancelling observer tasks), unregisters the
11173/// translator, and clears the listener `GlobalRef`. Idempotent — calling
11174/// twice or before any subscribe is a no-op.
11175#[cfg(all(feature = "sync", feature = "bluetooth"))]
11176#[no_mangle]
11177pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_unsubscribeOutboundFramesJni(
11178 _env: JNIEnv,
11179 _class: JClass,
11180 handle: i64,
11181) {
11182 // Drop the FanoutHandle first so no further frames are fanned out
11183 // while we're tearing down.
11184 let _ = OUTBOUND_FRAME_FANOUT.lock().unwrap().take();
11185
11186 if handle != 0 {
11187 let node_owner = unsafe { Arc::from_raw(handle as *const PeatNode) };
11188 node_owner.runtime.block_on(async {
11189 // Unregister both translators that the lite-bridge build
11190 // registered (ble + ble-lite). Each call independently
11191 // rejects "translator not registered", so the order doesn't
11192 // matter and a missing entry on either side is benign.
11193 #[cfg(feature = "lite-bridge")]
11194 {
11195 let _ = node_owner
11196 .transport_manager
11197 .unregister_translator(peat_mesh::transport::BLE_LITE_BRIDGE)
11198 .await;
11199 }
11200 let _ = node_owner
11201 .transport_manager
11202 .unregister_translator("ble")
11203 .await;
11204 });
11205 std::mem::forget(node_owner);
11206 }
11207
11208 *OUTBOUND_FRAME_LISTENER.lock().unwrap() = None;
11209
11210 #[cfg(target_os = "android")]
11211 android_log("unsubscribeOutboundFramesJni: subscription torn down");
11212}
11213
11214/// Dispatch an outbound frame to the registered Kotlin listener.
11215/// Attaches the current tokio worker thread to the JVM if needed.
11216#[cfg(all(feature = "sync", feature = "bluetooth"))]
11217fn dispatch_outbound_frame(transport_id: &str, collection: &str, bytes: &[u8]) {
11218 // Snapshot then drop locks before JNI work — see dispatch_document_change.
11219 let Some(listener) = clone_listener(&OUTBOUND_FRAME_LISTENER) else {
11220 return;
11221 };
11222 let Some(java_vm) = clone_java_vm() else {
11223 return;
11224 };
11225
11226 let mut env = match java_vm.attach_current_thread() {
11227 Ok(e) => e,
11228 Err(e) => {
11229 #[cfg(target_os = "android")]
11230 android_log(&format!("dispatch_outbound_frame: attach failed: {:?}", e));
11231 let _ = e;
11232 return;
11233 }
11234 };
11235
11236 let transport_jstr = match env.new_string(transport_id) {
11237 Ok(s) => s,
11238 Err(_) => return,
11239 };
11240 let collection_jstr = match env.new_string(collection) {
11241 Ok(s) => s,
11242 Err(_) => return,
11243 };
11244 let bytes_jarr = match env.byte_array_from_slice(bytes) {
11245 Ok(a) => a,
11246 Err(e) => {
11247 #[cfg(target_os = "android")]
11248 android_log(&format!(
11249 "dispatch_outbound_frame: byte_array_from_slice failed: {:?}",
11250 e
11251 ));
11252 let _ = e;
11253 return;
11254 }
11255 };
11256
11257 if let Err(e) = env.call_method(
11258 &listener,
11259 "onFrame",
11260 "(Ljava/lang/String;Ljava/lang/String;[B)V",
11261 &[
11262 JValue::Object(&transport_jstr),
11263 JValue::Object(&collection_jstr),
11264 JValue::Object(&bytes_jarr),
11265 ],
11266 ) {
11267 #[cfg(target_os = "android")]
11268 android_log(&format!(
11269 "dispatch_outbound_frame: call_method failed: {:?}",
11270 e
11271 ));
11272 let _ = e;
11273 let _ = env.exception_describe();
11274 let _ = env.exception_clear();
11275 }
11276}
11277
11278// =============================================================================
11279// Blob Transfer JNI (ADR-060)
11280// =============================================================================
11281
11282/// JNI: Enable blob transfer on the PeatNode.
11283///
11284/// Kotlin signature:
11285/// `external fun enableBlobTransferJni(handle: Long, bindAddr: String?):
11286/// Boolean`
11287#[cfg(feature = "sync")]
11288#[no_mangle]
11289pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_enableBlobTransferJni(
11290 mut env: JNIEnv,
11291 _class: JClass,
11292 handle: i64,
11293 bind_addr: JString,
11294) -> jboolean {
11295 if handle == 0 {
11296 return 0;
11297 }
11298 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
11299
11300 let addr_str: Option<String> = if bind_addr.is_null() {
11301 None
11302 } else {
11303 env.get_string(&bind_addr).ok().map(|s| s.into())
11304 };
11305 let bind: Option<std::net::SocketAddr> =
11306 addr_str.and_then(|s| if s.is_empty() { None } else { s.parse().ok() });
11307
11308 let result = match node.enable_blob_transfer(bind) {
11309 Ok(()) => 1,
11310 Err(e) => {
11311 #[cfg(target_os = "android")]
11312 android_log(&format!("enableBlobTransferJni: {}", e));
11313 0
11314 }
11315 };
11316 std::mem::forget(node);
11317 result
11318}
11319
11320/// JNI: Add a known blob peer.
11321///
11322/// Kotlin signature:
11323/// `external fun blobAddPeerJni(handle: Long, peerIdHex: String, address:
11324/// String): Boolean`
11325#[cfg(feature = "sync")]
11326#[no_mangle]
11327pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_blobAddPeerJni(
11328 mut env: JNIEnv,
11329 _class: JClass,
11330 handle: i64,
11331 peer_id_hex: JString,
11332 address: JString,
11333) -> jboolean {
11334 if handle == 0 {
11335 return 0;
11336 }
11337 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
11338
11339 let peer_hex: String = match env.get_string(&peer_id_hex) {
11340 Ok(s) => s.into(),
11341 Err(_) => {
11342 std::mem::forget(node);
11343 return 0;
11344 }
11345 };
11346 let addr: String = match env.get_string(&address) {
11347 Ok(s) => s.into(),
11348 Err(_) => {
11349 std::mem::forget(node);
11350 return 0;
11351 }
11352 };
11353
11354 let result = match node.blob_add_peer(&peer_hex, &addr) {
11355 Ok(()) => 1,
11356 Err(e) => {
11357 #[cfg(target_os = "android")]
11358 android_log(&format!("blobAddPeerJni: {}", e));
11359 0
11360 }
11361 };
11362 std::mem::forget(node);
11363 result
11364}
11365
11366/// JNI: Store bytes as a blob. Returns the content hash as a hex string.
11367///
11368/// Kotlin signature:
11369/// `external fun blobPutJni(handle: Long, data: ByteArray, contentType:
11370/// String): String?`
11371#[cfg(feature = "sync")]
11372#[no_mangle]
11373pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_blobPutJni(
11374 mut env: JNIEnv,
11375 _class: JClass,
11376 handle: i64,
11377 data: jni::objects::JByteArray,
11378 content_type: JString,
11379) -> jstring {
11380 if handle == 0 {
11381 return std::ptr::null_mut();
11382 }
11383 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
11384
11385 let bytes = match env.convert_byte_array(&data) {
11386 Ok(b) => b,
11387 Err(_) => {
11388 std::mem::forget(node);
11389 return std::ptr::null_mut();
11390 }
11391 };
11392 let ct: String = match env.get_string(&content_type) {
11393 Ok(s) => s.into(),
11394 Err(_) => {
11395 std::mem::forget(node);
11396 return std::ptr::null_mut();
11397 }
11398 };
11399
11400 let result = match node.blob_put(&bytes, &ct) {
11401 Ok(hash) => env.new_string(&hash).ok().map(|s| s.into_raw()),
11402 Err(e) => {
11403 #[cfg(target_os = "android")]
11404 android_log(&format!("blobPutJni: {}", e));
11405 None
11406 }
11407 };
11408 std::mem::forget(node);
11409 result.unwrap_or(std::ptr::null_mut())
11410}
11411
11412/// JNI: Fetch blob bytes by hash. Returns byte[] or null.
11413///
11414/// Kotlin signature:
11415/// `external fun blobGetJni(handle: Long, hashHex: String): ByteArray?`
11416#[cfg(feature = "sync")]
11417#[no_mangle]
11418pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_blobGetJni(
11419 mut env: JNIEnv,
11420 _class: JClass,
11421 handle: i64,
11422 hash_hex: JString,
11423) -> jni::objects::JByteArray<'static> {
11424 if handle == 0 {
11425 return JByteArray::default();
11426 }
11427 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
11428
11429 let hash: String = match env.get_string(&hash_hex) {
11430 Ok(s) => s.into(),
11431 Err(_) => {
11432 std::mem::forget(node);
11433 return JByteArray::default();
11434 }
11435 };
11436
11437 let result = match node.blob_get(&hash) {
11438 Ok(bytes) => env.byte_array_from_slice(&bytes).ok(),
11439 Err(e) => {
11440 #[cfg(target_os = "android")]
11441 android_log(&format!("blobGetJni: {}", e));
11442 None
11443 }
11444 };
11445 std::mem::forget(node);
11446 // Safety: JByteArray has no lifetime on the default — transmute is needed
11447 // because the JNI return type doesn't carry a lifetime parameter.
11448 result
11449 .map(|arr| unsafe { std::mem::transmute(arr) })
11450 .unwrap_or(JByteArray::default())
11451}
11452
11453/// JNI: Check if blob exists locally.
11454///
11455/// Kotlin signature:
11456/// `external fun blobExistsLocallyJni(handle: Long, hashHex: String): Boolean`
11457#[cfg(feature = "sync")]
11458#[no_mangle]
11459pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_blobExistsLocallyJni(
11460 mut env: JNIEnv,
11461 _class: JClass,
11462 handle: i64,
11463 hash_hex: JString,
11464) -> jboolean {
11465 if handle == 0 {
11466 return 0;
11467 }
11468 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
11469
11470 let hash: String = match env.get_string(&hash_hex) {
11471 Ok(s) => s.into(),
11472 Err(_) => {
11473 std::mem::forget(node);
11474 return 0;
11475 }
11476 };
11477
11478 let result = if node.blob_exists_locally(&hash) {
11479 1
11480 } else {
11481 0
11482 };
11483 std::mem::forget(node);
11484 result
11485}
11486
11487/// JNI: Get blob endpoint ID as hex string (or null if blob transfer disabled).
11488///
11489/// Kotlin signature:
11490/// `external fun blobEndpointIdJni(handle: Long): String?`
11491#[cfg(feature = "sync")]
11492#[no_mangle]
11493pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_blobEndpointIdJni(
11494 mut env: JNIEnv,
11495 _class: JClass,
11496 handle: i64,
11497) -> jstring {
11498 if handle == 0 {
11499 return std::ptr::null_mut();
11500 }
11501 let node = unsafe { Arc::from_raw(handle as *const PeatNode) };
11502
11503 let result = match node.blob_endpoint_id() {
11504 Some(id) => env.new_string(&id).ok().map(|s| s.into_raw()),
11505 None => None,
11506 };
11507 std::mem::forget(node);
11508 result.unwrap_or(std::ptr::null_mut())
11509}
11510
11511// =============================================================================
11512// JNI Native Method Registration
11513// =============================================================================
11514//
11515// Android's linker namespace isolation prevents normal JNI symbol lookup.
11516// We provide a nativeInit function that Kotlin must call after System.load()
11517// to explicitly register the native methods.
11518
11519/// Register native methods for PeatJni class
11520///
11521/// This must be called from Kotlin after System.load() to register native
11522/// methods. Android's classloader isolation prevents JNI_OnLoad from finding
11523/// the class.
11524///
11525/// Kotlin usage:
11526/// ```kotlin
11527/// companion object {
11528/// init {
11529/// System.load(libPath)
11530/// nativeInit()
11531/// }
11532/// @JvmStatic external fun nativeInit()
11533/// }
11534/// ```
11535#[no_mangle]
11536pub extern "system" fn Java_com_defenseunicorns_peat_PeatJni_nativeInit(
11537 mut env: JNIEnv,
11538 class: JClass,
11539) {
11540 use jni::NativeMethod;
11541
11542 let methods: Vec<NativeMethod> = vec![
11543 NativeMethod {
11544 name: "peatVersion".into(),
11545 sig: "()Ljava/lang/String;".into(),
11546 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_peatVersion as *mut c_void,
11547 },
11548 NativeMethod {
11549 name: "testJni".into(),
11550 sig: "()Ljava/lang/String;".into(),
11551 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_testJni as *mut c_void,
11552 },
11553 #[cfg(target_os = "android")]
11554 NativeMethod {
11555 name: "setAndroidContextJni".into(),
11556 // (Ljava/lang/Object;)V — Kotlin `Any` lowers to java.lang.Object.
11557 sig: "(Ljava/lang/Object;)V".into(),
11558 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_setAndroidContextJni as *mut c_void,
11559 },
11560 #[cfg(target_os = "android")]
11561 NativeMethod {
11562 name: "verifyAndroidContextJni".into(),
11563 sig: "()Z".into(),
11564 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_verifyAndroidContextJni as *mut c_void,
11565 },
11566 #[cfg(feature = "sync")]
11567 NativeMethod {
11568 name: "createNodeJni".into(),
11569 sig: "(Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;)J".into(),
11570 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_createNodeJni as *mut c_void,
11571 },
11572 #[cfg(feature = "sync")]
11573 NativeMethod {
11574 name: "getGlobalNodeHandleJni".into(),
11575 sig: "()J".into(),
11576 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getGlobalNodeHandleJni as *mut c_void,
11577 },
11578 #[cfg(feature = "sync")]
11579 NativeMethod {
11580 name: "clearGlobalNodeHandleJni".into(),
11581 sig: "()V".into(),
11582 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_clearGlobalNodeHandleJni as *mut c_void,
11583 },
11584 #[cfg(feature = "sync")]
11585 NativeMethod {
11586 name: "nodeIdJni".into(),
11587 sig: "(J)Ljava/lang/String;".into(),
11588 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_nodeIdJni as *mut c_void,
11589 },
11590 #[cfg(feature = "sync")]
11591 NativeMethod {
11592 name: "peerCountJni".into(),
11593 sig: "(J)I".into(),
11594 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_peerCountJni as *mut c_void,
11595 },
11596 #[cfg(feature = "sync")]
11597 NativeMethod {
11598 name: "connectedPeersJni".into(),
11599 sig: "(J)Ljava/lang/String;".into(),
11600 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_connectedPeersJni as *mut c_void,
11601 },
11602 #[cfg(feature = "sync")]
11603 NativeMethod {
11604 name: "requestSyncJni".into(),
11605 sig: "(J)Z".into(),
11606 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_requestSyncJni as *mut c_void,
11607 },
11608 #[cfg(feature = "sync")]
11609 NativeMethod {
11610 name: "endpointSocketAddrJni".into(),
11611 sig: "(J)Ljava/lang/String;".into(),
11612 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_endpointSocketAddrJni as *mut c_void,
11613 },
11614 #[cfg(feature = "sync")]
11615 NativeMethod {
11616 name: "getDocumentJni".into(),
11617 sig: "(JLjava/lang/String;Ljava/lang/String;)Ljava/lang/String;".into(),
11618 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getDocumentJni as *mut c_void,
11619 },
11620 #[cfg(feature = "sync")]
11621 NativeMethod {
11622 name: "forceStoreErrorForTestingJni".into(),
11623 sig: "(J)Z".into(),
11624 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_forceStoreErrorForTestingJni
11625 as *mut c_void,
11626 },
11627 #[cfg(feature = "sync")]
11628 NativeMethod {
11629 name: "startSyncJni".into(),
11630 sig: "(J)Z".into(),
11631 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_startSyncJni as *mut c_void,
11632 },
11633 #[cfg(feature = "sync")]
11634 NativeMethod {
11635 name: "freeNodeJni".into(),
11636 sig: "(J)V".into(),
11637 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_freeNodeJni as *mut c_void,
11638 },
11639 #[cfg(feature = "sync")]
11640 NativeMethod {
11641 name: "getCellsJni".into(),
11642 sig: "(J)Ljava/lang/String;".into(),
11643 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getCellsJni as *mut c_void,
11644 },
11645 #[cfg(feature = "sync")]
11646 NativeMethod {
11647 name: "getTracksJni".into(),
11648 sig: "(J)Ljava/lang/String;".into(),
11649 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getTracksJni as *mut c_void,
11650 },
11651 #[cfg(feature = "sync")]
11652 NativeMethod {
11653 name: "getNodesJni".into(),
11654 sig: "(J)Ljava/lang/String;".into(),
11655 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getNodesJni as *mut c_void,
11656 },
11657 #[cfg(feature = "sync")]
11658 NativeMethod {
11659 name: "getCommandsJni".into(),
11660 sig: "(J)Ljava/lang/String;".into(),
11661 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getCommandsJni as *mut c_void,
11662 },
11663 #[cfg(feature = "sync")]
11664 NativeMethod {
11665 name: "publishNodeJni".into(),
11666 sig: "(JLjava/lang/String;)Z".into(),
11667 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_publishNodeJni as *mut c_void,
11668 },
11669 #[cfg(feature = "sync")]
11670 NativeMethod {
11671 name: "getMarkersJni".into(),
11672 sig: "(J)Ljava/lang/String;".into(),
11673 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getMarkersJni as *mut c_void,
11674 },
11675 #[cfg(feature = "sync")]
11676 NativeMethod {
11677 name: "publishMarkerJni".into(),
11678 sig: "(JLjava/lang/String;)Z".into(),
11679 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_publishMarkerJni as *mut c_void,
11680 },
11681 #[cfg(feature = "sync")]
11682 NativeMethod {
11683 name: "publishDocumentJni".into(),
11684 sig: "(JLjava/lang/String;Ljava/lang/String;)Ljava/lang/String;".into(),
11685 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_publishDocumentJni as *mut c_void,
11686 },
11687 #[cfg(feature = "sync")]
11688 NativeMethod {
11689 name: "publishDocumentWithOriginJni".into(),
11690 sig: "(JLjava/lang/String;Ljava/lang/String;Ljava/lang/String;)Ljava/lang/String;"
11691 .into(),
11692 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_publishDocumentWithOriginJni
11693 as *mut c_void,
11694 },
11695 #[cfg(all(feature = "sync", feature = "bluetooth"))]
11696 NativeMethod {
11697 name: "ingestPositionJni".into(),
11698 sig: "(JLjava/lang/String;)Ljava/lang/String;".into(),
11699 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_ingestPositionJni as *mut c_void,
11700 },
11701 #[cfg(all(feature = "sync", feature = "bluetooth"))]
11702 NativeMethod {
11703 name: "ingestInboundFrameJni".into(),
11704 sig: "(JLjava/lang/String;[B)Ljava/lang/String;".into(),
11705 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_ingestInboundFrameJni as *mut c_void,
11706 },
11707 #[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
11708 NativeMethod {
11709 name: "ingestInboundLiteFrameJni".into(),
11710 sig: "(JLjava/lang/String;[B)Ljava/lang/String;".into(),
11711 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_ingestInboundLiteFrameJni as *mut c_void,
11712 },
11713 #[cfg(feature = "sync")]
11714 NativeMethod {
11715 name: "connectPeerJni".into(),
11716 sig: "(JLjava/lang/String;Ljava/lang/String;)Z".into(),
11717 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_connectPeerJni as *mut c_void,
11718 },
11719 #[cfg(feature = "sync")]
11720 NativeMethod {
11721 name: "createNodeWithConfigJni".into(),
11722 sig: "(Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;ZLjava/lang/String;)J"
11723 .into(),
11724 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_createNodeWithConfigJni as *mut c_void,
11725 },
11726 // peat#925: the four subscription methods
11727 // (subscribe/unsubscribeDocumentChangesJni,
11728 // subscribe/unsubscribeOutboundFramesJni) are intentionally NOT
11729 // registered via nativeInit because their signatures reference
11730 // consumer-supplied listener interfaces
11731 // (`com/defenseunicorns/peat/DocumentChangeListener`,
11732 // `com/defenseunicorns/peat/OutboundFrameListener`) that don't
11733 // exist in peat-ffi's own `PeatJni.kt` — see the comment block at
11734 // peat-ffi/android/src/main/kotlin/.../PeatJni.kt:27-34 which
11735 // documents the "consumers declare these externs locally" pattern.
11736 //
11737 // The Rust extern fns `Java_com_defenseunicorns_peat_PeatJni_*`
11738 // are still exported and reachable via JNI's auto-lookup-by-name
11739 // convention: any consumer (peat-atak-plugin, downstream apps)
11740 // that declares `external fun subscribeDocumentChangesJni(...)`
11741 // alongside its `DocumentChangeListener` interface gets the
11742 // function resolved via dlsym at first call.
11743 //
11744 // Why these were here: ADR-059 Slice 1.b's outbound-frame
11745 // wiring was developed against a peat-atak-plugin build that
11746 // DID declare the listener interfaces; the `NativeMethod`
11747 // entries were copy-pasted from that build's lockstep
11748 // registration table without re-checking peat-ffi's own
11749 // PeatJni.kt surface.
11750 //
11751 // What went wrong: `JNI_OnLoad → nativeInit → RegisterNatives`
11752 // tries to bind every entry to a corresponding member on
11753 // `com.defenseunicorns.peat.PeatJni`. The DocumentChangeListener
11754 // / OutboundFrameListener signatures reference Kotlin classes
11755 // that don't exist. CheckJNI (active on debug-instrumented
11756 // builds, which is the AndroidJUnit harness configuration on
11757 // the Galaxy Tab A9+ CI runner) aborts the process on
11758 // registration mismatch — `Fatal signal 6 (SIGABRT), code -1
11759 // (SI_QUEUE)` in tid == JUnit-runner-tid, ~12ms after
11760 // `System.loadLibrary("peat_ffi")` returns. The post-
11761 // IrohTransport timing of the abort in earlier logcats was
11762 // misleading — the actual fault is during `System.loadLibrary`
11763 // which the test harness only logs after the abort propagates.
11764 // Blob transfer (ADR-060)
11765 #[cfg(feature = "sync")]
11766 NativeMethod {
11767 name: "enableBlobTransferJni".into(),
11768 sig: "(JLjava/lang/String;)Z".into(),
11769 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_enableBlobTransferJni as *mut c_void,
11770 },
11771 #[cfg(feature = "sync")]
11772 NativeMethod {
11773 name: "blobAddPeerJni".into(),
11774 sig: "(JLjava/lang/String;Ljava/lang/String;)Z".into(),
11775 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_blobAddPeerJni as *mut c_void,
11776 },
11777 #[cfg(feature = "sync")]
11778 NativeMethod {
11779 name: "blobPutJni".into(),
11780 sig: "(J[BLjava/lang/String;)Ljava/lang/String;".into(),
11781 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_blobPutJni as *mut c_void,
11782 },
11783 #[cfg(feature = "sync")]
11784 NativeMethod {
11785 name: "blobGetJni".into(),
11786 sig: "(JLjava/lang/String;)[B".into(),
11787 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_blobGetJni as *mut c_void,
11788 },
11789 #[cfg(feature = "sync")]
11790 NativeMethod {
11791 name: "blobExistsLocallyJni".into(),
11792 sig: "(JLjava/lang/String;)Z".into(),
11793 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_blobExistsLocallyJni as *mut c_void,
11794 },
11795 #[cfg(feature = "sync")]
11796 NativeMethod {
11797 name: "blobEndpointIdJni".into(),
11798 sig: "(J)Ljava/lang/String;".into(),
11799 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_blobEndpointIdJni as *mut c_void,
11800 },
11801 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
11802 NativeMethod {
11803 name: "bleSetStartedJni".into(),
11804 sig: "(JZ)V".into(),
11805 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_bleSetStartedJni as *mut c_void,
11806 },
11807 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
11808 NativeMethod {
11809 name: "bleAddPeerJni".into(),
11810 sig: "(JLjava/lang/String;)V".into(),
11811 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_bleAddPeerJni as *mut c_void,
11812 },
11813 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
11814 NativeMethod {
11815 name: "bleRemovePeerJni".into(),
11816 sig: "(JLjava/lang/String;)V".into(),
11817 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_bleRemovePeerJni as *mut c_void,
11818 },
11819 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
11820 NativeMethod {
11821 name: "bleIsAvailableJni".into(),
11822 sig: "(J)Z".into(),
11823 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_bleIsAvailableJni as *mut c_void,
11824 },
11825 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
11826 NativeMethod {
11827 name: "blePeerCountJni".into(),
11828 sig: "(J)I".into(),
11829 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_blePeerCountJni as *mut c_void,
11830 },
11831 ];
11832
11833 // Register native methods - the class is passed in from Kotlin so it's valid
11834 if let Err(_e) = env.register_native_methods(&class, &methods) {
11835 // Log error but don't crash - caller will see methods not registered
11836 let _ = env.exception_describe();
11837 let _ = env.exception_clear();
11838 }
11839}
11840
11841/// Bridge `tracing` events into android logcat (peat#850).
11842///
11843/// peat-mesh and peat-protocol emit per-doc sync results, transport
11844/// errors, and other diagnostics via `tracing::error!` /
11845/// `tracing::warn!` / `tracing::info!` / `tracing::debug!`. Without
11846/// a subscriber installed these events go nowhere on Android — which
11847/// is how the marker-sync silent-failure bug went un-diagnosed until
11848/// peat-ffi `request_sync` got its own `android_log` (peat#848).
11849///
11850/// This subscriber routes every tracing event matching the filter
11851/// to logcat under the `PeatRust` tag, **with the tracing `Level`
11852/// mapped to the corresponding Android log priority** so
11853/// `adb logcat *:W` / `*:E` priority filtering surfaces peat-mesh's
11854/// `warn!` / `error!` events. Priority mapping (Android NDK
11855/// convention): `ERROR→6, WARN→5, INFO→4, DEBUG→3, TRACE→2`.
11856///
11857/// Implementation uses a custom `tracing_subscriber::Layer<S>` impl
11858/// (not the `fmt-layer` + custom `Write` pipeline) because the
11859/// formatted-bytes interface only sees the rendered string, not the
11860/// originating `Event`'s metadata. The Layer pulls
11861/// `event.metadata().level()` directly and dispatches to
11862/// `__android_log_write` with the mapped priority. peat#851 round-5.
11863///
11864/// Idempotent via `OnceLock` — safe to call multiple times. Failures
11865/// to install (another subscriber already global) are logged once
11866/// and ignored, never panic.
11867///
11868/// The level defaults to INFO; override with `PEAT_TRACING_LEVEL=debug`
11869/// (or any `tracing-subscriber::EnvFilter` directive) at process
11870/// launch via an environment variable on the Android side. Going
11871/// below INFO is verbose — fine for active diagnostic, not for
11872/// steady-state.
11873#[cfg(target_os = "android")]
11874fn init_android_tracing() {
11875 use std::sync::OnceLock;
11876 static INITIALIZED: OnceLock<()> = OnceLock::new();
11877 INITIALIZED.get_or_init(|| {
11878 use std::ffi::CString;
11879 use std::fmt::Write as _;
11880 use std::os::raw::c_char;
11881 use tracing::field::{Field, Visit};
11882 use tracing::{Event, Level, Subscriber};
11883 use tracing_subscriber::layer::{Context, SubscriberExt};
11884 use tracing_subscriber::util::SubscriberInitExt;
11885 use tracing_subscriber::{EnvFilter, Layer};
11886
11887 extern "C" {
11888 fn __android_log_write(prio: i32, tag: *const c_char, text: *const c_char) -> i32;
11889 }
11890
11891 // Tag is a compile-time constant — allocate the CString once
11892 // for the lifetime of the process, not on every log event.
11893 fn tag_ptr() -> *const c_char {
11894 static TAG: OnceLock<CString> = OnceLock::new();
11895 TAG.get_or_init(|| CString::new("PeatRust").expect("static tag"))
11896 .as_ptr()
11897 }
11898
11899 /// Visitor that flattens an event's fields into a single
11900 /// string. Treats the `message` field (where `info!("X")`'s
11901 /// argument lands) specially so it's not prefixed with
11902 /// `message=`. Other fields render as `name=value`.
11903 #[derive(Default)]
11904 struct FieldStringifier(String);
11905 impl Visit for FieldStringifier {
11906 fn record_debug(&mut self, field: &Field, value: &dyn std::fmt::Debug) {
11907 if !self.0.is_empty() {
11908 self.0.push(' ');
11909 }
11910 if field.name() == "message" {
11911 // Debug-format strips the surrounding quotes if
11912 // the value is a `&str` literal, which matches
11913 // how the fmt-layer rendered messages previously.
11914 let _ = write!(self.0, "{:?}", value);
11915 } else {
11916 let _ = write!(self.0, "{}={:?}", field.name(), value);
11917 }
11918 }
11919 fn record_str(&mut self, field: &Field, value: &str) {
11920 if !self.0.is_empty() {
11921 self.0.push(' ');
11922 }
11923 if field.name() == "message" {
11924 self.0.push_str(value);
11925 } else {
11926 let _ = write!(self.0, "{}={}", field.name(), value);
11927 }
11928 }
11929 }
11930
11931 /// `Level → Android NDK priority` mapping. Verbose=2,
11932 /// Debug=3, Info=4, Warn=5, Error=6. Constants live in
11933 /// `android/log.h`; we hardcode them rather than pulling in
11934 /// the `ndk-sys` crate just for five integers.
11935 fn android_priority(level: &Level) -> i32 {
11936 match *level {
11937 Level::ERROR => 6,
11938 Level::WARN => 5,
11939 Level::INFO => 4,
11940 Level::DEBUG => 3,
11941 Level::TRACE => 2,
11942 }
11943 }
11944
11945 struct AndroidLayer;
11946 impl<S: Subscriber> Layer<S> for AndroidLayer {
11947 fn on_event(&self, event: &Event<'_>, _ctx: Context<'_, S>) {
11948 let metadata = event.metadata();
11949 let prio = android_priority(metadata.level());
11950
11951 let mut visitor = FieldStringifier::default();
11952 event.record(&mut visitor);
11953 // Prefix with the target (typically the source crate
11954 // / module path) so a logcat reader can grep for
11955 // `peat_mesh::storage::automerge_sync` without
11956 // needing the priority signal alone.
11957 let formatted = if visitor.0.is_empty() {
11958 metadata.target().to_string()
11959 } else {
11960 format!("{}: {}", metadata.target(), visitor.0)
11961 };
11962
11963 // Cap each entry well under logcat's per-line limit
11964 // (~4 KiB). The source string is valid UTF-8, so we
11965 // must truncate on a char boundary — walk back from
11966 // byte LIMIT to a UTF-8 leading byte. Worst case 3
11967 // bytes back, O(1).
11968 const LIMIT: usize = 3500;
11969 let bytes = formatted.as_bytes();
11970 let truncated: &[u8] = if bytes.len() > LIMIT {
11971 let mut cut = LIMIT;
11972 while cut > 0 && (bytes[cut] & 0b1100_0000) == 0b1000_0000 {
11973 cut -= 1;
11974 }
11975 &bytes[..cut]
11976 } else {
11977 bytes
11978 };
11979
11980 if let Ok(c_msg) = CString::new(truncated) {
11981 unsafe {
11982 __android_log_write(prio, tag_ptr(), c_msg.as_ptr());
11983 }
11984 }
11985 }
11986 }
11987
11988 let env_filter = EnvFilter::try_from_env("PEAT_TRACING_LEVEL")
11989 .unwrap_or_else(|_| EnvFilter::new("info"));
11990
11991 let result = tracing_subscriber::registry()
11992 .with(env_filter)
11993 .with(AndroidLayer)
11994 .try_init();
11995
11996 match result {
11997 Ok(()) => android_log("init_android_tracing: subscriber installed"),
11998 Err(e) => android_log(&format!(
11999 "init_android_tracing: subscriber NOT installed (already set?): {}",
12000 e
12001 )),
12002 }
12003 });
12004}
12005
12006/// Install a `std::panic::set_hook` that writes the panic payload +
12007/// file:line + (best-effort) backtrace to logcat under the `PeatFFI`
12008/// tag before chaining to the default handler. Idempotent via
12009/// `OnceLock`.
12010///
12011/// Why this exists: on Android, the default panic handler writes to
12012/// stderr which logcat never captures, so an `unwrap()` in a worker
12013/// thread aborts the process with only a bionic SIGABRT trace whose
12014/// frames are stripped Rust symbols. With this hook installed, the
12015/// panic message + source location lands in the existing PeatFFI
12016/// logcat stream that AndroidJUnit and `adb logcat` already
12017/// surface.
12018#[cfg(target_os = "android")]
12019fn install_android_panic_hook() {
12020 use std::sync::OnceLock;
12021 static INSTALLED: OnceLock<()> = OnceLock::new();
12022 INSTALLED.get_or_init(|| {
12023 let default_hook = std::panic::take_hook();
12024 std::panic::set_hook(Box::new(move |info| {
12025 let payload = info
12026 .payload()
12027 .downcast_ref::<&str>()
12028 .copied()
12029 .or_else(|| info.payload().downcast_ref::<String>().map(String::as_str))
12030 .unwrap_or("<non-string panic payload>");
12031 let location = info
12032 .location()
12033 .map(|l| format!("{}:{}:{}", l.file(), l.line(), l.column()))
12034 .unwrap_or_else(|| "<unknown location>".to_string());
12035 let thread = std::thread::current();
12036 let thread_name = thread.name().unwrap_or("<unnamed>");
12037 android_log(&format!(
12038 "PANIC in thread '{}' at {}: {}",
12039 thread_name, location, payload
12040 ));
12041 default_hook(info);
12042 }));
12043 android_log("install_android_panic_hook: panic hook installed");
12044 });
12045}
12046
12047/// JNI_OnLoad - Called when library is loaded via System.loadLibrary()
12048///
12049/// This is our chance to register native methods while we have access to
12050/// the JNI environment from inside the library's linker namespace.
12051#[no_mangle]
12052#[allow(non_snake_case)]
12053#[allow(clippy::not_unsafe_ptr_arg_deref)] // JNI ABI requires raw pointer params
12054pub extern "C" fn JNI_OnLoad(vm: *mut JavaVM, _reserved: *mut c_void) -> jint {
12055 // Log that we're being called
12056 #[cfg(target_os = "android")]
12057 android_log("JNI_OnLoad called for peat_ffi");
12058
12059 // Bridge `tracing` events (peat-mesh's per-doc sync warnings,
12060 // peat-protocol's sync coordinator events, etc.) into logcat
12061 // under the `PeatRust` tag. peat#850 — previous attempts at
12062 // tracing init "caused issues" per the prior comment here; this
12063 // implementation uses a minimal in-process writer with no JNI
12064 // re-entry and `try_init` so it's a no-op if another subscriber
12065 // was already set.
12066 #[cfg(target_os = "android")]
12067 init_android_tracing();
12068
12069 // Forward Rust panics to logcat before the default hook aborts
12070 // the process. Without this, an `unwrap()` deep in a worker
12071 // thread aborts with no diagnostic — Android's default panic
12072 // path writes to stderr which logcat never captures, and the
12073 // process exits via SIGABRT with only a bionic backtrace whose
12074 // frames are stripped Rust symbols. peat#925 follow-on: makes
12075 // future panics in the iroh/rustls/aws-lc-rs/redb code paths
12076 // self-diagnose in the existing PeatFFI logcat tag.
12077 #[cfg(target_os = "android")]
12078 install_android_panic_hook();
12079
12080 // Initialize `ndk-context`'s global JavaVM cell. The crate is
12081 // pulled in transitively by the iroh 1.0.0-rc.0 cascade
12082 // (swarm-discovery / iroh-mdns-address-lookup / iroh-dns →
12083 // hickory-resolver) and panics with "android context was not
12084 // initialized" the first time any Android-aware code in that
12085 // subtree resolves the global context. Without this call,
12086 // every `createNodeJni` SIGABRT's mid-bind. Surfaced by the
12087 // panic hook above:
12088 // PANIC in thread '<unnamed>' at ndk-context-0.1.1/src/lib.rs:72:
12089 // android context was not initialized
12090 //
12091 // **Safety boundary of the null-context init below.** We pass
12092 // our `JavaVM*` (definitely available — it's the argument to
12093 // JNI_OnLoad) and `null` for the Android `Context` jobject (NOT
12094 // available from JNI_OnLoad — JNI_OnLoad runs before any
12095 // Application/Activity has been instantiated by the framework).
12096 // Code paths that consult only the JVM (mDNS multicast worker,
12097 // swarm-discovery sender, iroh thread attachment) get served by
12098 // this init alone. Code paths that genuinely need the
12099 // *Context* itself — hickory-resolver's Android system-DNS
12100 // probe via ConnectivityManager, NDK asset-manager access,
12101 // app-private file paths — will hit `ndk_context::android_context().context()`
12102 // and panic on the null. Consumers exercising those paths
12103 // (any iroh deployment using DNS-based discovery — relay, pkarr,
12104 // non-mDNS peer lookups) MUST call `setAndroidContextJni` from
12105 // their `Application.onCreate` before `createNodeJni`. peat-ffi's
12106 // own surface tests don't reach those paths, but a downstream
12107 // consumer hitting them without `setAndroidContextJni` would
12108 // get a `PANIC in thread '<unnamed>' at ndk-context-0.1.1/...:
12109 // android context was not initialized` line via the panic hook
12110 // above and a SIGABRT — same diagnostic the null-context
12111 // discovery in this very PR surfaced. peat#925 QA WARNING-1.
12112 #[cfg(target_os = "android")]
12113 unsafe {
12114 ndk_context::initialize_android_context(vm as *mut c_void, std::ptr::null_mut());
12115 android_log("JNI_OnLoad: ndk_context::initialize_android_context(vm, null) done");
12116 }
12117
12118 // Store JavaVM globally for callbacks from any thread
12119 let java_vm = unsafe {
12120 match jni::JavaVM::from_raw(vm) {
12121 Ok(jvm) => jvm,
12122 Err(_) => {
12123 #[cfg(target_os = "android")]
12124 android_log("JNI_OnLoad: Failed to create JavaVM from raw pointer");
12125 return jni::sys::JNI_ERR;
12126 }
12127 }
12128 };
12129 *JAVA_VM.lock().unwrap() = Some(java_vm);
12130
12131 // Get JNIEnv from JavaVM
12132 let mut env = unsafe {
12133 let mut env_ptr: *mut jni::sys::JNIEnv = std::ptr::null_mut();
12134 let get_env_result = (**vm).GetEnv.unwrap()(
12135 vm,
12136 &mut env_ptr as *mut _ as *mut *mut c_void,
12137 JNI_VERSION_1_6 as i32,
12138 );
12139 if get_env_result != jni::sys::JNI_OK as i32 {
12140 #[cfg(target_os = "android")]
12141 android_log("JNI_OnLoad: GetEnv failed");
12142 return jni::sys::JNI_ERR;
12143 }
12144 match JNIEnv::from_raw(env_ptr) {
12145 Ok(env) => env,
12146 Err(_) => {
12147 #[cfg(target_os = "android")]
12148 android_log("JNI_OnLoad: JNIEnv::from_raw failed");
12149 return jni::sys::JNI_ERR;
12150 }
12151 }
12152 };
12153
12154 // Try to find PeerEventManager class and store global reference for callbacks
12155 let peer_event_manager_class = "com/defenseunicorns/peat/PeerEventManager";
12156 match env.find_class(peer_event_manager_class) {
12157 Ok(class) => match env.new_global_ref(class) {
12158 Ok(global_ref) => {
12159 *PEER_EVENT_MANAGER_CLASS.lock().unwrap() = Some(global_ref);
12160 #[cfg(target_os = "android")]
12161 android_log("JNI_OnLoad: PeerEventManager class found and cached");
12162 }
12163 Err(_) => {
12164 #[cfg(target_os = "android")]
12165 android_log("JNI_OnLoad: Failed to create global ref for PeerEventManager");
12166 }
12167 },
12168 Err(_) => {
12169 // CRITICAL: clear the pending ClassNotFoundException
12170 // before any further JNI call. Without this, the very
12171 // next find_class (for PeatJni at line 9418) detects a
12172 // pending exception and the JNI runtime aborts the
12173 // process with SIGABRT. Consumers that don't ship a
12174 // PeerEventManager (anything other than peat-atak-plugin)
12175 // crash at System.loadLibrary("peat_ffi"). Surfaced by
12176 // peat-mesh#145 / peat#887.
12177 let _ = env.exception_clear();
12178 #[cfg(target_os = "android")]
12179 android_log(
12180 "JNI_OnLoad: PeerEventManager class not found (OK if loading before class init)",
12181 );
12182 }
12183 }
12184
12185 #[cfg(target_os = "android")]
12186 android_log("JNI_OnLoad: Got JNIEnv, looking for PeatJni class...");
12187
12188 // Try to find the PeatJni class and register natives
12189 let class_name = "com/defenseunicorns/peat/PeatJni";
12190 match env.find_class(class_name) {
12191 Ok(class) => {
12192 #[cfg(target_os = "android")]
12193 android_log("JNI_OnLoad: Found PeatJni class, registering natives...");
12194
12195 // Register native methods
12196 use jni::NativeMethod;
12197 let methods: Vec<NativeMethod> = vec![
12198 NativeMethod {
12199 name: "nativeInit".into(),
12200 sig: "()V".into(),
12201 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_nativeInit as *mut c_void,
12202 },
12203 NativeMethod {
12204 name: "peatVersion".into(),
12205 sig: "()Ljava/lang/String;".into(),
12206 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_peatVersion as *mut c_void,
12207 },
12208 NativeMethod {
12209 name: "testJni".into(),
12210 sig: "()Ljava/lang/String;".into(),
12211 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_testJni as *mut c_void,
12212 },
12213 #[cfg(target_os = "android")]
12214 NativeMethod {
12215 name: "setAndroidContextJni".into(),
12216 sig: "(Ljava/lang/Object;)V".into(),
12217 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_setAndroidContextJni
12218 as *mut c_void,
12219 },
12220 #[cfg(target_os = "android")]
12221 NativeMethod {
12222 name: "verifyAndroidContextJni".into(),
12223 sig: "()Z".into(),
12224 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_verifyAndroidContextJni
12225 as *mut c_void,
12226 },
12227 #[cfg(feature = "sync")]
12228 NativeMethod {
12229 name: "createNodeJni".into(),
12230 sig: "(Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;)J".into(),
12231 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_createNodeJni as *mut c_void,
12232 },
12233 #[cfg(feature = "sync")]
12234 NativeMethod {
12235 name: "getGlobalNodeHandleJni".into(),
12236 sig: "()J".into(),
12237 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getGlobalNodeHandleJni
12238 as *mut c_void,
12239 },
12240 #[cfg(feature = "sync")]
12241 NativeMethod {
12242 name: "clearGlobalNodeHandleJni".into(),
12243 sig: "()V".into(),
12244 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_clearGlobalNodeHandleJni
12245 as *mut c_void,
12246 },
12247 #[cfg(feature = "sync")]
12248 NativeMethod {
12249 name: "nodeIdJni".into(),
12250 sig: "(J)Ljava/lang/String;".into(),
12251 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_nodeIdJni as *mut c_void,
12252 },
12253 #[cfg(feature = "sync")]
12254 NativeMethod {
12255 name: "peerCountJni".into(),
12256 sig: "(J)I".into(),
12257 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_peerCountJni as *mut c_void,
12258 },
12259 #[cfg(feature = "sync")]
12260 NativeMethod {
12261 name: "connectedPeersJni".into(),
12262 sig: "(J)Ljava/lang/String;".into(),
12263 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_connectedPeersJni as *mut c_void,
12264 },
12265 #[cfg(feature = "sync")]
12266 NativeMethod {
12267 name: "requestSyncJni".into(),
12268 sig: "(J)Z".into(),
12269 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_requestSyncJni as *mut c_void,
12270 },
12271 #[cfg(feature = "sync")]
12272 NativeMethod {
12273 name: "endpointSocketAddrJni".into(),
12274 sig: "(J)Ljava/lang/String;".into(),
12275 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_endpointSocketAddrJni
12276 as *mut c_void,
12277 },
12278 #[cfg(feature = "sync")]
12279 NativeMethod {
12280 name: "getDocumentJni".into(),
12281 sig: "(JLjava/lang/String;Ljava/lang/String;)Ljava/lang/String;".into(),
12282 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getDocumentJni as *mut c_void,
12283 },
12284 #[cfg(feature = "sync")]
12285 NativeMethod {
12286 name: "forceStoreErrorForTestingJni".into(),
12287 sig: "(J)Z".into(),
12288 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_forceStoreErrorForTestingJni
12289 as *mut c_void,
12290 },
12291 #[cfg(feature = "sync")]
12292 NativeMethod {
12293 name: "startSyncJni".into(),
12294 sig: "(J)Z".into(),
12295 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_startSyncJni as *mut c_void,
12296 },
12297 #[cfg(feature = "sync")]
12298 NativeMethod {
12299 name: "freeNodeJni".into(),
12300 sig: "(J)V".into(),
12301 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_freeNodeJni as *mut c_void,
12302 },
12303 #[cfg(feature = "sync")]
12304 NativeMethod {
12305 name: "getCellsJni".into(),
12306 sig: "(J)Ljava/lang/String;".into(),
12307 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getCellsJni as *mut c_void,
12308 },
12309 #[cfg(feature = "sync")]
12310 NativeMethod {
12311 name: "getTracksJni".into(),
12312 sig: "(J)Ljava/lang/String;".into(),
12313 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getTracksJni as *mut c_void,
12314 },
12315 #[cfg(feature = "sync")]
12316 NativeMethod {
12317 name: "getNodesJni".into(),
12318 sig: "(J)Ljava/lang/String;".into(),
12319 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getNodesJni as *mut c_void,
12320 },
12321 #[cfg(feature = "sync")]
12322 NativeMethod {
12323 name: "getCommandsJni".into(),
12324 sig: "(J)Ljava/lang/String;".into(),
12325 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getCommandsJni as *mut c_void,
12326 },
12327 #[cfg(feature = "sync")]
12328 NativeMethod {
12329 name: "getMarkersJni".into(),
12330 sig: "(J)Ljava/lang/String;".into(),
12331 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_getMarkersJni as *mut c_void,
12332 },
12333 #[cfg(feature = "sync")]
12334 NativeMethod {
12335 name: "publishMarkerJni".into(),
12336 sig: "(JLjava/lang/String;)Z".into(),
12337 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_publishMarkerJni
12338 as *mut c_void,
12339 },
12340 #[cfg(feature = "sync")]
12341 NativeMethod {
12342 name: "publishNodeJni".into(),
12343 sig: "(JLjava/lang/String;)Z".into(),
12344 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_publishNodeJni
12345 as *mut c_void,
12346 },
12347 #[cfg(feature = "sync")]
12348 NativeMethod {
12349 name: "publishDocumentJni".into(),
12350 sig: "(JLjava/lang/String;Ljava/lang/String;)Ljava/lang/String;".into(),
12351 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_publishDocumentJni
12352 as *mut c_void,
12353 },
12354 #[cfg(feature = "sync")]
12355 NativeMethod {
12356 name: "publishDocumentWithOriginJni".into(),
12357 sig: "(JLjava/lang/String;Ljava/lang/String;Ljava/lang/String;)\
12358 Ljava/lang/String;"
12359 .into(),
12360 fn_ptr:
12361 Java_com_defenseunicorns_peat_PeatJni_publishDocumentWithOriginJni
12362 as *mut c_void,
12363 },
12364 #[cfg(all(feature = "sync", feature = "bluetooth"))]
12365 NativeMethod {
12366 name: "ingestPositionJni".into(),
12367 sig: "(JLjava/lang/String;)Ljava/lang/String;".into(),
12368 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_ingestPositionJni
12369 as *mut c_void,
12370 },
12371 #[cfg(all(feature = "sync", feature = "bluetooth"))]
12372 NativeMethod {
12373 name: "ingestInboundFrameJni".into(),
12374 sig: "(JLjava/lang/String;[B)Ljava/lang/String;".into(),
12375 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_ingestInboundFrameJni
12376 as *mut c_void,
12377 },
12378 #[cfg(all(feature = "sync", feature = "bluetooth", feature = "lite-bridge"))]
12379 NativeMethod {
12380 name: "ingestInboundLiteFrameJni".into(),
12381 sig: "(JLjava/lang/String;[B)Ljava/lang/String;".into(),
12382 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_ingestInboundLiteFrameJni
12383 as *mut c_void,
12384 },
12385 #[cfg(feature = "sync")]
12386 NativeMethod {
12387 name: "connectPeerJni".into(),
12388 sig: "(JLjava/lang/String;Ljava/lang/String;)Z".into(),
12389 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_connectPeerJni as *mut c_void,
12390 },
12391 #[cfg(feature = "sync")]
12392 NativeMethod {
12393 name: "createNodeWithConfigJni".into(),
12394 sig: "(Ljava/lang/String;Ljava/lang/String;Ljava/lang/String;ZLjava/lang/String;)J"
12395 .into(),
12396 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_createNodeWithConfigJni
12397 as *mut c_void,
12398 },
12399 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
12400 NativeMethod {
12401 name: "bleSetStartedJni".into(),
12402 sig: "(JZ)V".into(),
12403 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_bleSetStartedJni as *mut c_void,
12404 },
12405 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
12406 NativeMethod {
12407 name: "bleAddPeerJni".into(),
12408 sig: "(JLjava/lang/String;)V".into(),
12409 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_bleAddPeerJni as *mut c_void,
12410 },
12411 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
12412 NativeMethod {
12413 name: "bleRemovePeerJni".into(),
12414 sig: "(JLjava/lang/String;)V".into(),
12415 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_bleRemovePeerJni as *mut c_void,
12416 },
12417 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
12418 NativeMethod {
12419 name: "bleIsAvailableJni".into(),
12420 sig: "(J)Z".into(),
12421 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_bleIsAvailableJni as *mut c_void,
12422 },
12423 #[cfg(all(feature = "sync", feature = "bluetooth", target_os = "android"))]
12424 NativeMethod {
12425 name: "blePeerCountJni".into(),
12426 sig: "(J)I".into(),
12427 fn_ptr: Java_com_defenseunicorns_peat_PeatJni_blePeerCountJni as *mut c_void,
12428 },
12429 ];
12430
12431 match env.register_native_methods(&class, &methods) {
12432 Ok(_) => {
12433 #[cfg(target_os = "android")]
12434 android_log("JNI_OnLoad: Native methods registered successfully!");
12435 }
12436 Err(_) => {
12437 #[cfg(target_os = "android")]
12438 android_log("JNI_OnLoad: Failed to register native methods");
12439 let _ = env.exception_describe();
12440 let _ = env.exception_clear();
12441 }
12442 }
12443 }
12444 Err(_) => {
12445 #[cfg(target_os = "android")]
12446 android_log(
12447 "JNI_OnLoad: PeatJni class not found (this is OK if loading before class init)",
12448 );
12449 // Class not loaded yet - this is OK, nativeInit will be called
12450 // later
12451 }
12452 }
12453
12454 JNI_VERSION_1_6
12455}
12456
12457/// Log to Android logcat
12458#[cfg(target_os = "android")]
12459fn android_log(msg: &str) {
12460 use std::ffi::CString;
12461 use std::os::raw::c_char;
12462
12463 let tag = CString::new("PeatFFI").unwrap();
12464 let msg = CString::new(msg).unwrap();
12465
12466 unsafe {
12467 // Android log priority INFO = 4
12468 extern "C" {
12469 fn __android_log_write(prio: i32, tag: *const c_char, text: *const c_char) -> i32;
12470 }
12471 __android_log_write(4, tag.as_ptr(), msg.as_ptr());
12472 }
12473}
12474
12475/// Notify Java PeerEventManager of a peer connected event
12476#[cfg(feature = "sync")]
12477fn notify_peer_connected(peer_id: &str) {
12478 notify_peer_event("notifyPeerConnected", peer_id, None);
12479}
12480
12481/// Notify Java PeerEventManager of a peer disconnected event
12482#[cfg(feature = "sync")]
12483fn notify_peer_disconnected(peer_id: &str, reason: &str) {
12484 notify_peer_event("notifyPeerDisconnected", peer_id, Some(reason));
12485}
12486
12487/// Helper to call PeerEventManager static methods
12488#[cfg(feature = "sync")]
12489fn notify_peer_event(method_name: &str, peer_id: &str, reason: Option<&str>) {
12490 let java_vm_guard = JAVA_VM.lock().unwrap();
12491 let java_vm = match java_vm_guard.as_ref() {
12492 Some(vm) => vm,
12493 None => {
12494 #[cfg(target_os = "android")]
12495 android_log("notify_peer_event: No JavaVM available");
12496 return;
12497 }
12498 };
12499
12500 // Check if we already have the class cached
12501 let mut class_guard = PEER_EVENT_MANAGER_CLASS.lock().unwrap();
12502
12503 // If not cached, try to find it now (lazy loading)
12504 if class_guard.is_none() {
12505 #[cfg(target_os = "android")]
12506 android_log("notify_peer_event: PeerEventManager class not cached, trying to find it...");
12507
12508 // Attach current thread to get env for class lookup
12509 if let Ok(mut env) = java_vm.attach_current_thread() {
12510 let peer_event_manager_class = "com/defenseunicorns/peat/PeerEventManager";
12511 if let Ok(class) = env.find_class(peer_event_manager_class) {
12512 if let Ok(global_ref) = env.new_global_ref(class) {
12513 *class_guard = Some(global_ref);
12514 #[cfg(target_os = "android")]
12515 android_log("notify_peer_event: PeerEventManager class found and cached!");
12516 }
12517 } else {
12518 // Clear the pending ClassNotFoundException for the
12519 // same reason as the JNI_OnLoad branch above
12520 // (peat#887). A consumer without PeerEventManager
12521 // is fine — peer events just don't get notified.
12522 let _ = env.exception_clear();
12523 #[cfg(target_os = "android")]
12524 android_log("notify_peer_event: PeerEventManager class not found");
12525 }
12526 }
12527 }
12528
12529 let class_ref = match class_guard.as_ref() {
12530 Some(c) => c,
12531 None => {
12532 #[cfg(target_os = "android")]
12533 android_log("notify_peer_event: PeerEventManager class not available");
12534 return;
12535 }
12536 };
12537
12538 // Attach current thread to JVM
12539 let mut env = match java_vm.attach_current_thread() {
12540 Ok(env) => env,
12541 Err(e) => {
12542 #[cfg(target_os = "android")]
12543 android_log(&format!(
12544 "notify_peer_event: Failed to attach thread: {:?}",
12545 e
12546 ));
12547 return;
12548 }
12549 };
12550
12551 // Create Java string for peer_id
12552 let peer_id_jstring = match env.new_string(peer_id) {
12553 Ok(s) => s,
12554 Err(_) => {
12555 #[cfg(target_os = "android")]
12556 android_log("notify_peer_event: Failed to create peer_id string");
12557 return;
12558 }
12559 };
12560
12561 // Call the appropriate method
12562 let result = if let Some(reason) = reason {
12563 // notifyPeerDisconnected(String peerId, String reason)
12564 let reason_jstring = match env.new_string(reason) {
12565 Ok(s) => s,
12566 Err(_) => {
12567 #[cfg(target_os = "android")]
12568 android_log("notify_peer_event: Failed to create reason string");
12569 return;
12570 }
12571 };
12572 env.call_static_method(
12573 class_ref,
12574 method_name,
12575 "(Ljava/lang/String;Ljava/lang/String;)V",
12576 &[
12577 JValue::Object(&peer_id_jstring),
12578 JValue::Object(&reason_jstring),
12579 ],
12580 )
12581 } else {
12582 // notifyPeerConnected(String peerId)
12583 env.call_static_method(
12584 class_ref,
12585 method_name,
12586 "(Ljava/lang/String;)V",
12587 &[JValue::Object(&peer_id_jstring)],
12588 )
12589 };
12590
12591 if let Err(e) = result {
12592 #[cfg(target_os = "android")]
12593 android_log(&format!("notify_peer_event: Method call failed: {:?}", e));
12594 let _ = env.exception_describe();
12595 let _ = env.exception_clear();
12596 } else {
12597 #[cfg(target_os = "android")]
12598 android_log(&format!(
12599 "notify_peer_event: {} called for {}",
12600 method_name, peer_id
12601 ));
12602 }
12603}