spacebar/rust-nex
6
0
Fork 0
Code Issues Pull requests Projects Releases Packages 50 Wiki Activity Actions

feat(secure): add access level to bypass maintenance to the .env config

Browse source
This commit is contained in:
DJMrTV 2025-02-05 16:18:51 +01:00
commit 0d90665698
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/protocols/mod.rs
View file

@ -22,6 +22,12 @@ static IS_MAINTENANCE: Lazy<bool> = Lazy::new(|| {
.map(|v| v.parse().expect("IS_MAINTENANCE should be a boolean value")) .map(|v| v.parse().expect("IS_MAINTENANCE should be a boolean value"))
.unwrap_or(false) .unwrap_or(false)
}); });
static BYPASS_LEVEL: Lazy<i32> = Lazy::new(|| {
env::var("MAINTENANCE_BYPASS_MINIMUM_ACCESS_LEVEL")
.ok()
.map(|v| v.parse().expect("IS_MAINTENANCE should be a boolean value"))
.unwrap_or(3)
});
pub fn block_if_maintenance<'a>(rmcmessage: &'a RMCMessage, conn: &'a mut ConnectionData) -> Pin<Box<(dyn Future<Output=Option<RMCResponse>> + Send + 'a)>> { pub fn block_if_maintenance<'a>(rmcmessage: &'a RMCMessage, conn: &'a mut ConnectionData) -> Pin<Box<(dyn Future<Output=Option<RMCResponse>> + Send + 'a)>> {
@ -30,7 +36,7 @@ pub fn block_if_maintenance<'a>(rmcmessage: &'a RMCMessage, conn: &'a mut Connec
if let Some(secure_conn) = active_conn.active_secure_connection_data.as_ref() { if let Some(secure_conn) = active_conn.active_secure_connection_data.as_ref() {
if let Ok(mut client) = grpc::account::Client::new().await { if let Ok(mut client) = grpc::account::Client::new().await {
if let Ok(client_data) = client.get_user_data(secure_conn.pid).await{ if let Ok(client_data) = client.get_user_data(secure_conn.pid).await{
if client_data.access_level >= 2{ if client_data.access_level >= *BYPASS_LEVEL{
return None; return None;
} }
} }