use crate::source::{Query, ScopeError, Snapshot, equal_id};
use crate::timestamp::{Purpose, Verified};
use core::marker::PhantomData;
use structured_asn1::memory::{HashWork, MemoryError, OwnedMap, Work};
pub struct Coverage<N: HashWork, D, G, W: Work, I> {
snapshot: Snapshot<D, G>,
signature: N,
timestamp: N,
purpose: Purpose,
items: OwnedMap<N, bool, W>,
issuer: PhantomData<I>,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum CoverageError<N> {
Scope(ScopeError<N>),
Resource {
node: N,
cause: MemoryError,
},
Foreign {
node: N,
},
}
impl<N: Copy + Eq + HashWork, D: Copy + Eq, G: Copy + Eq, W: Work, I> Coverage<N, D, G, W, I> {
#[must_use]
pub fn new(
_issuer: I,
snapshot: Snapshot<D, G>,
signature: N,
timestamp: N,
purpose: Purpose,
items: OwnedMap<N, bool, W>,
) -> Self {
Self {
snapshot,
signature,
timestamp,
purpose,
items,
issuer: PhantomData,
}
}
pub fn selects<Q: Query<Id = N, Document = D, Generation = G>>(
&self,
node: N,
query: &Q,
) -> Result<bool, CoverageError<N>> {
if self.snapshot != query.snapshot() {
return Err(CoverageError::Foreign { node });
}
query.admit(node, 1).map_err(CoverageError::Scope)?;
self.items
.get(&node, query.execution())
.map(|value| value == Some(&true))
.map_err(|cause| CoverageError::Resource { node, cause })
}
pub fn covers<Q: Query<Id = N, Document = D, Generation = G>>(
&self,
evidence: &Verified<N, D, G, I>,
node: N,
query: &Q,
) -> Result<bool, CoverageError<N>> {
self.require_evidence(evidence, node, query)?;
self.selects(node, query)
}
pub fn require_evidence<Q: Query<Id = N, Document = D, Generation = G>>(
&self,
evidence: &Verified<N, D, G, I>,
node: N,
query: &Q,
) -> Result<(), CoverageError<N>> {
if self.snapshot != query.snapshot() {
return Err(CoverageError::Foreign { node });
}
query.admit(node, 1).map_err(CoverageError::Scope)?;
self.require_identity(
evidence.token().snapshot,
evidence.signature(),
evidence.property(),
evidence.purpose(),
node,
query.execution(),
)
}
fn require_identity<Q: Work>(
&self,
snapshot: Snapshot<D, G>,
signature: N,
timestamp: N,
purpose: Purpose,
node: N,
execution: &Q,
) -> Result<(), CoverageError<N>> {
let resource = |cause| CoverageError::Resource { node, cause };
if snapshot != self.snapshot
|| !equal_id(signature, self.signature, execution).map_err(resource)?
|| !equal_id(timestamp, self.timestamp, execution).map_err(resource)?
|| purpose != self.purpose
{
return Err(CoverageError::Foreign { node });
}
Ok(())
}
}
#[cfg(test)]
mod tests;