use std::fmt::{Display, Formatter};
use bc_components::{Digest, DigestProvider, tags};
use dcbor::prelude::*;
#[derive(Debug, Clone)]
enum KnownValueName {
Static(&'static str),
Dynamic(String),
}
#[derive(Clone, Debug)]
pub struct KnownValue {
value: u64,
assigned_name: Option<KnownValueName>,
}
impl KnownValue {
pub fn new(value: u64) -> Self { Self { value, assigned_name: None } }
pub fn new_with_name<T: Into<u64>>(
value: T,
assigned_name: String,
) -> Self {
Self {
value: value.into(),
assigned_name: Some(KnownValueName::Dynamic(assigned_name)),
}
}
pub const fn new_with_static_name(value: u64, name: &'static str) -> Self {
Self {
value,
assigned_name: Some(KnownValueName::Static(name)),
}
}
pub fn value(&self) -> u64 { self.value }
pub fn assigned_name(&self) -> Option<&str> {
match &self.assigned_name {
Some(KnownValueName::Static(name)) => Some(name),
Some(KnownValueName::Dynamic(name)) => Some(name),
None => None,
}
}
pub fn name(&self) -> String {
match &self.assigned_name {
Some(KnownValueName::Static(name)) => name.to_string(),
Some(KnownValueName::Dynamic(name)) => name.clone(),
None => self.value.to_string(),
}
}
}
impl PartialEq for KnownValue {
fn eq(&self, other: &Self) -> bool { self.value == other.value }
}
impl Eq for KnownValue {}
impl std::hash::Hash for KnownValue {
fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
self.value.hash(state);
}
}
impl Display for KnownValue {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
match &self.assigned_name {
Some(KnownValueName::Static(name)) => write!(f, "{}", name),
Some(KnownValueName::Dynamic(name)) => write!(f, "{}", name),
None => write!(f, "{}", self.value),
}
}
}
impl DigestProvider for KnownValue {
fn digest(&self) -> Digest {
Digest::from_image(self.tagged_cbor().to_cbor_data())
}
}
impl CBORTagged for KnownValue {
fn cbor_tags() -> Vec<Tag> { tags_for_values(&[tags::TAG_KNOWN_VALUE]) }
}
impl From<KnownValue> for CBOR {
fn from(value: KnownValue) -> Self { value.tagged_cbor() }
}
impl TryFrom<CBOR> for KnownValue {
type Error = dcbor::Error;
fn try_from(cbor: CBOR) -> dcbor::Result<Self> {
Self::from_tagged_cbor(cbor)
}
}
impl CBORTaggedEncodable for KnownValue {
fn untagged_cbor(&self) -> CBOR { self.value.into() }
}
impl CBORTaggedDecodable for KnownValue {
fn from_untagged_cbor(cbor: CBOR) -> dcbor::Result<Self> {
let value = u64::try_from(cbor)?;
Ok(Self::new(value))
}
}
impl From<u64> for KnownValue {
fn from(value: u64) -> Self { KnownValue::new(value) }
}
impl From<i32> for KnownValue {
fn from(value: i32) -> Self { KnownValue::new(value as u64) }
}
impl From<usize> for KnownValue {
fn from(value: usize) -> Self { KnownValue::new(value as u64) }
}