module Sequel::MSSQL::DatasetMethods

# File lib/sequel/adapters/shared/mssql.rb, line 472
def complex_expression_sql_append(sql, op, args)
  case op
  when :'||'
    super(sql, :+, args)
  when :LIKE, :"NOT LIKE"
    super(sql, op, args.map{|a| LiteralString.new("(#{literal(a)} COLLATE #{CASE_SENSITIVE_COLLATION})")})
  when :ILIKE, :"NOT ILIKE"
    super(sql, (op == :ILIKE ? :LIKE : :"NOT LIKE"), args.map{|a| LiteralString.new("(#{literal(a)} COLLATE #{CASE_INSENSITIVE_COLLATION})")})
  when :<<
    sql << complex_expression_arg_pairs(args){|a, b| "(#{literal(a)} * POWER(2, #{literal(b)}))"}
  when :>>
    sql << complex_expression_arg_pairs(args){|a, b| "(#{literal(a)} / POWER(2, #{literal(b)}))"}
  when :extract
    part = args.at(0)
    raise(Sequel::Error, "unsupported extract argument: #{part.inspect}") unless format = EXTRACT_MAP[part]
    if part == :second
      expr = literal(args.at(1))
      sql << DATEPART_SECOND_OPEN << format.to_s << COMMA << expr << DATEPART_SECOND_MIDDLE << expr << DATEPART_SECOND_CLOSE
    else
      sql << DATEPART_OPEN << format.to_s << COMMA
      literal_append(sql, args.at(1))
      sql << PAREN_CLOSE
    end
  else
    super
  end
end

# File lib/sequel/adapters/shared/mssql.rb, line 501
def constant_sql_append(sql, constant)
  if c = CONSTANT_MAP[constant]
    sql << c
  else
    super
  end
end

# File lib/sequel/adapters/shared/mssql.rb, line 510
def cross_apply(table)
  join_table(:cross_apply, table)
end

# File lib/sequel/adapters/shared/mssql.rb, line 515
def disable_insert_output
  clone(:disable_insert_output=>true)
end

# File lib/sequel/adapters/shared/mssql.rb, line 528
def emulated_function_sql_append(sql, f)
  case f.f
  when :char_length
    literal_append(sql, SQL::Function.new(:len, Sequel.join([f.args.first, 'x'])) - 1)
  when :trim
    literal_append(sql, SQL::Function.new(:ltrim, SQL::Function.new(:rtrim, f.args.first)))
  else
    super
  end
end

# File lib/sequel/adapters/shared/mssql.rb, line 520
def escape_like(string)
  string.gsub(/[\%_\[\]]/){|m| "\\#{m}"}
end

# File lib/sequel/adapters/shared/mssql.rb, line 540
def full_text_search(cols, terms, opts = OPTS)
  terms = "\"#{terms.join('" OR "')}\"" if terms.is_a?(Array)
  filter("CONTAINS (?, ?)", cols, terms)
end

# File lib/sequel/adapters/shared/mssql.rb, line 546
def insert_select(*values)
  return unless supports_insert_select?
  naked.clone(default_server_opts(:sql=>output(nil, [SQL::ColumnAll.new(:inserted)]).insert_sql(*values))).single_record
end

# File lib/sequel/adapters/shared/mssql.rb, line 552
def into(table)
  clone(:into => table)
end

# File lib/sequel/adapters/shared/mssql.rb, line 467
def mssql_unicode_strings
  defined?(@mssql_unicode_strings) ? @mssql_unicode_strings : (@mssql_unicode_strings = db.mssql_unicode_strings)
end

# File lib/sequel/adapters/shared/mssql.rb, line 557
def multi_insert_sql(columns, values)
  c = false
  sql = LiteralString.new('')
  u = UNION_ALL
  values.each do |v|
    sql << u if c
    sql << SELECT_SPACE
    expression_list_append(sql, v)
    c ||= true
  end
  [insert_sql(columns, sql)]
end

# File lib/sequel/adapters/shared/mssql.rb, line 571
def nolock
  lock_style(:dirty)
end

# File lib/sequel/adapters/shared/mssql.rb, line 576
def outer_apply(table)
  join_table(:outer_apply, table)
end

# File lib/sequel/adapters/shared/mssql.rb, line 592
def output(into, values)
  raise(Error, "SQL Server versions 2000 and earlier do not support the OUTPUT clause") unless supports_output_clause?
  output = {}
  case values
    when Hash
      output[:column_list], output[:select_list] = values.keys, values.values
    when Array
      output[:select_list] = values
  end
  output[:into] = into
  clone({:output => output})
end

# File lib/sequel/adapters/shared/mssql.rb, line 606
def quoted_identifier_append(sql, name)
  sql << BRACKET_OPEN << name.to_s.gsub(/\]/, DOUBLE_BRACKET_CLOSE) << BRACKET_CLOSE
end

# File lib/sequel/adapters/shared/mssql.rb, line 611
def server_version
  db.server_version(@opts[:server])
end

# File lib/sequel/adapters/shared/mssql.rb, line 616
def supports_group_cube?
  is_2005_or_later?
end

# File lib/sequel/adapters/shared/mssql.rb, line 621
def supports_group_rollup?
  is_2005_or_later?
end

# File lib/sequel/adapters/shared/mssql.rb, line 626
def supports_insert_select?
  supports_output_clause? && !opts[:disable_insert_output]
end

# File lib/sequel/adapters/shared/mssql.rb, line 631
def supports_intersect_except?
  is_2005_or_later?
end

# File lib/sequel/adapters/shared/mssql.rb, line 636
def supports_is_true?
  false
end

# File lib/sequel/adapters/shared/mssql.rb, line 641
def supports_join_using?
  false
end

# File lib/sequel/adapters/shared/mssql.rb, line 646
def supports_modifying_joins?
  is_2005_or_later?
end

# File lib/sequel/adapters/shared/mssql.rb, line 651
def supports_multiple_column_in?
  false
end

# File lib/sequel/adapters/shared/mssql.rb, line 656
def supports_output_clause?
  is_2005_or_later?
end

# File lib/sequel/adapters/shared/mssql.rb, line 666
def supports_where_true?
  false
end

# File lib/sequel/adapters/shared/mssql.rb, line 661
def supports_window_functions?
  true
end