mtproto 0.0.9 → 0.0.11

data/lib/mtproto/tl/objects/get_history.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+module MTProto
+  module TL
+    class GetHistory
+      include Binary
+
+      CONSTRUCTOR = 0x4423e6c5
+
+      def initialize(peer:, offset_id: 0, offset_date: 0, add_offset: 0, limit: 100, max_id: 0, min_id: 0)
+        @peer = peer
+        @offset_id = offset_id
+        @offset_date = offset_date
+        @add_offset = add_offset
+        @limit = limit
+        @max_id = max_id
+        @min_id = min_id
+      end
+
+      def body
+        result = u32_b(CONSTRUCTOR)
+        result += serialize_input_peer
+        result += u32_b(@offset_id)
+        result += u32_b(@offset_date)
+        result += u32_b(@add_offset)
+        result += u32_b(@limit)
+        result += u32_b(@max_id)
+        result += u32_b(@min_id)
+        result += u64_b(0) # hash
+        result
+      end
+
+      private
+
+      def serialize_input_peer
+        case @peer[:type]
+        when :user
+          u32_b(Constructors::INPUT_PEER_USER) + u64_b(@peer[:id]) + u64_b(@peer[:access_hash] || 0)
+        when :chat
+          u32_b(Constructors::INPUT_PEER_CHAT) + u64_b(@peer[:id])
+        when :channel
+          u32_b(Constructors::INPUT_PEER_CHANNEL) + u64_b(@peer[:id]) + u64_b(@peer[:access_hash] || 0)
+        else
+          raise "Unknown peer type: #{@peer[:type]}"
+        end
+      end
+    end
+  end
+end

data/lib/mtproto/tl/objects/get_password.rb

@@ -5,10 +5,8 @@ module MTProto
     class GetPassword
       include Binary
 
-      CONSTRUCTOR = 0x548a30f5
-
       def body
-        u32_b(CONSTRUCTOR)
+        u32_b(Constructors::ACCOUNT_GET_PASSWORD)
       end
     end
   end

data/lib/mtproto/tl/objects/get_state.rb

@@ -5,10 +5,8 @@ module MTProto
     class GetState
       include Binary
 
-      CONSTRUCTOR = 0xedd4882a
-
       def body
-        u32_b(CONSTRUCTOR)
+        u32_b(Constructors::UPDATES_GET_STATE)
       end
     end
   end

data/lib/mtproto/tl/objects/get_users.rb

@@ -5,15 +5,11 @@ module MTProto
     class GetUsers
       include Binary
 
-      CONSTRUCTOR = 0x0d91a548
-      VECTOR_CONSTRUCTOR = 0x1cb5c415
-      INPUT_USER_SELF = 0xf7c1b13f
-
       def body
-        u32_b(CONSTRUCTOR) +
-          u32_b(VECTOR_CONSTRUCTOR) +
+        u32_b(Constructors::USERS_GET_USERS) +
+          u32_b(Constructors::VECTOR) +
           u32_b(1) +
-          u32_b(INPUT_USER_SELF)
+          u32_b(Constructors::INPUT_USER_SELF)
       end
     end
   end

data/lib/mtproto/{type → tl/objects}/gzip_packed.rb

@@ -4,10 +4,8 @@ require 'zlib'
 require 'stringio'
 
 module MTProto
-  module Type
+  module TL
     module GzipPacked
-      CONSTRUCTOR = 0x3072cfa1
-
       module_function
 
       def unpack(data)

data/lib/mtproto/tl/objects/help_config.rb

@@ -3,9 +3,6 @@
 module MTProto
   module TL
     class HelpConfig
-      CONSTRUCTOR = 0xcc1a241e
-      CONSTRUCTOR_ALT = 0x3072cfa1
-
       attr_reader :flags, :date, :expires, :test_mode, :this_dc, :dc_options
 
       def initialize(flags:, date:, expires:, test_mode:, this_dc:, dc_options:)
@@ -19,7 +16,9 @@ module MTProto
 
       def self.parse(data)
         constructor = data[0, 4].unpack1('L<')
