extern crate oxipng;

use oxipng::OutFile;
use oxipng::colors::{BitDepth, ColorType};
use oxipng::png;
use std::collections::HashSet;
use std::fs::remove_file;
use std::path::Path;
use std::path::PathBuf;

fn get_opts(input: &Path) -> (OutFile, oxipng::Options) {
    let mut options = oxipng::Options::default();
    options.verbosity = None;
    options.force = true;
    let mut filter = HashSet::new();
    filter.insert(0);
    options.filter = filter;

    (OutFile::Path(Some(input.with_extension("out.png").to_owned())), options)
}

fn test_it_converts(
    input: &Path,
    output: &OutFile,
    opts: &oxipng::Options,
    color_type_in: ColorType,
    bit_depth_in: BitDepth,
    color_type_out: ColorType,
    bit_depth_out: BitDepth,
) {
    let png = png::PngData::new(input, opts.fix_errors).unwrap();

    assert_eq!(png.ihdr_data.color_type, color_type_in);
    assert_eq!(png.ihdr_data.bit_depth, bit_depth_in);
    assert_eq!(png.ihdr_data.interlaced, 1);

    match oxipng::optimize(input, output, opts) {
        Ok(_) => (),
        Err(x) => panic!("{}", x),
    };
    let output = output.path().unwrap();
    assert!(output.exists());

    let png = match png::PngData::new(output, opts.fix_errors) {
        Ok(x) => x,
        Err(x) => {
            remove_file(output).ok();
            panic!("{}", x)
        }
    };

    assert_eq!(png.ihdr_data.color_type, color_type_out);
    assert_eq!(png.ihdr_data.bit_depth, bit_depth_out);

    remove_file(output).ok();
}

#[test]
fn interlaced_rgba_16_should_be_rgba_16() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_rgba_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::RGBA,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_rgba_16_should_be_rgba_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_rgba_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::RGBA,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_8_should_be_rgba_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_8_should_be_rgba_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Eight,
        ColorType::RGBA,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_16_should_be_rgb_16() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_rgb_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::RGB,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_rgba_16_should_be_rgb_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_rgb_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::RGB,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_8_should_be_rgb_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_8_should_be_rgb_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Eight,
        ColorType::RGB,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_16_should_be_palette_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_palette_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::Indexed,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_8_should_be_palette_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_8_should_be_palette_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_16_should_be_palette_4() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_palette_4.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::Indexed,
        BitDepth::Four,
    );
}

#[test]
fn interlaced_rgba_8_should_be_palette_4() {
    let input = PathBuf::from("tests/files/interlaced_rgba_8_should_be_palette_4.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Four,
    );
}

#[test]
fn interlaced_rgba_16_should_be_palette_2() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_palette_2.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::Indexed,
        BitDepth::Two,
    );
}

#[test]
fn interlaced_rgba_8_should_be_palette_2() {
    let input = PathBuf::from("tests/files/interlaced_rgba_8_should_be_palette_2.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Two,
    );
}

