omt 0.9.0-alpha

use std::fs::File;
use std::io::BufReader;

use resize::{Pixel, Type};
use rgb::FromSlice;

use crate::xcassets::appiconset::AppIconSet;

pub struct Xcassets {}

impl Xcassets {
	pub fn generate(input: &str, mode: &str, output: &str) -> anyhow::Result<u32> {
		let content_json_filename = format!("{}/Contents.json", output);
		let mut app_icon_set = match AppIconSet::load(&content_json_filename) {
			Ok(ais) => ais,
			Err(e) => {
				println!("ERROR: AppIconSet::load failed with {:?}", e);
				anyhow::bail!("problem with output file");
				//				return Err(OmError::Generic("problem with output file".to_string()));
			},
		};

		// load the source // :TODO: handle errors better
		let decoder = png::Decoder::new(BufReader::new(File::open(&input).unwrap()));
		//	    let (info, mut reader) = decoder.read_info().unwrap();
		let mut reader = decoder.read_info().unwrap();

		let info = reader.info();
		println!("{:#?}", info);
		let width = info.width;
		let height = info.height;

		//	    let mut src = vec![0; info.buffer_size()];
		let mut src = vec![0; reader.output_buffer_size().unwrap()];
		reader.next_frame(&mut src).unwrap();

		let src = src;

		let basename = "icon";
		let suffix = ".png";

		let mut number_of_changes = 0;
		let force = mode == "force";
		for mut ie in app_icon_set.images.iter_mut() {
			//			println!("{:?}", ie );
			let do_update = force;
			if !do_update {
				// :TODO: check fill
				// :TODO: check update
			}

			if do_update {
				// figure out size
				let scale = match &ie.scale[..] {
					"1x" => 1.0,
					"2x" => 2.0,
					"3x" => 3.0,
					// _ => return Err(OmError::Generic("Unsupported scale".to_string())),
					_ => anyhow::bail!("Unsupported scale"),
				};
				let size = ie.size.split("x").collect::<Vec<&str>>();
				let (w, h) = (size[0], size[1]);
				let w: f32 = w.parse::<f32>().expect("");
				let h: f32 = h.parse::<f32>().expect("");

				//				let scale = 1.0;
				//				let w = 1024.0;
				//				let h = 1024.0;

				println!("scale: {:?}, size: {:?} x {:?}", scale, w, h);

				let sw = (w * scale) as usize;
				let sh = (h * scale) as usize;

				// figure out filename
				let name = format!("{}-{}x{}@{}x{}", basename, sw, sh, scale, suffix);
				println!("{} <= scale: {:?}, size: {:?} x {:?}", name, scale, sw, sh);

				// create image at correct size

				let mut dst = vec![0; 3 * sw * sh];

				let mut resizer = match resize::new(
					width as usize,
					height as usize,
					sw,
					sh,
					Pixel::RGB8,
					Type::Lanczos3,
					//	Type::Mitchell,
				) {
					Ok(r) => r,
					Err(e) => {
						//return Err(OmError::Generic("Error creating resizer".to_string()));
						anyhow::bail!("Error creating resizer: {}", &e);
					},
				};

				match resizer.resize(src.as_rgb(), dst.as_rgb_mut()) {
					Ok(r) => r,
					Err(e) => {
						//return Err(OmError::Generic("Error resizing".to_string()));
						anyhow::bail!("Error resizing: {}", &e);
					},
				};

				let scaled_filename = format!("{}/{}", output, name);

				let outfh = File::create(scaled_filename).unwrap();
				let mut encoder = png::Encoder::new(outfh, sw as u32, sh as u32);
				encoder.set_color(png::ColorType::Rgb);
				encoder.set_depth(png::BitDepth::Eight);
				encoder
					.write_header()
					.unwrap()
					.write_image_data(&dst)
					.unwrap();
				//    			encoder.write_header().unwrap().write_image_data(&src).unwrap();

				ie.filename = Some(name);
				number_of_changes += 1;
			}
		}

		if number_of_changes > 0 {
			//			println!("{:#?}", app_icon_set);
			match app_icon_set.save(&content_json_filename) {
				Ok(_) => {},
				//				Err(_e) => return Err(OmError::Generic("Error saving app icon set".to_string())),
				Err(_e) => anyhow::bail!("Error saving app icon set"),
			}
		}

		Ok(number_of_changes)
	}
}