GeorgeKaraszi/ActiveRecordExtended

# frozen_string_literal: true

require "arel/nodes/binary"
require "arel/nodes/function"

module Arel
  module Nodes
    unless ActiveRecordExtended::AR_VERSION_GTE_6_1
      ["Contains", "Overlaps"].each { |binary_node_name| const_set(binary_node_name, Class.new(::Arel::Nodes::Binary)) }
    end

    [
      "ContainsHStore",
      "ContainsArray",
      "ContainedInArray"
    ].each { |binary_node_name| const_set(binary_node_name, Class.new(::Arel::Nodes::Binary)) }

    [
      "RowToJson",
      "JsonBuildObject",
      "JsonbBuildObject",
      "ToJson",
      "ToJsonb",
      "Array",
      "ArrayAgg"
    ].each do |function_node_name|
      func_klass = Class.new(::Arel::Nodes::Function) do
        def initialize(*args)
          super
          return if @expressions.is_a?(::Array)

          @expressions = @expressions.is_a?(::Arel::Nodes::Node) ? [@expressions] : [::Arel.sql(@expressions)]
        end
      end

      const_set(function_node_name, func_klass)
    end

    module Inet
      [
        "Contains",
        "ContainsEquals",
        "ContainedWithin",
        "ContainedWithinEquals",
        "ContainsOrContainedWithin"
      ].each { |binary_node_name| const_set(binary_node_name, Class.new(::Arel::Nodes::Binary)) }
    end
  end
end