pluck_all 2.0.0 → 2.1.0

data/lib/pluck_all/hooks.rb

@@ -0,0 +1,21 @@
+module PluckAll
+  class Hooks
+    class << self
+      def init
+        require 'pluck_all/models/active_record_extension' if require_if_exists('active_record')
+        require 'pluck_all/models/mongoid_extension' if require_if_exists('mongoid')
+      end
+
+      private
+
+      def require_if_exists(path)
+        require path
+        return true
+      rescue LoadError, Gem::LoadError
+        return false
+      end
+    end
+  end
+end
+
+PluckAll::Hooks.init

data/lib/pluck_all/models/active_record_extension.rb

@@ -0,0 +1,130 @@
+require 'rails_compatibility/attribute_types'
+require_relative 'patches/deserialize'
+
+class ActiveRecord::Relation
+  def cast_need_columns(column_names, _klass = nil)
+    @pluck_all_cast_need_columns = column_names.map(&:to_s)
+    @pluck_all_cast_klass = _klass
+    return self
+  end
+
+  def select_all(*column_names)
+    relation = clone
+    relation.select_values = [].freeze # cannot use `unscope(:select)` in Rails 3
+    return klass.connection.select_all(relation.select(column_names).to_sql)
+  end
+
+  if Gem::Version.new(ActiveRecord::VERSION::STRING) < Gem::Version.new('4.0.0')
+    def pluck_all(*column_names, cast_uploader_url: true)
+      column_names.map!(&to_sql_column_name)
+      result = select_all(*column_names)
+      result.map! do |attributes| # This map! behaves different to array#map!
+        initialized_attributes = klass.initialize_attributes(attributes)
+        attributes.each do |key, _attribute|
+          attributes[key] = klass.type_cast_attribute(key, initialized_attributes) # TODO: 現在AS過後的type cast會有一點問題
+        end
+        cast_carrier_wave_uploader_url(attributes) if cast_uploader_url
+        next attributes
+      end
+    end
+
+    private
+
+    def to_sql_column_name
+      proc do |column_name|
+        if column_name.is_a?(Arel::Attributes::Attribute)
+          "#{column_name.relation.name}.#{column_name.name}"
+        elsif column_name.is_a?(Symbol) && column_names.include?(column_name.to_s)
+          "#{connection.quote_table_name(table_name)}.#{connection.quote_column_name(column_name)}"
+        else
+          column_name.to_s
+        end
+      end
+    end
+  else
+    def pluck_all(*column_names, cast_uploader_url: true)
+      column_names.map!(&to_sql_column_name)
+      if has_include?(column_names.first)
+        # The `construct_relation_for_association_calculations` method was removed at Rails 5.2.
+        relation = Gem::Version.new(ActiveRecord::VERSION::STRING) >= Gem::Version.new('5.2.0') ? apply_join_dependency : construct_relation_for_association_calculations
+        return relation.pluck_all(*column_names)
+      end
+      result = select_all(*column_names)
+      attribute_types = RailsCompatibility.attribute_types(klass)
+      result.map! do |attributes| # This map! behaves different to array#map!
+        attributes.each do |key, attribute|
+          attributes[key] = result.send(:column_type, key, attribute_types).deserialize(attribute) # TODO: 現在AS過後的type cast會有一點問題，但似乎原生的pluck也有此問題
+        end
+        cast_carrier_wave_uploader_url(attributes) if cast_uploader_url
+        next attributes
+      end
+    end
+
+    private
+
+    def to_sql_column_name
+      proc do |column_name|
+        if column_name.is_a?(Arel::Attributes::Attribute)
+          "#{column_name.relation.name}.#{column_name.name}"
+        elsif column_name.is_a?(Symbol) && attribute_alias?(column_name)
+          attribute_alias(column_name)
+        else
+          column_name.to_s
+        end
+      end
+    end
+  end
+
+  # ----------------------------------------------------------------
+  # ● Support casting CarrierWave url
+  # ----------------------------------------------------------------
+  def cast_carrier_wave_uploader_url(attributes)
+    if defined?(CarrierWave) && klass.respond_to?(:uploaders)
+      @pluck_all_cast_klass ||= klass
+      @pluck_all_uploaders ||= @pluck_all_cast_klass.uploaders.select{|key, _uploader| attributes.key?(key.to_s) }
+      @pluck_all_uploaders.each do |key, _uploader|
+        {}.tap do |hash|
+          @pluck_all_cast_need_columns.each{|k| hash[k] = attributes[k] } if @pluck_all_cast_need_columns
+          obj = @pluck_all_cast_klass.instantiate(hash)
+          obj[key] = attributes[key_s = key.to_s]
+          # https://github.com/carrierwaveuploader/carrierwave/blob/87c37b706c560de6d01816f9ebaa15ce1c51ed58/lib/carrierwave/mount.rb#L142
+          attributes[key_s] = obj.send(key)
+        end
+      end
+    end
+    return attributes
+  end
+end
+
+class ActiveRecord::Relation
+  if Gem::Version.new(ActiveRecord::VERSION::STRING) < Gem::Version.new('4.0.2')
+    def pluck_array(*args)
+      return pluck_all(*args, cast_uploader_url: false).map do |hash|
+        result = hash.values # P.S. 這裡是相信ruby 1.9以後，hash.values的順序跟insert的順序一樣。
+        next (args.one? ? result.first : result)
+      end
+    end
+  else
+    alias pluck_array pluck if not method_defined?(:pluck_array)
+  end
+end
+
+class << ActiveRecord::Base
+  def cast_need_columns(*args)
+    where(nil).cast_need_columns(*args)
+  end
+
+  def pluck_all(*args)
+    where(nil).pluck_all(*args)
+  end
+
+  def pluck_array(*args)
+    where(nil).pluck_array(*args)
+  end
+end
+
+module ActiveRecord::NullRelation
+  def pluck_all(*_args)
+    []
+  end
+end

