Implement pretty-print for augumented script set.

crlf0710 · crlf0710 · commit a5d5f09bea41 · 2020-06-12T13:01:02.000+08:00
diff --git a/src/mixed_script.rs b/src/mixed_script.rs
@@ -1,11 +1,12 @@
 //! [Mixed-script detection](https://www.unicode.org/reports/tr39/#Mixed_Script_Detection)
 
+use core::fmt::{self, Debug};
 use unicode_script::{Script, ScriptExtension};
 
 /// An Augmented script set, as defined by UTS 39
 ///
 /// https://www.unicode.org/reports/tr39/#def-augmented-script-set
-#[derive(Copy, Clone, PartialEq, Debug, Hash, Eq)]
+#[derive(Copy, Clone, PartialEq, Hash, Eq)]
 pub struct AugmentedScriptSet {
     /// The base ScriptExtension value
     pub base: ScriptExtension,
@@ -72,6 +73,38 @@ impl Default for AugmentedScriptSet {
     }
 }
 
+impl Debug for AugmentedScriptSet {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        if self.is_empty() {
+            write!(f, "AugmentedScriptSet {{∅}}")?;
+        } else if self.is_all() {
+            write!(f, "AugmentedScriptSet {{ALL}}")?;
+        } else {
+            write!(f, "AugmentedScriptSet {{")?;
+            let mut first_entry = true;
+            let hanb = if self.hanb { Some("Hanb") } else { None };
+            let jpan = if self.jpan { Some("Jpan") } else { None };
+            let kore = if self.kore { Some("Kore") } else { None };
+            for writing_system in None
+                .into_iter()
+                .chain(hanb)
+                .chain(jpan)
+                .chain(kore)
+                .chain(self.base.iter().map(Script::short_name))
+            {
+                if !first_entry {
+                    write!(f, ", ")?;
+                } else {
+                    first_entry = false;
+                }
+                write!(f, "{}", writing_system)?;
+            }
+            write!(f, "}}")?;
+        }
+        Ok(())
+    }
+}
+
 impl AugmentedScriptSet {
     /// Intersect this set with another
     pub fn intersect_with(&mut self, other: Self) {
diff --git a/src/tests.rs b/src/tests.rs
@@ -77,3 +77,40 @@ fn test_potential_mixed_script_detection() {
     assert!(is_potential_mixed_script_confusable_char('A'));
     assert!(!is_potential_mixed_script_confusable_char('D'));
 }
+
+#[test]
+fn test_augmented_script_set() {
+    use crate::mixed_script::AugmentedScriptSet;
+    let augmented_script_sets = vec![
+        AugmentedScriptSet::default(),
+        AugmentedScriptSet::from('0'),
+        AugmentedScriptSet::from('a'),
+        AugmentedScriptSet::from('μ'),
+        AugmentedScriptSet::from('汉'),
+        AugmentedScriptSet::from('ひ'),
+        AugmentedScriptSet::from('カ'),
+        AugmentedScriptSet::from('한'),
+        AugmentedScriptSet::from("汉ひ"),
+        AugmentedScriptSet::from("汉a"),
+        AugmentedScriptSet::from("汉μ"),
+        AugmentedScriptSet::from("〆切"),
+    ];
+    let debug_output = vec![
+        "AugmentedScriptSet {ALL}",
+        "AugmentedScriptSet {ALL}",
+        "AugmentedScriptSet {Latn}",
+        "AugmentedScriptSet {Grek}",
+        "AugmentedScriptSet {Hanb, Jpan, Kore, Hani}",
+        "AugmentedScriptSet {Jpan, Hira}",
+        "AugmentedScriptSet {Jpan, Kana}",
+        "AugmentedScriptSet {Kore, Hang}",
+        "AugmentedScriptSet {Jpan}",
+        "AugmentedScriptSet {∅}",
+        "AugmentedScriptSet {∅}",
+        "AugmentedScriptSet {Hanb, Jpan, Kore, Hani}",
+    ];
+
+    for (ss, output) in augmented_script_sets.into_iter().zip(debug_output) {
+        assert_eq!(format!("{:?}", ss), output);
+    }
+}
