ptth/crates/ptth_relay/src/config.rs

113 lines
2.5 KiB
Rust
Raw Normal View History

2020-11-29 18:37:33 +00:00
// False positive with itertools::process_results
#![allow (clippy::redundant_closure)]
use std::{
2020-11-29 18:37:33 +00:00
collections::HashMap,
convert::{TryFrom},
iter::FromIterator,
path::Path,
};
use crate::{
errors::ConfigError,
key_validity::*,
};
// Stuff we need to load from the config file and use to
// set up the HTTP server
pub mod file {
2020-11-29 18:37:33 +00:00
use serde::Deserialize;
use crate::key_validity::*;
#[derive (Deserialize)]
pub struct Server {
// This is duplicated in the hashmap, but it's not a problem
2020-11-30 15:52:15 +00:00
pub name: String,
pub tripcode: BlakeHashWrapper,
pub display_name: Option <String>,
}
#[derive (Deserialize)]
pub struct DevMode {
}
// Stuff that's identical between the file and the runtime structures
#[derive (Default, Deserialize)]
pub struct Isomorphic {
#[serde (default)]
pub enable_scraper_api: bool,
// If any of these fields are used, we are in dev mode and have to
// show extra warnings, since some auth may be weakened
pub dev_mode: Option <DevMode>,
}
#[derive (Deserialize)]
pub struct Config {
#[serde (flatten)]
pub iso: Isomorphic,
pub port: Option <u16>,
pub servers: Vec <Server>,
// Adding a DB will take a while, so I'm moving these out of dev mode.
pub scraper_keys: Vec <ScraperKey <Valid30Days>>,
}
}
// Stuff we actually need at runtime
pub struct Config {
pub iso: file::Isomorphic,
pub port: Option <u16>,
pub servers: HashMap <String, file::Server>,
pub scraper_keys: HashMap <String, ScraperKey <Valid30Days>>,
}
impl TryFrom <file::Config> for Config {
type Error = ConfigError;
fn try_from (f: file::Config) -> Result <Self, Self::Error> {
let servers = f.servers.into_iter ()
.map (|server| Ok::<_, ConfigError> ((server.name.clone (), server)));
let servers = itertools::process_results (servers, |i| HashMap::from_iter (i))?;
let scraper_keys = if f.iso.enable_scraper_api {
HashMap::from_iter (f.scraper_keys.into_iter ().map (|key| (key.hash.encode_base64 (), key)))
}
else {
Default::default ()
};
Ok (Self {
iso: f.iso,
port: f.port,
servers,
scraper_keys,
})
}
}
impl Config {
pub async fn from_file (path: &Path) -> Result <Self, ConfigError> {
use tokio::prelude::*;
let mut f = tokio::fs::File::open (path).await?;
2020-11-29 18:37:33 +00:00
let mut buffer = vec! [0_u8; 4096];
let bytes_read = f.read (&mut buffer).await?;
buffer.truncate (bytes_read);
let config_s = String::from_utf8 (buffer)?;
let new_config: file::Config = toml::from_str (&config_s)?;
Self::try_from (new_config)
}
}