mirror of https://github.com/mastodon/flodgatt
78 lines
2.8 KiB
Rust
78 lines
2.8 KiB
Rust
use super::{postgres_cfg_types::*, EnvVar};
|
|
use crate::err::FatalErr;
|
|
|
|
use url::Url;
|
|
use urlencoding;
|
|
|
|
type Result<T> = std::result::Result<T, FatalErr>;
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct Postgres {
|
|
pub user: PgUser,
|
|
pub host: PgHost,
|
|
pub password: PgPass,
|
|
pub database: PgDatabase,
|
|
pub port: PgPort,
|
|
pub ssl_mode: PgSslMode,
|
|
}
|
|
|
|
impl EnvVar {
|
|
fn update_with_postgres_url(mut self, url_str: &str) -> Result<Self> {
|
|
let url = Url::parse(url_str)?;
|
|
let none_if_empty = |s: String| if s.is_empty() { None } else { Some(s) };
|
|
|
|
for (k, v) in url.query_pairs().into_owned() {
|
|
match k.to_string().as_str() {
|
|
"user" => self.maybe_add_env_var("DB_USER", Some(v.to_string())),
|
|
"password" => self.maybe_add_env_var("DB_PASS", Some(v.to_string())),
|
|
"host" => self.maybe_add_env_var("DB_HOST", Some(v.to_string())),
|
|
"sslmode" => self.maybe_add_env_var("DB_SSLMODE", Some(v.to_string())),
|
|
_ => Err(FatalErr::config(
|
|
"POSTGRES_URL",
|
|
&k,
|
|
"a URL with parameters `password`, `user`, `host`, and `sslmode` only",
|
|
))?,
|
|
}
|
|
}
|
|
|
|
self.maybe_add_env_var("DB_PORT", url.port());
|
|
self.maybe_add_env_var("DB_PASS", url.password());
|
|
self.maybe_add_env_var(
|
|
"DB_HOST",
|
|
url.host()
|
|
.map(|host| urlencoding::decode(&host.to_string()))
|
|
.transpose()?,
|
|
);
|
|
self.maybe_add_env_var("DB_USER", none_if_empty(url.username().to_string()));
|
|
self.maybe_add_env_var("DB_NAME", none_if_empty(url.path()[1..].to_string()));
|
|
Ok(self)
|
|
}
|
|
}
|
|
|
|
impl Postgres {
|
|
/// Configure Postgres and return a connection
|
|
pub fn from_env(env: EnvVar) -> Result<Self> {
|
|
let env = match env.get("DATABASE_URL").cloned() {
|
|
Some(url_str) => env.update_with_postgres_url(&url_str)?,
|
|
None => env,
|
|
};
|
|
|
|
let cfg = Self {
|
|
user: PgUser::default().maybe_update(env.get("DB_USER"))?,
|
|
host: PgHost::default().maybe_update(env.get("DB_HOST"))?,
|
|
password: PgPass::default().maybe_update(env.get("DB_PASS"))?,
|
|
database: PgDatabase::default().maybe_update(env.get("DB_NAME"))?,
|
|
port: PgPort::default().maybe_update(env.get("DB_PORT"))?,
|
|
ssl_mode: PgSslMode::default().maybe_update(env.get("DB_SSLMODE"))?,
|
|
};
|
|
Ok(cfg)
|
|
}
|
|
|
|
// // use openssl::ssl::{SslConnector, SslMethod};
|
|
// // use postgres_openssl::MakeTlsConnector;
|
|
// // let mut builder = SslConnector::builder(SslMethod::tls())?;
|
|
// // builder.set_ca_file("/etc/ssl/cert.pem")?;
|
|
// // let connector = MakeTlsConnector::new(builder.build());
|
|
// // TODO: add TLS support, remove `NoTls`
|
|
}
|