datex-embedded 0.0.1

Embedded support for datex-core
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

use crate::{
    esp::timestamp_generator::TimestampGenerator,
    setup::global_initializer::{
        SetupInitializer, WifiCredentials, WifiInitializer,
    },
};
use alloc::rc::Rc;
use datex_core::network::com_hub::ComHub;
use embassy_executor::Spawner;
use embassy_net::Stack;
use esp_hal::{peripherals, rtc_cntl::Rtc};
use sntpc::NtpTimestampGenerator;

pub struct EspWifiInitializer {
    pub wifi: peripherals::WIFI<'static>,
}

impl WifiInitializer for EspWifiInitializer {
    #[cfg(feature = "wifi")]
    async fn init_wifi_stack(
        self,
        spawner: &Spawner,
        credentials: WifiCredentials,
    ) -> Stack<'static> {
        crate::esp::wifi::init_wifi_stack(spawner, self.wifi, credentials).await
    }
}

pub struct EspSetupInitializer {
    pub rtc: Rtc<'static>,
}

impl SetupInitializer for EspSetupInitializer {
    fn register_com_interface_factories(
        &self,
        stack: &Option<Stack<'static>>,
        com_hub: Rc<ComHub>,
    ) {
        #[cfg(feature = "websocket-client")]
        {
            use crate::interfaces::websocket_client_interface_embedded::{
                WebSocketClientInterfaceEmbeddedGlobalState,
                WebSocketClientInterfaceSetupDataEmbedded,
            };
            if let Some(stack) = stack {
                use esp_hal::rng::Rng;

                WebSocketClientInterfaceEmbeddedGlobalState::set_global_state(
                    WebSocketClientInterfaceEmbeddedGlobalState {
                        stack: stack.clone(),
                        rng: Rc::new(Rng::new()),
                    },
                );
                com_hub.register_async_interface_factory::<WebSocketClientInterfaceSetupDataEmbedded>();
            }
        }
        #[cfg(feature = "tcp-client")]
        {
            use crate::interfaces::tcp_client_interface_embedded::{
                TCPClientInterfaceSetupDataEmbedded,
                TcpClientInterfaceEmbeddedGlobalState,
            };
            if let Some(stack) = stack {
                use esp_hal::rng::Rng;

                TcpClientInterfaceEmbeddedGlobalState::set_global_state(
                    TcpClientInterfaceEmbeddedGlobalState {
                        stack: *stack,
                        rng: Rc::new(Rng::new()),
                    },
                );
                com_hub.register_async_interface_factory::<TCPClientInterfaceSetupDataEmbedded>();
            }
        }
    }

    async fn set_current_time(&self, current_time_us: u64) {
        self.rtc.set_current_time_us(current_time_us);
    }

    fn get_timestamp_generator(&self) -> impl NtpTimestampGenerator + Copy {
        TimestampGenerator::new(&self.rtc)
    }
}