index.js 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. const fs = require('fs-extra');
  2. const FileDetector = require('../../FileDetector');
  3. //порядок важен
  4. const convertClassFactory = [
  5. require('./ConvertJpegPng'),
  6. require('./ConvertEpub'),
  7. require('./ConvertDjvu'),
  8. require('./ConvertPdf'),
  9. require('./ConvertRtf'),
  10. require('./ConvertDocX'),
  11. require('./ConvertFb3'),
  12. require('./ConvertDoc'),
  13. require('./ConvertMobi'),
  14. require('./ConvertFb2'),
  15. require('./ConvertSamlib'),
  16. require('./ConvertSites'),
  17. require('./ConvertHtml'),
  18. ];
  19. class BookConverter {
  20. constructor(config) {
  21. this.detector = new FileDetector();
  22. this.convertFactory = [];
  23. for (const convertClass of convertClassFactory) {
  24. this.convertFactory.push(new convertClass(config));
  25. }
  26. }
  27. async convertToFb2(inputFiles, outputFile, opts, callback, abort) {
  28. if (abort && abort())
  29. throw new Error('abort');
  30. const selectedFileType = await this.detector.detectFile(inputFiles.selectedFile);
  31. const data = await fs.readFile(inputFiles.selectedFile);
  32. const convertOpts = Object.assign({}, opts, {inputFiles, callback, abort, dataType: selectedFileType});
  33. let result = false;
  34. for (const convert of this.convertFactory) {
  35. result = await convert.run(data, convertOpts);
  36. if (result) {
  37. await fs.writeFile(outputFile, result);
  38. break;
  39. }
  40. }
  41. if (!result && inputFiles.nesting) {
  42. result = await this.convertToFb2(inputFiles.nesting, outputFile, opts, callback, abort);
  43. }
  44. if (!result) {
  45. if (selectedFileType)
  46. throw new Error(`Этот формат файла не поддерживается: ${selectedFileType.mime}`);
  47. else {
  48. throw new Error(`Не удалось определить формат файла: ${opts.url}`);
  49. }
  50. }
  51. callback(100);
  52. return result;
  53. }
  54. }
  55. module.exports = BookConverter;