chore: Add a extension trait for Tags type

Signed-off-by: Awiteb <a@4rs.nl>
This commit is contained in:
Awiteb
2025-05-05 14:03:13 +00:00
parent fb77a8ca14
commit 99e46d7633
3 changed files with 49 additions and 25 deletions

View File

@@ -14,6 +14,9 @@
// You should have received a copy of the GNU General Public License // You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://gnu.org/licenses/gpl-3.0.html>. // along with this program. If not, see <https://gnu.org/licenses/gpl-3.0.html>.
/// Extension traits for nostr types.
pub mod traits;
/// Utility functions for nostr.
pub mod utils; pub mod utils;
use std::time::Duration; use std::time::Duration;

36
src/nostr_utils/traits.rs Normal file
View File

@@ -0,0 +1,36 @@
// n34 - A CLI to interact with NIP-34 and other stuff related to codes in nostr
// Copyright (C) 2025 Awiteb <a@4rs.nl>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://gnu.org/licenses/gpl-3.0.html>.
use nostr::event::{TagKind, TagStandard, Tags};
/// A trait to add helper instance function to [`Tags`] type
#[easy_ext::ext(TagsExt)]
impl Tags {
/// Search for the given tag and map it value to a function
pub fn map_tag<T>(&self, kind: TagKind, f: impl FnOnce(&TagStandard) -> T) -> Option<T> {
self.find_standardized(kind).map(f)
}
/// Search for the given tag and map it value to a function. If the tag not
/// found return the default `T`
pub fn dmap_tag<T>(&self, kind: TagKind, f: impl FnOnce(&TagStandard) -> T) -> T
where
T: Default,
{
self.map_tag(kind, f).unwrap_or_default()
}
}

View File

@@ -21,6 +21,8 @@ use nostr::{
nips::nip34::GitRepositoryAnnouncement, nips::nip34::GitRepositoryAnnouncement,
}; };
use super::traits::TagsExt;
/// Returns the value of the given tag /// Returns the value of the given tag
fn tag_value(tag: &TagStandard) -> String { fn tag_value(tag: &TagStandard) -> String {
tag.clone().to_vec().remove(1) tag.clone().to_vec().remove(1)
@@ -48,33 +50,16 @@ where
/// Convert [`Event`] to [`GitRepositoryAnnouncement`] /// Convert [`Event`] to [`GitRepositoryAnnouncement`]
pub fn event_into_repo(event: Event, repo_id: impl Into<String>) -> GitRepositoryAnnouncement { pub fn event_into_repo(event: Event, repo_id: impl Into<String>) -> GitRepositoryAnnouncement {
let tags = &event.tags;
GitRepositoryAnnouncement { GitRepositoryAnnouncement {
id: repo_id.into(), id: repo_id.into(),
name: event.tags.find_standardized(TagKind::Name).map(tag_value), name: tags.map_tag(TagKind::Name, tag_value),
description: event description: tags.map_tag(TagKind::Description, tag_value),
.tags
.find_standardized(TagKind::Description)
.map(tag_value),
web: event
.tags
.find_standardized(TagKind::Web)
.map(tag_values)
.unwrap_or_default(),
clone: event
.tags
.find_standardized(TagKind::Clone)
.map(tag_values)
.unwrap_or_default(),
relays: event
.tags
.find_standardized(TagKind::Relays)
.map(tag_values)
.unwrap_or_default(),
euc: None, euc: None,
maintainers: event web: tags.dmap_tag(TagKind::Web, tag_values),
.tags clone: tags.dmap_tag(TagKind::Clone, tag_values),
.find_standardized(TagKind::Maintainers) relays: tags.dmap_tag(TagKind::Relays, tag_values),
.map(tag_values) maintainers: tags.dmap_tag(TagKind::Maintainers, tag_values),
.unwrap_or_default(),
} }
} }