use clap::Parser;
use image::Rgba;
use std::{
    collections::HashMap,
    fs::File,
    io::{BufRead, BufReader},
    path::PathBuf,
};

#[derive(Parser)]
struct Args {
    input: PathBuf,
    output: PathBuf,
}

fn main() {
    let Args {
        input,
        output: out_path,
    } = Args::parse();

    let palette = include_str!("../../../textures/palette.csv")
        .split('\n')
        .filter(|l| !l.is_empty())
        .map(|s| {
            let (c, s) = s.split_once(",").unwrap();
            let (r, s) = s.split_once(",").unwrap();
            let (g, s) = s.split_once(",").unwrap();
            let (b, a) = s.split_once(",").unwrap();
            (
                c.chars().next().unwrap(),
                [
                    r.parse().unwrap(),
                    g.parse().unwrap(),
                    b.parse().unwrap(),
                    a.parse().unwrap(),
                ],
            )
        })
        .collect::<HashMap<_, _>>();

    let file = BufReader::new(File::open(&input).unwrap());
    let tex = file.lines().map(Result::unwrap).collect::<Vec<String>>();
    let (width, height) = (tex[0].len(), tex.len());

    let mut output = image::ImageBuffer::<Rgba<u8>, Vec<_>>::new(width as u32, height as u32);

    for (y, line) in tex.iter().enumerate() {
        if line.is_empty() {
            continue;
        }
        for (x, char) in line.chars().enumerate() {
            output.put_pixel(x as u32, y as u32, Rgba(*palette.get(&char).unwrap()));
        }
    }

    output.save(out_path).unwrap();
}