Merge pull request #16 from setiawanhu/dev

Updating behavior change on MadelineProto API instance
setiawanhu · May 20, 2021 · 014d352 · 014d352
2 parents ba8f3cb + 2a303fe
commit 014d352
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/src/ClientMessages.php b/src/ClientMessages.php
@@ -16,7 +16,7 @@ class ClientMessages
      *
      * @param messages $messages
      */
-    public function __construct(messages $messages)
+    public function __construct($messages)
     {
         $this->messages = $messages;
     }

diff --git a/src/MadelineProto.php b/src/MadelineProto.php
@@ -46,11 +46,12 @@ public function completePhoneLogin(string $code): TelegramObject
     {
         $response = new TelegramObject($this->client->completePhoneLogin($code));
 
-        switch ($response->return_type) {
-            case Account::PASSWORD:
-                throw new NeedTwoFactorAuthException($response);
-            case Account::NEED_SIGN_UP:
-                throw new SignUpNeededException();
+        if ($response->return_type == Account::PASSWORD) {
+            throw new NeedTwoFactorAuthException($response);
+        }
+
+        if ($response->return_type == Account::NEED_SIGN_UP) {
+            throw new SignUpNeededException();
         }
 
         return $response;
@@ -145,7 +146,7 @@ public function isLoggedIn(): bool
      *
      * @return ClientMessages
      */
-    public function messages()
+    public function messages(): ClientMessages
     {
         return new ClientMessages($this->client->messages);
     }