blog.rs

use anyhow::{bail, Error};
use fnv::FnvHashMap;
use warp::hyper::StatusCode;

use crate::{
    favicon::Favicon, image::MyImage, load::load_from_path, misc::Misc, path::UrlPath, post::Post,
    resp::Response, style::Style,
};

pub struct Blog {
    pages: Vec<Post>,
    posts: Vec<Post>,
    imgs: Vec<MyImage>,
    misc: Vec<Misc>,
    custom_style: Option<Style>,
    favicon: Option<Favicon>,
}

impl Blog {
    pub fn new() -> Result<Self, Error> {
        let pages = load_from_path("pages", "pages")?;
        let posts = load_from_path("posts", "posts")?;
        let imgs = load_from_path("assets/img", "")?;
        let misc = load_from_path("assets/misc", "")?;
        let custom_style = Style::load_if_exists("assets/style.css")?;
        let favicon = Favicon::load_if_exists("assets/favicon.png")?;

        Ok(Self {
            pages,
            posts,
            imgs,
            misc,
            custom_style,
            favicon,
        })
    }

    fn home(&mut self) -> Result<String, Error> {
        let mut content =
            r#"<div class="centered"><h1>Stephen's Site</h1></div>Welcome to my corner of the Internet. Occasionally I will do something kind of interesting and post about it here. Bear with me - I'm a developer, not a writer, so I'm bad at keeping this place up to date! For a more complete list of my work, please check out my <a href="https://gitlab.scd31.com/stephen">Git repository</a>.<h2>Recent Posts</h2><table>"#
                .to_string();

        for post in &self.posts {
            content.push_str(&post.link()?);
        }

        content.push_str("</table>");

        Ok(dress_page(
            "Stephen's Site",
            &content,
            &self.pages,
            &self.favicon,
        ))
    }
}

pub struct RenderedBlog {
    pages: FnvHashMap<UrlPath, Response>,
    not_found: Response,
}

impl RenderedBlog {
    pub fn get(&self, path: &str) -> (StatusCode, Response) {
        self.pages
            .get(&UrlPath::new(path))
            .map(|x| (StatusCode::OK, x.clone()))
            .unwrap_or((StatusCode::NOT_FOUND, self.not_found.clone()))
    }
}

impl TryFrom<Blog> for RenderedBlog {
    type Error = Error;

    fn try_from(mut b: Blog) -> Result<Self, Error> {
        b.pages.sort();
        b.posts.sort();

        let mut pages = FnvHashMap::default();

        for p in &b.posts {
            let body = dress_page(&p.title, &p.html()?, &b.pages, &b.favicon);

            insert_path(&mut pages, &p.url, Response::html(body))?;
        }

        for p in &b.pages {
            let body = dress_page(&p.title, &p.html()?, &b.pages, &b.favicon);

            insert_path(&mut pages, &p.url, Response::html(body))?;
        }

        let mut style = include_str!("assets/style.css").to_string();
        if let Some(s) = &b.custom_style {
            style.push_str(&s.content);
        }

        insert_path(&mut pages, "/style.css", Response::css(style))?;

        let home = b.home()?;
        insert_path(&mut pages, "/", Response::html(home))?;

        for img in b.imgs {
            let (original_url, original, thumbnail_url, thumbnail) = img.into_responses()?;
            insert_path(&mut pages, &thumbnail_url, thumbnail)?;
            insert_path(&mut pages, &original_url, original)?;
        }

        for m in b.misc {
            insert_path(&mut pages, &m.url.clone(), m.response())?;
        }

        let not_found = Response::html(dress_page(
            "Page not found",
            include_str!("assets/404.html"),
            &b.pages,
            &b.favicon,
        ));

        if let Some(fav) = b.favicon {
            for (p, r) in fav.responses {
                insert_path(&mut pages, p, r)?;
            }
        }

        dbg!(pages.keys());

        Ok(Self { not_found, pages })
    }
}

fn insert_path(
    pages: &mut FnvHashMap<UrlPath, Response>,
    path: &str,
    value: Response,
) -> anyhow::Result<()> {
    let path = UrlPath::new(path);
    if pages.contains_key(&path) {
        bail!(
            "Cannot serve {} - already serving an asset with the same path",
            path
        );
    }

    pages.insert(path, value);

    Ok(())
}

fn dress_page(title: &str, content: &str, pages: &[Post], favicon: &Option<Favicon>) -> String {
    let favicons = match favicon {
        Some(f) => f.html(),
        None => "",
    };

    let mut page_links = String::new();
    for p in pages {
        page_links.push_str(&p.link_short());
    }

    format!(
        r#"<!DOCTYPE html><html lang="en"><head><meta name="viewport" content="width=device-width, initial-scale=1"><link rel="stylesheet" href="/style.css" />{favicons}<title>{title}</title></head><body><div><a href="/">Home</a>{page_links}<img src="/img/logo.png" class="align-right" /></div><hr>{content}</body></html>"#
    )
}