use crate::marshal::resolver::handler;
use commonware_cryptography::{Digest, PublicKey};
use commonware_p2p::{Blocker, Provider, Receiver, Sender};
use commonware_resolver::p2p;
use commonware_runtime::{BufferPooler, Clock, Metrics, Spawner};
use commonware_utils::channel::mpsc;
use rand::Rng;
use std::time::Duration;
pub struct Config<P, C, B>
where
P: PublicKey,
C: Provider<PublicKey = P>,
B: Blocker<PublicKey = P>,
{
pub public_key: P,
pub peer_provider: C,
pub blocker: B,
pub mailbox_size: usize,
pub initial: Duration,
pub timeout: Duration,
pub fetch_retry_timeout: Duration,
pub priority_requests: bool,
pub priority_responses: bool,
}
pub fn init<E, C, B, D, S, R, P>(
ctx: &E,
config: Config<P, C, B>,
backfill: (S, R),
) -> (
mpsc::Receiver<handler::Message<D>>,
p2p::Mailbox<handler::Request<D>, P>,
)
where
E: BufferPooler + Rng + Spawner + Clock + Metrics,
C: Provider<PublicKey = P>,
B: Blocker<PublicKey = P>,
D: Digest,
S: Sender<PublicKey = P>,
R: Receiver<PublicKey = P>,
P: PublicKey,
{
let (sender, receiver) = mpsc::channel(config.mailbox_size);
let handler = handler::Handler::new(sender);
let (resolver_engine, resolver) = p2p::Engine::new(
ctx.with_label("resolver"),
p2p::Config {
peer_provider: config.peer_provider,
blocker: config.blocker,
consumer: handler.clone(),
producer: handler,
mailbox_size: config.mailbox_size,
me: Some(config.public_key),
initial: config.initial,
timeout: config.timeout,
fetch_retry_timeout: config.fetch_retry_timeout,
priority_requests: config.priority_requests,
priority_responses: config.priority_responses,
},
);
resolver_engine.start(backfill);
(receiver, resolver)
}