kindlerb 0.1.1 → 1.0.1

data/_document.yml.sample

@@ -1,7 +1,7 @@
-doc_uuid: 
-title:
-author:
-publisher:
-subject: News
-date: 2012-01-04
-
+doc_uuid: 
+title:
+author:
+publisher:
+subject: News
+date: 2012-01-04
+

data/bin/setupkindlerb

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+require 'kindlerb'
+
+Kindlerb.download
+
+puts "Kindlerb is successfully set up!"

data/kindlerb.gemspec

@@ -1,26 +1,25 @@
-# -*- encoding: utf-8 -*-
-$:.push File.expand_path("../lib", __FILE__)
-require "kindlerb"
-
-Gem::Specification.new do |s|
-  s.name        = "kindlerb"
-  s.version     = Kindlerb::VERSION
-  s.platform    = Gem::Platform::RUBY
-  s.required_ruby_version = '>= 1.9.0'
-
-  s.authors     = ["Daniel Choi"]
-  s.email       = ["dhchoi@gmail.com"]
-  s.homepage    = "http://github.com/danchoi/kindlerb"
-  s.summary     = %q{Kindle eperiodical generator}
-  s.description = %q{Kindle eperiodical generator}
-
-  s.rubyforge_project = "kindlerb"
-
-  s.files         = `git ls-files`.split("\n")
-  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
-  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
-  s.require_paths = ["lib"]
-
-  s.add_dependency 'nokogiri'
-  s.add_dependency 'mustache'
-end
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "kindlerb"
+  s.version     = '1.0.1'
+  s.platform    = Gem::Platform::RUBY
+  s.required_ruby_version = '>= 2.0.0'
+
+  s.authors     = ["Daniel Choi", "Emir Aydin"]
+  s.email       = ["dhchoi@gmail.com", "emir@emiraydin.com"]
+  s.homepage    = "http://github.com/danchoi/kindlerb"
+  s.summary     = %q{Kindle eperiodical generator}
+  s.description = %q{Kindle eperiodical generator}
+
+  s.rubyforge_project = "kindlerb"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency 'nokogiri'
+  s.add_dependency 'mustache'
+end

data/lib/kindlerb.rb

