refactor(doc): add helper functions for DocNode (#7083)

1 files changed, 7 insertions, 99 deletions

diff --git a/cli/doc/module.rs b/cli/doc/module.rs
index cd0fdbd45..69effdde8 100644
--- a/cli/doc/module.rs
+++ b/cli/doc/module.rs
@@ -3,7 +3,6 @@ use swc_common::Spanned;
 
 use super::parser::DocParser;
 use super::DocNode;
-use super::DocNodeKind;
 
 pub fn get_doc_node_for_export_decl(
   doc_parser: &DocParser,
@@ -19,55 +18,16 @@ pub fn get_doc_node_for_export_decl(
     Decl::Class(class_decl) => {
       let (name, class_def) =
         super::class::get_doc_for_class_decl(doc_parser, class_decl);
-      DocNode {
-        kind: DocNodeKind::Class,
-        name,
-        location,
-        js_doc,
-        class_def: Some(class_def),
-        function_def: None,
-        variable_def: None,
-        enum_def: None,
-        type_alias_def: None,
-        namespace_def: None,
-        interface_def: None,
-        import_def: None,
-      }
+      DocNode::class(name, location, js_doc, class_def)
     }
     Decl::Fn(fn_decl) => {
       let (name, function_def) =
         super::function::get_doc_for_fn_decl(doc_parser, fn_decl);
-      DocNode {
-        kind: DocNodeKind::Function,
-        name,
-        location,
-        js_doc,
-        function_def: Some(function_def),
-        class_def: None,
-        variable_def: None,
-        enum_def: None,
-        type_alias_def: None,
-        namespace_def: None,
-        interface_def: None,
-        import_def: None,
-      }
+      DocNode::function(name, location, js_doc, function_def)
     }
     Decl::Var(var_decl) => {
       let (name, var_def) = super::variable::get_doc_for_var_decl(var_decl);
-      DocNode {
-        kind: DocNodeKind::Variable,
-        name,
-        location,
-        js_doc,
-        variable_def: Some(var_def),
-        function_def: None,
-        class_def: None,
-        enum_def: None,
-        type_alias_def: None,
-        namespace_def: None,
-        interface_def: None,
-        import_def: None,
-      }
+      DocNode::variable(name, location, js_doc, var_def)
     }
     Decl::TsInterface(ts_interface_decl) => {
       let (name, interface_def) =
@@ -75,20 +35,7 @@ pub fn get_doc_node_for_export_decl(
           doc_parser,
           ts_interface_decl,
         );
-      DocNode {
-        kind: DocNodeKind::Interface,
-        name,
-        location,
-        js_doc,
-        interface_def: Some(interface_def),
-        variable_def: None,
-        function_def: None,
-        class_def: None,
-        enum_def: None,
-        type_alias_def: None,
-        namespace_def: None,
-        import_def: None,
-      }
+      DocNode::interface(name, location, js_doc, interface_def)
     }
     Decl::TsTypeAlias(ts_type_alias) => {
       let (name, type_alias_def) =
@@ -96,56 +43,17 @@ pub fn get_doc_node_for_export_decl(
           doc_parser,
           ts_type_alias,
         );
-      DocNode {
-        kind: DocNodeKind::TypeAlias,
-        name,
-        location,
-        js_doc,
-        type_alias_def: Some(type_alias_def),
-        interface_def: None,
-        variable_def: None,
-        function_def: None,
-        class_def: None,
-        enum_def: None,
-        namespace_def: None,
-        import_def: None,
-      }
+      DocNode::type_alias(name, location, js_doc, type_alias_def)
     }
     Decl::TsEnum(ts_enum) => {
       let (name, enum_def) =
         super::r#enum::get_doc_for_ts_enum_decl(doc_parser, ts_enum);
-      DocNode {
-        kind: DocNodeKind::Enum,
-        name,
-        location,
-        js_doc,
-        enum_def: Some(enum_def),
-        type_alias_def: None,
-        interface_def: None,
-        variable_def: None,
-        function_def: None,
-        class_def: None,
-        namespace_def: None,
-        import_def: None,
-      }
+      DocNode::r#enum(name, location, js_doc, enum_def)
     }
     Decl::TsModule(ts_module) => {
       let (name, namespace_def) =
         super::namespace::get_doc_for_ts_module(doc_parser, ts_module);
-      DocNode {
-        kind: DocNodeKind::Namespace,
-        name,
-        location,
-        js_doc,
-        namespace_def: Some(namespace_def),
-        enum_def: None,
-        type_alias_def: None,
-        interface_def: None,
-        variable_def: None,
-        function_def: None,
-        class_def: None,
-        import_def: None,
-      }
+      DocNode::namespace(name, location, js_doc, namespace_def)
     }
   }
 }