/*
    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 super::{
    assets::rocket_uri_macro_r_item_assets,
    error::MyError,
    sort::{filter_and_sort_nodes, NodeFilterSort, NodeFilterSortForm},
};
use crate::{
    database::Database,
    routes::{
        api::AcceptJson,
        progress::rocket_uri_macro_r_player_watched,
        ui::{
            account::session::Session,
            assets::AssetRole,
            layout::{DynLayoutPage, LayoutPage},
            player::{rocket_uri_macro_r_player, PlayerConfig},
        },
    },
    uri,
};
use anyhow::{anyhow, Context, Result};
use jellybase::permission::NodePermissionExt;
use jellycommon::{
    user::{NodeUserData, WatchedState},
    MediaInfo, NodeKind, NodePublic, Rating, SourceTrackKind,
};
use rocket::{get, serde::json::Json, Either, State};

/// This function is a stub and only useful for use in the uri! macro.
#[get("/n/<id>")]
pub fn r_library_node(id: String) {
    drop(id)
}

#[get("/n/<id>?<filter..>")]
pub async fn r_library_node_filter<'a>(
    session: Session,
    id: &'a str,
    db: &'a State<Database>,
    aj: AcceptJson,
    filter: NodeFilterSort,
) -> Result<Either<DynLayoutPage<'a>, Json<NodePublic>>, MyError> {
    let node = db
        .node
        .get(&id.to_string())
        .context("retrieving library node")?
        .only_if_permitted(&session.user.permissions)
        .ok_or(anyhow!("node does not exist"))?
        .public;

    let udata = db
        .user_node
        .get(&(session.user.name.clone(), id.to_string()))?
        .unwrap_or_default();

    if *aj {
        return Ok(Either::Right(Json(node)));
    }

    let mut children = node
        .children
        .iter()
        .map(|c| Ok(db.get_node_with_userdata(c, &session)?))
        .collect::<anyhow::Result<Vec<_>>>()?
        .into_iter()
        .collect();

    filter_and_sort_nodes(&filter, &mut children);

    Ok(Either::Left(LayoutPage {
        title: node.title.to_string(),
        content: markup::new! {
            @NodePage { node: &node, id: &id, udata: &udata, children: &children, filter: &filter }
        },
        ..Default::default()
    }))
}

markup::define! {
    NodeCard<'a>(id: &'a str, node: &'a NodePublic, udata: &'a NodeUserData) {
        @let cls = format!("node card poster {}", match node.kind {NodeKind::Channel => "aspect-square", NodeKind::Video => "aspect-thumb", NodeKind::Collection => "aspect-land", _ => "aspect-port"});
        div[class=cls] {
            .poster {
                a[href=uri!(r_library_node(id))] {
                    img[src=uri!(r_item_assets(id, AssetRole::Poster, Some(1024)))];
                }
                .cardhover.item {
                    @if !(matches!(node.kind, NodeKind::Collection | NodeKind::Channel)) {
                        a.play.icon[href=&uri!(r_player(id, PlayerConfig::default()))] { "play_arrow" }
                    }
                    @Props { node, udata }
                }
            }
            div.title {
                a[href=uri!(r_library_node(id))] {
                    @node.title
                }
            }
        }
    }
    NodePage<'a>(id: &'a str, node: &'a NodePublic, udata: &'a NodeUserData, children: &'a Vec<(String, NodePublic, NodeUserData)>, filter: &'a NodeFilterSort) {
        @if !matches!(node.kind, NodeKind::Collection) {
            img.backdrop[src=uri!(r_item_assets(id, AssetRole::Backdrop, Some(2048)))];
        }
        .page.node {
            @if !matches!(node.kind, NodeKind::Collection) {
                div.bigposter { img[src=uri!(r_item_assets(id, AssetRole::Poster, Some(2048)))]; }
            }
            .title {
                h1 { @node.title }
                @if node.media.is_some() { a.play[href=&uri!(r_player(id, PlayerConfig::default()))] { "Watch now" }}
                @if !matches!(node.kind, NodeKind::Collection | NodeKind::Channel) {
                    @match udata.watched {
                        WatchedState::None |
                        WatchedState::Progress(_) => {
                            form.mark_watched[method="POST", action=uri!(r_player_watched(id, true))] {
                                input[type="submit", value="Mark Watched"];
                            }
                        }
                        WatchedState::Watched => {
                            form.mark_unwatched[method="POST", action=uri!(r_player_watched(id, false))] {
                                input[type="submit", value="Mark Unwatched"];
                            }
                        }
                    }
                }
            }
            .details {
                @Props { node, udata }
                h3 { @node.tagline }
                @if let Some(description) = &node.description {
                    p { @for line in description.lines() { @line br; } }
                }
                @if let Some(media) = &node.media {
                    details {
                        summary { "Tracks" }
                        ol { @for track in &media.tracks {
                            li { @format!("{track}") }
                        }}
                    }
                }
            }
            @if matches!(node.kind, NodeKind::Collection | NodeKind::Channel) {
                @if matches!(node.kind, NodeKind::Collection) {
                    @if let Some(parent) = &node.path.last().cloned() {
                        a.dirup[href=uri!(r_library_node(parent))] { "Go up" }
                    }
                }
                @NodeFilterSortForm { f: filter }
            }
            @match node.kind {
                NodeKind::Collection | NodeKind::Channel => {
                    ul.children {@for (id, node, udata) in children.iter() {
                        li { @NodeCard { id, node, udata } }
                    }}
                }
                NodeKind::Series => {
                    ol { @for (id, c, _) in children.iter() {
                        li { a[href=uri!(r_library_node(id))] { @c.title } }
                    }}
                }
                _ => {}
            }
        }
    }

    Props<'a>(node: &'a NodePublic, udata: &'a NodeUserData) {
        .props {
            @if let Some(m) = &node.media {
                p { @format_duration(m.duration) }
                p { @m.resolution_name() }
            }
            @if let Some(d) = &node.release_date {
                p { @d.format("%Y-%m-%d").to_string() }
            }
            @if !node.children.is_empty() {
                p { @format!("{} items", node.children.len()) }
            }
            @for (kind, value) in &node.ratings {
                @match kind {
                    Rating::YoutubeLikes => {p{ @format_count(*value as usize) " Likes" }}
                    Rating::YoutubeViews => {p{ @format_count(*value as usize) " Views" }}
                    Rating::YoutubeFollowers => {p{ @format_count(*value as usize) " Subscribers" }}
                    Rating::RottenTomatoes => {p.rating{ @value " Tomatoes" }}
                    Rating::Metacritic => {p{ "Metacritic Score: " @value }}
                    Rating::Imdb => {p.rating{ "IMDb " @value }}
                    Rating::Tmdb => {p.rating{ "TMDB " @value }}
                }
            }
            @if let Some(f) = &node.federated {
                p.federation { @f }
            }
            @match udata.watched {
                WatchedState::None => {}
                WatchedState::Progress(x) => { p.progress { "Watched up to " @format_duration(x) } }
                WatchedState::Watched => { p.watched { "Watched" } }
            }
        }
    }
}

pub fn format_duration(mut d: f64) -> String {
    let mut s = String::new();
    for (unit, k) in [("h", 60. * 60.), ("m", 60.), ("s", 1.)] {
        let mut h = 0;
        while d > k {
            d -= k;
            h += 1;
        }
        if h > 0 {
            s += &format!("{h}{unit}")
        }
    }
    s
}

pub trait DatabaseNodeUserDataExt {
    fn get_node_with_userdata(
        &self,
        id: &str,
        session: &Session,
    ) -> Result<(String, NodePublic, NodeUserData)>;
}
impl DatabaseNodeUserDataExt for Database {
    fn get_node_with_userdata(
        &self,
        id: &str,
        session: &Session,
    ) -> Result<(String, NodePublic, NodeUserData)> {
        Ok((
            id.to_owned(),
            self.node
                .get(&id.to_owned())?
                .ok_or(anyhow!("node does not exist: {id}"))?
                .public,
            self.user_node
                .get(&(session.user.name.to_owned(), id.to_owned()))?
                .unwrap_or_default(),
        ))
    }
}

trait MediaInfoExt {
    fn resolution_name(&self) -> &'static str;
}
impl MediaInfoExt for MediaInfo {
    fn resolution_name(&self) -> &'static str {
        let mut maxw = 0;
        for t in &self.tracks {
            match &t.kind {
                SourceTrackKind::Video { width, .. } => maxw = maxw.max(*width),
                _ => (),
            }
        }

        match maxw {
            7680.. => "8K",
            3840.. => "4K",
            2560.. => "WQHD",
            1920.. => "Full HD",
            1280.. => "HD 720p",
            640.. => "NTSC",
            _ => "Unkown",
        }
    }
}

fn format_count(n: impl Into<usize>) -> String {
    let n: usize = n.into();

    if n >= 1_000_000 {
        format!("{:.1}M", n as f32 / 1_000_000.)
    } else if n >= 1_000 {
        format!("{:.1}k", n as f32 / 1_000.)
    } else {
        format!("{n}")
    }
}