librepages/src/pages/dash/home.rs

/*
 * Copyright (C) 2022  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 std::cell::RefCell;

use actix_identity::Identity;
use actix_web::http::header::ContentType;
use serde::{Deserialize, Serialize};
use tera::Context;

use super::get_auth_middleware;
use crate::db::Site;
use crate::errors::ServiceResult;
use crate::pages::errors::*;
use crate::settings::Settings;
use crate::AppCtx;

use super::TemplateSiteEvent;

pub use super::*;

pub const DASH_HOME: TemplateFile = TemplateFile::new("dash_home", "pages/dash/index.html");

pub struct Home {
    ctx: RefCell<Context>,
}

#[derive(Clone, Debug, Deserialize, Serialize, Eq, PartialEq)]
pub struct TemplateSite {
    pub site: Site,
    pub view: String,
    pub last_update: Option<TemplateSiteEvent>,
}

impl TemplateSite {
    pub fn new(site: Site, last_update: Option<TemplateSiteEvent>) -> Self {
        let view = PAGES.dash.site.get_view(site.pub_id.clone());
        Self {
            site,
            last_update,
            view,
        }
    }
}

impl CtxError for Home {
    fn with_error(&self, e: &ReadableError) -> String {
        self.ctx.borrow_mut().insert(ERROR_KEY, e);
        self.render()
    }
}

impl Home {
    pub fn new(settings: &Settings, sites: Option<&[TemplateSite]>) -> Self {
        let ctx = RefCell::new(context(settings));
        if let Some(sites) = sites {
            ctx.borrow_mut().insert(PAYLOAD_KEY, sites);
        }
        Self { ctx }
    }

    pub fn render(&self) -> String {
        TEMPLATES
            .render(DASH_HOME.name, &self.ctx.borrow())
            .unwrap()
    }
}

async fn get_site_data(ctx: &AppCtx, id: &Identity) -> ServiceResult<Vec<TemplateSite>> {
    let db_sites = ctx.db.list_all_sites(&id.identity().unwrap()).await?;
    let mut sites = Vec::with_capacity(db_sites.len());
    for site in db_sites {
        // TODO: impl method on DB to get latest "update" event
        let last_update = ctx
            .db
            .get_latest_update_event(&site.hostname)
            .await?
            .map(|e| e.into());
        sites.push(TemplateSite::new(site, last_update));
    }
    Ok(sites)
}

#[actix_web_codegen_const_routes::get(path = "PAGES.dash.home", wrap = "get_auth_middleware()")]
#[tracing::instrument(name = "Dashboard homepage", skip(ctx, id))]
pub async fn get_home(ctx: AppCtx, id: Identity) -> PageResult<impl Responder, Home> {
    let sites = get_site_data(&ctx, &id)
        .await
        .map_err(|e| PageError::new(Home::new(&ctx.settings, None), e))?;
    let home = Home::new(&ctx.settings, Some(&sites)).render();
    let html = ContentType::html();
    Ok(HttpResponse::Ok().content_type(html).body(home))
}

pub fn services(cfg: &mut web::ServiceConfig) {
    cfg.service(get_home);
}

#[cfg(test)]
mod tests {
    use actix_web::http::StatusCode;
    use actix_web::test;

    use crate::ctx::ArcCtx;
    use crate::tests;
    use crate::*;

    use super::PAGES;

    #[actix_rt::test]
    async fn postgres_dash_home_works() {
        let (_, ctx) = tests::get_ctx().await;
        dashboard_home_works(ctx.clone()).await;
    }

    async fn dashboard_home_works(ctx: ArcCtx) {
        const NAME: &str = "testdashuser";
        const EMAIL: &str = "testdashuser@foo.com";
        const PASSWORD: &str = "longpassword";

        let _ = ctx.delete_user(NAME, PASSWORD).await;
        let (_, signin_resp) = ctx.register_and_signin(NAME, EMAIL, PASSWORD).await;
        let cookies = get_cookie!(signin_resp);
        let app = get_app!(ctx).await;

        let resp = get_request!(&app, PAGES.dash.home, cookies.clone());
        assert_eq!(resp.status(), StatusCode::OK);
        let res = String::from_utf8(test::read_body(resp).await.to_vec()).unwrap();
        println!("before adding site: {res}");
        assert!(res.contains("Nothing to show"));

        let page = ctx.add_test_site(NAME.into()).await;

        let resp = get_request!(&app, PAGES.dash.home, cookies.clone());
        assert_eq!(resp.status(), StatusCode::OK);
        let res = String::from_utf8(test::read_body(resp).await.to_vec()).unwrap();
        println!("after adding site: {res}");
        assert!(!res.contains("Nothing here"));
        assert!(res.contains(&page.domain));
        assert!(res.contains(&page.repo));

        let _ = ctx.delete_user(NAME, PASSWORD).await;
    }
}