intercom-rails 0.2.17 → 0.2.18

data/lib/intercom-rails/import.rb

@@ -33,11 +33,24 @@ module IntercomRails
       end
     end
 
+    # Check for ActiveRecord OR Mongoid
+    def active_record?(user_klass)
+      (defined?(ActiveRecord::Base) && (user_klass < ActiveRecord::Base))
+    end
+
+    def mongoid?(user_klass)
+      (defined?(Mongoid::Document) && (user_klass < Mongoid::Document))
+    end
+
+    def supported_orm?(user_klass)
+      active_record?(user_klass) || mongoid?(user_klass)
+    end
+
     def assert_runnable
       raise ImportError, "You can only import your users from your production environment" unless Rails.env.production?
       raise ImportError, "We couldn't find your user class, please set one in config/initializers/intercom_rails.rb" unless user_klass.present?
       info "Found user class: #{user_klass}"
-      raise ImportError, "Only ActiveRecord models are supported" unless defined?(ActiveRecord::Base) && (user_klass < ActiveRecord::Base)
+      raise ImportError, "Only ActiveRecord and Mongoid models are supported" unless supported_orm?(user_klass)
       raise ImportError, "Please add an Intercom API Key to config/initializers/intercom.rb" unless IntercomRails.config.api_key.present?
       info "Intercom API key found"
     end
@@ -67,22 +80,33 @@ module IntercomRails
     private
     MAX_BATCH_SIZE = 100
     def batches
-      user_klass.find_in_batches(:batch_size => MAX_BATCH_SIZE) do |users|
-        users_for_wire = users.map do |u| 
-          user_proxy = Proxy::User.new(u)
-          next unless user_proxy.valid?
-          
-          for_wire = user_proxy.to_hash
-          companies = Proxy::Company.companies_for_user(user_proxy)
-          for_wire.merge!(:companies => companies.map(&:to_hash)) if companies.present?
-
-          for_wire
-        end.compact
-
-        yield(prepare_batch(users_for_wire), users_for_wire.count) unless users_for_wire.count.zero?
+      if active_record?(user_klass)
+        user_klass.find_in_batches(:batch_size => MAX_BATCH_SIZE) do |users|
+          users_for_wire = map_to_users_for_wire(users)
+
+          yield(prepare_batch(users_for_wire), users_for_wire.count) unless users_for_wire.count.zero?
+        end
+      elsif mongoid?(user_klass)
+        0.step(user_klass.all.count, MAX_BATCH_SIZE) do |offset|
+          users_for_wire = map_to_users_for_wire(user_klass.limit(MAX_BATCH_SIZE).skip(offset))
+          yield(prepare_batch(users_for_wire), users_for_wire.count) unless users_for_wire.count.zero?
+        end
       end
     end
 
+    def map_to_users_for_wire(users)
+      users.map do |u|
+        user_proxy = Proxy::User.new(u)
+        next unless user_proxy.valid?
+
+        for_wire = user_proxy.to_hash
+        companies = Proxy::Company.companies_for_user(user_proxy)
+        for_wire.merge!(:companies => companies.map(&:to_hash)) if companies.present?
+
+        for_wire
+      end.compact
+    end
+
     def prepare_batch(batch)
       {:users => batch}.to_json
     end

data/lib/intercom-rails/version.rb

@@ -1,3 +1,3 @@
 module IntercomRails
-  VERSION = "0.2.17"
+  VERSION = "0.2.18"
 end

data/test/import_test_setup.rb

@@ -11,6 +11,28 @@ module ActiveRecord
   class Base; end
 end
 
+module Mongoid
+  module Document 
+    def self.included(klass)
+      klass.extend ClassMethods
+    end
+
+    module ClassMethods
+      def all
+        @_users ||= User.all
+      end
+
+      def limit(*args)
+        self
+      end
+
+      def skip(*args)
+        all
+      end
+    end
+  end
+end
+
 class User
 
   attr_reader :id, :email, :name

data/test/intercom-rails/import_unit_test.rb

@@ -25,14 +25,14 @@ class ImportUnitTest < MiniTest::Unit::TestCase
     assert_equal exception.message, "We couldn't find your user class, please set one in config/initializers/intercom_rails.rb"
   end
 
-  def test_run_with_non_activerecord_user_class
+  def test_run_with_unsupported_user_class
     IntercomRails::Import.any_instance.stub(:user_klass).and_return(Class)
 
     exception = assert_raises IntercomRails::ImportError do
       IntercomRails::Import.run
     end
     
-    assert_equal exception.message, "Only ActiveRecord models are supported"
+    assert_equal exception.message, "Only ActiveRecord and Mongoid models are supported"
   end
 
   def test_run_with_no_api_key
@@ -45,6 +45,19 @@ class ImportUnitTest < MiniTest::Unit::TestCase
     assert_equal exception.message, "Please add an Intercom API Key to config/initializers/intercom.rb"
   end
 
+  def test_mongoid
+    klass = Class.new
+    klass.class_eval do
+      include Mongoid::Document
+    end
+    IntercomRails::Import.any_instance.stub(:user_klass).and_return(klass)
+
+    @import = IntercomRails::Import.new
+    @import.should_receive(:map_to_users_for_wire).with(klass.all).and_call_original
+    @import.should_receive(:send_users).and_return('failed' => [])
+    @import.run
+  end
+
   def test_status_output
     @import = IntercomRails::Import.new(:status_enabled => true)
     @import.stub(:send_users).and_return('failed' => [1])
@@ -62,7 +75,7 @@ class ImportUnitTest < MiniTest::Unit::TestCase
 ..F
 * Successfully created 2 users
 * Failed to create 1 user, this is likely due to bad data
-output
+    output
     $stdout.flush
 
     assert_equal expected_output, @output.string

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: intercom-rails
 version: !ruby/object:Gem::Version
-  version: 0.2.17
+  version: 0.2.18
   prerelease: 
 platform: ruby
 authors:
@@ -11,7 +11,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-04-11 00:00:00.000000000 Z
+date: 2013-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -66,17 +66,17 @@ dependencies:
   requirement: !ruby/object:Gem::Requirement
     none: false
     requirements:
-    - - ! '>='
+    - - ~>
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '2.12'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     none: false
     requirements:
-    - - ! '>='
+    - - ~>
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '2.12'
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
@@ -194,12 +194,18 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
+      segments:
+      - 0
+      hash: -3918465583784430054
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
+      segments:
+      - 0
+      hash: -3918465583784430054
 requirements: []
 rubyforge_project: intercom-rails
 rubygems_version: 1.8.23
@@ -218,4 +224,3 @@ test_files:
 - test/intercom-rails/script_tag_helper_test.rb
 - test/intercom-rails/script_tag_test.rb
 - test/test_setup.rb
-has_rdoc: