mtproto 0.0.5 → 0.0.7

data/lib/mtproto/tl/rpc/users/users.rb

@@ -0,0 +1,99 @@
+# frozen_string_literal: true
+
+require_relative '../../gzip_packed'
+require_relative '../../rpc_error'
+
+module MTProto
+  module TL
+    module RPC
+      module Users
+        class Users
+          VECTOR_CONSTRUCTOR = 0x1cb5c415
+          USER_CONSTRUCTOR = 0x020b1422
+
+          def self.parse(response)
+            offset = 0
+
+            constructor = response[offset, 4].unpack1('L<')
+            offset += 4
+
+            if constructor == TL::GzipPacked::CONSTRUCTOR
+              response = TL::GzipPacked.unpack(response)
+              constructor = response[0, 4].unpack1('L<')
+              offset = 4
+            end
+
+            if constructor == TL::RpcError::CONSTRUCTOR
+              error = TL::RpcError.deserialize(response)
+              raise MTProto::RpcError.new(error.error_code, error.error_message)
+            end
+
+            # Expect Vector<User>
+            raise "Expected Vector constructor, got 0x#{constructor.to_s(16)}" unless constructor == VECTOR_CONSTRUCTOR
+
+            count = response[offset, 4].unpack1('L<')
+            offset += 4
+
+            users = []
+            count.times do
+              user_constructor = response[offset, 4].unpack1('L<')
+              offset += 4
+
+              next unless user_constructor == USER_CONSTRUCTOR
+
+              # Parse User: flags:# flags2:# id:long access_hash:flags.0?long first_name:flags.1?string ...
+              flags = response[offset, 4].unpack1('L<')
+              offset += 4
+
+              flags2 = response[offset, 4].unpack1('L<')
+              offset += 4
+
+              id = response[offset, 8].unpack1('Q<')
+              offset += 8
+
+              access_hash = nil
+              if (flags & (1 << 0)) != 0
+                access_hash = response[offset, 8].unpack1('Q<')
+                offset += 8
+              end
+
+              first_name = nil
+              if (flags & (1 << 1)) != 0
+                len = response[offset].unpack1('C')
+                offset += 1
+                first_name = response[offset, len]
+                offset += len
+                padding = (4 - ((1 + len) % 4)) % 4
+                offset += padding
+              end
+
+              last_name = nil
+              if (flags & (1 << 2)) != 0
+                len = response[offset].unpack1('C')
+                offset += 1
+                last_name = response[offset, len]
+                offset += len
+                padding = (4 - ((1 + len) % 4)) % 4
+                offset += padding
+              end
+
+              users << {
+                id: id,
+                access_hash: access_hash,
+                first_name: first_name,
+                last_name: last_name,
+                flags: flags,
+                flags2: flags2
+              }
+
+              # Skip remaining fields - we don't need them for this test
+              break
+            end
+
+            users
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mtproto/tl/sent_code.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+module MTProto
+  module TL
+    class SentCode
+      CONSTRUCTOR = 0x5e002502
+
+      attr_reader :flags, :type, :phone_code_hash, :next_type, :timeout
+
+      def self.deserialize(data)
+        offset = 4
+
+        flags = data[offset, 4].unpack1('L<')
+        offset += 4
+
+        type, type_bytes_read = SentCodeType.deserialize_from(data[offset..])
+        offset += type_bytes_read
+
+        phone_code_hash_length = data[offset].ord
+        offset += 1
+
+        phone_code_hash = data[offset, phone_code_hash_length]
+        offset += phone_code_hash_length
+
+        padding = (4 - ((phone_code_hash_length + 1) % 4)) % 4
+        offset += padding
+
+        next_type = nil
+        if (flags & (1 << 1)) != 0
+          next_type, next_type_bytes_read = CodeType.deserialize_from(data[offset..])
+          offset += next_type_bytes_read
+        end
+
+        timeout = nil
+        if (flags & (1 << 2)) != 0
+          timeout = data[offset, 4].unpack1('L<')
+        end
+
+        new(
+          flags: flags,
+          type: type,
+          phone_code_hash: phone_code_hash,
+          next_type: next_type,
+          timeout: timeout
+        )
+      end
+
+      def initialize(flags:, type:, phone_code_hash:, next_type: nil, timeout: nil)
+        @flags = flags
+        @type = type
+        @phone_code_hash = phone_code_hash
+        @next_type = next_type
+        @timeout = timeout
+      end
+
+      def to_h
+        {
+          phone_code_hash: @phone_code_hash,
+          type: @type,
+          next_type: @next_type,
+          timeout: @timeout
+        }.compact
+      end
+    end
+
+    module SentCodeType
+      def self.deserialize_from(data)
+        constructor = data[0, 4].unpack1('L<')
+        offset = 4
+
+        case constructor
+        when 0x3dbb5986 # auth.sentCodeTypeApp
+          length = data[offset, 4].unpack1('L<')
+          offset += 4
+          [{ _: :sent_code_type_app, length: length }, offset]
+        when 0xc000bba2 # auth.sentCodeTypeSms
+          length = data[offset, 4].unpack1('L<')
+          offset += 4
+          [{ _: :sent_code_type_sms, length: length }, offset]
+        when 0x5353e5a7 # auth.sentCodeTypeCall
+          length = data[offset, 4].unpack1('L<')
+          offset += 4
+          [{ _: :sent_code_type_call, length: length }, offset]
+        when 0xab03c6d9 # auth.sentCodeTypeFlashCall
+          pattern_length = data[offset].ord
+          offset += 1
+          pattern = data[offset, pattern_length]
+          offset += pattern_length
+          padding = (4 - ((pattern_length + 1) % 4)) % 4
+          offset += padding
+          [{ _: :sent_code_type_flash_call, pattern: pattern }, offset]
+        when 0x82006484 # auth.sentCodeTypeMissedCall
+          prefix_length = data[offset].ord
+          offset += 1
+          prefix = data[offset, prefix_length]
+          offset += prefix_length
+          padding = (4 - ((prefix_length + 1) % 4)) % 4
+          offset += padding
+          length = data[offset, 4].unpack1('L<')
+          offset += 4
+          [{ _: :sent_code_type_missed_call, prefix: prefix, length: length }, offset]
+        else
+          raise "Unknown SentCodeType constructor: 0x#{constructor.to_s(16)}"
+        end
+      end
+    end
+
+    module CodeType
+      def self.deserialize_from(data)
+        constructor = data[0, 4].unpack1('L<')
+        offset = 4
+
+        case constructor
+        when 0x72a3158c # codeTypeSms
+          [{ _: :code_type_sms }, offset]
+        when 0x741cd3e3 # codeTypeCall
+          [{ _: :code_type_call }, offset]
+        when 0x226ccefb # codeTypeFlashCall
+          [{ _: :code_type_flash_call }, offset]
+        when 0xd61ad6ee # codeTypeMissedCall
+          [{ _: :code_type_missed_call }, offset]
+        else
+          raise "Unknown CodeType constructor: 0x#{constructor.to_s(16)}"
+        end
+      end
+    end
+  end
+end

