src/string.rs

   1 // Copyright 2015 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 // Format string literals.
  12
  13 use regex::Regex;
  14 use unicode_segmentation::UnicodeSegmentation;
  15
  16 use config::Config;
  17 use shape::Shape;
  18 use utils::wrap_str;
  19
  20 const MIN_STRING: usize = 10;
  21
  22 /// Describes the layout of a piece of text.
  23 pub struct StringFormat<'a> {
  24     /// The opening sequence of characters for the piece of text
  25     pub opener: &'a str,
  26     /// The closing sequence of characters for the piece of text
  27     pub closer: &'a str,
  28     /// The opening sequence of characters for a line
  29     pub line_start: &'a str,
  30     /// The closing sequence of characters for a line
  31     pub line_end: &'a str,
  32     /// The allocated box to fit the text into
  33     pub shape: Shape,
  34     /// Trim trailing whitespaces
  35     pub trim_end: bool,
  36     pub config: &'a Config,
  37 }
  38
  39 impl<'a> StringFormat<'a> {
  40     pub fn new(shape: Shape, config: &'a Config) -> StringFormat<'a> {
  41         StringFormat {
  42             opener: "\"",
  43             closer: "\"",
  44             line_start: " ",
  45             line_end: "\\",
  46             shape,
  47             trim_end: false,
  48             config,
  49         }
  50     }
  51
  52     /// Returns the maximum number of graphemes that is possible on a line while taking the
  53     /// indentation into account.
  54     ///
  55     /// If we cannot put at least a single character per line, the rewrite won't succeed.
  56     fn max_chars_with_indent(&self) -> Option<usize> {
  57         Some(
  58             self.shape
  59                 .width
  60                 .checked_sub(self.opener.len() + self.line_end.len() + 1)?
  61                 + 1,
  62         )
  63     }
  64
  65     /// Like max_chars_with_indent but the indentation is not substracted.
  66     /// This allows to fit more graphemes from the string on a line when
  67     /// SnippetState::EndWithLineFeed.
  68     fn max_chars_without_indent(&self) -> Option<usize> {
  69         Some(self.config.max_width().checked_sub(self.line_end.len())?)
  70     }
  71 }
  72
  73 pub fn rewrite_string<'a>(orig: &str, fmt: &StringFormat<'a>) -> Option<String> {
  74     let max_chars_with_indent = fmt.max_chars_with_indent()?;
  75     let max_chars_without_indent = fmt.max_chars_without_indent()?;
  76     let indent_with_newline = fmt.shape.indent.to_string_with_newline(fmt.config);
  77     let indent_without_newline = fmt.shape.indent.to_string(fmt.config);
  78
  79     // Strip line breaks.
  80     // With this regex applied, all remaining whitespaces are significant
  81     let strip_line_breaks_re = Regex::new(r"([^\\](\\\\)*)\\[\n\r][[:space:]]*").unwrap();
  82     let stripped_str = strip_line_breaks_re.replace_all(orig, "$1");
  83
  84     let graphemes = UnicodeSegmentation::graphemes(&*stripped_str, false).collect::<Vec<&str>>();
  85
  86     // `cur_start` is the position in `orig` of the start of the current line.
  87     let mut cur_start = 0;
  88     let mut result = String::with_capacity(
  89         stripped_str
  90             .len()
  91             .checked_next_power_of_two()
  92             .unwrap_or(usize::max_value()),
  93     );
  94     result.push_str(fmt.opener);
  95
  96     // Snip a line at a time from `stripped_str` until it is used up. Push the snippet
  97     // onto result.
  98     let mut cur_max_chars = max_chars_with_indent;
  99     let is_bareline_ok = fmt.line_start.is_empty() || is_whitespace(fmt.line_start);
 100     loop {
 101         // All the input starting at cur_start fits on the current line
 102         if graphemes.len() - cur_start <= cur_max_chars {
 103             for (i, grapheme) in graphemes[cur_start..].iter().enumerate() {
 104                 if is_line_feed(grapheme) {
 105                     // take care of blank lines
 106                     result = trim_right_but_line_feed(fmt.trim_end, result);
 107                     result.push_str("\n");
 108                     if !is_bareline_ok && cur_start + i + 1 < graphemes.len() {
 109                         result.push_str(&indent_without_newline);
 110                         result.push_str(fmt.line_start);
 111                     }
 112                 } else {
 113                     result.push_str(grapheme);
 114                 }
 115             }
 116             result = trim_right_but_line_feed(fmt.trim_end, result);
 117             break;
 118         }
 119
 120         // The input starting at cur_start needs to be broken
 121         match break_string(
 122             cur_max_chars,
 123             fmt.trim_end,
 124             fmt.line_end,
 125             &graphemes[cur_start..],
 126         ) {
 127             SnippetState::LineEnd(line, len) => {
 128                 result.push_str(&line);
 129                 result.push_str(fmt.line_end);
 130                 result.push_str(&indent_with_newline);
 131                 result.push_str(fmt.line_start);
 132                 cur_max_chars = max_chars_with_indent;
 133                 cur_start += len;
 134             }
 135             SnippetState::EndWithLineFeed(line, len) => {
 136                 if line == "\n" && fmt.trim_end {
 137                     result = result.trim_right().to_string();
 138                 }
 139                 result.push_str(&line);
 140                 if is_bareline_ok {
 141                     // the next line can benefit from the full width
 142                     cur_max_chars = max_chars_without_indent;
 143                 } else {
 144                     result.push_str(&indent_without_newline);
 145                     result.push_str(fmt.line_start);
 146                     cur_max_chars = max_chars_with_indent;
 147                 }
 148                 cur_start += len;
 149             }
 150             SnippetState::EndOfInput(line) => {
 151                 result.push_str(&line);
 152                 break;
 153             }
 154         }
 155     }
 156
 157     result.push_str(fmt.closer);
 158     wrap_str(result, fmt.config.max_width(), fmt.shape)
 159 }
 160
 161 /// Returns the index to the end of the url if the given string includes an
 162 /// URL or alike. Otherwise, returns None;
 163 fn detect_url(s: &[&str], index: usize) -> Option<usize> {
 164     let start = match s[..=index].iter().rposition(|g| is_whitespace(g)) {
 165         Some(pos) => pos + 1,
 166         None => 0,
 167     };
 168     if s.len() < start + 8 {
 169         return None;
 170     }
 171     let prefix = s[start..start + 8].join("");
 172     if prefix.starts_with("https://")
 173         || prefix.starts_with("http://")
 174         || prefix.starts_with("ftp://")
 175         || prefix.starts_with("file://")
 176     {
 177         match s[index..].iter().position(|g| is_whitespace(g)) {
 178             Some(pos) => Some(index + pos - 1),
 179             None => Some(s.len() - 1),
 180         }
 181     } else {
 182         None
 183     }
 184 }
 185
 186 /// Trims whitespaces to the right except for the line feed character.
 187 fn trim_right_but_line_feed(trim_end: bool, result: String) -> String {
 188     let whitespace_except_line_feed = |c: char| c.is_whitespace() && c != '\n';
 189     if trim_end && result.ends_with(whitespace_except_line_feed) {
 190         result
 191             .trim_right_matches(whitespace_except_line_feed)
 192             .to_string()
 193     } else {
 194         result
 195     }
 196 }
 197
 198 /// Result of breaking a string so it fits in a line and the state it ended in.
 199 /// The state informs about what to do with the snippet and how to continue the breaking process.
 200 #[derive(Debug, PartialEq)]
 201 enum SnippetState {
 202     /// The input could not be broken and so rewriting the string is finished.
 203     EndOfInput(String),
 204     /// The input could be broken and the returned snippet should be ended with a
 205     /// `[StringFormat::line_end]`. The next snippet needs to be indented.
 206     ///
 207     /// The returned string is the line to print out and the number is the length that got read in
 208     /// the text being rewritten. That length may be greater than the returned string if trailing
 209     /// whitespaces got trimmed.
 210     LineEnd(String, usize),
 211     /// The input could be broken but a newline is present that cannot be trimmed. The next snippet
 212     /// to be rewritten *could* use more width than what is specified by the given shape. For
 213     /// example with a multiline string, the next snippet does not need to be indented, allowing
 214     /// more characters to be fit within a line.
 215     ///
 216     /// The returned string is the line to print out and the number is the length that got read in
 217     /// the text being rewritten.
 218     EndWithLineFeed(String, usize),
 219 }
 220
 221 fn not_whitespace_except_line_feed(g: &str) -> bool {
 222     is_line_feed(g) || !is_whitespace(g)
 223 }
 224
 225 /// Break the input string at a boundary character around the offset `max_chars`. A boundary
 226 /// character is either a punctuation or a whitespace.
 227 fn break_string(max_chars: usize, trim_end: bool, line_end: &str, input: &[&str]) -> SnippetState {
 228     let break_at = |index /* grapheme at index is included */| {
 229         // Take in any whitespaces to the left/right of `input[index]` while
 230         // preserving line feeds
 231         let index_minus_ws = input[0..=index]
 232             .iter()
 233             .rposition(|grapheme| not_whitespace_except_line_feed(grapheme))
 234             .unwrap_or(index);
 235         // Take into account newlines occuring in input[0..=index], i.e., the possible next new
 236         // line. If there is one, then text after it could be rewritten in a way that the available
 237         // space is fully used.
 238         for (i, grapheme) in input[0..=index].iter().enumerate() {
 239             if is_line_feed(grapheme) {
 240                 if i <= index_minus_ws {
 241                     let mut line = input[0..i].join("");
 242                     if trim_end {
 243                         line = line.trim_right().to_string();
 244                     }
 245                     return SnippetState::EndWithLineFeed(format!("{}\n", line), i + 1);
 246                 }
 247                 break;
 248             }
 249         }
 250
 251         let mut index_plus_ws = index;
 252         for (i, grapheme) in input[index + 1..].iter().enumerate() {
 253             if !trim_end && is_line_feed(grapheme) {
 254                 return SnippetState::EndWithLineFeed(
 255                     input[0..=index + 1 + i].join("").to_string(),
 256                     index + 2 + i,
 257                 );
 258             } else if not_whitespace_except_line_feed(grapheme) {
 259                 index_plus_ws = index + i;
 260                 break;
 261             }
 262         }
 263
 264         if trim_end {
 265             SnippetState::LineEnd(
 266                 input[0..=index_minus_ws].join("").to_string(),
 267                 index_plus_ws + 1,
 268             )
 269         } else {
 270             SnippetState::LineEnd(
 271                 input[0..=index_plus_ws].join("").to_string(),
 272                 index_plus_ws + 1,
 273             )
 274         }
 275     };
 276
 277     // Find the position in input for breaking the string
 278     if line_end.is_empty()
 279         && trim_end
 280         && !is_whitespace(input[max_chars - 1])
 281         && is_whitespace(input[max_chars])
 282     {
 283         // At a breaking point already
 284         // The line won't invalidate the rewriting because:
 285         // - no extra space needed for the line_end character
 286         // - extra whitespaces to the right can be trimmed
 287         return break_at(max_chars - 1);
 288     }
 289     if let Some(url_index_end) = detect_url(input, max_chars) {
 290         let index_plus_ws = url_index_end
 291             + input[url_index_end..]
 292                 .iter()
 293                 .skip(1)
 294                 .position(|grapheme| not_whitespace_except_line_feed(grapheme))
 295                 .unwrap_or(0);
 296         return if trim_end {
 297             SnippetState::LineEnd(
 298                 input[..=url_index_end].join("").to_string(),
 299                 index_plus_ws + 1,
 300             )
 301         } else {
 302             return SnippetState::LineEnd(
 303                 input[..=index_plus_ws].join("").to_string(),
 304                 index_plus_ws + 1,
 305             );
 306         };
 307     }
 308     match input[0..max_chars]
 309         .iter()
 310         .rposition(|grapheme| is_whitespace(grapheme))
 311     {
 312         // Found a whitespace and what is on its left side is big enough.
 313         Some(index) if index >= MIN_STRING => break_at(index),
 314         // No whitespace found, try looking for a punctuation instead
 315         _ => match input[0..max_chars]
 316             .iter()
 317             .rposition(|grapheme| is_punctuation(grapheme))
 318         {
 319             // Found a punctuation and what is on its left side is big enough.
 320             Some(index) if index >= MIN_STRING => break_at(index),
 321             // Either no boundary character was found to the left of `input[max_chars]`, or the line
 322             // got too small. We try searching for a boundary character to the right.
 323             _ => match input[max_chars..]
 324                 .iter()
 325                 .position(|grapheme| is_whitespace(grapheme) || is_punctuation(grapheme))
 326             {
 327                 // A boundary was found after the line limit
 328                 Some(index) => break_at(max_chars + index),
 329                 // No boundary to the right, the input cannot be broken
 330                 None => SnippetState::EndOfInput(input.join("").to_string()),
 331             },
 332         },
 333     }
 334 }
 335
 336 fn is_line_feed(grapheme: &str) -> bool {
 337     grapheme.as_bytes()[0] == b'\n'
 338 }
 339
 340 fn is_whitespace(grapheme: &str) -> bool {
 341     grapheme.chars().all(|c| c.is_whitespace())
 342 }
 343
 344 fn is_punctuation(grapheme: &str) -> bool {
 345     match grapheme.as_bytes()[0] {
 346         b':' | b',' | b';' | b'.' => true,
 347         _ => false,
 348     }
 349 }
 350
 351 #[cfg(test)]
 352 mod test {
 353     use super::{break_string, detect_url, rewrite_string, SnippetState, StringFormat};
 354     use config::Config;
 355     use shape::{Indent, Shape};
 356     use unicode_segmentation::UnicodeSegmentation;
 357
 358     #[test]
 359     fn issue343() {
 360         let config = Default::default();
 361         let fmt = StringFormat::new(Shape::legacy(2, Indent::empty()), &config);
 362         rewrite_string("eq_", &fmt);
 363     }
 364
 365     #[test]
 366     fn should_break_on_whitespace() {
 367         let string = "Placerat felis. Mauris porta ante sagittis purus.";
 368         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 369         assert_eq!(
 370             break_string(20, false, "", &graphemes[..]),
 371             SnippetState::LineEnd("Placerat felis. ".to_string(), 16)
 372         );
 373         assert_eq!(
 374             break_string(20, true, "", &graphemes[..]),
 375             SnippetState::LineEnd("Placerat felis.".to_string(), 16)
 376         );
 377     }
 378
 379     #[test]
 380     fn should_break_on_punctuation() {
 381         let string = "Placerat_felis._Mauris_porta_ante_sagittis_purus.";
 382         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 383         assert_eq!(
 384             break_string(20, false, "", &graphemes[..]),
 385             SnippetState::LineEnd("Placerat_felis.".to_string(), 15)
 386         );
 387     }
 388
 389     #[test]
 390     fn should_break_forward() {
 391         let string = "Venenatis_tellus_vel_tellus. Aliquam aliquam dolor at justo.";
 392         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 393         assert_eq!(
 394             break_string(20, false, "", &graphemes[..]),
 395             SnippetState::LineEnd("Venenatis_tellus_vel_tellus. ".to_string(), 29)
 396         );
 397         assert_eq!(
 398             break_string(20, true, "", &graphemes[..]),
 399             SnippetState::LineEnd("Venenatis_tellus_vel_tellus.".to_string(), 29)
 400         );
 401     }
 402
 403     #[test]
 404     fn nothing_to_break() {
 405         let string = "Venenatis_tellus_vel_tellus";
 406         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 407         assert_eq!(
 408             break_string(20, false, "", &graphemes[..]),
 409             SnippetState::EndOfInput("Venenatis_tellus_vel_tellus".to_string())
 410         );
 411     }
 412
 413     #[test]
 414     fn significant_whitespaces() {
 415         let string = "Neque in sem.      \n      Pellentesque tellus augue.";
 416         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 417         assert_eq!(
 418             break_string(15, false, "", &graphemes[..]),
 419             SnippetState::EndWithLineFeed("Neque in sem.      \n".to_string(), 20)
 420         );
 421         assert_eq!(
 422             break_string(25, false, "", &graphemes[..]),
 423             SnippetState::EndWithLineFeed("Neque in sem.      \n".to_string(), 20)
 424         );
 425
 426         assert_eq!(
 427             break_string(15, true, "", &graphemes[..]),
 428             SnippetState::LineEnd("Neque in sem.".to_string(), 19)
 429         );
 430         assert_eq!(
 431             break_string(25, true, "", &graphemes[..]),
 432             SnippetState::EndWithLineFeed("Neque in sem.\n".to_string(), 20)
 433         );
 434     }
 435
 436     #[test]
 437     fn big_whitespace() {
 438         let string = "Neque in sem.            Pellentesque tellus augue.";
 439         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 440         assert_eq!(
 441             break_string(20, false, "", &graphemes[..]),
 442             SnippetState::LineEnd("Neque in sem.            ".to_string(), 25)
 443         );
 444         assert_eq!(
 445             break_string(20, true, "", &graphemes[..]),
 446             SnippetState::LineEnd("Neque in sem.".to_string(), 25)
 447         );
 448     }
 449
 450     #[test]
 451     fn newline_in_candidate_line() {
 452         let string = "Nulla\nconsequat erat at massa. Vivamus id mi.";
 453
 454         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 455         assert_eq!(
 456             break_string(25, false, "", &graphemes[..]),
 457             SnippetState::EndWithLineFeed("Nulla\n".to_string(), 6)
 458         );
 459         assert_eq!(
 460             break_string(25, true, "", &graphemes[..]),
 461             SnippetState::EndWithLineFeed("Nulla\n".to_string(), 6)
 462         );
 463
 464         let mut config: Config = Default::default();
 465         config.set().max_width(27);
 466         let fmt = StringFormat::new(Shape::legacy(25, Indent::empty()), &config);
 467         let rewritten_string = rewrite_string(string, &fmt);
 468         assert_eq!(
 469             rewritten_string,
 470             Some("\"Nulla\nconsequat erat at massa. \\\n Vivamus id mi.\"".to_string())
 471         );
 472     }
 473
 474     #[test]
 475     fn last_line_fit_with_trailing_whitespaces() {
 476         let string = "Vivamus id mi.  ";
 477         let config: Config = Default::default();
 478         let mut fmt = StringFormat::new(Shape::legacy(25, Indent::empty()), &config);
 479
 480         fmt.trim_end = true;
 481         let rewritten_string = rewrite_string(string, &fmt);
 482         assert_eq!(rewritten_string, Some("\"Vivamus id mi.\"".to_string()));
 483
 484         fmt.trim_end = false; // default value of trim_end
 485         let rewritten_string = rewrite_string(string, &fmt);
 486         assert_eq!(rewritten_string, Some("\"Vivamus id mi.  \"".to_string()));
 487     }
 488
 489     #[test]
 490     fn last_line_fit_with_newline() {
 491         let string = "Vivamus id mi.\nVivamus id mi.";
 492         let config: Config = Default::default();
 493         let fmt = StringFormat {
 494             opener: "",
 495             closer: "",
 496             line_start: "// ",
 497             line_end: "",
 498             shape: Shape::legacy(100, Indent::from_width(&config, 4)),
 499             trim_end: true,
 500             config: &config,
 501         };
 502
 503         let rewritten_string = rewrite_string(string, &fmt);
 504         assert_eq!(
 505             rewritten_string,
 506             Some("Vivamus id mi.\n    // Vivamus id mi.".to_string())
 507         );
 508     }
 509
 510     #[test]
 511     fn overflow_in_non_string_content() {
 512         let comment = "Aenean metus.\nVestibulum ac lacus. Vivamus porttitor";
 513         let config: Config = Default::default();
 514         let fmt = StringFormat {
 515             opener: "",
 516             closer: "",
 517             line_start: "// ",
 518             line_end: "",
 519             shape: Shape::legacy(30, Indent::from_width(&config, 8)),
 520             trim_end: true,
 521             config: &config,
 522         };
 523
 524         assert_eq!(
 525             rewrite_string(comment, &fmt),
 526             Some(
 527                 "Aenean metus.\n        // Vestibulum ac lacus. Vivamus\n        // porttitor"
 528                     .to_string()
 529             )
 530         );
 531     }
 532
 533     #[test]
 534     fn overflow_in_non_string_content_with_line_end() {
 535         let comment = "Aenean metus.\nVestibulum ac lacus. Vivamus porttitor";
 536         let config: Config = Default::default();
 537         let fmt = StringFormat {
 538             opener: "",
 539             closer: "",
 540             line_start: "// ",
 541             line_end: "@",
 542             shape: Shape::legacy(30, Indent::from_width(&config, 8)),
 543             trim_end: true,
 544             config: &config,
 545         };
 546
 547         assert_eq!(
 548             rewrite_string(comment, &fmt),
 549             Some(
 550                 "Aenean metus.\n        // Vestibulum ac lacus. Vivamus@\n        // porttitor"
 551                     .to_string()
 552             )
 553         );
 554     }
 555
 556     #[test]
 557     fn blank_line_with_non_empty_line_start() {
 558         let config: Config = Default::default();
 559         let mut fmt = StringFormat {
 560             opener: "",
 561             closer: "",
 562             line_start: "// ",
 563             line_end: "",
 564             shape: Shape::legacy(30, Indent::from_width(&config, 4)),
 565             trim_end: true,
 566             config: &config,
 567         };
 568
 569         let comment = "Aenean metus. Vestibulum\n\nac lacus. Vivamus porttitor";
 570         assert_eq!(
 571             rewrite_string(comment, &fmt),
 572             Some(
 573                 "Aenean metus. Vestibulum\n    //\n    // ac lacus. Vivamus porttitor".to_string()
 574             )
 575         );
 576
 577         fmt.shape = Shape::legacy(15, Indent::from_width(&config, 4));
 578         let comment = "Aenean\n\nmetus. Vestibulum ac lacus. Vivamus porttitor";
 579         assert_eq!(
 580             rewrite_string(comment, &fmt),
 581             Some(
 582                 r#"Aenean
 583     //
 584     // metus. Vestibulum
 585     // ac lacus. Vivamus
 586     // porttitor"#
 587                     .to_string()
 588             )
 589         );
 590     }
 591
 592     #[test]
 593     fn retain_blank_lines() {
 594         let config: Config = Default::default();
 595         let fmt = StringFormat {
 596             opener: "",
 597             closer: "",
 598             line_start: "// ",
 599             line_end: "",
 600             shape: Shape::legacy(20, Indent::from_width(&config, 4)),
 601             trim_end: true,
 602             config: &config,
 603         };
 604
 605         let comment = "Aenean\n\nmetus. Vestibulum ac lacus.\n\n";
 606         assert_eq!(
 607             rewrite_string(comment, &fmt),
 608             Some(
 609                 "Aenean\n    //\n    // metus. Vestibulum ac\n    // lacus.\n    //\n".to_string()
 610             )
 611         );
 612
 613         let comment = "Aenean\n\nmetus. Vestibulum ac lacus.\n";
 614         assert_eq!(
 615             rewrite_string(comment, &fmt),
 616             Some("Aenean\n    //\n    // metus. Vestibulum ac\n    // lacus.\n".to_string())
 617         );
 618
 619         let comment = "Aenean\n        \nmetus. Vestibulum ac lacus.";
 620         assert_eq!(
 621             rewrite_string(comment, &fmt),
 622             Some("Aenean\n    //\n    // metus. Vestibulum ac\n    // lacus.".to_string())
 623         );
 624     }
 625
 626     #[test]
 627     fn boundary_on_edge() {
 628         let config: Config = Default::default();
 629         let mut fmt = StringFormat {
 630             opener: "",
 631             closer: "",
 632             line_start: "// ",
 633             line_end: "",
 634             shape: Shape::legacy(13, Indent::from_width(&config, 4)),
 635             trim_end: true,
 636             config: &config,
 637         };
 638
 639         let comment = "Aenean metus. Vestibulum ac lacus.";
 640         assert_eq!(
 641             rewrite_string(comment, &fmt),
 642             Some("Aenean metus.\n    // Vestibulum ac\n    // lacus.".to_string())
 643         );
 644
 645         fmt.trim_end = false;
 646         let comment = "Vestibulum ac lacus.";
 647         assert_eq!(
 648             rewrite_string(comment, &fmt),
 649             Some("Vestibulum \n    // ac lacus.".to_string())
 650         );
 651
 652         fmt.trim_end = true;
 653         fmt.line_end = "\\";
 654         let comment = "Vestibulum ac lacus.";
 655         assert_eq!(
 656             rewrite_string(comment, &fmt),
 657             Some("Vestibulum\\\n    // ac lacus.".to_string())
 658         );
 659     }
 660
 661     #[test]
 662     fn detect_urls() {
 663         let string = "aaa http://example.org something";
 664         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 665         assert_eq!(detect_url(&graphemes, 8), Some(21));
 666
 667         let string = "https://example.org something";
 668         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 669         assert_eq!(detect_url(&graphemes, 0), Some(18));
 670
 671         let string = "aaa ftp://example.org something";
 672         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 673         assert_eq!(detect_url(&graphemes, 8), Some(20));
 674
 675         let string = "aaa file://example.org something";
 676         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 677         assert_eq!(detect_url(&graphemes, 8), Some(21));
 678
 679         let string = "aaa http not an url";
 680         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 681         assert_eq!(detect_url(&graphemes, 6), None);
 682
 683         let string = "aaa file://example.org";
 684         let graphemes = UnicodeSegmentation::graphemes(&*string, false).collect::<Vec<&str>>();
 685         assert_eq!(detect_url(&graphemes, 8), Some(21));
 686     }
 687 }