Adds support for tracking series and generating pages based on them.

[quanlib.git] / book.rb

diff --git a/book.rb b/book.rb
index 3cf513957c0a2cc4173266857c95952b8cc0bb14..aa61930bc357fb89dcc861085eedd38921b1f654 100644 (file)
--- a/book.rb
+++ b/book.rb
@@ -4,19 +4,24 @@ require 'zip'
 
 require 'author'
 require 'cover'
+require 'store'
 
 class Book
   @@DC_NS_URL = 'http://purl.org/dc/elements/1.1/'
 
-  def initialize(fileName)
+  def initialize(store)
     @author = nil
     @cover = nil
     @description = nil
-    @path = fileName
-    @series = nil
+    @path = nil
+    @series_id = nil
+    @store = store
     @title = nil
     @volume = nil
+  end
 
+  def loadFromFile(fileName)
+    @path = fileName
     parseFileName!(fileName)
   end
 
@@ -35,14 +40,30 @@ class Book
     return false
   end
 
+  def author
+    return @author
+  end
+
+  def author=(value)
+    @author = value
+  end
+
   def cover
     return @cover
   end
 
+  def cover=(value)
+    @cover = value
+  end
+
   def description
     @description
   end
 
+  def description=(value)
+    @description = value
+  end
+
   def heading
     result = []
 
@@ -52,12 +73,13 @@ class Book
       result.push('<i>(Unknown title)</i>')
     end
     if nil != @author
-      result.push('<i>by ' + @author.to_s() + '</i>')
+      result.push('<i>by ' + @author.reading_order + '</i>')
     end
     
     seriesInfo = []
-    if nil != @series
-      seriesInfo.push(@series.to_s)
+    series = @store.load_series(@series_id)
+    if nil != series and nil != series.descr
+      seriesInfo.push(series.descr.to_s)
     end
     if nil != @volume
       seriesInfo.push(@volume.to_s)
@@ -72,10 +94,10 @@ class Book
   def inspect
     data = []
     if nil != @author
-      data.push('author="' + @author.to_s + '"')
+      data.push('author="' + @author.inspect + '"')
     end
-    if nil != @series
-      data.push('series="' + @series + '"')
+    if nil != @series_id
+      data.push('series_id="' + @series_id.to_s() + '"')
     end
     if nil != @volume
       data.push('volume="' + @volume + '"')
@@ -96,30 +118,58 @@ class Book
     @path
   end
 
+  def path=(value)
+    @path = value
+  end
+
+  def series_id
+    @series_id
+  end
+  
+  def series_id=(value)
+    @series_id = value
+  end
+
   def to_s
     return inspect()
   end
 
+  def title
+    @title
+  end
+
+  def title=(value)
+    @title = value
+  end
+
+  def volume
+    @volume
+  end
+
+  def volume=(value)
+    @volume = value
+  end
+
   protected
   def isUpper?(c)
     return /[[:upper:]]/.match(c)
   end
 
   protected
-  def massageAuthor(input)
+  def massage_author(input)
     if nil == input
       return nil
     end
 
-    result = ""
+    reading_order = ""
     input.each_char do |c|
-      if isUpper?(c) and (result.length > 0)
-        result += " "
-      end
-      result += c
+      if isUpper?(c) and (reading_order.length > 0)
+        reading_order += " "
+     end
+      reading_order += c
     end
-    
-    return result
+
+    return reading_order
   end
 
   # Returns (series, volumeNo, titleText)
@@ -156,9 +206,13 @@ class Book
   protected
   def parseFileName!(fileName)
     parts = fileName.split('/')
-    (@series, @volume, @title) = processTitle(parts[-1])
+    (series_code, @volume, @title) = processTitle(parts[-1])
     if parts.length > 1
-      @author = massageAuthor(parts[-2])
+      grouping = parts[-2]
+      reading_order = massage_author(grouping)
+      sort_order = nil
+      @author = Author.new(grouping, reading_order, sort_order)
+      @series_id = @store.get_series(grouping, series_code)
     end
 
     if fileName.downcase.end_with?(".epub")
@@ -173,6 +227,7 @@ class Book
       Zip::File.open(fileName) do |zipfile|
         entry = zipfile.find_entry('META-INF/container.xml')
         if nil == entry
+          puts 'No META-INF/container.xml, skipping book ' + fileName
           return
         end
         contXml = zipfile.read('META-INF/container.xml')
@@ -198,22 +253,25 @@ class Book
     #-------
     # Author
 
+    grouping = @author.grouping
+    reading_order = @author.reading_order
+    sort_order = @author.sort_order
+
     creators = opfDoc.css('dc|creator', 'dc' => @@DC_NS_URL)
     if (creators.length > 0)
       creator = creators[0]
       if nil != creator
         role = creator['opf:role']
         if 'aut' == role
-          name = creator.content
-          parts = name.split(' ')
-          if parts.length > 1
-            surname = parts[-1]
-            givenNames = parts[0..-2].join(' ')
-            @author = Author.new(surname, givenNames)
-          else
-            @author = Author.new(name, '')
+          reading_order = creator.content
+
+          file_as = creator['opf:file-as']
+          if nil != file_as
+            sort_order = file_as
           end
         end
+
+        @author = Author.new(grouping, reading_order, sort_order)
       end
     end
 
@@ -248,22 +306,24 @@ class Book
       content = m['content']
 
       if 'calibre:series' == name
-        @series = content
+        # TODO:  Dynamically create a new series?
+        # @series_id = content
       elsif 'calibre:series-index' == name
         @volume = content
       elsif 'cover' == name
         coverId = content
+        #puts 'File ' + @path + ' coverId ' + coverId
       end
     end
 
     #---------------
     # Load the cover
 
-    @cover = loadCover(zipfile, opfPath, opfDoc, coverId)
+    @cover = load_cover(zipfile, opfPath, opfDoc, coverId)
   end
 
   protected
-  def loadCover(zipfile, opfPath, opfDoc, coverId)
+  def load_cover(zipfile, opfPath, opfDoc, coverId)
     coverFile = nil
     if nil == coverId
       coverId = "cover-image"