X-Git-Url: http://jaekl.net/gitweb/?p=quanlib.git;a=blobdiff_plain;f=main.rb;h=e294b4a15e3391a0e4b84d41235b2eee1265ca31;hp=2d7f68b25c33c6bf45dbb0e909e17e3146f53e5b;hb=HEAD;hpb=b0ed11b1a03aab548e3c274dfd89732e8e11892f
diff --git a/main.rb b/main.rb
index 2d7f68b..e294b4a 100644
--- a/main.rb
+++ b/main.rb
@@ -1,67 +1,56 @@
-require 'walkdir'
-
-outputDir = 'output'
+require_relative 'navigator'
+require_relative 'page'
+require_relative 'store'
+require_relative 'walk_dir'
+
+@outputDir = 'output'
+
+@config_file = 'quanlib.ini'
+@skip_class = false
+
+def handleArg(arg)
+ if arg.start_with?("--config=")
+ @config_file = arg[9..-1]
+ puts 'Using config file "' + @config_file + '".'
+ elsif "--purge" == arg
+ puts 'Purging database...'
+ @store.dropSchema()
+ if File.exists?(@store.unclassified_csv)
+ File.delete(@store.unclassified_csv)
+ end
+ elsif "--skip-class" == arg
+ puts 'Skipping load of classification table.'
+ @skip_class = true
+ elsif arg.start_with?("--")
+ abort('ERROR: Unrecognized option "' + arg + '".')
+ end
+end
-books = []
-imageCount = 0
+@store = Store.new(@config_file)
+@store.connect()
for arg in ARGV
- w = WalkDir.new(arg)
- books += (w.books)
+ handleArg(arg)
end
-if ! Dir.exist?(outputDir)
- Dir.mkdir(outputDir)
+@store.init_db(@skip_class)
+
+for arg in ARGV
+ if ! arg.start_with?("--")
+ puts 'Scanning directory "' + arg + '"...'
+ w = WalkDir.new(@config_file, arg)
+ w.books
+ end
end
-open(outputDir + '/index.html', 'w') do |fd|
- fd.puts ''
- fd.puts ' '
- fd.puts ' '
- fd.puts ' Books'
- fd.puts ' '
- fd.puts ' '
- fd.puts ' '
-
- for book in books
- image = nil
- if nil != book.cover
- imageCount += 1
- (path, mimeType) = book.cover.writeImage(outputDir, 'image' + imageCount.to_s)
- image = '
'
- else
- image = '(No cover image)'
- end
+@store.cross_reference_lists
- fd.puts ' '
- fd.puts ' | ' + image + ' | '
+puts 'Creating output...'
- heading = book.heading()
- description = book.description()
- if nil != description
- fd.puts ' |
'
- else
- fd.puts ' ' + heading + ' | '
- end
-
- fd.puts '