diff --git a/src/server/environment.rs b/src/server/environment.rs index 18346be4..7345cc8d 100644 --- a/src/server/environment.rs +++ b/src/server/environment.rs @@ -4,12 +4,16 @@ use std::collections::HashMap; use serde_json::Value; pub struct RestEnvironment { + env_type: RpcEnvironmentType, result_attributes: HashMap, } impl RestEnvironment { - pub fn new() -> Self { - Self { result_attributes: HashMap::new() } + pub fn new(env_type: RpcEnvironmentType) -> Self { + Self { + result_attributes: HashMap::new(), + env_type, + } } } @@ -22,4 +26,8 @@ impl RpcEnvironment for RestEnvironment { fn get_result_attrib(&self, name: &str) -> Option<&Value> { self.result_attributes.get(name) } + + fn env_type(&self) -> RpcEnvironmentType { + self.env_type + } } diff --git a/src/server/rest.rs b/src/server/rest.rs index 8dc80d85..eb938c53 100644 --- a/src/server/rest.rs +++ b/src/server/rest.rs @@ -397,7 +397,7 @@ pub fn handle_request(api: Arc, req: Request) -> BoxFut { println!("REQUEST {} {}", method, path); println!("COMPO {:?}", components); - let mut rpcenv = RestEnvironment::new(); + let rpcenv = RestEnvironment::new(RpcEnvironmentType::PRIVILEDGED); if comp_len >= 1 && components[0] == "api2" { println!("GOT API REQUEST");