arel_extensions 0.8.0

data/gemfiles/Gemfile.rails3

@@ -0,0 +1,10 @@
+source "https://rubygems.org"
+
+gemspec
+
+group :test do
+	gem "sqlite3", :platform => [:ruby, :mswin, :mingw]
+	# for JRuby
+	gem "jdbc-sqlite3", :platform => :jrubyend
+	gem 'activerecord', '~> 3.0'
+end

data/gemfiles/Gemfile.rails5

@@ -0,0 +1,10 @@
+source "https://rubygems.org"
+
+gemspec
+
+group :test do
+	gem "sqlite3", :platform => [:ruby, :mswin, :mingw]
+	# for JRuby
+	gem "jdbc-sqlite3", :platform => :jrubyend
+	gem 'activerecord', '~> 5.0'
+end

data/init/mssql.sql

@@ -0,0 +1,6 @@
+CREATE FUNCTION TRIM(@string VARCHAR(MAX))
+RETURNS VARCHAR(MAX)
+BEGIN
+RETURN LTRIM(RTRIM(@string))
+END
+GO

data/init/oracle.sql

@@ -0,0 +1,31 @@
+CREATE OR REPLACE FUNCTION find_in_set(
+  i_value  IN  VARCHAR2,
+  i_list   IN  VARCHAR2,
+  i_delim  IN  VARCHAR2 DEFAULT ','
+) RETURN INT DETERMINISTIC
+AS
+  p_result       INT       := 0;
+  p_start        NUMBER(5) := 1;
+  p_end          NUMBER(5);
+  c_len CONSTANT NUMBER(5) := LENGTH( i_list );
+  c_ld  CONSTANT NUMBER(5) := LENGTH( i_delim );
+BEGIN
+  IF c_len > 0 THEN
+    p_end := INSTR( i_list, i_delim, p_start );
+    WHILE p_end > 0 LOOP
+      p_result := p_result + 1;
+      IF ( SUBSTR( i_list, p_start, p_end - p_start ) = i_value )
+      THEN
+        RETURN p_result;
+      END IF;
+      p_start := p_end + c_ld;
+      p_end := INSTR( i_list, i_delim, p_start );
+    END LOOP;
+    IF p_start <= c_len + 1
+       AND SUBSTR( i_list, p_start, c_len - p_start + 1 ) = i_value
+    THEN
+      RETURN p_result + 1;
+    END IF;
+  END IF;
+  RETURN 0;
+END;

data/init/postgresql.sql

@@ -0,0 +1,12 @@
+CREATE OR REPLACE FUNCTION public.find_in_set(n INTEGER, s TEXT)
+ RETURNS BOOLEAN
+ LANGUAGE sql
+AS $function$
+    select bool(int4(z.row_number))
+    from
+    (
+        select row_number() over(), y.x
+        from (select unnest(('{' || $2 || '}')::int[]) as x) as y
+    ) as z
+where z.x = $1
+$function$

data/init/sqlite.sql

@@ -0,0 +1 @@
+

data/lib/arel_extensions.rb

@@ -0,0 +1,84 @@
+require 'arel'
+
+# UnaryOperation|Grouping|Extract < Unary < Arel::Nodes::Node
+# Equality|Regexp|Matches < Binary < Arel::Nodes::Node
+# Count|NamedFunction < Function < Arel::Nodes::Node
+
+# pure Arel internals improvements
+Arel::Nodes::Binary.class_eval do
+  include Arel::AliasPredication
+end
+
+Arel::Nodes::Unary.class_eval do
+  include Arel::Math
+  include Arel::AliasPredication
+  include Arel::Expressions
+end
+
+Arel::Nodes::Grouping.class_eval do
+  include Arel::Math
+  include Arel::AliasPredication
+  include Arel::Expressions
+end
+
+Arel::Nodes::Function.class_eval do
+  include Arel::Math
+  include Arel::Expressions
+end
+
+require 'arel_extensions/version'
+require 'arel_extensions/attributes'
+require 'arel_extensions/visitors'
+require 'arel_extensions/nodes'
+require 'arel_extensions/comparators'
+require 'arel_extensions/date_duration'
+require 'arel_extensions/null_functions'
+require 'arel_extensions/math'
+require 'arel_extensions/math_functions'
+require 'arel_extensions/string_functions'
+
+require 'arel_extensions/insert_manager'
+
+module Arel
+  def self.rand
+    ArelExtensions::Nodes::Rand.new
+  end
+end
+
+Arel::Attributes::Attribute.class_eval do
+  include Arel::Math
+  include ArelExtensions::Attributes
+end
+
+Arel::Nodes::Function.class_eval do
+  include ArelExtensions::Comparators
+  include ArelExtensions::DateDuration
+  include ArelExtensions::MathFunctions
+  include ArelExtensions::StringFunctions
+end
+
+Arel::Nodes::Unary.class_eval do
+  include ArelExtensions::Math
+  include ArelExtensions::Attributes
+  include ArelExtensions::MathFunctions
+  include ArelExtensions::Comparators
+end
+
+Arel::Nodes::Binary.class_eval do
+  include ArelExtensions::Math
+  include ArelExtensions::Attributes
+  include ArelExtensions::MathFunctions
+  include ArelExtensions::Comparators
+end
+
+Arel::Nodes::Equality.class_eval do
+  include ArelExtensions::Comparators
+  include ArelExtensions::DateDuration
+  include ArelExtensions::MathFunctions
+  include ArelExtensions::StringFunctions
+end
+
+
+Arel::InsertManager.class_eval do
+  include ArelExtensions::InsertManager
+end

