use crate::dat::Dat;
use crate::dat_key::DatKey;
use crate::dat_kid::Kid;
use crate::dat_payload::DatPayload;
use crate::error::DatError;
use crate::signature_key::SignatureKeyOutOption;
use itertools::Itertools;
use std::sync::RwLock;
pub struct DatBank<T: Kid> {
issuance_key: RwLock<Option<DatKey<T>>>,
verifying_keys: RwLock<Vec<DatKey<T>>>,
}
impl <T: Kid> DatBank<T> {
pub fn new() -> Self {
DatBank {
issuance_key: RwLock::new(None),
verifying_keys: RwLock::new(vec![]),
}
}
pub fn to_dat(&self, plain: &str, secure: &str) -> Result<String, DatError> {
if let Some(key) = self.issuance_key.read().unwrap().as_ref() {
key.to_dat(plain, secure)
} else {
Err(DatError::SigningKeyNotExists)
}
}
pub fn to_payload(&self, dat: String) -> Result<DatPayload, DatError> {
let dat: Dat<T> = dat.try_into()?;
let kid = dat.kid();
if let Some(key) = self.verifying_keys.read().unwrap().iter().find(|e| &e.kid == kid) {
key.to_payload(dat)
} else {
Err(DatError::KidNotFound)
}
}
pub fn to_payload_without_verify(&self, dat: String) -> Result<DatPayload, DatError> {
let dat: Dat<T> = dat.try_into()?;
let kid = dat.kid();
if let Some(key) = self.verifying_keys.read().unwrap().iter().find(|e| &e.kid == kid) {
key.to_payload_without_verify(dat)
} else {
Err(DatError::KidNotFound)
}
}
pub fn export_kids(&self) -> Vec<T> {
self.verifying_keys.read().unwrap().iter().map(|key| key.kid.clone()).collect()
}
pub fn export(&self, signature_key_out_option: SignatureKeyOutOption) -> String {
self.verifying_keys.read().unwrap().iter().map(|key| key.format(signature_key_out_option).unwrap()).join("\n")
}
pub fn export_keys(&self) -> Vec<DatKey<T>> {
self.verifying_keys.read().unwrap().clone()
}
pub fn import(&self, format: &str, clear: bool) -> Result<(), DatError> {
let new_keys = format.lines()
.filter(|e| !e.is_empty())
.map(|e| e.parse::<DatKey<T>>())
.collect::<Result<Vec<DatKey<T>>, DatError>>()?;
self.import_keys(new_keys, clear)
}
pub fn import_keys(&self, new_keys: Vec<DatKey<T>>, clear: bool) -> Result<(), DatError> {
let mut keys = if clear {
vec![]
} else {
self.verifying_keys.read().unwrap().clone()
};
for key in new_keys {
if !keys.contains(&key) {
keys.push(key);
}
}
let keys = keys.into_iter()
.filter(|key| !key.expired_verifying())
.sorted_by(|a, b| a.issue_begin.cmp(&b.issue_begin))
.collect::<Vec<DatKey<T>>>();
let issuance_key_set: Option<DatKey<T>> = keys.iter()
.rev()
.find(|e| e.issuable())
.cloned();
*self.issuance_key.write().unwrap() = issuance_key_set;
*self.verifying_keys.write().unwrap() = keys;
Ok(())
}
}