arel_toolkit 0.1.0 → 0.2.0

data/lib/arel/extensions/current_timestamp.rb

@@ -0,0 +1,22 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    class CurrentTimestamp < TimeWithPrecision
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_CurrentTimestamp(o, collector)
+        collector << 'current_timestamp'
+        collector << "(#{o.precision.to_i})" if o.precision
+        collector
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/current_user.rb

@@ -0,0 +1,20 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    class CurrentUser < Arel::Nodes::Node
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_CurrentUser(_o, collector)
+        collector << 'current_user'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/default_values.rb

@@ -0,0 +1,21 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.5/sql-insert.html
+    class DefaultValues < Arel::Nodes::Node
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_DefaultValues(_o, collector)
+        collector << 'DEFAULT VALUES'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/delete_statement.rb

@@ -0,0 +1,49 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.5/sql-insert.html
+    Arel::Nodes::DeleteStatement.class_eval do
+      attr_accessor :using
+      attr_accessor :with
+      attr_accessor :returning
+    end
+  end
+
+  module Visitors
+    class ToSql
+      # rubocop:disable Metrics/AbcSize
+      def visit_Arel_Nodes_DeleteStatement(o, collector)
+        if o.with
+          collector = visit o.with, collector
+          collector << SPACE
+        end
+
+        collector << 'DELETE FROM '
+        collector = visit o.relation, collector
+
+        if o.using
+          collector << ' USING '
+          collector = inject_join o.using, collector, ', '
+        end
+
+        if o.wheres.any?
+          collector << WHERE
+          collector = inject_join o.wheres, collector, AND
+        end
+
+        unless o.returning.empty?
+          collector << ' RETURNING '
+          collector = inject_join o.returning, collector, ', '
+        end
+
+        maybe_visit o.limit, collector
+      end
+      # rubocop:enable Metrics/AbcSize
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/distinct_from.rb

@@ -0,0 +1,22 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    class DistinctFrom < Arel::Nodes::Binary
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_DistinctFrom(o, collector)
+        visit o.left, collector
+        collector << ' IS DISTINCT FROM '
+        visit o.right, collector
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/equality.rb

@@ -0,0 +1,30 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Equality(o, collector)
+        right = o.right
+
+        collector = visit o.left, collector
+
+        case right
+        when Arel::Nodes::Unknown, Arel::Nodes::False, Arel::Nodes::True
+          collector << ' IS '
+          visit right, collector
+
+        when NilClass
+          collector << ' IS NULL'
+
+        else
+          collector << ' = '
+          visit right, collector
+        end
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/except_all.rb

@@ -0,0 +1,21 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    class ExceptAll < Binary
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_ExceptAll(o, collector)
+        collector << '( '
+        infix_value(o, collector, ' EXCEPT ALL ') << ' )'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/exponentiation.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.1/functions-array.html
+    class Exponentiation < InfixOperation
+      def initialize(left, right)
+        super(:^, left, right)
+      end
+    end
+  end
+end

data/lib/arel/extensions/factorial.rb

@@ -0,0 +1,33 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.4/functions-math.html
+    class Factorial < Arel::Nodes::Unary
+      attr_accessor :prefix
+
+      def initialize(expr, prefix)
+        super(expr)
+        @prefix = prefix
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Factorial(o, collector)
+        if o.prefix
+          collector << '!! '
+          visit o.expr, collector
+        else
+          visit o.expr, collector
+          collector << ' !'
+        end
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/UncommunicativeMethodParamName
+# rubocop:enable Naming/MethodName

data/lib/arel/extensions/function.rb

