customerio 1.0.0 → 3.0.0

data/lib/customerio/base_client.rb

@@ -0,0 +1,87 @@
+require 'net/http'
+require 'multi_json'
+
+module Customerio
+  DEFAULT_TIMEOUT  = 10
+
+  class InvalidRequest < RuntimeError; end
+  class InvalidResponse < RuntimeError
+    attr_reader :code, :response
+
+    def initialize(code, body, response=nil)
+      @message = body
+      @code = code
+      @response = response
+
+      super(@message)
+    end
+  end
+
+  class BaseClient
+    def initialize(auth, options = {})
+      @auth = auth
+      @timeout = options[:timeout] || DEFAULT_TIMEOUT
+      @base_uri = options[:url]
+    end
+
+    def request(method, path, body = nil, headers = {})
+      execute(method, path, body, headers)
+    end
+
+    def request_and_verify_response(method, path, body = nil, headers = {})
+      verify_response(request(method, path, body, headers))
+    end
+
+    private
+
+    def execute(method, path, body = nil, headers = {})
+      uri = URI.join(@base_uri, path)
+
+      session = Net::HTTP.new(uri.host, uri.port)
+      session.use_ssl = (uri.scheme == 'https')
+      session.open_timeout = @timeout
+      session.read_timeout = @timeout
+
+      req = request_class(method).new(uri.path)
+
+      if @auth.has_key?(:site_id) && @auth.has_key?(:api_key)
+        req.initialize_http_header(headers)
+        req.basic_auth @auth[:site_id], @auth[:api_key]
+      else
+        headers['Authorization'] = "Bearer #{@auth[:app_key]}"
+        req.initialize_http_header(headers)
+      end
+
+      if !body.nil?
+        req.add_field('Content-Type', 'application/json')
+        req.body = MultiJson.dump(body)
+      end
+
+      session.start do |http|
+        http.request(req)
+      end
+    end
+
+    def request_class(method)
+      case method
+      when :post
+        Net::HTTP::Post
+      when :put
+        Net::HTTP::Put
+      when :delete
+        Net::HTTP::Delete
+      else
+        raise InvalidRequest.new("Invalid request method #{method.inspect}")
+      end
+    end
+
+    def verify_response(response)
+      case response
+      when Net::HTTPSuccess then
+        response
+      else
+        raise InvalidResponse.new(response.code, response.body, response)
+      end
+    end
+  end
+end

data/lib/customerio/client.rb

@@ -1,28 +1,15 @@
-require 'net/http'
-require 'multi_json'
+require "addressable/uri"
 
 module Customerio
-  DEFAULT_BASE_URI = 'https://track.customer.io'
-  DEFAULT_TIMEOUT  = 10
-
   class Client
-    class MissingIdAttributeError < RuntimeError; end
-    class InvalidRequest < RuntimeError; end
-    class InvalidResponse < RuntimeError
-      attr_reader :response
+    DEFAULT_TRACK_URL = 'https://track.customer.io'
 
-      def initialize(message, response)
-        @message = message
-        @response = response
-      end
-    end
+    class MissingIdAttributeError < RuntimeError; end
+    class ParamError < RuntimeError; end
 
-    def initialize(site_id, secret_key, options = {})
-      @username = site_id
-      @password = secret_key
-      @json = options.has_key?(:json) ? options[:json] : true
-      @base_uri = options[:base_uri] || DEFAULT_BASE_URI
-      @timeout = options[:timeout] || DEFAULT_TIMEOUT
+    def initialize(site_id, api_key, options = {})
+      options[:url] = DEFAULT_TRACK_URL if options[:url].nil? || options[:url].empty?
+      @client = Customerio::BaseClient.new({ site_id: site_id, api_key: api_key }, options)
     end
 
     def identify(attributes)
@@ -30,117 +17,113 @@ module Customerio
     end
 
     def delete(customer_id)
-      verify_response(request(:delete, customer_path(customer_id)))
+      raise ParamError.new("customer_id must be a non-empty string") if is_empty?(customer_id)
+      @client.request_and_verify_response(:delete, customer_path(customer_id))
     end
 
