snapcat 0.0.1

data/lib/snapcat/crypt.rb

@@ -0,0 +1,35 @@
+module Snapcat
+  module Crypt
+    extend self
+
+    CIPHER = 'AES-128-ECB'
+    ENCRYPTION_KEY = 'M02cnQ51Ji97vwT4'
+
+    def decrypt(data)
+      cipher = OpenSSL::Cipher.new(CIPHER)
+      cipher.decrypt
+      cipher.key = ENCRYPTION_KEY
+      decrypted_data = ''
+
+      data.bytes.each_slice(16) do |slice|
+        decrypted_data += cipher.update(slice.map(&:chr).join)
+      end
+
+      decrypted_data += cipher.final
+    end
+
+    def encrypt(data)
+      cipher = OpenSSL::Cipher.new(CIPHER)
+      cipher.encrypt
+      cipher.key = ENCRYPTION_KEY
+      cipher.update(pkcs5_pad(data)) + cipher.final
+    end
+
+    private
+
+    def pkcs5_pad(data, blocksize = 16)
+      pad = blocksize - (data.length % blocksize)
+      "#{data}#{pad.chr * pad}"
+    end
+  end
+end

data/lib/snapcat/friend.rb

@@ -0,0 +1,57 @@
+module Snapcat
+  class Friend
+    ALLOWED_FIELD_CONVERSIONS = {
+      can_see_custom_stories: :can_see_custom_stories,
+      display: :display_name,
+      name: :username,
+      type: :type
+    }
+
+    attr_reader *ALLOWED_FIELD_CONVERSIONS.values
+
+    def initialize(data = {})
+      humanize_data(data)
+      @type = Type.new(@type)
+    end
+
+    private
+
+    def humanize_data(data)
+      ALLOWED_FIELD_CONVERSIONS.each do |api_field, human_field|
+        instance_variable_set(
+          "@#{human_field}",
+          data[human_field] || data[api_field]
+        )
+      end
+    end
+
+    class Type
+      CONFIRMED = 0
+      UNCONFIRMED = 1
+      BLOCKED = 2
+      DELETED = 3
+
+      attr_reader :code
+
+      def initialize(code)
+        @code = code
+      end
+
+      def blocked?
+        @code == BLOCKED
+      end
+
+      def confirmed?
+        @code == CONFIRMED
+      end
+
+      def deleted?
+        @code == DELETED
+      end
+
+      def unconfirmed?
+        @code == UNCONFIRMED
+      end
+    end
+  end
+end

data/lib/snapcat/media.rb

@@ -0,0 +1,81 @@
+module Snapcat
+  class Media
+    def initialize(data, type_code = nil)
+      @data = Crypt.decrypt(data)
+      @type = Type.new(code: type_code, data: @data)
+    end
+
+    def image?
+      @type.image?
+    end
+
+    def file_extension
+      if image?
+        'jpg'
+      elsif video?
+        'mp4'
+      end
+    end
+
+    def generate_id(username)
+      "#{username.upcase}~#{Timestamp.macro}"
+    end
+
+    def to_s
+      @data
+    end
+
+    def type_code
+      @type.code
+    end
+
+    def video?
+      @type.video?
+    end
+
+    class Type
+      IMAGE = 0
+      VIDEO = 1
+      VIDEO_NOAUDIO = 2
+      FRIEND_REQUEST = 3
+      FRIEND_REQUEST_IMAGE = 4
+      FRIEND_REQUEST_VIDEO = 5
+      FRIEND_REQUEST_VIDEO_NOAUDIO = 6
+
+      attr_reader :code
+
+      def initialize(options = {})
+        @code = code_from(options[:code], options[:data])
+      end
+
+      def image?
+        [IMAGE, FRIEND_REQUEST_IMAGE].include? @code
+      end
+
+      def video?
+        [
+          VIDEO, VIDEO_NOAUDIO, FRIEND_REQUEST_VIDEO, FRIEND_REQUEST_VIDEO_NOAUDIO
+        ].include? @code
+      end
+
+      private
+
+      def code_from(code, data)
+        if code
+          code
+        else
+          code_from_data(data)
+        end
+      end
+
+      def code_from_data(data)
+        case data.to_s[0..1]
+        when "\x00\x00".force_encoding('ASCII-8BIT')
+          VIDEO
+        when "\xFF\xD8".force_encoding('ASCII-8BIT')
+          IMAGE
+        end
+      end
+    end
+  end
+end

