Support relative links using base option

marcbachmann · Feb 16, 2016 · b791d6d · b791d6d
1 parent 02748ee
commit b791d6d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/bin/index.js b/bin/index.js
@@ -2,6 +2,7 @@
 
 var fs = require('fs')
 var pdf = require('../')
+var path = require('path')
 
 var args = process.argv.slice(2)
 
@@ -22,7 +23,10 @@ function help () {
 
 function htmlpdf (source, destination) {
   var html = fs.readFileSync(source, 'utf8')
-  pdf.create(html).toFile(destination, function (err, res) {
+  var options = {
+    base: 'file://' + path.resolve(source)
+  }
+  pdf.create(html, options).toFile(destination, function (err, res) {
     if (err) throw err
   })
 }
diff --git a/src/scripts/pdf_a4_portrait.coffee b/src/scripts/pdf_a4_portrait.coffee
@@ -32,7 +32,11 @@ page = webpage.create()
 if options.httpHeaders
   page.customHeaders = options.httpHeaders
 
-page.content = json.html
+if options.base
+  page.setContent(json.html, options.base)
+else
+  page.setContent(json.html, null)
+
 page.viewportSize = vp if vp = options.viewportSize
 totalPages = 0