/*
    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>
*/
pub mod config;
pub mod helpers;
pub mod r#impl;
pub mod jhls;
pub mod seek_index;
pub mod stream;
pub mod user;

pub use chrono;

use bincode::{Decode, Encode};

#[cfg(feature = "rocket")]
use rocket::{FromFormField, UriDisplayQuery};
use serde::{Deserialize, Serialize};
use std::{collections::BTreeMap, path::PathBuf};

#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
pub struct Node {
    #[serde(default)]
    pub public: NodePublic,
    #[serde(default)]
    pub private: NodePrivate,
}

#[rustfmt::skip]
#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode,Decode)]
pub struct NodePrivate {
    #[serde(default)] pub id: Option<String>,
    #[serde(default)] pub poster: Option<AssetLocation>,
    #[serde(default)] pub backdrop: Option<AssetLocation>,
    #[serde(default)] pub source: Option<Vec<TrackSource>>,
}

#[rustfmt::skip]
#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
pub struct NodePublic {
    #[serde(default)] pub kind: Option<NodeKind>,
    #[serde(default)] pub title: Option<String>,
    #[serde(default)] pub id: Option<String>,
    #[serde(default)] pub path: Vec<String>,
    #[serde(default)] pub children: Vec<String>,
    #[serde(default)] pub tagline: Option<String>,
    #[serde(default)] pub description: Option<String>,
    #[serde(default)] pub release_date: Option<i64>, // in unix millis
    #[serde(default)] pub index: Option<usize>,
    #[serde(default)] pub media: Option<MediaInfo>,
    #[serde(default)] pub ratings: BTreeMap<Rating, f64>,
    #[serde(default)] pub federated: Option<String>,
}

#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
pub struct ImportOptions {
    pub id: String,
    pub sources: Vec<ImportSource>,
}

#[derive(Debug, Clone, Deserialize, Serialize, Encode, Decode)]
#[serde(rename_all = "snake_case")]
pub enum ImportSource {
    Override(Node),
    Tmdb {
        id: u64,
    },
    Trakt {
        kind: TraktKind,
        id: u64,
    },
    AutoChildren {
        path: Option<PathBuf>,
    },
    Media {
        location: AssetLocation,
        #[serde(default)]
        ignore_metadata: bool,
        #[serde(default)]
        ignore_attachments: bool,
        #[serde(default)]
        ignore_chapters: bool,
    },
    Federated {
        host: String,
    },
}

#[derive(Debug, Clone, Deserialize, Serialize, Hash, PartialEq, Eq, Encode, Decode)]
#[serde(rename_all = "snake_case")]
pub enum AssetLocation {
    Cache(PathBuf),
    Library(PathBuf),
    Assets(PathBuf),
    Temp(PathBuf),
    Media(PathBuf),
}

#[rustfmt::skip]
#[derive(Debug, Clone, Copy, Deserialize, Serialize, PartialEq, Eq, Default, Encode,Decode)]
#[cfg_attr(feature = "rocket", derive(FromFormField, UriDisplayQuery))]
#[serde(rename_all = "snake_case")]
pub enum NodeKind {
    #[cfg_attr(feature = "rocket", field(value = "movie"))] #[default] Movie,
    #[cfg_attr(feature = "rocket", field(value = "video"))] Video,
    #[cfg_attr(feature = "rocket", field(value = "collection"))] Collection,
    #[cfg_attr(feature = "rocket", field(value = "channel"))] Channel,
    #[cfg_attr(feature = "rocket", field(value = "show"))] Show,
    #[cfg_attr(feature = "rocket", field(value = "series"))] Series,
    #[cfg_attr(feature = "rocket", field(value = "season"))] Season,
    #[cfg_attr(feature = "rocket", field(value = "episode"))] Episode,
}

#[derive(Debug, Clone, Deserialize, Serialize, Encode, Decode)]
#[serde(rename_all = "snake_case")]
pub enum TrackSource {
    Local(LocalTrack),
    Remote(usize),
}

pub type TrackID = usize;

#[derive(Debug, Clone, Deserialize, Serialize, Encode, Decode, Hash)]
pub struct LocalTrack {
    pub path: PathBuf,
    pub track: TrackID,
    pub codec_private: Option<Vec<u8>>,
}

#[derive(Debug, Clone, Deserialize, Serialize, Encode, Decode)]
pub struct MediaInfo {
    pub duration: f64, // in seconds
    pub tracks: Vec<SourceTrack>,
    #[serde(default)]
    pub chapters: Vec<Chapter>,
}

#[derive(Debug, Clone, Deserialize, Serialize, Default, Encode, Decode)]
pub struct Chapter {
    pub time_start: Option<f64>,
    pub time_end: Option<f64>,
    pub labels: Vec<(String, String)>,
}

#[derive(Debug, Clone, Deserialize, Serialize, Encode, Decode)]
pub struct SourceTrack {
    pub kind: SourceTrackKind,
    pub name: String,
    pub codec: String,
    pub language: String,
    pub default_duration: Option<u64>,
    #[serde(default)]
    pub federated: Vec<String>,
}

#[derive(
    Debug, Clone, Deserialize, Serialize, PartialEq, Eq, PartialOrd, Ord, Hash, Encode, Decode,
)]
#[serde(rename_all = "snake_case")]
pub enum Rating {
    Imdb,
    Tmdb,
    RottenTomatoes,
    Metacritic,
    YoutubeViews,
    YoutubeLikes,
    YoutubeFollowers,
    Trakt,
}

#[derive(Debug, Clone, Deserialize, Serialize, Encode, Decode)]
#[serde(rename_all = "snake_case")]
pub enum SourceTrackKind {
    Video {
        width: u64,
        height: u64,
        fps: f64,
    },
    Audio {
        channels: usize,
        sample_rate: f64,
        bit_depth: usize,
    },
    Subtitles,
}

#[rustfmt::skip]
#[derive(Debug, Clone, Copy, Encode, Decode)]
#[cfg_attr(feature = "rocket", derive(FromFormField, UriDisplayQuery))]
pub enum AssetRole {
    #[cfg_attr(feature = "rocket", field(value = "poster"))] Poster,
    #[cfg_attr(feature = "rocket", field(value = "backdrop"))] Backdrop,
}

#[derive(Debug, Serialize, Deserialize, Clone, Copy, Encode, Decode)]
#[serde(rename_all = "snake_case")]
pub enum TraktKind {
    Movie,
    Show,
    Season,
    Episode,
    Person,
    User,
}

impl TraktKind {
    pub fn singular(self) -> &'static str {
        match self {
            TraktKind::Movie => "movie",
            TraktKind::Show => "show",
            TraktKind::Season => "season",
            TraktKind::Episode => "episode",
            TraktKind::Person => "person",
            TraktKind::User => "user",
        }
    }
    pub fn plural(self) -> &'static str {
        match self {
            TraktKind::Movie => "movies",
            TraktKind::Show => "shows",
            TraktKind::Season => "seasons",
            TraktKind::Episode => "episodes",
            TraktKind::Person => "people",
            TraktKind::User => "users", // //! not used in API
        }
    }
}
impl ToString for TraktKind {
    fn to_string(&self) -> String {
        match self {
            TraktKind::Movie => "Movie",
            TraktKind::Show => "Show",
            TraktKind::Season => "Season",
            TraktKind::Episode => "Episode",
            TraktKind::Person => "Person",
            TraktKind::User => "User",
        }
        .to_string()
    }
}