/*
    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
    Copyright (C) 2023 metamuffin <metamuffin.org>
*/
use markup::Render;
use rocket::{
    futures::FutureExt,
    http::ContentType,
    response::{self, Responder},
    Request, Response,
};
use std::{future::Future, io::Cursor, pin::Pin};
use tokio::io::AsyncRead;

pub mod account;
pub mod error;
pub mod home;
pub mod layout;
pub mod node;
pub mod player;
pub mod style;

pub struct HtmlTemplate<'a>(pub markup::DynRender<'a>);

impl<'r> Responder<'r, 'static> for HtmlTemplate<'_> {
    fn respond_to(self, _req: &'r Request<'_>) -> response::Result<'static> {
        let mut out = String::new();
        self.0.render(&mut out).unwrap();
        Response::build()
            .header(ContentType::HTML)
            .sized_body(out.as_bytes().len(), Cursor::new(out))
            .ok()
    }
}

pub struct Defer(Pin<Box<dyn Future<Output = String> + Send>>);

impl AsyncRead for Defer {
    fn poll_read(
        mut self: std::pin::Pin<&mut Self>,
        cx: &mut std::task::Context<'_>,
        buf: &mut tokio::io::ReadBuf<'_>,
    ) -> std::task::Poll<std::io::Result<()>> {
        match self.0.poll_unpin(cx) {
            std::task::Poll::Ready(r) => {
                buf.put_slice(r.as_bytes());
                std::task::Poll::Ready(Ok(()))
            }
            std::task::Poll::Pending => std::task::Poll::Pending,
        }
    }
}