lib

encode.rs (5247B)

---

 #[cfg(not(feature = "std"))]
 use alloc::{
     string::{String, ToString},
     vec,
     vec::Vec,
 };
 
 use radroots_event::{
     envelope::kind::KIND_RESOURCE_AREA, farm::resource_cap::ResourceHarvestCap, tag::name::TAG_D,
 };
 
 use crate::d_tag::validate_d_tag;
 use crate::error::EventEncodeError;
 
 #[cfg(feature = "json")]
 use radroots_event::envelope::kind::KIND_RESOURCE_HARVEST_CAP;
 #[cfg(feature = "json")]
 use radroots_event::wire::Nip01EventWireParts;
 
 const TAG_A: &str = "a";
 const TAG_P: &str = "p";
 const TAG_T: &str = "t";
 const TAG_KEY: &str = "key";
 const TAG_CATEGORY: &str = "category";
 const TAG_START: &str = "start";
 const TAG_END: &str = "end";
 
 fn push_tag(tags: &mut Vec<Vec<String>>, key: &str, value: &str) {
     tags.push(vec![key.to_string(), value.to_string()]);
 }
 
 fn resource_area_address(cap: &ResourceHarvestCap) -> Result<String, EventEncodeError> {
     let area = &cap.resource_area;
     if area.pubkey.trim().is_empty() {
         return Err(EventEncodeError::EmptyRequiredField("resource_area.pubkey"));
     }
     if area.d_tag.trim().is_empty() {
         return Err(EventEncodeError::EmptyRequiredField("resource_area.d_tag"));
     }
     validate_d_tag(&area.d_tag, "resource_area.d_tag")?;
     let mut addr = String::new();
     addr.push_str(&KIND_RESOURCE_AREA.to_string());
     addr.push(':');
     addr.push_str(&area.pubkey);
     addr.push(':');
     addr.push_str(&area.d_tag);
     Ok(addr)
 }
 
 pub fn resource_harvest_cap_build_tags(
     cap: &ResourceHarvestCap,
 ) -> Result<Vec<Vec<String>>, EventEncodeError> {
     if cap.d_tag.trim().is_empty() {
         return Err(EventEncodeError::EmptyRequiredField("d_tag"));
     }
     validate_d_tag(&cap.d_tag, "d_tag")?;
     if cap.product.key.trim().is_empty() {
         return Err(EventEncodeError::EmptyRequiredField("product.key"));
     }
     let mut tags = Vec::new();
     push_tag(&mut tags, TAG_D, &cap.d_tag);
     let addr = resource_area_address(cap)?;
     push_tag(&mut tags, TAG_A, &addr);
     push_tag(&mut tags, TAG_P, &cap.resource_area.pubkey);
     push_tag(&mut tags, TAG_KEY, &cap.product.key);
     if let Some(category) = cap.product.category.as_deref()
         && !category.trim().is_empty()
     {
         push_tag(&mut tags, TAG_CATEGORY, category);
     }
     push_tag(&mut tags, TAG_START, &cap.start.to_string());
     push_tag(&mut tags, TAG_END, &cap.end.to_string());
     if let Some(items) = cap.tags.as_ref() {
         for item in items.iter().filter(|v| !v.trim().is_empty()) {
             push_tag(&mut tags, TAG_T, item);
         }
     }
     Ok(tags)
 }
 
 #[cfg(feature = "json")]
 pub fn to_wire_parts(cap: &ResourceHarvestCap) -> Result<Nip01EventWireParts, EventEncodeError> {
     to_wire_parts_with_kind(cap, KIND_RESOURCE_HARVEST_CAP)
 }
 
 #[cfg(feature = "json")]
 pub fn to_wire_parts_with_kind(
     cap: &ResourceHarvestCap,
     kind: u32,
 ) -> Result<Nip01EventWireParts, EventEncodeError> {
     if kind != KIND_RESOURCE_HARVEST_CAP {
         return Err(EventEncodeError::InvalidKind(kind));
     }
     let tags = resource_harvest_cap_build_tags(cap)?;
     let content = serde_json::to_string(cap).map_err(|_| EventEncodeError::Json)?;
     Ok(Nip01EventWireParts {
         kind,
         content,
         tags,
     })
 }
 
 #[cfg(test)]
 mod tests {
     use super::*;
     use crate::test_fixtures::FIXTURE_ALICE_PUBLIC_KEY_HEX;
     use radroots_core::{Decimal, Quantity, Unit};
     use radroots_event::farm::resource_area::ResourceAreaRef;
     use radroots_event::farm::resource_cap::ResourceHarvestProduct;
 
     fn sample_cap_with_category(category: Option<&str>) -> ResourceHarvestCap {
         ResourceHarvestCap {
             d_tag: "AAAAAAAAAAAAAAAAAAAABA".to_string(),
             resource_area: ResourceAreaRef {
                 pubkey: FIXTURE_ALICE_PUBLIC_KEY_HEX.to_string(),
                 d_tag: "AAAAAAAAAAAAAAAAAAAAAw".to_string(),
             },
             product: ResourceHarvestProduct {
                 key: "nutmeg".to_string(),
                 category: category.map(|value| value.to_string()),
             },
             start: 1,
             end: 2,
             cap_quantity: Quantity::try_new(Decimal::from(1000u32), Unit::MassG).unwrap(),
             display_amount: None,
             display_unit: None,
             display_label: None,
             tags: None,
         }
     }
 
     #[test]
     fn resource_harvest_cap_build_tags_omits_blank_category() {
         let tags = resource_harvest_cap_build_tags(&sample_cap_with_category(Some(" ")))
             .expect("resource harvest cap tags");
         assert!(
             !tags
                 .iter()
                 .any(|tag| tag.first().map(|v| v.as_str()) == Some("category"))
         );
 
         let tags = resource_harvest_cap_build_tags(&sample_cap_with_category(None))
             .expect("resource harvest cap tags");
         assert!(
             !tags
                 .iter()
                 .any(|tag| tag.first().map(|v| v.as_str()) == Some("category"))
         );
 
         let tags = resource_harvest_cap_build_tags(&sample_cap_with_category(Some("spice")))
             .expect("resource harvest cap tags");
         assert!(
             tags.iter()
                 .any(|tag| tag.first().map(|v| v.as_str()) == Some("category"))
         );
     }
 }