module ActiveRecord::Import::SQLite3Adapter

def affected_rows
  result = execute('SELECT changes();')
  result.first[0]
end

def insert_many(sql, values, *args) # :nodoc:
  number_of_inserts = 0
  ids = []
  base_sql, post_sql = if sql.is_a?( String )
    [sql, '']
  elsif sql.is_a?( Array )
    [sql.shift, sql.join( ' ' )]
  end
  value_sets = ::ActiveRecord::Import::ValueSetsRecordsParser.parse(values,
    max_records: SQLITE_LIMIT_COMPOUND_SELECT)
  transaction(requires_new: true) do
    value_sets.each do |value_set|
      number_of_inserts += 1
      sql2insert = base_sql + value_set.join( ',' ) + post_sql
      last_insert_id = insert( sql2insert, *args )
      if last_insert_id > 0
        first_insert_id = last_insert_id - affected_rows + 1
        ids.concat((first_insert_id..last_insert_id).to_a)
      end
    end
  end
  [number_of_inserts, ids]
end

def next_value_for_sequence(sequence_name)
  %{nextval('#{sequence_name}')}
end

def pre_sql_statements( options)
  sql = []
  # Options :recursive and :on_duplicate_key_ignore are mutually exclusive
  if (options[:ignore] || options[:on_duplicate_key_ignore]) && !options[:recursive]
    sql << "OR IGNORE"
  end
  sql + super
end

def support_setting_primary_key_of_imported_objects?
  true
end

def supports_import?(current_version = sqlite_version)
  if current_version >= MIN_VERSION_FOR_IMPORT
    true
  else
    false
  end
end