rpub 0.4.0 → 0.5.0

data/lib/rpub/command.rb

@@ -0,0 +1,32 @@
+module Rpub
+  class Command
+    attr_reader :args, :options
+
+    def initialize(args, options)
+      @args, @options = args, options
+      run
+    end
+
+    protected
+
+    def run
+      raise 'Must be implemented by subclass'
+    end
+
+    def book
+      @book ||= Book.new(context)
+    end
+
+    def context
+      @context ||= Context.new({
+        :layout => options.layout,
+        :styles => options.styles,
+        :config => options.config
+      })
+    end
+
+    def source
+      @source ||= Rpub.source
+    end
+  end
+end

data/lib/rpub/commands/clean.rb

@@ -1,54 +1,9 @@
 module Rpub
   module Commands
-    class Clean < Base
-      include CompilationHelpers
-
-      identifier 'clean'
-
-      def invoke
-        super
-        remove create_book.filename
-        remove 'preview.html'
-      end
-
-    private
-
-      def parser
-        OptionParser.new do |opts|
-          opts.banner = <<-EOS
-Usage: rpub clean [-d]'
-
-Clean up all generated files, such as the standard generated .epub
-file, package files and preview files.
-
-Options:
-EOS
-
-          opts.separator ''
-
-          opts.on '-d', '--dry-run', 'Dry-run: only list files to be removed' do
-            @dry_run = true
-          end
-
-          opts.separator ''
-          opts.separator 'Generic options:'
-          opts.separator ''
-
-          opts.on_tail '-h', '--help', 'Display this message' do
-            puts opts
-            exit
-          end
-        end
-      end
-
-      def remove(filename)
-        if File.exist?(filename)
-          unless @dry_run
-            File.unlink(filename)
-          else
-            puts filename
-          end
-        end
+    class Clean < Command
+      def run
+        source.remove book.filename, options.dryrun
+        source.remove 'preview.html', options.dryrun
       end
     end
   end

data/lib/rpub/commands/compile.rb

@@ -1,54 +1,9 @@
 module Rpub
   module Commands
-    class Compile < Base
-      include CompilationHelpers
-
-      identifier 'compile'
-
-      def invoke
-        super
-        book = create_book
-        Compressor.open(book.filename) do |zip|
-          Epub.new(book, File.read(styles)).manifest_in(zip)
-        end
-      end
-
-    private
-
-      def parser
-        OptionParser.new do |opts|
-          opts.banner = <<-EOS
-Usage: rpub compile [options]
-
-Compile your Markdown-formatted input files to a valid .epub output
-file using the options described in config.yml. This will use the
-layout.html and styles.css files in your project directory if
-present.
-
-Options:
-EOS
-          opts.separator ''
-
-          opts.on '-l', '--layout FILENAME', 'Specify an explicit layout file to use' do |filename|
-            @layout = filename
-          end
-
-          opts.on '-s', '--styles FILENAME', 'Specify an explicit stylesheet file to use' do |filename|
-            @styles = filename
-          end
-
-          opts.on '-c', '--config FILENAME', 'Specify an explicit configuration file to use' do |filename|
-            @config_file = filename
-          end
-
-          opts.separator ''
-          opts.separator 'Generic options:'
-          opts.separator ''
-
-          opts.on_tail '-h', '--help', 'Display this message' do
-            puts opts
-            exit
-          end
+    class Compile < Command
+      def run
+        Rpub::Compressor.open(book.filename) do |zip|
+          Rpub::Epub.new(book, source.read(context.styles)).manifest_in(zip)
         end
       end
     end

data/lib/rpub/commands/generate.rb

@@ -1,78 +1,21 @@
 module Rpub
   module Commands
-    class Generate < Base
-      identifier 'generate'
-
-      def initialize(*args)
-        super
-        @all = true
-      end
-
-      def invoke
-        super
-
-        if (@styles.nil? && @all) || (!@styles.nil? && @styles)
-          write_file Rpub.support_file('styles.css')
+    class Generate < Command
+      def run
+        all = [options.styles, options.layout, options.config].all? do |b|
+          b.nil? || b == false
         end
 
-        if (@layout.nil? && @all) || (!@layout.nil? && @layout)
-          write_file Rpub.support_file('layout.html')
+        if (options.styles.nil? && all) || (!options.styles.nil? && options.styles)
+          source.write 'styles.css', source.read(Rpub.support_file('styles.css'))
         end
 
