68 lines
1.9 KiB
Rust
68 lines
1.9 KiB
Rust
use crate::api_schema::router::*;
|
|
|
|
use std::collections::HashMap;
|
|
use std::path::{PathBuf};
|
|
|
|
use hyper::Method;
|
|
|
|
pub struct ApiConfig {
|
|
basedir: PathBuf,
|
|
router: &'static Router,
|
|
aliases: HashMap<String, PathBuf>,
|
|
env_type: RpcEnvironmentType,
|
|
}
|
|
|
|
impl ApiConfig {
|
|
|
|
pub fn new<B: Into<PathBuf>>(basedir: B, router: &'static Router, env_type: RpcEnvironmentType) -> Self {
|
|
Self {
|
|
basedir: basedir.into(),
|
|
router: router,
|
|
aliases: HashMap::new(),
|
|
env_type,
|
|
}
|
|
}
|
|
|
|
pub fn find_method(&self, components: &[&str], method: Method, uri_param: &mut HashMap<String, String>) -> &'static MethodDefinition {
|
|
|
|
if let Some(info) = self.router.find_route(components, uri_param) {
|
|
return match method {
|
|
Method::GET => &info.get,
|
|
Method::PUT => &info.put,
|
|
Method::POST => &info.post,
|
|
Method::DELETE => &info.delete,
|
|
_ => &MethodDefinition::None,
|
|
};
|
|
}
|
|
&MethodDefinition::None
|
|
}
|
|
|
|
pub fn find_alias(&self, components: &[&str]) -> PathBuf {
|
|
|
|
let mut prefix = String::new();
|
|
let mut filename = self.basedir.clone();
|
|
let comp_len = components.len();
|
|
if comp_len >= 1 {
|
|
prefix.push_str(components[0]);
|
|
if let Some(subdir) = self.aliases.get(&prefix) {
|
|
filename.push(subdir);
|
|
for i in 1..comp_len { filename.push(components[i]) }
|
|
} else {
|
|
for i in 0..comp_len { filename.push(components[i]) }
|
|
}
|
|
}
|
|
filename
|
|
}
|
|
|
|
pub fn add_alias<S, P>(&mut self, alias: S, path: P)
|
|
where S: Into<String>,
|
|
P: Into<PathBuf>,
|
|
{
|
|
self.aliases.insert(alias.into(), path.into());
|
|
}
|
|
|
|
pub fn env_type(&self) -> RpcEnvironmentType {
|
|
self.env_type
|
|
}
|
|
}
|