class Fixpoint

# File lib/fixpoint.rb, line 36
def exists?(fixname)
  File.exist?(fixpoint_path(fixname))
end

# File lib/fixpoint.rb, line 64
def fixpoint_path(fixname)
  fspath = self.fixpoints_path
  raise Fixpoint::Error, 'Can not automatically infer the base path for the specs, please set `rspec_config.fixpoints_path` explicitly' if fspath.nil?
  raise Fixpoint::Error, "Please create the fixpoints folder (and maybe create a .gitkeep): #{fspath}" if !File.exist?(fspath)

  File.join(fspath, "#{fixname}.yml")
end

# File lib/fixpoint.rb, line 48
def from_database(conn)
  new(read_database_records(conn))
end

# File lib/fixpoint.rb, line 40
def from_file(fixname)
  raise Fixpoint::Error, "The requested fixpoint (\"#{fixname}\") could not be found. Re-run the test which stores the fixpoint." unless exists?(fixname)

  file_path = fixpoint_path(fixname)
  new(YAML.load_file(file_path))
end

# File lib/fixpoint.rb, line 106
def initialize(records_in_tables)
  @records_in_tables = records_in_tables
end

# File lib/fixpoint.rb, line 52
def remove(fixname)
  FileUtils.rm_f(fixpoint_path(fixname))
end

# File lib/fixpoint.rb, line 59
def reset_pk_sequences!(conn)
  return unless conn.respond_to?(:reset_pk_sequence!)
  conn.tables.each { |table_name| conn.reset_pk_sequence!(table_name) }
end

# File lib/fixpoint.rb, line 74
def fixpoints_path
  return RSpec.configuration.fixpoints_path unless RSpec.configuration.fixpoints_path.nil?
  return Rails.root.join(RSpec.configuration.default_path, FIXPOINT_FOLDER) if defined?(Rails)
  # now this is ugly, but necessary. we go up from the current example's path until we find the spec folder...
  return nil if RSpec.current_example.nil?
  spec_path = Pathname.new(RSpec.current_example.file_path).ascend.find { |pn| pn.basename.to_s == RSpec.configuration.default_path }.expand_path

  File.join(spec_path, FIXPOINT_FOLDER)
end

# File lib/fixpoint.rb, line 84
def read_database_records(conn)
  # adapted from: https://yizeng.me/2017/07/16/generate-rails-test-fixtures-yaml-from-database-dump/
  tables = conn.tables
  tables.reject! { |table_name| TABLES_TO_SKIP.include?(table_name) }

  tables.each_with_object({}) do |table_name, acc|
    result = conn.select_all("SELECT * FROM #{conn.quote_table_name(table_name)}")
    next if result.count.zero?

    rows = result.to_a
    rows.sort_by! { |row| row['id'] } if result.columns.include?('id') # let's make the order of items stable
    # fix jsonb columns by re-parsing them, so they are not saved as string to the yaml file
    jsonb_columns = result.column_types.select { |_, col_type| col_type.type == :jsonb }.collect { |col_name, _| col_name }
    rows.collect! { |row| jsonb_columns.each {|jcol| row[jcol] = JSON.parse(row[jcol] || 'null') }; row }

    acc[table_name] = rows
  end
end

# File lib/fixpoint.rb, line 110
def load_into_database(conn)
  # Here some more pointers on implementation details of fixtures:
  # - https://github.com/rails/rails/blob/2998672fc22f0d5e1a79a29ccb60d0d0e627a430/activerecord/lib/active_record/fixtures.rb#L612
  # - http://api.rubyonrails.org/v5.2.4/classes/ActiveRecord/FixtureSet.html#method-c-create_fixtures
  # - https://github.com/rails/rails/blob/67feba0c822d64741d574dfea808c1a2feedbcfc/activerecord/test/cases/fixtures_test.rb
  #
  # Note from the past (useful if we want to get back to using Rails' +create_fixtures+ method)
  # we used to do: ActiveRecord::FixtureSet.create_fixtures(folder_path, filename_without_extension) # this will also clear the table
  # but we abandoned this approach because we want to one file per fixpoint (not one file per table)
  # ActiveRecord::FixtureSet.reset_cache # create_fixtures does use only the table name as cache key. we always invalidate the cache because we may want to read different fixpoints but with the same table names

  # let's remove all data
  conn.tables.each { |table| conn.select_all("DELETE FROM #{conn.quote_table_name(table)}") }

  # actually insert
  conn.insert_fixtures_set(@records_in_tables)
  self.class.reset_pk_sequences!(conn)
end

# File lib/fixpoint.rb, line 143
def records_for_table(table_name, ignore_columns = [])
  strip_columns_from_records(@records_in_tables[table_name], table_name, ignore_columns)
end

# File lib/fixpoint.rb, line 129
def save_to_file(fixname)
  file_path = self.class.fixpoint_path(fixname)
  FileUtils.mkdir_p(File.dirname(file_path))
  File.write(file_path, contents_for_file)
end

# File lib/fixpoint.rb, line 135
def table_names
  @records_in_tables.keys
end

# File lib/fixpoint.rb, line 149
def contents_for_file
  YAML.dump(@records_in_tables)
end

# File lib/fixpoint.rb, line 154
def strip_columns_from_records(records, table_name, columns)
  return nil if records.nil?

  if columns.last.is_a?(Hash) # columns has the a table names at the end (e.g. [:created_at, :updated_at, users: [:password_hash]])
    columns = columns.dup
    all_table_scoped = columns.pop.stringify_keys
    table_scoped = all_table_scoped[table_name]
    columns += table_scoped if table_scoped
  end
  columns = columns.collect(&:to_s)

  records.collect do |attributes|
    attributes.reject { |col, _value| columns.include?(col) }
  end
end