Add support for enums and property signatures

uniqueiniquity · uniqueiniquity · commit 976c25c672b9 · 2017-10-30T15:05:55.000-07:00
diff --git a/src/services/jsDoc.ts b/src/services/jsDoc.ts
@@ -259,8 +259,6 @@ namespace ts.JsDoc {
     }
     function getCommentOwnerInfo(tokenAtPos: Node): CommentOwnerInfo | undefined {
         // TODO: add support for:
-        // - enums/enum members
-        // - property declarations
         // - potentially property assignments
         for (let commentOwner = tokenAtPos; commentOwner; commentOwner = commentOwner.parent) {
             switch (commentOwner.kind) {
@@ -273,6 +271,9 @@ namespace ts.JsDoc {
 
                 case SyntaxKind.ClassDeclaration:
                 case SyntaxKind.InterfaceDeclaration:
+                case SyntaxKind.PropertySignature:
+                case SyntaxKind.EnumDeclaration:
+                case SyntaxKind.EnumMember:
                     return { commentOwner };
 
                 case SyntaxKind.VariableStatement: {
diff --git a/tests/cases/fourslash/docCommentTemplateInterfaces.ts b/tests/cases/fourslash/docCommentTemplateInterfaces.ts
diff --git a/tests/cases/fourslash/docCommentTemplateInterfacesAndEnums.ts b/tests/cases/fourslash/docCommentTemplateInterfacesAndEnums.ts
@@ -0,0 +1,50 @@
+/// <reference path='fourslash.ts' />
+
+/////*interfaceFoo*/
+////interface Foo {
+////    /*propertybar*/
+////    bar: any;
+////
+////    /*methodbaz*/
+////    baz(message: any): void;
+////}
+////
+/////*enumStatus*/
+////const enum Status {
+////    /*memberOpen*/
+////    Open,
+////
+////    /*memberClosed*/
+////    Closed
+////}
+
+verify.docCommentTemplateAt("interfaceFoo", /*expectedOffset*/ 8,
+`/**
+ * 
+ */`);
+
+verify.docCommentTemplateAt("propertybar", /*expectedOffset*/ 12,
+   `/**
+     * 
+     */`);
+
+verify.docCommentTemplateAt("methodbaz", /*expectedOffset*/ 12,
+   `/**
+     * 
+     * @param message
+     */`);
+
+verify.docCommentTemplateAt("enumStatus", /*expectedOffset*/ 8,
+`/**
+ * 
+ */`);
+
+verify.docCommentTemplateAt("memberOpen", /*expectedOffset*/ 12,
+   `/**
+     * 
+     */`);
+
+verify.docCommentTemplateAt("memberClosed", /*expectedOffset*/ 12,
+   `/**
+     * 
+     */`);
