feat: Add Node.getEndLineNumber()

dsherret · Feb 4, 2018 · f00ca4c · f00ca4c
1 parent f4ba433
commit f00ca4c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/compiler/common/Node.ts b/src/compiler/common/Node.ts
@@ -754,7 +754,16 @@ export class Node<NodeType extends ts.Node = ts.Node> {
      * @param includeJsDocComment - Whether to include the JS doc comment or not.
      */
     getStartLineNumber(includeJsDocComment?: boolean) {
-        return this.getSourceFile().getLineNumberFromPos(this.getStartLinePos(includeJsDocComment));
+        return this.sourceFile.getLineNumberFromPos(this.getStartLinePos(includeJsDocComment));
+    }
+
+    /**
+     * Gets the line number of the end of the node.
+     */
+    getEndLineNumber() {
+        const sourceFileText = this.sourceFile.getFullText();
+        const endLinePos = getPreviousMatchingPos(sourceFileText, this.getEnd(), char => char === "\n");
+        return this.sourceFile.getLineNumberFromPos(endLinePos);
     }
 
     /**

diff --git a/src/tests/compiler/common/nodeTests.ts b/src/tests/compiler/common/nodeTests.ts
@@ -323,6 +323,13 @@ describe(nameof(Node), () => {
         });
     });
 
+    describe(nameof<Node>(n => n.getEndLineNumber), () => {
+        it("should get the end line number of the node", () => {
+            const {firstChild} = getInfoFromText<ClassDeclaration>("\n\nclass MyClass {\n\n    prop: string;\n}");
+            expect(firstChild.getEndLineNumber()).to.equal(6);
+        });
+    });
+
     describe(nameof<Node>(n => n.getStart), () => {
         function doTest(text: string, expectedPos: number, includeJsDocComment?: boolean) {
             const {firstChild} = getInfoFromText(text);