feat(fmt): add all_params config - same as all but split single p…

…aram too (#9176) fet(fmt): add all_params config - smae as all but split single param too
foundry-rs · Oct 23, 2024 · b1e9365 · b1e9365
1 parent 4c84dc7
commit b1e9365
Show file tree

Hide file tree

Showing 3 changed files with 747 additions and 4 deletions.
diff --git a/crates/config/src/fmt.rs b/crates/config/src/fmt.rs
@@ -162,6 +162,8 @@ pub enum MultilineFuncHeaderStyle {
     /// If function params or attrs are multiline.
     /// split the rest
     All,
+    /// Same as `All` but writes function params multiline even when there is a single param.
+    AllParams,
 }
 
 impl Default for FormatterConfig {

diff --git a/crates/fmt/src/formatter.rs b/crates/fmt/src/formatter.rs
@@ -1628,7 +1628,8 @@ impl<'a, W: Write> Formatter<'a, W> {
                             fmt.config.multiline_func_header,
                             MultilineFuncHeaderStyle::ParamsFirst |
                                 MultilineFuncHeaderStyle::ParamsFirstMulti |
-                                MultilineFuncHeaderStyle::All
+                                MultilineFuncHeaderStyle::All |
+                                MultilineFuncHeaderStyle::AllParams
                         );
                     params_multiline = should_multiline ||
                         multiline ||
@@ -1637,11 +1638,12 @@ impl<'a, W: Write> Formatter<'a, W> {
                             &params,
                             ",",
                         )?;
-                    // Write new line if we have only one parameter and params first set.
+                    // Write new line if we have only one parameter and params first all multi set.
                     if params.len() == 1 &&
                         matches!(
                             fmt.config.multiline_func_header,
-                            MultilineFuncHeaderStyle::ParamsFirst
+                            MultilineFuncHeaderStyle::ParamsFirst |
+                                MultilineFuncHeaderStyle::AllParams
                         )
                     {
                         writeln!(fmt.buf())?;
@@ -1736,7 +1738,10 @@ impl<'a, W: Write> Formatter<'a, W> {
 
         let should_multiline = header_multiline &&
             if params_multiline {
-                matches!(self.config.multiline_func_header, MultilineFuncHeaderStyle::All)
+                matches!(
+                    self.config.multiline_func_header,
+                    MultilineFuncHeaderStyle::All | MultilineFuncHeaderStyle::AllParams
+                )
             } else {
                 matches!(
                     self.config.multiline_func_header,