93 lines
2.7 KiB
JavaScript
93 lines
2.7 KiB
JavaScript
const axios = require('axios');
|
|
const fs = require('fs/promises');
|
|
const {createWriteStream} = require('fs');
|
|
const path = require('path');
|
|
|
|
module.exports = Self => {
|
|
Self.remoteMethod('download', {
|
|
description: 'Processes the image download queue',
|
|
accessType: 'WRITE',
|
|
http: {
|
|
path: `/download`,
|
|
verb: 'POST',
|
|
},
|
|
});
|
|
|
|
Self.download = async() => {
|
|
const models = Self.app.models;
|
|
const tempContainer = await models.TempContainer.container(
|
|
'salix-image'
|
|
);
|
|
const tempPath = path.join(
|
|
tempContainer.client.root,
|
|
tempContainer.name
|
|
);
|
|
const maxAttempts = 3;
|
|
const collectionName = 'catalog';
|
|
|
|
let tempFilePath;
|
|
let queueRow;
|
|
try {
|
|
queueRow = await Self.findOne(
|
|
{
|
|
fields: ['id', 'itemFk', 'url', 'attempts'],
|
|
where: {
|
|
url: {neq: null},
|
|
attempts: {
|
|
lt: maxAttempts,
|
|
},
|
|
},
|
|
order: 'priority, attempts, updated',
|
|
}
|
|
);
|
|
|
|
if (!queueRow) return;
|
|
|
|
const fileName = `${queueRow.itemFk}.png`;
|
|
tempFilePath = path.join(tempPath, fileName);
|
|
|
|
// Download remote image
|
|
const response = await axios.get(queueRow.url, {
|
|
responseType: 'stream',
|
|
});
|
|
|
|
const writeStream = createWriteStream(tempFilePath);
|
|
await new Promise((resolve, reject) => {
|
|
writeStream.on('open', () => response.data.pipe(writeStream));
|
|
writeStream.on('finish', () => resolve());
|
|
writeStream.on('error', error => reject(error));
|
|
});
|
|
|
|
await models.Image.resize({
|
|
collectionName: collectionName,
|
|
srcFile: tempFilePath,
|
|
fileName: fileName,
|
|
entityId: queueRow.itemFk
|
|
});
|
|
|
|
try {
|
|
await fs.unlink(tempFilePath);
|
|
} catch (error) { }
|
|
|
|
await queueRow.destroy();
|
|
|
|
// Restart queue
|
|
Self.download();
|
|
} catch (error) {
|
|
if (queueRow.attempts < maxAttempts) {
|
|
await queueRow.updateAttributes({
|
|
error: error,
|
|
attempts: queueRow.attempts + 1,
|
|
updated: Date.vnNew(),
|
|
});
|
|
}
|
|
|
|
try {
|
|
await fs.unlink(tempFilePath);
|
|
} catch (error) { }
|
|
|
|
Self.download();
|
|
}
|
|
};
|
|
};
|