From 95bc1feef725ae87f301c9fd5cb27f8d70e3e3e2 Mon Sep 17 00:00:00 2001
From: realaravinth <realaravinth@batsense.net>
Date: Sun, 9 May 2021 19:45:25 +0530
Subject: [PATCH] panel: overview

---
 src/pages/panel/mod.rs          | 32 ++++++++++++++++++--------------
 src/pages/panel/sitekey/list.rs | 18 ++++++++++++------
 src/pages/panel/sitekey/mod.rs  |  2 +-
 templates/panel/index.html      | 19 +++++++++++++++++++
 4 files changed, 50 insertions(+), 21 deletions(-)

diff --git a/src/pages/panel/mod.rs b/src/pages/panel/mod.rs
index 45cb2f8a..dea76452 100644
--- a/src/pages/panel/mod.rs
+++ b/src/pages/panel/mod.rs
@@ -15,32 +15,36 @@
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
-use actix_web::{HttpResponse, Responder};
-use lazy_static::lazy_static;
+use actix_identity::Identity;
+use actix_web::{web, HttpResponse, Responder};
 use sailfish::TemplateOnce;
 
 pub mod sitekey;
 
+use crate::errors::PageResult;
+use crate::Data;
+use sitekey::list::{get_list_sitekeys, SiteKeys};
+
 #[derive(TemplateOnce, Clone)]
 #[template(path = "panel/index.html")]
-pub struct IndexPage;
+pub struct IndexPage {
+    sitekeys: SiteKeys,
+}
 
-const PAGE: &str = "Dashboard";
-
-impl Default for IndexPage {
-    fn default() -> Self {
-        IndexPage
+impl IndexPage {
+    fn new(sitekeys: SiteKeys) -> Self {
+        IndexPage { sitekeys }
     }
 }
 
-lazy_static! {
-    static ref INDEX: String = IndexPage::default().render_once().unwrap();
-}
+const PAGE: &str = "Dashboard";
 
-async fn panel() -> impl Responder {
-    HttpResponse::Ok()
+async fn panel(data: web::Data<Data>, id: Identity) -> PageResult<impl Responder> {
+    let sitekeys = get_list_sitekeys(&data, &id).await?;
+    let body = IndexPage::new(sitekeys).render_once().unwrap();
+    Ok(HttpResponse::Ok()
         .content_type("text/html; charset=utf-8")
-        .body(&*INDEX)
+        .body(body))
 }
 
 pub fn services(cfg: &mut actix_web::web::ServiceConfig) {
diff --git a/src/pages/panel/sitekey/list.rs b/src/pages/panel/sitekey/list.rs
index 54363407..1072927b 100644
--- a/src/pages/panel/sitekey/list.rs
+++ b/src/pages/panel/sitekey/list.rs
@@ -37,7 +37,17 @@ impl IndexPage {
     }
 }
 
+/// render a list of all sitekeys that a user has
 pub async fn list_sitekeys(data: web::Data<Data>, id: Identity) -> PageResult<impl Responder> {
+    let res = get_list_sitekeys(&data, &id).await?;
+    let body = IndexPage::new(res).render_once().unwrap();
+    Ok(HttpResponse::Ok()
+        .content_type("text/html; charset=utf-8")
+        .body(body))
+}
+
+/// utility function to get a list of all sitekeys that a user owns
+pub async fn get_list_sitekeys(data: &Data, id: &Identity) -> PageResult<SiteKeys> {
     let username = id.identity().unwrap();
     let res = sqlx::query_as!(
         MCaptchaDetails,
@@ -47,14 +57,10 @@ pub async fn list_sitekeys(data: web::Data<Data>, id: Identity) -> PageResult<im
     )
     .fetch_all(&data.db)
     .await?;
-
-    let body = IndexPage::new(res).render_once().unwrap();
-    Ok(HttpResponse::Ok()
-        .content_type("text/html; charset=utf-8")
-        .body(body))
+    Ok(res)
 }
 
-type SiteKeys = Vec<MCaptchaDetails>;
+pub type SiteKeys = Vec<MCaptchaDetails>;
 
 #[cfg(test)]
 mod test {
diff --git a/src/pages/panel/sitekey/mod.rs b/src/pages/panel/sitekey/mod.rs
index 6945defb..22840c34 100644
--- a/src/pages/panel/sitekey/mod.rs
+++ b/src/pages/panel/sitekey/mod.rs
@@ -16,7 +16,7 @@
 */
 
 mod add;
-mod list;
+pub mod list;
 mod view;
 
 pub mod routes {
diff --git a/templates/panel/index.html b/templates/panel/index.html
index 267b23ef..ef449257 100644
--- a/templates/panel/index.html
+++ b/templates/panel/index.html
@@ -6,5 +6,24 @@
   <. include!("./help-banner/index.html"); .>
   <div class="inner-container">
 
+    <ul class="sitekey-list__box">
+      <h1 class="sitekey-list__title">Your Sitekeys</h1>
+      <. for sitekey in sitekeys.iter() { .>
+      <a href="/sitekey/<.= sitekey.key .>/view" class="sitekey-list__item-container">
+        <li class="sitekey-list__item">
+
+          <span class="sitekey-list__name">
+            <.= sitekey.name .>
+          </span>
+          <span class="sitekey-list__key">
+            <.= sitekey.key .>
+          </span>
+        </li>
+      </a>
+
+      <. } .>
+    </ul>
+
+
   </div>
 <. include!("../components/footers.html"); .>