AndroidJavaObjectExtensions.cs

/*
 * Modified MIT License
 *
 * Copyright 2023 OneSignal
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * 1. The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * 2. All copies of substantial portions of the Software may only be used in connection
 * with services provided by OneSignal.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

using System.Collections.Generic;
using UnityEngine;

// ReSharper disable InconsistentNaming
namespace OneSignalSDK.Android.Utilities
{
    /// <summary>
    /// Conversion methods for common Java types wrapped by <see cref="AndroidJavaObject"/>
    /// </summary>
    internal static class AndroidJavaObjectExtensions
    {
        /// <summary>
        /// Converts from a Java class which implements a toJSONObject method which returns a JSONObject representation
        /// of that object to a Serializable class in Unity
        /// </summary>
        public static TModel ToSerializable<TModel>(this AndroidJavaObject source)
        {
            if (source == default)
                return default;

            var json = source.Call<AndroidJavaObject>("toJSONObject");
            var jsonStr = json.Call<string>("toString");
            var serialized = JsonUtility.FromJson<TModel>(jsonStr);

            return serialized;
        }

        /*
         * JSONObject
         */

        /// <summary>
        /// Converts from a Java org.json.JSONObject to a <see cref="Dictionary{TKey,TValue}"/>
        /// </summary>
        public static Dictionary<string, object> JSONObjectToDictionary(
            this AndroidJavaObject source
        ) =>
            source != null
                ? Json.Deserialize(source.Call<string>("toString")) as Dictionary<string, object>
                : null;

        /// <summary>
        /// Converts from a <see cref="Dictionary{TKey,TValue}"/> to a Java org.json.JSONObject
        /// </summary>
        public static AndroidJavaObject ToJSONObject<TKey, TValue>(
            this Dictionary<TKey, TValue> source
        ) => new AndroidJavaObject("org.json.JSONObject", Json.Serialize(source));

        /*
         * Map
         */

        /// <summary>
        /// Converts from a Java java.util.Map to a <see cref="Dictionary{TKey,TValue}"/>
        /// </summary>
        public static Dictionary<string, string> MapToDictionary(this AndroidJavaObject source)
        {
            if (source == null)
                return null;

            var entries = source.Call<AndroidJavaObject>("entrySet");
            var iter = entries.Call<AndroidJavaObject>("iterator");

            var ret = new Dictionary<string, string>();

            while (iter.Call<bool>("hasNext"))
            {
                var entry = iter.Call<AndroidJavaObject>("next");
                var key = entry.Call<string>("getKey");
                var valueJO = entry.Call<AndroidJavaObject>("getValue");

                ret[key] = valueJO.Call<string>("toString");
            }

            return ret;
        }

        /// <summary>
        /// Converts from a <see cref="Dictionary{TKey,TValue}"/> to a Java java.util.Map
        /// </summary>
        public static AndroidJavaObject ToMap(this Dictionary<string, string> source)
        {
            var map = new AndroidJavaObject("java.util.HashMap");
            var put = AndroidJNIHelper.GetMethodID(
                map.GetRawClass(),
                "put",
                "(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;"
            );

            var entryArgs = new object[2];
            foreach (var kv in source)
            {
                using (var key = new AndroidJavaObject("java.lang.String", kv.Key))
                using (var value = new AndroidJavaObject("java.lang.String", kv.Value))
                {
                    entryArgs[0] = key;
                    entryArgs[1] = value;
                    AndroidJNI.CallObjectMethod(
                        map.GetRawObject(),
                        put,
                        AndroidJNIHelper.CreateJNIArgArray(entryArgs)
                    );
                }
            }

            return map;
        }

        public static AndroidJavaObject ToMap(this Dictionary<string, object> source)
        {
            if (source == null)
                return null;

            var map = new AndroidJavaObject("java.util.HashMap");
            var put = AndroidJNIHelper.GetMethodID(
                map.GetRawClass(),
                "put",
                "(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;"
            );

            var entryArgs = new object[2];
            foreach (var kv in source)
            {
                using (var key = new AndroidJavaObject("java.lang.String", kv.Key))
                {
                    AndroidJavaObject value = null;

                    // Preserve original data types
                    switch (kv.Value)
                    {
                        case string stringValue:
                            value = new AndroidJavaObject("java.lang.String", stringValue);
                            break;
                        case int intValue:
                            value = new AndroidJavaObject("java.lang.Integer", intValue);
                            break;
                        case long longValue:
                            value = new AndroidJavaObject("java.lang.Long", longValue);
                            break;
                        case float floatValue:
                            value = new AndroidJavaObject("java.lang.Float", floatValue);
                            break;
                        case double doubleValue:
                            value = new AndroidJavaObject("java.lang.Double", doubleValue);
                            break;
                        case bool boolValue:
                            value = new AndroidJavaObject("java.lang.Boolean", boolValue);
                            break;
                        // Check IDictionary before IList: some dict-like types
                        // (e.g. Newtonsoft's JObject) also implement IList.
                        case System.Collections.IDictionary dictValue:
                            value = new AndroidJavaObject(
                                "org.json.JSONObject",
                                Json.Serialize(dictValue)
                            );
                            break;
                        case System.Collections.IList listValue:
                            value = new AndroidJavaObject(
                                "org.json.JSONArray",
                                Json.Serialize(listValue)
                            );
                            break;
                        case null:
                            value = null;
                            break;
                        default:
                            // Fallback to string representation for unsupported types
                            value = new AndroidJavaObject("java.lang.String", kv.Value.ToString());
                            break;
                    }

                    try
                    {
                        entryArgs[0] = key;
                        entryArgs[1] = value;
                        AndroidJNI.CallObjectMethod(
                            map.GetRawObject(),
                            put,
                            AndroidJNIHelper.CreateJNIArgArray(entryArgs)
                        );
                    }
                    finally
                    {
                        value?.Dispose();
                    }
                }
            }

            return map;
        }

        public static AndroidJavaObject ToArrayList(this string[] keys)
        {
            AndroidJavaObject arrayList = new AndroidJavaObject("java.util.ArrayList");

            foreach (string key in keys)
            {
                arrayList.Call<bool>("add", key);
            }

            return arrayList;
        }
    }
}