use crate::logger::{log_info, log_warn, WriteLog};
use crate::{error::TransportError, DlcDevKitDlcManager, Oracle, Storage, Transport};
use async_trait::async_trait;
use bitcoin::secp256k1::PublicKey;
use lightning_net_tokio::connect_outbound;
use std::sync::Arc;
use tokio::sync::watch;
pub(crate) mod peer_manager;
pub use peer_manager::LightningTransport;
#[async_trait]
impl Transport for LightningTransport {
fn name(&self) -> String {
"lightning".into()
}
fn public_key(&self) -> PublicKey {
self.node_id
}
async fn send_message(&self, counterparty: PublicKey, message: ddk_messages::Message) {
log_info!(
self.logger,
"Sending message to counter_party={}",
counterparty.to_string()
);
if self.peer_manager.peer_by_node_id(&counterparty).is_some() {
self.message_handler.send_message(counterparty, message);
self.peer_manager.process_events();
} else {
log_warn!(
self.logger,
"Not connected to counterparty. Message not sent. counter_party={}",
counterparty.to_string()
);
}
}
async fn start<S: Storage, O: Oracle>(
&self,
mut stop_signal: watch::Receiver<bool>,
manager: Arc<DlcDevKitDlcManager<S, O>>,
) -> Result<(), TransportError> {
let listen_handle = self.listen(stop_signal.clone());
let process_handle = self.process_messages(stop_signal.clone(), manager.clone());
tokio::select! {
_ = stop_signal.changed() => Ok(()),
res = listen_handle => res.map_err(|e| TransportError::Listen(e.to_string()))?,
res = process_handle => res.map_err(|e| TransportError::MessageProcessing(e.to_string()))?,
}
}
async fn connect_outbound(&self, pubkey: PublicKey, host: &str) {
connect_outbound(self.peer_manager.clone(), pubkey, host.parse().unwrap()).await;
}
}