amplitude-api 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 69388981ec7465494e2f8894a0f8d324430d8280
-  data.tar.gz: 8a97fd8316f0581b18a0420ab5c6a455094626d2
+  metadata.gz: a052a8eba1f08e0c286851d27a6bf2ce7e93cabe
+  data.tar.gz: 420237d4d92990243c11fce24998ba314178258d
 SHA512:
-  metadata.gz: c17936fe48ef36da5ced4833c86a64f57f0df31e8d06909a0d60d55a2ecf2fe2a26b2692d675d6c015d36ed71d7d87ea14bd08ab0034384dbe035a8b81ae98b4
-  data.tar.gz: c4b6ed7af9745f5a75647080c08f9a7ee952977b03d305409f36d285d752da3fdf65830574e7b091deaa72df9242346060d34e403e14270e08119c9840f48324
+  metadata.gz: 19691aa9910e89f15bab32bf9ea73a3f051a8a50d6340cee1ab7e6e0d6fef1874c7cc23f80fb838773b32b6072479b9843813655875ed77d9fb2c0d2198a72cb
+  data.tar.gz: 36882567908c18fba0cd0b1a2f4a60b389bf84a7c2c9610b89bd7622674e79437687f7c52f25bd3841659412a0e761b794e5deaddc9e2ddb22f27025e8f66d95

data/lib/amplitude-api.rb

@@ -2,9 +2,11 @@ require 'json'
 require 'bundler/setup'
 require 'typhoeus'
 require 'amplitude-api/event'
+require 'amplitude-api/identification'
 
 class AmplitudeAPI
-  URI_STRING = "https://api.amplitude.com/httpapi"
+  TRACK_URI_STRING = "https://api.amplitude.com/httpapi"
+  IDENTIFY_URI_STRING = "https://api.amplitude.com/identify"
 
   USER_WITH_NO_ACCOUNT = "user who doesn't have an account"
 
@@ -13,6 +15,8 @@ class AmplitudeAPI
     #   @return [ String ] an Amplitude API Key
     attr_accessor :api_key
 
+    # ==== Event Tracking related methods
+
     # Send a single event immediately to the AmplitudeAPI
     #
     # @param [ String ] event_name a string that describes the event, e.g. "clicked on Home"
@@ -25,37 +29,81 @@ class AmplitudeAPI
       track(event)
     end
 
-    # @overload body(event)
+
+    # @overload track_body(event)
     #   @param [ AmplitudeAPI::Event ]
     #
-    # @overload body([events])
+    # @overload track_body([events])
     #   @param [ Array<AmplitudeAPI::Event> ]
     #
     # @return [ Hash ]
     #
     # Converts a series of AmplitudeAPI::Event objects into a body
     # suitable for the Amplitude API
-    def body(*events)
+    def track_body(*events)
       event_body = events.flatten.map do |event|
         event.to_hash
       end
