use crate::types::Height;
use bytes::{Buf, BufMut};
use commonware_codec::{varint::UInt, Codec, EncodeSize, Error, Read, ReadExt, Write};
use commonware_cryptography::{Digest, Digestible, Hasher};
use std::fmt::Debug;
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct Block<D: Digest, C> {
pub context: C,
pub parent: D,
pub height: Height,
pub timestamp: u64,
digest: D,
}
impl<D: Digest, C: Codec> Block<D, C> {
fn compute_digest<H: Hasher<Digest = D>>(
context: &C,
parent: &D,
height: Height,
timestamp: u64,
) -> D {
let mut hasher = H::new();
hasher.update(parent);
hasher.update(&height.get().to_be_bytes());
hasher.update(&context.encode());
hasher.update(×tamp.to_be_bytes());
hasher.finalize()
}
pub fn new<H: Hasher<Digest = D>>(
context: C,
parent: D,
height: Height,
timestamp: u64,
) -> Self {
let digest = Self::compute_digest::<H>(&context, &parent, height, timestamp);
Self {
context,
parent,
height,
timestamp,
digest,
}
}
}
impl<D: Digest, C: Write> Write for Block<D, C> {
fn write(&self, writer: &mut impl BufMut) {
self.context.write(writer);
self.parent.write(writer);
self.height.write(writer);
UInt(self.timestamp).write(writer);
self.digest.write(writer);
}
}
impl<D: Digest, C: Read<Cfg = ()>> Read for Block<D, C> {
type Cfg = ();
fn read_cfg(reader: &mut impl Buf, _: &Self::Cfg) -> Result<Self, Error> {
let context = C::read(reader)?;
let parent = D::read(reader)?;
let height = Height::read(reader)?;
let timestamp = UInt::read(reader)?.into();
let digest = D::read(reader)?;
Ok(Self {
context,
parent,
height,
timestamp,
digest,
})
}
}
impl<D: Digest, C: EncodeSize> EncodeSize for Block<D, C> {
fn encode_size(&self) -> usize {
self.context.encode_size()
+ self.parent.encode_size()
+ self.height.encode_size()
+ UInt(self.timestamp).encode_size()
+ self.digest.encode_size()
}
}
impl<D: Digest, C: Clone + Send + Sync + 'static> Digestible for Block<D, C> {
type Digest = D;
fn digest(&self) -> D {
self.digest
}
}
impl<D: Digest, C: Clone + Send + Sync + 'static> crate::Heightable for Block<D, C> {
fn height(&self) -> Height {
self.height
}
}
impl<D: Digest, C: Codec<Cfg = ()> + Clone + Send + Sync + 'static> crate::Block for Block<D, C> {
fn parent(&self) -> Self::Digest {
self.parent
}
}
impl<D: Digest, C: Codec<Cfg = ()> + Clone + Send + Sync + 'static> crate::CertifiableBlock
for Block<D, C>
{
type Context = C;
fn context(&self) -> Self::Context {
self.context.clone()
}
}