data/lib/mtproto/transport/tcp_connection.rb

@@ -17,7 +17,7 @@ module MTProto
         @socket = nil
       end
 
-      def connect
+      def connect!
         return if connected?
 
         @socket = TCPSocket.new(@host, @port)

data/lib/mtproto/updates_poller.rb

@@ -0,0 +1,111 @@
+# frozen_string_literal: true
+
+require_relative 'rpc/get_updates_state'
+require_relative 'rpc/get_updates_difference'
+
+module MTProto
+  class UpdatesPoller
+    attr_reader :state, :running
+
+    def initialize(client, poll_interval: 1.0)
+      @client = client
+      @poll_interval = poll_interval
+      @state = nil
+      @running = false
+      @on_message_callbacks = []
+      @on_update_callbacks = []
+      @users_cache = {} # Cache user_id => access_hash
+    end
+
+    def on_message(&block)
+      @on_message_callbacks << block
+    end
+
+    def on_update(&block)
+      @on_update_callbacks << block
+    end
+
+    def start
+      raise 'Already running' if @running
+      raise 'Auth key not generated' unless @client.auth_key
+
+      @running = true
+
+      @state = RPC::GetUpdatesState.new(@client).call
+
+      poll_loop
+    end
+
+    def stop
+      @running = false
+    end
+
+    private
+
+    def poll_loop
+      while @running
+        begin
+          difference = RPC::GetUpdatesDifference.new(@client).call(
+            pts: @state[:pts],
+            date: @state[:date],
+            qts: @state[:qts]
+          )
+
+          process_difference(difference)
+
+          if difference[:state]
+            @state = @state.merge(difference[:state])
+          elsif difference[:type] == :too_long
+            @state = RPC::GetUpdatesState.new(@client).call
+          elsif difference[:date]
+            @state[:date] = difference[:date]
+          end
+
+          sleep @poll_interval
+        rescue MTProto::RpcError => e
+          puts "RPC Error during polling: #{e.message}"
+          sleep @poll_interval
+        rescue StandardError => e
+          puts "Error during polling: #{e.class} - #{e.message}"
+          puts e.backtrace.first(5)
+          sleep @poll_interval
+        end
+      end
+    end
+
+    def process_difference(difference)
+      case difference[:type]
+      when :empty
+        nil
+      when :difference, :slice, :short_message
+        if difference[:users]
+          difference[:users].each do |user|
+            if user[:access_hash]
+              @users_cache[user[:id]] = user[:access_hash]
+            end
+          end
+        end
+
+        if difference[:new_messages] && !difference[:new_messages].empty?
+          difference[:new_messages].each do |message|
+            process_message(message)
+          end
+        end
+
+        @on_update_callbacks.each { |callback| callback.call(difference) }
+      when :too_long
+        puts "Update gap too long (pts=#{difference[:pts]}), refreshing state..."
+      end
+    end
+
+    def process_message(message)
+      return unless message
+
+      if message[:user_id] && @users_cache[message[:user_id]]
+        message[:access_hash] = @users_cache[message[:user_id]]
+      end
+
+      @on_message_callbacks.each { |callback| callback.call(message) }
+    end
+  end
+end

