Add #remove on JsxElement and JsxSelfClosingElement

packages/ts-morph/src/compiler/ast/jsx/JsxElement.ts

@@ -1,5 +1,5 @@
-import { errors, nameof, ts } from "@ts-morph/common";
-import { insertIntoParentTextRange } from "../../../manipulation";
+import { errors, nameof, SyntaxKind, ts } from "@ts-morph/common";
+import { insertIntoParentTextRange, removeChildren } from "../../../manipulation";
 import { JsxElementSpecificStructure, JsxElementStructure, StructureKind } from "../../../structures";
 import { WriterFunction } from "../../../types";
 import { printTextFromStringOrWriter } from "../../../utils";
@@ -104,6 +104,21 @@ export class JsxElement extends JsxElementBase<ts.JsxElement> {
     delete structure.children;
     return structure;
   }
+
+  /**
+   * Removes the JSX element.
+   */
+  remove() {
+    const parentKind = this.getParent()?.getKind()
+
+    if (!(parentKind === SyntaxKind.JsxElement || parentKind === SyntaxKind.JsxOpeningElement || parentKind === SyntaxKind.JsxFragment)) {
+      throw new errors.InvalidOperationError(`Error removing JsxElement: parent is ${this.getParent()?.getKindName() ?? '(no parent)'} and therefore the node cannot be removed. Only JsxElements with JsxElement/JsxOpeningElement/JsxFragment parent can be removed`)
+    }
+
+    return removeChildren({
+      children: [this],
+    })
+  }
 }
 
 function setText(element: JsxElement, newText: string) {

packages/ts-morph/src/compiler/ast/jsx/JsxSelfClosingElement.ts

@@ -1,5 +1,6 @@
-import { ts } from "@ts-morph/common";
+import { errors, ts, SyntaxKind } from "@ts-morph/common";
 import { JsxSelfClosingElementSpecificStructure, JsxSelfClosingElementStructure, StructureKind } from "../../../structures";
+import { removeChildren } from "../../../manipulation";
 import { callBaseGetStructure } from "../callBaseGetStructure";
 import { callBaseSet } from "../callBaseSet";
 import { PrimaryExpression } from "../expression";
@@ -25,4 +26,19 @@ export class JsxSelfClosingElement extends JsxSelfClosingElementBase<ts.JsxSelfC
       kind: StructureKind.JsxSelfClosingElement,
     });
   }
+
+  /**
+   * Removes the JSX self-closing element.
+   */
+   remove() {
+    const parentKind = this.getParent()?.getKind()
+
+    if (!(parentKind === SyntaxKind.JsxElement || parentKind === SyntaxKind.JsxOpeningElement || parentKind === SyntaxKind.JsxFragment)) {
+      throw new errors.InvalidOperationError(`Error removing JsxSelfClosingElement: parent is ${this.getParent()?.getKindName() ?? '(no parent)'} and therefore the node cannot be removed. Only JsxSelfClosingElements with JsxElement/JsxOpeningElement/JsxFragment parent can be removed`)
+    }
+
+    return removeChildren({
+      children: [this],
+    })
+  }
 }

packages/ts-morph/src/tests/compiler/ast/jsx/jsxElementTests.ts

@@ -1,6 +1,6 @@
 import { errors, nameof, SyntaxKind } from "@ts-morph/common";
 import { expect } from "chai";
-import { JsxElement } from "../../../../compiler";
+import { JsxElement, JsxSelfClosingElement } from "../../../../compiler";
 import { JsxAttributeStructure, JsxElementStructure, StructureKind } from "../../../../structures";
 import { getInfoFromTextWithDescendant, OptionalKindAndTrivia, OptionalTrivia } from "../../testHelpers";
 
@@ -159,4 +159,34 @@ const v = <div>
       });
     });
   });
+
+  describe(nameof<JsxElement>("remove"), () => {
+    function doRemove(text: string) {
+      const { descendant, sourceFile } = getInfo(text);
+      descendant.remove();
+    }
+
+    function doTestWithJsxElementChild(text: string, expected: string) {
+      const { descendant, sourceFile } = getInfo(text);
+      (descendant.getFirstDescendantByKind(SyntaxKind.JsxElement) as JsxElement).remove();
+      expect(sourceFile.getFullText()).to.equal(expected);
+    }
+
+    it("should not remove the root JsxElement", () => {
+      let error = null;
+
+      try {
+        doRemove(`var t = (<jsx></jsx>);`);
+      }
+      catch (err) {
+        error = err;
+      }
+
+      expect(error).to.be.instanceOf(errors.InvalidOperationError);
+    });
+
+    it("should remove the JsxElement child", () => {
+      doTestWithJsxElementChild(`var t = (<jsx><jsx2></jsx2></jsx>);`, `var t = (<jsx></jsx>);`);
+    });
+  });
 });

packages/ts-morph/src/tests/compiler/ast/jsx/jsxSelfClosingElementTests.ts

@@ -1,4 +1,4 @@
-import { nameof, SyntaxKind } from "@ts-morph/common";
+import { errors,  nameof, SyntaxKind } from "@ts-morph/common";
 import { expect } from "chai";
 import { JsxSelfClosingElement } from "../../../../compiler";
 import { JsxAttributeStructure, JsxSelfClosingElementStructure, StructureKind } from "../../../../structures";
@@ -167,4 +167,34 @@ describe("JsxSelfClosingElement", () => {
       expect(descendant.getFullText()).to.equal(`<jsx // comment1\n    {...props1} // comment2\n    {...props2} />`);
     });
   });
+
+  describe(nameof<JsxSelfClosingElement>("remove"), () => {
+    function doRemove(text: string) {
+      const { descendant, sourceFile } = getInfo(text);
+      descendant.remove();
+    }
+
+    function doTestWithJsxSelfClosingElementChild(text: string, expected: string) {
+      const { descendant, sourceFile } = getInfo(text);
+      descendant.remove();
+      expect(sourceFile.getFullText()).to.equal(expected);
+    }
+
+    it("should not remove the root JsxSelfClosingElement", () => {
+      let error = null;
+
+      try {
+        doRemove(`var t = (<jsx />);`);
+      }
+      catch (err) {
+        error = err;
+      }
+
+      expect(error).to.be.instanceOf(errors.InvalidOperationError);
+    });
+
+    it("should remove the JsxSelfClosingElement child", () => {
+      doTestWithJsxSelfClosingElementChild(`var t = (<jsx><jsx2 /></jsx>);`, `var t = (<jsx></jsx>);`);
+    });
+  });
 });