require 'arel/visitors/compat'

module Arel
  module Visitors
    class Derby < Arel::Visitors::ToSql

      # @private
      STR_1 = ' '

      if ArJdbc::AR42
        def visit_Arel_Nodes_SelectStatement(o, a = nil)
          a = o.cores.inject(a) { |c, x| visit_Arel_Nodes_SelectCore(x, c) }
          unless o.orders.empty?
            a << ' ORDER BY '
            last = o.orders.length - 1
            o.orders.each_with_index do |x, i|
              visit(x, a);  a << ', ' unless last == i
            end
          end
          if o.offset
            a << STR_1; visit(o.offset, a)
          end
          if o.limit
            a << STR_1; visit(o.limit, a)
          end
          if o.lock
            a << STR_1; visit(o.lock, a)
          end
          a
        end
      else
        def visit_Arel_Nodes_SelectStatement(o, a = nil)
          sql = o.cores.map { |x| do_visit(x, a) }.join
          sql << " ORDER BY #{o.orders.map { |x| do_visit x, a }.join(', ')}" unless o.orders.empty?
          sql << " #{do_visit o.offset, a}" if o.offset
          sql << " #{do_visit o.limit, a}" if o.limit
          sql << " #{do_visit o.lock, a}" if o.lock
          sql
        end
      end

      def visit_Arel_Nodes_Limit(o, a = nil)
        limit = "FETCH FIRST #{limit_for(o)} ROWS ONLY"
        a << limit if a
        limit
      end

      def visit_Arel_Nodes_Offset(o, a = nil)
        if a
          a << 'OFFSET '
          do_visit(o.value, a)
          a << ' ROWS'
        else
          "OFFSET #{do_visit o.value, a} ROWS"
        end
      end

      # This generates SELECT...FOR UPDATE, but it will only work if the
      # current transaction isolation level is set to SERIALIZABLE.  Otherwise,
      # locks aren't held for the entire transaction.
      def visit_Arel_Nodes_Lock o, a = nil
        do_visit o.expr, a
      end

      # @private
      VALUES_DEFAULT = 'VALUES ( DEFAULT )' # NOTE: marker set by ArJdbc::Derby

      if ArJdbc::AR42
        def visit_Arel_Nodes_InsertStatement o, a = nil
          a << "INSERT INTO "
          visit(o.relation, a)

          values = o.values

          if o.columns.any?
            cols = o.columns.map { |x| quote_column_name x.name }
            a << ' (' << cols.join(', ') << ') '
          elsif o.values.eql? VALUES_DEFAULT
            cols = o.relation.engine.columns.map { |c| c.name }
            a << ' (' << cols.join(', ') << ')'
            a << ' VALUES '
            a << ' (' << cols.map { 'DEFAULT' }.join(', ') << ')'
            values = false
          end
          visit(values, a) if values
          a
        end
      elsif Arel::VERSION >= '4.0' # AR 4.0 ... AREL 5.0 since AR >= 4.1
        def visit_Arel_Nodes_InsertStatement o, a = nil
          sql = 'INSERT INTO '
          sql << visit(o.relation, a)

          values = o.values

          if o.columns.any?
            cols = o.columns.map { |x| quote_column_name x.name }
            sql << ' (' << cols.join(', ') << ') '
          elsif o.values.eql? VALUES_DEFAULT
            cols = o.relation.engine.columns.map { |c| c.name }
            sql << ' (' << cols.join(', ') << ')'
            sql << ' VALUES '
            sql << ' (' << cols.map { 'DEFAULT' }.join(', ') << ')'
            values = false
          end

          sql << visit(values, a) if values
          sql
        end
      end
    end
  end
end