1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
//! Main BT lifecycle support
#[cfg(target_os = "android")]
#[macro_use]
extern crate lazy_static;
pub mod hal;
pub mod hci;
pub mod link;
use crate::hal::rootcanal_hal::RootcanalConfig;
use crate::hal::snoop::{SnoopConfig, SnoopMode};
use bt_common::GrpcFacade;
use gddi::{module, Registry, RegistryBuilder, Stoppable};
use std::sync::Arc;
use tokio::runtime::Runtime;
module! {
stack_module,
submodules {
crate::hal::hal_module,
crate::hci::hci_module,
}
}
/// Central state manager
pub struct Stack {
registry: Arc<Registry>,
rt: Arc<Runtime>,
}
impl Stack {
/// Construct a new Stack
pub async fn new(rt: Arc<Runtime>) -> Self {
let registry = Arc::new(RegistryBuilder::new().register_module(stack_module).build());
registry.inject(rt.clone()).await;
Self { registry, rt }
}
/// Helper to set the rootcanal port
pub async fn set_rootcanal_port(&self, port: Option<u16>) {
if let Some(port) = port {
self.registry.inject(RootcanalConfig::new("127.0.0.1", port)).await;
}
}
/// Configures snoop with defaults
pub async fn use_default_snoop(&self) {
self.configure_snoop(None).await;
}
/// Configures snoop. If the path is provided, full logging is turned on
pub async fn configure_snoop(&self, path: Option<String>) {
let mut config = SnoopConfig::default();
if let Some(path) = path {
config.set_path(path);
config.set_mode(SnoopMode::Full);
}
self.registry.inject(config).await;
}
/// Helper forwarding to underlying registry
pub async fn get<T: 'static + Clone + Send + Sync + Stoppable>(&self) -> T {
self.registry.get::<T>().await
}
/// Get, but blocks the current thread.
pub fn get_blocking<T: 'static + Clone + Send + Sync + Stoppable>(&self) -> T {
self.rt.block_on(self.get::<T>())
}
/// Helper to get a grpc service
pub async fn get_grpc<T: 'static + Clone + Send + Sync + GrpcFacade + Stoppable>(
&self,
) -> grpcio::Service {
self.get::<T>().await.into_grpc()
}
/// Stop the stack
pub async fn stop(&mut self) {
self.registry.stop_all().await;
}
/// Stop, but blocks the current thread.
pub fn stop_blocking(&mut self) {
let rt = self.rt.clone();
rt.block_on(self.stop());
}
/// Get a clone of the underlying runtime for this stack
pub fn get_runtime(&self) -> Arc<Runtime> {
self.rt.clone()
}
}
|
