37 lines
1.0 KiB
Rust
37 lines
1.0 KiB
Rust
use std::io::Write;
|
|
|
|
use shared::db_type::DBType;
|
|
|
|
#[derive(Clone, Copy)]
|
|
pub enum EnvFileType {
|
|
DotEnv,
|
|
Yaml,
|
|
}
|
|
|
|
#[derive(Clone)]
|
|
pub struct EnvFile {
|
|
pub file_type: EnvFileType,
|
|
pub db_type: DBType,
|
|
pub db_url: String,
|
|
}
|
|
|
|
impl EnvFile {
|
|
pub fn write(self, path: impl AsRef<std::path::Path>) {
|
|
let path_ref = path.as_ref();
|
|
println!("Config file path: {}", path_ref.display());
|
|
let mut config_file =
|
|
std::fs::File::create(path_ref).expect("Failed to create config file");
|
|
//
|
|
self._write_line(&mut config_file, "DB_TYPE", &self.db_type.to_string());
|
|
self._write_line(&mut config_file, "DATABASE_URL", &self.db_url.to_string())
|
|
}
|
|
|
|
fn _write_line(&self, file: &mut std::fs::File, key: &str, value: &str) {
|
|
match self.file_type {
|
|
EnvFileType::DotEnv => writeln!(file, "{}={}", key, value),
|
|
EnvFileType::Yaml => writeln!(file, "{}: \"{}\"", key, value),
|
|
}
|
|
.expect("Failed to write to config file");
|
|
}
|
|
}
|