invisiblellama-repub 0.2.1

data/lib/repub/app/profile.rb

@@ -0,0 +1,91 @@
+require 'delegate'
+
+module Repub
+  class App
+    module Profile
+
+      PROFILE_KEYS = %w[css encoding fixup helper metadata remove rx selectors].map {|k| k.to_sym}
+      
+      def load_profile(name = nil)
+        name ||= 'default'
+        profile = Profile.new
+        profile[name] ||= {}
+        PROFILE_KEYS.each { |key| options[key] = profile[name][key] if profile[name][key] }
+        profile.save
+        profile[name]
+      end
+      
+      def write_profile(name = nil)
+        name ||= 'default'
+        profile = Profile.new
+        profile[name] ||= {}
+        PROFILE_KEYS.each { |key| profile[name][key] = options[key] }
+        profile.save
+        profile[name]
+      end
+      
+      def dump_profile(name = nil)
+        name ||= 'default'
+        profile = Profile.new
+        if p = profile[name]
+          keys = p.keys.map{|k| k.to_s }.sort.map{|k| k.to_sym }
+          keys.each do |k|
+            v = p[k]
+            next if v.nil? || (v.respond_to?(:empty?) && v.empty?)
+            case k
+            when :selectors
+              printf("%4s%-6s\n", '', k)
+              selector_keys = v.keys.map{|k| k.to_s }.sort.map{|k| k.to_sym }
+              selector_keys.each { |sk| printf("%8s%-12s %s\n", '', sk, v[sk]) }
+            when :remove
+              printf("%4s%-6s\n", '', k)
+              v.each { |rk| printf("%20s %s\n", '', rk) }
+            when :rx
+              printf("%4s%-6s\n", '', k)
+              v.each { |rk| printf("%20s %s\n", '', rk) }
+            when :metadata
+              printf("%4s%-6s\n", '', k)
+              metadata_keys = v.keys.map{|k| k.to_s }.sort.map{|k| k.to_sym }
+              metadata_keys.each { |mk| printf("%8s%-12s %s\n", '', mk, v[mk]) }
+            else
+              printf("%4s%-16s %s\n", '', k, v)
+            end
+          end
+        end
+      end
+      
+      def list_profiles
+        profile = Profile.new
+        if profile.empty?
+          puts "No saved profiles."
+        else
+          profile.keys.sort.each do |name|
+            puts "#{name}:"
+            dump_profile(name)
+          end
+        end
+      end
+      
+      class Profile < DelegateClass(Hash)
+        def initialize
+          @profiles = YAML.load_file(Profile.path)
+        rescue
+        ensure
+          @profiles ||= {}
+          super(@profiles)
+        end
+        
+        def self.path
+          File.join(App.data_path, 'profiles')
+        end
+        
+        def save
+          File.open(Profile.path, 'w') do |f|
+            YAML.dump(@profiles, f)
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/repub/app/utility.rb

@@ -0,0 +1,57 @@
+require 'tmpdir'
+
+# Convenience method to avoid long ifs with nil? and empty?
+# If receiver is nil or empty? is _true_, returns _value_
+# Otherwise, returns self
+#
+class Object
+  def if_blank(value)
+    self.nil? || self.empty? ? value : self
+  end
+end
+
+# Windows still has one-click installer based on Ruby 1.8.6
+# Add this for compatibility (from Ruby 1.8.7 tmpdir.rb)
+#
+if not Dir.respond_to? :mktmpdir
+  class << Dir
+    def mktmpdir(prefix_suffix=nil, tmpdir=nil)
+      case prefix_suffix
+      when nil
+        prefix = "d"
+        suffix = ""
+      when String
+        prefix = prefix_suffix
+        suffix = ""
+      when Array
+        prefix = prefix_suffix[0]
+        suffix = prefix_suffix[1]
+      else
+        raise ArgumentError, "unexpected prefix_suffix: #{prefix_suffix.inspect}"
+      end
+      tmpdir ||= Dir.tmpdir
+      t = Time.now.strftime("%Y%m%d")
+      n = nil
+      begin
+        path = "#{tmpdir}/#{prefix}#{t}-#{$$}-#{rand(0x100000000).to_s(36)}"
+        path << "-#{n}" if n
+        path << suffix
+        Dir.mkdir(path, 0700)
+      rescue Errno::EEXIST
+        n ||= 0
+        n += 1
+        retry
+      end
+    
+      if block_given?
+        begin
+          yield path
+        ensure
+          FileUtils.remove_entry_secure path
+        end
+      else
+        path
+      end
+    end
+  end
+end

data/lib/repub/epub.rb

