From 9bf00c9536d79ae5a5a1c66908d15986672a66cb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stanis=C5=82aw=20Chmiela?= <sjchmiela@gmail.com>
Date: Tue, 5 May 2020 16:26:19 +0200
Subject: [PATCH] [@unimodules/core] Add JSONObjectArguments

---
 .../core/arguments/JSONObjectArguments.java   | 104 ++++++++++++++++++
 .../core/arguments/ReadableArguments.java     |   7 ++
 2 files changed, 111 insertions(+)
 create mode 100644 packages/@unimodules/core/android/src/main/java/org/unimodules/core/arguments/JSONObjectArguments.java

diff --git a/packages/@unimodules/core/android/src/main/java/org/unimodules/core/arguments/JSONObjectArguments.java b/packages/@unimodules/core/android/src/main/java/org/unimodules/core/arguments/JSONObjectArguments.java
new file mode 100644
index 0000000000000..fa4328cf23a05
--- /dev/null
+++ b/packages/@unimodules/core/android/src/main/java/org/unimodules/core/arguments/JSONObjectArguments.java
@@ -0,0 +1,104 @@
+package org.unimodules.core.arguments;
+
+import android.os.Bundle;
+
+import org.json.JSONArray;
+import org.json.JSONObject;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+
+public class JSONObjectArguments implements ReadableArguments {
+  private JSONObject mObject;
+
+  public JSONObjectArguments(JSONObject object) {
+    mObject = object;
+  }
+
+  @Override
+  public Collection<String> keys() {
+    Collection<String> keys = new ArrayList<>();
+    Iterator<String> iterator = mObject.keys();
+    while (iterator.hasNext()) {
+      keys.add(iterator.next());
+    }
+    return keys;
+  }
+
+  @Override
+  public boolean containsKey(String key) {
+    return mObject.has(key);
+  }
+
+  @Override
+  public Object get(String key) {
+    return mObject.opt(key);
+  }
+
+  @Override
+  public boolean getBoolean(String key, boolean defaultValue) {
+    return mObject.optBoolean(key, defaultValue);
+  }
+
+  @Override
+  public double getDouble(String key, double defaultValue) {
+    return mObject.optDouble(key, defaultValue);
+  }
+
+  @Override
+  public int getInt(String key, int defaultValue) {
+    return mObject.optInt(key, defaultValue);
+  }
+
+  @Override
+  public String getString(String key, String defaultValue) {
+    return mObject.optString(key, defaultValue);
+  }
+
+  @Override
+  @SuppressWarnings("unchecked")
+  public List getList(String key, List defaultValue) {
+    JSONArray jsonArray = mObject.optJSONArray(key);
+    if (jsonArray == null) {
+      return defaultValue;
+    }
+    List list = new ArrayList();
+    for (int i = 0; i < jsonArray.length(); i++) {
+      Object value = jsonArray.opt(i);
+      list.add(value);
+    }
+    return list;
+  }
+
+  @Override
+  @SuppressWarnings("unchecked")
+  public Map getMap(String key, Map defaultValue) {
+    JSONObject jsonObject = mObject.optJSONObject(key);
+    if (jsonObject == null) {
+      return defaultValue;
+    }
+    Bundle bundle = new JSONObjectArguments(jsonObject).toBundle();
+    Map map = new HashMap();
+    for (String subKey : bundle.keySet()) {
+      Object value = bundle.get(subKey);
+      if (value != null) {
+        map.put(subKey, value);
+      }
+    }
+    return map;
+  }
+
+  @Override
+  public boolean isEmpty() {
+    return mObject.length() > 0;
+  }
+
+  @Override
+  public int size() {
+    return mObject.length();
+  }
+}
diff --git a/packages/@unimodules/core/android/src/main/java/org/unimodules/core/arguments/ReadableArguments.java b/packages/@unimodules/core/android/src/main/java/org/unimodules/core/arguments/ReadableArguments.java
index 1c2b9ab5d076c..520b6194e2591 100644
--- a/packages/@unimodules/core/android/src/main/java/org/unimodules/core/arguments/ReadableArguments.java
+++ b/packages/@unimodules/core/android/src/main/java/org/unimodules/core/arguments/ReadableArguments.java
@@ -2,6 +2,8 @@
 
 import android.os.Bundle;
 
+import org.json.JSONObject;
+
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
@@ -55,6 +57,8 @@ default ReadableArguments getArguments(String key) {
     Object value = get(key);
     if (value instanceof Map) {
       return new MapArguments((Map) value);
+    } else if (value instanceof JSONObject) {
+      return new JSONObjectArguments((JSONObject) value);
     }
     return null;
   }
@@ -83,6 +87,9 @@ default Bundle toBundle() {
         bundle.putBundle(key, new MapArguments((Map) value).toBundle());
       } else if (value instanceof Bundle) {
         bundle.putBundle(key, (Bundle) value);
+      } else if (value instanceof JSONObject) {
+        JSONObject jsonObject = (JSONObject) value;
+        bundle.putBundle(key, new JSONObjectArguments(jsonObject).toBundle());
       } else {
         throw new UnsupportedOperationException("Could not put a value of " + value.getClass() + " to bundle.");
       }