kristoferssolo/tg-relay-rs
1
0
Fork 0
mirror of https://github.com/kristoferssolo/tg-relay-rs.git synced 2025-12-20 11:04:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor(handlers): make lazy_regex macro

Browse Source
This commit is contained in:
Kristofers Solo 2025-10-27 11:51:15 +02:00
parent 89dec3387b
commit 90805674c6
Signed by: kristoferssolo
GPG Key ID: 74FF8144483D82C8
5 changed files with 35 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/handlers/instagram.rs
View File

@ -1,21 +1,14 @@
use crate::download::{download_instagram, process_download_result};
use crate::error::Result;
use crate::handlers::SocialHandler;
use regex::Regex;
use std::sync::OnceLock;
use crate::lazy_regex;
use teloxide::{Bot, types::ChatId};
use tracing::info;
static SHORTCODE_RE: OnceLock<Regex> = OnceLock::new();
fn shortcode_regex() -> &'static Regex {
SHORTCODE_RE.get_or_init(|| {
Regex::new(
r"https?://(?:www\.)?(?:instagram\.com|instagr\.am)/(?:p|reel|tv)/([A-Za-z0-9_-]+)",
)
.expect("filed to compile regex")
})
}
lazy_regex!(
URL_RE,
r#"https?://(?:www\.)?(?:instagram\.com|instagr\.am)/(?:p|reel|tv)/([A-Za-z0-9_-]+)"#
);
/// Handler for Instagram posts / reels / tv
#[derive(Clone, Default)]
@ -36,7 +29,7 @@ impl SocialHandler for InstagramHandler {
}
fn try_extract(&self, text: &str) -> Option<String> {
shortcode_regex()
regex()
.captures(text)
.and_then(|c| c.get(0).map(|m| m.as_str().to_owned()))
}

11
src/handlers/mod.rs
View File

@ -19,6 +19,17 @@ pub use twitter::TwitterHandler;
#[cfg(feature = "youtube")]
pub use youtube::YouTubeShortsHandler;
#[macro_export]
macro_rules! lazy_regex {
($name:ident, $pattern:expr) => {
static $name: std::sync::OnceLock<regex::Regex> = std::sync::OnceLock::new();
fn regex() -> &'static regex::Regex {
$name.get_or_init(|| regex::Regex::new($pattern).expect("failed to compile regex"))
}
};
}
#[async_trait::async_trait]
pub trait SocialHandler: Send + Sync {
/// Short name used for logging etc.

17
src/handlers/tiktok.rs
View File

@ -1,22 +1,17 @@
use crate::{
download::{download_tiktok, process_download_result},
error::Result,
lazy_regex,
};
use regex::Regex;
use std::sync::OnceLock;
use teloxide::{Bot, types::ChatId};
use tracing::info;
use crate::handlers::SocialHandler;
static SHORTCODE_RE: OnceLock<Regex> = OnceLock::new();
fn shortcode_regex() -> &'static Regex {
SHORTCODE_RE.get_or_init(|| {
Regex::new(r"https?://(?:www\.)?(?:vm|vt|tt|tik)\.tiktok\.com/([A-Za-z0-9_-]+)[/?#]?")
.expect("filed to compile regex")
})
}
lazy_regex!(
URL_RE,
r#"https?://(?:www\.)?(?:vm|vt|tt|tik)\.tiktok\.com/([A-Za-z0-9_-]+)[/?#]?"#
);
/// Handler for Tiktok
#[derive(Clone, Default)]
@ -37,7 +32,7 @@ impl SocialHandler for TiktokHandler {
}
fn try_extract(&self, text: &str) -> Option<String> {
shortcode_regex()
regex()
.captures(text)
.and_then(|c| c.get(0).map(|m| m.as_str().to_owned()))
}

19
src/handlers/twitter.rs
View File

@ -1,24 +1,17 @@
use crate::{
download::{download_twitter, process_download_result},
error::Result,
lazy_regex,
};
use regex::Regex;
use std::sync::OnceLock;
use teloxide::{Bot, types::ChatId};
use tracing::info;
use crate::handlers::SocialHandler;
static SHORTCODE_RE: OnceLock<Regex> = OnceLock::new();
fn shortcode_regex() -> &'static Regex {
SHORTCODE_RE.get_or_init(|| {
Regex::new(
r"https?://(?:www\.)?twitter\.com/([A-Za-z0-9_]+(?:/[A-Za-z0-9_]+)?)/status/(\d{1,20})",
)
.expect("filed to compile regex")
})
}
lazy_regex!(
URL_RE,
r#"https?://(?:www\.)?twitter\.com/([A-Za-z0-9_]+(?:/[A-Za-z0-9_]+)?)/status/(\d{1,20})"#
);
/// Handler for Tiktok
#[derive(Clone, Default)]
@ -39,7 +32,7 @@ impl SocialHandler for TwitterHandler {
}
fn try_extract(&self, text: &str) -> Option<String> {
shortcode_regex()
regex()
.captures(text)
.and_then(|c| c.get(0).map(|m| m.as_str().to_owned()))
}

17
src/handlers/youtube.rs
View File

@ -1,21 +1,16 @@
use crate::handlers::SocialHandler;
use crate::lazy_regex;
use crate::{
download::{download_ytdlp, process_download_result},
error::Result,
};
use regex::Regex;
use std::sync::OnceLock;
use teloxide::{Bot, types::ChatId};
use tracing::info;
static SHORTCODE_RE: OnceLock<Regex> = OnceLock::new();
fn shortcode_regex() -> &'static Regex {
SHORTCODE_RE.get_or_init(|| {
Regex::new(r"https?:\/\/(?:www\.)?youtube\.com\/shorts\/[A-Za-z0-9_-]+(?:\?[^\s]*)?")
.expect("filed to compile regex")
})
}
lazy_regex!(
URL_RE,
r#"https?:\/\/(?:www\.)?youtube\.com\/shorts\/[A-Za-z0-9_-]+(?:\?[^\s]*)?"#
);
/// Handler for `YouTube Shorts` (and short youtu.be links)
#[derive(Clone, Default)]
@ -36,7 +31,7 @@ impl SocialHandler for YouTubeShortsHandler {
}
fn try_extract(&self, text: &str) -> Option<String> {
shortcode_regex().find(text).map(|m| m.as_str().to_owned())
regex().find(text).map(|m| m.as_str().to_owned())
}
async fn handle(&self, bot: &Bot, chat_id: ChatId, url: String) -> Result<()> {