arel_extensions 0.8.0

data/lib/arel_extensions/nodes/rtrim.rb

@@ -0,0 +1,29 @@
+module ArelExtensions
+  module Nodes
+    class Rtrim < Arel::Nodes::Function
+
+      def initialize  left,right,aliaz = nil
+        tab = Array.new
+        tab << left
+        tab << right
+
+        super(tab, aliaz)
+      end
+
+      def left
+        @expressions.first
+      end
+
+      def right
+        @expressions[1]
+      end
+
+
+      def as other
+        Arel::Nodes::As.new self, Arel::Nodes::SqlLiteral.new(other)
+      end
+
+
+    end
+  end
+end

data/lib/arel_extensions/nodes/soundex.rb

@@ -0,0 +1,23 @@
+module ArelExtensions
+  module Nodes
+    class Soundex < Arel::Nodes::Function
+
+
+      def initialize other, aliaz = nil
+        tab = Array.new
+        tab << other
+        super(tab, aliaz)
+      end
+
+      def expr
+        @expressions.first
+      end
+
+
+      def as other
+        Arel::Nodes::As.new self, Arel::Nodes::SqlLiteral.new(other)
+      end
+
+    end
+  end
+end

data/lib/arel_extensions/nodes/sum.rb

@@ -0,0 +1,23 @@
+module ArelExtensions
+  module Nodes
+    class Sum < Arel::Nodes::Function
+
+
+      def initialize other, aliaz = nil
+        tab = Array.new
+        tab << other
+        super(tab, aliaz)
+      end
+
+      def expr
+        @expressions.first
+      end
+
+
+      def as other
+        Arel::Nodes::As.new self, Arel::Nodes::SqlLiteral.new(other)
+      end
+
+    end
+  end
+end

data/lib/arel_extensions/nodes/trim.rb

@@ -0,0 +1,26 @@
+module ArelExtensions
+  module Nodes
+    class Trim < Arel::Nodes::Function
+
+      def initialize  left,right, aliaz = nil
+        tab = Array.new
+        tab << left
+        tab << right
+        super(tab, aliaz)
+      end
+
+      def left
+        @expressions.first
+      end
+
+      def right
+        @expressions[1]
+      end
+
+
+      def as other
+        Arel::Nodes::As.new self, Arel::Nodes::SqlLiteral.new(other)
+      end
+    end
+  end
+end

data/lib/arel_extensions/nodes/wday.rb

@@ -0,0 +1,23 @@
+module ArelExtensions
+  module Nodes
+    class Wday < Arel::Nodes::Function
+
+
+      def initialize other, aliaz = nil
+        tab = Array.new
+        tab << other
+        super(tab, aliaz)
+      end
+
+
+      def date
+        @expressions.first
+      end
+
+      def as other
+        Arel::Nodes::As.new self, Arel::Nodes::SqlLiteral.new(other)
+      end
+
+    end
+  end
+end

data/lib/arel_extensions/null_functions.rb

@@ -0,0 +1,16 @@
+module ArelExtensions
+  module NullFunctions
+
+    #ISNULL function lets you return an alternative value when an expression is NULL.
+    def isnull(other)
+      ArelExtensions::Nodes::Isnull.new self, other
+    end
+    # returns the first non-null expr in the expression list. You must specify at least two expressions.
+    #If all occurrences of expr evaluate to null, then the function returns null.
+    def coalesce (*args)
+      args.unshift(self)
+      ArelExtensions::Nodes::Coalesce.new args
+    end
+
+  end
+end

data/lib/arel_extensions/string_functions.rb

@@ -0,0 +1,68 @@
+module ArelExtensions
+  module StringFunctions
+
+    #*FindInSet function .......
+    def &(other)
+      ArelExtensions::Nodes::FindInSet.new [other, self]
+    end
+
+    #LENGTH function returns the length of the value in a text field.
+    def length
+      ArelExtensions::Nodes::Length.new [self]
+    end
+
+    #LOCATE function returns the first starting position of a string in another string.
+    #If string1 or string2 is NULL then it returns NULL. If string1 not found in string2 then it returns 0.
+    def locate val
+      ArelExtensions::Nodes::Locate.new [self, val]
+    end
+
+    #SOUNDEX function returns a character string containing the phonetic representation of char.
+    def soundex
+      ArelExtensions::Nodes::Soundex.new self
+    end
+
+    def imatches others, escape = nil
+      ArelExtensions::Nodes::IMatches.new self, others, escape
+    end
+
+    def imatches_any others, escape = nil
+      grouping_any :imatches, others, escape
+    end
+
+    def imatches_all others, escape = nil
+      grouping_all :imatches, others, escape, escape
+    end
+
+    def idoes_not_match others, escape = nil
+      ArelExtensions::Nodes::IDoesNotMatch.new self, others, escape
+    end
+
+    def idoes_not_match_any others, escape = nil
+      grouping_any :idoes_not_match, others, escape
+    end
+
+    def idoes_not_match_all others, escape = nil
+      grouping_all :idoes_not_match, others, escape
+    end
+
+    #REPLACE function replaces a sequence of characters in a string with another set of characters, not case-sensitive.
+    def replace left, right
+      ArelExtensions::Nodes::Replace.new self, left, right
+    end
+
+    #Function returns a string after removing left, right or the both prefixes or suffixes int argument
+    def trim other
+      ArelExtensions::Nodes::Trim.new self,other
+    end
+
+    def ltrim other
+      ArelExtensions::Nodes::Ltrim.new self,other
+    end
+
+    def rtrim other
+      ArelExtensions::Nodes::Rtrim.new self, other
+    end
+
+  end
+end

