+require 'store'
require 'walkdir'
outputDir = 'output'
books = []
imageCount = 0
+def handleArg(arg)
+ if "--purge" == arg
+ puts 'Purging database...'
+ @store.dropSchema()
+ elsif arg.start_with?("--")
+ abort('ERROR: Unrecognized option "' + arg + '".')
+ end
+end
+
+@store = Store.new()
+@store.connect()
+
+for arg in ARGV
+ handleArg(arg)
+end
+
+@store.init_db()
+
for arg in ARGV
- w = WalkDir.new(arg)
- books += (w.books)
+ if ! arg.start_with?("--")
+ puts 'Scanning directory "' + arg + '"...'
+ w = WalkDir.new(@store, arg)
+ books += (w.books)
+ end
end
+puts 'Creating output...'
+
if ! Dir.exist?(outputDir)
Dir.mkdir(outputDir)
end
fd.puts "</html>"
end
+@store.disconnect()