diff --git a/router.rb b/router.rb index 32fac0b..2eba6b1 100644 --- a/router.rb +++ b/router.rb @@ -11,7 +11,7 @@ def run filepath = gets.chomp tables = schema_parser(filepath) db = generate_xml_db(tables) - filename = filepath.match(/\/(\w+)\/db\/schema.rb/)[1] + filename = filepath.match(/\/([\w-]+)\/db\/schema.rb/)[1] xml_path = "#{File.dirname(filepath)}/#{filename}.xml" File.open(xml_path, 'wb') { |file| file.write(db) } when 2