Adds player resources

Author: coding-chimp

Rakefile

@@ -2,8 +2,8 @@ require 'bundler/gem_tasks'
 require 'rake/testtask'
 
 Rake::TestTask.new(:test) do |t|
-  t.libs << "test"
-  t.libs << "lib"
+  t.libs << 'test'
+  t.libs << 'lib'
   t.test_files = FileList['test/**/*_test.rb']
 end
 

lib/onesignal.rb

@@ -2,9 +2,16 @@
 require 'onesignal/client'
 require 'onesignal/request'
 require 'onesignal/request_error'
+require 'onesignal/app_id_missing_error'
 require 'onesignal/models/base_model'
 require 'onesignal/models/app'
+require 'onesignal/models/player'
 require 'onesignal/resources/app_resource'
+require 'onesignal/resources/player_resource'
 
 module OneSignal
+  DEVICE_TYPES = [
+    'ios', 'android', 'amazon', 'mpns', 'chrome_app', 'chrome_website', 'wns',
+    'safari', 'firefox'
+  ]
 end

lib/onesignal/app_id_missing_error.rb

@@ -0,0 +1,7 @@
+module OneSignal
+  class AppIdMissingError < StandardError
+    def initialize(message = 'An app ID is needed for this resource')
+      super(message)
+    end
+  end
+end

lib/onesignal/client.rb

@@ -12,5 +12,17 @@ def initialize(auth_token:, app_id: nil)
     def apps
       @apps ||= AppResource.new(self)
     end
+
+    def players
+      ensure_app_id_presence
+
+      @players ||= PlayerResource.new(self)
+    end
+
+    private
+
+    def ensure_app_id_presence
+      raise AppIdMissingError if app_id.nil?
+    end
   end
 end

lib/onesignal/models/app.rb

@@ -1,11 +1,37 @@
+require 'date'
+
 module OneSignal
   class App < BaseModel
+    DATE_REGEX = /\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}.\d{3}Z/
+
     attr_accessor :apns_certificates, :apns_env, :basic_auth_key, :chrome_key,
       :chrome_web_default_notification_icon, :chrome_web_gcm_sender_id,
       :chrome_web_origin, :chrome_web_sub_domain, :created_at, :gcm_key, :id,
       :messagable_players, :name, :players, :safari_apns_cetificate,
       :safari_icon_128_128, :safari_icon_16_16, :safari_icon_256_256,
       :safari_icon_32_32, :safari_icon_64_64, :safari_push_id,
       :safari_site_origin, :site_name, :updated_at
+
+    def created_at=(time)
+      parse_time('created_at', time)
+    end
+
+    def updated_at=(time)
+      parse_time('updated_at', time)
+    end
+
+    private
+
+    def parse_time(attribute, value)
+      if string_date?(value)
+        value = Time.parse(value)
+      end
+
+      instance_variable_set("@#{attribute}", value)
+    end
+
+    def string_date?(date)
+      date.is_a?(String) && date.match(DATE_REGEX)
+    end
   end
 end

lib/onesignal/models/base_model.rb

@@ -1,21 +1,11 @@
-require 'date'
-
 module OneSignal
   class BaseModel
-    DATE_REGEX = /\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}.\d{3}Z/
-
     def self.to_proc
       method(:new).to_proc
     end
 
     def initialize(attributes = {})
-      attributes.each do |key, value|
-        if string_date?(value)
-          value = DateTime.parse(value)
-        end
-
-        instance_variable_set("@#{key}", value)
-      end
+      attributes.each { |key, value| send("#{key}=", value) }
     end
 
     def inspect
@@ -25,11 +15,5 @@ def inspect
 
       "<#{self.class.name} #{values}>"
     end
-
-    private
-
-    def string_date?(date)
-      date.is_a?(String) && date.match(DATE_REGEX)
-    end
   end
 end

lib/onesignal/models/player.rb

