2023-07-17 10:46:32 +03:00
|
|
|
use axum::{
|
|
|
|
extract::State,
|
|
|
|
response::{Html, IntoResponse},
|
|
|
|
routing::get,
|
|
|
|
Router,
|
|
|
|
};
|
2023-07-17 09:01:10 +03:00
|
|
|
use dashmap::DashMap;
|
2023-07-17 14:18:29 +03:00
|
|
|
use data::{Art, ArtKind, Data};
|
|
|
|
use error::AppResult;
|
2023-07-17 09:01:10 +03:00
|
|
|
use http::Uri;
|
2023-07-17 09:38:54 +03:00
|
|
|
use std::{
|
|
|
|
ops::Deref,
|
2023-07-17 10:46:32 +03:00
|
|
|
sync::{Arc, Mutex},
|
2023-07-17 09:38:54 +03:00
|
|
|
};
|
2023-07-17 09:01:10 +03:00
|
|
|
|
2023-07-17 14:18:29 +03:00
|
|
|
mod data;
|
2023-07-17 09:01:10 +03:00
|
|
|
mod error;
|
|
|
|
|
|
|
|
#[tokio::main]
|
|
|
|
async fn main() {
|
2023-07-17 14:18:29 +03:00
|
|
|
let arts_file_path = get_conf("ARTS_PATH");
|
|
|
|
let arts = std::fs::read_to_string(&arts_file_path).unwrap();
|
2023-07-17 09:38:54 +03:00
|
|
|
let state = AppState::new(Data::parse(&arts).unwrap());
|
|
|
|
|
|
|
|
std::thread::spawn({
|
|
|
|
use signal_hook::{consts::SIGUSR2, iterator::Signals};
|
|
|
|
|
|
|
|
let state = state.clone();
|
|
|
|
move || {
|
|
|
|
let mut signals = Signals::new(&[SIGUSR2]).unwrap();
|
|
|
|
for _ in signals.forever() {
|
2023-07-17 14:18:29 +03:00
|
|
|
let data = std::fs::read_to_string(&arts_file_path).unwrap();
|
2023-07-17 09:38:54 +03:00
|
|
|
state.data.lock().unwrap().reload(&data).unwrap();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
});
|
|
|
|
|
|
|
|
let app = Router::new().route("/", get(show_art)).with_state(state);
|
2023-07-17 09:01:10 +03:00
|
|
|
|
|
|
|
let listener = tokio::net::TcpListener::bind("127.0.0.1:3000")
|
|
|
|
.await
|
|
|
|
.unwrap();
|
|
|
|
println!("listening on {}", listener.local_addr().unwrap());
|
|
|
|
axum::serve(listener, app).await.unwrap();
|
|
|
|
}
|
|
|
|
|
2023-07-17 11:03:42 +03:00
|
|
|
async fn show_art(state: State<AppState>) -> AppResult<axum::response::Response> {
|
2023-07-17 09:38:54 +03:00
|
|
|
let art = state.data.lock().unwrap().pick_random_art().clone();
|
2023-07-17 10:46:32 +03:00
|
|
|
let image_link = if let Some(image_link) = state.direct_links.get(&art.url) {
|
|
|
|
image_link.to_string()
|
2023-07-17 09:01:10 +03:00
|
|
|
} else {
|
|
|
|
let image_link = match art.kind {
|
|
|
|
ArtKind::Twitter => fetch_twitter_image_link(&state.http, &art.url).await?,
|
2023-07-17 13:58:47 +03:00
|
|
|
ArtKind::Safebooru => fetch_safebooru_image_link(&state.http, &art.url).await?,
|
2023-07-17 09:01:10 +03:00
|
|
|
};
|
2023-07-17 10:46:32 +03:00
|
|
|
state
|
|
|
|
.direct_links
|
|
|
|
.insert(art.url.clone(), image_link.clone());
|
|
|
|
image_link
|
|
|
|
};
|
|
|
|
|
|
|
|
let page = render_page(&art, &image_link);
|
|
|
|
Ok(page.into_response())
|
2023-07-17 09:01:10 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
fn render_page(art: &Art, image_link: &str) -> Html<String> {
|
2023-07-17 14:18:29 +03:00
|
|
|
let title = get_conf("SITE_TITLE");
|
|
|
|
let embed_title = get_conf("EMBED_TITLE");
|
|
|
|
let embed_content = get_conf("EMBED_DESC");
|
|
|
|
let embed_color = get_conf("EMBED_COLOR");
|
|
|
|
|
2023-07-17 09:01:10 +03:00
|
|
|
let body_style =
|
|
|
|
"margin: 0px; background: #0e0e0e; height: 100vh; width: 100vw; display: flex;";
|
|
|
|
let img_style = "display: block; margin: auto; max-height: 100vh; max-width: 100vw;";
|
|
|
|
let about_style = "position: absolute; bottom: 0; font-size: 0.75em; color: #ffffff; background-color: #0e0e0eaa;";
|
|
|
|
let content = maud::html! {
|
|
|
|
(maud::DOCTYPE)
|
|
|
|
head {
|
|
|
|
meta charset="utf8";
|
2023-07-17 14:18:29 +03:00
|
|
|
meta property="og:title" content=(embed_title);
|
|
|
|
meta property="og:description" content=(embed_content);
|
|
|
|
meta name="theme-color" content=(embed_color);
|
|
|
|
title { (title) }
|
2023-07-17 09:01:10 +03:00
|
|
|
}
|
|
|
|
body style=(body_style) {
|
|
|
|
img style=(img_style) src=(image_link);
|
|
|
|
a style=(format!("{about_style} left: 0;")) href=(art.url) target="_blank" {
|
|
|
|
"source: " (art.url)
|
|
|
|
}
|
|
|
|
a style=(format!("{about_style} right: 0;")) href="https://gaze.systems" target="_blank" {
|
|
|
|
"website made by dusk"
|
2023-07-18 01:00:16 +03:00
|
|
|
br;
|
|
|
|
"report problems / feedback @ yusdacra on Discord"
|
2023-07-17 09:01:10 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
Html(content.into_string())
|
|
|
|
}
|
|
|
|
|
2023-07-17 13:58:47 +03:00
|
|
|
async fn fetch_safebooru_image_link(http: &reqwest::Client, url: &Uri) -> AppResult<String> {
|
|
|
|
let mut id = String::new();
|
|
|
|
for (name, value) in form_urlencoded::parse(url.query().unwrap().as_bytes()) {
|
|
|
|
if name == "id" {
|
|
|
|
id = value.into_owned();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if id.is_empty() {
|
|
|
|
return Err("no id?".into());
|
|
|
|
}
|
|
|
|
|
|
|
|
let url = format!("https://safebooru.org/index.php?page=dapi&s=post&q=index&json=1&id={id}");
|
|
|
|
let req = http.get(url).build()?;
|
|
|
|
let resp = http.execute(req).await?.error_for_status()?;
|
|
|
|
let data: Vec<serde_json::Map<String, serde_json::Value>> = resp.json().await?;
|
|
|
|
|
|
|
|
let image_filename = data[0].get("image").unwrap().as_str().unwrap();
|
|
|
|
let image_directory = data[0].get("directory").unwrap().as_str().unwrap();
|
|
|
|
|
|
|
|
Ok(format!(
|
|
|
|
"http://safebooru.org/images/{image_directory}/{image_filename}"
|
|
|
|
))
|
|
|
|
}
|
|
|
|
|
2023-07-17 09:01:10 +03:00
|
|
|
async fn fetch_twitter_image_link(http: &reqwest::Client, url: &Uri) -> AppResult<String> {
|
|
|
|
let fxurl = Uri::builder()
|
|
|
|
.scheme("https")
|
|
|
|
.authority("d.fxtwitter.com")
|
|
|
|
.path_and_query(url.path_and_query().unwrap().clone())
|
|
|
|
.build()?
|
|
|
|
.to_string();
|
|
|
|
let req = http.get(fxurl).build()?;
|
|
|
|
let resp = http.execute(req).await?.error_for_status()?;
|
|
|
|
let link = resp
|
|
|
|
.headers()
|
|
|
|
.get(http::header::LOCATION)
|
|
|
|
.unwrap()
|
|
|
|
.to_str()?
|
|
|
|
.to_owned();
|
|
|
|
Ok(link)
|
2023-07-17 02:42:55 +03:00
|
|
|
}
|
2023-07-17 14:18:29 +03:00
|
|
|
|
|
|
|
fn get_conf(name: &str) -> String {
|
|
|
|
std::env::var(name).unwrap()
|
|
|
|
}
|
|
|
|
|
|
|
|
struct InternalAppState {
|
|
|
|
// cached direct links to images
|
|
|
|
direct_links: DashMap<Uri, String>,
|
|
|
|
data: Mutex<Data>,
|
|
|
|
http: reqwest::Client,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Clone)]
|
|
|
|
struct AppState {
|
|
|
|
internal: Arc<InternalAppState>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl AppState {
|
|
|
|
fn new(data: Data) -> Self {
|
|
|
|
Self {
|
|
|
|
internal: Arc::new(InternalAppState {
|
|
|
|
data: Mutex::new(data),
|
|
|
|
direct_links: Default::default(),
|
|
|
|
http: reqwest::ClientBuilder::new()
|
|
|
|
.redirect(reqwest::redirect::Policy::none())
|
|
|
|
.build()
|
|
|
|
.unwrap(),
|
|
|
|
}),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Deref for AppState {
|
|
|
|
type Target = InternalAppState;
|
|
|
|
|
|
|
|
fn deref(&self) -> &Self::Target {
|
|
|
|
&self.internal
|
|
|
|
}
|
|
|
|
}
|