use crate::error::TinifyError;
use crate::sync::source::Source;
use std::path::Path;

/// The Tinify Client.
pub struct Client {
  source: Source,
}

impl Client {
  pub(crate) fn new<K>(key: K) -> Self
  where
    K: AsRef<str>,
  {
    Self {
      source: Source::new(Some(key.as_ref())),
    }
  }

  /// Choose a file to compress.
  pub fn from_file<P>(self, path: P) -> Result<Source, TinifyError>
  where
    P: AsRef<Path>,
  {
    self.source.from_file(path)
  }

  /// Choose a buffer to compress.
  pub fn from_buffer(self, buffer: &[u8]) -> Result<Source, TinifyError> {
    self.source.from_buffer(buffer)
  }

  /// Choose an url image to compress.
  pub fn from_url<P>(self, url: P) -> Result<Source, TinifyError>
  where
    P: AsRef<str> + Into<String>,
  {
    self.source.from_url(url)
  }
}

#[cfg(test)]
mod tests {
  use super::*;
  use crate::convert::Convert;
  use crate::convert::Type;
  use crate::resize::Method;
  use crate::resize::Resize;
  use assert_matches::assert_matches;
  use dotenv::dotenv;
  use imagesize::size;
  use reqwest::blocking::Client as ReqwestClient;
  use std::env;
  use std::ffi::OsStr;
  use std::fs;

  fn get_key() -> String {
    dotenv().ok();
    match env::var("KEY") {
      Ok(key) => key,
      Err(_err) => panic!("No such file or directory."),
    }
  }

  #[test]
  fn test_invalid_key() {
    let client = Client::new("invalid");
    let request = client
      .from_url("https://tinypng.com/images/panda-happy.png")
      .unwrap_err();

    assert_matches!(request, TinifyError::ClientError { .. });
  }

  #[test]
  fn test_compress_from_file() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./optimized.jpg");
    let tmp_image = Path::new("./tmp_image.jpg");
    let _ = Client::new(key).from_file(tmp_image)?.to_file(output);
    let actual = fs::metadata(tmp_image)?.len();
    let expected = fs::metadata(output)?.len();

    assert_eq!(actual, 124814);
    assert_eq!(expected, 102051);

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_compress_from_buffer() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./optimized.jpg");
    let tmp_image = Path::new("./tmp_image.jpg");
    let buffer = fs::read(tmp_image).unwrap();
    let _ = Client::new(key).from_buffer(&buffer)?.to_file(output);
    let actual = fs::metadata(tmp_image)?.len();
    let expected = fs::metadata(output)?.len();

    assert_eq!(actual, 124814);
    assert_eq!(expected, 102051);

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_compress_from_url() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./optimized.jpg");
    let remote_image = "https://tinypng.com/images/panda-happy.png";
    let _ = Client::new(key).from_url(remote_image)?.to_file(output);
    let expected = fs::metadata(output)?.len();
    let actual = ReqwestClient::new().get(remote_image).send()?;

    if let Some(content_length) = actual.content_length() {
      assert_eq!(content_length, 30734);
    }

    assert_eq!(expected, 26324);

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_save_to_file() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./optimized.jpg");
    let tmp_image = Path::new("./tmp_image.jpg");
    let _ = Client::new(key).from_file(tmp_image)?.to_file(output);

    assert!(output.exists());

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_save_to_bufffer() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./optimized.jpg");
    let tmp_image = Path::new("./tmp_image.jpg");
    let client = Client::new(key);
    let buffer = client.from_file(tmp_image)?.to_buffer()?;

    assert_eq!(buffer.capacity(), 102051);

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_resize_scale_width() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./tmp_resized.jpg");
    let _ = Client::new(key)
      .from_file("./tmp_image.jpg")?
      .resize(Resize {
        method: Method::Scale,
        width: Some(400),
        height: None,
      })?
      .to_file(output);

    let (width, height) = match size(output) {
      Ok(dim) => (dim.width, dim.height),
      Err(err) => panic!("Error getting dimensions: {:?}", err),
    };

