From b65346d49c15a023e2a96c9b3e479683a896150d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Damir=20Jeli=C4=87?= <poljar@termina.org.uk>
Date: Sun, 14 Oct 2018 11:50:07 +0200
Subject: [PATCH] server: Add device deletion functionality.

---
 matrix/commands.py |  2 +-
 matrix/server.py   | 54 +++++++++++++++++++++++++++++++++++-----------
 2 files changed, 43 insertions(+), 13 deletions(-)

diff --git a/matrix/commands.py b/matrix/commands.py
index 8aefad9..322579a 100644
--- a/matrix/commands.py
+++ b/matrix/commands.py
@@ -635,7 +635,7 @@ def matrix_devices_command_cb(data, buffer, args):
             if not parsed_args.subcommand or parsed_args.subcommand == "list":
                 server.devices()
             elif parsed_args.subcommand == "delete":
-                server.delete_device(parsed_args.delete_device)
+                server.delete_device(parsed_args.device_id)
 
             return W.WEECHAT_RC_OK
 
diff --git a/matrix/server.py b/matrix/server.py
index f2d75ed..0bd96c7 100644
--- a/matrix/server.py
+++ b/matrix/server.py
@@ -37,6 +37,8 @@ from nio import (
     KeysQueryResponse,
     KeysClaimResponse,
     DevicesResponse,
+    DeleteDevicesAuthResponse,
+    DeleteDevicesResponse,
     TransportResponse,
     TransportType,
     RoomMessagesResponse,
@@ -44,6 +46,9 @@ from nio import (
     EncryptionError,
     GroupEncryptionError,
     OlmTrustError,
+    ErrorResponse,
+    SyncError,
+    LoginError,
 )
 
 from . import globals as G
@@ -237,6 +242,8 @@ class MatrixServer(object):
         self.send_buffer = b""                           # type: bytes
         self.device_check_timestamp = None               # type: Optional[int]
 
+        self.device_deletion_queue = dict()
+
         self.own_message_queue = dict()  # type: Dict[str, OwnMessage]
         self.encryption_queue = defaultdict(deque)  \
             # type: DefaultDict[str, Deque[EncrytpionQueueItem]]
@@ -587,8 +594,10 @@ class MatrixServer(object):
         _, request = self.client.devices()
         self.send_or_queue(request)
 
-    def delete_device(self, device_id):
-        # TODO implement this
+    def delete_device(self, device_id, auth=None):
+        uuid, request = self.client.delete_devices([device_id], auth)
+        self.device_deletion_queue[uuid] = device_id
+        self.send_or_queue(request)
         return
 
     def room_send_state(self, room_buffer, body, event_type):
@@ -906,15 +915,30 @@ class MatrixServer(object):
                 W.bar_item_update("buffer_modes")
                 W.bar_item_update("matrix_modes")
 
-    def handle_transport_response(self, response):
-        self.error(
-            ("Error with response of type type: {}, " "error code {}").format(
-                response.request_info.type, response.status_code
-            )
-        )
+    def handle_delete_device_auth(self, response):
+        device_id = self.device_deletion_queue.pop(response.uuid, None)
 
-        # TODO better error handling.
-        if response.request_info.type in (RequestType.sync, RequestType.login):
+        if not device_id:
+            return
+
+        for flow in response.flows:
+            if "m.login.password" in flow["stages"]:
+                session = response.session
+                auth = {
+                    "type": "m.login.password",
+                    "session": session,
+                    "user": self.client.user_id,
+                    "password": self.config.password
+                }
+                self.delete_device(device_id, auth)
+                return
+
+        self.error("No supported auth method for device deletion found.")
+
+    def handle_error_response(self, response):
+        self.error("Error: {}".format(str(response)))
+
+        if isinstance(response, (SyncError, LoginError)):
             self.disconnect()
 
     def handle_response(self, response):
@@ -930,8 +954,8 @@ class MatrixServer(object):
         self.lag_done = True
         W.bar_item_update("lag")
 
-        if isinstance(response, TransportResponse):
-            self.handle_transport_response(response)
+        if isinstance(response, ErrorResponse):
+            self.handle_error_response(response)
 
         elif isinstance(response, LoginResponse):
             self._handle_login(response)
@@ -948,6 +972,12 @@ class MatrixServer(object):
         elif isinstance(response, DevicesResponse):
             self.handle_devices_response(response)
 
+        elif isinstance(response, DeleteDevicesAuthResponse):
+            self.handle_delete_device_auth(response)
+
+        elif isinstance(response, DeleteDevicesResponse):
+            self.info("Device successfully deleted")
+
         elif isinstance(response, KeysQueryResponse):
             self.keys_queried = False