@@ -0,0 +1,3 @@
+require 'repub/epub/container'
+require 'repub/epub/content'
+require 'repub/epub/toc'

data/lib/repub/epub/container.rb

@@ -0,0 +1,28 @@
+require 'rubygems'
+require 'builder'
+
+module Repub
+  module Epub
+  
+  class Container
+    def to_xml
+      out = ''
+      builder = Builder::XmlMarkup.new(:target => out, :indent => 4)
+      builder.instruct!
+      builder.container :xmlns => "urn:oasis:names:tc:opendocument:xmlns:container", :version => "1.0" do
+        builder.rootfiles do
+          builder.rootfile 'full-path' => "content.opf", 'media-type' => "application/oebps-package+xml"
+        end
+      end
+      out
+    end
+  
+    def save(path = 'container.xml')
+      File.open(path, 'w') do |f|
+        f << to_xml
+      end
+    end
+  end
+
+  end
+end

data/lib/repub/epub/content.rb

@@ -0,0 +1,153 @@
+require 'rubygems'
+require 'builder'
+
+module Repub
+  module Epub
+  
+  class Content
+    
+    def initialize(uid)
+      @metadata = Metadata.new('Untitled', 'en', uid, Date.today.to_s)
+      @css_counter = 0
+      @img_counter = 0
+      @html_counter = 0
+      @manifest_items = []
+      @spine_items = []
+      @manifest_items << ContentItem.new('ncx', 'toc.ncx', 'application/x-dtbncx+xml')
+    end
+    
+    class Metadata < Struct.new(
+        :title,
+        :language,
+        :identifier,
+        :date,
+        :subject,
+        :description,
+        :relation,
+        :creator,
+        :publisher,
+        :rights
+      )
+    end
+    
+    attr_reader :metadata
+    
+    def add_page_template(href = 'page-template.xpgt', id = 'pt')
+      @manifest_items << ContentItem.new(id, href, 'application/vnd.adobe-page-template+xml')
+    end
+    
+    def add_stylesheet(href, id = nil)
+      @manifest_items << ContentItem.new(id || "css_#{@css_counter += 1}", href, 'text/css')
+    end
+    
+    def add_image(href, id = nil)
+      image_type = case(href.strip.downcase)
+        when /.*\.(jpeg|jpg)$/
+          'image/jpeg'
+        when /.*\.png$/
+          'image/png'
+        when /.*\.gif$/
+          'image/gif'
+        when /.*\.svg$/
+          'image/svg+xml'
+        else
+          raise 'Unsupported image type'
+      end
+      @manifest_items << ContentItem.new(id || "img_#{@img_counter += 1}", href, image_type)
+    end
+    
+    def add_document(href, id = nil)
+      manifest_item = ContentItem.new(id || "item_#{@html_counter += 1}", href, 'application/xhtml+xml')
+      @manifest_items << manifest_item
+      @spine_items << manifest_item
+    end
+    
+    def to_xml
+      out = ''
+      builder = Builder::XmlMarkup.new(:target => out, :indent => 4)
+      builder.instruct!
+      builder.package :xmlns => "http://www.idpf.org/2007/opf",
+          'unique-identifier' => "dcidid",
+          'version' => "2.0" do
+        metadata_to_xml(builder)
+        manifest_to_xml(@manifest_items, builder)
+        spine_to_xml(@spine_items, builder)
+      end
+      out
+    end
+    
+    def save(path = 'content.opf')
+      File.open(path, 'w') do |f|
+        f << to_xml
+      end
+    end
+    
+    private
+    
+    def metadata_to_xml(builder)
+      builder.metadata 'xmlns:dc' => "http://purl.org/dc/elements/1.1/",
+        'xmlns:dcterms' => "http://purl.org/dc/terms/",
+        'xmlns:xsi' => "http://www.w3.org/2001/XMLSchema-instance",
+        'xmlns:opf' => "http://www.idpf.org/2007/opf" do
+          # Required elements
+          builder.dc :title do
+            builder << @metadata.title
+          end
+          builder.dc :language, 'xsi:type' => "dcterms:RFC3066" do
+            builder << @metadata.language
+          end
+          builder.dc :identifier, :id => 'dcidid', 'opf:scheme' => 'URI' do
+            builder << @metadata.identifier
+          end
+          # Optional elements
+          builder.dc :subject do
+            builder << @metadata.subject
+          end if @metadata.subject
+          builder.dc :description do
+            builder << @metadata.description
+          end if @metadata.description
+          builder.dc :relation do
+            builder << @metadata.relation
+          end if @metadata.relation
+          builder.dc :creator do
+            builder << @metadata.creator
+          end if @metadata.creator
+          builder.dc :publisher do
+            builder << @metadata.publisher
+          end if @metadata.publisher
+          builder.dc :date do
+            builder << @metadata.date.to_s
+          end if @metadata.date
+          builder.dc :rights do
+            builder << @metadata.rights
+          end if @metadata.rights
+      end
+    end
+    
+    class ContentItem < Struct.new(
+        :id,
+        :href,
+        :media_type
+      )
+    end
+    
+    def manifest_to_xml(manifest_items, builder)
+      builder.manifest do
+        manifest_items.each do |i|
+          builder.item :id => i[:id], :href => i[:href], 'media-type' => i[:media_type]
+        end
+      end
+    end
+    
+    def spine_to_xml(spine_items, builder)
+      builder.spine do
+        spine_items.each do |i|
+          builder.itemref :idref => i[:id]
+        end
+      end
+    end
+  
+  end
+  
+  end
+end