-        raise UnexpectedConstructorError, constructor unless [CONSTRUCTOR, CONSTRUCTOR_ALT].include?(constructor)
+        unless [Constructors::CONFIG, Constructors::GZIP_PACKED].include?(constructor)
+          raise UnexpectedConstructorError, constructor
+        end
 
         offset = 4
         flags = data[offset, 4].unpack1('L<')
@@ -28,7 +27,7 @@ module MTProto
         offset += 4
         expires = data[offset, 4].unpack1('L<')
         offset += 4
-        test_mode = data[offset, 4].unpack1('L<') == 0x997275b5
+        test_mode = data[offset, 4].unpack1('L<') == Constructors::BOOL_TRUE
         offset += 4
         this_dc = data[offset, 4].unpack1('L<')
         offset += 4

data/lib/mtproto/tl/objects/import_login_token.rb

@@ -5,8 +5,6 @@ module MTProto
     class ImportLoginToken
       include Binary
 
-      CONSTRUCTOR = 0x95ac5ce4
-
       attr_reader :token
 
       def initialize(token:)
@@ -14,7 +12,7 @@ module MTProto
       end
 
       def body
-        u32_b(CONSTRUCTOR) + serialize_tl_bytes(@token)
+        u32_b(Constructors::AUTH_IMPORT_LOGIN_TOKEN) + serialize_tl_bytes(@token)
       end
 
       private

data/lib/mtproto/tl/objects/init_connection.rb

@@ -5,8 +5,6 @@ module MTProto
     class InitConnection
       include Binary
 
-      CONSTRUCTOR = 0xc1cd5ea9
-
       attr_reader :api_id, :device_model, :system_version, :app_version,
         :system_lang_code, :lang_pack, :lang_code, :query
 
@@ -25,7 +23,7 @@ module MTProto
       def body
         flags = 0
 
-        result = u32_b(CONSTRUCTOR)
+        result = u32_b(Constructors::INIT_CONNECTION)
         result += u32_b(flags)
         result += u32_b(@api_id)
         result += serialize_tl_string(@device_model)

data/lib/mtproto/tl/objects/invoke_with_layer.rb

@@ -5,8 +5,6 @@ module MTProto
     class InvokeWithLayer
       include Binary
 
-      CONSTRUCTOR = 0xda9b0d0d
-
       attr_reader :layer, :query
 
       def initialize(layer:, query:)
@@ -15,7 +13,7 @@ module MTProto
       end
 
       def body
-        u32_b(CONSTRUCTOR) + u32_b(@layer) + @query.body
+        u32_b(Constructors::INVOKE_WITH_LAYER) + u32_b(@layer) + @query.body
       end
     end
   end

data/lib/mtproto/tl/objects/login_token.rb

@@ -5,10 +5,6 @@ require_relative 'authorization'
 module MTProto
   module TL
     class LoginToken
-      CONSTRUCTOR_TOKEN = 0x629f1980
-      CONSTRUCTOR_MIGRATE_TO = 0x068e9916
-      CONSTRUCTOR_SUCCESS = 0x390d5c5e
-
       attr_reader :type, :expires, :token, :dc_id, :authorization
 
       def initialize(type:, expires: nil, token: nil, dc_id: nil, authorization: nil)
@@ -35,11 +31,11 @@ module MTProto
         constructor = data[0, 4].unpack1('L<')
 
         case constructor
-        when CONSTRUCTOR_TOKEN
+        when Constructors::AUTH_LOGIN_TOKEN
           parse_token(data)
-        when CONSTRUCTOR_MIGRATE_TO
+        when Constructors::AUTH_LOGIN_TOKEN_MIGRATE_TO
           parse_migrate_to(data)
-        when CONSTRUCTOR_SUCCESS
+        when Constructors::AUTH_LOGIN_TOKEN_SUCCESS
           new(type: :success, authorization: Authorization.parse(data[4..]))
         else
           raise UnexpectedConstructorError, constructor

data/lib/mtproto/{type → tl/objects}/message.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module MTProto
-  module Type
+  module TL
     class Message
       attr_reader :auth_key_id, :msg_id, :body
 

data/lib/mtproto/tl/objects/messages.rb