-      post_body = {
+
+      {
         api_key: self.api_key,
         event: JSON.generate(event_body)
       }
     end
 
+
     # @overload track(event)
     #   @param [ AmplitudeAPI::Event ] Send a single event to the Amplitude API
     #
     # @overload track([events])
     #   @param [ Array<AmplitudeAPI::Event> ] Send an array of events in a single request to Amplitude
     #
-    # @return [ Typhoeus::Response ] 
+    # @return [ Typhoeus::Response ]
     #
     # Send one or more Events to the Amplitude API
     def track(*events)
-      Typhoeus.post(URI_STRING, body: body(events))
+      Typhoeus.post(TRACK_URI_STRING, body: track_body(events))
+    end
+
+    # ==== Identification related methods
+
+    def send_identify(user_id, user_properties = {})
+      identification = AmplitudeAPI::Identification.new(user_id: user_id, user_properties: user_properties)
+      identify(identification)
+    end
+
+    # @overload identify_body(identification)
+    #   @param [ AmplitudeAPI::Identification ]
+    #
+    # @overload identify_body([identifications])
+    #   @param [ Array<AmplitudeAPI::Identification> ]
+    #
+    # @return [ Hash ]
+    #
+    # Converts a series of AmplitudeAPI::Identification objects into a body
+    # suitable for the Amplitude Identify API
+    def identify_body(*identifications)
+      identification_body = identifications.flatten.map do |identification|
+        identification.to_hash
+      end
+
+      {
+        api_key: self.api_key,
+        identification: JSON.generate(identification_body)
+      }
+    end
+
+    # @overload identify(identification)
+    #   @param [ AmplitudeAPI::Identify ] Send a single identify to the Amplitude API
+    #
+    # @overload identify([identifications])
+    #   @param [ Array<AmplitudeAPI::Identify> ] Send an array of identifications in a single request to Amplitude
+    #
+    # @return [ Typhoeus::Response ]
+    #
+    # Send one or more Identifications to the Amplitude Identify API
+    def identify(*identifications)
+      Typhoeus.post(IDENTIFY_URI_STRING, body: identify_body(identifications))
     end
   end
 end

data/lib/amplitude-api/identification.rb

@@ -0,0 +1,49 @@
+class AmplitudeAPI
+  class Identification
+    # @!attribute [ rw ] user_id
+    #   @return [ String ] the user_id to be sent to Amplitude
+    attr_accessor :user_id
+    # @!attribute [ rw ] user_properties
+    #   @return [ String ] the user_properties to be attached to the Amplitude Identify
+    attr_accessor :user_properties
+
+    # Create a new Identification
+    #
+    # @param [ String ] user_id a user_id to associate with the identification
+    # @param [ Hash ] user_properties various properties to attach to the user identification
+    def initialize(user_id: "", user_properties: {})
+      self.user_id = user_id
+      self.user_properties = user_properties
+    end
+
+    def user_id=(value)
+      @user_id =
+        if value.respond_to?(:id)
+          value.id
+        else
+          value || AmplitudeAPI::USER_WITH_NO_ACCOUNT
+        end
+    end
+
+    # @return [ Hash ] A serialized Event
+    #
+    # Used for serialization and comparison
+    def to_hash
+      {
+        user_id: self.user_id,
+        user_properties: self.user_properties
+      }
+    end
+
+    # @return [ true, false ]
+    #
+    # Compares +to_hash+ for equality
+    def ==(other)
+      if other.respond_to?(:to_hash)
+        self.to_hash == other.to_hash
+      else
+        false
+      end
+    end
+  end
+end

data/lib/amplitude-api/version.rb

@@ -1,3 +1,3 @@
 class AmplitudeAPI
-  VERSION = '0.0.3'
+  VERSION = '0.0.4'
 end

data/spec/lib/amplitude-api/identification_spec.rb

@@ -0,0 +1,44 @@
+require 'spec_helper'
+
+describe AmplitudeAPI::Identification do
+  User = Struct.new(:id)
+
+  context "with a user object" do
+    describe "#body" do
+      it "populates with the user's id" do
+        identification = AmplitudeAPI::Identification.new(user_id: User.new(123))
+        expect(identification.to_hash[:user_id]).to eq(123)
+      end
+    end
+  end
+
+  context "with a user id" do
+    describe "#body" do
+      it "populates with the user's id" do
+        identification = AmplitudeAPI::Identification.new(user_id: 123)
+        expect(identification.to_hash[:user_id]).to eq(123)
+      end
+    end
+  end
+
+  context "without a user" do
+    describe "#body" do
+      it "populates with the unknown user" do
+        identification = AmplitudeAPI::Identification.new(user_id: nil)
+        expect(identification.to_hash[:user_id]).to eq(AmplitudeAPI::USER_WITH_NO_ACCOUNT)
+      end
+    end
+  end
+
+  describe '#body' do
+    it "includes the user id" do
+      identification = AmplitudeAPI::Identification.new(user_id: 123)
+      expect(identification.to_hash[:user_id]).to eq(123)
+    end
+
+    it "includes arbitrary user properties" do
+      identification = AmplitudeAPI::Identification.new(user_id: 123, user_properties: {first_name: 'John', last_name: 'Doe'})
+      expect(identification.to_hash[:user_properties]).to eq(first_name: 'John', last_name: 'Doe')
+    end
+  end
+end

data/spec/lib/amplitude_api_spec.rb

@@ -11,7 +11,7 @@ describe AmplitudeAPI do
         event = AmplitudeAPI::Event.new(user_id: 123, event_type: 'clicked on sign up')
         body = {api_key: AmplitudeAPI.api_key, event: JSON.generate([event.to_hash])}
 
-        expect(Typhoeus).to receive(:post).with(AmplitudeAPI::URI_STRING, body: body)
+        expect(Typhoeus).to receive(:post).with(AmplitudeAPI::TRACK_URI_STRING, body: body)
 
         AmplitudeAPI.track(event)
       end
@@ -23,13 +23,38 @@ describe AmplitudeAPI do
         event2 = AmplitudeAPI::Event.new(user_id: 456, event_type: 'liked a widget')
         body = {api_key: AmplitudeAPI.api_key, event: JSON.generate([event.to_hash, event2.to_hash])}
 
-        expect(Typhoeus).to receive(:post).with(AmplitudeAPI::URI_STRING, body: body)
+        expect(Typhoeus).to receive(:post).with(AmplitudeAPI::TRACK_URI_STRING, body: body)
 
         AmplitudeAPI.track([event, event2])
       end
     end
   end
 
+  describe ".identify" do
+    context "with a single identification" do
+      it "sends the identification to Amplitude" do
+        identification = AmplitudeAPI::Identification.new(user_id: 123, user_properties: {first_name: 'John', last_name: 'Doe'})
+        body = {api_key: AmplitudeAPI.api_key, identification: JSON.generate([identification.to_hash])}
+
+        expect(Typhoeus).to receive(:post).with(AmplitudeAPI::IDENTIFY_URI_STRING, body: body)
+
+        AmplitudeAPI.identify(identification)
+      end
+    end
+
+    context "with multiple identifications" do
+      it "sends all identifications in a single request" do
+        identification = AmplitudeAPI::Identification.new(user_id: 123, user_properties: {first_name: 'Julian', last_name: 'Ponce'})
+        identification2 = AmplitudeAPI::Identification.new(user_id: 456, user_properties: {first_name: 'John', last_name: 'Doe'})
+        body = {api_key: AmplitudeAPI.api_key, identification: JSON.generate([identification.to_hash, identification2.to_hash])}
+
+        expect(Typhoeus).to receive(:post).with(AmplitudeAPI::IDENTIFY_URI_STRING, body: body)
+
+        AmplitudeAPI.identify([identification, identification2])
+      end
+    end
+  end
+
   describe ".initializer " do
     it "initializes event without parameter" do
       event = AmplitudeAPI::Event.new()
@@ -77,16 +102,43 @@ describe AmplitudeAPI do
     end
   end
 
+  describe ".send_identify" do
+    it "sends an identify to AmplitudeAPI" do
+      identification = AmplitudeAPI::Identification.new(user_id: @user, user_properties: {first_name: 'John', last_name: 'Doe'})
+      expect(AmplitudeAPI).to receive(:identify).with(identification)
+
+      AmplitudeAPI.send_identify(@user, {first_name: 'John', last_name: 'Doe'})
+    end
+
+    context "the user is nil" do
+      it "sends an identify with the no account user" do
+        identification = AmplitudeAPI::Identification.new(user_id: nil, user_properties: {first_name: 'John', last_name: 'Doe'})
+        expect(AmplitudeAPI).to receive(:identify).with(identification)
+
+        AmplitudeAPI.send_identify(nil, {first_name: 'John', last_name: 'Doe'})
+      end
+    end
+
+    context "the user is a user_id" do
+      it "sends an identify to AmplitudeAPI" do
+        identification = AmplitudeAPI::Identification.new(user_id: 123, user_properties: {first_name: 'John', last_name: 'Doe'})
+        expect(AmplitudeAPI).to receive(:identify).with(identification)
+
+        AmplitudeAPI.send_identify(@user.id, {first_name: 'John', last_name: 'Doe'})
+      end
+    end
+  end
+
   describe "#body" do
     it "should add an api key" do
       event = AmplitudeAPI::Event.new(user_id: @user, event_type: "test_event", event_properties: {test_property: 1})
-      body = AmplitudeAPI.body(event)
+      body = AmplitudeAPI.track_body(event)
       expect(body[:api_key]).to eq('stub api key')
     end
 
     it "should create an event" do
       event = AmplitudeAPI::Event.new(user_id: 23, event_type: "test_event", event_properties: {foo: "bar"})
-      body = AmplitudeAPI.body(event)
+      body = AmplitudeAPI.track_body(event)
 
       expected = JSON.generate([{event_type: "test_event", user_id: 23, event_properties: {foo: "bar"}}])
       expect(body[:event]).to eq(expected)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: amplitude-api
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Alex Rakoczy
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-09-28 00:00:00.000000000 Z
+date: 2016-02-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -96,9 +96,11 @@ files:
 - amplitude-api.gemspec
 - lib/amplitude-api.rb
 - lib/amplitude-api/event.rb
+- lib/amplitude-api/identification.rb
 - lib/amplitude-api/version.rb
 - readme.md
 - spec/lib/amplitude-api/event_spec.rb
+- spec/lib/amplitude-api/identification_spec.rb
 - spec/lib/amplitude_api_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/toothrot/amplitude-api
@@ -121,11 +123,12 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.5
+rubygems_version: 2.4.5.1
 signing_key: 
 specification_version: 4
 summary: Send events to the Amplitude API
 test_files:
 - spec/lib/amplitude-api/event_spec.rb
+- spec/lib/amplitude-api/identification_spec.rb
 - spec/lib/amplitude_api_spec.rb
 - spec/spec_helper.rb