data/lib/pluck_all/{mongoid_pluck_all.rb → models/mongoid_extension.rb}

@@ -1,5 +1,15 @@
 # frozen_string_literal: true
 module Mongoid
+  module Document::ClassMethods
+    def pluck_array(*fields)
+      where(nil).pluck_array(*fields)
+    end
+
+    def pluck_all(*fields)
+      where(nil).pluck_all(*fields)
+    end
+  end
+
   module Findable
     delegate :pluck_all, :pluck_array, to: :with_default_scope
   end
@@ -21,9 +31,7 @@ module Mongoid
       def pluck_array(*fields)
         normalized_select = get_normalized_select(fields)
         get_query_data(normalized_select).reduce([]) do |plucked, doc|
-          values = normalized_select.keys.map do |n|
-            n =~ /\./ ? doc[n.partition('.')[0]] : doc[n]
-          end
+          values = normalized_select.keys.map(&plucked_value_mapper(:array, doc))
           plucked << (values.size == 1 ? values.first : values)
         end
       end
@@ -31,23 +39,30 @@ module Mongoid
       def pluck_all(*fields)
         normalized_select = get_normalized_select(fields)
         get_query_data(normalized_select).reduce([]) do |plucked, doc|
-          values = normalized_select.keys.map do |n|
-            [n, n =~ /\./ ? doc[n.partition('.')[0]] : doc[n]]
-          end.to_h
-          plucked << values
+          values = normalized_select.keys.map(&plucked_value_mapper(:all, doc))
+          plucked << values.to_h
         end
       end
 
       private
 
+      def plucked_value_mapper(type, doc)
+        proc do |n|
+          values = [n, n =~ /\./ ? doc[n.partition('.')[0]] : doc[n]]
+          case type
+          when :array then values[1]
+          when :all   then values
+          end
+        end
+      end
+
       def get_query_data(normalized_select)
         return (@view ? @view.projection(normalized_select) : query.dup.select(normalized_select))
       end
 
       def get_normalized_select(fields)