@@ -0,0 +1,24 @@
+module OneSignal
+  class Player < BaseModel
+    attr_accessor :ad_id, :amount_spent, :badge_count, :created_at,
+      :device_model, :device_os, :device_type, :game_version, :identifier,
+      :invalid_identifier, :language, :last_active, :session_count, :tags,
+      :timezone
+
+    def created_at=(time)
+      @created_at = Time.at(time)
+    end
+
+    def device_type=(type)
+      if type.is_a?(Integer)
+        type = OneSignal::DEVICE_TYPES[type]
+      end
+
+      @device_type = type
+    end
+
+    def last_active=(time)
+      @last_active = Time.at(time)
+    end
+  end
+end

lib/onesignal/request.rb

@@ -6,25 +6,38 @@ def initialize(client)
       @client = client
     end
 
-    def get(path)
-      request(path, Net::HTTP::Get)
+    def get(path, params = {})
+      uri = uri(path)
+
+      if @client.app_id
+        params[:app_id] = @client.app_id
+      end
+
+      uri.query = URI.encode_www_form(params)
+
+      request(uri, Net::HTTP::Get)
     end
 
     def post(path, params)
-      request(path, Net::HTTP::Post, params)
+      uri = uri(path)
+      request(uri(path), Net::HTTP::Post, params)
     end
 
     def put(path, params)
-      request(path, Net::HTTP::Put, params)
+      uri = uri(path)
+      request(uri, Net::HTTP::Put, params)
     end
 
     private
 
-    def request(path, klass, params = nil)
-      uri = URI.parse(OneSignal::Client::BASE_URL + path)
+    def uri(path)
+      URI.parse(OneSignal::Client::BASE_URL + path)
+    end
+
+    def request(uri, klass, params = nil)
       request = klass.new(uri.request_uri)
 
-      if @client.app_id
+      if params && @client.app_id
         params[:app_id] = @client.app_id
       end
 

lib/onesignal/request_error.rb

@@ -1,23 +1,22 @@
 module OneSignal
   class RequestError < StandardError
-    attr_reader :message
     attr_reader :http_status
     attr_reader :http_body
 
     def initialize(message, http_status, http_body)
-      @message = message
       @http_status = http_status
       @http_body = http_body
 
       if @http_status
-        @message += " - http status : #{@http_status}"
+        message += " - http status : #{@http_status}"
       end
 
       if @http_body
-        @message += " - http body : #{@http_body}"
+        message += " - http body : #{@http_body}"
       end
 
-      super(@message)
+      super(message)
     end
   end
 end
+

lib/onesignal/resources/player_resource.rb

@@ -0,0 +1,64 @@
+module OneSignal
+  class PlayerResource
+    def initialize(client)
+      @client = client
+    end
+
+    def all(params = {})
+      response = request.get('/players', params)
+      players = JSON.parse(response.body)['players']
+
+      players.map(&OneSignal::Player)
+    end
+
+    def find(id)
+      response = request.get("/players/#{id}")
+      player = JSON.parse(response.body)
+
+      OneSignal::Player.new(player)
+    end
+
+    def create(params)
+      response = request.post('/players', params)
+
+      JSON.parse(response.body)['id']
+    end
+
+    def update(id, params)
+      request.put("/players/#{id}", params)
+
+      true
+    end
+
+    def on_session(id, params)
+      request.post("/players/#{id}/on_session", params)
+
+      true
+    end
+
+    def on_purchase(id, params)
+      request.post("/players/#{id}/on_purchase", params)
+
+      true
+    end
+
+    def on_focus(id, params)
+      request.post("/players/#{id}/on_focus", params)
+
+      true
+    end
+
+    def csv_export(id)
+      params = { app_id: id }
+      response = request.post('/players/csv_export', params)
+
+      JSON.parse(response.body)['csv_file_url']
+    end
+
+    private
+
+    def request
+      @request ||= OneSignal::Request.new(@client)
+    end
+  end
+end