pdftohtmljs_uwazi/lib/pdftohtml.js

PDF to HTML (pdf2htmlex) shell wrapper

var Promise = require('bluebird');
var shell = require('shelljs');

function pdftohtml (filename, outfile, options) {
  this.options = options || {};
  this.options.additional = [filename];
  if (typeof outfile !== "undefined" && outfile !== null) {
    this.options.additional.push(outfile);
  }

  pdftohtml.prototype._preset = function(preset) {
    try {
      var module = require('./presets/' + preset);
      if (typeof module.load === 'function') {
        module.load(this);
      }
      return this;
    }
    catch (err) {
      try {
        var module = require(preset);
        if (typeof module.load === 'function') {
          module.load(this);
        }
        return this;
      }
      catch (err) {
        //pass
      }
    }
    throw new Error('preset ' + preset + ' could not be loaded');
  };

  pdftohtml.prototype.add_options = function(optionArray) {
    if (typeof optionArray.length !== undefined) {
        var self = this;
        optionArray.forEach(function(el) {
          var firstSpace = el.indexOf(' ');
          if (firstSpace > 0) {
            var param = el.substr(0, firstSpace);
            var val = el.substr(firstSpace + 1).replace(/ /g, "\\ ");
            self.options.additional.push(param, val);
          } else {
            self.options.additional.push(el);
          }
        });
    }
    return this;
  };

  pdftohtml.prototype.convert = function(preset) {
    var preset = preset || "default";
    var self = this;
    return new Promise(function (resolve, reject) {
      self._preset(preset);
      var child = shell.exec('pdf2htmlEX ' + self.options.additional.join(' '), {async:true, silent:true});

      var error = '';

      child.stdout.on('data', function(data){
        // pdf2htmlEX writes out to stderr
      });

      child.stderr.on('data', function(data){
        error += data;
        if (self.options.progress
              && typeof self.options.progress === "function") {
          var lastLine = data.split(/\r\n|\r|\n/g);
          var ll = lastLine[lastLine.length - 2];
          var progress;
          if (ll) {
            progress = ll.split(/Working\: ([0-9\d]+)\/([0-9\d]+)/ig);
          }
          if (progress && progress.length > 1) {
            // build progress report object
            var ret = {
              current: parseInt(progress[1]),
              total: parseInt(progress[2])
            };
            self.options.progress(ret);
          }
        }
      });

      child.on('exit', function(code, signal){
        if (code === 0) {
          resolve();
        }
        else {
          if (!shell.which('pdf2htmlEX')) {
            shell.echo('Sorry, this script requires pdf2htmlEX.');
            shell.echo('Install it from http://github.com/coolwanglu/pdf2htmlEX');
            throw new Error("Please install pdf2htmlEX");
          }
          throw new Error("Error code: "+ code);
        }
      });
    });
  };

  pdftohtml.prototype.progress = function(callback) {
    this.options.progress = callback;
    return this;
  };

}

// module exports
exports = module.exports = function(filename, args) {
  return new pdftohtml(filename, args);
};