Fix QR code logins

Author: danog

psalm-baseline.xml

@@ -7187,6 +7187,7 @@
       <code><![CDATA[$tl['layer'] ?? -1]]></code>
       <code><![CDATA[$tl['layer'] ?? -1]]></code>
       <code><![CDATA[$type]]></code>
+      <code><![CDATA[$type['connection']]]></code>
       <code><![CDATA[$type['subtype']]]></code>
       <code><![CDATA[$type['type']]]></code>
       <code><![CDATA[$type['type']]]></code>

src/Connection.php

@@ -228,6 +228,10 @@ public function getDatacenterID(): string
     {
         return $this->datacenterId;
     }
+    public function getDatacenter(): int
+    {
+        return $this->datacenter;
+    }
     /**
      * Get connection context.
      */

src/MTProto.php

@@ -1996,6 +1996,11 @@ public function populateConfig(array $config): void
     {
         $this->config = $config;
     }
+    /** @internal */
+    public function handleAuthorization(array $authorization, Connection $connection): void
+    {
+        EventLoop::queue($this->processAuthorization(...), $authorization, $connection->getDatacenter());
+    }
     /**
      * @internal
      */
@@ -2005,6 +2010,7 @@ public function getConstructorAfterDeserializationCallbacks(): array
         return [
             'help.support' => [$this->populateSupportUser(...)],
             'config' => [$this->populateConfig(...)],
+            'auth.authorization' => [$this->handleAuthorization(...)],
         ];
     }
     /**

src/MTProtoSession/ResponseHandler.php

@@ -236,8 +236,6 @@ private function handleResponse(MTProtoIncomingMessage $message, ?int $requestId
             }
             $request->reply(static fn () => RPCErrorException::make('Received bad_msg_notification: ' . MTProto::BAD_MSG_ERROR_CODES[$response['error_code']], $response['error_code'], $request->constructor));
             return;
-        } elseif ($constructor === 'auth.authorization') {
-            EventLoop::queue($this->API->processAuthorization(...), $response, $this->datacenter);
         }
 
         if (isset($response['_']) && !$this->shared->auth->isCdn) {

src/TL/TL.php

@@ -1111,7 +1111,7 @@ public function deserialize($stream, array $type)
         if ($type['encrypted']) {
             if (isset($this->afterConstructorDeserialization[$x['_']])) {
                 foreach ($this->afterConstructorDeserialization[$x['_']] as $callback) {
-                    $callback($x);
+                    $callback($x, $type['connection']);
                 }
             } elseif ($x['_'] === 'rpc_result'
                 && isset($type['connection']->new_outgoing[$x['req_msg_id']])

src/TL/TLCallback.php

@@ -20,6 +20,7 @@
 
 namespace danog\MadelineProto\TL;
 
+use danog\MadelineProto\Connection;
 use danog\MadelineProto\MTProto\MTProtoOutgoingMessage;
 
 /**
@@ -28,7 +29,7 @@
  *
  * @psalm-type TBeforeConstructorSerialization=Closure(array): mixed
  * @psalm-type TBeforeConstructorDeserialization=Closure(string): void
- * @psalm-type TAfterConstructorDeserialization=Closure(array): void
+ * @psalm-type TAfterConstructorDeserialization=Closure(array, Connection): void
  * @psalm-type TTypeMismatch=Closure(array): mixed
  *
  * @internal Interface for managing TL serialization callbacks.