view sitekey

src/pages/panel/sitekey/mod.rs

@@ -17,6 +17,7 @@
 
 mod add;
 mod list;
+mod view;
 
 pub mod routes {
     pub struct Sitekey {
@@ -53,4 +54,11 @@ pub fn services(cfg: &mut actix_web::web::ServiceConfig) {
         Methods::ProtectGet,
         list::list_sitekeys
     );
+
+    define_resource!(
+        cfg,
+        PAGES.panel.sitekey.view,
+        Methods::ProtectGet,
+        view::view_sitekey
+    );
 }

src/pages/panel/sitekey/view.rs

@@ -0,0 +1,139 @@
+/*
+* Copyright (C) 2021  Aravinth Manivannan <realaravinth@batsense.net>
+*
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU Affero General Public License as
+* published by the Free Software Foundation, either version 3 of the
+* License, or (at your option) any later version.
+*
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU Affero General Public License for more details.
+*
+* You should have received a copy of the GNU Affero General Public License
+* along with this program.  If not, see <https://www.gnu.org/licenses/>.
+*/
+
+use actix_identity::Identity;
+use actix_web::{web, HttpResponse, Responder};
+use sailfish::TemplateOnce;
+
+use crate::errors::*;
+use crate::Data;
+
+const PAGE: &str = "SiteKeys";
+
+#[derive(Clone)]
+struct McaptchaConfig {
+    config_id: i32,
+    duration: i32,
+    name: String,
+}
+
+#[derive(Clone)]
+struct Level {
+    difficulty_factor: i32,
+    visitor_threshold: i32,
+}
+#[derive(TemplateOnce, Clone)]
+#[template(path = "panel/sitekey/view/index.html")]
+struct IndexPage {
+    duration: u32,
+    name: String,
+    levels: Vec<Level>,
+}
+
+impl IndexPage {
+    fn new(config: McaptchaConfig, levels: Vec<Level>) -> Self {
+        IndexPage {
+            duration: config.duration as u32,
+            name: config.name,
+            levels,
+        }
+    }
+}
+
+pub async fn view_sitekey(
+    path: web::Path<String>,
+    data: web::Data<Data>,
+    id: Identity,
+) -> PageResult<impl Responder> {
+    let username = id.identity().unwrap();
+    let key = path.0;
+
+    let config = sqlx::query_as!(
+        McaptchaConfig,
+        "SELECT config_id, duration, name from mcaptcha_config WHERE
+        key = $1 AND
+        user_id = (SELECT ID FROM mcaptcha_users WHERE name = $2) ",
+        &key,
+        &username,
+    )
+    .fetch_one(&data.db)
+    .await?;
+
+    let levels = sqlx::query_as!(
+        Level,
+        "SELECT difficulty_factor, visitor_threshold from mcaptcha_levels WHERE config_id = $1",
+        &config.config_id
+    )
+    .fetch_all(&data.db)
+    .await?;
+
+    let body = IndexPage::new(config, levels).render_once().unwrap();
+    Ok(HttpResponse::Ok()
+        .content_type("text/html; charset=utf-8")
+        .body(body))
+}
+
+#[cfg(test)]
+mod test {
+    use actix_web::http::StatusCode;
+    use actix_web::test;
+    use actix_web::web::Bytes;
+
+    use crate::tests::*;
+    use crate::*;
+
+    #[actix_rt::test]
+    async fn view_sitekey_work() {
+        const NAME: &str = "viewsitekeyuser";
+        const PASSWORD: &str = "longpassworddomain";
+        const EMAIL: &str = "viewsitekeyuser@a.com";
+
+        {
+            let data = Data::new().await;
+            delete_user(NAME, &data).await;
+        }
+
+        register_and_signin(NAME, EMAIL, PASSWORD).await;
+        let (data, _, signin_resp, key) = add_levels_util(NAME, PASSWORD).await;
+        let cookies = get_cookie!(signin_resp);
+
+        let mut app = get_app!(data).await;
+
+        let url = format!("/sitekey/{}/view", &key.key);
+
+        let list_sitekey_resp = test::call_service(
+            &mut app,
+            test::TestRequest::get()
+                .uri(&url)
+                .cookie(cookies.clone())
+                .to_request(),
+        )
+        .await;
+
+        assert_eq!(list_sitekey_resp.status(), StatusCode::OK);
+
+        let body: Bytes = test::read_body(list_sitekey_resp).await;
+        let body = String::from_utf8(body.to_vec()).unwrap();
+
+        assert!(body.contains(&key.name));
+
+        assert!(body.contains(&L1.visitor_threshold.to_string()));
+        assert!(body.contains(&L1.difficulty_factor.to_string()));
+        assert!(body.contains(&L2.difficulty_factor.to_string()));
+        assert!(body.contains(&L2.visitor_threshold.to_string()));
+    }
+}