url prefix processing

src/api/v1/pow/mod.rs

@@ -15,6 +15,9 @@
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
+//use actix_cors::Cors;
+//use lazy_static::lazy_static;
+
 pub mod get_config;
 pub mod verify_pow;
 pub mod verify_token;
@@ -22,3 +25,12 @@ pub mod verify_token;
 pub use super::mcaptcha::duration::GetDurationResp;
 pub use super::mcaptcha::is_authenticated;
 pub use super::mcaptcha::levels::I32Levels;
+
+//lazy_static! {
+//    pub static ref CORS: Cors = Cors::default()
+//        .allow_any_origin()
+//        .allowed_methods(vec!["POST"])
+//        .allow_any_header()
+//        .max_age(0)
+//        .send_wildcard();
+//}

src/settings.rs

@@ -29,6 +29,7 @@ pub struct Server {
     pub domain: String,
     pub cookie_secret: String,
     pub ip: String,
+    pub url_prefix: Option<String>,
 }
 
 #[derive(Debug, Clone, Deserialize)]
@@ -42,6 +43,15 @@ impl Server {
     pub fn get_ip(&self) -> String {
         format!("{}:{}", self.ip, self.port)
     }
+
+    fn check_url_prefix(&mut self) {
+        if let Some(prefix) = self.url_prefix.clone() {
+            self.url_prefix = Some(prefix.trim().into());
+            if prefix.trim().is_empty() {
+                panic!("URL prefix is set to empty string")
+            }
+        }
+    }
 }
 
 #[derive(Debug, Clone, Deserialize)]
@@ -162,3 +172,24 @@ fn set_database_url(s: &mut Config) {
     )
     .expect("Couldn't set databse url");
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn url_prefix_test() {
+        let mut settings = Settings::new().unwrap();
+        assert!(settings.server.url_prefix.is_none());
+        settings.server.url_prefix = Some("test".into());
+        settings.server.check_url_prefix();
+    }
+
+    #[test]
+    #[should_panic]
+    fn url_prefix_panic_test() {
+        let mut settings = Settings::new().unwrap();
+        settings.server.url_prefix = Some("    ".into());
+        settings.server.check_url_prefix();
+    }
+}