use crunchyroll_rs::media::VariantData;
use crunchyroll_rs::{Episode, Locale, Media, Movie};
use std::time::Duration;

#[derive(Clone)]
pub struct Format {
    pub id: String,
    pub title: String,
    pub description: String,
    pub number: u32,
    pub audio: Locale,

    pub duration: Duration,
    pub stream: VariantData,

    pub series_id: String,
    pub series_name: String,

    pub season_id: String,
    pub season_title: String,
    pub season_number: u32,
}

impl Format {
    pub fn new_from_episode(episode: Media<Episode>, stream: VariantData) -> Self {
        Self {
            id: episode.id,
            title: episode.title,
            description: episode.description,
            number: episode.metadata.episode_number,
            audio: episode.metadata.audio_locale,

            duration: episode.metadata.duration.to_std().unwrap(),
            stream,

            series_id: episode.metadata.series_id,
            series_name: episode.metadata.series_title,

            season_id: episode.metadata.season_id,
            season_title: episode.metadata.season_title,
            season_number: episode.metadata.season_number,
        }
    }

    pub fn new_from_movie(movie: Media<Movie>, stream: VariantData) -> Self {
        Self {
            id: movie.id,
            title: movie.title,
            description: movie.description,
            number: 1,
            audio: Locale::ja_JP,

            duration: movie.metadata.duration.to_std().unwrap(),
            stream,

            series_id: movie.metadata.movie_listing_id.clone(),
            series_name: movie.metadata.movie_listing_title.clone(),

            season_id: movie.metadata.movie_listing_id,
            season_title: movie.metadata.movie_listing_title,
            season_number: 1,
        }
    }
}

pub fn format_string(s: String, format: &Format) -> String {
    s.replace("{title}", &format.title)
        .replace("{series_name}", &format.series_name)
        .replace("{season_name}", &format.season_title)
        .replace("{audio}", &format.audio.to_string())
        .replace("{resolution}", &format.stream.resolution.to_string())
        .replace("{season_number}", &format.season_number.to_string())
        .replace("{episode_number}", &format.number.to_string())
        .replace("{series_id}", &format.series_id)
        .replace("{season_id}", &format.season_id)
        .replace("{episode_id}", &format.id)
}