Merge pull request #2441 from topecongiro/issue-2438

Return `None` when format_code_block may have failed

rustfmt-core/src/lib.rs

@@ -564,18 +564,29 @@ pub fn format_code_block(code_snippet: &str, config: &Config) -> Option<String>
     // Wrap the given code block with `fn main()` if it does not have one.
     let fn_main_prefix = "fn main() {\n";
     let snippet = fn_main_prefix.to_owned() + code_snippet + "\n}";
+    let mut result = String::with_capacity(snippet.len());
+    let mut is_first = true;
 
     // Trim "fn main() {" on the first line and "}" on the last line,
     // then unindent the whole code block.
-    format_snippet(&snippet, config).map(|s| {
+    let formatted = format_snippet(&snippet, config)?;
     // 2 = "}\n"
-        s[fn_main_prefix.len()..s.len().checked_sub(2).unwrap_or(0)]
-            .lines()
-            .map(|line| {
-                if line.len() > config.tab_spaces() {
+    let block_len = formatted.len().checked_sub(2).unwrap_or(0);
+    for line in formatted[fn_main_prefix.len()..block_len].lines() {
+        if !is_first {
+            result.push('\n');
+        } else {
+            is_first = false;
+        }
+        let trimmed_line = if line.len() > config.max_width() {
+            // If there are lines that are larger than max width, we cannot tell
+            // whether we have succeeded but have some comments or strings that
+            // are too long, or we have failed to format code block. We will be
+            // conservative and just return `None` in this case.
+            return None;
+        } else if line.len() > config.tab_spaces() {
             // Make sure that the line has leading whitespaces.
-                    let indent_str =
-                        Indent::from_width(config, config.tab_spaces()).to_string(config);
+            let indent_str = Indent::from_width(config, config.tab_spaces()).to_string(config);
             if line.starts_with(indent_str.as_ref()) {
                 let offset = if config.hard_tabs() {
                     1
@@ -588,11 +599,10 @@ pub fn format_code_block(code_snippet: &str, config: &Config) -> Option<String>
             }
         } else {
             line
+        };
+        result.push_str(trimmed_line);
     }
-            })
-            .collect::<Vec<_>>()
-            .join("\n")
-    })
+    Some(result)
 }
 
 pub fn format_input<T: Write>(
@@ -781,6 +791,12 @@ fn test_format_snippet() {
         assert!(test_format_inner(format_snippet, snippet, expected));
     }
 
+    #[test]
+    fn test_format_code_block_fail() {
+        let code_block = "this_line_is_100_characters_long_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx(x, y, z);";
+        assert!(format_code_block(code_block, &Config::default()).is_none());
+    }
+
     #[test]
     fn test_format_code_block() {
         // simple code block

rustfmt-core/tests/source/macro_rules.rs

@@ -50,6 +50,15 @@ macro_rules! m {
 }
 }
 
+
+// #2438
+macro_rules! m {
+    () => {
+        this_line_is_99_characters_long_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx(
+        ); // this line is drifting
+    };
+}
+
 // #2439
 macro_rules! m {
     (

rustfmt-core/tests/target/macro_rules.rs

@@ -42,6 +42,14 @@ mod macro_item {
     }
 }
 
+// #2438
+macro_rules! m {
+    () => {
+        this_line_is_99_characters_long_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx(
+        ); // this line is drifting
+    };
+}
+
 // #2439
 macro_rules! m {
     (