data/lib/mtproto/version.rb

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

data/lib/mtproto.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require_relative 'mtproto/version'
+require_relative 'mtproto/errors'
 require_relative 'mtproto/transport/abridged_packet_codec'
 require_relative 'mtproto/transport/tcp_connection'
 require_relative 'mtproto/tl/message'
@@ -14,6 +15,8 @@ require_relative 'mtproto/tl/new_session_created'
 require_relative 'mtproto/tl/rpc_error'
 require_relative 'mtproto/tl/gzip_packed'
 require_relative 'mtproto/tl/config'
+require_relative 'mtproto/tl/code_settings'
+require_relative 'mtproto/tl/sent_code'
 require_relative 'mtproto/crypto/rsa_key'
 require_relative 'mtproto/crypto/factorization'
 require_relative 'mtproto/crypto/aes_ige'
@@ -26,6 +29,16 @@ require_relative 'mtproto/auth_key_generator'
 require_relative 'mtproto/session'
 require_relative 'mtproto/encrypted_message'
 require_relative 'mtproto/client'
+require_relative 'mtproto/rpc/ping'
+require_relative 'mtproto/rpc/get_config'
+require_relative 'mtproto/rpc/send_code'
+require_relative 'mtproto/rpc/sign_in'
+require_relative 'mtproto/rpc/get_users'
+require_relative 'mtproto/rpc/send_message'
+require_relative 'mtproto/rpc/get_updates_state'
+require_relative 'mtproto/rpc/get_updates_difference'
+require_relative 'mtproto/rpc/get_contacts'
+require_relative 'mtproto/updates_poller'
 
 module MTProto
 end

metadata

@@ -1,14 +1,28 @@
 --- !ruby/object:Gem::Specification
 name: mtproto
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.7
 platform: ruby
 authors:
 - Artem Levenkov
 bindir: bin
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: base64
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.2'
 description: A Ruby library for Telegram MTProto protocol with TCP abridged transport
   support
 email:
@@ -17,19 +31,17 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".env.example"
 - ".ruby-version"
 - Rakefile