@@ -0,0 +1,162 @@
+# frozen_string_literal: true
+
+require_relative '../schema'
+
+module MTProto
+  module TL
+    class Messages
+      Message = Struct.new(:id, :from_id, :date, :text, keyword_init: true)
+
+      MESSAGES_MESSAGES = 0x8c718e87
+      MESSAGES_SLICE = 0x762b263d
+      MESSAGES_CHANNEL = 0xc776ba4e
+
+      MESSAGE = Constructors::MESSAGE
+      MESSAGE_SERVICE = 0x7a800e0a
+      MESSAGE_EMPTY = 0x90a6ca84
+
+      VALID_CONSTRUCTORS = [MESSAGES_MESSAGES, MESSAGES_SLICE, MESSAGES_CHANNEL].freeze
+
+      attr_reader :messages, :count
+
+      def initialize(messages:, count: nil)
+        @messages = messages
+        @count = count
+      end
+
+      def slice?
+        !@count.nil?
+      end
+
+      def self.schema
+        @schema ||= Schema.new(
+          File.expand_path('../../../../data/tl-schema.json', __dir__)
+        )
+      end
+
+      def self.parse(data)
+        constructor = data[0, 4].unpack1('L<')
+        raise UnexpectedConstructorError, constructor unless VALID_CONSTRUCTORS.include?(constructor)
+
+        offset = 4
+        count = nil
+
+        case constructor
+        when MESSAGES_MESSAGES
+          # messages:Vector<Message> chats:Vector<Chat> users:Vector<User>
+        when MESSAGES_SLICE
+          # flags:# count:int next_rate:flags.0?int offset_id_offset:flags.2?int
+          # search_flood:flags.3?SearchPostsFlood messages:Vector<Message> ...
+          flags = data[offset, 4].unpack1('L<')
+          offset += 4
+          count = data[offset, 4].unpack1('L<')
+          offset += 4
+          offset += 4 if flags.anybits?(1 << 0) # next_rate
+          offset += 4 if flags.anybits?(1 << 2) # offset_id_offset
+          offset = schema.skip(data, offset) if flags.anybits?(1 << 3) # search_flood
+        when MESSAGES_CHANNEL
+          # flags:# pts:int count:int offset_id_offset:flags.2?int
+          # messages:Vector<Message> topics:Vector<ForumTopic> ...
+          flags = data[offset, 4].unpack1('L<')
+          offset += 4
+          offset += 4 # pts
+          count = data[offset, 4].unpack1('L<')
+          offset += 4
+          offset += 4 if flags.anybits?(1 << 2) # offset_id_offset
+        end
+
+        messages, offset = parse_messages_vector(data, offset)
+
+        # skip topics vector for channelMessages
+        schema.skip_vector(data, offset) { |d, o| schema.skip(d, o) } if constructor == MESSAGES_CHANNEL
+
+        # skip chats and users (we don't need them here)
+
+        new(messages: messages, count: count)
+      end
+
+      class << self
+        private
+
+        def parse_messages_vector(data, offset)
+          offset += 4 # vector constructor
+          count = data[offset, 4].unpack1('L<')
+          offset += 4
+
+          messages = []
+          count.times do
+            constructor = data[offset, 4].unpack1('L<')
+
+            if constructor == MESSAGE
+              msg, = parse_message(data, offset)
+              messages << msg if msg
+            end
+
+            offset = schema.skip(data, offset)
+          end
+          [messages, offset]
+        end
+
+        # message#9815cec8 flags:# flags2:# id:int
+        #   from_id:flags.8?Peer ... date:int message:string ...
+        def parse_message(data, offset)
+          offset += 4 # constructor
+          flags = data[offset, 4].unpack1('L<')
+          offset += 4
+          flags2 = data[offset, 4].unpack1('L<')
+          offset += 4
+          id = data[offset, 4].unpack1('l<')
+          offset += 4
+
+          from_id = nil
+          _, from_id, offset = parse_peer(data, offset) if flags.anybits?(1 << 8)
+          offset += 4 if flags.anybits?(1 << 29) # from_boosts_applied
+          _, _, offset = parse_peer(data, offset) # peer_id (skip)
+          _, _, offset = parse_peer(data, offset) if flags.anybits?(1 << 28) # saved_peer_id
+          offset = schema.skip(data, offset) if flags.anybits?(1 << 2) # fwd_from
+          offset += 8 if flags.anybits?(1 << 11) # via_bot_id
+          offset += 8 if flags2.anybits?(1 << 0) # via_business_bot_id
+          offset = schema.skip(data, offset) if flags.anybits?(1 << 3) # reply_to
+
+          date = data[offset, 4].unpack1('L<')
+          offset += 4
+
+          text, offset = read_tl_string(data, offset)
+
+          msg = Message.new(id: id, from_id: from_id, date: date, text: text)
+          [msg, offset]
+        end
+
+        def parse_peer(data, offset)
+          constructor = data[offset, 4].unpack1('L<')
+          offset += 4
+          case constructor
+          when Constructors::PEER_USER
+            [:user, data[offset, 8].unpack1('Q<'), offset + 8]
+          when Constructors::PEER_CHAT
+            [:chat, data[offset, 8].unpack1('Q<'), offset + 8]
+          when Constructors::PEER_CHANNEL
+            [:channel, data[offset, 8].unpack1('Q<'), offset + 8]
+          else
+            raise "Unknown peer constructor: 0x#{constructor.to_s(16)}"
+          end
+        end
+
+        def read_tl_string(data, offset)
+          first_byte = data.getbyte(offset)
+          if first_byte == 254
+            len = data.getbyte(offset + 1) |
+                  (data.getbyte(offset + 2) << 8) |
+                  (data.getbyte(offset + 3) << 16)
+            total = 4 + len
+          else
+            len = first_byte
+            total = 1 + len
+          end
+          padding = (4 - (total % 4)) % 4
+          [data[offset + (total - len), len].force_encoding('UTF-8'), offset + total + padding]
+        end
+      end
+    end
+  end
+end

