feat: compatible with magic-string-stack

package.json

@@ -48,6 +48,7 @@
     "bumpp": "^9.4.0",
     "eslint": "^9.1.0",
     "fast-glob": "^3.3.2",
+    "magic-string-stack": "^0.1.1",
     "prettier": "^3.2.5",
     "tsup": "^8.0.2",
     "tsx": "^4.7.2",

src/index.ts

@@ -21,8 +21,11 @@ export class MagicStringAST implements MagicString {
       /** offset of node */
       offset?: number
     },
+    private prototype: typeof MagicString = typeof str === 'string'
+      ? MagicString
+      : (str.constructor as any),
   ) {
-    this.s = typeof str === 'string' ? new MagicString(str, options) : str
+    this.s = typeof str === 'string' ? new prototype(str, options) : str
     this.offset = options?.offset ?? 0
     return new Proxy(this.s, {
       get: (target, p, receiver) => {
@@ -86,11 +89,15 @@ export class MagicStringAST implements MagicString {
     let newS: MagicString
     if (isEmptyNodes(node)) newS = this.s.snip(0, 0)
     else newS = this.s.snip(...this.getNodePos(node, offset))
-    return new MagicStringAST(newS, { offset: this.offset })
+    return new MagicStringAST(newS, { offset: this.offset }, this.prototype)
   }
 
   clone(): this {
-    return new MagicStringAST(this.s.clone(), { offset: this.offset }) as any
+    return new MagicStringAST(
+      this.s.clone(),
+      { offset: this.offset },
+      this.prototype,
+    ) as any
   }
 
   toString(): string {

tests/index.test.ts

@@ -1,4 +1,5 @@
 import { expect, test } from 'vitest'
+import MagicStringStack from 'magic-string-stack'
 import { MagicString, MagicStringAST } from '../src'
 
 test('basic', () => {
@@ -37,3 +38,19 @@ test('empty array', () => {
   expect(s.overwriteNode([], 'new').toString()).toBe(ORIGINAL)
   expect(s.moveNode([], 0).toString()).toBe(ORIGINAL)
 })
+
+test('compatible with magic-string-stack', () => {
+  const s = new MagicStringAST(
+    'hello world',
+    undefined,
+    MagicStringStack,
+  ) as MagicStringAST & MagicStringStack
+  expect(s.commit).toBeDefined()
+  expect(s.clone().commit).toBeDefined()
+
+  const s2 = new MagicStringAST(
+    new MagicStringStack('hello world'),
+  ) as MagicStringAST & MagicStringStack
+  expect(s2.commit).toBeDefined()
+  expect(s2.clone().commit).toBeDefined()
+})