use crate::database::{article::Article, PsqlData};
use axum::{
    extract::{Extension, Path},
    http::StatusCode,
    response::IntoResponse,
    routing::{get, Router},
};
use sqlx::PgPool;
use std::{collections::HashMap, error::Error};

use super::templates::{ArticleTemplate, PostFooterTemplate, PostTemplate, HtmlTemplate};

pub fn get_router() -> Router {
    Router::new()
        .route("/", get(post))
        .route("/:article", get(article))
}

async fn post(Extension(pool): Extension<PgPool>) -> impl IntoResponse {
    let post_page = PostTemplate {
        articles: match get_articles_date_sorted(&pool).await {
            Ok(list) => list,
            Err(_) => Vec::new(),
        },
    };
    HtmlTemplate(post_page)
}

async fn article(
    Extension(pool): Extension<PgPool>,
    Path(params): Path<HashMap<String, String>>,
) -> Result<impl IntoResponse, StatusCode> {
    let article_id: &String = params.get("article").unwrap();
    let article: Article = match Article::read_by_reference(&pool, article_id).await {
        Ok(a) => *a,
        Err(_) => return Err(StatusCode::NOT_FOUND),
    };
    let footer = PostFooterTemplate {
        previous: match article.previous {
            Some(a) => a,
            None => "".to_string(),
        },
        next: match article.next {
            Some(a) => a,
            None => "".to_string(),
        },
    };
    let template = ArticleTemplate {
        content: article.content,
        footer: footer.to_string(),
    };
    Ok(HtmlTemplate(template))
}

pub async fn get_articles_date_sorted(pool: &PgPool) -> Result<Vec<Article>, Box<dyn Error>> {
    let mut articles: Vec<Article> = match Article::read_all(pool).await {
        Ok(a) => a.iter().map(|x| *x.clone()).collect(),
        Err(_) => Vec::new(),
    };

    articles.sort_by(|a, b| b.date.cmp(&a.date));
    Ok(articles)
}
