diff options
author | alyx <alyx@aleteoryx.me> | 2023-08-09 15:04:41 -0400 |
---|---|---|
committer | alyx <alyx@aleteoryx.me> | 2023-08-09 15:04:41 -0400 |
commit | d0fc16dde4cf1c58769dfff058509677f601eb8e (patch) | |
tree | e9721c547732a6082eac922cc661ea8e1aaadbff /src/config.rs | |
parent | 40372f0cf8a95f4d5bdf78fc522328b776482dbf (diff) | |
download | lfm_embed-d0fc16dde4cf1c58769dfff058509677f601eb8e.tar.gz lfm_embed-d0fc16dde4cf1c58769dfff058509677f601eb8e.tar.bz2 lfm_embed-d0fc16dde4cf1c58769dfff058509677f601eb8e.zip |
config done
Diffstat (limited to 'src/config.rs')
-rw-r--r-- | src/config.rs | 96 |
1 files changed, 96 insertions, 0 deletions
diff --git a/src/config.rs b/src/config.rs new file mode 100644 index 0000000..ef9c709 --- /dev/null +++ b/src/config.rs @@ -0,0 +1,96 @@ +use std::collections::{HashMap, HashSet}; +use std::error::Error; +use std::sync::LazyLock; +use std::sync::Arc; +use std::future::Future; +use std::pin::Pin; +use std::fs; +use std::time::*; +use duration_str as ds; + +use super::cache::AsyncCache; +use super::deserialize::{GetRecentTracks, GetUserInfo, Track, User}; + +use reqwest::Client; +use dotenv::var; +use tokio::sync::RwLock; + +pub static STATE: LazyLock<Arc<State>> = LazyLock::new(|| { + State::new() +}); + +fn getter(username: &String) -> Pin<Box<dyn Future<Output = Result<(User, Track), &'static str>>>> { + Box::pin(async{Err("nope")}) +} + +type Cache = Arc<RwLock<AsyncCache<String, (User, Track), fn(&String) -> Pin<Box<dyn Future<Output = Result<(User, Track), &'static str>>>>>>>; +#[derive(Debug)] +enum Whitelist { + Exclusive{cache: Cache, whitelist: HashSet<String>}, + Open{default_cache: Cache, whitelist_cache: Cache, whitelist: HashSet<String>} +} +#[derive(Debug)] +pub struct State { + api_key: Arc<str>, + whitelist: Whitelist, + port: u16, + custom_themes: HashMap<String, Arc<str>>, + send_refresh_header: bool +} + +impl State { + fn new() -> Arc<Self> { + Arc::new(State { + api_key: var("LFME_API_KEY").expect("API key must be set").into(), + port: var("LFME_PORT").map(|p| p.parse().expect("cannot parse as a port number")).unwrap_or(9999), + send_refresh_header: var("LFME_SET_HEADER").map(|h| &h == "1").unwrap_or(false), + + custom_themes: { + if let Ok(themes_dir) = var("LFME_THEMES_DIR") { + fs::read_dir(themes_dir).expect("error reading LFME_THEMES_DIR") + .map(|a| a.expect("error reading LFME_THEMES_DIR")) + .filter_map(|a| { + let path = a.path(); + if fs::metadata(&path).map(|m| m.is_file()).unwrap_or(false) && + path.extension() == Some("css".as_ref()) { + Some((path.file_stem().unwrap().to_str().expect("bad filename").to_string(), fs::read_to_string(&path).expect("couldn't read theme CSS").into())) + } + else { None } + }) + .collect() + } + else { HashMap::new() } + }, + + whitelist: { + let cache_from_var = |v: &str, d: u64| -> Cache { + let refresh = var(v).map(|r| ds::parse(&r).expect("bad duration string")).unwrap_or_else(|_| Duration::from_secs(d)); + Arc::new(RwLock::new(AsyncCache::new(refresh, getter as _))) + }; + let default_cache = || cache_from_var("LFME_DEFAULT_REFRESH", 300); + let whitelist_cache = || cache_from_var("LFME_WHITELIST_REFRESH", 60); + + let load_whitelist = || -> Option<HashSet<String>> { + var("LFME_WHITELIST").ok().map( + |w| w.split(",").map(|s| s.trim().to_string()).collect() + ) + }; + + match var("LFME_WHITELIST_MODE").map(|m| m.to_ascii_lowercase()).unwrap_or_else(|_| "open".into()).as_str() { + "open" => { + Whitelist::Open{default_cache: default_cache(), whitelist_cache: whitelist_cache(), whitelist: load_whitelist().unwrap_or_default()} + }, + "exclusive" => { + Whitelist::Exclusive{cache: whitelist_cache(), whitelist: load_whitelist().expect("LFME_WHITELIST not set, unable to serve anyone")} + }, + m => { + panic!("Bad whitelist mode: `{m}`"); + } + } + } + }) + } + + pub fn api_key(&self) -> Arc<str> { self.api_key.clone() } + pub fn port(&self) -> u16 { self.port } +} |