markdown-pdf/phantom/render.js

Markdown to PDF converter

var system = require('system')
var page = require('webpage').create()
var fs = require('fs')
var os = require('system').os

// Read in arguments
var args = ['in', 'out', 'cwd', 'runningsPath', 'cssPath', 'highlightCssPath', 'paperFormat', 'paperOrientation', 'paperBorder', 'renderDelay', 'loadTimeout'].reduce(function (args, name, i) {
  args[name] = system.args[i + 1]
  return args
}, {})

var html5bpPath = page.libraryPath + '/../html5bp'

// Resources don't load in windows with file protocol
var isWin = os.name === 'windows'
var protocol = isWin ? 'file:///' : 'file://'

var html = fs.read(html5bpPath + '/index.html')
  .replace(/\{\{baseUrl\}\}/g, protocol + html5bpPath)
  .replace('{{content}}', fs.read(args.in))

page.setContent(html, protocol + args.cwd + '/markdown-pdf.html')

// Add custom CSS to the page
page.evaluate(function (cssPaths) {
  var head = document.querySelector('head')

  cssPaths.forEach(function (cssPath) {
    var css = document.createElement('link')
    css.rel = 'stylesheet'
    css.href = cssPath

    head.appendChild(css)
  })
}, [args.cssPath, args.highlightCssPath].map(function (cssPath) {
  return (isWin ? protocol : '') + cssPath
}))

// Set the PDF paper size
page.paperSize = paperSize(args.runningsPath, { format: args.paperFormat, orientation: args.paperOrientation, border: isJson(args.paperBorder) ? JSON.parse(args.paperBorder) : args.paperBorder })

args.renderDelay = parseInt(args.renderDelay, 10)

if (args.renderDelay) {
  setTimeout(render, args.renderDelay)
} else {
  var loadTimeout = setTimeout(render, parseInt(args.loadTimeout, 10))
  page.onLoadFinished = function () {
    clearTimeout(loadTimeout)
    render()
  }
}

function render () {
  page.render(args.out)
  page.close()
  window.phantom.exit(0)
}

function paperSize (runningsPath, obj) {
  var runnings = require(runningsPath)

  // encapsulate .contents into phantom.callback()
  //   Why does phantomjs not support Array.prototype.forEach?!
  var keys = ['header', 'footer']

  for (var i = 0; i < keys.length; i++) {
    var which = keys[i]

    if (runnings[which] && runnings[which].contents && typeof runnings[which].contents === 'function') {
      obj[which] = {
        contents: window.phantom.callback(runnings[which].contents)
      }

      if (runnings[which].height) {
        obj[which].height = runnings[which].height
      }
    }
  }

  return obj
}

function isJson (str) { try { JSON.parse(str) } catch (e) { return false } return true }