fix(fmt): properly format enums (#6637)

crates/fmt/src/formatter.rs

@@ -2038,31 +2038,27 @@ impl<'a, W: Write> Visitor for Formatter<'a, W> {
         let enum_name = enumeration.name.safe_unwrap_mut();
         let mut name =
             self.visit_to_chunk(enum_name.loc.start(), Some(enum_name.loc.end()), enum_name)?;
-        name.content = format!("enum {}", name.content);
-        self.write_chunk(&name)?;
-
+        name.content = format!("enum {} ", name.content);
         if enumeration.values.is_empty() {
+            self.write_chunk(&name)?;
             self.write_empty_brackets()?;
         } else {
-            self.surrounded(
-                SurroundingChunk::new(
-                    "{",
-                    Some(enumeration.values.first_mut().unwrap().safe_unwrap().loc.start()),
-                    None,
-                ),
-                SurroundingChunk::new("}", None, Some(enumeration.loc.end())),
-                |fmt, _multiline| {
-                    let values = fmt.items_to_chunks(
-                        Some(enumeration.loc.end()),
-                        enumeration.values.iter_mut().map(|ident| {
-                            let ident = ident.safe_unwrap_mut();
-                            (ident.loc, ident)
-                        }),
-                    )?;
-                    fmt.write_chunks_separated(&values, ",", true)?;
-                    Ok(())
-                },
-            )?;
+            name.content.push('{');
+            self.write_chunk(&name)?;
+
+            self.indented(1, |fmt| {
+                let values = fmt.items_to_chunks(
+                    Some(enumeration.loc.end()),
+                    enumeration.values.iter_mut().map(|ident| {
+                        let ident = ident.safe_unwrap_mut();
+                        (ident.loc, ident)
+                    }),
+                )?;
+                fmt.write_chunks_separated(&values, ",", true)?;
+                writeln!(fmt.buf())?;
+                Ok(())
+            })?;
+            write_chunk!(self, "}}")?;
         }
 
         Ok(())

crates/fmt/testdata/EnumVariants/fmt.sol

@@ -0,0 +1,19 @@
+interface I {
+    enum Empty {}
+
+    /// A modification applied to either `msg.sender` or `tx.origin`. Returned by `readCallers`.
+    enum CallerMode {
+        /// No caller modification is currently active.
+        None
+    }
+
+    /// A modification applied to either `msg.sender` or `tx.origin`. Returned by `readCallers`.
+    enum CallerMode2 {
+        /// No caller modification is currently active.
+        None,
+        /// No caller modification is currently active2.
+        Some
+    }
+
+    function bar() public {}
+}

crates/fmt/testdata/EnumVariants/original.sol

@@ -0,0 +1,23 @@
+interface I {
+    enum Empty {
+
+    }
+
+    /// A modification applied to either `msg.sender` or `tx.origin`. Returned by `readCallers`.
+    enum CallerMode
+    {/// No caller modification is currently active.
+        None
+    }
+
+    /// A modification applied to either `msg.sender` or `tx.origin`. Returned by `readCallers`.
+    enum CallerMode2
+    {/// No caller modification is currently active.
+        None,/// No caller modification is currently active2.
+
+        Some
+    }
+
+    function bar() public {
+
+    }
+}

crates/fmt/tests/formatter.rs

@@ -229,6 +229,7 @@ test_directories! {
     EmitStatement,
     Repros,
     BlockComments,
+    EnumVariants,
 }
 
 test_dir!(SortedImports, TestConfig::skip_compare_ast_eq());