@@ -0,0 +1,68 @@
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # Postgres: https://www.postgresql.org/docs/9.1/functions-comparison.html
+    Arel::Nodes::Function.class_eval do
+      # postgres only: https://www.postgresql.org/docs/9.5/functions-aggregate.html
+      attr_accessor :orders
+      attr_accessor :filter
+      attr_accessor :within_group
+      attr_accessor :variardic
+
+      def initialize(expr, aliaz = nil)
+        super()
+        @expressions = expr
+        @alias       = aliaz && SqlLiteral.new(aliaz)
+        @distinct    = false
+        @orders      = []
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      # rubocop:disable Metrics/PerceivedComplexity
+      # rubocop:disable Metrics/CyclomaticComplexity
+      # rubocop:disable Metrics/AbcSize
+      def aggregate(name, o, collector)
+        collector << "#{name}("
+        collector << 'DISTINCT ' if o.distinct
+        collector << 'VARIADIC ' if o.variardic
+
+        collector = inject_join(o.expressions, collector, ', ')
+
+        if o.within_group
+          collector << ')'
+          collector << ' WITHIN GROUP ('
+        end
+
+        if o.orders.any?
+          collector << SPACE unless o.within_group
+          collector << 'ORDER BY '
+          collector = inject_join o.orders, collector, ', '
+        end
+
+        collector << ')'
+
+        if o.filter
+          collector << ' FILTER(WHERE '
+          visit o.filter, collector
+          collector << ')'
+        end
+
+        if o.alias
+          collector << ' AS '
+          visit o.alias, collector
+        else
+          collector
+        end
+      end
+      # rubocop:enable Metrics/PerceivedComplexity
+      # rubocop:enable Metrics/CyclomaticComplexity
+      # rubocop:enable Metrics/AbcSize
+    end
+  end
+end
+
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/generate_series.rb

@@ -0,0 +1,9 @@
+module Arel
+  module Nodes
+    class GenerateSeries < Arel::Nodes::NamedFunction
+      def initialize(args)
+        super 'GENERATE_SERIES', args
+      end
+    end
+  end
+end

data/lib/arel/extensions/greatest.rb

@@ -0,0 +1,9 @@
+module Arel
+  module Nodes
+    class Greatest < Arel::Nodes::NamedFunction
+      def initialize(args)
+        super 'GREATEST', args
+      end
+    end
+  end
+end

data/lib/arel/extensions/indirection.rb

@@ -0,0 +1,32 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    class Indirection < Arel::Nodes::Node
+      attr_reader :arg
+      attr_reader :indirection
+
+      def initialize(arg, indirection)
+        super()
+
+        @arg = arg
+        @indirection = indirection
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Indirection(o, collector)
+        visit(o.arg, collector)
+        collector << '['
+        visit(o.indirection, collector)
+        collector << ']'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/infer.rb

@@ -0,0 +1,35 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.5/sql-insert.html
+    class Infer < Arel::Nodes::Node
+      attr_accessor :name
+      attr_accessor :indexes
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Infer(o, collector)
+        if o.name
+          collector << 'ON CONSTRAINT '
+          collector << o.name
+          collector << SPACE
+        end
+
+        if o.indexes
+          collector << '('
+          inject_join o.indexes, collector, ', '
+          collector << ') '
+        end
+
+        collector
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/insert_statement.rb

@@ -0,0 +1,69 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.5/sql-insert.html
+    Arel::Nodes::InsertStatement.class_eval do
+      attr_accessor :with
+      attr_accessor :on_conflict
+      attr_accessor :override
+      attr_accessor :returning
+    end
+  end
+
+  module Visitors
+    class ToSql
+      # rubocop:disable Metrics/CyclomaticComplexity
+      # rubocop:disable Metrics/AbcSize
+      # rubocop:disable Metrics/PerceivedComplexity
+      def visit_Arel_Nodes_InsertStatement(o, collector)
+        if o.with
+          collector = visit o.with, collector
+          collector << SPACE
+        end
+
+        collector << 'INSERT INTO '
+        collector = visit o.relation, collector
+        if o.columns.any?
+          collector << " (#{o.columns.map do |x|
+            quote_column_name x.name
+          end.join ', '})"
+        end
+
+        case o.override
+        when 0
+          collector << ''
+        when 1
+          collector << ' OVERRIDING USER VALUE'
+        when 2
+          collector << ' OVERRIDING SYSTEM VALUE'
+        else
+          raise "Unknown override `#{o.override}`"
+        end
+
+        collector = if o.values
+                      maybe_visit o.values, collector
+                    elsif o.select
+                      maybe_visit o.select, collector
+                    else
+                      collector
+                    end
+
+        unless o.returning.empty?
+          collector << ' RETURNING '
+          collector = inject_join o.returning, collector, ', '
+        end
+
+        visit(o.on_conflict, collector) if o.on_conflict
+        collector
+      end
+      # rubocop:enable Metrics/CyclomaticComplexity
+      # rubocop:enable Metrics/AbcSize
+      # rubocop:enable Metrics/PerceivedComplexity
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName