AddListener/RemoveListener/ListenToAll/RemoveAllListeners (#23)

Author: 0xFirekeeper

Assets/Plugin/thirdweb.jslib

@@ -24,6 +24,30 @@ var plugin = {
         dynCall_viii(cb, idPtr, null, buffer);
       });
   },
+  ThirdwebInvokeListener: function (taskId, route, payload, action) {
+    // execute bridge call
+    window.bridge.invokeListener(
+      UTF8ToString(taskId),
+      UTF8ToString(route),
+      UTF8ToString(payload),
+      action,
+      (jsAction, jsTaskId, jsResult) => {
+        // Handle Task ID
+        var jsId = jsTaskId;
+        var jsIdSize = lengthBytesUTF8(jsId) + 1;
+        var jsIdBuffer = _malloc(jsIdSize);
+        stringToUTF8(jsId, jsIdBuffer, jsIdSize);
+
+        // Handle Result
+        var jsRes = jsResult;
+        var jsResSize = lengthBytesUTF8(jsRes) + 1;
+        var jsResBuffer = _malloc(jsResSize);
+        stringToUTF8(jsRes, jsResBuffer, jsResSize);
+
+        dynCall_vii(jsAction, jsIdBuffer, jsResBuffer);
+      }
+    );
+  },
   ThirdwebInitialize: function (chain, options) {
     window.bridge.initialize(UTF8ToString(chain), UTF8ToString(options));
   },

Assets/Thirdweb/Core/Scripts/Bridge.cs

@@ -26,7 +26,20 @@ public RequestMessageBody(string[] arguments)
             public string[] arguments;
         }
 
+        private struct GenericAction
+        {
+            public Type t;
+            public Delegate d;
+
+            public GenericAction(Type t, Delegate d)
+            {
+                this.t = t;
+                this.d = d;
+            }
+        }
+
         private static Dictionary<string, TaskCompletionSource<string>> taskMap = new Dictionary<string, TaskCompletionSource<string>>();
+        private static Dictionary<string, GenericAction> taskActionMap = new Dictionary<string, GenericAction>();
 
         [AOT.MonoPInvokeCallback(typeof(Action<string, string, string>))]
         private static void jsCallback(string taskId, string result, string error)
@@ -45,6 +58,16 @@ private static void jsCallback(string taskId, string result, string error)
             }
         }
 
+        [AOT.MonoPInvokeCallback(typeof(Action<string, string>))]
+        private static void jsAction(string taskId, string result)
+        {
+            if (taskActionMap.ContainsKey(taskId))
+            {
+                Type tempType = taskActionMap[taskId].t;
+                taskActionMap[taskId].d.DynamicInvoke(tempType == typeof(string) ? result : JsonConvert.DeserializeObject(result, tempType));
+            }
+        }
+
         public static void Initialize(string chainOrRPC, ThirdwebSDK.Options options)
         {
             if (Application.isEditor)
@@ -115,6 +138,21 @@ public static async Task<T> InvokeRoute<T>(string route, string[] body)
             return JsonConvert.DeserializeObject<Result<T>>(result).result;
         }
 
+        public static string InvokeListener<T>(string route, string[] body, Action<T> action)
+        {
+            if (Application.isEditor)
+            {
+                Debug.LogWarning("Interacting with the thirdweb SDK is not supported in the editor. Please build and run the app instead.");
+                return null;
+            }
+
+            string taskId = Guid.NewGuid().ToString();
+            taskActionMap[taskId] = new GenericAction(typeof(T), action);
+            var msg = Utils.ToJson(new RequestMessageBody(body));
+            ThirdwebInvokeListener(taskId, route, msg, jsAction);
+            return taskId;
+        }
+
         public static async Task FundWallet(FundWalletOptions payload)
         {
             if (Application.isEditor)
@@ -133,6 +171,8 @@ public static async Task FundWallet(FundWalletOptions payload)
         [DllImport("__Internal")]
         private static extern string ThirdwebInvoke(string taskId, string route, string payload, Action<string, string, string> cb);
         [DllImport("__Internal")]
+        private static extern string ThirdwebInvokeListener(string taskId, string route, string payload, Action<string, string> action);
+        [DllImport("__Internal")]
         private static extern string ThirdwebInitialize(string chainOrRPC, string options);
         [DllImport("__Internal")]
         private static extern string ThirdwebConnect(string taskId, string wallet, int chainId, Action<string, string, string> cb);

Assets/Thirdweb/Core/Scripts/Events.cs

@@ -1,5 +1,6 @@
 using System.Collections.Generic;
 using System.Threading.Tasks;
+using System;
 
 namespace Thirdweb
 {
@@ -33,5 +34,36 @@ public async Task<List<ContractEvent<object>>> GetAll(EventQueryOptions eventQue
         {
             return await Bridge.InvokeRoute<List<ContractEvent<object>>>(getRoute("getAllEvents"), Utils.ToJsonStringArray(eventQueryOptions));
         }
+
+        /// <summary>
+        /// Listens an event and executes callback
+        /// </summary>
+        /// <param name="eventName">Event name filter</param>
+        /// <param name="action">Callback action</param>
+        /// <typeparam name="T">Action return type</typeparam>
+        /// <returns>Task ID string</returns>
+        public string AddListener<T>(string eventName, Action<T> action)
+        {
+            return Bridge.InvokeListener<T>(getRoute("addEventListener"), Utils.ToJsonStringArray(eventName), action);
+        }
+
+        /// <summary>
+        /// Listens to all events and executes callback every time
+        /// </summary>
+        /// <param name="action">Callback action</param>
+        /// <typeparam name="T">Action return type</typeparam>
+        /// <returns>Task ID string</returns>
+        public string ListenToAll<T>(Action<T> action)
+        {
+            return Bridge.InvokeListener(getRoute("listenToAllEvents"), new string[] { }, action);
+        }
+
+        /// <summary>
+        /// Removes all event listeners
+        /// </summary>
+        public async Task<string> RemoveAllListeners()
+        {
+            return await Bridge.InvokeRoute<string>(getRoute("removeAllListeners"), new string[] { });
+        }
     }
 }