blob: d35d7d9deb9027104c089d7f9338308e60716f6d (
plain) (
blame)
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
121
122
123
124
125
|
//! Utilities to deal with character codes.
use crate::constant::TAB_SIZE;
use crate::tokenizer::Code;
/// Turn a string into codes.
pub fn parse(value: &str) -> Vec<Code> {
// Note: It’ll grow a bit bigger with each `Code::VirtualSpace`, smaller
// with `Code::CarriageReturnLineFeed`.
let mut codes: Vec<Code> = Vec::with_capacity(value.len());
let mut at_start = true;
let mut at_carriage_return = false;
let mut column = 1;
for char in value.chars() {
if at_start {
at_start = false;
if char == '\u{feff}' {
// Ignore.
continue;
}
}
// Send a CRLF.
if at_carriage_return && '\n' == char {
at_carriage_return = false;
codes.push(Code::CarriageReturnLineFeed);
} else {
// Send the previous CR: we’re not at a next `\n`.
if at_carriage_return {
at_carriage_return = false;
codes.push(Code::Char('\r'));
}
match char {
// Send a replacement character.
'\0' => {
column += 1;
codes.push(Code::Char(char::REPLACEMENT_CHARACTER));
}
// Send a tab and virtual spaces.
'\t' => {
let remainder = column % TAB_SIZE;
let mut virtual_spaces = if remainder == 0 {
0
} else {
TAB_SIZE - remainder
};
codes.push(Code::Char(char));
column += 1;
while virtual_spaces > 0 {
codes.push(Code::VirtualSpace);
column += 1;
virtual_spaces -= 1;
}
}
// Send an LF.
'\n' => {
column = 1;
codes.push(Code::Char(char));
}
// Don’t send anything yet.
'\r' => {
column = 1;
at_carriage_return = true;
}
// Send the char.
_ => {
column += 1;
codes.push(Code::Char(char));
}
}
};
}
// Send the last CR: we’re not at a next `\n`.
if at_carriage_return {
codes.push(Code::Char('\r'));
}
codes
}
/// Serialize codes, optionally expanding tabs.
pub fn serialize(codes: &[Code], expand_tabs: bool) -> String {
let mut at_tab = false;
// Note: It’ll grow a bit smaller with each
// `Code::Char('\t') | Code::VirtualSpace` if `expand_tabs` is false,
// and bigger with `Code::CarriageReturnLineFeed`,
let mut value = String::with_capacity(codes.len());
for code in codes {
let mut at_tab_next = false;
match code {
Code::CarriageReturnLineFeed => {
value.push_str("\r\n");
}
Code::Char(char) if *char == '\n' || *char == '\r' => {
value.push(*char);
}
Code::Char(char) if *char == '\t' => {
at_tab_next = true;
value.push(if expand_tabs { ' ' } else { *char });
}
Code::VirtualSpace => {
if !expand_tabs && at_tab {
continue;
}
value.push(' ');
}
Code::Char(char) => {
value.push(*char);
}
Code::None => {
unreachable!("unexpected EOF code in codes");
}
}
at_tab = at_tab_next;
}
value
}
|