-        normalized_select = fields.inject({}) do |hash, f|
+        fields.each_with_object({}) do |f, hash|
           hash[klass.database_field_name(f)] = 1
-          hash
         end
       end
     end

data/lib/pluck_all/models/patches/deserialize.rb

@@ -0,0 +1,13 @@
+module ActiveRecord
+  [
+    *([Type::Value, Type::Integer, Type::Serialized] if defined?(Type::Value)),
+    *([Enum::EnumType] if defined?(Enum::EnumType)),
+  ].each do |s|
+    s.class_eval do
+      if !method_defined?(:deserialize) && method_defined?(:type_cast_from_database)
+        # deserialize was changed to type_cast_from_database in Rails 5
+        alias_method :deserialize, :type_cast_from_database
+      end
+    end
+  end
+end

data/lib/pluck_all/version.rb

@@ -1,4 +1,4 @@
 # frozen_string_literal: true
 module PluckAll
-  VERSION = '2.0.0'
+  VERSION = '2.1.0'
 end

data/pluck_all.gemspec

@@ -4,35 +4,40 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 require 'pluck_all/version'
 
 Gem::Specification.new do |spec|
-  spec.name          = "pluck_all"
+  spec.name          = 'pluck_all'
   spec.version       = PluckAll::VERSION
-  spec.authors       = ["khiav reoy"]
-  spec.email         = ["mrtmrt15xn@yahoo.com.tw"]
+  spec.authors       = ['khiav reoy']
+  spec.email         = ['mrtmrt15xn@yahoo.com.tw']
 
-  spec.summary       = %q{Pluck multiple columns/attributes and return array of hashes. Support Rails 3, 4, 5.}
-  spec.description   = %q{Pluck multiple columns/attributes and return array of hashes. Support Rails 3, 4, 5. If you have a Rails 3 project, and want to pluck not only one column, feel free to use this gem and no need to worry about upgrading to Rails 4, 5 in the future will break this.}
-  spec.homepage      = "https://github.com/khiav223577/pluck_all"
-  spec.license       = "MIT"
+  spec.summary       = 'Pluck multiple columns/attributes and return array of hashes. Support Rails 3, 4, 5.'
+  spec.description   = 'Pluck multiple columns/attributes and return array of hashes. Support Rails 3, 4, 5. If you have a Rails 3 project, and want to pluck not only one column, feel free to use this gem and no need to worry about upgrading to Rails 4, 5 in the future will break this.'
+  spec.homepage      = 'https://github.com/khiav223577/pluck_all'
+  spec.license       = 'MIT'
 
   # Prevent pushing this gem to RubyGems.org by setting 'allowed_push_host', or
   # delete this section to allow pushing this gem to any host.
-  #if spec.respond_to?(:metadata)
+  # if spec.respond_to?(:metadata)
   #  spec.metadata['allowed_push_host'] = "TODO: Set to 'http://mygemserver.com'"
-  #else
+  # else
   #  raise "RubyGems 2.0 or newer is required to protect against public gem pushes."
-  #end
+  # end
 
