Merge pull request mooz#16 from dustinc/master

Add setPdfFileBaseName
leodisarli · Jan 31, 2018 · 65719c1 · 65719c1
2 parents 122b837 + 1ed9e37
commit 65719c1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -11,8 +11,8 @@ function PDFImage(pdfFilePath, options) {
   if (!options) options = {};
 
   this.pdfFilePath = pdfFilePath;
-  this.pdfFileBaseName = path.basename(pdfFilePath, ".pdf");
 
+  this.setPdfFileBaseName(options.pdfFileBaseName);
   this.setConvertOptions(options.convertOptions);
   this.setConvertExtension(options.convertExtension);
   this.useGM = options.graphicsMagick || false;
@@ -68,6 +68,9 @@ PDFImage.prototype = {
   setConvertOptions: function (convertOptions) {
     this.convertOptions = convertOptions || {};
   },
+  setPdfFileBaseName: function(pdfFileBaseName) {
+    this.pdfFileBaseName = pdfFileBaseName || path.basename(this.pdfFilePath, ".pdf");
+  },
   setConvertExtension: function (convertExtension) {
     this.convertExtension = convertExtension || "png";
   },

diff --git a/tests/test-main.js b/tests/test-main.js
@@ -16,6 +16,11 @@ describe("PDFImage", function () {
   it("should have correct basename", function () {
     expect(pdfImage.pdfFileBaseName).equal("test");
   });
+
+  it("should set custom basename", function() {
+    pdfImage.setPdfFileBaseName('custom-basename');
+    expect(pdfImage.pdfFileBaseName).equal("custom-basename");
+  });
 
   it("should return correct page path", function () {
     expect(pdfImage.getOutputImagePathForPage(1))