data/lib/mtproto/{type → tl/objects}/msg_container.rb

@@ -1,10 +1,8 @@
 # frozen_string_literal: true
 
 module MTProto
-  module Type
+  module TL
     class MsgContainer
-      CONSTRUCTOR = 0x73f1f8dc
-
       attr_reader :messages
 
       def self.deserialize(data)

data/lib/mtproto/{type → tl/objects}/new_session_created.rb

@@ -1,10 +1,8 @@
 # frozen_string_literal: true
 
 module MTProto
-  module Type
+  module TL
     class NewSessionCreated
-      CONSTRUCTOR = 0x9ec20908
-
       attr_reader :first_msg_id, :unique_id, :server_salt
 
       def self.deserialize(data)

data/lib/mtproto/tl/objects/pq_inner_data.rb

@@ -5,9 +5,6 @@ module MTProto
     class PQInnerData
       include Binary
 
-      CONSTRUCTOR_DC = 0xa9f55f95
-      CONSTRUCTOR_TEMP_DC = 0x56fddf88
-
       attr_reader :pq, :p, :q, :nonce, :server_nonce, :new_nonce, :dc, :expires_in
 
       def initialize(pq:, p:, q:, nonce:, server_nonce:, new_nonce:, dc:, expires_in: nil)
@@ -22,7 +19,7 @@ module MTProto
       end
 
       def body
-        constructor = @expires_in ? CONSTRUCTOR_TEMP_DC : CONSTRUCTOR_DC
+        constructor = @expires_in ? Constructors::PQ_INNER_DATA_TEMP_DC : Constructors::PQ_INNER_DATA_DC
 
         result = u32_b(constructor)
         result += serialize_tl_bytes(integer_to_bytes(@pq))

data/lib/mtproto/tl/objects/req_dh_params.rb

@@ -5,8 +5,6 @@ module MTProto
     class ReqDHParams
       include Binary
 
-      CONSTRUCTOR = 0xd712e4be
-
       attr_reader :nonce, :server_nonce, :p, :q, :public_key_fingerprint, :encrypted_data
 
       def initialize(nonce:, server_nonce:, p:, q:, public_key_fingerprint:, encrypted_data:)
@@ -22,7 +20,7 @@ module MTProto
       end
 
       def body
-        result = u32_b(CONSTRUCTOR)
+        result = u32_b(Constructors::REQ_DH_PARAMS)
         result += @nonce.bytes
         result += @server_nonce.bytes
         result += serialize_tl_bytes(integer_to_bytes(@p))

data/lib/mtproto/tl/objects/req_pq_multi.rb

