use alloc::string::{String, ToString};
use alloc::vec;
use crate::error::Error;
use crate::event::{Tag, impl_tag_codec_conversions};
use crate::nips::util::{
missing_tag_kind, take_and_parse_from_str, take_string, take_timestamp, unknown_tag,
};
use crate::types::Timestamp;
use crate::types::url::Url;
const TITLE: &str = "title";
const IMAGE: &str = "image";
const SUMMARY: &str = "summary";
const PUBLISHED_AT: &str = "published_at";
const HASHTAG: &str = "t";
#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub enum Nip23Tag {
Title(String),
Image(Url),
Summary(String),
PublishedAt(Timestamp),
Hashtag(String),
}
impl_tag_codec_conversions! {
Nip23Tag,
fn parse(tag) {
let mut iter = tag.into_iter();
let kind: S = iter.next().ok_or(missing_tag_kind())?;
match kind.as_ref() {
TITLE => Ok(Self::Title(take_string(&mut iter, "title")?)),
IMAGE => {
let url = parse_image_tag(iter)?;
Ok(Self::Image(url))
}
SUMMARY => Ok(Self::Summary(take_string(&mut iter, "summary")?)),
PUBLISHED_AT => {
let timestamp: Timestamp = take_timestamp(&mut iter)?;
Ok(Self::PublishedAt(timestamp))
}
HASHTAG => Ok(Self::Hashtag(
take_string(&mut iter, "hashtag")?.to_lowercase(),
)),
_ => Err(unknown_tag()),
}
}
fn to_tag(&self) {
match self {
Self::Title(title) => Tag::new(vec![String::from(TITLE), title.clone()]),
Self::Image(url) => Tag::new(vec![String::from(IMAGE), url.to_string()]),
Self::Summary(summary) => Tag::new(vec![String::from(SUMMARY), summary.clone()]),
Self::PublishedAt(timestamp) => {
Tag::new(vec![String::from(PUBLISHED_AT), timestamp.to_string()])
}
Self::Hashtag(hashtag) => Tag::new(vec![String::from(HASHTAG), hashtag.to_lowercase()]),
}
}
}
fn parse_image_tag<T, S>(mut iter: T) -> Result<Url, Error>
where
T: Iterator<Item = S>,
S: AsRef<str>,
{
let url: Url = take_and_parse_from_str(&mut iter, "image URL")?;
Ok(url)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_parse_title_tag() {
let tag = vec!["title", "Lorem Ipsum"];
let parsed = Nip23Tag::parse(&tag).unwrap();
assert_eq!(parsed, Nip23Tag::Title(String::from("Lorem Ipsum")));
assert_eq!(parsed.to_tag(), Tag::parse(tag).unwrap());
}
#[test]
fn test_parse_image_tag() {
let tag = vec!["image", "https://example.com/image.jpg"];
let parsed = Nip23Tag::parse(&tag).unwrap();
assert_eq!(
parsed,
Nip23Tag::Image(Url::parse("https://example.com/image.jpg").unwrap())
);
assert_eq!(parsed.to_tag(), Tag::parse(tag).unwrap());
}
#[test]
fn test_parse_summary_tag() {
let tag = vec!["summary", "Article summary"];
let parsed = Nip23Tag::parse(&tag).unwrap();
assert_eq!(parsed, Nip23Tag::Summary(String::from("Article summary")));
assert_eq!(parsed.to_tag(), Tag::parse(tag).unwrap());
}
#[test]
fn test_parse_published_at_tag() {
let tag = vec!["published_at", "1296962229"];
let parsed = Nip23Tag::parse(&tag).unwrap();
assert_eq!(parsed, Nip23Tag::PublishedAt(Timestamp::from(1296962229)));
assert_eq!(parsed.to_tag(), Tag::parse(tag).unwrap());
}
#[test]
fn test_parse_hashtag_tag() {
let tag = vec!["t", "Placeholder"];
let parsed = Nip23Tag::parse(&tag).unwrap();
assert_eq!(parsed, Nip23Tag::Hashtag(String::from("placeholder")));
assert_eq!(
parsed.to_tag(),
Tag::parse(vec!["t", "placeholder"]).unwrap()
);
}
}