-    def track(*args)
-      attributes = extract_attributes(args)
+    def suppress(customer_id)
+      raise ParamError.new("customer_id must be a non-empty string") if is_empty?(customer_id)
+      @client.request_and_verify_response(:post, suppress_path(customer_id))
+    end
 
-      if args.length == 1
-        # Only passed in an event name, create an anonymous event
-        event_name = args.first
-        create_anonymous_event(event_name, attributes)
-      else
-        # Passed in a customer id and an event name.
-        # Track the event for the given customer
-        customer_id, event_name = args
+    def unsuppress(customer_id)
+      raise ParamError.new("customer_id must be a non-empty string") if is_empty?(customer_id)
+      @client.request_and_verify_response(:post, unsuppress_path(customer_id))
+    end
 
-        create_customer_event(customer_id, event_name, attributes)
-      end
+    def track(customer_id, event_name, attributes = {})
+      raise ParamError.new("customer_id must be a non-empty string") if is_empty?(customer_id)
+      raise ParamError.new("event_name must be a non-empty string") if is_empty?(event_name)
+
+      create_customer_event(customer_id, event_name, attributes)
     end
 
     def anonymous_track(event_name, attributes = {})
+      raise ParamError.new("event_name must be a non-empty string") if is_empty?(event_name)
       create_anonymous_event(event_name, attributes)
     end
 
-    private
-
-    def create_or_update(attributes = {})
-      attributes = Hash[attributes.map { |(k,v)| [ k.to_sym, v ] }]
+    def add_device(customer_id, device_id, platform, data={})
+      raise ParamError.new("customer_id must be a non-empty string") if is_empty?(customer_id)
+      raise ParamError.new("device_id must be a non-empty string") if is_empty?(device_id)
+      raise ParamError.new("platform must be a non-empty string") if is_empty?(platform)
 
-      raise MissingIdAttributeError.new("Must provide a customer id") unless attributes[:id]
+      if data.nil?
+        data = {}
+      end
 
-      url = customer_path(attributes[:id])
+      raise ParamError.new("data parameter must be a hash") unless data.is_a?(Hash)
 
-      verify_response(request(:put, url, attributes))
+      @client.request_and_verify_response(:put, device_path(customer_id), {
+        :device => data.update({
+          :id => device_id,
+          :platform => platform,
+        })
+      })
     end
 
-    def create_customer_event(customer_id, event_name, attributes = {})
-      create_event("#{customer_path(customer_id)}/events", event_name, attributes)
-    end
+    def delete_device(customer_id, device_id)
+      raise ParamError.new("customer_id must be a non-empty string") if is_empty?(customer_id)
+      raise ParamError.new("device_id must be a non-empty string") if is_empty?(device_id)
 
-    def create_anonymous_event(event_name, attributes = {})
-      create_event("/api/v1/events", event_name, attributes)
+      @client.request_and_verify_response(:delete, device_id_path(customer_id, device_id))
     end
 
-    def create_event(url, event_name, attributes = {})
-      body = { :name => event_name, :data => attributes }
-      body[:timestamp] = attributes[:timestamp] if valid_timestamp?(attributes[:timestamp])
-      verify_response(request(:post, url, body))
+    private
+
+    def escape(val)
+      # CGI.escape is recommended for escaping, however, it doesn't correctly escape spaces.
+      Addressable::URI.encode_component(val.to_s, Addressable::URI::CharacterClasses::UNRESERVED)
     end
 
-    def customer_path(id)
-      "/api/v1/customers/#{id}"
+    def device_path(customer_id)
+      "/api/v1/customers/#{escape(customer_id)}/devices"
     end
 
-    def valid_timestamp?(timestamp)
-      timestamp && timestamp.is_a?(Integer) && timestamp > 999999999 && timestamp < 100000000000
+    def device_id_path(customer_id, device_id)
+      "/api/v1/customers/#{escape(customer_id)}/devices/#{escape(device_id)}"
     end
 
+    def customer_path(id)
+      "/api/v1/customers/#{escape(id)}"
+    end
 
-    def verify_response(response)
-      if response.code.to_i >= 200 && response.code.to_i < 300
-        response
-      else
-        raise InvalidResponse.new("Customer.io API returned an invalid response: #{response.code}", response)
-      end
+    def suppress_path(customer_id)
+      "/api/v1/customers/#{escape(customer_id)}/suppress"
     end
 
-    def extract_attributes(args)
-      hash = args.last.is_a?(Hash) ? args.pop : {}
-      hash.inject({}){ |hash, (k,v)| hash[k.to_sym] = v; hash }
+    def unsuppress_path(customer_id)
+      "/api/v1/customers/#{escape(customer_id)}/unsuppress"
     end
 
-    def request(method, path, body = nil, headers = {})
-      uri = URI.join(@base_uri, path)
+    def create_or_update(attributes = {})
+      attributes = Hash[attributes.map { |(k,v)| [ k.to_sym, v ] }]
+      raise MissingIdAttributeError.new("Must provide a customer id") if is_empty?(attributes[:id])
 
-      session = Net::HTTP.new(uri.host, uri.port)
-      session.use_ssl = (uri.scheme == 'https')
-      session.open_timeout = @timeout
-      session.read_timeout = @timeout
+      url = customer_path(attributes[:id])
+      @client.request_and_verify_response(:put, url, attributes)
+    end
 
-      req = request_class(method).new(uri.path)
-      req.initialize_http_header(headers)
-      req.basic_auth @username, @password
+    def create_customer_event(customer_id, event_name, attributes = {})
+      create_event("#{customer_path(customer_id)}/events", event_name, attributes)
+    end
 
-      add_request_body(req, body) unless body.nil?
+    def create_anonymous_event(event_name, attributes = {})
+      create_event("/api/v1/events", event_name, attributes)
+    end
 
-      session.start do |http|
-        http.request(req)
-      end
+    def create_event(url, event_name, attributes = {})
+      body = { :name => event_name, :data => attributes }
+      body[:timestamp] = attributes[:timestamp] if valid_timestamp?(attributes[:timestamp])
+      @client.request_and_verify_response(:post, url, body)
     end
 
-    def request_class(method)
-      case method
-      when :post
-        Net::HTTP::Post
-      when :put
-        Net::HTTP::Put
-      when :delete
-        Net::HTTP::Delete
-      else
-        raise InvalidRequest.new("Invalid request method #{method.inspect}")
-      end
+    def valid_timestamp?(timestamp)
+      timestamp && timestamp.is_a?(Integer) && timestamp > 999999999 && timestamp < 100000000000
     end
 
-    def add_request_body(req, body)
-      if @json
-        req.add_field('Content-Type', 'application/json')
-        req.body = MultiJson.dump(body)
-      else
-        req.add_field('Content-Type', 'application/x-www-form-urlencoded')
-        req.body = ParamEncoder.to_params(body)
-      end
+    def is_empty?(val)
+      val.nil? || (val.is_a?(String) && val.strip == "")
     end
   end
 end

data/lib/customerio/requests/send_email_request.rb

@@ -0,0 +1,49 @@
+require 'base64'
+
+module Customerio
+  class SendEmailRequest
+    attr_reader :message
+
+    def initialize(opts)
+      @message = opts.delete_if { |field| invalid_field?(field) }
+      @message[:attachments] = {}
+      @message[:headers] = {}
+    end
+
+    def attach(name, data, encode: true)
+      raise "attachment #{name} already exists" if @message[:attachments].has_key?(name)
+      @message[:attachments][name] = encode ? Base64.strict_encode64(data) : data
+    end
+
+    private
+
+    REQUIRED_FIELDS = %i(to identifiers)
+
+    OPTIONAL_FIELDS = %i(
+      transactional_message_id
+      message_data
+      headers
+      preheader
+      from
+      reply_to
+      bcc
+      subject
+      body
+      plaintext_body
+      amp_body
+      fake_bcc
+      disable_message_retention
+      send_to_unsubscribed
+      tracked
+      queue_draft
+    )
+
+    def invalid_field?(field)
+      !REQUIRED_FIELDS.include?(field) && !OPTIONAL_FIELDS.include?(field)
+    end
+
+    def encode(data)
+      Base64.strict_encode64(data)
+    end
+  end
+end

data/lib/customerio/version.rb

@@ -1,3 +1,3 @@
 module Customerio
