Code cleaning

setiawanhu · May 20, 2021 · 2a303fe · 2a303fe
1 parent 3f89407
commit 2a303fe
Showing 1 changed file with 7 additions and 6 deletions.
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);
     }