2017-09-17 04:30:34 +00:00
|
|
|
module Nokogiri
|
|
|
|
module XML
|
|
|
|
# Patch to add class?
|
|
|
|
class Node
|
|
|
|
def class?(*classes)
|
|
|
|
present = false
|
2017-09-27 16:44:55 +00:00
|
|
|
if attribute('class')
|
2017-09-17 04:30:34 +00:00
|
|
|
present = true
|
|
|
|
classes.each do |klass|
|
|
|
|
present &&= self['class'].include? klass
|
|
|
|
end
|
|
|
|
end
|
|
|
|
present
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2017-09-15 06:10:06 +00:00
|
|
|
# https://stackoverflow.com/a/42533209/368328
|
2017-11-12 21:16:21 +00:00
|
|
|
def command?(cmd)
|
|
|
|
exts = ENV['PATHEXT'] ? ENV['PATHEXT'].split(';') : ['']
|
|
|
|
ENV['PATH'].split(File::PATH_SEPARATOR).each do |path|
|
|
|
|
exts.each { |ext|
|
|
|
|
exe = File.join(path, "#{cmd}#{ext}")
|
|
|
|
return exe if File.executable?(exe) && !File.directory?(exe)
|
|
|
|
}
|
|
|
|
end
|
|
|
|
return nil
|
2017-09-15 06:10:06 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def commands?(commands)
|
2017-09-17 04:30:34 +00:00
|
|
|
commands.map { |c| command? c }
|
2017-09-15 06:10:06 +00:00
|
|
|
end
|
|
|
|
|
2017-09-26 16:52:47 +00:00
|
|
|
def format_match(format, format_to_match)
|
|
|
|
[:all, format_to_match].include? format
|
2017-09-15 06:10:06 +00:00
|
|
|
end
|
|
|
|
|
2017-09-27 16:44:55 +00:00
|
|
|
def gen_epub(name, format)
|
|
|
|
return unless format_match(format, :epub)
|
|
|
|
begin
|
|
|
|
require 'paru/pandoc'
|
|
|
|
Paru::Pandoc.new do
|
|
|
|
from 'html'
|
|
|
|
to 'epub'
|
|
|
|
epub_metadata "metadata/#{name}.xml"
|
|
|
|
epub_cover_image "covers/#{name}.jpg"
|
2017-11-06 21:30:20 +00:00
|
|
|
data_dir Dir.pwd
|
2017-09-27 16:44:55 +00:00
|
|
|
output "books/#{name}.epub"
|
|
|
|
end.convert File.read("books/#{name}.html")
|
|
|
|
puts '[epub] Generated EPUB file'
|
|
|
|
rescue LoadError
|
|
|
|
puts "[error] Can't generate EPUB without paru"
|
2017-09-15 06:10:06 +00:00
|
|
|
end
|
2017-09-17 04:30:34 +00:00
|
|
|
end
|
2017-09-15 06:10:06 +00:00
|
|
|
|
2017-09-27 16:44:55 +00:00
|
|
|
def gen_mobi(name, format)
|
|
|
|
if command?('ebook-convert') && format_match(format, :mobi)
|
2017-09-15 06:10:06 +00:00
|
|
|
# Convert epub to a mobi
|
2017-09-15 08:28:26 +00:00
|
|
|
`ebook-convert books/#{name}.epub books/#{name}.mobi`
|
2017-09-17 04:30:34 +00:00
|
|
|
puts '[mobi] Generated MOBI file'
|
2017-09-15 06:10:06 +00:00
|
|
|
else
|
|
|
|
puts "[error] Can't generate MOBI without ebook-convert"
|
|
|
|
end
|
2017-09-17 04:30:34 +00:00
|
|
|
end
|
2017-09-15 06:10:06 +00:00
|
|
|
|
2017-09-27 16:44:55 +00:00
|
|
|
def gen_pdf(name, format)
|
|
|
|
if commands?(%w[pandoc convert wkhtmltopdf pdftk]) && format_match(format, :pdf)
|
2017-09-15 06:10:06 +00:00
|
|
|
# Generate PDF as well
|
|
|
|
# First, lets make a better css version of the html
|
2017-11-06 21:30:20 +00:00
|
|
|
`pandoc books/#{name}.html -s -c ../epub.css -o books/#{name}_pdf.html`
|
2017-09-17 04:30:34 +00:00
|
|
|
puts '[pdf] Generated html for pdf'
|
2017-09-15 06:10:06 +00:00
|
|
|
|
|
|
|
# Print the pdf_html file to pdf
|
2017-09-17 04:30:34 +00:00
|
|
|
`wkhtmltopdf books/#{name}_pdf.html books/#{name}-nocover.pdf`
|
|
|
|
puts '[pdf] Generated PDF without cover'
|
2017-09-15 06:10:06 +00:00
|
|
|
|
|
|
|
# Join the cover and pdf together
|
2017-09-15 08:28:26 +00:00
|
|
|
`pdftk covers/#{name}.pdf books/#{name}-nocover.pdf cat output books/#{name}.pdf`
|
2017-09-17 04:30:34 +00:00
|
|
|
puts '[pdf] Generated PDF file'
|
2017-09-15 06:10:06 +00:00
|
|
|
else
|
2017-09-17 04:30:34 +00:00
|
|
|
puts '[error] Please check README for PDF dependencies'
|
2017-09-15 06:10:06 +00:00
|
|
|
end
|
|
|
|
end
|
2017-09-17 04:30:34 +00:00
|
|
|
|
2017-09-27 16:44:55 +00:00
|
|
|
def generate(name, format = :all)
|
|
|
|
gen_epub(name, format)
|
|
|
|
gen_mobi(name, format)
|
|
|
|
gen_pdf(name, format)
|
2017-09-17 04:30:34 +00:00
|
|
|
end
|