data/lib/repub/epub/toc.rb

@@ -0,0 +1,139 @@
+require 'rubygems'
+require 'builder'
+
+module Repub
+  module Epub
+  
+  class Toc
+    
+    def initialize(uid)
+      @head = Head.new(uid)
+      @doc_title = DocTitle.new('Untitled')
+      @nav_map = NavMap.new
+    end
+
+    def title
+      @doc_title.text
+    end
+    
+    def title=(text)
+      @doc_title = DocTitle.new(text)
+    end
+    
+    attr_reader :nav_map
+    
+    def to_xml
+      out = ''
+      builder = Builder::XmlMarkup.new(:target => out, :indent => 4)
+      builder.instruct!
+      builder.declare! :DOCTYPE, :ncx, :PUBLIC, "-//NISO//DTD ncx 2005-1//EN", "http://www.daisy.org/z3986/2005/ncx-2005-1.dtd"
+      builder.ncx :xmlns => "http://www.daisy.org/z3986/2005/ncx/", :version => "2005-1" do
+        @nav_map.calc_depth_and_play_order
+        @head.depth = @nav_map.depth
+        @head.to_xml(builder)
+        @doc_title.to_xml(builder)
+        @nav_map.to_xml(builder)
+      end
+      out
+    end
+    
+    def save(path = 'toc.ncx')
+      File.open(path, 'w') do |f|
+        f << to_xml
+      end
+    end
+    
+    class Head < Struct.new(
+        :uid
+      )
+      
+      attr_accessor :depth
+      
+      def to_xml(builder)
+        builder.head do 
+          builder.meta :name => "dtb:uid", :content => self.uid
+          builder.meta :name => "dtb:depth", :content => @depth
+          builder.meta :name => "dtb:totalPageCount", :content => 0
+          builder.meta :name => "dtb:dtb:maxPageNumber", :content => 0
+        end
+      end
+    end
+    
+    class DocTitle < Struct.new(
+        :text
+      )
+      
+      def to_xml(builder)
+        builder.docTitle do 
+          builder.text self.text
+        end
+      end
+    end
+    
+    class NavMap
+      class NavPoint < Struct.new(
+          :title,
+          :src
+        )
+        
+        def initialize(title, src)
+          super
+          @@last_play_order = 0
+          @play_order = 0
+          @child_points = []
+        end
+        
+        def add_nav_point(title, src)
+          nav_point = NavPoint.new(title, src)
+          @child_points << nav_point
+          nav_point
+        end
+        
+        def to_xml(builder)
+          builder.navPoint :id => "navPoint-#{@play_order}", :playOrder => @play_order do
+            builder.navLabel do
+              builder.text self.title
+            end
+            builder.content :src => self.src
+            @child_points.each { |child_point| child_point.to_xml(builder) }
+          end
+        end
+        
+        def calc_depth_and_play_order(nav_map, depth)
+          nav_map.depth = depth
+          @play_order = @@last_play_order += 1
+          @child_points.each { |child_point| child_point.calc_depth_and_play_order(nav_map, depth + 1) }
+        end
+      end
+      
+      def initialize
+        @nav_points = []
+        @depth = 1
+      end
+      
+      def add_nav_point(title, src)
+        nav_point = NavPoint.new(title, src)
+        @nav_points << nav_point
+        nav_point
+      end
+    
+      attr_reader :depth
+      
+      def depth=(value)
+        @depth = value if value > @depth
+      end
+      
+      def calc_depth_and_play_order
+        @nav_points.each { |nav_point| nav_point.calc_depth_and_play_order(self, 1) }
+      end
+      
+      def to_xml(builder)
+        builder.navMap do
+          @nav_points.each { |nav_point| nav_point.to_xml(builder) }
+        end
+      end
+    end
+  end
+
+  end
+end