1 use anyhow::{anyhow, Result as AnyResult};
3 collections::{BTreeMap, HashSet, VecDeque},
5 fs::{read_to_string, File},
6 io::{Error as IoError, Write},
10 #[derive(Copy, Clone, PartialEq, Eq, Ord, PartialOrd)]
18 type CPNumber = usize;
20 fn process_converter<'a>(
21 fields: &Vec<&'a str>,
22 codepages: &mut BTreeMap<CPNumber, BTreeMap<Source, Vec<&'a str>>>,
24 if fields.is_empty() || fields[0] == "{" {
28 let mut cps: BTreeMap<Source, CPNumber> = BTreeMap::new();
29 let mut iana = VecDeque::new();
30 let mut other = VecDeque::new();
32 let mut iter = fields.iter().peekable();
33 while let Some(&name) = iter.next() {
34 if iter.next_if(|&&s| s == "{").is_some() {
35 let mut standards = HashSet::new();
37 let &standard = iter.next().expect("missing `}` in list of standards");
41 standards.insert(standard);
44 if standards.contains("IANA*") {
45 iana.push_front(name);
46 } else if standards.contains("IANA") {
48 } else if standards.iter().any(|&s| s.ends_with('*')) {
49 other.push_front(name);
51 other.push_back(name);
54 // Untagged names are completely nonstandard.
58 if let Some(number) = name.strip_prefix("cp") {
59 if let Ok(number) = number.parse::<CPNumber>() {
60 cps.insert(Source::CP, number);
64 if let Some(number) = name.strip_prefix("windows-") {
65 if let Ok(number) = number.parse::<CPNumber>() {
66 cps.insert(Source::Windows, number);
70 if let Some(number) = name.strip_prefix("ibm-") {
71 if let Ok(number) = number.parse::<CPNumber>() {
72 cps.insert(Source::IBM, number);
77 // If there are no tagged names then this is completely nonstandard.
78 if iana.is_empty() && other.is_empty() {
82 let all: Vec<&str> = iana.into_iter().chain(other.into_iter()).collect();
83 for (source, number) in cps {
86 .or_insert_with(BTreeMap::new)
87 .insert(source, all.clone());
92 codepages: &BTreeMap<CPNumber, BTreeMap<Source, Vec<&str>>>,
94 ) -> Result<(), IoError> {
95 let mut file = File::create(file_name)?;
98 use lazy_static::lazy_static;
99 use std::collections::HashMap;
102 static ref CODEPAGE_NUMBER_TO_NAME: HashMap<u32, &'static str> = {
103 let mut map = HashMap::new();
106 for (&cpnumber, value) in codepages.iter() {
107 let source = value.keys().max().unwrap();
108 let name = value[source][0];
109 writeln!(file, " map.insert({cpnumber}, \"{name}\");")?;
111 write!(file, "{}", "\
117 let mut names: BTreeMap<&str, BTreeMap<Source, Vec<CPNumber>>> = BTreeMap::new();
118 for (&cpnumber, value) in codepages.iter() {
119 for (&source, value2) in value.iter() {
120 for &name in value2.iter() {
123 .or_insert_with(BTreeMap::new)
125 .or_insert_with(Vec::new)
131 for (&name, value) in names.iter() {
132 for (_source, numbers) in value.iter().rev() {
133 println!(" {{ {}, \"{name}\" }},", numbers[0]);
141 fn main() -> AnyResult<()> {
142 println!("cargo:rerun-if-changed=build.rs");
144 let input_file = Path::new(env!("CARGO_MANIFEST_DIR")).join("../src/data/convrtrs.txt");
145 println!("cargo:rerun-if-changed={}", input_file.to_string_lossy());
146 let input = read_to_string(&input_file)
147 .map_err(|e| anyhow!("{}: read failed ({e})", input_file.to_string_lossy()))?;
149 let mut codepages: BTreeMap<CPNumber, BTreeMap<Source, Vec<&str>>> = BTreeMap::new();
150 let mut converter: Vec<&str> = Vec::new();
151 for line in input.lines() {
154 .map(|position| &line[..position])
157 if !line.starts_with(&[' ', '\t']) {
158 process_converter(&converter, &mut codepages);
161 converter.extend(line.split_whitespace());
163 process_converter(&converter, &mut codepages);
165 let output_file_name = Path::new(&var_os("OUT_DIR").unwrap()).join("encodings.rs");
167 write_output(&codepages, &output_file_name)
168 .map_err(|e| anyhow!("{}: write failed ({e})", output_file_name.to_string_lossy()))?;