feat: 支持 JS 对象和数组转 Map 和 Array

HarlonWang · HarlonWang · commit e0ee90db99c3 · 2024-09-27T16:22:40.000+08:00
diff --git a/wrapper-android/src/androidTest/java/com/whl/quickjs/wrapper/QuickJSTest.java b/wrapper-android/src/androidTest/java/com/whl/quickjs/wrapper/QuickJSTest.java
@@ -1194,4 +1194,19 @@ public void testAsyncErrorInByteCode() {
         context.destroy();
     }
 
+    @Test
+    public void testObjectToMap() {
+        QuickJSContext context = createContext();
+        JSObject ret = (JSObject) context.evaluate("var a = {'a': '123', 'b': [1, 2]};a.c = a;;a;");
+        assertEquals("{a=123, b=[1, 2]}", ret.toMap().toString());
+        ret.release();
+
+        JSArray array = (JSArray) context.evaluate("var b = [{a: { c : 'xxx'}}, 2, 'qqq', 1.22]; b.push(b); b;");
+        assertEquals("[{a={c=xxx}}, 2, qqq, 1.22]", array.toArray().toString());
+        array.release();
+
+        assertEquals("{a={a=123, b=[1, 2]}, b=[{a={c=xxx}}, 2, qqq, 1.22], Infinity=-9223372036854775808, NaN=NaN, Math={LN2=0.6931471805599453, LN10=2.302585092994046, LOG2E=1.4426950408889634, E=2.718281828459045, SQRT2=1.4142135623730951, LOG10E=0.4342944819032518, PI=3.141592653589793, SQRT1_2=0.7071067811865476}, undefined=null}", context.getGlobalObject().toMap().toString());
+        context.destroy();
+    }
+
 }
diff --git a/wrapper-java/src/main/java/com/whl/quickjs/wrapper/JSObject.java b/wrapper-java/src/main/java/com/whl/quickjs/wrapper/JSObject.java
@@ -1,5 +1,8 @@
 package com.whl.quickjs.wrapper;
 
+import java.util.ArrayList;
+import java.util.HashMap;
+
 public interface JSObject {
 
     void setStackTrace(Throwable trace);
@@ -51,4 +54,8 @@ public interface JSObject {
      * 引用计数减一，目前仅将对象返回到 JavaScript 中的场景中使用。
      */
     void decrementRefCount();
+
+    HashMap<String, Object> toMap();
+
+    ArrayList<Object> toArray();
 }
diff --git a/wrapper-java/src/main/java/com/whl/quickjs/wrapper/QuickJSArray.java b/wrapper-java/src/main/java/com/whl/quickjs/wrapper/QuickJSArray.java
@@ -1,5 +1,8 @@
 package com.whl.quickjs.wrapper;
 
+import java.util.ArrayList;
+import java.util.HashMap;
+
 /**
  * Created by Harlon Wang on 2024/2/13.
  */
@@ -26,4 +29,17 @@ public void set(Object value, int index) {
         checkRefCountIsZero();
         getContext().set(this, value, index);
     }
+
+    @Override
+    public HashMap<String, Object> toMap() {
+        throw new UnsupportedOperationException("Array types are not yet supported for conversion to map. You should use toArray.");
+    }
+
+    @Override
+    public ArrayList<Object> toArray() {
+        ArrayList<Object> arrayList = new ArrayList<>(length());
+        convertToMap(this, arrayList);
+        circulars.clear();
+        return arrayList;
+    }
 }
diff --git a/wrapper-java/src/main/java/com/whl/quickjs/wrapper/QuickJSFunction.java b/wrapper-java/src/main/java/com/whl/quickjs/wrapper/QuickJSFunction.java
@@ -1,5 +1,8 @@
 package com.whl.quickjs.wrapper;
 
+import java.util.ArrayList;
+import java.util.HashMap;
+
 /**
  * Created by Harlon Wang on 2024/2/12.
  */
