70 lines
2.0 KiB
JavaScript
Raw Normal View History

'use strict';
const _ = require('lodash');
module.exports = async (entry, files, { model, source }) => {
const entity = strapi.getModel(model, source);
if (!_.has(strapi.plugins, 'upload')) return entry;
const uploadService = strapi.plugins.upload.services.upload;
const findModelFromUploadPath = path => {
if (path.length === 0) return { model, source };
2019-11-07 18:11:23 +01:00
let currentPath = [];
let tmpModel = entity;
let modelName = model;
let sourceName;
2019-11-07 18:11:23 +01:00
for (let i = 0; i < path.length; i++) {
if (!tmpModel) return {};
2019-11-07 18:11:23 +01:00
const part = path[i];
const attr = tmpModel.attributes[part];
2019-11-07 18:11:23 +01:00
currentPath.push(part);
// ignore array indexes => handled in the dynamic zone section
if (_.isFinite(_.toNumber(path[i]))) {
continue;
}
if (!attr) return {};
2019-10-22 18:01:03 +02:00
if (attr.type === 'component') {
modelName = attr.component;
tmpModel = strapi.components[attr.component];
2019-11-07 18:11:23 +01:00
} else if (attr.type === 'dynamiczone') {
const entryIdx = path[i + 1]; // get component index
modelName = _.get(entry, [...currentPath, entryIdx]).__component; // get component type
tmpModel = strapi.components[modelName];
} else if (_.has(attr, 'model') || _.has(attr, 'collection')) {
sourceName = attr.plugin;
modelName = attr.model || attr.collection;
tmpModel = strapi.getModel(attr.model || attr.collection, source);
} else {
return {};
}
}
return { model: modelName, source: sourceName };
};
const doUpload = async (key, files) => {
const parts = key.split('.');
const [path, field] = [_.initial(parts), _.last(parts)];
const { model, source } = findModelFromUploadPath(path);
if (model) {
const id = _.get(entry, path.concat('id'));
return uploadService.uploadToEntity(
2019-11-07 18:11:23 +01:00
{ id, model },
{ [field]: files },
source
);
}
};
await Promise.all(Object.keys(files).map(key => doUpload(key, files[key])));
};