use std::sync::LazyLock;
use super::StructuredPrinter;
use super::TagHandler;
use crate::common::get_tag_attr;
use crate::dummy::IdentityHandler;
use crate::markup5ever_rcdom;
use markup5ever_rcdom::Handle;
use regex::Regex;
static YOUTUBE_PATTERN: LazyLock<Regex> =
LazyLock::new(|| Regex::new(r"www\.youtube(?:-nocookie)?\.com/embed/([-\w]+)").unwrap());
static INSTAGRAM_PATTERN: LazyLock<Regex> =
LazyLock::new(|| Regex::new(r"www\.instagram\.com/p/([-\w]+)/embed").unwrap());
static VK_PATTERN: LazyLock<Regex> = LazyLock::new(|| {
Regex::new(r"vk\.com/video_ext\.php\?oid=(-?\d+)&id=(\d+)&hash=(.*)").unwrap()
});
#[allow(dead_code)]
static YANDEX_MUSIC_TRACK_PATTERN: LazyLock<Regex> =
LazyLock::new(|| Regex::new(r"https://music.yandex.ru/iframe/#track/(\d+)/(\d+)").unwrap());
#[allow(dead_code)]
static YANDEX_MUSIC_ALBUM_PATTERN: LazyLock<Regex> =
LazyLock::new(|| Regex::new(r"https://music.yandex.ru/iframe/#album/(\d+)").unwrap());
#[derive(Default)]
pub struct IframeHandler;
impl TagHandler for IframeHandler {
fn handle(&mut self, tag: &Handle, printer: &mut StructuredPrinter) {
printer.insert_newline();
printer.insert_newline();
let src = get_tag_attr(tag, "src");
if src.is_none() {
return;
}
let src = src.unwrap();
if let Some(capture) = YOUTUBE_PATTERN.captures(&src) {
let media_id = capture.get(1).map_or("", |m| m.as_str());
printer.append_str(&format!("[](https://www.youtube.com/watch?v={mid})", mid = media_id));
return;
}
if let Some(capture) = INSTAGRAM_PATTERN.captures(&src) {
let media_id = capture.get(1).map_or("", |m| m.as_str());
printer.append_str(&format!("[](https://www.instagram.com/p/{mid}/embed/)", mid = media_id));
return;
}
if let Some(capture) = VK_PATTERN.captures(&src) {
let owner_id = capture.get(1).map_or("", |m| m.as_str());
let video_id = capture.get(2).map_or("", |m| m.as_str());
let _hash = capture.get(3).map_or("", |m| m.as_str());
printer.append_str(&format!("[](https://vk.com/video{oid}_{vid})", oid = owner_id, vid = video_id));
return;
}
let mut identity = IdentityHandler;
identity.handle(tag, printer);
}
fn after_handle(&mut self, printer: &mut StructuredPrinter) {
printer.insert_newline();
printer.insert_newline();
}
fn skip_descendants(&self) -> bool {
true
}
}