Make sure formatter doesn't remove internal annotation

Author: giacomocavalieri

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## Unreleased
+
+### Formatter
+
+- Fixed a bug where the `@internal` annotation wouldn't be displayed.
+
 ## v1.1.0-rc2 - 2024-04-10
 
 ### Compiler

compiler-core/src/format.rs

@@ -2175,7 +2175,8 @@ impl<'a> Documentable<'a> for &'a ArgNames {
 fn pub_(publicity: Publicity) -> Document<'static> {
     match publicity {
         Publicity::Public => "pub ".to_doc(),
-        Publicity::Private | Publicity::Internal => nil(),
+        Publicity::Private => nil(),
+        Publicity::Internal => "@internal".to_doc().append(line()).append("pub "),
     }
 }
 

compiler-core/src/format/tests.rs

@@ -5905,3 +5905,32 @@ fn comments_inside_binop_chain() {
 "#
     );
 }
+
+#[test]
+fn internal_attribute_on_function() {
+    assert_format!(
+        r#"@internal
+pub fn main() {
+  todo
+}
+"#
+    );
+}
+
+#[test]
+fn internal_attribute_on_type() {
+    assert_format!(
+        r#"@internal
+pub type Type
+"#
+    );
+}
+
+#[test]
+fn internal_attribute_on_const() {
+    assert_format!(
+        r#"@internal
+pub const wibble = 1
+"#
+    );
+}