-- ext/aes_ige/Makefile
-- ext/aes_ige/aes_ige.bundle
 - ext/aes_ige/aes_ige.c
 - ext/aes_ige/extconf.rb
-- ext/factorization/Makefile
 - ext/factorization/extconf.rb
-- ext/factorization/factorization.bundle
 - ext/factorization/factorization.c
 - lib/mtproto.rb
 - lib/mtproto/auth_key_generator.rb
 - lib/mtproto/client.rb
+- lib/mtproto/client/rpc.rb
 - lib/mtproto/crypto/aes_ige.rb
 - lib/mtproto/crypto/auth_key_helper.rb
 - lib/mtproto/crypto/dh_key_exchange.rb
@@ -39,20 +51,59 @@ files:
 - lib/mtproto/crypto/rsa_key.rb
 - lib/mtproto/crypto/rsa_pad.rb
 - lib/mtproto/encrypted_message.rb
+- lib/mtproto/errors.rb
+- lib/mtproto/message_id.rb
+- lib/mtproto/rpc/get_config.rb
+- lib/mtproto/rpc/get_contacts.rb
+- lib/mtproto/rpc/get_updates_difference.rb
+- lib/mtproto/rpc/get_updates_state.rb
+- lib/mtproto/rpc/get_users.rb
+- lib/mtproto/rpc/ping.rb
+- lib/mtproto/rpc/send_code.rb
+- lib/mtproto/rpc/send_message.rb
+- lib/mtproto/rpc/sign_in.rb
 - lib/mtproto/session.rb
+- lib/mtproto/tl/auth_key/dh_gen_response.rb
+- lib/mtproto/tl/auth_key/req_dh_params.rb
+- lib/mtproto/tl/auth_key/req_pq_multi.rb
+- lib/mtproto/tl/auth_key/res_pq.rb
+- lib/mtproto/tl/auth_key/server_dh_params.rb
+- lib/mtproto/tl/auth_key/set_client_dh_params.rb
 - lib/mtproto/tl/bad_msg_notification.rb
 - lib/mtproto/tl/client_dh_inner_data.rb
+- lib/mtproto/tl/code_settings.rb
 - lib/mtproto/tl/config.rb
 - lib/mtproto/tl/gzip_packed.rb
 - lib/mtproto/tl/message.rb
+- lib/mtproto/tl/method_builder.rb
 - lib/mtproto/tl/msg_container.rb
 - lib/mtproto/tl/new_session_created.rb
 - lib/mtproto/tl/p_q_inner_data.rb
+- lib/mtproto/tl/rpc/auth/authorization.rb
+- lib/mtproto/tl/rpc/auth/send_code.rb
+- lib/mtproto/tl/rpc/auth/sent_code.rb
+- lib/mtproto/tl/rpc/auth/sign_in.rb
+- lib/mtproto/tl/rpc/contacts/contacts.rb
+- lib/mtproto/tl/rpc/contacts/get_contacts.rb
+- lib/mtproto/tl/rpc/help/config.rb
+- lib/mtproto/tl/rpc/help/get_config.rb
+- lib/mtproto/tl/rpc/messages/send_message.rb
+- lib/mtproto/tl/rpc/messages/updates.rb
+- lib/mtproto/tl/rpc/ping.rb
+- lib/mtproto/tl/rpc/pong.rb
+- lib/mtproto/tl/rpc/updates/difference.rb
+- lib/mtproto/tl/rpc/updates/get_difference.rb
+- lib/mtproto/tl/rpc/updates/get_state.rb
+- lib/mtproto/tl/rpc/updates/state.rb
+- lib/mtproto/tl/rpc/users/get_users.rb
+- lib/mtproto/tl/rpc/users/users.rb
 - lib/mtproto/tl/rpc_error.rb
+- lib/mtproto/tl/sent_code.rb
 - lib/mtproto/tl/serializer.rb
 - lib/mtproto/tl/server_dh_inner_data.rb
 - lib/mtproto/transport/abridged_packet_codec.rb
 - lib/mtproto/transport/tcp_connection.rb
+- lib/mtproto/updates_poller.rb
 - lib/mtproto/version.rb
 - tmp/.keep
 homepage: https://github.com/alev-pro/mtproto-ruby