db_utils.rb

require 'data_mapper'
require 'json'
require 'yaml'
require 'pry'
require 'nokogiri'
require 'open-uri'
Dir.glob("models/*.rb").each {|x| require_relative x}

def import_model(model_class)
  infile = File.new("dumped_#{model_class.to_s.downcase}.json")
  json_data = JSON.parse(infile.read)
  count = 0
  json_data.each do |row|
    count += 1
    model_class.create(row)
  end
  p "Importer #{count} instances of #{model_class}"
end

def dump_model(model_class)
  collector = []
  model_class.all.each do |m|
    collector << JSON.parse(m.to_json)
  end
  outfile = File.new("dumped_#{model_class.to_s.downcase}.json", "w")
  outfile.write(collector.to_json)
end

def import(destructive)
  DataMapper.auto_migrate! if destructive
  import_model(Monster)
  import_model(User)
end

def export
  dump_model(User)
  dump_model(Monster)
end

def run_upgrade
  DataMapper.auto_upgrade!
end

def console
  binding.pry
end

def print_usage
  puts "USAGE: ruby db_utils.rb [IMPORT|EXPORT|CONSOLE|MIGRATE]"
end

config = YAML.load(File.read("database_config.yaml"))
DataMapper.setup(:default, config)
DataMapper.finalize
command = ARGV.first
if command.nil?
  print_usage
else
  case command
    when 'migrate'
      run_upgrade
    when 'import'
      import(true)
    when 'append'
      import(false)
    when 'export'
      export
    when 'console'
      console
    else
      print_usage
  end
end