data/lib/arel_extensions/version.rb

@@ -0,0 +1,4 @@
+# -*- encoding : utf-8 -*-
+module ArelExtensions
+  VERSION = "0.8.0".freeze
+end

data/lib/arel_extensions/visitors.rb

@@ -0,0 +1,6 @@
+require 'arel_extensions/visitors/to_sql'
+require 'arel_extensions/visitors/mysql'
+require 'arel_extensions/visitors/oracle'
+require 'arel_extensions/visitors/postgresql'
+require 'arel_extensions/visitors/sqlite'
+require 'arel_extensions/visitors/mssql'

data/lib/arel_extensions/visitors/ibm_db.rb

@@ -0,0 +1,206 @@
+module ArelExtensions
+  module Visitors
+    Arel::Visitors::IBM_DB.class_eval do
+
+      def visit_ArelExtensions_Nodes_Ceil o, collector
+        collector << "CEILING("
+       if((o.expr).is_a?(Arel::Attributes::Attribute))
+        collector = visit o.expr, collector
+      else
+          collector <<"#{o.expr}"
+       end
+        collector << ")"
+        collector
+      end
+
+      def visit_ArelExtensions_Nodes_Concat o, collector
+        arg = o.left.relation.engine.columns.find{|c| c.name == o.left.name.to_s}.type
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector << "CONCAT("
+          collector = visit o.left, collector
+          collector<< ","
+          collector = visit o.right, collector
+          collector << ")"
+        elsif ( arg == :date || arg == :datetime)
+          collector = visit o.left, collector
+          collector<< "+"
+          collector << "#{o.right} days"
+        else
+          collector << "CONCAT("
+          collector = visit o.left, collector
+          collector<< ","
+          collector <<"#{o.right})"
+        end
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_Coalesce o, collector
+        collector << "COALESCE("
+        if(o.left.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.left, collector
+        else
+          collector << "#{o.left}"
+        end
+           o.other.each { |a|
+          collector << ","
+          if(a.is_a?(Arel::Attributes::Attribute))
+            collector = visit a, collector
+          else
+            collector << "#{a}"
+          end
+        }
+        collector << ")"
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_DateDiff  o, collector
+
+        collector << "DAY("
+        collector = visit o.left, collector
+        collector<< ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector<< "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_Duration o , collector
+        #visit left for period
+        if(o.left == "d")
+          collector << "DAY("
+        elsif(o.left == "m")
+          collector << "MONTH("
+        elsif (o.left == "w")
+          collector << "WEEK"
+        elsif (o.left == "y")
+          collector << "YEAR("
+        end
+        #visit right
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_Findis o, collector
+
+      end
+
+      def visit_ArelExtensions_Nodes_Isnull o, collector
+        collector << "COALESCE("
+        collector = visit o.left, collector
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector <<")"
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_Replace o, collector
+        collector << "REPLACE("
+        collector = visit o.expr,collector
+        collector << ","
+        if(o.left.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.left, collector
+        else
+          collector << "'#{o.left}'"
+        end
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector <<")"
+        collector
+      end
+
+
+
+      def visit_ArelExtensions_Nodes_Soundex o, collector
+        collector << "SOUNDEX("
+       if((o.expr).is_a?(Arel::Attributes::Attribute))
+        collector = visit o.expr, collector
+      else
+          collector <<"'#{o.expr}'"
+       end
+        collector << ")"
+        collector
+      end
+
+
+      def visit_Arel_Nodes_Regexp o , collector
+        collector = visit o.left, collector
+        collector << " REGEXP "
+        collector << "'#{o.right}'"
+        collector
+      end
+
+
+
+      def visit_Arel_Nodes_NotRegexp o , collector
+        collector = visit o.left, collector
+        collector << " NOT REGEXP "
+        collector << "'#{o.right}'"
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_Trim o , collector
+          collector << "TRIM("
+        collector = visit o.left, collector
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+      def visit_ArelExtensions_Nodes_Ltrim o , collector
+        collector << "LTRIM("
+        collector = visit o.left, collector
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_Rtrim o , collector
+          collector << "RTRIM("
+        collector = visit o.left, collector
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+
+    end
+  end
+end

data/lib/arel_extensions/visitors/mssql.rb

@@ -0,0 +1,213 @@
+module ArelExtensions
+  module Visitors
+    Arel::Visitors::MSSQL.class_eval do
+
+      # Math Functions
+      def visit_ArelExtensions_Nodes_Ceil o, collector
+        collector << "CEILING("
+        collector = visit o.expr, collector
+        collector << ")"
+        collector
+      end
+
+      def visit_ArelExtensions_Nodes_Concat o, collector
+        arg = o.left.relation.engine.columns.find{|c| c.name == o.left.name.to_s}.type
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.left, collector
+          collector << ' + '
+          collector = visit o.right, collector
+          collector
+        elsif ( arg == :date || arg == :datetime)
+          collector << "DATEADD(day,#{o.right},"
+          collector = visit o.left, collector
+          collector
+        else
+          collector = visit o.left, collector
+          collector << " + '"
+          collector << "#{o.right}'"
+          collector
+        end
+      end
+
+      def visit_ArelExtensions_Nodes_Coalesce o, collector
+        collector << "COALESCE("
+        if(o.left.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.left, collector
+        else
+          collector << "#{o.left}"
+        end
+           o.other.each { |a|
+          collector << ","
+          if(a.is_a?(Arel::Attributes::Attribute))
+            collector = visit a, collector
+          else
+            collector << "#{a}"
+          end
+        }
+        collector << ")"
+        collector
+      end
+
+      def visit_ArelExtensions_Nodes_DateDiff o, collector
+
+          collector << "DATEDIFF(day,"
+          collector = visit o.left, collector
+          collector<< ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector<< "'#{o.right}'"
+
+        end
+        collector << ")"
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_Duration o, collector
+        #visit left for period
+        if(o.left == "d")
+          collector << "DAY("
+        elsif(o.left == "m")
+          collector << "MONTH("
+        elsif (o.left == "w")
+          collector << "WEEK"
+        elsif (o.left == "y")
+          collector << "YEAR("
+        end
+        #visit right
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+
+     def visit_ArelExtensions_Nodes_Findis o,collector
+     end
+
+
+     def visit_ArelExtensions_Nodes_Length o, collector
+       collector << "LEN("
+       collector = visit o.expr, collector
+       collector << ")"
+       collector
+     end
+
+
+     def visit_ArelExtensions_Nodes_Locate o, collector
+       collector << "CHARINDEX("
+       if(o.val.is_a?(Arel::Attributes::Attribute))
+         collector = visit o.val, collector
+       else
+         collector << "'#{o.val}'"
+       end
+       collector << ","
+       collector = visit o.expr, collector
+       collector << ")"
+       collector
+     end
+
+      def visit_ArelExtensions_Nodes_Isnull o, collector
+        collector << "ISNULL("
+        collector = visit o.left, collector
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+        collector = visit o.right, collector
+        else
+        collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+      def visit_ArelExtensions_Nodes_Replace o, collector
+         collector << "REPLACE("
+         collector = visit o.expr,collector
+         collector << ","
+         if(o.left.is_a?(Arel::Attributes::Attribute))
+           collector = visit o.left, collector
+         else
+           collector << "'#{o.left}'"
+         end
+         collector << ","
+         if(o.right.is_a?(Arel::Attributes::Attribute))
+           collector = visit o.right, collector
+         else
+           collector << "'#{o.right}'"
+         end
+         collector << ")"
+         collector
+      end
+
+
+      # SQL Server does not know about REGEXP
+      def visit_Arel_Nodes_Regexp o, collector
+        collector = visit o.left, collector
+        collector << "LIKE '% #{o.right}%'"
+        collector
+      end
+
+      def visit_Arel_Nodes_NotRegexp o, collector
+        collector = visit o.left, collector
+        collector << "NOT LIKE '% #{o.right}%'"
+        collector
+      end
+
+      def visit_ArelExtensions_Nodes_Soundex o, collector
+        collector << "SOUNDEX("
+       if((o.expr).is_a?(Arel::Attributes::Attribute))
+        collector = visit o.expr, collector
+      else
+          collector << "'#{o.expr}'"
+       end
+        collector << ")"
+        collector
+      end
+
+      def visit_ArelExtensions_Nodes_Trim o , collector
+          collector << "TRIM("
+        collector = visit o.left, collector
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+      def visit_ArelExtensions_Nodes_Ltrim o , collector
+          collector << "LTRIM("
+        collector = visit o.left, collector
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+
+      def visit_ArelExtensions_Nodes_Rtrim o , collector
+          collector << "RTRIM("
+        collector = visit o.left, collector
+        collector << ","
+        if(o.right.is_a?(Arel::Attributes::Attribute))
+          collector = visit o.right, collector
+        else
+          collector << "'#{o.right}'"
+        end
+        collector << ")"
+        collector
+      end
+
+    end
+  end
+end