-  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
-  spec.bindir        = "exe"
-  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
-  spec.require_paths = ["lib"]
+  spec.files         = `git ls-files -z`.split("\x0").reject{|f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = 'exe'
+  spec.executables   = spec.files.grep(%r{^exe/}){|f| File.basename(f) }
+  spec.require_paths = ['lib']
+  spec.metadata      = {
+    'homepage_uri'      => 'https://github.com/khiav223577/pluck_all',
+    'changelog_uri'     => 'https://github.com/khiav223577/pluck_all/blob/master/CHANGELOG.md',
+    'source_code_uri'   => 'https://github.com/khiav223577/pluck_all',
+    'documentation_uri' => 'https://www.rubydoc.info/gems/pluck_all',
+    'bug_tracker_uri'   => 'https://github.com/khiav223577/pluck_all/issues',
+  }
 
-  spec.add_development_dependency "bundler", "~> 1.11"
-  spec.add_development_dependency "rake", "~> 12.0"
-  spec.add_development_dependency "sqlite3", "~> 1.3"
-  spec.add_development_dependency "minitest", "~> 5.0"
-  spec.add_development_dependency "mongoid", ">= 3.1.7"
-
-  spec.add_dependency "activerecord", ">= 3"
+  spec.add_dependency 'activesupport', '>= 3.0.0'
+  spec.add_dependency 'rails_compatibility', '>= 0.0.2'
 
+  spec.add_development_dependency 'bundler', '>= 1.17', '< 3.x'
+  spec.add_development_dependency 'rake', '~> 12.0'
+  spec.add_development_dependency 'minitest', '~> 5.0'
 end

metadata

@@ -1,99 +1,91 @@
 --- !ruby/object:Gem::Specification
 name: pluck_all
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 2.1.0
 platform: ruby
 authors:
 - khiav reoy
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-05-27 00:00:00.000000000 Z
+date: 2020-10-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: bundler
+  name: activesupport
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.11'
-  type: :development
+        version: 3.0.0
+  type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.11'
+        version: 3.0.0
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: rails_compatibility
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '12.0'
-  type: :development
+        version: 0.0.2
+  type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '12.0'
+        version: 0.0.2
 - !ruby/object:Gem::Dependency
-  name: sqlite3
+  name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.17'
+    - - "<"
       - !ruby/object:Gem::Version
-        version: '1.3'
+        version: 3.x
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.17'
+    - - "<"
       - !ruby/object:Gem::Version
-        version: '1.3'
+        version: 3.x
 - !ruby/object:Gem::Dependency
-  name: minitest
+  name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '5.0'
+        version: '12.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '5.0'
+        version: '12.0'
 - !ruby/object:Gem::Dependency
-  name: mongoid
+  name: minitest
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: 3.1.7
+        version: '5.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: 3.1.7
-- !ruby/object:Gem::Dependency
-  name: activerecord
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '3'
-  type: :runtime
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '3'
+        version: '5.0'
 description: Pluck multiple columns/attributes and return array of hashes. Support
   Rails 3, 4, 5. If you have a Rails 3 project, and want to pluck not only one column,
   feel free to use this gem and no need to worry about upgrading to Rails 4, 5 in
@@ -105,6 +97,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".rubocop.yml"
 - ".travis.yml"
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
@@ -113,19 +106,31 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
-- gemfiles/3.2.gemfile
-- gemfiles/4.2.gemfile
-- gemfiles/5.0.gemfile
-- gemfiles/5.1.gemfile
-- gemfiles/5.2.gemfile
+- gemfiles/active_record_32.gemfile
+- gemfiles/active_record_42.gemfile
+- gemfiles/active_record_50.gemfile
+- gemfiles/active_record_51.gemfile
+- gemfiles/active_record_52.gemfile
+- gemfiles/active_record_60.gemfile
+- gemfiles/mongoid_54.gemfile
+- gemfiles/mongoid_64.gemfile
+- gemfiles/mongoid_70.gemfile
 - lib/pluck_all.rb
-- lib/pluck_all/mongoid_pluck_all.rb
+- lib/pluck_all/hooks.rb
+- lib/pluck_all/models/active_record_extension.rb
+- lib/pluck_all/models/mongoid_extension.rb
+- lib/pluck_all/models/patches/deserialize.rb
 - lib/pluck_all/version.rb
 - pluck_all.gemspec
 homepage: https://github.com/khiav223577/pluck_all
 licenses:
 - MIT
-metadata: {}
+metadata:
+  homepage_uri: https://github.com/khiav223577/pluck_all
+  changelog_uri: https://github.com/khiav223577/pluck_all/blob/master/CHANGELOG.md
+  source_code_uri: https://github.com/khiav223577/pluck_all
+  documentation_uri: https://www.rubydoc.info/gems/pluck_all
+  bug_tracker_uri: https://github.com/khiav223577/pluck_all/issues
 post_install_message: 
 rdoc_options: []
 require_paths:
@@ -141,8 +146,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.6
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: Pluck multiple columns/attributes and return array of hashes. Support Rails