feat: #204 - Add WithStatement.

src/compiler/statement/WithStatement.ts

@@ -0,0 +1,29 @@
+import * as ts from "typescript";
+import {removeStatementedNodeChild} from "./../../manipulation";
+import {ChildOrderableNode} from "./../base";
+import {Expression} from "./../common";
+import {Statement} from "./Statement";
+
+export const WithStatementBase = ChildOrderableNode(Statement);
+export class WithStatement extends WithStatementBase<ts.WithStatement> {
+    /**
+     * Gets this with statement's expression.
+     */
+    getExpression() {
+        return this.getNodeFromCompilerNode(this.compilerNode.expression) as Expression;
+    }
+
+    /**
+     * Gets this with statement's statement.
+     */
+    getStatement() {
+        return this.getNodeFromCompilerNode(this.compilerNode.statement) as Statement;
+    }
+
+    /**
+     * Removes this with statement.
+     */
+    remove() {
+        removeStatementedNodeChild(this);
+    }
+}

src/compiler/statement/index.ts

@@ -20,3 +20,4 @@ export * from "./VariableDeclarationList";
 export * from "./VariableDeclarationType";
 export * from "./VariableStatement";
 export * from "./WhileStatement";
+export * from "./WithStatement";

src/factories/nodeToWrapperMappings.ts

@@ -81,6 +81,7 @@ export const nodeToWrapperMappings: { [key: number]: any } = {
     [ts.SyntaxKind.JSDocPropertyTag]: compiler.JSDocPropertyTag,
     [ts.SyntaxKind.SemicolonToken]: compiler.Node,
     [ts.SyntaxKind.WhileStatement]: compiler.WhileStatement,
+    [ts.SyntaxKind.WithStatement]: compiler.WithStatement,
     // literals
     [ts.SyntaxKind.AnyKeyword]: compiler.Expression,
     [ts.SyntaxKind.BooleanKeyword]: compiler.Expression,

src/tests/compiler/statement/withStatementTests.ts

@@ -0,0 +1,36 @@
+import * as ts from "typescript";
+import {expect} from "chai";
+import {WithStatement} from "./../../../compiler";
+import {getInfoFromText} from "./../testHelpers";
+
+function getInfoFromTextWithForInStatement(text: string) {
+    const obj = getInfoFromText(text);
+    const withStatement = (
+        obj.sourceFile.getFirstDescendantByKindOrThrow(ts.SyntaxKind.WithStatement)
+    ) as WithStatement;
+    return {...obj, withStatement};
+}
+
+describe(nameof(WithStatement), () => {
+    describe(nameof<WithStatement>(n => n.getExpression), () => {
+        function doTest(text: string, expectedText: string) {
+            const {withStatement} = getInfoFromTextWithForInStatement(text);
+            expect(withStatement.getExpression().getText()).to.equal(expectedText);
+        }
+
+        it("should get the correct statement", () => {
+            doTest("with (Math) { PI; }", "Math");
+        });
+    });
+
+    describe(nameof<WithStatement>(n => n.getStatement), () => {
+        function doTest(text: string, expectedText: string) {
+            const {withStatement} = getInfoFromTextWithForInStatement(text);
+            expect(withStatement.getStatement().getText()).to.equal(expectedText);
+        }
+
+        it("should get the correct statement", () => {
+            doTest("with (Math) { PI; }", "{ PI; }");
+        });
+    });
+});