84 lines
2.1 KiB
Rust
84 lines
2.1 KiB
Rust
|
use crate::database::db_conn;
|
||
|
|
use crate::model::entity::meta;
|
||
|
|
use rand::prelude::*;
|
||
|
|
use sea_orm::{prelude::*, ActiveValue};
|
||
|
|
use std::sync::Mutex;
|
||
|
|
|
||
|
|
type Meta = meta::Model;
|
||
|
|
|
||
|
|
static CACHE: Mutex<Option<Meta>> = Mutex::new(None);
|
||
|
|
fn update_cache(meta: &Meta) {
|
||
|
|
let _ = CACHE.lock().map(|mut cache| *cache = Some(meta.clone()));
|
||
|
|
}
|
||
|
|
|
||
|
|
#[crate::export]
|
||
|
|
pub async fn fetch_meta(use_cache: bool) -> Result<Meta, DbErr> {
|
||
|
|
// try using cache
|
||
|
|
if use_cache {
|
||
|
|
if let Some(cache) = CACHE.lock().ok().and_then(|cache| cache.clone()) {
|
||
|
|
return Ok(cache);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// try fetching from db
|
||
|
|
let db = db_conn().await?;
|
||
|
|
let meta = meta::Entity::find().one(db).await?;
|
||
|
|
if let Some(meta) = meta {
|
||
|
|
update_cache(&meta);
|
||
|
|
return Ok(meta);
|
||
|
|
}
|
||
|
|
|
||
|
|
// create a new meta object and insert into db
|
||
|
|
let meta = meta::Entity::insert(meta::ActiveModel {
|
||
|
|
id: ActiveValue::Set("x".to_owned()),
|
||
|
|
..Default::default()
|
||
|
|
})
|
||
|
|
.exec_with_returning(db)
|
||
|
|
.await?;
|
||
|
|
update_cache(&meta);
|
||
|
|
Ok(meta)
|
||
|
|
}
|
||
|
|
|
||
|
|
#[crate::export(object)]
|
||
|
|
pub struct PugArgs {
|
||
|
|
pub img: Option<String>,
|
||
|
|
pub title: String,
|
||
|
|
pub instance_name: String,
|
||
|
|
pub desc: Option<String>,
|
||
|
|
pub icon: Option<String>,
|
||
|
|
pub splash_icon: Option<String>,
|
||
|
|
pub theme_color: Option<String>,
|
||
|
|
pub random_motd: String,
|
||
|
|
pub private_mode: Option<bool>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[crate::export]
|
||
|
|
pub fn meta_to_pug_args(meta: Meta) -> PugArgs {
|
||
|
|
let mut rng = rand::thread_rng();
|
||
|
|
|
||
|
|
let splash_icon = meta
|
||
|
|
.custom_splash_icons
|
||
|
|
.choose(&mut rng)
|
||
|
|
.map(|s| s.to_owned())
|
||
|
|
.or_else(|| meta.icon_url.to_owned());
|
||
|
|
|
||
|
|
let random_motd = meta
|
||
|
|
.custom_motd
|
||
|
|
.choose(&mut rng)
|
||
|
|
.map(|s| s.to_owned())
|
||
|
|
.unwrap_or_else(|| "Loading...".to_owned());
|
||
|
|
|
||
|
|
let name = meta.name.unwrap_or_else(|| "Firefish".to_owned());
|
||
|
|
PugArgs {
|
||
|
|
img: meta.banner_url,
|
||
|
|
title: name.clone(),
|
||
|
|
instance_name: name.clone(),
|
||
|
|
desc: meta.description,
|
||
|
|
icon: meta.icon_url,
|
||
|
|
splash_icon,
|
||
|
|
theme_color: meta.theme_color,
|
||
|
|
random_motd,
|
||
|
|
private_mode: meta.private_mode,
|
||
|
|
}
|
||
|
|
}
|