    assert_eq!((width, height), (400, 200));

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_resize_scale_height() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./tmp_resized.jpg");
    let _ = Client::new(key)
      .from_file("./tmp_image.jpg")?
      .resize(Resize {
        method: Method::Scale,
        width: None,
        height: Some(400),
      })?
      .to_file(output);

    let (width, height) = match size(output) {
      Ok(dim) => (dim.width, dim.height),
      Err(err) => panic!("Error getting dimensions: {:?}", err),
    };

    assert_eq!((width, height), (800, 400));

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_resize_fit() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./tmp_resized.jpg");
    let _ = Client::new(key)
      .from_file("./tmp_image.jpg")?
      .resize(Resize {
        method: Method::Fit,
        width: Some(400),
        height: Some(200),
      })?
      .to_file(output);

    let (width, height) = match size(output) {
      Ok(dim) => (dim.width, dim.height),
      Err(err) => panic!("Error getting dimensions: {:?}", err),
    };

    assert_eq!((width, height), (400, 200));

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_resize_cover() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./tmp_resized.jpg");
    let _ = Client::new(key)
      .from_file("./tmp_image.jpg")?
      .resize(Resize {
        method: Method::Cover,
        width: Some(400),
        height: Some(200),
      })?
      .to_file(output);

    let (width, height) = match size(output) {
      Ok(dim) => (dim.width, dim.height),
      Err(err) => panic!("Error getting dimensions: {:?}", err),
    };

    assert_eq!((width, height), (400, 200));

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_resize_thumb() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./tmp_resized.jpg");
    let _ = Client::new(key)
      .from_file("./tmp_image.jpg")?
      .resize(Resize {
        method: Method::Thumb,
        width: Some(400),
        height: Some(200),
      })?
      .to_file(output);

    let (width, height) = match size(output) {
      Ok(dim) => (dim.width, dim.height),
      Err(err) => panic!("Error getting dimensions: {:?}", err),
    };

    assert_eq!((width, height), (400, 200));

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_error_transparent_png_to_jpeg() -> Result<(), TinifyError> {
    let key = get_key();
    let convert = Convert {
      r#type: vec![Type::Jpeg],
    };
    let request = Client::new(key)
      .from_url("https://tinypng.com/images/panda-happy.png")?
      .convert(convert)?
      .to_file(Path::new("./tmp_transparent.jpg"))
      .unwrap_err();

    assert_matches!(request, TinifyError::ClientError { .. });

    Ok(())
  }

  #[test]
  fn test_convert_from_jpg_to_png() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./panda-sticker.png");
    let convert = Convert {
      r#type: vec![Type::Png],
    };
    let _ = Client::new(key)
      .from_file("./tmp_image.jpg")?
      .convert(convert)?
      .to_file(output);

    let extension = output.extension().and_then(OsStr::to_str).unwrap();

    assert_eq!(extension, "png");

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_convert_from_jpg_to_webp() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./panda-sticker.webp");
    let convert = Convert {
      r#type: vec![Type::Webp],
    };
    let _ = Client::new(key)
      .from_file("./tmp_image.jpg")?
      .convert(convert)?
      .to_file(output);

    let extension = output.extension().and_then(OsStr::to_str).unwrap();

    assert_eq!(extension, "webp");

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_convert_smallest_type() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./panda-sticker.webp");
    let convert = Convert {
      r#type: vec![Type::Jpeg, Type::Png, Type::Webp],
    };
    let _ = Client::new(key)
      .from_url("https://tinypng.com/images/panda-happy.png")?
      .convert(convert)?
      .to_file(output);

    let extension = output.extension().and_then(OsStr::to_str).unwrap();

    assert_eq!(extension, "webp");

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }

  #[test]
  fn test_convert_smallest_wildcard_type() -> Result<(), TinifyError> {
    let key = get_key();
    let output = Path::new("./panda-sticker.webp");
    let convert = Convert {
      r#type: vec![Type::WildCard],
    };
    let _ = Client::new(key)
      .from_url("https://tinypng.com/images/panda-happy.png")?
      .convert(convert)?
      .to_file(output);

    let extension = output.extension().and_then(OsStr::to_str).unwrap();

    assert_eq!(extension, "webp");

    if output.exists() {
      fs::remove_file(output)?;
    }

    Ok(())
  }
}