#[test]
fn interlaced_rgba_16_should_be_palette_1() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_palette_1.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_rgba_8_should_be_palette_1() {
    let input = PathBuf::from("tests/files/interlaced_rgba_8_should_be_palette_1.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_rgba_16_should_be_grayscale_alpha_16() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_grayscale_alpha_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::GrayscaleAlpha,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_rgba_16_should_be_grayscale_alpha_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_grayscale_alpha_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::GrayscaleAlpha,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_8_should_be_grayscale_alpha_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_8_should_be_grayscale_alpha_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Eight,
        ColorType::GrayscaleAlpha,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_16_should_be_grayscale_16() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_grayscale_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::Grayscale,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_rgba_16_should_be_grayscale_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_16_should_be_grayscale_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Sixteen,
        ColorType::Grayscale,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgba_8_should_be_grayscale_8() {
    let input = PathBuf::from("tests/files/interlaced_rgba_8_should_be_grayscale_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGBA,
        BitDepth::Eight,
        ColorType::Grayscale,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgb_16_should_be_rgb_16() {
    let input = PathBuf::from("tests/files/interlaced_rgb_16_should_be_rgb_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Sixteen,
        ColorType::RGB,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_rgb_16_should_be_rgb_8() {
    let input = PathBuf::from("tests/files/interlaced_rgb_16_should_be_rgb_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Sixteen,
        ColorType::RGB,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgb_8_should_be_rgb_8() {
    let input = PathBuf::from("tests/files/interlaced_rgb_8_should_be_rgb_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Eight,
        ColorType::RGB,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgb_16_should_be_palette_8() {
    let input = PathBuf::from("tests/files/interlaced_rgb_16_should_be_palette_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Sixteen,
        ColorType::Indexed,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgb_8_should_be_palette_8() {
    let input = PathBuf::from("tests/files/interlaced_rgb_8_should_be_palette_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgb_16_should_be_palette_4() {
    let input = PathBuf::from("tests/files/interlaced_rgb_16_should_be_palette_4.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Sixteen,
        ColorType::Indexed,
        BitDepth::Four,
    );
}

#[test]
fn interlaced_rgb_8_should_be_palette_4() {
    let input = PathBuf::from("tests/files/interlaced_rgb_8_should_be_palette_4.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Four,
    );
}

#[test]
fn interlaced_rgb_16_should_be_palette_2() {
    let input = PathBuf::from("tests/files/interlaced_rgb_16_should_be_palette_2.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Sixteen,
        ColorType::Indexed,
        BitDepth::Two,
    );
}

#[test]
fn interlaced_rgb_8_should_be_palette_2() {
    let input = PathBuf::from("tests/files/interlaced_rgb_8_should_be_palette_2.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Two,
    );
}

#[test]
fn interlaced_rgb_16_should_be_palette_1() {
    let input = PathBuf::from("tests/files/interlaced_rgb_16_should_be_palette_1.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Sixteen,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_rgb_8_should_be_palette_1() {
    let input = PathBuf::from("tests/files/interlaced_rgb_8_should_be_palette_1.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_rgb_16_should_be_grayscale_16() {
    let input = PathBuf::from("tests/files/interlaced_rgb_16_should_be_grayscale_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Sixteen,
        ColorType::Grayscale,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_rgb_16_should_be_grayscale_8() {
    let input = PathBuf::from("tests/files/interlaced_rgb_16_should_be_grayscale_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Sixteen,
        ColorType::Grayscale,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_rgb_8_should_be_grayscale_8() {
    let input = PathBuf::from("tests/files/interlaced_rgb_8_should_be_grayscale_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Eight,
        ColorType::Grayscale,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_palette_8_should_be_palette_8() {
    let input = PathBuf::from("tests/files/interlaced_palette_8_should_be_palette_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_palette_8_should_be_palette_4() {
    let input = PathBuf::from("tests/files/interlaced_palette_8_should_be_palette_4.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Four,
    );
}

#[test]
fn interlaced_palette_4_should_be_palette_4() {
    let input = PathBuf::from("tests/files/interlaced_palette_4_should_be_palette_4.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Four,
        ColorType::Indexed,
        BitDepth::Four,
    );
}

#[test]
fn interlaced_palette_8_should_be_palette_2() {
    let input = PathBuf::from("tests/files/interlaced_palette_8_should_be_palette_2.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::Two,
    );
}

#[test]
fn interlaced_palette_4_should_be_palette_2() {
    let input = PathBuf::from("tests/files/interlaced_palette_4_should_be_palette_2.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Four,
        ColorType::Indexed,
        BitDepth::Two,
    );
}

#[test]
fn interlaced_palette_2_should_be_palette_2() {
    let input = PathBuf::from("tests/files/interlaced_palette_2_should_be_palette_2.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Two,
        ColorType::Indexed,
        BitDepth::Two,
    );
}

#[test]
fn interlaced_palette_8_should_be_palette_1() {
    let input = PathBuf::from("tests/files/interlaced_palette_8_should_be_palette_1.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_palette_4_should_be_palette_1() {
    let input = PathBuf::from("tests/files/interlaced_palette_4_should_be_palette_1.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Four,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_palette_2_should_be_palette_1() {
    let input = PathBuf::from("tests/files/interlaced_palette_2_should_be_palette_1.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Two,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_palette_1_should_be_palette_1() {
    let input = PathBuf::from("tests/files/interlaced_palette_1_should_be_palette_1.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::One,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_grayscale_alpha_16_should_be_grayscale_alpha_16() {
    let input =
        PathBuf::from("tests/files/interlaced_grayscale_alpha_16_should_be_grayscale_alpha_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::GrayscaleAlpha,
        BitDepth::Sixteen,
        ColorType::GrayscaleAlpha,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_grayscale_alpha_16_should_be_grayscale_alpha_8() {
    let input =
        PathBuf::from("tests/files/interlaced_grayscale_alpha_16_should_be_grayscale_alpha_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::GrayscaleAlpha,
        BitDepth::Sixteen,
        ColorType::GrayscaleAlpha,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_grayscale_alpha_8_should_be_grayscale_alpha_8() {
    let input =
        PathBuf::from("tests/files/interlaced_grayscale_alpha_8_should_be_grayscale_alpha_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::GrayscaleAlpha,
        BitDepth::Eight,
        ColorType::GrayscaleAlpha,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_grayscale_alpha_16_should_be_grayscale_16() {
    let input =
        PathBuf::from("tests/files/interlaced_grayscale_alpha_16_should_be_grayscale_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::GrayscaleAlpha,
        BitDepth::Sixteen,
        ColorType::Grayscale,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_grayscale_alpha_16_should_be_grayscale_8() {
    let input =
        PathBuf::from("tests/files/interlaced_grayscale_alpha_16_should_be_grayscale_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::GrayscaleAlpha,
        BitDepth::Sixteen,
        ColorType::Grayscale,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_grayscale_alpha_8_should_be_grayscale_8() {
    let input = PathBuf::from("tests/files/interlaced_grayscale_alpha_8_should_be_grayscale_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::GrayscaleAlpha,
        BitDepth::Eight,
        ColorType::Grayscale,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_grayscale_16_should_be_grayscale_16() {
    let input = PathBuf::from("tests/files/interlaced_grayscale_16_should_be_grayscale_16.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Grayscale,
        BitDepth::Sixteen,
        ColorType::Grayscale,
        BitDepth::Sixteen,
    );
}

#[test]
fn interlaced_grayscale_16_should_be_grayscale_8() {
    let input = PathBuf::from("tests/files/interlaced_grayscale_16_should_be_grayscale_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Grayscale,
        BitDepth::Sixteen,
        ColorType::Grayscale,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_grayscale_8_should_be_grayscale_8() {
    let input = PathBuf::from("tests/files/interlaced_grayscale_8_should_be_grayscale_8.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Grayscale,
        BitDepth::Eight,
        ColorType::Grayscale,
        BitDepth::Eight,
    );
}

#[test]
fn interlaced_small_files() {
    let input = PathBuf::from("tests/files/interlaced_small_files.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::Indexed,
        BitDepth::Eight,
        ColorType::Indexed,
        BitDepth::One,
    );
}

#[test]
fn interlaced_odd_width() {
    let input = PathBuf::from("tests/files/interlaced_odd_width.png");
    let (output, opts) = get_opts(&input);

    test_it_converts(
        &input,
        &output,
        &opts,
        ColorType::RGB,
        BitDepth::Eight,
        ColorType::RGB,
        BitDepth::Eight,
    );
}