-  VERSION = "1.0.0"
+  VERSION = "3.0.0"
 end

data/spec/api_client_spec.rb

@@ -0,0 +1,130 @@
+require 'spec_helper'
+require 'multi_json'
+require 'base64'
+require 'tempfile'
+
+describe Customerio::APIClient do
+  let(:app_key) { "appkey" }
+
+  let(:client)   { Customerio::APIClient.new(app_key) }
+  let(:response) { double("Response", code: 200) }
+
+  def api_uri(path)
+    "https://api.customer.io#{path}"
+  end
+
+  def request_headers
+    { 'Authorization': "Bearer #{app_key}", 'Content-Type': 'application/json' }
+  end
+
+  def json(data)
+    MultiJson.dump(data)
+  end
+
+  describe "#send_email" do
+    it "sends a POST request to the /api/send/email path" do
+      req = Customerio::SendEmailRequest.new(
+        identifiers: {
+          id: 'c1',
+        },
+        transactional_message_id: 1,
+      )
+
+      stub_request(:post, api_uri('/v1/send/email'))
+        .with(headers: request_headers, body: req.message)
+        .to_return(status: 200, body: { delivery_id: 1 }.to_json, headers: {})
+
+      client.send_email(req).should eq({ "delivery_id" => 1 })
+    end
+
+    it "handles validation failures (400)" do
+      req = Customerio::SendEmailRequest.new(
+        identifiers: {
+          id: 'c1',
+        },
+        transactional_message_id: 1,
+      )
+
+      err_json = { meta: { error: "example error" } }.to_json
+
+      stub_request(:post, api_uri('/v1/send/email'))
+        .with(headers: request_headers, body: req.message)
+        .to_return(status: 400, body: err_json, headers: {})
+
+      lambda { client.send_email(req) }.should(
+        raise_error(Customerio::InvalidResponse) { |error|
+          error.message.should eq "example error"
+          error.code.should eq "400"
+        }
+      )
+    end
+
+    it "handles other failures (5xx)" do
+      req = Customerio::SendEmailRequest.new(
+        identifiers: {
+          id: 'c1',
+        },
+        transactional_message_id: 1,
+      )
+
+      stub_request(:post, api_uri('/v1/send/email'))
+        .with(headers: request_headers, body: req.message)
+        .to_return(status: 500, body: "Server unavailable", headers: {})
+
+      lambda { client.send_email(req) }.should(
+        raise_error(Customerio::InvalidResponse) { |error|
+          error.message.should eq "Server unavailable"
+          error.code.should eq "500"
+        }
+      )
+    end
+
+    it "allows attaching file content without encoding" do
+      content = 'sample content'
+
+      req = Customerio::SendEmailRequest.new(
+        customer_id: 'c1',
+        transactional_message_id: 1,
+      )
+
+      req.attach('test', content, encode: false)
+      req.message[:attachments]['test'].should eq content
+
+      stub_request(:post, api_uri('/v1/send/email'))
+        .with(headers: request_headers, body: req.message)
+        .to_return(status: 200, body: { delivery_id: 1 }.to_json, headers: {})
+
+      client.send_email(req)
+    end
+
+    it "allows attaching files with encoding (default)" do
+      content = 'sample content'
+
+      req = Customerio::SendEmailRequest.new(
+        customer_id: 'c1',
+        transactional_message_id: 1,
+      )
+
+      req.attach('test', content)
+      req.message[:attachments]['test'].should eq Base64.strict_encode64(content)
+
+      stub_request(:post, api_uri('/v1/send/email'))
+        .with(headers: request_headers, body: req.message)
+        .to_return(status: 200, body: { delivery_id: 1 }.to_json, headers: {})
+
+      client.send_email(req)
+    end
+
+    it "raises error when attaching the same key again" do
+      req = Customerio::SendEmailRequest.new(
+        customer_id: 'c1',
+        transactional_message_id: 1,
+      )
+
+      req.attach('test', 'test-content')
+
+      lambda { req.attach('test', '') }.should raise_error(/attachment test already exists/)
+      req.message[:attachments].should eq({ "test" => Base64.strict_encode64("test-content") })
+    end
+  end
+end