rm_engine 0.1.0

My engine for risk-management
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use color_eyre::eyre::Result;
use v_utils::{
	io::{ExpandedPath, Percent},
	macros::MyConfigPrimitives,
};

#[derive(Clone, Debug, Default, MyConfigPrimitives)]
pub struct AppConfig {
	pub default_risk_percent_balance: Percent,
	pub default_sl: Percent,
	pub binance: BinanceConfig,
	pub bybit: BybitConfig,
}

#[derive(Clone, Debug, Default, MyConfigPrimitives)]
pub struct BinanceConfig {
	pub key: String,
	pub secret: String,
}

#[derive(Clone, Debug, Default, MyConfigPrimitives)]
pub struct BybitConfig {
	pub key: String,
	pub secret: String,
}

impl AppConfig {
	pub fn read(path: Option<ExpandedPath>) -> Result<Self> {
		let app_name = env!("CARGO_PKG_NAME");
		let config_dir = env!("XDG_CONFIG_HOME");
		let locations = [
			format!("{config_dir}/{app_name}"),
			format!("{config_dir}/{app_name}/config"), //
		];

		let mut builder = config::Config::builder().add_source(config::Environment::default());

		match path {
			Some(path) => {
				let builder = builder.add_source(config::File::with_name(&path.to_string()).required(true));
				Ok(builder.build()?.try_deserialize()?)
			}
			None => {
				for location in locations.iter() {
					builder = builder.add_source(config::File::with_name(location).required(false));
				}
				let raw: config::Config = builder.build()?;

				match raw.try_deserialize() {
					Ok(config) => Ok(config),
					Err(e) => {
						eprintln!("Config file does not exist or is invalid:");
						Err(e.into())
					}
				}
			}
		}
	}
}