data/lib/arel_extensions/attributes.rb

@@ -0,0 +1,26 @@
+require 'arel_extensions/comparators'
+require 'arel_extensions/date_duration'
+require 'arel_extensions/math'
+require 'arel_extensions/math_functions'
+require 'arel_extensions/null_functions'
+require 'arel_extensions/string_functions'
+
+module ArelExtensions
+  module Attributes
+	include ArelExtensions::Comparators
+	include ArelExtensions::DateDuration
+	include ArelExtensions::Math
+	include ArelExtensions::MathFunctions
+	include ArelExtensions::NullFunctions
+	include ArelExtensions::StringFunctions
+
+  	def ==(other)
+      Arel::Nodes::Equality.new self, Arel::Nodes.build_quoted(other, self)
+  	end
+
+    def !=(other)
+      Arel::Nodes::NotEqual.new self, Arel::Nodes.build_quoted(other, self)
+    end
+
+  end
+end

data/lib/arel_extensions/comparators.rb

@@ -0,0 +1,59 @@
+module ArelExtensions
+  module Comparators
+
+    def >(other)
+      Arel::Nodes::GreaterThan.new self, Arel::Nodes.build_quoted(other, self)
+    end
+
+    def >=(other)
+      Arel::Nodes::GreaterThanOrEqual.new self, Arel::Nodes.build_quoted(other, self)
+    end
+
+    def <(other)
+      Arel::Nodes::LessThan.new self, Arel::Nodes.build_quoted(other, self)
+    end
+
+    def <=(other)
+      Arel::Nodes::LessThanOrEqual.new self, Arel::Nodes.build_quoted(other, self)
+    end
+
+
+    #REGEXP function
+    #Pattern matching using regular expressions
+    def =~(other)
+#      arg = self.relation.engine.connection.schema_cache.columns_hash(self.relation.table_name)[self.name.to_s].type
+#      if arg == :string || arg == :text
+        Arel::Nodes::Regexp.new self, convert_regexp(other)
+#      end
+    end
+
+    #NOT_REGEXP function
+    #Negation of Regexp
+    def !~(other)
+#      arg = self.relation.engine.connection.schema_cache.columns_hash(self.relation.table_name)[self.name.to_s].type
+#      if arg == :string || arg == :text
+        Arel::Nodes::NotRegexp.new self, convert_regexp(other)
+#      end
+    end
+
+    private
+    #Function use for not_regexp
+    def convert_regexp(other)
+      case other
+      when String
+        #Do nothing
+      when Regexp
+        other = other.source.gsub('\A','^')
+        other.gsub!('\Z','$')
+        other.gsub!('\d','[0-9]')
+        other.gsub!('\D','[^0-9]')
+        other.gsub!('\w','[0-9A-Za-z]')
+        other.gsub!('\W','[^A-Za-z0-9_]')
+      else
+        raise(ArgumentError)
+      end
+      Arel::Nodes.build_quoted(other, self)
+    end
+
+  end
+end

data/lib/arel_extensions/date_duration.rb

@@ -0,0 +1,28 @@
+module ArelExtensions
+    module DateDuration
+      #function returns the year (as a number) given a date value.
+      def year
+        ArelExtensions::Nodes::Duration.new "y",self
+      end
+
+    #function returns the month (as a number) given a date value.
+      def month
+        ArelExtensions::Nodes::Duration.new "m",self
+      end
+
+      #function returns the  week (as a number) given a date value.
+      def week
+        ArelExtensions::Nodes::Duration.new "w",self
+      end
+
+      #function returns the month (as a number) given a date value.
+      def day
+        ArelExtensions::Nodes::Duration.new "d",self
+      end
+
+      def wday
+        ArelExtensions::Nodes::Wday.new self
+      end
+
+    end
+end

data/lib/arel_extensions/insert_manager.rb

@@ -0,0 +1,33 @@
+require 'arel'
+
+module ArelExtensions
+  module InsertManager
+
+	def bulk_insert(cols, data)
+		case cols.first
+		when String, Symbol
+			cols.each { |c|
+				@ast.columns << @ast.relation[c]
+			}
+		when Array
+			if String === cols.first.first
+				@ast.columns = cols.map {|c| [@ast.relation[c.first]] }
+			elsif Arel::Attributes::Attribute == cols.first.first
+				@ast.columns = cols
+			end
+		when NilClass
+			@ast.columns = @ast.relation.columns
+		end
+		self.values = BulkValues.new(@ast.columns, data)
+	end
+
+	class BulkValues < Arel::Nodes::Node
+		attr_accessor :left, :cols
+		def initialize(cols, values)
+	  		@left = values
+	  		@cols = cols
+	  	end
+	end
+
+  end
+end

