50 lines
1.8 KiB
Rust
50 lines
1.8 KiB
Rust
use async_trait::async_trait;
|
|
use fancy_regex::{Captures, Regex};
|
|
use reqwest::Client;
|
|
use htmlescape::decode_html;
|
|
use crate::bot::{Context, Trigger};
|
|
|
|
pub struct Title {
|
|
http: Client,
|
|
title_regex: Regex
|
|
}
|
|
|
|
impl Title {
|
|
pub fn new() -> anyhow::Result<Self> {
|
|
Ok(Title {
|
|
http: Client::new(),
|
|
title_regex: Regex::new(r"(?<=<title>)(.*)(?=</title>)")?
|
|
})
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl Trigger for Title {
|
|
async fn execute<'a>(&mut self, _msg: Context<'a>, captures: Captures<'a>) -> anyhow::Result<String> {
|
|
let url = captures.get(0).unwrap().as_str();
|
|
tracing::debug!("url: {}", url);
|
|
|
|
let request = self.http.get(url).build()?;
|
|
let response = self.http.execute(request).await?;
|
|
let headers = response.headers();
|
|
return if let Some(header) = headers.get("Content-Type") {
|
|
let mime = header.to_str()?;
|
|
if mime.contains("text/html") {
|
|
let body = response.text().await?;
|
|
if let Some(tm) = self.title_regex.find(&body)? {
|
|
let title_match = &body[tm.start()..tm.end()];
|
|
let result = decode_html(title_match).unwrap_or_else(|_| title_match.to_string());
|
|
Ok(format!("\x039[Title]\x0311 {}", result))
|
|
} else {
|
|
Ok("\x039[Title]\x0311 No title".into())
|
|
}
|
|
} else {
|
|
let content_length = response.content_length().map(|l| (l/1024).to_string());
|
|
let size = content_length.as_deref().unwrap_or("unknown");
|
|
Ok(format!("\x039[Title]\x0311 File: {}; {}kb", mime, size))
|
|
}
|
|
} else {
|
|
Ok("\x039[Title]\x0311 No Content-Type header".into())
|
|
}
|
|
}
|
|
} |