From b8c632e6a9d670b72e8f51e6a4bfde74bb320930 Mon Sep 17 00:00:00 2001
From: Stefan Agner <stefan@agner.ch>
Date: Thu, 28 Mar 2024 00:18:16 +0100
Subject: [PATCH] WIP [Python]: Implement asyncio variant of CallAsync

Call Matter SDK in a asyncio friendly way.

TODO: Locking?
---
 src/controller/python/chip/ChipStack.py | 38 +++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/src/controller/python/chip/ChipStack.py b/src/controller/python/chip/ChipStack.py
index 6c0eb4dc1409ea..0a5cb808238965 100644
--- a/src/controller/python/chip/ChipStack.py
+++ b/src/controller/python/chip/ChipStack.py
@@ -26,6 +26,7 @@
 
 from __future__ import absolute_import, print_function
 
+import asyncio
 import builtins
 import logging
 import os
@@ -367,6 +368,43 @@ def Call(self, callFunct, timeoutMs: int = None):
             return self.callbackRes
         return res
 
+    async def CallAsync(self, callFunct, timeoutMs: int = None):
+        class AsyncSimpleCallableHandle:
+            def __init__(self, callback, loop, future):
+                self._callback = callback
+                self._loop = loop
+                self._future = future
+                self._result = None
+                self._exception = None
+
+            def _done(self):
+                if self._exception:
+                    self._future.set_exception(self._exception)
+                else:
+                    self._future.set_result(self._result)
+
+            def __call__(self):
+                try:
+                    self._result = self._callback()
+                except Exception as ex:
+                    self._exception = ex
+                self._loop.call_soon_threadsafe(self._done)
+                pythonapi.Py_DecRef(py_object(self))
+
+        loop = asyncio.get_event_loop()
+        future = loop.create_future()
+        callObj = AsyncSimpleCallableHandle(callFunct, loop, future)
+        pythonapi.Py_IncRef(py_object(callObj))
+
+        res = self._ChipStackLib.pychip_DeviceController_PostTaskOnChipThread(
+            self.cbHandleChipThreadRun, py_object(callObj))
+
+        if not res.is_success:
+            pythonapi.Py_DecRef(py_object(callObj))
+            raise res.to_exception()
+
+        return await asyncio.wait_for(future, timeoutMs / 1000 if timeoutMs else None)
+
     def CallAsyncWithCallback(self, callFunct):
         '''Run a Python function on CHIP stack, and wait for the application specific response.
         This function is a wrapper of PostTaskOnChipThread, which includes some handling of application specific logics.