@@ -5,8 +5,6 @@ module MTProto
     class ReqPqMulti
       include Binary
 
-      CONSTRUCTOR = 0xbe7e8ef1
-
       attr_reader :nonce
 
       def initialize(nonce)
@@ -16,7 +14,7 @@ module MTProto
       end
 
       def body
-        u32_b(CONSTRUCTOR) + @nonce.bytes
+        u32_b(Constructors::REQ_PQ_MULTI) + @nonce.bytes
       end
     end
   end

data/lib/mtproto/tl/objects/res_pq.rb

@@ -5,14 +5,12 @@ module MTProto
     class ResPq
       extend Binary
 
-      CONSTRUCTOR = 0x05162463
-
       attr_reader :nonce, :server_nonce, :pq, :fingerprints
 
       def self.parse(message)
         data = message.body
         constructor = b_u32(data[0, 4])
-        raise "Unexpected constructor: 0x#{constructor.to_s(16)}" unless constructor == CONSTRUCTOR
+        raise "Unexpected constructor: 0x#{constructor.to_s(16)}" unless constructor == Constructors::RES_PQ
 
         offset = 4
 
@@ -38,7 +36,7 @@ module MTProto
 
         vector_constructor = b_u32(data[offset, 4])
         offset += 4
-        raise 'Expected vector constructor' unless vector_constructor == 0x1cb5c415
+        raise 'Expected vector constructor' unless vector_constructor == Constructors::VECTOR
 
         fingerprints_count = b_u32(data[offset, 4])
         offset += 4

data/lib/mtproto/{type → tl/objects}/rpc_error.rb

@@ -1,10 +1,8 @@
 # frozen_string_literal: true
 
 module MTProto
-  module Type
+  module TL
     class RpcError
-      CONSTRUCTOR = 0x2144ca19
-
       attr_reader :error_code, :error_message
 
       def self.deserialize(data)

data/lib/mtproto/tl/objects/send_code.rb

@@ -5,9 +5,6 @@ module MTProto
     class SendCode
       include Binary
 
-      CONSTRUCTOR = 0xa677244f
-      CODE_SETTINGS = 0xad253d78
-
       attr_reader :phone_number, :api_id, :api_hash
 
       def initialize(phone_number:, api_id:, api_hash:)
@@ -17,7 +14,7 @@ module MTProto
       end
 
       def body
-        u32_b(CONSTRUCTOR) +
+        u32_b(Constructors::AUTH_SEND_CODE) +
           serialize_tl_string(@phone_number) +
           u32_b(@api_id) +
           serialize_tl_string(@api_hash) +
@@ -27,7 +24,7 @@ module MTProto
       private
 
       def code_settings
-        u32_b(CODE_SETTINGS) + u32_b(0)
+        u32_b(Constructors::CODE_SETTINGS) + u32_b(0)
       end
 
       def serialize_tl_string(str)

data/lib/mtproto/tl/objects/sent_code.rb

@@ -3,8 +3,6 @@
 module MTProto
   module TL
     class SentCode
-      CONSTRUCTOR = 0x5e002502
-
       attr_reader :flags, :type, :phone_code_hash, :next_type, :timeout
 
       def initialize(flags:, type:, phone_code_hash:, next_type: nil, timeout: nil)
@@ -17,7 +15,7 @@ module MTProto
 
       def self.parse(data)
         constructor = data[0, 4].unpack1('L<')
-        raise UnexpectedConstructorError, constructor unless constructor == CONSTRUCTOR
+        raise UnexpectedConstructorError, constructor unless constructor == Constructors::AUTH_SENT_CODE
 
         offset = 4
         flags = data[offset, 4].unpack1('L<')
@@ -44,7 +42,9 @@ module MTProto
           offset += 4
 
           case constructor
-          when 0x3dbb5986, 0xc000bba2, 0x5353e5a7 # app, sms, call
+          when Constructors::AUTH_SENT_CODE_TYPE_APP,
+               Constructors::AUTH_SENT_CODE_TYPE_SMS,
+               Constructors::AUTH_SENT_CODE_TYPE_CALL
             length = data[offset, 4].unpack1('L<')
             offset += 4
             [{ length: length }, offset]

data/lib/mtproto/tl/objects/server_dh_inner_data.rb

