feat(rome_js_parser): export default function in declare module #3485

crates/rome_js_formatter/src/generated.rs

@@ -6318,6 +6318,21 @@ impl IntoFormat<crate::JsFormatContext> for rome_js_syntax::JsFunctionExportDefa
         FormatOwnedWithRule :: new (self , crate :: js :: declarations :: function_export_default_declaration :: FormatJsFunctionExportDefaultDeclaration :: default ())
     }
 }
+impl FormatRule < rome_js_syntax :: TsDeclareFunctionExportDefaultDeclaration > for crate :: ts :: declarations :: declare_function_export_default_declaration :: FormatTsDeclareFunctionExportDefaultDeclaration { type Context = JsFormatContext ; # [inline (always)] fn fmt (& self , node : & rome_js_syntax :: TsDeclareFunctionExportDefaultDeclaration , f : & mut JsFormatter) -> FormatResult < () > { FormatNodeRule :: < rome_js_syntax :: TsDeclareFunctionExportDefaultDeclaration > :: fmt (self , node , f) } }
+impl AsFormat for rome_js_syntax::TsDeclareFunctionExportDefaultDeclaration {
+    type Format < 'a > = FormatRefWithRule < 'a , rome_js_syntax :: TsDeclareFunctionExportDefaultDeclaration , crate :: ts :: declarations :: declare_function_export_default_declaration :: FormatTsDeclareFunctionExportDefaultDeclaration > ;
+    fn format(&self) -> Self::Format<'_> {
+        FormatRefWithRule :: new (self , crate :: ts :: declarations :: declare_function_export_default_declaration :: FormatTsDeclareFunctionExportDefaultDeclaration :: default ())
+    }
+}
+impl IntoFormat<crate::JsFormatContext>
+    for rome_js_syntax::TsDeclareFunctionExportDefaultDeclaration
+{
+    type Format = FormatOwnedWithRule < rome_js_syntax :: TsDeclareFunctionExportDefaultDeclaration , crate :: ts :: declarations :: declare_function_export_default_declaration :: FormatTsDeclareFunctionExportDefaultDeclaration > ;
+    fn into_format(self) -> Self::Format {
+        FormatOwnedWithRule :: new (self , crate :: ts :: declarations :: declare_function_export_default_declaration :: FormatTsDeclareFunctionExportDefaultDeclaration :: default ())
+    }
+}
 impl FormatRule<rome_js_syntax::JsExportNamedShorthandSpecifier>
     for crate::js::module::export_named_shorthand_specifier::FormatJsExportNamedShorthandSpecifier
 {

crates/rome_js_formatter/src/js/any/export_default_declaration.rs

@@ -14,10 +14,10 @@ impl FormatRule<JsAnyExportDefaultDeclaration> for FormatJsAnyExportDefaultDecla
             JsAnyExportDefaultDeclaration::JsFunctionExportDefaultDeclaration(node) => {
                 node.format().fmt(f)
             }
-            JsAnyExportDefaultDeclaration::TsDeclareFunctionDeclaration(node) => {
+            JsAnyExportDefaultDeclaration::TsInterfaceDeclaration(node) => node.format().fmt(f),
+            JsAnyExportDefaultDeclaration::TsDeclareFunctionExportDefaultDeclaration(node) => {
                 node.format().fmt(f)
             }
-            JsAnyExportDefaultDeclaration::TsInterfaceDeclaration(node) => node.format().fmt(f),
         }
     }
 }

crates/rome_js_formatter/src/js/declarations/function_declaration.rs

@@ -6,7 +6,8 @@ use rome_formatter::{write, RemoveSoftLinesBuffer};
 use rome_js_syntax::{
     JsAnyBinding, JsFunctionBody, JsFunctionDeclaration, JsFunctionExportDefaultDeclaration,
     JsFunctionExpression, JsParameters, JsSyntaxToken, TsAnyReturnType,
-    TsDeclareFunctionDeclaration, TsReturnTypeAnnotation, TsType, TsTypeParameters,
+    TsDeclareFunctionDeclaration, TsDeclareFunctionExportDefaultDeclaration,
+    TsReturnTypeAnnotation, TsType, TsTypeParameters,
 };
 use rome_rowan::{declare_node_union, SyntaxResult};
 
@@ -24,7 +25,8 @@ declare_node_union! {
         JsFunctionDeclaration |
         JsFunctionExpression |
         JsFunctionExportDefaultDeclaration |
-        TsDeclareFunctionDeclaration
+        TsDeclareFunctionDeclaration |
+        TsDeclareFunctionExportDefaultDeclaration
 }
 
 #[derive(Copy, Clone, Debug, Default)]
@@ -42,6 +44,9 @@ impl FormatFunction {
                 declaration.async_token()
             }
             FormatFunction::TsDeclareFunctionDeclaration(member) => member.async_token(),
