From bc6db177a8742cbd41f9559289c6aa38c3551834 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?poljar=20=28Damir=20Jeli=C4=87=29?= <poljar@termina.org.uk>
Date: Mon, 12 Feb 2018 12:27:20 +0100
Subject: [PATCH] Add login message class.

---
 matrix/api.py | 28 ++++++++++++++++++++++++----
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/matrix/api.py b/matrix/api.py
index 3e307c1..040e289 100644
--- a/matrix/api.py
+++ b/matrix/api.py
@@ -329,6 +329,25 @@ class MatrixGenericMessage():
         self.request = request_func(**func_args)
 
 
+class MatrixLoginMessage(MatrixGenericMessage):
+    def __init__(self, client, user, password, device_name, device_id=None):
+        data = {
+            "user": user,
+            "password": password,
+            "device_name": device_name
+        }
+
+        if device_id:
+            data["device_id"] = device_id
+
+        MatrixGenericMessage.__init__(
+            self,
+            MessageType.LOGIN,
+            client.login,
+            data
+        )
+
+
 class MatrixSendMessage(MatrixGenericMessage):
     def __init__(self, client, room_id, formatted_message):
         self.room_id = room_id
@@ -389,9 +408,10 @@ def matrix_sync(server):
 
 def matrix_login(server):
     # type: (MatrixServer) -> None
-    message = MatrixMessage(
-        server,
-        OPTIONS,
-        MessageType.LOGIN
+    message = MatrixLoginMessage(
+        server.client,
+        server.user,
+        server.password,
+        server.device_name
     )
     server.send_or_queue(message)