feat(js/ts): add support for class prop documentation (#638)

helper/src/typescript/docs/jsdoc.yaml

@@ -91,3 +91,14 @@ templates:
        * @returns {% if show_types %}{{ "{" }}{{ return_type | default(value="[TODO:type]") }}{{ "}" }}{% endif %} [TODO:description]
       {% endif %}
        */
+
+  class_property:
+    node_types:
+      - public_field_definition
+    template: |
+      /**
+       * [TODO:description]
+      {% if show_types %}
+       * @type {{ "{" }}{{ type | default(value="[TODO:type]") }}{{ "}" }}
+      {% endif %}
+       */

helper/src/typescript/parser.rs

@@ -89,6 +89,8 @@ impl<'a> TypescriptParser<'a> {
                         }
                     },
 
+                    "public_field_definition" => Some(self.parse_class_property(&child_node)),
+
                     "class" | "class_declaration" => Some(self.parse_class(&child_node)),
 
                     _ => None,
@@ -99,6 +101,22 @@ impl<'a> TypescriptParser<'a> {
         None
     }
 
+    fn parse_class_property(&self, node: &Node) -> Result<Map<String, Value>, String> {
+        let mut tokens = Map::new();
+
+        for child_node in node.children(&mut node.walk()) {
+            match child_node.kind() {
+                "type_annotation" => {
+                    let type_node = child_node.children(&mut child_node.walk()).last().unwrap();
+                    tokens.insert("type".to_string(), Value::String(self.get_node_text(&type_node)));
+                },
+                _ => {},
+            }
+        }
+
+        Ok(tokens)
+    }
+
     fn parse_class(&self, node: &Node) -> Result<Map<String, Value>, String> {
         let mut tokens = Map::new();
 

test/filetypes/typescript/class-properties.vader

@@ -0,0 +1,53 @@
+# ==============================================================================
+# Functions with destructured parameters.
+# ==============================================================================
+Given typescript (class with multiple properties):
+  class TestClass {
+    foo
+    private bar;
+    baz: int
+    private bax: str = "test";
+    private bax2: str = "test";
+  }
+
+Do (run doge):
+  :2\<CR>
+  \<C-d>
+  :7\<CR>
+  \<C-d>
+  :12\<CR>
+  \<C-d>
+  :17\<CR>
+  \<C-d>
+  :let g:doge_javascript_settings['omit_redundant_param_types'] = 1\<CR>
+  :22\<CR>
+  \<C-d>
+  :let g:doge_javascript_settings['omit_redundant_param_types'] = 0\<CR>
+
+Expect typescript (each class prop has a docblock):
+  class TestClass {
+    /**
+     * [TODO:description]
+     * @type {[TODO:type]}
+     */
+    foo
+    /**
+     * [TODO:description]
+     * @type {[TODO:type]}
+     */
+    private bar;
+    /**
+     * [TODO:description]
+     * @type {int}
+     */
+    baz: int
+    /**
+     * [TODO:description]
+     * @type {str}
+     */
+    private bax: str = "test";
+    /**
+     * [TODO:description]
+     */
+    private bax2: str = "test";
+  }