+            FormatFunction::TsDeclareFunctionExportDefaultDeclaration(member) => {
+                member.async_token()
+            }
         }
     }
 
@@ -53,6 +58,9 @@ impl FormatFunction {
                 declaration.function_token()
             }
             FormatFunction::TsDeclareFunctionDeclaration(member) => member.function_token(),
+            FormatFunction::TsDeclareFunctionExportDefaultDeclaration(member) => {
+                member.function_token()
+            }
         }
     }
 
@@ -64,6 +72,7 @@ impl FormatFunction {
                 declaration.star_token()
             }
             FormatFunction::TsDeclareFunctionDeclaration(_) => None,
+            FormatFunction::TsDeclareFunctionExportDefaultDeclaration(_) => None,
         }
     }
 
@@ -73,6 +82,7 @@ impl FormatFunction {
             FormatFunction::JsFunctionExpression(expression) => Ok(expression.id()),
             FormatFunction::JsFunctionExportDefaultDeclaration(declaration) => Ok(declaration.id()),
             FormatFunction::TsDeclareFunctionDeclaration(member) => member.id().map(Some),
+            FormatFunction::TsDeclareFunctionExportDefaultDeclaration(member) => Ok(member.id()),
         }
     }
 
@@ -84,6 +94,9 @@ impl FormatFunction {
                 declaration.type_parameters()
             }
             FormatFunction::TsDeclareFunctionDeclaration(member) => member.type_parameters(),
+            FormatFunction::TsDeclareFunctionExportDefaultDeclaration(member) => {
+                member.type_parameters()
+            }
         }
     }
 
@@ -95,6 +108,9 @@ impl FormatFunction {
                 declaration.parameters()
             }
             FormatFunction::TsDeclareFunctionDeclaration(member) => member.parameters(),
+            FormatFunction::TsDeclareFunctionExportDefaultDeclaration(member) => {
+                member.parameters()
+            }
         }
     }
 
@@ -108,6 +124,9 @@ impl FormatFunction {
                 declaration.return_type_annotation()
             }
             FormatFunction::TsDeclareFunctionDeclaration(member) => member.return_type_annotation(),
+            FormatFunction::TsDeclareFunctionExportDefaultDeclaration(member) => {
+                member.return_type_annotation()
+            }
         }
     }
 
@@ -119,6 +138,7 @@ impl FormatFunction {
                 Some(declaration.body()?)
             }
             FormatFunction::TsDeclareFunctionDeclaration(_) => None,
+            FormatFunction::TsDeclareFunctionExportDefaultDeclaration(_) => None,
         })
     }
 

crates/rome_js_formatter/src/lib.rs

@@ -787,17 +787,9 @@ function() {
     // use this test check if your snippet prints as you wish, without using a snapshot
     fn quick_test() {
         let src = r#"
-// different output than prettier
-function Component4() {
-
-  return (
-    <div>
-      {fn(datadatadatadatadatadatadatadatadatadatadatadatadatadatadatadata)}{' '}
-      <div/>
-    </div>
-  );
+declare module 'x' {
+  export default function (option: any): void
 }
-
 "#;
         let syntax = SourceType::tsx();
         let tree = parse(src, FileId::zero(), syntax);

crates/rome_js_formatter/src/ts/declarations/declare_function_export_default_declaration.rs

@@ -0,0 +1,25 @@
+use crate::js::declarations::function_declaration::FormatFunction;
+use crate::prelude::*;
+use crate::utils::FormatStatementSemicolon;
+use rome_formatter::write;
+use rome_js_syntax::TsDeclareFunctionExportDefaultDeclaration;
+
+#[derive(Debug, Clone, Default)]
+pub struct FormatTsDeclareFunctionExportDefaultDeclaration;
+impl FormatNodeRule<TsDeclareFunctionExportDefaultDeclaration>
+    for FormatTsDeclareFunctionExportDefaultDeclaration
+{
+    fn fmt_fields(
+        &self,
+        node: &TsDeclareFunctionExportDefaultDeclaration,
+        f: &mut JsFormatter,
+    ) -> FormatResult<()> {
+        write![
+            f,
+            [
+                FormatFunction::from(node.clone()),
+                FormatStatementSemicolon::new(node.semicolon_token().as_ref())
+            ]
+        ]
+    }
+}

crates/rome_js_formatter/src/ts/declarations/mod.rs

@@ -1,6 +1,7 @@
 //! Generated file, do not edit by hand, see `xtask/codegen`
 
 pub(crate) mod declare_function_declaration;
+pub(crate) mod declare_function_export_default_declaration;
 pub(crate) mod enum_declaration;
 pub(crate) mod external_module_declaration;
 pub(crate) mod global_declaration;