From 0a1008a127d4885d28725f9dbaafe05a13c344f1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Damir=20Jeli=C4=87?= <poljar@termina.org.uk>
Date: Sun, 28 Apr 2019 21:15:14 +0200
Subject: [PATCH] client: Send out outgoing to-device messages.

---
 pantalaimon/client.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/pantalaimon/client.py b/pantalaimon/client.py
index cd50e4d..140b4b1 100644
--- a/pantalaimon/client.py
+++ b/pantalaimon/client.py
@@ -57,6 +57,26 @@ class PanClient(AsyncClient):
         self.task = task
         return task
 
+    async def _to_device(self, message):
+        response = await self.to_device(message)
+        return message, response
+
+    async def send_to_device_messages(self):
+        if not self.outgoing_to_device_messages:
+            return
+
+        tasks = []
+
+        for message in self.outgoing_to_device_messages:
+            task = asyncio.create_task(self._to_device(message))
+            tasks.append(task)
+
+        responses = await asyncio.gather(*tasks)
+
+        for message, response in responses:
+            if response.transport_response.status == 200:
+                self.mark_to_device_message_as_sent(message)
+
     async def loop(self):
         self.loop_running = True
         self.loop_stopped.clear()
@@ -75,6 +95,8 @@ class PanClient(AsyncClient):
                     await asyncio.sleep(5)
                     continue
 
+                await self.send_to_device_messages()
+
                 if self.should_upload_keys:
                     await self.keys_upload()