@@ -60,4 +63,13 @@ public void callVoid(Object... args) {
         }
     }
 
+    @Override
+    public HashMap<String, Object> toMap() {
+        throw new UnsupportedOperationException("JSFunction types do not support conversion to map or array.");
+    }
+
+    @Override
+    public ArrayList<Object> toArray() {
+        throw new UnsupportedOperationException("JSFunction types do not support conversion to map or array.");
+    }
 }
diff --git a/wrapper-java/src/main/java/com/whl/quickjs/wrapper/QuickJSObject.java b/wrapper-java/src/main/java/com/whl/quickjs/wrapper/QuickJSObject.java
@@ -2,7 +2,9 @@
 
 import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
+import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.HashMap;
 
 /**
  * Created by Harlon Wang on 2024/2/12.
@@ -13,6 +15,7 @@ public class QuickJSObject implements JSObject {
     private final long pointer;
     private int refCount;
     private Throwable stackTrace;
+    protected ArrayList<Long> circulars;
 
     public QuickJSObject(QuickJSContext context, long pointer) {
         this.context = context;
@@ -239,6 +242,89 @@ public void decrementRefCount() {
         refCount--;
     }
 
+    @Override
+    public HashMap<String, Object> toMap() {
+        HashMap<String, Object> objectMap = new HashMap<>();
+        convertToMap(this, objectMap);
+        circulars.clear();
+        return objectMap;
+    }
+
+    @Override
+    public ArrayList<Object> toArray() {
+        throw new UnsupportedOperationException("Object types are not yet supported for conversion to array. You should use toMap.");
+    }
+
+    protected void convertToMap(Object target, Object map) {
+        if (circulars == null) {
+            circulars = new ArrayList<>();
+        }
+        if (circulars.contains(((JSObject) target).getPointer())) {
+            // Circular reference objects, no processing needed.
+            return;
+        }
+
+        circulars.add(((JSObject) target).getPointer());
+
+        boolean isArray = target instanceof JSArray;
+        JSArray array = isArray ? (JSArray) target : ((JSObject) target).getNames();
+        int length = array.length();
+        for (int i = 0; i < length; i++) {
+            String key = null;
+            Object value;
+            if (isArray) {
+                value = array.get(i);
+            } else {
+                key = (String) array.get(i);
+                value = ((JSObject) target).getProperty(key);
+            }
+
+            if (value instanceof JSFunction) {
+                // Unsupported type.
+                ((JSFunction) value).release();
+                continue;
+            }
+
+            if (value instanceof JSArray) {
+                ArrayList<Object> list = new ArrayList<>(((JSArray) value).length());
+                convertToMap(value, list);
+                if (!list.isEmpty()) {
+                    if (map instanceof HashMap) {
+                        ((HashMap<String, Object>) map).put(key, list);
+                    } else if (map instanceof ArrayList){
+                        ((ArrayList<Object>) map).add(list);
+                    }
+                }
+                ((JSArray) value).release();
+                continue;
+            }
+
+            if (value instanceof JSObject) {
+                HashMap<String, Object> valueMap = new HashMap<>();
+                convertToMap(value, valueMap);
+                if (!valueMap.isEmpty()) {
+                    if (map instanceof HashMap) {
+                        ((HashMap<String, Object>) map).put(key, valueMap);
+                    } else if (map instanceof ArrayList){
+                        ((ArrayList<Object>) map).add(valueMap);
+                    }
+                }
+                ((JSObject) value).release();
+                continue;
+            }
+
+            // Primitive types.
+            if (map instanceof HashMap) {
+                ((HashMap<String, Object>) map).put(key, value);
+            } else if (map instanceof ArrayList){
+                ((ArrayList<Object>) map).add(value);
+            }
+        }
+        if (!isArray) {
+            array.release();
+        }
+    }
+
     public int getRefCount() {
         return refCount;
     }
