mtproto 0.0.20 → 0.0.21

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9c0529af4d0277b14c02e0a618a865e67234aaf956873d749579f0aae5d2b4c8
-  data.tar.gz: c519588e92e960402dfe515e9b202e8f9c83a05ca6a21952a05d339b031c8878
+  metadata.gz: 7b7ded80a63fcfa91991547984a7ffd73198a66216b920106accd235958fde49
+  data.tar.gz: ec61fe873b71bac923edf826c93a2c46c57ef4b602e92e17a2fc9891aa9ad052
 SHA512:
-  metadata.gz: 782793b348cdd81fd8bfb43ceedd459bef20c0c2033c5b36068a392db9b3cc796273c0debfe9ddf0a1c2965b6b7aebe4a591f8b6afc35fed180e116c40d24a34
-  data.tar.gz: e2221dcca9e4e15f92b88140ef5eeb523e838a1a88687497e87a0241b2c196f2a6c70b841fbfecca36fccbb2ac6f68fa2be912dbe6cdc94f2c32a7175cec9cc5
+  metadata.gz: cd4efbd5e89da27a91baa98abcc8b6dbffc113b3598ea954c2a20daa31510e5b9ad830cfd1c279ed04fca04a4550baf1c657d286fffd12d1ca452f86f27af473
+  data.tar.gz: 8ad12d3ff4820cbc0da0b3ccd00b81d4fc30e47c2412c88cd86c8fa15997026d2323255682410e2aa25f02bc4c34d0845aebee1f949d7cbc5fcbddc182e11ae9

data/lib/mtproto/bot_authorizer.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+require_relative 'dc'
+require_relative 'errors'
+
+module MTProto
+  # Logs a bot in with auth.importBotAuthorization, following a USER_MIGRATE_X
+  # redirect. A bot token is bound to a home DC; importing it from any other DC
+  # answers RPC 303 USER_MIGRATE_X naming that DC. On that reply we drop the seed
+  # connection and re-import against a fresh connection to the named DC. The DC
+  # switch lives here rather than in the DC-bound Client (mirroring
+  # FileDownloader); addresses come from MTProto::DC, never hardcoded per bot.
+  #
+  # Runs inside an Async reactor: the default connect starts a receiver.
+  class BotAuthorizer
+    def initialize(api_id:, api_hash:, public_key:, test_mode: false, timeout: 10, connect: nil)
+      @api_id = api_id
+      @api_hash = api_hash
+      @public_key = public_key
+      @test_mode = test_mode
+      @timeout = timeout
+      @connect = connect || method(:default_connect)
+    end
+
+    # Returns a connected, key-exchanged, receiving, initialised, authorized
+    # Client on the bot's home DC. seed_dc is only where the search starts; the
+    # real DC is discovered from the migrate redirect.
+    def authorize(bot_auth_token, seed_dc: 2)
+      dc = seed_dc
+      visited = []
+      loop do
+        visited << dc
+        client = @connect.call(dc)
+        begin
+          client.api.import_bot_authorization(bot_auth_token)
+          return client
+        rescue RpcError => e
+          raise if e.migrate_dc.nil?
+
+          client.disconnect!
+          dc = e.migrate_dc
+          raise Error, "bot authorization keeps migrating (revisited DC #{dc})" if visited.include?(dc)
+        end
+      end
+    end
+
+    private
+
+    def default_connect(dc)
+      addr = DC.address(dc, test: @test_mode)
+      client = Client.new(
+        api_id: @api_id, api_hash: @api_hash,
+        host: addr[:host], port: addr[:port],
+        public_key: @public_key, dc_number: dc, test_mode: @test_mode, timeout: @timeout
+      )
+      client.connect!
+      client.exchange_keys!
+      client.start_receiving!
+      client.init_connection!
+      client
+    end
+  end
+end

data/lib/mtproto/errors.rb

@@ -31,6 +31,12 @@ module MTProto
     def flood_wait_seconds
       error_message&.match(/FLOOD_WAIT_(\d+)/)&.[](1)&.to_i
     end
+
+    # A 303 *_MIGRATE_N reply means the account/bot lives on DC N: reconnect there
+    # and retry. Returns that DC number, or nil for a non-migrate error.
+    def migrate_dc
+      error_message&.match(/\A(?:USER|NETWORK|PHONE)_MIGRATE_(\d+)\z/)&.[](1)&.to_i
+    end
   end
 
   class UnexpectedConstructorError < Error

data/lib/mtproto/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module MTProto
-  VERSION = '0.0.20'
+  VERSION = '0.0.21'
 end

data/lib/mtproto.rb

@@ -39,6 +39,7 @@ require_relative 'mtproto/session'
 require_relative 'mtproto/encrypted_message'
 require_relative 'mtproto/client'
 require_relative 'mtproto/file_downloader'
+require_relative 'mtproto/bot_authorizer'
 
 module MTProto
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mtproto
 version: !ruby/object:Gem::Version
-  version: 0.0.20
+  version: 0.0.21
 platform: ruby
 authors:
 - Artem Levenkov
@@ -60,6 +60,7 @@ files:
 - lib/mtproto.rb
 - lib/mtproto/auth_key_generator.rb
 - lib/mtproto/binary.rb
+- lib/mtproto/bot_authorizer.rb
 - lib/mtproto/client.rb
 - lib/mtproto/client/api.rb
 - lib/mtproto/client/api/check_password.rb