mirror of
https://github.com/crate-ci/typos.git
synced 2024-12-23 08:02:15 -05:00
chore(dict): Automate more cleanup
This commit is contained in:
parent
6cc3d16d3f
commit
09f096a096
3 changed files with 2245 additions and 2248 deletions
2228
crates/typos-dict/assets/words.csv
vendored
2228
crates/typos-dict/assets/words.csv
vendored
File diff suppressed because it is too large
Load diff
File diff suppressed because it is too large
Load diff
|
@ -1,9 +1,10 @@
|
||||||
use std::collections::BTreeMap;
|
use std::collections::BTreeMap;
|
||||||
|
use std::collections::BTreeSet;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::collections::HashSet;
|
use std::collections::HashSet;
|
||||||
use unicase::UniCase;
|
use unicase::UniCase;
|
||||||
|
|
||||||
type Dict = BTreeMap<UniCase<String>, Vec<String>>;
|
type Dict = BTreeMap<UniCase<String>, BTreeSet<String>>;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn verify() {
|
fn verify() {
|
||||||
|
@ -30,7 +31,9 @@ fn generate<W: std::io::Write>(file: &mut W, dict: &[u8]) {
|
||||||
let mut typo = i.next().expect("typo").to_owned();
|
let mut typo = i.next().expect("typo").to_owned();
|
||||||
typo.make_ascii_lowercase();
|
typo.make_ascii_lowercase();
|
||||||
let typo = UniCase::new(typo);
|
let typo = UniCase::new(typo);
|
||||||
rows.entry(typo).or_insert_with(Vec::new).extend(i.map(|c| {
|
rows.entry(typo)
|
||||||
|
.or_insert_with(BTreeSet::new)
|
||||||
|
.extend(i.map(|c| {
|
||||||
let mut c = c.to_owned();
|
let mut c = c.to_owned();
|
||||||
c.make_ascii_lowercase();
|
c.make_ascii_lowercase();
|
||||||
c
|
c
|
||||||
|
@ -41,7 +44,7 @@ fn generate<W: std::io::Write>(file: &mut W, dict: &[u8]) {
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.filter(|(t, _)| is_word(t))
|
.filter(|(t, _)| is_word(t))
|
||||||
.filter_map(|(t, c)| {
|
.filter_map(|(t, c)| {
|
||||||
let new_c: Vec<_> = c.into_iter().filter(|c| is_word(c)).collect();
|
let new_c: BTreeSet<_> = c.into_iter().filter(|c| is_word(c)).collect();
|
||||||
if new_c.is_empty() {
|
if new_c.is_empty() {
|
||||||
None
|
None
|
||||||
} else {
|
} else {
|
||||||
|
@ -53,7 +56,7 @@ fn generate<W: std::io::Write>(file: &mut W, dict: &[u8]) {
|
||||||
let varcon_words = varcon_words();
|
let varcon_words = varcon_words();
|
||||||
let allowed_words = allowed_words();
|
let allowed_words = allowed_words();
|
||||||
let word_variants = proper_word_variants();
|
let word_variants = proper_word_variants();
|
||||||
let rows: Dict = rows
|
let rows: Vec<_> = rows
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.filter(|(typo, _)| {
|
.filter(|(typo, _)| {
|
||||||
let is_disallowed = varcon_words.contains(&unicase::UniCase::new(typo));
|
let is_disallowed = varcon_words.contains(&unicase::UniCase::new(typo));
|
||||||
|
@ -71,29 +74,34 @@ fn generate<W: std::io::Write>(file: &mut W, dict: &[u8]) {
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
.map(|(typo, corrections)| {
|
.map(|(typo, corrections)| {
|
||||||
let mut new_corrections = vec![];
|
let mut new_corrections = BTreeSet::new();
|
||||||
for correction in corrections {
|
for correction in corrections {
|
||||||
let correction = word_variants
|
let correction = word_variants
|
||||||
.get(correction.as_str())
|
.get(correction.as_str())
|
||||||
.and_then(|words| find_best_match(&typo, correction.as_str(), words))
|
.and_then(|words| find_best_match(&typo, correction.as_str(), words))
|
||||||
.unwrap_or(&correction);
|
.unwrap_or(&correction);
|
||||||
new_corrections.push(correction.to_owned());
|
new_corrections.insert(correction.to_owned());
|
||||||
}
|
}
|
||||||
(typo, new_corrections)
|
(typo, new_corrections)
|
||||||
})
|
})
|
||||||
.collect();
|
.collect();
|
||||||
|
let mut dict = Dict::new();
|
||||||
|
for (bad, good) in rows {
|
||||||
|
let current = dict.entry(bad).or_default();
|
||||||
|
current.extend(good);
|
||||||
|
}
|
||||||
|
|
||||||
let corrections: std::collections::HashSet<_> =
|
let corrections: std::collections::HashSet<_> =
|
||||||
rows.values().flatten().map(ToOwned::to_owned).collect();
|
dict.values().flatten().map(ToOwned::to_owned).collect();
|
||||||
let rows: Vec<_> = rows
|
let rows: Vec<_> = dict
|
||||||
.into_iter()
|
.into_iter()
|
||||||
.filter(|(typo, _)| !corrections.contains(typo.as_str()))
|
.filter(|(typo, _)| !corrections.contains(typo.as_str()))
|
||||||
.collect();
|
.collect();
|
||||||
|
|
||||||
let mut wtr = csv::WriterBuilder::new().flexible(true).from_writer(file);
|
let mut wtr = csv::WriterBuilder::new().flexible(true).from_writer(file);
|
||||||
for (typo, corrections) in rows {
|
for (typo, corrections) in rows {
|
||||||
let mut row = corrections;
|
let mut row = vec![typo.as_str().to_owned()];
|
||||||
row.insert(0, typo.as_str().to_owned());
|
row.extend(corrections);
|
||||||
wtr.write_record(&row).unwrap();
|
wtr.write_record(&row).unwrap();
|
||||||
}
|
}
|
||||||
wtr.flush().unwrap();
|
wtr.flush().unwrap();
|
||||||
|
|
Loading…
Reference in a new issue