@@ -3,13 +3,13 @@
 module MTProto
   module TL
     class ServerDHInnerData
-      CONSTRUCTOR = 0xb5890dba
-
       attr_reader :nonce, :server_nonce, :g, :dh_prime, :g_a, :server_time
 
       def self.parse(bytes)
         constructor = bytes[0, 4].unpack1('L<')
-        raise "Unexpected constructor: 0x#{constructor.to_s(16)}" unless constructor == CONSTRUCTOR
+        unless constructor == Constructors::SERVER_DH_INNER_DATA
+          raise "Unexpected constructor: 0x#{constructor.to_s(16)}"
+        end
 
         offset = 4
 

data/lib/mtproto/tl/objects/server_dh_params.rb

@@ -5,14 +5,14 @@ module MTProto
     class ServerDHParams
       extend Binary
 
-      CONSTRUCTOR = 0xd0e8075c
-
       attr_reader :nonce, :server_nonce, :encrypted_answer
 
       def self.parse(message)
         data = message.body
         constructor = b_u32(data[0, 4])
-        raise "Unexpected constructor: 0x#{constructor.to_s(16)}" unless constructor == CONSTRUCTOR
+        unless constructor == Constructors::SERVER_DH_PARAMS_OK
+          raise "Unexpected constructor: 0x#{constructor.to_s(16)}"
+        end
 
         offset = 4
 

data/lib/mtproto/tl/objects/set_client_dh_params.rb

@@ -5,8 +5,6 @@ module MTProto
     class SetClientDHParams
       include Binary
 
-      CONSTRUCTOR = 0xf5045f1f
-
       attr_reader :nonce, :server_nonce, :encrypted_data
 
       def initialize(nonce:, server_nonce:, encrypted_data:)
@@ -19,7 +17,7 @@ module MTProto
       end
 
       def body
-        result = u32_b(CONSTRUCTOR)
+        result = u32_b(Constructors::SET_CLIENT_DH_PARAMS)
         result += @nonce.bytes
         result += @server_nonce.bytes
         result += serialize_tl_bytes(@encrypted_data.bytes)

data/lib/mtproto/tl/objects/sign_in.rb

@@ -5,8 +5,6 @@ module MTProto
     class SignIn
       include Binary
 
-      CONSTRUCTOR = 0x8d52a951
-
       attr_reader :phone_number, :phone_code_hash, :phone_code
 
       def initialize(phone_number:, phone_code_hash:, phone_code:)
@@ -18,7 +16,7 @@ module MTProto
       def body
         flags = 1 << 0
 
-        u32_b(CONSTRUCTOR) +
+        u32_b(Constructors::AUTH_SIGN_IN) +
           u32_b(flags) +
           serialize_tl_string(@phone_number) +
           serialize_tl_string(@phone_code_hash) +

data/lib/mtproto/tl/objects/update.rb

@@ -3,9 +3,6 @@
 module MTProto
   module TL
     class Update
-      CONSTRUCTOR_NEW_MESSAGE = 0x1f2b0afd
-      MESSAGE_CONSTRUCTOR = 0x9815cec8
-
       attr_reader :type, :text
 
       def initialize(type, text: nil)
@@ -18,7 +15,7 @@ module MTProto
         type = type_name(constructor_id)
 
         case constructor_id
-        when CONSTRUCTOR_NEW_MESSAGE
+        when Constructors::UPDATE_NEW_MESSAGE
           text = extract_new_message_text(bytes)
           new(type, text: text)
         else
@@ -30,7 +27,7 @@ module MTProto
         private
 
         def type_name(constructor_id)
-          name = Constructors::NAMES.fetch(constructor_id)
+          name = ConstructorNames::NAMES.fetch(constructor_id)
           name.sub(/^update/, '')
               .gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
               .gsub(/([a-z\d])([A-Z])/, '\1_\2')
@@ -41,7 +38,7 @@ module MTProto
           offset = 4 # skip updateNewMessage constructor
 
           msg_constructor = bytes[offset, 4].unpack1('L<')
-          return nil unless msg_constructor == MESSAGE_CONSTRUCTOR
+          return nil unless msg_constructor == Constructors::MESSAGE
 
           offset += 4
           flags = bytes[offset, 4].unpack1('L<')