2022-09-07 13:06:03 +05:30
/*
* Copyright ( C ) 2022 Aravinth Manivannan < realaravinth @ batsense . net >
*
* This program is free software : you can redistribute it and / or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation , either version 3 of the
* License , or ( at your option ) any later version .
*
* This program is distributed in the hope that it will be useful ,
* but WITHOUT ANY WARRANTY ; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the
* GNU Affero General Public License for more details .
*
* You should have received a copy of the GNU Affero General Public License
* along with this program . If not , see < https ://www.gnu.org/licenses/>.
* /
use std ::env ;
use std ::path ::Path ;
use config ::{ Config , ConfigError , Environment , File } ;
2022-09-09 17:14:55 +05:30
use derive_more ::Display ;
2022-09-07 13:06:03 +05:30
use log ::warn ;
use serde ::Deserialize ;
2022-09-09 17:14:55 +05:30
use serde ::Serialize ;
2022-09-07 13:06:03 +05:30
use url ::Url ;
2022-12-29 01:38:48 +05:30
#[ derive(Debug, Clone, Deserialize) ]
pub struct Dashboard {
pub api_key : String ,
}
2022-09-07 13:06:03 +05:30
#[ derive(Debug, Clone, Deserialize) ]
pub struct Server {
pub port : u32 ,
pub domain : String ,
pub ip : String ,
pub url_prefix : Option < String > ,
pub proxy_has_tls : bool ,
}
impl Server {
#[ cfg(not(tarpaulin_include)) ]
pub fn get_ip ( & self ) -> String {
format! ( " {} : {} " , self . ip , self . port )
}
}
2022-09-09 17:14:55 +05:30
#[ derive(Deserialize, Serialize, Display, Eq, PartialEq, Clone, Debug) ]
#[ serde(rename_all = " lowercase " ) ]
pub enum DBType {
#[ display(fmt = " postgres " ) ]
Postgres ,
// #[display(fmt = "maria")]
// Maria,
}
impl DBType {
fn from_url ( url : & Url ) -> Result < Self , ConfigError > {
match url . scheme ( ) {
// "mysql" => Ok(Self::Maria),
" postgres " = > Ok ( Self ::Postgres ) ,
_ = > Err ( ConfigError ::Message ( " Unknown database type " . into ( ) ) ) ,
}
}
}
#[ derive(Debug, Clone, Deserialize) ]
pub struct Database {
pub url : String ,
pub pool : u32 ,
pub database_type : DBType ,
}
2022-09-07 13:06:03 +05:30
#[ derive(Debug, Clone, Deserialize) ]
pub struct Settings {
pub debug : bool ,
2022-09-09 17:14:55 +05:30
pub database : Database ,
2022-09-07 13:06:03 +05:30
pub server : Server ,
pub source_code : String ,
2022-12-29 01:38:48 +05:30
pub dash : Dashboard ,
2022-09-07 13:06:03 +05:30
}
#[ cfg(not(tarpaulin_include)) ]
impl Settings {
pub fn new ( ) -> Result < Self , ConfigError > {
let mut s = Config ::new ( ) ;
const CURRENT_DIR : & str = " ./config/default.toml " ;
2022-09-08 18:00:26 +05:30
const ETC : & str = " /etc/lpforms/config.toml " ;
2022-09-07 13:06:03 +05:30
2022-09-08 18:14:27 +05:30
if let Ok ( path ) = env ::var ( " LPFORMS_CONFIG " ) {
2022-09-07 13:06:03 +05:30
s . merge ( File ::with_name ( & path ) ) ? ;
} else if Path ::new ( CURRENT_DIR ) . exists ( ) {
// merging default config from file
s . merge ( File ::with_name ( CURRENT_DIR ) ) ? ;
} else if Path ::new ( ETC ) . exists ( ) {
s . merge ( File ::with_name ( ETC ) ) ? ;
} else {
log ::warn! ( " configuration file not found " ) ;
}
2022-09-08 18:00:26 +05:30
s . merge ( Environment ::with_prefix ( " LPFORMS " ) . separator ( " _ " ) ) ? ;
2022-09-07 13:06:03 +05:30
check_url ( & s ) ;
match env ::var ( " PORT " ) {
Ok ( val ) = > {
s . set ( " server.port " , val ) . unwrap ( ) ;
}
Err ( e ) = > warn! ( " couldn't interpret PORT: {} " , e ) ,
}
2022-09-09 17:14:55 +05:30
match env ::var ( " DATABASE_URL " ) {
Ok ( val ) = > {
let url = Url ::parse ( & val ) . expect ( " couldn't parse Database URL " ) ;
s . set ( " database.url " , url . to_string ( ) ) . unwrap ( ) ;
let database_type = DBType ::from_url ( & url ) . unwrap ( ) ;
s . set ( " database.database_type " , database_type . to_string ( ) )
. unwrap ( ) ;
}
Err ( _e ) = > {
set_database_url ( & mut s ) ;
}
}
// // setting default values
// #[cfg(test)]
// s.set("database.pool", 2.to_string())
// .expect("Couldn't set database pool count");
2022-09-07 13:06:03 +05:30
match s . try_into ::< Self > ( ) {
Ok ( val ) = > {
Ok ( val )
} ,
Err ( e ) = > Err ( ConfigError ::Message ( format! ( " \n \n Error: {} . If it says missing fields, then please refer to https://github.com/mCaptcha/mcaptcha#configuration to learn more about how mcaptcha reads configuration \n \n " , e ) ) ) ,
}
}
}
2022-09-09 17:14:55 +05:30
fn set_database_url ( s : & mut Config ) {
s . set (
" database.url " ,
format! (
r "postgres://{}:{}@{}:{}/{}" ,
s . get ::< String > ( " database.username " )
. expect ( " Couldn't access database username " ) ,
s . get ::< String > ( " database.password " )
. expect ( " Couldn't access database password " ) ,
s . get ::< String > ( " database.hostname " )
. expect ( " Couldn't access database hostname " ) ,
s . get ::< String > ( " database.port " )
. expect ( " Couldn't access database port " ) ,
s . get ::< String > ( " database.name " )
. expect ( " Couldn't access database name " )
) ,
)
. expect ( " Couldn't set database url " ) ;
}
2022-09-07 13:06:03 +05:30
#[ cfg(not(tarpaulin_include)) ]
fn check_url ( s : & Config ) {
let url = s
. get ::< String > ( " source_code " )
. expect ( " Couldn't access source_code " ) ;
Url ::parse ( & url ) . expect ( " Please enter a URL for source_code in settings " ) ;
}