mtproto 0.0.9 → 0.0.11

data/ext/aes_ige/extconf.rb

@@ -16,4 +16,4 @@ abort 'AES_ige_encrypt function not found in OpenSSL library.' unless have_func(
 # Suppress deprecation warnings for OpenSSL 3.0
 $CFLAGS += ' -Wno-deprecated-declarations'
 
-create_makefile('aes_ige/aes_ige')
+create_makefile('aes_ige')

data/ext/factorization/extconf.rb

@@ -2,4 +2,4 @@
 
 require 'mkmf'
 
-create_makefile('factorization/factorization')
+create_makefile('factorization')

data/lib/mtproto/client/api/get_dialogs.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require_relative '../../tl/objects/get_dialogs'
+require_relative '../../tl/objects/dialogs'
+
+module MTProto
+  class Client
+    class API
+      def get_dialogs(offset_date: 0, offset_id: 0, offset_peer: nil, limit: 100, exclude_pinned: false, folder_id: nil)
+        rpc_call(
+          TL::GetDialogs.new(
+            offset_date: offset_date, offset_id: offset_id,
+            offset_peer: offset_peer, limit: limit,
+            exclude_pinned: exclude_pinned, folder_id: folder_id
+          ),
+          TL::Dialogs
+        ).body
+      end
+    end
+  end
+end

data/lib/mtproto/client/api/get_history.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative '../../tl/objects/get_history'
+require_relative '../../tl/objects/messages'
+
+module MTProto
+  class Client
+    class API
+      def get_history(peer:, offset_id: 0, offset_date: 0, add_offset: 0, limit: 100, max_id: 0, min_id: 0)
+        rpc_call(
+          TL::GetHistory.new(
+            peer: peer, offset_id: offset_id, offset_date: offset_date,
+            add_offset: add_offset, limit: limit, max_id: max_id, min_id: min_id
+          ),
+          TL::Messages
+        ).body
+      end
+    end
+  end
+end

data/lib/mtproto/client/api.rb

@@ -6,6 +6,8 @@ require_relative 'api/export_login_token'
 require_relative 'api/import_login_token'
 require_relative 'api/check_password'
 require_relative 'api/get_users'
+require_relative 'api/get_dialogs'
+require_relative 'api/get_history'
 require_relative 'api/get_updates_state'
 require_relative 'api/get_updates_difference'
 
@@ -22,6 +24,12 @@ module MTProto
         response = @client.rpc.call(request, response_class)
         response.wait!(@client.timeout)
         response
+      rescue RpcError => e
+        raise unless e.flood_wait?
+
+        warn "[MTProto] FLOOD_WAIT: retrying in #{e.flood_wait_seconds}s..."
+        sleep(e.flood_wait_seconds)
+        retry
       end
     end
   end

data/lib/mtproto/client/rpc.rb

@@ -6,9 +6,6 @@ require_relative 'rpc/response'
 module MTProto
   class Client
     class RPC
-      CONSTRUCTOR_RPC_RESULT = 0xf35c6d01
-      CONSTRUCTOR_BAD_SERVER_SALT = 0xedab447b
-
       attr_reader :pending_requests
 
       def initialize(client)
@@ -48,16 +45,16 @@ module MTProto
         result = response_body[12..]
 
         result_constructor = result[0, 4].unpack1('L<')
-        if result_constructor == Type::GzipPacked::CONSTRUCTOR
-          result = Type::GzipPacked.unpack(result)
+        if result_constructor == TL::Constructors::GZIP_PACKED
+          result = TL::GzipPacked.unpack(result)
           result_constructor = result[0, 4].unpack1('L<')
         end
 
         response = @pending_requests.delete(req_msg_id)
         return unless response
 
-        if result_constructor == Type::RpcError::CONSTRUCTOR
-          error = Type::RpcError.deserialize(result)
+        if result_constructor == TL::Constructors::RPC_ERROR
+          error = TL::RpcError.deserialize(result)
           response.signal_error(RpcError.new(error.error_code, error.error_message))
         else
           response.signal(result)

data/lib/mtproto/client.rb

@@ -8,7 +8,7 @@ require 'async/condition'
 require_relative 'transport/tcp_connection'
 require_relative 'transport/connection'
 require_relative 'transport/abridged_packet_codec'
-require_relative 'type/message'
+require_relative 'tl/objects/message'
 require_relative 'client/rpc'
 require_relative 'client/api'
 require_relative 'tl/objects/invoke_with_layer'
@@ -226,13 +226,13 @@ module MTProto
       constructor = response_body[0, 4].unpack1('L<')
 
       case constructor
-      when RPC::CONSTRUCTOR_BAD_SERVER_SALT
+      when TL::Constructors::BAD_SERVER_SALT
         rpc.handle_bad_server_salt(response_body, self)
-      when Type::NewSessionCreated::CONSTRUCTOR
+      when TL::Constructors::NEW_SESSION_CREATED
         handle_new_session(response_body)
-      when Type::MsgContainer::CONSTRUCTOR
+      when TL::Constructors::MSG_CONTAINER
         handle_container(response_body)
-      when RPC::CONSTRUCTOR_RPC_RESULT
+      when TL::Constructors::RPC_RESULT
         rpc.handle_rpc_result(response_body)
       else
         @on_update_callbacks.each { |cb| cb.call(constructor, response_body) }
@@ -240,12 +240,12 @@ module MTProto
     end
 
     def handle_new_session(response_body)
-      session_info = Type::NewSessionCreated.deserialize(response_body)
+      session_info = TL::NewSessionCreated.deserialize(response_body)
       @server_salt = session_info.server_salt
     end
 
     def handle_container(response_body)
-      container = Type::MsgContainer.deserialize(response_body)
+      container = TL::MsgContainer.deserialize(response_body)
 
       container.messages.each do |msg|
         process_message(msg[:body])

data/lib/mtproto/errors.rb

@@ -23,6 +23,14 @@ module MTProto
       @error_message = error_message
       super("RPC Error #{error_code}: #{error_message}")
     end
+
+    def flood_wait?
+      error_message&.start_with?('FLOOD_WAIT_')
+    end
+
+    def flood_wait_seconds
+      error_message&.match(/FLOOD_WAIT_(\d+)/)&.[](1)&.to_i
+    end
   end
 
   class UnexpectedConstructorError < Error