RbotFAQ: rbot-registry-load.rb

#!/usr/bin/env ruby
# load rbot's registry from dump

require 'optparse'
require 'fileutils'

registry_dir = "#{ENV['HOME']}/.rbot/registry"
dump_dir = Dir.pwd
db_load = 'db_load'

OptionParser.new do |opts|
  opts.banner = "Usage: rbot-registry-load.rb [options]"

  opts.on("-r", "--registry-dir DIR", "Path to the registry directory") do |v|
    registry_dir = v
  end

  opts.on("-d", "--dump-dir DIR", "Directory containing the dump") do |v|
    dump_dir = v
  end

  opts.on("-l", "--db-load NAME", "Name of the db_load program") do |v|
    db_load = v
  end

end.parse!

p db_load

FileUtils.cd dump_dir
dirs, files = Dir['**/*'].partition{|f| File.directory?(f)}
dirs.each {|d| FileUtils.mkdir_p("#{registry_dir}/#{d}")}
files.each {|f| `#{db_load} -f #{f} #{registry_dir}/#{$1}` if f =~ /\A(.*\.db)\.dump\Z/}