arel_toolkit 0.2.0 → 0.4.3

data/lib/arel/extensions/range_function.rb

@@ -5,15 +5,23 @@ module Arel
   module Nodes
     # Postgres: https://www.postgresql.org/docs/9.4/sql-select.html
     class RangeFunction < Arel::Nodes::Unary
+      attr_reader :is_rowsfrom
+
+      def initialize(*args, is_rowsfrom:, **kwargs)
+        @is_rowsfrom = is_rowsfrom
+        super(*args, **kwargs)
+      end
     end
   end
 
   module Visitors
     class ToSql
       def visit_Arel_Nodes_RangeFunction(o, collector)
-        collector << 'ROWS FROM ('
+        collector << 'ROWS FROM (' if o.is_rowsfrom
         visit o.expr, collector
-        collector << ')'
+        collector << ')' if o.is_rowsfrom
+
+        collector
       end
     end
   end

data/lib/arel/extensions/row.rb

@@ -4,14 +4,9 @@
 module Arel
   module Nodes
     # Postgres: https://www.postgresql.org/docs/9.2/sql-expressions.html
-    class Row < Arel::Nodes::Unary
-      attr_reader :row_format
-
-      def initialize(args, row_format)
-        super(args)
-
-        @row_format = row_format
-      end
+    class Row < Arel::Nodes::Binary
+      alias expr left
+      alias row_format right
     end
   end
 

data/lib/arel/extensions/select_core.rb

@@ -0,0 +1,73 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+# rubocop:disable Metrics/AbcSize
+
+module Arel
+  module Nodes
+    class SelectCore < Arel::Nodes::Node
+      attr_accessor :into
+      attr_accessor :top
+
+      private
+
+      def hash
+        [
+          @source, @set_quantifier, @projections, @optimizer_hints,
+          @wheres, @groups, @havings, @windows, @comment, @top, @into
+        ].hash
+      end
+
+      def eql?(other)
+        super &&
+          top == other.top &&
+          into == other.into
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_SelectCore(o, collector)
+        collector << 'SELECT'
+
+        collector = maybe_visit o.set_quantifier, collector
+
+        collect_nodes_for o.projections, collector, ' '
+
+        maybe_visit o.into, collector
+
+        if o.source && !o.source.empty?
+          collector << ' FROM '
+          collector = visit o.source, collector
+        end
+
+        collect_nodes_for o.wheres, collector, ' WHERE ', ' AND '
+        collect_nodes_for o.groups, collector, ' GROUP BY '
+        unless o.havings.empty?
+          collector << ' HAVING '
+          inject_join o.havings, collector, ' AND '
+        end
+        collect_nodes_for o.windows, collector, ' WINDOW '
+
+        collector
+      end
+    end
+
+    class Dot
+      module SelectCoreExtension
+        def visit_Arel_Nodes_SelectCore(o)
+          super
+
+          visit_edge o, 'into'
+          visit_edge o, 'top'
+        end
+      end
+
+      prepend SelectCoreExtension
+    end
+  end
+end
+
+# rubocop:enable Metrics/AbcSize
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/select_manager.rb

@@ -0,0 +1,25 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  class SelectManager
+    def ==(other)
+      other.is_a?(self.class) && @ast == other.ast && @ctx == other.ctx
+    end
+
+    protected
+
+    attr_reader :ctx
+  end
+
+  module Visitors
+    class Dot
+      def visit_Arel_SelectManager(o)
+        visit_edge(o, 'ast')
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/select_statement.rb

@@ -3,21 +3,43 @@
 
 module Arel
   module Nodes
-    Arel::Nodes::SelectStatement.class_eval do
-      # For INSERT statements
-      attr_accessor :values_lists
-      attr_accessor :union
-      attr_writer :cores
+    class SelectStatement
+      module SelectStatementExtension
+        # For INSERT statements
+        attr_accessor :values_lists
+        attr_accessor :union
+        attr_writer :cores
+      end
+
+      prepend SelectStatementExtension
     end
   end
 
   module Visitors
     class ToSql
-      alias old_visit_Nodes_SelectStatement visit_Arel_Nodes_SelectStatement
-      def visit_Arel_Nodes_SelectStatement(o, collector)
-        visit(o.union, collector) if o.union
-        old_visit_Nodes_SelectStatement(o, collector)
+      module SelectStatementExtension
+        def visit_Arel_Nodes_SelectStatement(o, collector)
+          visit(o.union, collector) if o.union
+          super
+        end
+      end
+
+      prepend SelectStatementExtension
+    end
+
+    class Dot
+      module SelectStatementExtension
+        def visit_Arel_Nodes_SelectStatement(o)
+          super
+
+          visit_edge o, 'lock'
+          visit_edge o, 'with'
+          visit_edge o, 'union'
+          visit_edge o, 'values_lists'
+        end
       end
+
+      prepend SelectStatementExtension
     end
   end
 end

data/lib/arel/extensions/session_user.rb

@@ -13,6 +13,10 @@ module Arel
         collector << 'session_user'
       end
     end
+
+    class Dot
+      alias visit_Arel_Nodes_SessionUser terminal
+    end
   end
 end
 

data/lib/arel/extensions/set_to_default.rb

@@ -14,6 +14,10 @@ module Arel
         collector << 'DEFAULT'
       end
     end
+
+    class Dot
+      alias visit_Arel_Nodes_SetToDefault terminal
+    end
   end
 end
 

data/lib/arel/extensions/substring.rb

