make some stuff configurable via env vars

This commit is contained in:
dusk 2023-07-17 14:18:29 +03:00
parent 13e7465c88
commit 20bde1f002
Signed by: dusk
SSH Key Fingerprint: SHA256:Abmvag+juovVufZTxyWY8KcVgrznxvBjQpJesv071Aw
4 changed files with 135 additions and 126 deletions

View File

@ -1,4 +0,0 @@
https://twitter.com/FFJ_OFF/status/1679888209705340928
https://twitter.com/rauchoi_ovu/status/1680486681663922177
https://twitter.com/lk0_71604/status/1680425494741938179
https://twitter.com/stellaofastra/status/1680217448900096001

79
src/data.rs Normal file
View File

@ -0,0 +1,79 @@
use std::{collections::HashMap, str::FromStr};
use http::Uri;
use crate::error::{AppError, AppResult};
#[derive(Clone)]
pub(crate) enum ArtKind {
Twitter,
Safebooru,
}
impl FromStr for ArtKind {
type Err = AppError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"twitter.com" => Ok(Self::Twitter),
"safebooru.org" => Ok(Self::Safebooru),
_ => Err("not support website".into()),
}
}
}
#[derive(Clone)]
pub(crate) struct Art {
pub(crate) url: Uri,
pub(crate) kind: ArtKind,
}
impl FromStr for Art {
type Err = AppError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let url: Uri = s.parse()?;
let kind: ArtKind = url.authority().unwrap().host().parse()?;
Ok(Self { url, kind })
}
}
pub(crate) struct Data {
// actual arts
art: Vec<Art>,
art_indices: HashMap<Uri, usize>,
}
impl Data {
pub(crate) fn parse(data: &str) -> AppResult<Self> {
let mut this = Self {
art: Default::default(),
art_indices: Default::default(),
};
for entry in data.lines() {
let art: Art = entry.parse()?;
this.art_indices.insert(art.url.clone(), this.art.len());
this.art.push(art);
}
Ok(this)
}
pub(crate) fn pick_random_art(&self) -> &Art {
let no = fastrand::usize(0..self.art.len());
&self.art[no]
}
pub(crate) fn reload(&mut self, data: &str) -> AppResult<()> {
for entry in data.lines() {
let art: Art = entry.parse()?;
if !self.art_indices.contains_key(&art.url) {
self.art_indices.insert(art.url.clone(), self.art.len());
self.art.push(art);
}
}
Ok(())
}
}

View File

@ -5,6 +5,8 @@ use http::StatusCode;
type BoxedError = Box<dyn std::error::Error>; type BoxedError = Box<dyn std::error::Error>;
pub(crate) type AppResult<T> = Result<T, AppError>;
#[derive(Debug)] #[derive(Debug)]
pub(crate) struct AppError { pub(crate) struct AppError {
internal: BoxedError, internal: BoxedError,

View File

@ -5,133 +5,21 @@ use axum::{
Router, Router,
}; };
use dashmap::DashMap; use dashmap::DashMap;
use error::AppError; use data::{Art, ArtKind, Data};
use error::AppResult;
use http::Uri; use http::Uri;
use std::{ use std::{
collections::HashMap,
ops::Deref, ops::Deref,
str::FromStr,
sync::{Arc, Mutex}, sync::{Arc, Mutex},
}; };
mod data;
mod error; mod error;
type AppResult<T> = Result<T, AppError>;
#[derive(Clone)]
enum ArtKind {
Twitter,
Safebooru,
}
impl FromStr for ArtKind {
type Err = AppError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"twitter.com" => Ok(Self::Twitter),
"safebooru.org" => Ok(Self::Safebooru),
_ => Err("not support website".into()),
}
}
}
#[derive(Clone)]
struct Art {
url: Uri,
kind: ArtKind,
}
impl FromStr for Art {
type Err = AppError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let url: Uri = s.parse()?;
let kind: ArtKind = url.authority().unwrap().host().parse()?;
Ok(Self { url, kind })
}
}
struct Data {
// actual arts
art: Vec<Art>,
art_indices: HashMap<Uri, usize>,
}
impl Data {
fn parse(data: &str) -> AppResult<Self> {
let mut this = Self {
art: Default::default(),
art_indices: Default::default(),
};
for entry in data.lines() {
let art: Art = entry.parse()?;
this.art_indices.insert(art.url.clone(), this.art.len());
this.art.push(art);
}
Ok(this)
}
fn pick_random_art(&self) -> &Art {
let no = fastrand::usize(0..self.art.len());
&self.art[no]
}
fn reload(&mut self, data: &str) -> AppResult<()> {
for entry in data.lines() {
let art: Art = entry.parse()?;
if !self.art_indices.contains_key(&art.url) {
self.art_indices.insert(art.url.clone(), self.art.len());
self.art.push(art);
}
}
Ok(())
}
}
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
}
}
const ARTS_PATH: &str = "arts.txt";
#[tokio::main] #[tokio::main]
async fn main() { async fn main() {
let arts = std::fs::read_to_string(ARTS_PATH).unwrap(); let arts_file_path = get_conf("ARTS_PATH");
let arts = std::fs::read_to_string(&arts_file_path).unwrap();
let state = AppState::new(Data::parse(&arts).unwrap()); let state = AppState::new(Data::parse(&arts).unwrap());
std::thread::spawn({ std::thread::spawn({
@ -141,7 +29,7 @@ async fn main() {
move || { move || {
let mut signals = Signals::new(&[SIGUSR2]).unwrap(); let mut signals = Signals::new(&[SIGUSR2]).unwrap();
for _ in signals.forever() { for _ in signals.forever() {
let data = std::fs::read_to_string(ARTS_PATH).unwrap(); let data = std::fs::read_to_string(&arts_file_path).unwrap();
state.data.lock().unwrap().reload(&data).unwrap(); state.data.lock().unwrap().reload(&data).unwrap();
} }
} }
@ -176,6 +64,11 @@ async fn show_art(state: State<AppState>) -> AppResult<axum::response::Response>
} }
fn render_page(art: &Art, image_link: &str) -> Html<String> { fn render_page(art: &Art, image_link: &str) -> Html<String> {
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");
let body_style = let body_style =
"margin: 0px; background: #0e0e0e; height: 100vh; width: 100vw; display: flex;"; "margin: 0px; background: #0e0e0e; height: 100vh; width: 100vw; display: flex;";
let img_style = "display: block; margin: auto; max-height: 100vh; max-width: 100vw;"; let img_style = "display: block; margin: auto; max-height: 100vh; max-width: 100vw;";
@ -184,10 +77,10 @@ fn render_page(art: &Art, image_link: &str) -> Html<String> {
(maud::DOCTYPE) (maud::DOCTYPE)
head { head {
meta charset="utf8"; meta charset="utf8";
meta property="og:title" content="random pm art here!!"; meta property="og:title" content=(embed_title);
meta property="og:description" content="click NOW to see a random PM art"; meta property="og:description" content=(embed_content);
meta name="theme-color" content="#bd0000"; meta name="theme-color" content=(embed_color);
title { "random pm art" } title { (title) }
} }
body style=(body_style) { body style=(body_style) {
img style=(img_style) src=(image_link); img style=(img_style) src=(image_link);
@ -243,3 +136,42 @@ async fn fetch_twitter_image_link(http: &reqwest::Client, url: &Uri) -> AppResul
.to_owned(); .to_owned();
Ok(link) Ok(link)
} }
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
}
}