use derive_builder::Builder;
use getopts::Options;
use pdf_forms::{Form, FieldType};
use serde::{Deserialize, Serialize};

use std::env;
use std::fs::{File, OpenOptions};
use std::io::{Read, Write};
use std::path::Path;

#[derive(Debug, Default, Deserialize, Serialize)]
struct TextForm<'a> {
    #[serde(borrow)]
    fields: Vec<Field<'a>>,
}

#[derive(Debug, Builder, Deserialize, Serialize)]
struct Field<'a> {
    id: usize,

    #[builder(default)]
    value: Option<&'a str>,

    #[builder(default)]
    state: Option<bool>,
}

fn main() {
    let args: Vec<String> = env::args().collect();

    let mut opts = Options::new();
    opts.optflag("", "fill", "fill in the form using a markup file");
    opts.optflag("", "generate", "generate helper files to fill in the form");

    let opt_matches = opts.parse(&args[1..]).unwrap();

    if opt_matches.free.is_empty() {
        return;
    }

    let form_path = Path::new(&opt_matches.free[0]);
    let form_file_prefix = form_path.file_stem().unwrap().to_str().unwrap();
    let toml_file_name = format!("{}.toml", form_file_prefix);
    let toml_path = form_path.with_file_name(toml_file_name);
    let output_file_name = format!("{}-filled.pdf", form_file_prefix);

    let mut form = Form::load(form_path).unwrap();

    if opt_matches.opt_present("fill") {
        fill(
            toml_path,
            form_path.with_file_name(output_file_name),
            &mut form,
        );

        return;
    } else if opt_matches.opt_present("generate") {
        let ids_path = form_path.with_file_name(
            format!("{}-ids.pdf", form_file_prefix)
        );

        generate_fill_helpers(
            toml_path,
            ids_path,
            &mut form,
        );
    }
}

fn generate_fill_helpers<P: AsRef<Path>>(
    data_path: P,
    output_path: P,
    form: &mut Form,
) {
    let mut data = TextForm::default();

    for i in 0..form.len() {
        let field_type = form.get_type(i);

        match field_type {
            FieldType::Text => {
                form.set_text(i, format!("{}", i)).unwrap();

                data.fields.push(FieldBuilder::default()
                    .id(i)
                    .value(Some(""))
                    .build()
                    .unwrap()
                );
            },
            FieldType::CheckBox => {
                data.fields.push(FieldBuilder::default()
                    .id(i)
                    .state(Some(false))
                    .build()
                    .unwrap()
                );
            },
            _ => (),
        }
    }

    let mut toml_file = OpenOptions::new()
        .create_new(true)
        .write(true)
        .open(data_path)
        .unwrap();
    toml_file.write_all(&toml::to_vec(&data).unwrap()).unwrap();

    form.save(output_path).unwrap();
}

fn fill<P: AsRef<Path>>(
    data_path: P,
    output_path: P,
    form: &mut Form,
) {
    let mut buf = Vec::new();
    let mut file = File::open(data_path).unwrap();
    file.read_to_end(&mut buf).unwrap();

    let data: TextForm = toml::from_slice(&buf).unwrap();

    for field in data.fields {
        if let Some(value) = field.value {
            form.set_text(field.id, value.to_owned()).unwrap();
        } else if let Some(state) = field.state {
            form.set_check_box(field.id, state).unwrap();
        }
    }

    form.save(output_path).unwrap();
}