reader.js 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. import axios from 'axios';
  2. import {sleep} from '../share/utils';
  3. const api = axios.create({
  4. baseURL: '/api/reader'
  5. });
  6. const workerApi = axios.create({
  7. baseURL: '/api/worker'
  8. });
  9. class Reader {
  10. async loadBook(url, callback) {
  11. const refreshPause = 200;
  12. if (!callback) callback = () => {};
  13. let response = await api.post('/load-book', {type: 'url', url});
  14. const workerId = response.data.workerId;
  15. if (!workerId)
  16. throw new Error('Неверный ответ api');
  17. callback({totalSteps: 4});
  18. let i = 0;
  19. while (1) {// eslint-disable-line no-constant-condition
  20. callback(response.data);
  21. if (response.data.state == 'finish') {//воркер закончил работу, можно скачивать кешированный на сервере файл
  22. callback({step: 4});
  23. const book = await this.loadCachedBook(response.data.path, callback);
  24. return Object.assign({}, response.data, {data: book.data});
  25. }
  26. if (response.data.state == 'error') {
  27. let errMes = response.data.error;
  28. if (errMes.indexOf('getaddrinfo') >= 0 ||
  29. errMes.indexOf('ECONNRESET') >= 0 ||
  30. errMes.indexOf('EINVAL') >= 0 ||
  31. errMes.indexOf('404') >= 0)
  32. errMes = `Ресурс не найден по адресу: ${response.data.url}`;
  33. throw new Error(errMes);
  34. }
  35. if (i > 0)
  36. await sleep(refreshPause);
  37. i++;
  38. if (i > 30*1000/refreshPause) {//30 сек ждем телодвижений воркера
  39. throw new Error('Слишком долгое время ожидания');
  40. }
  41. //проверка воркера
  42. const prevProgress = response.data.progress;
  43. response = await workerApi.post('/get-state', {workerId});
  44. i = (prevProgress != response.data.progress ? 1 : i);
  45. }
  46. }
  47. async loadCachedBook(url, callback){
  48. const response = await axios.head(url);
  49. let estSize = 1000000;
  50. if (response.headers['content-length']) {
  51. estSize = response.headers['content-length'];
  52. }
  53. const options = {
  54. onDownloadProgress: progress => {
  55. while (progress.loaded > estSize) estSize *= 1.5;
  56. if (callback)
  57. callback({state: 'loading', progress: Math.round((progress.loaded*100)/estSize)});
  58. }
  59. }
  60. //загрузка
  61. return await axios.get(url, options);
  62. }
  63. async uploadFile(file, maxUploadFileSize, callback) {
  64. if (!maxUploadFileSize)
  65. maxUploadFileSize = 10*1024*1024;
  66. if (file.size > maxUploadFileSize)
  67. throw new Error(`Размер файла превышает ${maxUploadFileSize} байт`);
  68. let formData = new FormData();
  69. formData.append('file', file);
  70. const options = {
  71. headers: {
  72. 'Content-Type': 'multipart/form-data'
  73. },
  74. onUploadProgress: progress => {
  75. const total = (progress.total ? progress.total : progress.loaded + 200000);
  76. if (callback)
  77. callback({state: 'upload', progress: Math.round((progress.loaded*100)/total)});
  78. }
  79. };
  80. let response = await api.post('/upload-file', formData, options);
  81. if (response.data.state == 'error')
  82. throw new Error(response.data.error);
  83. const url = response.data.url;
  84. if (!url)
  85. throw new Error('Неверный ответ api');
  86. return url;
  87. }
  88. }
  89. export default new Reader();