-        if (@config.nil? && @all) || (!@config.nil? && @config)
-          write_file Rpub.support_file('config.yml')
-        end
-      end
-
-    private
-
-      def write_file(file)
-        output_file = File.basename(file)
-        if File.exist?(output_file)
-          puts "Not overriding #{output_file}"
-          return
-        end
-        File.open(output_file, 'w') do |f|
-          f.write File.read(file)
+        if (options.layout.nil? && all) || (!options.layout.nil? && options.layout)
+          source.write 'layout.html', source.read(Rpub.support_file('layout.html'))
         end
-      end
-
-      def parser
-        OptionParser.new do |opts|
-          opts.banner = <<-EOS
-Usage: rpub generate [-slach]
-
-Generate one or more standard files to get started with a new project.  By
-default an entire skeleton project is generated, but by passing the -s, -l, -c
-options you can generate just a single file.
-
-Options:
-EOS
-          opts.separator ' '
-
-          opts.on '-s', '--[no-]styles', 'Generate default stylesheet' do |v|
-            @all = false if v
-            @styles = v
-          end
-
-          opts.on '-l', '--[no-]layout', 'Generate default HTML layout' do |v|
-            @all = false if v
-            @layout = v
-          end
-
-          opts.on '-c', '--[no-]config', 'Generate default configuration' do |v|
-            @all = false if v
-            @config = v
-          end
-
-          opts.separator ''
-          opts.separator 'Generic options:'
-          opts.separator ''
 
-          opts.on_tail '-h', '--help', 'Display this message' do
-            puts opts
-            exit
-          end
+        if (options.config.nil? && all) || (!options.config.nil? && options.config)
+          source.write 'config.yml', source.read(Rpub.support_file('config.yml'))
         end
       end
     end

data/lib/rpub/commands/package.rb

@@ -1,42 +1,14 @@
 module Rpub
   module Commands
-    class Package < Base
-      include CompilationHelpers
-
-      identifier 'package'
-
-      def invoke
-        super
-        Compile.new(options).invoke
-        return unless config.has_key?('package_file')
-        Compressor.open(config.fetch('package_file')) do |zip|
-          zip.store_file create_book.filename, File.read(create_book.filename)
-          config.fetch('package') { [] }.each do |file|
-            zip.compress_file file, File.read(file)
-          end
-        end
-      end
-
-    private
-
-      def parser
-        OptionParser.new do |opts|
-          opts.banner = <<-EOS
-Usage: rpub package
-
-Compile your ebook to an ePub file and package it into an archive together with
-optional other files for easy distibution. You might want to include a README
-file, a license or other promotion material.
-
-Options:
-EOS
-          opts.separator ''
-          opts.separator 'Generic options:'
-          opts.separator ''
-
-          opts.on_tail '-h', '--help', 'Display this message' do
-            puts opts
-            exit
+    class Package < Command
+      def run
+        Compile.new(args, options)
+        if context.config.package_file
+          Rpub::Compressor.open(context.config.package_file) do |zip|
+            zip.store_file book.filename, source.read(book.filename)
+            Array(context.config.package).each do |file|
+              zip.compress_file file, source.read(file)
+            end
           end
         end
       end

data/lib/rpub/commands/preview.rb

@@ -1,59 +1,9 @@
 module Rpub
   module Commands
