everygarf 2.7.0

use std::fmt::Display;

use chrono::{Datelike, NaiveDate};
use clap::ValueEnum;
use reqwest::Client;

use crate::dates::{date_month_to_string, date_to_string};

#[derive(Clone, Copy, Debug)]
pub struct Api<'a> {
    pub source: Source,
    pub proxy: Option<&'a str>,
}

#[derive(Clone, Copy, Debug, ValueEnum)]
pub enum Source {
    Gocomics,
    Fandom,
}

pub async fn check_proxy_service(client: &Client, proxy: &str) -> Result<(), reqwest::Error> {
    client.get(proxy).send().await?.error_for_status()?;
    Ok(())
}

impl<'a> Api<'a> {
    pub fn get_page_url(&self, date: NaiveDate) -> String {
        let url = self.source.get_page_url(date);
        match self.proxy {
            None => url,
            Some(proxy) => proxy.to_string() + "?" + &url,
        }
    }
}

impl Default for Source {
    fn default() -> Self {
        Self::Gocomics
    }
}

impl Display for Source {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.to_possible_value().unwrap().get_name())
    }
}

impl Source {
    fn get_page_url(&self, date: NaiveDate) -> String {
        // TODO: make consistent
        match &self {
            Self::Gocomics => {
                let date_string = date_to_string(date, "/", false);
                format!("https://www.gocomics.com/garfield/{}", date_string)
            }

            Self::Fandom => {
                let month_string = date_month_to_string(date);
                let date_string = date_to_string(date, "-", false);
                format!(
                    "https://garfield.fandom.com/wiki/Garfield,_{month}_{year}_comic_strips?file={date}.gif",
                    month = month_string,
                    year = date.year(),
                    date = date_string,
                )
            }
        }
    }

    pub fn find_image_url<'a>(&self, body: &'a str) -> Option<&'a str> {
        // TODO: make consistent
        match &self {
            Self::Gocomics => {
                let char_index = body.find("https://assets.amuniversal.com")?;
                body.get(char_index..char_index + 63)
            }

            Self::Fandom => {
                const IMAGE_URL_BASE: &str = "https://static.wikia.nocookie.net";

                let char_index_left = body.find(IMAGE_URL_BASE)?;

                let mut char_index_right = char_index_left + IMAGE_URL_BASE.len() + 1;
                let mut chars = body.chars().skip(char_index_right);

                while chars.next().is_some_and(|ch| ch != '"') {
                    char_index_right += 1;
                }

                body.get(char_index_left..char_index_right)
            }
        }
    }
}