Alex Crichton | 76a5cc8 | 2017-05-23 07:01:44 -0700 | [diff] [blame] | 1 | use std::ascii; |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 2 | use std::borrow::Borrow; |
| 3 | use std::cell::RefCell; |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 4 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 5 | use std::cmp; |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 6 | use std::collections::HashMap; |
| 7 | use std::fmt; |
| 8 | use std::iter; |
David Tolnay | 041bcd4 | 2017-06-03 09:18:04 -0700 | [diff] [blame] | 9 | use std::marker::PhantomData; |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 10 | use std::ops; |
| 11 | use std::rc::Rc; |
| 12 | use std::str::FromStr; |
| 13 | use std::vec; |
| 14 | |
| 15 | use proc_macro; |
David Tolnay | b103266 | 2017-05-31 15:52:28 -0700 | [diff] [blame] | 16 | use unicode_xid::UnicodeXID; |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 17 | use strnom::{Cursor, PResult, skip_whitespace, block_comment, whitespace, word_break}; |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 18 | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 19 | use {TokenTree, TokenNode, Delimiter, Spacing}; |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 20 | |
David Tolnay | 977f828 | 2017-05-31 17:41:33 -0700 | [diff] [blame] | 21 | #[derive(Clone, Debug)] |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 22 | pub struct TokenStream { |
| 23 | inner: Vec<TokenTree>, |
| 24 | } |
| 25 | |
| 26 | #[derive(Debug)] |
| 27 | pub struct LexError; |
| 28 | |
| 29 | impl TokenStream { |
| 30 | pub fn empty() -> TokenStream { |
| 31 | TokenStream { inner: Vec::new() } |
| 32 | } |
| 33 | |
| 34 | pub fn is_empty(&self) -> bool { |
| 35 | self.inner.len() == 0 |
| 36 | } |
| 37 | } |
| 38 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 39 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | a9dbc18 | 2017-12-30 14:50:13 -0500 | [diff] [blame] | 40 | fn get_cursor(src: &str) -> Cursor { |
| 41 | // Create a dummy file & add it to the codemap |
| 42 | CODEMAP.with(|cm| { |
| 43 | let mut cm = cm.borrow_mut(); |
| 44 | let name = format!("<parsed string {}>", cm.files.len()); |
| 45 | let span = cm.add_file(&name, src); |
| 46 | Cursor { |
| 47 | rest: src, |
| 48 | off: span.lo, |
| 49 | } |
| 50 | }) |
| 51 | } |
| 52 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 53 | #[cfg(not(procmacro2_semver_exempt))] |
Nika Layzell | a9dbc18 | 2017-12-30 14:50:13 -0500 | [diff] [blame] | 54 | fn get_cursor(src: &str) -> Cursor { |
| 55 | Cursor { |
| 56 | rest: src, |
Nika Layzell | a9dbc18 | 2017-12-30 14:50:13 -0500 | [diff] [blame] | 57 | } |
| 58 | } |
| 59 | |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 60 | impl FromStr for TokenStream { |
| 61 | type Err = LexError; |
| 62 | |
| 63 | fn from_str(src: &str) -> Result<TokenStream, LexError> { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 64 | // Create a dummy file & add it to the codemap |
Nika Layzell | a9dbc18 | 2017-12-30 14:50:13 -0500 | [diff] [blame] | 65 | let cursor = get_cursor(src); |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 66 | |
| 67 | match token_stream(cursor) { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 68 | Ok((input, output)) => { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 69 | if skip_whitespace(input).len() != 0 { |
| 70 | Err(LexError) |
| 71 | } else { |
David Tolnay | 8e976c6 | 2017-06-01 12:12:29 -0700 | [diff] [blame] | 72 | Ok(output.0) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 73 | } |
| 74 | } |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 75 | Err(LexError) => Err(LexError), |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 76 | } |
| 77 | } |
| 78 | } |
| 79 | |
| 80 | impl fmt::Display for TokenStream { |
| 81 | fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { |
| 82 | let mut joint = false; |
| 83 | for (i, tt) in self.inner.iter().enumerate() { |
| 84 | if i != 0 && !joint { |
| 85 | write!(f, " ")?; |
| 86 | } |
| 87 | joint = false; |
| 88 | match tt.kind { |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 89 | TokenNode::Group(delim, ref stream) => { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 90 | let (start, end) = match delim { |
| 91 | Delimiter::Parenthesis => ("(", ")"), |
| 92 | Delimiter::Brace => ("{", "}"), |
| 93 | Delimiter::Bracket => ("[", "]"), |
| 94 | Delimiter::None => ("", ""), |
| 95 | }; |
Alex Crichton | 852d53d | 2017-05-19 19:25:08 -0700 | [diff] [blame] | 96 | if stream.0.inner.len() == 0 { |
| 97 | write!(f, "{} {}", start, end)? |
| 98 | } else { |
| 99 | write!(f, "{} {} {}", start, stream, end)? |
| 100 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 101 | } |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 102 | TokenNode::Term(ref sym) => write!(f, "{}", sym.as_str())?, |
| 103 | TokenNode::Op(ch, ref op) => { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 104 | write!(f, "{}", ch)?; |
| 105 | match *op { |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 106 | Spacing::Alone => {} |
| 107 | Spacing::Joint => joint = true, |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 108 | } |
| 109 | } |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 110 | TokenNode::Literal(ref literal) => { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 111 | write!(f, "{}", literal)?; |
| 112 | // handle comments |
| 113 | if (literal.0).0.starts_with("/") { |
| 114 | write!(f, "\n")?; |
| 115 | } |
| 116 | } |
| 117 | } |
| 118 | } |
| 119 | |
| 120 | Ok(()) |
| 121 | } |
| 122 | } |
| 123 | |
| 124 | impl From<proc_macro::TokenStream> for TokenStream { |
| 125 | fn from(inner: proc_macro::TokenStream) -> TokenStream { |
| 126 | inner.to_string().parse().expect("compiler token stream parse failed") |
| 127 | } |
| 128 | } |
| 129 | |
| 130 | impl From<TokenStream> for proc_macro::TokenStream { |
| 131 | fn from(inner: TokenStream) -> proc_macro::TokenStream { |
| 132 | inner.to_string().parse().expect("failed to parse to compiler tokens") |
| 133 | } |
| 134 | } |
| 135 | |
| 136 | |
| 137 | impl From<TokenTree> for TokenStream { |
| 138 | fn from(tree: TokenTree) -> TokenStream { |
| 139 | TokenStream { inner: vec![tree] } |
| 140 | } |
| 141 | } |
| 142 | |
| 143 | impl iter::FromIterator<TokenStream> for TokenStream { |
| 144 | fn from_iter<I: IntoIterator<Item=TokenStream>>(streams: I) -> Self { |
| 145 | let mut v = Vec::new(); |
| 146 | |
| 147 | for stream in streams.into_iter() { |
| 148 | v.extend(stream.inner); |
| 149 | } |
| 150 | |
| 151 | TokenStream { inner: v } |
| 152 | } |
| 153 | } |
| 154 | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 155 | pub type TokenTreeIter = vec::IntoIter<TokenTree>; |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 156 | |
| 157 | impl IntoIterator for TokenStream { |
| 158 | type Item = TokenTree; |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 159 | type IntoIter = TokenTreeIter; |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 160 | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 161 | fn into_iter(self) -> TokenTreeIter { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 162 | self.inner.into_iter() |
| 163 | } |
| 164 | } |
| 165 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 166 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | b35a9a3 | 2017-12-30 14:34:35 -0500 | [diff] [blame] | 167 | #[derive(Clone, PartialEq, Eq, Debug)] |
| 168 | pub struct FileName(String); |
| 169 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 170 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | b35a9a3 | 2017-12-30 14:34:35 -0500 | [diff] [blame] | 171 | impl fmt::Display for FileName { |
| 172 | fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { |
| 173 | self.0.fmt(f) |
| 174 | } |
| 175 | } |
| 176 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 177 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 178 | #[derive(Clone, PartialEq, Eq)] |
| 179 | pub struct SourceFile { |
Nika Layzell | b35a9a3 | 2017-12-30 14:34:35 -0500 | [diff] [blame] | 180 | name: FileName, |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 181 | } |
| 182 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 183 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 184 | impl SourceFile { |
| 185 | /// Get the path to this source file as a string. |
Nika Layzell | b35a9a3 | 2017-12-30 14:34:35 -0500 | [diff] [blame] | 186 | pub fn path(&self) -> &FileName { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 187 | &self.name |
| 188 | } |
| 189 | |
| 190 | pub fn is_real(&self) -> bool { |
| 191 | // XXX(nika): Support real files in the future? |
| 192 | false |
| 193 | } |
| 194 | } |
| 195 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 196 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | b35a9a3 | 2017-12-30 14:34:35 -0500 | [diff] [blame] | 197 | impl AsRef<FileName> for SourceFile { |
| 198 | fn as_ref(&self) -> &FileName { |
| 199 | self.path() |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 200 | } |
| 201 | } |
| 202 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 203 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 204 | impl fmt::Debug for SourceFile { |
| 205 | fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { |
| 206 | f.debug_struct("SourceFile") |
Nika Layzell | b35a9a3 | 2017-12-30 14:34:35 -0500 | [diff] [blame] | 207 | .field("path", &self.path()) |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 208 | .field("is_real", &self.is_real()) |
| 209 | .finish() |
| 210 | } |
| 211 | } |
| 212 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 213 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 214 | #[derive(Clone, Copy, Debug, PartialEq, Eq)] |
| 215 | pub struct LineColumn { |
| 216 | pub line: usize, |
| 217 | pub column: usize, |
| 218 | } |
| 219 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 220 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 221 | thread_local! { |
| 222 | static CODEMAP: RefCell<Codemap> = RefCell::new(Codemap { |
| 223 | // NOTE: We start with a single dummy file which all call_site() and |
| 224 | // def_site() spans reference. |
| 225 | files: vec![FileInfo { |
| 226 | name: "<unspecified>".to_owned(), |
| 227 | span: Span { lo: 0, hi: 0 }, |
| 228 | lines: vec![0], |
| 229 | }], |
| 230 | }); |
| 231 | } |
| 232 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 233 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 234 | struct FileInfo { |
| 235 | name: String, |
| 236 | span: Span, |
| 237 | lines: Vec<usize>, |
| 238 | } |
| 239 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 240 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 241 | impl FileInfo { |
| 242 | fn offset_line_column(&self, offset: usize) -> LineColumn { |
| 243 | assert!(self.span_within(Span { lo: offset as u32, hi: offset as u32 })); |
| 244 | let offset = offset - self.span.lo as usize; |
| 245 | match self.lines.binary_search(&offset) { |
| 246 | Ok(found) => LineColumn { |
| 247 | line: found + 1, |
| 248 | column: 0 |
| 249 | }, |
| 250 | Err(idx) => LineColumn { |
| 251 | line: idx, |
| 252 | column: offset - self.lines[idx - 1] |
| 253 | }, |
| 254 | } |
| 255 | } |
| 256 | |
| 257 | fn span_within(&self, span: Span) -> bool { |
| 258 | span.lo >= self.span.lo && span.hi <= self.span.hi |
| 259 | } |
| 260 | } |
| 261 | |
| 262 | /// Computes the offsets of each line in the given source string. |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 263 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | a0a7c3d | 2017-12-30 14:52:39 -0500 | [diff] [blame] | 264 | fn lines_offsets(s: &str) -> Vec<usize> { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 265 | let mut lines = vec![0]; |
| 266 | let mut prev = 0; |
Nika Layzell | a0a7c3d | 2017-12-30 14:52:39 -0500 | [diff] [blame] | 267 | while let Some(len) = s[prev..].find('\n') { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 268 | prev += len + 1; |
| 269 | lines.push(prev); |
| 270 | } |
| 271 | lines |
| 272 | } |
| 273 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 274 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 275 | struct Codemap { |
| 276 | files: Vec<FileInfo>, |
| 277 | } |
| 278 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 279 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 280 | impl Codemap { |
| 281 | fn next_start_pos(&self) -> u32 { |
| 282 | // Add 1 so there's always space between files. |
| 283 | // |
| 284 | // We'll always have at least 1 file, as we initialize our files list |
| 285 | // with a dummy file. |
| 286 | self.files.last().unwrap().span.hi + 1 |
| 287 | } |
| 288 | |
| 289 | fn add_file(&mut self, name: &str, src: &str) -> Span { |
Nika Layzell | a0a7c3d | 2017-12-30 14:52:39 -0500 | [diff] [blame] | 290 | let lines = lines_offsets(src); |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 291 | let lo = self.next_start_pos(); |
| 292 | // XXX(nika): Shouild we bother doing a checked cast or checked add here? |
| 293 | let span = Span { lo: lo, hi: lo + (src.len() as u32) }; |
| 294 | |
| 295 | self.files.push(FileInfo { |
| 296 | name: name.to_owned(), |
| 297 | span: span, |
| 298 | lines: lines, |
| 299 | }); |
| 300 | |
| 301 | span |
| 302 | } |
| 303 | |
| 304 | fn fileinfo(&self, span: Span) -> &FileInfo { |
| 305 | for file in &self.files { |
| 306 | if file.span_within(span) { |
| 307 | return file; |
| 308 | } |
| 309 | } |
| 310 | panic!("Invalid span with no related FileInfo!"); |
| 311 | } |
| 312 | } |
| 313 | |
Alex Crichton | e6085b7 | 2017-11-21 07:24:25 -0800 | [diff] [blame] | 314 | #[derive(Clone, Copy, Debug)] |
David Tolnay | ddfca05 | 2017-12-31 10:41:24 -0500 | [diff] [blame] | 315 | pub struct Span { |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 316 | #[cfg(procmacro2_semver_exempt)] |
David Tolnay | ddfca05 | 2017-12-31 10:41:24 -0500 | [diff] [blame] | 317 | lo: u32, |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 318 | #[cfg(procmacro2_semver_exempt)] |
David Tolnay | ddfca05 | 2017-12-31 10:41:24 -0500 | [diff] [blame] | 319 | hi: u32, |
| 320 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 321 | |
| 322 | impl Span { |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 323 | #[cfg(not(procmacro2_semver_exempt))] |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 324 | pub fn call_site() -> Span { |
David Tolnay | 79105e5 | 2017-12-31 11:03:04 -0500 | [diff] [blame] | 325 | Span {} |
| 326 | } |
| 327 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 328 | #[cfg(procmacro2_semver_exempt)] |
David Tolnay | 79105e5 | 2017-12-31 11:03:04 -0500 | [diff] [blame] | 329 | pub fn call_site() -> Span { |
| 330 | Span { lo: 0, hi: 0 } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 331 | } |
Alex Crichton | e6085b7 | 2017-11-21 07:24:25 -0800 | [diff] [blame] | 332 | |
| 333 | pub fn def_site() -> Span { |
David Tolnay | 79105e5 | 2017-12-31 11:03:04 -0500 | [diff] [blame] | 334 | Span::call_site() |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 335 | } |
| 336 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 337 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 338 | pub fn source_file(&self) -> SourceFile { |
| 339 | CODEMAP.with(|cm| { |
| 340 | let cm = cm.borrow(); |
| 341 | let fi = cm.fileinfo(*self); |
| 342 | SourceFile { |
Nika Layzell | b35a9a3 | 2017-12-30 14:34:35 -0500 | [diff] [blame] | 343 | name: FileName(fi.name.clone()), |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 344 | } |
| 345 | }) |
| 346 | } |
| 347 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 348 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 349 | pub fn start(&self) -> LineColumn { |
| 350 | CODEMAP.with(|cm| { |
| 351 | let cm = cm.borrow(); |
| 352 | let fi = cm.fileinfo(*self); |
| 353 | fi.offset_line_column(self.lo as usize) |
| 354 | }) |
| 355 | } |
| 356 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 357 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 358 | pub fn end(&self) -> LineColumn { |
| 359 | CODEMAP.with(|cm| { |
| 360 | let cm = cm.borrow(); |
| 361 | let fi = cm.fileinfo(*self); |
| 362 | fi.offset_line_column(self.hi as usize) |
| 363 | }) |
| 364 | } |
| 365 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 366 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 367 | pub fn join(&self, other: Span) -> Option<Span> { |
| 368 | CODEMAP.with(|cm| { |
| 369 | let cm = cm.borrow(); |
| 370 | // If `other` is not within the same FileInfo as us, return None. |
| 371 | if !cm.fileinfo(*self).span_within(other) { |
| 372 | return None; |
| 373 | } |
| 374 | Some(Span { |
| 375 | lo: cmp::min(self.lo, other.lo), |
| 376 | hi: cmp::max(self.hi, other.hi), |
| 377 | }) |
| 378 | }) |
Alex Crichton | e6085b7 | 2017-11-21 07:24:25 -0800 | [diff] [blame] | 379 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 380 | } |
| 381 | |
David Tolnay | 8ad3e3e | 2017-06-03 16:45:00 -0700 | [diff] [blame] | 382 | #[derive(Copy, Clone)] |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 383 | pub struct Term { |
David Tolnay | 041bcd4 | 2017-06-03 09:18:04 -0700 | [diff] [blame] | 384 | intern: usize, |
| 385 | not_send_sync: PhantomData<*const ()>, |
| 386 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 387 | |
| 388 | thread_local!(static SYMBOLS: RefCell<Interner> = RefCell::new(Interner::new())); |
| 389 | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 390 | impl<'a> From<&'a str> for Term { |
| 391 | fn from(string: &'a str) -> Term { |
| 392 | Term { |
David Tolnay | 041bcd4 | 2017-06-03 09:18:04 -0700 | [diff] [blame] | 393 | intern: SYMBOLS.with(|s| s.borrow_mut().intern(string)), |
| 394 | not_send_sync: PhantomData, |
| 395 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 396 | } |
| 397 | } |
| 398 | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 399 | impl ops::Deref for Term { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 400 | type Target = str; |
| 401 | |
| 402 | fn deref(&self) -> &str { |
| 403 | SYMBOLS.with(|interner| { |
| 404 | let interner = interner.borrow(); |
David Tolnay | 041bcd4 | 2017-06-03 09:18:04 -0700 | [diff] [blame] | 405 | let s = interner.get(self.intern); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 406 | unsafe { |
| 407 | &*(s as *const str) |
| 408 | } |
| 409 | }) |
| 410 | } |
| 411 | } |
| 412 | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 413 | impl fmt::Debug for Term { |
David Tolnay | 8ad3e3e | 2017-06-03 16:45:00 -0700 | [diff] [blame] | 414 | fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 415 | f.debug_tuple("Term").field(&&**self).finish() |
David Tolnay | 8ad3e3e | 2017-06-03 16:45:00 -0700 | [diff] [blame] | 416 | } |
| 417 | } |
| 418 | |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 419 | struct Interner { |
| 420 | string_to_index: HashMap<MyRc, usize>, |
| 421 | index_to_string: Vec<Rc<String>>, |
| 422 | } |
| 423 | |
| 424 | #[derive(Hash, Eq, PartialEq)] |
| 425 | struct MyRc(Rc<String>); |
| 426 | |
| 427 | impl Borrow<str> for MyRc { |
| 428 | fn borrow(&self) -> &str { |
| 429 | &self.0 |
| 430 | } |
| 431 | } |
| 432 | |
| 433 | impl Interner { |
| 434 | fn new() -> Interner { |
| 435 | Interner { |
| 436 | string_to_index: HashMap::new(), |
| 437 | index_to_string: Vec::new(), |
| 438 | } |
| 439 | } |
| 440 | |
| 441 | fn intern(&mut self, s: &str) -> usize { |
| 442 | if let Some(&idx) = self.string_to_index.get(s) { |
| 443 | return idx |
| 444 | } |
| 445 | let s = Rc::new(s.to_string()); |
| 446 | self.index_to_string.push(s.clone()); |
| 447 | self.string_to_index.insert(MyRc(s), self.index_to_string.len() - 1); |
| 448 | self.index_to_string.len() - 1 |
| 449 | } |
| 450 | |
| 451 | fn get(&self, idx: usize) -> &str { |
| 452 | &self.index_to_string[idx] |
| 453 | } |
| 454 | } |
| 455 | |
David Tolnay | 977f828 | 2017-05-31 17:41:33 -0700 | [diff] [blame] | 456 | #[derive(Clone, Debug)] |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 457 | pub struct Literal(String); |
| 458 | |
Alex Crichton | 852d53d | 2017-05-19 19:25:08 -0700 | [diff] [blame] | 459 | impl Literal { |
Alex Crichton | 9c2fb0a | 2017-05-26 08:49:31 -0700 | [diff] [blame] | 460 | pub fn byte_char(byte: u8) -> Literal { |
Alex Crichton | 76a5cc8 | 2017-05-23 07:01:44 -0700 | [diff] [blame] | 461 | match byte { |
| 462 | 0 => Literal(format!("b'\\0'")), |
| 463 | b'\"' => Literal(format!("b'\"'")), |
| 464 | n => { |
| 465 | let mut escaped = "b'".to_string(); |
| 466 | escaped.extend(ascii::escape_default(n).map(|c| c as char)); |
| 467 | escaped.push('\''); |
| 468 | Literal(escaped) |
| 469 | } |
| 470 | } |
| 471 | } |
| 472 | |
Alex Crichton | 9c2fb0a | 2017-05-26 08:49:31 -0700 | [diff] [blame] | 473 | pub fn byte_string(bytes: &[u8]) -> Literal { |
Alex Crichton | 852d53d | 2017-05-19 19:25:08 -0700 | [diff] [blame] | 474 | let mut escaped = "b\"".to_string(); |
| 475 | for b in bytes { |
| 476 | match *b { |
| 477 | b'\0' => escaped.push_str(r"\0"), |
| 478 | b'\t' => escaped.push_str(r"\t"), |
| 479 | b'\n' => escaped.push_str(r"\n"), |
| 480 | b'\r' => escaped.push_str(r"\r"), |
| 481 | b'"' => escaped.push_str("\\\""), |
| 482 | b'\\' => escaped.push_str("\\\\"), |
| 483 | b'\x20' ... b'\x7E' => escaped.push(*b as char), |
| 484 | _ => escaped.push_str(&format!("\\x{:02X}", b)), |
| 485 | } |
| 486 | } |
| 487 | escaped.push('"'); |
| 488 | Literal(escaped) |
| 489 | } |
Alex Crichton | 76a5cc8 | 2017-05-23 07:01:44 -0700 | [diff] [blame] | 490 | |
| 491 | pub fn doccomment(s: &str) -> Literal { |
| 492 | Literal(s.to_string()) |
| 493 | } |
Alex Crichton | 9c2fb0a | 2017-05-26 08:49:31 -0700 | [diff] [blame] | 494 | |
David Tolnay | 114990e | 2018-01-05 11:17:08 -0800 | [diff] [blame^] | 495 | pub fn float(n: f64) -> Literal { |
| 496 | if !n.is_finite() { |
| 497 | panic!("Invalid float literal {}", n); |
| 498 | } |
| 499 | let mut s = n.to_string(); |
| 500 | if !s.contains('.') { |
| 501 | s += ".0"; |
| 502 | } |
| 503 | Literal(s) |
Alex Crichton | 9c2fb0a | 2017-05-26 08:49:31 -0700 | [diff] [blame] | 504 | } |
| 505 | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 506 | pub fn integer(s: i64) -> Literal { |
Alex Crichton | 9c2fb0a | 2017-05-26 08:49:31 -0700 | [diff] [blame] | 507 | Literal(s.to_string()) |
| 508 | } |
Alex Crichton | 3131662 | 2017-05-26 12:54:47 -0700 | [diff] [blame] | 509 | |
| 510 | pub fn raw_string(s: &str, pounds: usize) -> Literal { |
| 511 | let mut ret = format!("r"); |
| 512 | ret.extend((0..pounds).map(|_| "#")); |
| 513 | ret.push('"'); |
| 514 | ret.push_str(s); |
| 515 | ret.push('"'); |
| 516 | ret.extend((0..pounds).map(|_| "#")); |
| 517 | Literal(ret) |
| 518 | } |
| 519 | |
| 520 | pub fn raw_byte_string(s: &str, pounds: usize) -> Literal { |
Alex Crichton | 7ed6d28 | 2017-05-26 13:42:50 -0700 | [diff] [blame] | 521 | let mut ret = format!("br"); |
Alex Crichton | 3131662 | 2017-05-26 12:54:47 -0700 | [diff] [blame] | 522 | ret.extend((0..pounds).map(|_| "#")); |
| 523 | ret.push('"'); |
| 524 | ret.push_str(s); |
| 525 | ret.push('"'); |
| 526 | ret.extend((0..pounds).map(|_| "#")); |
| 527 | Literal(ret) |
| 528 | } |
Alex Crichton | 852d53d | 2017-05-19 19:25:08 -0700 | [diff] [blame] | 529 | } |
| 530 | |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 531 | impl fmt::Display for Literal { |
| 532 | fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { |
| 533 | self.0.fmt(f) |
| 534 | } |
| 535 | } |
| 536 | |
Alex Crichton | 9c2fb0a | 2017-05-26 08:49:31 -0700 | [diff] [blame] | 537 | macro_rules! ints { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 538 | ($($t:ty,)*) => {$( |
| 539 | impl From<$t> for Literal { |
| 540 | fn from(t: $t) -> Literal { |
Alex Crichton | 852d53d | 2017-05-19 19:25:08 -0700 | [diff] [blame] | 541 | Literal(format!(concat!("{}", stringify!($t)), t)) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 542 | } |
| 543 | } |
| 544 | )*} |
| 545 | } |
| 546 | |
Alex Crichton | 9c2fb0a | 2017-05-26 08:49:31 -0700 | [diff] [blame] | 547 | ints! { |
Alex Crichton | 852d53d | 2017-05-19 19:25:08 -0700 | [diff] [blame] | 548 | u8, u16, u32, u64, usize, |
| 549 | i8, i16, i32, i64, isize, |
Alex Crichton | 9c2fb0a | 2017-05-26 08:49:31 -0700 | [diff] [blame] | 550 | } |
| 551 | |
| 552 | macro_rules! floats { |
| 553 | ($($t:ty,)*) => {$( |
| 554 | impl From<$t> for Literal { |
| 555 | fn from(t: $t) -> Literal { |
| 556 | assert!(!t.is_nan()); |
| 557 | assert!(!t.is_infinite()); |
| 558 | Literal(format!(concat!("{}", stringify!($t)), t)) |
| 559 | } |
| 560 | } |
| 561 | )*} |
| 562 | } |
| 563 | |
| 564 | floats! { |
Alex Crichton | 852d53d | 2017-05-19 19:25:08 -0700 | [diff] [blame] | 565 | f32, f64, |
| 566 | } |
| 567 | |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 568 | impl<'a> From<&'a str> for Literal { |
| 569 | fn from(t: &'a str) -> Literal { |
| 570 | let mut s = t.chars().flat_map(|c| c.escape_default()).collect::<String>(); |
| 571 | s.push('"'); |
| 572 | s.insert(0, '"'); |
| 573 | Literal(s) |
| 574 | } |
| 575 | } |
| 576 | |
| 577 | impl From<char> for Literal { |
| 578 | fn from(t: char) -> Literal { |
Alex Crichton | 2d0cf0b | 2017-05-26 14:00:16 -0700 | [diff] [blame] | 579 | Literal(format!("'{}'", t.escape_default().collect::<String>())) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 580 | } |
| 581 | } |
| 582 | |
David Tolnay | 8e976c6 | 2017-06-01 12:12:29 -0700 | [diff] [blame] | 583 | named!(token_stream -> ::TokenStream, map!( |
| 584 | many0!(token_tree), |
| 585 | |trees| ::TokenStream(TokenStream { inner: trees }) |
| 586 | )); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 587 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 588 | #[cfg(not(procmacro2_semver_exempt))] |
David Tolnay | ddfca05 | 2017-12-31 10:41:24 -0500 | [diff] [blame] | 589 | fn token_tree(input: Cursor) -> PResult<TokenTree> { |
| 590 | let (input, kind) = token_kind(input)?; |
| 591 | Ok((input, TokenTree { |
| 592 | span: ::Span(Span {}), |
| 593 | kind: kind, |
| 594 | })) |
| 595 | } |
| 596 | |
David Tolnay | 1ebe397 | 2018-01-02 20:14:20 -0800 | [diff] [blame] | 597 | #[cfg(procmacro2_semver_exempt)] |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 598 | fn token_tree(input: Cursor) -> PResult<TokenTree> { |
| 599 | let input = skip_whitespace(input); |
| 600 | let lo = input.off; |
| 601 | let (input, kind) = token_kind(input)?; |
| 602 | let hi = input.off; |
| 603 | Ok((input, TokenTree { |
| 604 | span: ::Span(Span { |
| 605 | lo: lo, |
| 606 | hi: hi, |
| 607 | }), |
| 608 | kind: kind, |
| 609 | })) |
| 610 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 611 | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 612 | named!(token_kind -> TokenNode, alt!( |
| 613 | map!(delimited, |(d, s)| TokenNode::Group(d, s)) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 614 | | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 615 | map!(literal, TokenNode::Literal) // must be before symbol |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 616 | | |
Alex Crichton | 52725f7 | 2017-08-28 12:20:58 -0700 | [diff] [blame] | 617 | symbol |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 618 | | |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 619 | map!(op, |(op, kind)| TokenNode::Op(op, kind)) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 620 | )); |
| 621 | |
David Tolnay | 8e976c6 | 2017-06-01 12:12:29 -0700 | [diff] [blame] | 622 | named!(delimited -> (Delimiter, ::TokenStream), alt!( |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 623 | delimited!( |
| 624 | punct!("("), |
| 625 | token_stream, |
| 626 | punct!(")") |
| 627 | ) => { |ts| (Delimiter::Parenthesis, ts) } |
| 628 | | |
| 629 | delimited!( |
| 630 | punct!("["), |
| 631 | token_stream, |
| 632 | punct!("]") |
| 633 | ) => { |ts| (Delimiter::Bracket, ts) } |
| 634 | | |
| 635 | delimited!( |
| 636 | punct!("{"), |
| 637 | token_stream, |
| 638 | punct!("}") |
| 639 | ) => { |ts| (Delimiter::Brace, ts) } |
| 640 | )); |
| 641 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 642 | fn symbol(mut input: Cursor) -> PResult<TokenNode> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 643 | input = skip_whitespace(input); |
| 644 | |
| 645 | let mut chars = input.char_indices(); |
David Tolnay | a202d50 | 2017-06-01 12:26:55 -0700 | [diff] [blame] | 646 | |
| 647 | let lifetime = input.starts_with("'"); |
| 648 | if lifetime { |
| 649 | chars.next(); |
| 650 | } |
| 651 | |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 652 | match chars.next() { |
| 653 | Some((_, ch)) if UnicodeXID::is_xid_start(ch) || ch == '_' => {} |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 654 | _ => return Err(LexError), |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 655 | } |
| 656 | |
David Tolnay | 214c94c | 2017-06-01 12:42:56 -0700 | [diff] [blame] | 657 | let mut end = input.len(); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 658 | for (i, ch) in chars { |
| 659 | if !UnicodeXID::is_xid_continue(ch) { |
David Tolnay | 214c94c | 2017-06-01 12:42:56 -0700 | [diff] [blame] | 660 | end = i; |
| 661 | break; |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 662 | } |
| 663 | } |
| 664 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 665 | if lifetime && &input.rest[..end] != "'static" && KEYWORDS.contains(&&input.rest[1..end]) { |
David Tolnay | 214c94c | 2017-06-01 12:42:56 -0700 | [diff] [blame] | 666 | Err(LexError) |
| 667 | } else { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 668 | let a = &input.rest[..end]; |
Alex Crichton | 52725f7 | 2017-08-28 12:20:58 -0700 | [diff] [blame] | 669 | if a == "_" { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 670 | Ok((input.advance(end), TokenNode::Op('_', Spacing::Alone))) |
Alex Crichton | 52725f7 | 2017-08-28 12:20:58 -0700 | [diff] [blame] | 671 | } else { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 672 | Ok((input.advance(end), TokenNode::Term(::Term::intern(a)))) |
Alex Crichton | 52725f7 | 2017-08-28 12:20:58 -0700 | [diff] [blame] | 673 | } |
David Tolnay | 214c94c | 2017-06-01 12:42:56 -0700 | [diff] [blame] | 674 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 675 | } |
| 676 | |
David Tolnay | 214c94c | 2017-06-01 12:42:56 -0700 | [diff] [blame] | 677 | // From https://github.com/rust-lang/rust/blob/master/src/libsyntax_pos/symbol.rs |
| 678 | static KEYWORDS: &'static [&'static str] = &[ |
| 679 | "abstract", "alignof", "as", "become", "box", "break", "const", "continue", |
| 680 | "crate", "do", "else", "enum", "extern", "false", "final", "fn", "for", |
| 681 | "if", "impl", "in", "let", "loop", "macro", "match", "mod", "move", "mut", |
| 682 | "offsetof", "override", "priv", "proc", "pub", "pure", "ref", "return", |
| 683 | "self", "Self", "sizeof", "static", "struct", "super", "trait", "true", |
| 684 | "type", "typeof", "unsafe", "unsized", "use", "virtual", "where", "while", |
| 685 | "yield", |
| 686 | ]; |
| 687 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 688 | fn literal(input: Cursor) -> PResult<::Literal> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 689 | let input_no_ws = skip_whitespace(input); |
| 690 | |
| 691 | match literal_nocapture(input_no_ws) { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 692 | Ok((a, ())) => { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 693 | let start = input.len() - input_no_ws.len(); |
| 694 | let len = input_no_ws.len() - a.len(); |
| 695 | let end = start + len; |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 696 | Ok((a, ::Literal(Literal(input.rest[start..end].to_string())))) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 697 | } |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 698 | Err(LexError) => Err(LexError), |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 699 | } |
| 700 | } |
| 701 | |
| 702 | named!(literal_nocapture -> (), alt!( |
| 703 | string |
| 704 | | |
| 705 | byte_string |
| 706 | | |
| 707 | byte |
| 708 | | |
| 709 | character |
| 710 | | |
| 711 | float |
| 712 | | |
| 713 | int |
| 714 | | |
| 715 | boolean |
| 716 | | |
| 717 | doc_comment |
| 718 | )); |
| 719 | |
| 720 | named!(string -> (), alt!( |
| 721 | quoted_string |
| 722 | | |
| 723 | preceded!( |
| 724 | punct!("r"), |
| 725 | raw_string |
| 726 | ) => { |_| () } |
| 727 | )); |
| 728 | |
| 729 | named!(quoted_string -> (), delimited!( |
| 730 | punct!("\""), |
| 731 | cooked_string, |
| 732 | tag!("\"") |
| 733 | )); |
| 734 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 735 | fn cooked_string(input: Cursor) -> PResult<()> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 736 | let mut chars = input.char_indices().peekable(); |
| 737 | while let Some((byte_offset, ch)) = chars.next() { |
| 738 | match ch { |
| 739 | '"' => { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 740 | return Ok((input.advance(byte_offset), ())); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 741 | } |
| 742 | '\r' => { |
| 743 | if let Some((_, '\n')) = chars.next() { |
| 744 | // ... |
| 745 | } else { |
| 746 | break; |
| 747 | } |
| 748 | } |
| 749 | '\\' => { |
| 750 | match chars.next() { |
| 751 | Some((_, 'x')) => { |
| 752 | if !backslash_x_char(&mut chars) { |
| 753 | break |
| 754 | } |
| 755 | } |
| 756 | Some((_, 'n')) | |
| 757 | Some((_, 'r')) | |
| 758 | Some((_, 't')) | |
| 759 | Some((_, '\\')) | |
| 760 | Some((_, '\'')) | |
| 761 | Some((_, '"')) | |
| 762 | Some((_, '0')) => {} |
| 763 | Some((_, 'u')) => { |
| 764 | if !backslash_u(&mut chars) { |
| 765 | break |
| 766 | } |
| 767 | } |
| 768 | Some((_, '\n')) | Some((_, '\r')) => { |
| 769 | while let Some(&(_, ch)) = chars.peek() { |
| 770 | if ch.is_whitespace() { |
| 771 | chars.next(); |
| 772 | } else { |
| 773 | break; |
| 774 | } |
| 775 | } |
| 776 | } |
| 777 | _ => break, |
| 778 | } |
| 779 | } |
| 780 | _ch => {} |
| 781 | } |
| 782 | } |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 783 | Err(LexError) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 784 | } |
| 785 | |
| 786 | named!(byte_string -> (), alt!( |
| 787 | delimited!( |
| 788 | punct!("b\""), |
| 789 | cooked_byte_string, |
| 790 | tag!("\"") |
| 791 | ) => { |_| () } |
| 792 | | |
| 793 | preceded!( |
| 794 | punct!("br"), |
| 795 | raw_string |
| 796 | ) => { |_| () } |
| 797 | )); |
| 798 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 799 | fn cooked_byte_string(mut input: Cursor) -> PResult<()> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 800 | let mut bytes = input.bytes().enumerate(); |
| 801 | 'outer: while let Some((offset, b)) = bytes.next() { |
| 802 | match b { |
| 803 | b'"' => { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 804 | return Ok((input.advance(offset), ())); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 805 | } |
| 806 | b'\r' => { |
| 807 | if let Some((_, b'\n')) = bytes.next() { |
| 808 | // ... |
| 809 | } else { |
| 810 | break; |
| 811 | } |
| 812 | } |
| 813 | b'\\' => { |
| 814 | match bytes.next() { |
| 815 | Some((_, b'x')) => { |
| 816 | if !backslash_x_byte(&mut bytes) { |
| 817 | break |
| 818 | } |
| 819 | } |
| 820 | Some((_, b'n')) | |
| 821 | Some((_, b'r')) | |
| 822 | Some((_, b't')) | |
| 823 | Some((_, b'\\')) | |
| 824 | Some((_, b'0')) | |
| 825 | Some((_, b'\'')) | |
| 826 | Some((_, b'"')) => {} |
| 827 | Some((newline, b'\n')) | |
| 828 | Some((newline, b'\r')) => { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 829 | let rest = input.advance(newline + 1); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 830 | for (offset, ch) in rest.char_indices() { |
| 831 | if !ch.is_whitespace() { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 832 | input = rest.advance(offset); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 833 | bytes = input.bytes().enumerate(); |
| 834 | continue 'outer; |
| 835 | } |
| 836 | } |
| 837 | break; |
| 838 | } |
| 839 | _ => break, |
| 840 | } |
| 841 | } |
| 842 | b if b < 0x80 => {} |
| 843 | _ => break, |
| 844 | } |
| 845 | } |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 846 | Err(LexError) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 847 | } |
| 848 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 849 | fn raw_string(input: Cursor) -> PResult<()> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 850 | let mut chars = input.char_indices(); |
| 851 | let mut n = 0; |
| 852 | while let Some((byte_offset, ch)) = chars.next() { |
| 853 | match ch { |
| 854 | '"' => { |
| 855 | n = byte_offset; |
| 856 | break; |
| 857 | } |
| 858 | '#' => {} |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 859 | _ => return Err(LexError), |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 860 | } |
| 861 | } |
| 862 | for (byte_offset, ch) in chars { |
| 863 | match ch { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 864 | '"' if input.advance(byte_offset + 1).starts_with(&input.rest[..n]) => { |
| 865 | let rest = input.advance(byte_offset + 1 + n); |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 866 | return Ok((rest, ())) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 867 | } |
| 868 | '\r' => {} |
| 869 | _ => {} |
| 870 | } |
| 871 | } |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 872 | Err(LexError) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 873 | } |
| 874 | |
| 875 | named!(byte -> (), do_parse!( |
| 876 | punct!("b") >> |
| 877 | tag!("'") >> |
| 878 | cooked_byte >> |
| 879 | tag!("'") >> |
| 880 | (()) |
| 881 | )); |
| 882 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 883 | fn cooked_byte(input: Cursor) -> PResult<()> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 884 | let mut bytes = input.bytes().enumerate(); |
| 885 | let ok = match bytes.next().map(|(_, b)| b) { |
| 886 | Some(b'\\') => { |
| 887 | match bytes.next().map(|(_, b)| b) { |
| 888 | Some(b'x') => backslash_x_byte(&mut bytes), |
| 889 | Some(b'n') | |
| 890 | Some(b'r') | |
| 891 | Some(b't') | |
| 892 | Some(b'\\') | |
| 893 | Some(b'0') | |
| 894 | Some(b'\'') | |
| 895 | Some(b'"') => true, |
| 896 | _ => false, |
| 897 | } |
| 898 | } |
| 899 | b => b.is_some(), |
| 900 | }; |
| 901 | if ok { |
| 902 | match bytes.next() { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 903 | Some((offset, _)) => Ok((input.advance(offset), ())), |
| 904 | None => Ok((input.advance(input.len()), ())), |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 905 | } |
| 906 | } else { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 907 | Err(LexError) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 908 | } |
| 909 | } |
| 910 | |
| 911 | named!(character -> (), do_parse!( |
| 912 | punct!("'") >> |
| 913 | cooked_char >> |
| 914 | tag!("'") >> |
| 915 | (()) |
| 916 | )); |
| 917 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 918 | fn cooked_char(input: Cursor) -> PResult<()> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 919 | let mut chars = input.char_indices(); |
| 920 | let ok = match chars.next().map(|(_, ch)| ch) { |
| 921 | Some('\\') => { |
| 922 | match chars.next().map(|(_, ch)| ch) { |
| 923 | Some('x') => backslash_x_char(&mut chars), |
| 924 | Some('u') => backslash_u(&mut chars), |
| 925 | Some('n') | |
| 926 | Some('r') | |
| 927 | Some('t') | |
| 928 | Some('\\') | |
| 929 | Some('0') | |
| 930 | Some('\'') | |
| 931 | Some('"') => true, |
| 932 | _ => false, |
| 933 | } |
| 934 | } |
| 935 | ch => ch.is_some(), |
| 936 | }; |
| 937 | if ok { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 938 | match chars.next() { |
| 939 | Some((idx, _)) => Ok((input.advance(idx), ())), |
| 940 | None => Ok((input.advance(input.len()), ())), |
| 941 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 942 | } else { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 943 | Err(LexError) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 944 | } |
| 945 | } |
| 946 | |
| 947 | macro_rules! next_ch { |
| 948 | ($chars:ident @ $pat:pat $(| $rest:pat)*) => { |
| 949 | match $chars.next() { |
| 950 | Some((_, ch)) => match ch { |
| 951 | $pat $(| $rest)* => ch, |
| 952 | _ => return false, |
| 953 | }, |
| 954 | None => return false |
| 955 | } |
| 956 | }; |
| 957 | } |
| 958 | |
| 959 | fn backslash_x_char<I>(chars: &mut I) -> bool |
| 960 | where I: Iterator<Item = (usize, char)> |
| 961 | { |
| 962 | next_ch!(chars @ '0'...'7'); |
| 963 | next_ch!(chars @ '0'...'9' | 'a'...'f' | 'A'...'F'); |
| 964 | true |
| 965 | } |
| 966 | |
| 967 | fn backslash_x_byte<I>(chars: &mut I) -> bool |
| 968 | where I: Iterator<Item = (usize, u8)> |
| 969 | { |
| 970 | next_ch!(chars @ b'0'...b'9' | b'a'...b'f' | b'A'...b'F'); |
| 971 | next_ch!(chars @ b'0'...b'9' | b'a'...b'f' | b'A'...b'F'); |
| 972 | true |
| 973 | } |
| 974 | |
| 975 | fn backslash_u<I>(chars: &mut I) -> bool |
| 976 | where I: Iterator<Item = (usize, char)> |
| 977 | { |
| 978 | next_ch!(chars @ '{'); |
| 979 | next_ch!(chars @ '0'...'9' | 'a'...'f' | 'A'...'F'); |
David Tolnay | 8d10934 | 2017-12-25 18:24:45 -0500 | [diff] [blame] | 980 | loop { |
| 981 | let c = next_ch!(chars @ '0'...'9' | 'a'...'f' | 'A'...'F' | '_' | '}'); |
| 982 | if c == '}' { |
| 983 | return true; |
| 984 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 985 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 986 | } |
| 987 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 988 | fn float(input: Cursor) -> PResult<()> { |
David Tolnay | 744a6b8 | 2017-06-01 11:34:29 -0700 | [diff] [blame] | 989 | let (rest, ()) = float_digits(input)?; |
| 990 | for suffix in &["f32", "f64"] { |
| 991 | if rest.starts_with(suffix) { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 992 | return word_break(rest.advance(suffix.len())); |
David Tolnay | 744a6b8 | 2017-06-01 11:34:29 -0700 | [diff] [blame] | 993 | } |
| 994 | } |
| 995 | word_break(rest) |
| 996 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 997 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 998 | fn float_digits(input: Cursor) -> PResult<()> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 999 | let mut chars = input.chars().peekable(); |
| 1000 | match chars.next() { |
| 1001 | Some(ch) if ch >= '0' && ch <= '9' => {} |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1002 | _ => return Err(LexError), |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1003 | } |
| 1004 | |
| 1005 | let mut len = 1; |
| 1006 | let mut has_dot = false; |
| 1007 | let mut has_exp = false; |
| 1008 | while let Some(&ch) = chars.peek() { |
| 1009 | match ch { |
| 1010 | '0'...'9' | '_' => { |
| 1011 | chars.next(); |
| 1012 | len += 1; |
| 1013 | } |
| 1014 | '.' => { |
| 1015 | if has_dot { |
| 1016 | break; |
| 1017 | } |
| 1018 | chars.next(); |
| 1019 | if chars.peek() |
| 1020 | .map(|&ch| ch == '.' || UnicodeXID::is_xid_start(ch)) |
| 1021 | .unwrap_or(false) { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1022 | return Err(LexError); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1023 | } |
| 1024 | len += 1; |
| 1025 | has_dot = true; |
| 1026 | } |
| 1027 | 'e' | 'E' => { |
| 1028 | chars.next(); |
| 1029 | len += 1; |
| 1030 | has_exp = true; |
| 1031 | break; |
| 1032 | } |
| 1033 | _ => break, |
| 1034 | } |
| 1035 | } |
| 1036 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1037 | let rest = input.advance(len); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1038 | if !(has_dot || has_exp || rest.starts_with("f32") || rest.starts_with("f64")) { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1039 | return Err(LexError); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1040 | } |
| 1041 | |
| 1042 | if has_exp { |
| 1043 | let mut has_exp_value = false; |
| 1044 | while let Some(&ch) = chars.peek() { |
| 1045 | match ch { |
| 1046 | '+' | '-' => { |
| 1047 | if has_exp_value { |
| 1048 | break; |
| 1049 | } |
| 1050 | chars.next(); |
| 1051 | len += 1; |
| 1052 | } |
| 1053 | '0'...'9' => { |
| 1054 | chars.next(); |
| 1055 | len += 1; |
| 1056 | has_exp_value = true; |
| 1057 | } |
| 1058 | '_' => { |
| 1059 | chars.next(); |
| 1060 | len += 1; |
| 1061 | } |
| 1062 | _ => break, |
| 1063 | } |
| 1064 | } |
| 1065 | if !has_exp_value { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1066 | return Err(LexError); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1067 | } |
| 1068 | } |
| 1069 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1070 | Ok((input.advance(len), ())) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1071 | } |
| 1072 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1073 | fn int(input: Cursor) -> PResult<()> { |
David Tolnay | 744a6b8 | 2017-06-01 11:34:29 -0700 | [diff] [blame] | 1074 | let (rest, ()) = digits(input)?; |
| 1075 | for suffix in &[ |
| 1076 | "isize", |
| 1077 | "i8", |
| 1078 | "i16", |
| 1079 | "i32", |
| 1080 | "i64", |
| 1081 | "i128", |
| 1082 | "usize", |
| 1083 | "u8", |
| 1084 | "u16", |
| 1085 | "u32", |
| 1086 | "u64", |
| 1087 | "u128", |
| 1088 | ] { |
| 1089 | if rest.starts_with(suffix) { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1090 | return word_break(rest.advance(suffix.len())); |
David Tolnay | 744a6b8 | 2017-06-01 11:34:29 -0700 | [diff] [blame] | 1091 | } |
| 1092 | } |
| 1093 | word_break(rest) |
| 1094 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1095 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1096 | fn digits(mut input: Cursor) -> PResult<()> { |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1097 | let base = if input.starts_with("0x") { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1098 | input = input.advance(2); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1099 | 16 |
| 1100 | } else if input.starts_with("0o") { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1101 | input = input.advance(2); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1102 | 8 |
| 1103 | } else if input.starts_with("0b") { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1104 | input = input.advance(2); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1105 | 2 |
| 1106 | } else { |
| 1107 | 10 |
| 1108 | }; |
| 1109 | |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1110 | let mut len = 0; |
| 1111 | let mut empty = true; |
| 1112 | for b in input.bytes() { |
| 1113 | let digit = match b { |
| 1114 | b'0'...b'9' => (b - b'0') as u64, |
| 1115 | b'a'...b'f' => 10 + (b - b'a') as u64, |
| 1116 | b'A'...b'F' => 10 + (b - b'A') as u64, |
| 1117 | b'_' => { |
| 1118 | if empty && base == 10 { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1119 | return Err(LexError); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1120 | } |
| 1121 | len += 1; |
| 1122 | continue; |
| 1123 | } |
| 1124 | _ => break, |
| 1125 | }; |
| 1126 | if digit >= base { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1127 | return Err(LexError); |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1128 | } |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1129 | len += 1; |
| 1130 | empty = false; |
| 1131 | } |
| 1132 | if empty { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1133 | Err(LexError) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1134 | } else { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1135 | Ok((input.advance(len), ())) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1136 | } |
| 1137 | } |
| 1138 | |
| 1139 | named!(boolean -> (), alt!( |
| 1140 | keyword!("true") => { |_| () } |
| 1141 | | |
| 1142 | keyword!("false") => { |_| () } |
| 1143 | )); |
| 1144 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1145 | fn op(input: Cursor) -> PResult<(char, Spacing)> { |
David Tolnay | ea75c5f | 2017-05-31 23:40:33 -0700 | [diff] [blame] | 1146 | let input = skip_whitespace(input); |
| 1147 | match op_char(input) { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1148 | Ok((rest, ch)) => { |
David Tolnay | ea75c5f | 2017-05-31 23:40:33 -0700 | [diff] [blame] | 1149 | let kind = match op_char(rest) { |
Alex Crichton | 1a7f762 | 2017-07-05 17:47:15 -0700 | [diff] [blame] | 1150 | Ok(_) => Spacing::Joint, |
| 1151 | Err(LexError) => Spacing::Alone, |
David Tolnay | ea75c5f | 2017-05-31 23:40:33 -0700 | [diff] [blame] | 1152 | }; |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1153 | Ok((rest, (ch, kind))) |
David Tolnay | ea75c5f | 2017-05-31 23:40:33 -0700 | [diff] [blame] | 1154 | } |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1155 | Err(LexError) => Err(LexError), |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1156 | } |
| 1157 | } |
| 1158 | |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1159 | fn op_char(input: Cursor) -> PResult<char> { |
David Tolnay | ea75c5f | 2017-05-31 23:40:33 -0700 | [diff] [blame] | 1160 | let mut chars = input.chars(); |
| 1161 | let first = match chars.next() { |
| 1162 | Some(ch) => ch, |
| 1163 | None => { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1164 | return Err(LexError); |
David Tolnay | ea75c5f | 2017-05-31 23:40:33 -0700 | [diff] [blame] | 1165 | } |
| 1166 | }; |
| 1167 | let recognized = "~!@#$%^&*-=+|;:,<.>/?"; |
| 1168 | if recognized.contains(first) { |
Nika Layzell | f8d5f21 | 2017-12-11 14:07:02 -0500 | [diff] [blame] | 1169 | Ok((input.advance(first.len_utf8()), first)) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1170 | } else { |
David Tolnay | 1218e12 | 2017-06-01 11:13:45 -0700 | [diff] [blame] | 1171 | Err(LexError) |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1172 | } |
| 1173 | } |
| 1174 | |
Alex Crichton | 44bffbc | 2017-05-19 17:51:59 -0700 | [diff] [blame] | 1175 | named!(doc_comment -> (), alt!( |
| 1176 | do_parse!( |
| 1177 | punct!("//!") >> |
| 1178 | take_until!("\n") >> |
| 1179 | (()) |
| 1180 | ) |
| 1181 | | |
| 1182 | do_parse!( |
| 1183 | option!(whitespace) >> |
| 1184 | peek!(tag!("/*!")) >> |
| 1185 | block_comment >> |
| 1186 | (()) |
| 1187 | ) |
| 1188 | | |
| 1189 | do_parse!( |
| 1190 | punct!("///") >> |
| 1191 | not!(tag!("/")) >> |
| 1192 | take_until!("\n") >> |
| 1193 | (()) |
| 1194 | ) |
| 1195 | | |
| 1196 | do_parse!( |
| 1197 | option!(whitespace) >> |
| 1198 | peek!(tuple!(tag!("/**"), not!(tag!("*")))) >> |
| 1199 | block_comment >> |
| 1200 | (()) |
| 1201 | ) |
| 1202 | )); |