use alloc::boxed::Box;
use alloc::collections::BTreeSet;
use alloc::string::String;
use serde::{Deserialize, Serialize};
use umbral_pre::Capsule;
use crate::address::Address;
use crate::conditions::Conditions;
use crate::message_kit::MessageKit;
use crate::versioning::{
messagepack_deserialize, messagepack_serialize, ProtocolObject, ProtocolObjectInner,
};
#[derive(PartialEq, Debug, Serialize, Deserialize)]
pub struct RetrievalKit {
pub capsule: Capsule,
pub queried_addresses: BTreeSet<Address>,
pub conditions: Option<Conditions>,
}
impl RetrievalKit {
pub fn from_message_kit(message_kit: &MessageKit) -> Self {
Self {
capsule: message_kit.capsule.clone(),
queried_addresses: BTreeSet::<Address>::new(),
conditions: message_kit.conditions.clone(),
}
}
pub fn new(
capsule: &Capsule,
queried_addresses: impl IntoIterator<Item = Address>,
conditions: Option<&Conditions>,
) -> Self {
Self {
capsule: capsule.clone(),
queried_addresses: queried_addresses.into_iter().collect(),
conditions: conditions.cloned(),
}
}
}
impl<'a> ProtocolObjectInner<'a> for RetrievalKit {
fn brand() -> [u8; 4] {
*b"RKit"
}
fn version() -> (u16, u16) {
(2, 0)
}
fn unversioned_to_bytes(&self) -> Box<[u8]> {
messagepack_serialize(&self)
}
fn unversioned_from_bytes(minor_version: u16, bytes: &[u8]) -> Option<Result<Self, String>> {
if minor_version == 0 {
Some(messagepack_deserialize(bytes))
} else {
None
}
}
}
impl<'a> ProtocolObject<'a> for RetrievalKit {}