use std::collections::HashMap;
use std::iter::FromIterator;
use serde_json::{json, Value};
use crate::{data::PlutusDataJson, WError};
#[derive(Clone, Debug, PartialEq)]
pub struct Map<K, V>
where
K: Clone + PlutusDataJson,
V: Clone + PlutusDataJson,
{
pub map: Vec<(K, V)>,
}
impl<K, V> Map<K, V>
where
K: Clone + PlutusDataJson,
V: Clone + PlutusDataJson,
{
pub fn new(map_items: &[(K, V)]) -> Self {
Map {
map: map_items.to_vec(),
}
}
pub fn from_map(hash_map: HashMap<K, V>) -> Self {
Map {
map: hash_map.into_iter().collect(),
}
}
pub fn insert(&mut self, key: K, value: V) {
self.map.push((key, value));
}
}
impl<K, V> FromIterator<(K, V)> for Map<K, V>
where
K: Clone + PlutusDataJson,
V: Clone + PlutusDataJson,
{
fn from_iter<I: IntoIterator<Item = (K, V)>>(iter: I) -> Self {
let mut map = Map { map: Vec::new() };
for (key, value) in iter {
map.insert(key, value);
}
map
}
}
impl<K, V> PlutusDataJson for Map<K, V>
where
K: Clone + PlutusDataJson,
V: Clone + PlutusDataJson,
{
fn to_json(&self) -> Value {
let map_items_json: Vec<(Value, Value)> = self
.map
.iter()
.map(|(k, v)| (k.clone().to_json(), v.clone().to_json()))
.collect();
pairs(map_items_json)
}
fn from_json(value: &Value) -> Result<Self, WError> {
let map_json = value
.get("map")
.ok_or_else(|| WError::new("Map::from_json", "missing 'map' field"))?
.as_array()
.ok_or_else(|| WError::new("Map::from_json", "invalid 'map' value"))?;
let map = map_json
.iter()
.enumerate()
.map(|(i, item)| {
let k_value = item
.get("k")
.ok_or_else(|| WError::new("Map::from_json", "missing 'k' in map entry"))?;
let v_value = item
.get("v")
.ok_or_else(|| WError::new("Map::from_json", "missing 'v' in map entry"))?;
let k = K::from_json(k_value).map_err(WError::add_err_trace(
Box::leak(format!("Map::from_json[{}].k", i).into_boxed_str())
))?;
let v = V::from_json(v_value).map_err(WError::add_err_trace(
Box::leak(format!("Map::from_json[{}].v", i).into_boxed_str())
))?;
Ok((k, v))
})
.collect::<Result<Vec<(K, V)>, WError>>()?;
Ok(Map { map })
}
}
pub fn pairs<K: Into<Value>, V: Into<Value>>(items_map: Vec<(K, V)>) -> Value {
let map: Vec<Value> = items_map
.into_iter()
.map(|(k, v)| json!({"k": k.into(), "v": v.into()}))
.collect();
json!({ "map": map })
}
pub fn assoc_map<K: Into<Value>, V: Into<Value>>(items_map: Vec<(K, V)>) -> Value {
pairs(items_map)
}