active_record_sql_unionizer 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: a02c2215bc7bfb78cb28677fe3feb34c086ea0ba
-  data.tar.gz: cbb35c5d3fed7e8de039a6e933b6c3825ea582f8
+  metadata.gz: 6edeb9257c4226bca2f11ddf5f60659ebecd58b2
+  data.tar.gz: a8ea988d3fc032efdae9a2c6361c1929bd0cf7a8
 SHA512:
-  metadata.gz: bf18bb9ec9454df6c659d221f4870c56a7fcdb5f00e0db10f9d0ad3f5712037a25deac948abf9505c2e4b8d380ad6d2ff926085a73887a40803da32d45fcc39a
-  data.tar.gz: 54d73e6f4423452999ef0c787fddc5df1d9e6f104425ede3fb9f858d38f9a48d60138b675545aaadef9513cbbecc8678f467ccda4cf16fa84acdedc51df9dac5
+  metadata.gz: 20096ba4f489c6646f714fec2d9616442e96025e466bc83ec0e248342fa3e9b67423d894653ce58d2e218d609fd4109944be64b0f1a1bd58749ea51a2ec99f09
+  data.tar.gz: ab817088c9827fad57ffe147bb4d5d1fa462146489221d5222e75da105f2cb03cf7eb52771f23699772ce842ffee3eeb52d1928ce14c3cea26d0d6b5aba9bdaf

data/lib/active_record_sql_unionizer/active_record_sql_unionizer.rb

@@ -4,15 +4,79 @@ module ActiveRecordSqlUnionizer
   extend ActiveSupport::Concern
 
   module ClassMethods
+    #passed in objects can be valid SQL strings, ActiveRecord::Relation, and class methods
+    #as symbols that return an ActiveRecord::Relation.
+    # @param [Splat] queries
+    #
+    # @return [ActiveRecord::Relation]
     def unionize(*queries)
-      union_string = queries.map do |query|
-        query_string = query.is_a?(String) ? query : query.to_sql
+      unionizer_helper = UnionizerHelper.new
+
+      table_name = unionizer_helper.get_table_name(self)
+
+      sql_queries = queries.map do |query|
+        query_string =
+            case query
+              when String
+                query
+              when Symbol
+                unionizer_helper.handle_symbol_arg(self, query)
+              else
+                query.to_sql
+            end
+
         "(#{query_string})"
-      end.join(" UNION ")
-      union_string = "(#{union_string}) AS #{self.to_s.underscore.downcase.pluralize}"
+      end
+
+      union_string = unionizer_helper.construct_final_query_string(sql_queries, table_name)
 
       sql = self.connection.unprepared_statement {union_string}
       self.from(sql)
     end
   end
+
+  class UnionizerHelper
+    #for 'private' methods/ readability
+
+    # @param [Object] klass
+    #
+    # @return [String]
+    def get_table_name(klass)
+      table_klass = klass
+      while table_klass.superclass != ActiveRecord::Base
+        table_klass = table_klass.superclass
+      end
+      table_klass.to_s.underscore.downcase.pluralize
+    end
+
+    # @param [Symbol] arg
+    #
+    # @return [String, UnionizerError]
+    def handle_symbol_arg(klass, arg)
+      if klass.respond_to?(arg)
+        klass.send(arg).to_sql
+      else
+        raise(UnionizerError.new(klass.to_s, arg))
+      end
+    end
+
+    # @param [Array<String>] queries
+    # @param [String] table_name
+    #
+    # @return [String]
+    def construct_final_query_string(queries, table_name)
+      "(#{queries.join(" UNION ")}) AS #{table_name}"
+    end
+  end
+
+  class UnionizerError < StandardError
+    # @param [String] klass_name
+    # @param [Symbol] method_name
+    #
+    # @return [UnionizerError]
+    def initialize(klass_name, method_name)
+      msg = "ActiveRecordSqlUnionizer expected #{klass_name} to respond to #{method_name}, but it does not"
+      super(msg)
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: active_record_sql_unionizer
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Jeff Hostetler