1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
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");
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;
}
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(toml_path)
.unwrap();
toml_file.write_all(&toml::to_vec(&data).unwrap()).unwrap();
form.save(
form_path.with_file_name(
format!("{}-ids.pdf", form_file_prefix)
),
).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();
}
|