@@ -0,0 +1,46 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/10/functions-string.html
+    class Substring < Arel::Nodes::Node
+      attr_reader :string
+      attr_reader :pattern
+      attr_reader :escape
+
+      def initialize(string, pattern, escape)
+        @string = string
+        @pattern = pattern
+        @escape = escape
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Substring(o, collector)
+        collector << 'substring('
+        visit o.string, collector
+        collector << ' from '
+        visit o.pattern, collector
+        unless o.escape.nil?
+          collector << ' for '
+          visit o.escape, collector
+        end
+        collector << ')'
+      end
+    end
+
+    class Dot
+      def visit_Arel_Nodes_Substring(o)
+        visit_edge o, 'string'
+        visit_edge o, 'pattern'
+        visit_edge o, 'escape'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/table.rb

@@ -3,16 +3,15 @@
 # rubocop:disable Metrics/ParameterLists
 
 module Arel
-  module Nodes
-    Arel::Table.class_eval do
+  class Table
+    module TableExtension
       # postgres only: https://www.postgresql.org/docs/9.5/sql-select.html
       attr_accessor :only
-      # postgres only: https://www.postgresql.org/docs/9.5/ddl-schemas.html
+      # postgres only: https://www.postgresql.org/docs/10/ddl-schemas.html
       attr_accessor :schema_name
       # postgres only: https://www.postgresql.org/docs/9.1/catalog-pg-class.html
       attr_accessor :relpersistence
 
-      alias_method :old_initialize, :initialize
       def initialize(
         name,
         as: nil,
@@ -25,21 +24,55 @@ module Arel
         @schema_name = schema_name
         @relpersistence = relpersistence
 
-        old_initialize(name, as: as, type_caster: type_caster)
+        super(name, as: as, type_caster: type_caster)
       end
     end
+
+    prepend TableExtension
   end
 
   module Visitors
     class ToSql
-      alias old_visit_Arel_Table visit_Arel_Table
-      def visit_Arel_Table(o, collector)
-        collector << 'ONLY ' if o.only
+      module TableExtension
+        def visit_Arel_Table(o, collector)
+          collector << 'ONLY ' if o.only
+
+          case o.relpersistence
+          when 'p'
+            collector << ''
+
+          when 'u'
+            collector << 'UNLOGGED '
 
-        collector << "\"#{o.schema_name}\"." if o.schema_name
+          when 't'
+            collector << 'TEMPORARY '
 
-        old_visit_Arel_Table(o, collector)
+          else
+            raise "Unknown relpersistence `#{o.relpersistence}`"
+          end
+
+          collector << "\"#{o.schema_name}\"." if o.schema_name
+
+          super
+        end
       end
+
+      prepend TableExtension
+    end
+
+    class Dot
+      module TableExtension
+        def visit_Arel_Table(o)
+          super
+
+          visit_edge o, 'only'
+          visit_edge o, 'schema_name'
+          visit_edge o, 'relpersistence'
+          visit_edge o, 'type_caster'
+        end
+      end
+
+      prepend TableExtension
     end
   end
 end

data/lib/arel/extensions/time_with_precision.rb

@@ -10,4 +10,10 @@ module Arel
       end
     end
   end
+
+  module Visitors
+    class Dot
+      alias visit_Arel_Nodes_TimeWithPrecision terminal
+    end
+  end
 end

data/lib/arel/extensions/to_sql.rb

@@ -0,0 +1,27 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class ToSql
+      def visit_Arel_Attributes_Attribute(o, collector)
+        if o.relation
+          join_name = o.relation.table_alias || o.relation.name
+          collector << "#{quote_table_name join_name}.#{quote_column_name o.name}"
+        else
+          visit_Arel_Nodes_UnqualifiedColumn o, collector
+        end
+      end
+
+      alias visit_Arel_Attributes_Integer visit_Arel_Attributes_Attribute
+      alias visit_Arel_Attributes_Float visit_Arel_Attributes_Attribute
+      alias visit_Arel_Attributes_Decimal visit_Arel_Attributes_Attribute
+      alias visit_Arel_Attributes_String visit_Arel_Attributes_Attribute
+      alias visit_Arel_Attributes_Time visit_Arel_Attributes_Attribute
+      alias visit_Arel_Attributes_Boolean visit_Arel_Attributes_Attribute
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/top.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module Arel # :nodoc: all
+  module Nodes
+    class Top < Unary
+    end
+  end
+end

data/lib/arel/extensions/transaction.rb

@@ -0,0 +1,45 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+# rubocop:disable Metrics/CyclomaticComplexity
+# rubocop:disable Metrics/AbcSize
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/8.3/tutorial-transactions.html
+    class Transaction < Arel::Nodes::Binary
+      alias type left
+      alias options right
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Transaction(o, collector)
+        case o.type
+        when 0
+          collector << 'BEGIN'
+        when 2
+          collector << 'COMMIT'
+        when 3
+          collector << 'ROLLBACK'
+        when 4
+          collector << 'SAVEPOINT '
+          collector << o.options.join(' ')
+        when 5
+          collector << 'RELEASE SAVEPOINT '
+          collector << o.options.join(' ')
+        when 6
+          collector << 'ROLLBACK TO '
+          collector << o.options.join(' ')
+        else
+          raise "Unknown transaction type `#{o.type}`"
+        end
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName
+# rubocop:enable Metrics/CyclomaticComplexity
+# rubocop:enable Metrics/AbcSize