autocorrect 2.14.2

A linter and formatter for help you improve copywriting, to correct spaces, words, punctuations between CJK (Chinese, Japanese, Korean).
Documentation 
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

use std::borrow::Cow;

// autocorrect: false
enum SpaceMode {
    Add,
    Remove,
}

pub struct Strategery {
    space_mode: SpaceMode,
    reverse: bool,
    add_space_re: regex::Regex,
    add_space_reverse_re: regex::Regex,
    remove_space_re: regex::Regex,
    remove_space_reverse_re: regex::Regex,
}

impl Strategery {
    /// Create a new strategery object.
    pub fn new(one: &'static str, other: &'static str) -> Self {
        Strategery {
            space_mode: SpaceMode::Add,
            reverse: false,
            add_space_re: regexp!("({})({})", one, other),
            add_space_reverse_re: regexp!("({})({})", other, one),
            remove_space_re: regexp!("({})[ ]+({})", one, other),
            remove_space_reverse_re: regexp!("({})[ ]+({})", other, one),
        }
    }

    // Set Strategery for remove space.
    pub fn with_remove_space(mut self) -> Self {
        self.space_mode = SpaceMode::Remove;
        self
    }

    // Set Strategery for format by reverse again.
    pub fn with_reverse(mut self) -> Self {
        self.reverse = true;
        self
    }

    pub fn format<'a>(&self, text: &'a str) -> Cow<'a, str> {
        match self.space_mode {
            SpaceMode::Add => self.add_space(text),
            SpaceMode::Remove => self.remove_space(text),
        }
    }

    fn add_space<'a>(&self, text: &'a str) -> Cow<'a, str> {
        let out = self.add_space_re.replace_all(text, "$1 $2");
        if !self.reverse {
            return out;
        }

        match out {
            Cow::Borrowed(str) => self.add_space_reverse_re.replace_all(str, "$1 $2"),
            Cow::Owned(str) => match self.add_space_reverse_re.replace_all(&str, "$1 $2") {
                Cow::Borrowed(str) => Cow::Owned(str.to_owned()),
                Cow::Owned(str) => Cow::Owned(str),
            },
        }
    }

    fn remove_space<'a>(&self, text: &'a str) -> Cow<'a, str> {
        let out = self.remove_space_re.replace_all(text, "$1$2");
        if !self.reverse {
            return out;
        }

        match out {
            Cow::Borrowed(str) => self.remove_space_reverse_re.replace_all(str, "$1$2"),
            Cow::Owned(str) => match self.remove_space_reverse_re.replace_all(&str, "$1$2") {
                Cow::Borrowed(str) => Cow::Owned(str.to_owned()),
                Cow::Owned(str) => Cow::Owned(str),
            },
        }
    }
}