-    class Preview < Base
-      include CompilationHelpers
-
-      identifier 'preview'
-
-      def initialize(*args)
-        super
-        @filename = 'preview.html'
-      end
-
-      def invoke
-        super
-        return unless markdown_files.any?
-        File.open(@filename, 'w') do |f|
-          f.write move_styles_inline(Typogruby.improve(concatenated_document.to_html))
-        end
-      end
-
-    private
-
-      def move_styles_inline(html)
-        style_block = %Q{<style>\n#{File.read(styles)}\n</style>}
-        html.gsub %r{</head>}, style_block + "\n</head>"
-      end
-
-      def parser
-        OptionParser.new do |opts|
-          opts.banner = <<-EOS
-Usage: rpub preview [options]
-
-Generate a single-page HTML file for easy previewing of your content with the
-layout and styles used when generating .epub files. By default, the output
-file will be named "preview.html".
-
-Options:
-EOS
-          opts.separator ''
-          opts.on '-l', '--layout FILENAME', 'Specify an explicit layout file to use' do |filename|
-            @layout = filename
-          end
-
-          opts.on '-o', '--output FILENAME', 'Specify an explicit output file' do |filename|
-            @filename = filename
-          end
-
-          opts.separator ''
-          opts.separator 'Generic options:'
-          opts.separator ''
-
-          opts.on_tail '-h', '--help', 'Display this message' do
-            puts opts
-            exit
-          end
-        end
+    class Preview < Command
+      def run
+        options.default :output => 'preview.html'
+        source.force_write options.output, Rpub::Preview.new(context, source).formatted
       end
     end
   end

data/lib/rpub/commands/stats.rb

@@ -1,18 +1,14 @@
 module Rpub
   module Commands
-    class Stats < Base
-      identifier 'stats'
-      include CompilationHelpers
-
-      def invoke
-        super
-        text = Nokogiri::HTML(concatenated_document.to_html).xpath('//text()').to_s
+    class Stats < Command
+      def run
+        text = Rpub::Preview.new(context, source).text
         puts "#{text.words.size} words"
-        puts "#{text.words.size / 500} pages"
+        puts "#{(text.words.size.to_f / 500).ceil} pages"
         puts "#{text.sentences} sentences"
         puts "#{text.avg_sentence_length} avg sentence length"
-        puts "#{text.ari} ari"
-        puts "#{text.clf} clf"
+        puts format("%.2f ari", text.ari)
+        puts format("%.2f clf", text.clf)
       end
     end
   end

data/lib/rpub/compressor.rb

@@ -17,7 +17,7 @@ module Rpub
 
     # @param [String] filename of the archive to write to disk
     def initialize(filename)
-      @zip = Zip::ZipOutputStream.new(filename)
+      @zip = Zip::OutputStream.new(filename)
     end
 
     # Close the zip stream and write the file to disk.
@@ -30,7 +30,7 @@ module Rpub
     # @param [String] filename under the which the data should be stored
     # @param [#to_s] content to be compressed
     def store_file(filename, content)
-      zip.put_next_entry filename, nil, nil, Zip::ZipEntry::STORED, Zlib::NO_COMPRESSION
+      zip.put_next_entry filename, nil, nil, Zip::Entry::STORED, Zlib::NO_COMPRESSION
       zip.write content.to_s
     end
 
@@ -39,7 +39,7 @@ module Rpub
     # @param [String] filename under the which the data should be stored
     # @param [#to_s] content to be compressed
     def compress_file(filename, content)
-      zip.put_next_entry filename, nil, nil, Zip::ZipEntry::DEFLATED, Zlib::BEST_COMPRESSION
+      zip.put_next_entry filename, nil, nil, Zip::Entry::DEFLATED, Zlib::BEST_COMPRESSION
       zip.write content.to_s
     end
   end

data/lib/rpub/context.rb

@@ -0,0 +1,48 @@
+module Rpub
+  class Context
+    include Rpub.source
+
+    def initialize(options = {})
+      @config_file = options[:config]
+      @layout      = options[:layout]
+      @styles      = options[:styles]
+    end
+
+    def config
+      @config ||= OpenStruct.new(YAML.load_file(config_file) || {})
+    end
+
+    # @return [String] path to the current layout file (defaulting to built-in)
+    def layout
+      @layout ||= own_or_support_file('layout.html')
+    end
+
+    # @return [String] path to the current stylesheet file (defaulting to built-in)
+    def styles
+      @styles ||= own_or_support_file('styles.css')
+    end
+
+    def fonts
+      @fonts ||= read(styles).scan(/url\((?:'|")?([^'")]+\.[ot]tf)(?:'|")?\)/i).flatten
+    end
+
+    # All chapter input files loaded into strings. This does not include any of
+    # the files listed in the +ignore+ configuration key.
+    #
+    # @return [Array<String>]
+    def chapter_files
+      @chapter_files ||= filter_exceptions(source_files).sort.map(&method(:read))
+    end
+
+    private
+
+    def config_file
+      @config_file ||= own_or_support_file('config.yml')
+    end
+
+    def filter_exceptions(filenames)
+      return filenames unless config.ignore
+      filenames.reject(&config.ignore.method(:include?))
+    end
+  end
+end