@@ -1,146 +1,232 @@
-# encoding: utf-8
-
-unless `which kindlegen` =~ /kindlegen/
-  abort "Please install kindlegen on your path"
-end
-
-# extract nav structure
-
-require 'pathname'
-require 'yaml'
-require 'nokogiri'
-require 'mustache'
-require 'fileutils'
-
-# monkeypatch
-class String
-  def shorten(max)
-    length > max ? Array(self[0,max].split(/\s+/)[0..-2]).join(' ') + '...' : self
-  end
-end
-
-
-module Kindlerb
-  VERSION = '0.1.1'
-
-  def self.run
-
-    target_dir = Pathname.new(ARGV.first || '.')
-
-    opf_template = File.read(File.join(File.dirname(__FILE__), '..', "templates/opf.mustache"))
-    ncx_template = File.read(File.join(File.dirname(__FILE__), '..', "templates/ncx.mustache"))
-    contents_template = File.read(File.join(File.dirname(__FILE__), '..', "templates/contents.mustache"))
-    section_template = File.read(File.join(File.dirname(__FILE__), '..', "templates/section.mustache"))
-    masthead_gif = File.join(File.dirname(__FILE__), '..', "templates/masthead.gif")
-    cover_gif = File.join(File.dirname(__FILE__), '..', "templates/cover-image.gif")
-
-    Dir.chdir target_dir do
-      playorder = 1
-
-      images = []
-      manifest_items = []
-
-      unless File.exist?("_document.yml")
-        
-        puts "Usage: kindlerb [target file directory]"
-
-        abort "Missing _document.yml. Your input file tree is not structured correctly. Please read the README."
-      end
-
-      document = YAML::load_file("_document.yml")  
-
-      document[:spine_items] = []
-      section_html_files = []
-
-      sections = Dir['sections/*'].entries.sort.map.with_index {|section_dir| 
-        c = File.read(Pathname.new(section_dir) + '_section.txt')
-        c.force_encoding("UTF-8")
-        section_title = c.strip
-        articles = Dir[Pathname.new(section_dir) + '*'].entries.select {|x| File.basename(x) !~ /section/}.sort
-        section_html_files << (section_html_file = (Pathname.new(section_dir) + 'section.html').to_s)
-        idref = "item-#{section_dir.gsub(/\D/, '')}"
-
-        document[:spine_items] << {:idref => idref}
-        manifest_items << {
-          :href => section_html_file,
-          :media => "application/xhtml+xml",
-          :idref => idref
-        }
-
-        s = {
-          :path => section_dir,
-          :title => section_title.shorten(40),
-          :playorder => (playorder += 1),
-          :idref => idref,
-          :href => Pathname.new(section_dir) + 'section.html',
-          :articles => articles.map {|article_file|
-                doc = Nokogiri::HTML(File.read(article_file))
-                article_images = doc.search("img").map {|img| 
-                  mimetype =  img[:src] ? "image/#{File.extname(img[:src]).sub('.', '')}" : nil
-                  {:href => img[:src], :mimetype => mimetype}
-                }
-                images.push *article_images
-                title = doc.search("html/head/title").map(&:inner_text).first || "no title"
-                idref = "item-#{article_file.gsub(/\D/, '')}"
-                document[:spine_items] << {:idref => idref}
-                article = {
-                  :file => article_file,
-                  :href => article_file,
-                  :title => title, 
-                  :short_title => title.shorten(60),
-                  :author => doc.search("html/head/meta[@name=author]").map{|n|n[:content]}.first,
-                  :description => doc.search("html/head/meta[@name=description]").map{|n|n[:content]}.first,
-                  :playorder => (playorder += 1),
-                  :idref => idref
-                }
-                manifest_items << {
-                  :href => article[:file],
-                  :media => "application/xhtml+xml",
-                  :idref => article[:idref]
-                }
-                article
-            }
-        }
-
-        # Generate the section html
-        out = Mustache.render section_template, s
-        File.open(section_html_file, "w") {|f| f.puts out}
-        s
-
-      }
-
-      document[:first_article] = sections[0][:articles][0]
-      document['sections'] = sections
-
-
-      document[:manifest_items] = manifest_items + images.map.with_index {|img, idx| 
-        {
-          :href => img[:href],
-          :media => img[:mimetype],
-          :idref => "img-%03d" % idx
-        }
-      } 
-      document[:cover_mimetype] ||= "image/gif"
-     
-      opf = Mustache.render opf_template, document
-      File.open("kindlerb.opf", "w") {|f| f.puts opf}
-      puts "Wrote #{target_dir}/kindlerb.opf"
-
-      # NCX
-      ncx = Mustache.render ncx_template, document
-      File.open("nav-contents.ncx", "w") {|f| f.puts ncx}
-      puts "Wrote #{target_dir}/nav-contents.ncx"
-
-      # contents
-      contents = Mustache.render contents_template, document
-      File.open("contents.html", "w") {|f| f.puts contents}
-      puts "Wrote #{target_dir}/contents.html"
-
-      outfile = document['mobi_outfile']
-      puts "Writing #{outfile}"
-      cmd = "kindlegen -verbose -c2 -o #{outfile} kindlerb.opf && echo 'Wrote MOBI to #{outfile}'"
-      puts cmd
-      exec cmd
-    end
-  end
-end
+# encoding: utf-8
+
+# extract nav structure
+
+require 'pathname'
+require 'yaml'
+require 'nokogiri'
+require 'mustache'
+require 'fileutils'
+
+# monkeypatch
+class String
+  def shorten(max)
+    length > max ? Array(self[0,max].split(/\s+/)[0..-2]).join(' ') + '...' : self
+  end
+end
+
+module Kindlerb
+
+  def self.download
+
+    gem_path = Gem::Specification.find_by_name('kindlerb').gem_dir
+    ext_dir = gem_path + '/ext/'
+    bin_dir = gem_path + '/bin/'
+
+    # Define Kindlegen download files for different OS's
+    executable = 'kindlegen'
+    windows = false
+    compressed_file = case RbConfig::CONFIG['host_os']
+    when /mac|darwin/i
+      extract = 'unzip '
+      "KindleGen_Mac_i386_v2_9.zip"
+    when /linux|cygwin/i
+      extract = 'tar zxf '
+      "kindlegen_linux_2.6_i386_v2_9.tar.gz"
+    when /mingw32/i
+      windows = true
+      extract = 'unzip '
+      executable = 'kindlegen.exe'
+      "kindlegen_win32_v2_9.zip"
+    else
+      STDERR.puts "Host OS is not supported!"
+      exit(1)
+    end
+
+    url = 'http://kindlegen.s3.amazonaws.com/' + compressed_file
+
+    # Download and extract the Kindlegen file into gem's /etc folder
+    unless File.directory?(ext_dir)
+      FileUtils.mkdir_p(ext_dir)
+    end
+    system 'curl ' + url + ' -o ' + ext_dir + compressed_file
+    puts "Kindlegen downloaded: " + ext_dir + compressed_file
+    FileUtils.cd(ext_dir)
+    system extract + compressed_file
+
+    # Move the executable into gem's /bin folder
+    unless File.directory?(bin_dir)
+      FileUtils.mkdir_p(bin_dir)
+    end
+    moved = FileUtils.mv(ext_dir + executable, bin_dir)
+    puts "Kindlegen extracted to: " + bin_dir
+    # Clean up ext folder
+    if moved
+      FileUtils.rm_rf(ext_dir)
+    end
+
+    # Give exec permissions to Kindlegen file
+    exec_file = bin_dir + executable
+    if windows
+      cmd = "icacls #{exec_file} /T /C /grant Everyone:(f)"
+      system cmd
+    else
+      FileUtils.chmod 0754, exec_file
+    end
+    puts "Execution permissions granted to the user for Kindlegen executable"
+
+  end
+
+  # Returns the full path to executable Kindlegen file
+  def self.executable
+
+    gem_path = Gem::Specification.find_by_name('kindlerb').gem_dir
+
+    # Different extensions based on OS
+    kindlegen = case RbConfig::CONFIG['host_os']
+    when /mac|darwin/i
+      "kindlegen"
+    when /linux|cygwin/i
+      "kindlegen"
+    when /mingw32/i
+      "kindlegen.exe"
+    else
+      STDERR.puts "Kindlegen is not installed because host OS is not supported!"
+      exit(1)
+    end
+    
+    exec_path = gem_path + '/bin/' + kindlegen
+
+    return exec_path
+
+  end
+
+  def self.run(target_dir, verbose = false, compression_method = 'c2')
+
+    opf_template = File.read(File.join(File.dirname(__FILE__), '..', "templates/opf.mustache"))
+    ncx_template = File.read(File.join(File.dirname(__FILE__), '..', "templates/ncx.mustache"))
+    contents_template = File.read(File.join(File.dirname(__FILE__), '..', "templates/contents.mustache"))
+    section_template = File.read(File.join(File.dirname(__FILE__), '..', "templates/section.mustache"))
+    masthead_gif = File.join(File.dirname(__FILE__), '..', "templates/masthead.gif")
+    cover_gif = File.join(File.dirname(__FILE__), '..', "templates/cover-image.gif")
+
+    Dir.chdir target_dir do
+      playorder = 1
+
+      images = []
+      manifest_items = []
+
+      unless File.exist?("_document.yml")
+        
+        puts "Usage: kindlerb [target file directory]"
+
+        abort "Missing _document.yml. Your input file tree is not structured correctly. Please read the README."
+      end
+
+      document = YAML::load_file("_document.yml")  
+
+      document[:spine_items] = []
+      section_html_files = []
+
+      sections = Dir['sections/*'].entries.sort.map.with_index {|section_dir| 
+        c = File.read(Pathname.new(section_dir) + '_section.txt')
+        c.force_encoding("UTF-8")
+        section_title = c.strip
+        articles = Dir[Pathname.new(section_dir) + '*'].entries.select {|x| File.basename(x) !~ /section/}.sort
+        section_html_files << (section_html_file = (Pathname.new(section_dir) + 'section.html').to_s)
+        idref = "item-#{section_dir.gsub(/\D/, '')}"
+
+        document[:spine_items] << {:idref => idref}
+        manifest_items << {
+          :href => section_html_file,
+          :media => "application/xhtml+xml",
+          :idref => idref
+        }
+
+        s = {
+          :path => section_dir,
+          :title => section_title.shorten(40),
+          :playorder => (playorder += 1),
+          :idref => idref,
+          :href => Pathname.new(section_dir) + 'section.html',
+          :articles => articles.map {|article_file|
+                doc = Nokogiri::HTML(File.read(article_file, :encoding => 'UTF-8'))
+                article_images = doc.search("img").map {|img| 
+                  mimetype =  img[:src] ? "image/#{File.extname(img[:src]).sub('.', '')}" : nil
+                  {:href => img[:src], :mimetype => mimetype}
+                }
+                images.push *article_images
+                title = doc.search("html/head/title").map(&:inner_text).first || "no title"
+                idref = "item-#{article_file.gsub(/\D/, '')}"
+                document[:spine_items] << {:idref => idref}
+                article = {
+                  :file => article_file,
+                  :href => article_file,
+                  :title => title, 
+                  :short_title => title.shorten(60),
+                  :author => doc.search("html/head/meta[@name=author]").map{|n|n[:content]}.first,
+                  :description => doc.search("html/head/meta[@name=description]").map{|n|n[:content]}.first,
+                  :playorder => (playorder += 1),
+                  :idref => idref
+                }
+                manifest_items << {
+                  :href => article[:file],
+                  :media => "application/xhtml+xml",
+                  :idref => article[:idref]
+                }
+                article
+            }
+        }
+
+        # Generate the section html
+        out = Mustache.render section_template, s
+        File.open(section_html_file, "w") {|f| f.puts out}
+        s
+
+      }
+
+      document[:first_article] = sections[0][:articles][0]
+      document['sections'] = sections
+
+
+      document[:manifest_items] = manifest_items + images.map.with_index {|img, idx| 
+        {
+          :href => img[:href],
+          :media => img[:mimetype],
+          :idref => "img-%03d" % idx
+        }
+      } 
+      document[:cover_mimetype] ||= "image/gif"
+     
+      opf = Mustache.render opf_template, document
+      File.open("kindlerb.opf", "w") {|f| f.puts opf}
+      puts "Wrote #{target_dir}/kindlerb.opf"
+
+      # NCX
+      ncx = Mustache.render ncx_template, document
+      File.open("nav-contents.ncx", "w") {|f| f.puts ncx}
+      puts "Wrote #{target_dir}/nav-contents.ncx"
+
+      # contents
+      contents = Mustache.render contents_template, document
+      File.open("contents.html", "w") {|f| f.puts contents}
+      puts "Wrote #{target_dir}/contents.html"
+
+      outfile = document['mobi_outfile']
+      puts "Writing #{outfile}"
+      cmd = self.executable + "#{' -verbose' if verbose} -#{compression_method} -o #{outfile} kindlerb.opf && echo 'Wrote MOBI to #{outfile}'"
+      puts cmd
+      system cmd
+
+      # If the system call returns anything other than nil, the call was successful
+      # Because Kindlegen completes build successfully with warnings
+      successful = $?.exitstatus.nil? ? false : true
+      if successful
+        return true
+      else
+        return false
+      end
+
+    end
+  end
+end