data/lib/snapcat/requestor.rb

@@ -0,0 +1,106 @@
+module Snapcat
+  class Requestor
+    include HTTMultiParty
+
+    APP_VERSION = '6.0.0'
+    SECRET = 'iEk21fuwZApXlz93750dmW22pw389dPwOk'
+    STATIC_TOKEN = 'm198sOkJEn37DjqZ32lpRu76xmw288xSQ9'
+    HASH_PATTERN = '0001110111101110001111010101111011010001001110011000110001000110'
+
+    base_uri 'https://feelinsonice-hrd.appspot.com/bq/'
+
+    def initialize(username)
+      @auth_token = STATIC_TOKEN
+      @username = username
+    end
+
+    def request(endpoint, data = {})
+      response = self.class.post(
+        "/#{endpoint}",
+        { body: merge_defaults_with(data) }
+      )
+
+      additional_fields = additional_fields_for(data)
+      result = Snapcat::Response.new(response, additional_fields)
+
+      auth_token_from(result, endpoint)
+      result
+    end
+
+    def request_media(snap_id)
+      response = self.class.post(
+        '/blob',
+        { body: merge_defaults_with({ id: snap_id, username: @username }) }
+      )
+
+      Response.new(response)
+    end
+
+    def request_with_username(endpoint, data = {})
+      request(endpoint, data.merge({ username: @username }))
+    end
+
+    def request_upload(data, type = nil)
+      encrypted_data = Crypt.encrypt(data)
+      media = Media.new(encrypted_data, type)
+      file_extension = media.file_extension
+
+      begin
+        file = Tempfile.new(['snap', ".#{file_extension}"])
+        file.write(encrypted_data)
+        file.rewind
+
+        return request_with_username(
+          'upload',
+          data: file,
+          media_id: media.generate_id(@username),
+          type: media.type_code
+        )
+      ensure
+        file.close
+        file.unlink
+      end
+    end
+
+    private
+
+    def additional_fields_for(data)
+      if data[:media_id]
+        { media_id: data[:media_id] }
+      else
+        {}
+      end
+    end
+
+    def auth_token_from(result, endpoint)
+      if endpoint == 'logout'
+        @auth_token = STATIC_TOKEN
+      else
+        @auth_token = result.auth_token || @auth_token
+      end
+    end
+
+    def built_token(auth_token, timestamp)
+      hash_a = Digest::SHA256.new << "#{SECRET}#{auth_token}"
+      hash_b = Digest::SHA256.new << "#{timestamp}#{SECRET}"
+
+      HASH_PATTERN.split(//).each_index.inject('') do |final_string, index|
+        if HASH_PATTERN[index] == '1'
+          final_string << hash_b.to_s[index].to_s
+        else
+          final_string << hash_a.to_s[index].to_s
+        end
+      end
+    end
+
+    def merge_defaults_with(data)
+      now = Timestamp.micro
+
+      data.merge!({
+        req_token: built_token(@auth_token, now),
+        timestamp: now,
+        version: APP_VERSION
+      })
+    end
+  end
+end

data/lib/snapcat/response.rb

@@ -0,0 +1,43 @@
+module Snapcat
+  class Response
+    attr_reader :code, :data, :http_success
+
+    def initialize(response, additional_fields = {})
+      @data = formatted_result(response).merge(additional_fields)
+      @code = response.code
+      @http_success = response.success?
+    end
+
+    def auth_token
+      @data[:auth_token]
+    end
+
+    def success?
+      if !@data[:logged].nil?
+        !!@data[:logged]
+      else
+        @http_success
+      end
+    end
+
+    private
+
+    def response_empty?(response)
+      response.body.to_s.empty?
+    end
+
+    def formatted_result(response)
+      if !response_empty?(response)
+        if response.content_type == 'application/octet-stream'
+          { media: Media.new(response.body) }
+        elsif response.content_type == 'application/json'
+          JSON.parse(response.body, symbolize_names: true)
+        else
+          {}
+        end
+      else
+        {}
+      end
+    end
+  end
+end

data/lib/snapcat/snap.rb

@@ -0,0 +1,80 @@
+module Snapcat
+  class Snap
+    ALLOWED_FIELD_CONVERSIONS = {
+      broadcast: :broadcast,
+      broadcast_action_text: :broadcast_action_text,
+      broadcast_hide_timer: :broadcast_hide_timer,
+      broadcast_url: :broadcast_url,
+      c: :screenshot_count,
+      c_id: :media_id,
+      id: :id,
+      m: :media_type,
+      rp: :recipient,
+      sn: :sender,
+      st: :status,
+      sts: :sent,
+      ts: :opened
+    }
+
+    attr_reader *ALLOWED_FIELD_CONVERSIONS.values
+
+    def initialize(data = {})
+      humanize_data(data)
+      @status = Status.new(@status)
+      @media_type = Media::Type.new(code: @media_type)
+    end
+
+    def received?
+      !sent?
+    end
+
+    def sent?
+      !!media_id
+    end
+
+    private
+
+    def humanize_data(data)
+      ALLOWED_FIELD_CONVERSIONS.each do |api_field, human_field|
+        instance_variable_set(
+          "@#{human_field}",
+          data[human_field] || data[api_field]
+        )
+      end
+    end
+
+    class Status
+      NONE = -1
+      SENT = 0
+      DELIVERED = 1
+      OPENED = 2
+      SCREENSHOT = 3
+
+      attr_reader :code
+
+      def initialize(code)
+        @code = code
+      end
+
+      def delivered?
+        @code == DELIVERED
+      end
+
+      def opened?
+        @code == OPENED
+      end
+
+      def sent?
+        @code == SENT
+      end
+
+      def screenshot?
+        @code == SCREENSHOT
+      end
+
+      def none?
+        @code == NONE
+      end
+    end
+  end
+end

data/lib/snapcat/snapcat_error.rb

@@ -0,0 +1,4 @@
+module Snapcat
+  class Error < StandardError
+  end
+end

data/lib/snapcat/timestamp.rb

@@ -0,0 +1,17 @@
+module Snapcat
+  module Timestamp
+    extend self
+
+    def float
+      Time.now.to_f
+    end
+
+    def macro
+      float.floor
+    end
+
+    def micro
+      (float * 1000).floor
+    end
+  end
+end

data/lib/snapcat/user.rb

@@ -0,0 +1,46 @@
+module Snapcat
+  class User
+    module Privacy
+      EVERYONE = 0
+      FRIENDS = 1
+    end
+
+    attr_reader :data, :friends, :snaps_sent, :snaps_received
+
+    def initialize
+      @friends = []
+      @snaps_sent = []
+      @snaps_received = []
+    end
+
+    def data=(data)
+      set_friends(data[:friends])
+      set_snaps(data[:snaps])
+      @data = data
+    end
+
+    private
+
+    def set_friends(friends)
+      @friends = []
+
+      friends.each do |friend_data|
+        @friends << Friend.new(friend_data)
+      end
+    end
+
+    def set_snaps(snaps)
+      @snaps_received = []
+      @snaps_sent = []
+
+      snaps.each do |snap_data|
+        snap = Snap.new(snap_data)
+        if snap.sent?
+          @snaps_sent << snap
+        else
+          @snaps_received << snap
+        end
+      end
+    end
+  end
+end