data/lib/arel_extensions/math.rb

@@ -0,0 +1,48 @@
+module ArelExtensions
+    module Math
+      #function + between
+     #String and others (convert in string)  allows you to concatenate 2 or more strings together.
+     #Date and integer adds or subtracts a specified time interval from a date.
+      def +(other)
+        return ArelExtensions::Nodes::Concat.new(self.expressions + [other]) if self.is_a?(ArelExtensions::Nodes::Concat)
+        arg = self.relation.engine.connection.schema_cache.columns_hash(self.relation.table_name)[self.name.to_s].type
+        if arg == :integer || arg == :decimal || arg == :float
+          if other.is_a?(String)
+            other = other.to_i
+          end
+          Arel::Nodes::Grouping.new(Arel::Nodes::Addition.new self, other)
+        elsif arg == :datetime || arg == :date
+          ArelExtensions::Nodes::DateAdd.new [self, other]
+        elsif arg == :string
+          ArelExtensions::Nodes::Concat.new [self, other]
+        end        
+      end
+
+      #function returns the time between two dates
+      #function returns the susbration between two int
+      def -(other)
+        arg = self.relation.engine.connection.schema_cache.columns_hash(self.relation.table_name)[self.name.to_s].type
+        if (arg == :date || arg == :datetime)
+          case other
+          when Arel::Attributes::Attribute
+            arg2 = other.relation.engine.connection.schema_cache.columns_hash(other.relation.table_name)[other.name.to_s].type
+            if arg2 == :date || arg2 == :datetime
+              ArelExtensions::Nodes::DateDiff.new self, other
+            else
+              ArelExtensions::Nodes::DateSub.new self, other
+            end
+          when Arel::Nodes::Node, DateTime, Time, String, Date
+            ArelExtensions::Nodes::DateDiff.new self, other
+          when Fixnum
+            ArelExtensions::Nodes::DateSub.new self, other
+          end
+        else
+          if other.is_a?(String)
+            other = other.to_i
+          end
+          Arel::Nodes::Grouping.new(Arel::Nodes::Subtraction.new(self, other))
+        end
+      end
+
+    end
+end

data/lib/arel_extensions/math_functions.rb

@@ -0,0 +1,35 @@
+module ArelExtensions
+  module MathFunctions
+
+    # Abs function returns the absolute value of a number passed as argument #
+    def abs
+        ArelExtensions::Nodes::Abs.new [self]
+    end
+
+    # will rounded up any positive or negative decimal value within the function upwards #
+    def ceil
+        ArelExtensions::Nodes::Ceil.new [self]
+    end
+
+    # function rounded up any positive or negative decimal value down to the next least integer
+    def floor
+        ArelExtensions::Nodes::Floor.new [self]
+    end
+
+    #function that can be invoked to produce random numbers between 0 and 1
+#    def rand seed = nil
+#        ArelExtensions::Nodes::Rand.new [seed]
+#    end
+    alias_method :random, :rand
+
+    #function is used to round a numeric field to the number of decimals specified
+    def round precision = nil
+        if precision
+            ArelExtensions::Nodes::Round.new [self, Arel::Nodes.build_quoted(precision)]
+        else
+            ArelExtensions::Nodes::Round.new [self]
+        end
+    end
+
+  end
+end

data/lib/arel_extensions/nodes.rb

@@ -0,0 +1,27 @@
+require 'arel_extensions/nodes/function'
+# Math functions
+require 'arel_extensions/nodes/abs'
+require 'arel_extensions/nodes/ceil'
+require 'arel_extensions/nodes/floor'
+require 'arel_extensions/nodes/round'
+require 'arel_extensions/nodes/rand'
+require 'arel_extensions/nodes/sum'
+
+# String functions
+require 'arel_extensions/nodes/concat' if Arel::VERSION.to_i < 7
+require 'arel_extensions/nodes/length'
+require 'arel_extensions/nodes/locate'
+require 'arel_extensions/nodes/matches'
+require 'arel_extensions/nodes/find_in_set'
+require 'arel_extensions/nodes/replace'
+require 'arel_extensions/nodes/soundex'
+require 'arel_extensions/nodes/trim'
+require 'arel_extensions/nodes/ltrim'
+require 'arel_extensions/nodes/rtrim'
+
+
+require 'arel_extensions/nodes/coalesce'
+require 'arel_extensions/nodes/date_diff'
+require 'arel_extensions/nodes/duration'
+require 'arel_extensions/nodes/isnull'
+require 'arel_extensions/nodes/wday'

data/lib/arel_extensions/nodes/abs.rb

@@ -0,0 +1,6 @@
+module ArelExtensions
+  module Nodes
+    class Abs < Function
+    end
+  end
+end