json-as 1.1.7 → 1.1.9

package/transform/src/util.ts

@@ -1,5 +1,5 @@
 // Taken from https://github.com/as-pect/visitor-as/blob/master/src/simpleParser.ts
-import { Parser, Tokenizer, Source, SourceKind, Expression, Statement, NamespaceDeclaration, ClassDeclaration, DeclarationStatement, Range, Node } from "assemblyscript/dist/assemblyscript.js";
+import { Parser, Tokenizer, Source, SourceKind, Expression, Statement, NamespaceDeclaration, ClassDeclaration, DeclarationStatement, Range, Node, NodeKind } from "assemblyscript/dist/assemblyscript.js";
 import { ASTBuilder } from "./builder.js";
 
 export class SimpleParser {
@@ -54,3 +54,69 @@ export function isStdlib(s: Source | { range: Range }): boolean {
 export function toString(node: Node): string {
   return ASTBuilder.build(node);
 }
+
+export function replaceRef(node: Node, replacement: Node | Node[], ref: Node | Node[] | null): void {
+  if (!node || !ref) return;
+  const nodeExpr = stripExpr(node);
+
+  if (Array.isArray(ref)) {
+    for (let i = 0; i < ref.length; i++) {
+      if (stripExpr(ref[i]) === nodeExpr) {
+        if (Array.isArray(replacement)) ref.splice(i, 1, ...replacement);
+        else ref.splice(i, 1, replacement);
+        return; // Exit early after replacement
+      }
+    }
+  } else if (typeof ref === "object") {
+    for (const key of Object.keys(ref)) {
+      const current = ref[key] as Node | Node[];
+      if (Array.isArray(current)) {
+        for (let i = 0; i < current.length; i++) {
+          if (stripExpr(current[i]) === nodeExpr) {
+            if (Array.isArray(replacement)) current.splice(i, 1, ...replacement);
+            else current.splice(i, 1, replacement);
+            return;
+          }
+        }
+      } else if (stripExpr(current) === nodeExpr) {
+        ref[key] = replacement;
+        return;
+      }
+    }
+  }
+}
+
+export function cloneNode(input: Node | Node[] | null, seen = new WeakMap(), path = ""): Node | Node[] | null {
+  if (input === null || typeof input !== "object") return input;
+
+  if (Array.isArray(input)) {
+    return input.map((item, index) => cloneNode(item, seen, `${path}[${index}]`)) as Node | Node[] | null;
+  }
+
+  if (seen.has(input)) return seen.get(input);
+
+  const prototype = Object.getPrototypeOf(input);
+  const clone = Array.isArray(input) ? [] : Object.create(prototype);
+  seen.set(input, clone);
+
+  for (const key of Reflect.ownKeys(input)) {
+    const value = input[key];
+    const newPath = path ? `${path}.${String(key)}` : String(key);
+
+    if (newPath.endsWith(".source")) {
+      clone[key] = value;
+    } else if (value && typeof value === "object") {
+      clone[key] = cloneNode(value, seen, newPath);
+    } else {
+      clone[key] = value;
+    }
+  }
+
+  return clone as Node | Node[] | null;
+}
+
+export function stripExpr(node: Node): Node {
+  if (!node) return node;
+  if (node.kind == NodeKind.Expression) return node["expression"];
+  return node;
+}