yujunjun/strapi
1
0
Fork 0
mirror of https://github.com/strapi/strapi.git synced 2025-12-27 07:03:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: load storage providers at strapi 'register' step

Browse Source
This commit is contained in:
vincentbpro 2022-05-09 10:06:47 +02:00
parent 1e111d6fae
commit 715494585f
2 changed files with 59 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
packages/core/upload/server/bootstrap.js vendored
View File

@ -1,13 +1,9 @@
'use strict';
const _ = require('lodash');
module.exports = async ({ strapi }) => {
// set plugin store
const configurator = strapi.store({ type: 'plugin', name: 'upload', key: 'settings' });
strapi.plugin('upload').provider = createProvider(strapi.config.get('plugin.upload', {}));
// if provider config does not exist set one by default
const config = await configurator.get();
@ -24,62 +20,6 @@ module.exports = async ({ strapi }) => {
await registerPermissionActions();
};
const createProvider = config => {
const { providerOptions, actionOptions = {} } = config;
const providerName = _.toLower(config.provider);
let provider;
let modulePath;
try {
modulePath = require.resolve(`@strapi/provider-upload-${providerName}`);
} catch (error) {
if (error.code === 'MODULE_NOT_FOUND') {
modulePath = providerName;
} else {
throw error;
}
}
try {
provider = require(modulePath);
} catch (err) {
throw new Error(`Could not load upload provider "${providerName}".`);
}
const providerInstance = provider.init(providerOptions);
if (!providerInstance.delete) {
throw new Error(`The upload provider "${providerName}" doesn't implement the delete method.`);
}
if (!providerInstance.upload && !providerInstance.uploadStream) {
throw new Error(
`The upload provider "${providerName}" doesn't implement the uploadStream nor the upload method.`
);
}
if (!providerInstance.uploadStream) {
process.emitWarning(
`The upload provider "${providerName}" doesn't implement the uploadStream function. Strapi will fallback on the upload method. Some performance issues may occur.`
);
}
const wrappedProvider = _.mapValues(providerInstance, (method, methodName) => {
return async function(file, options = actionOptions[methodName]) {
return providerInstance[methodName](file, options);
};
});
return Object.assign(Object.create(baseProvider), wrappedProvider);
};
const baseProvider = {
extend(obj) {
Object.assign(this, obj);
},
};
const registerPermissionActions = async () => {
const actions = [
{

59
packages/core/upload/server/register.js
View File

@ -1,5 +1,6 @@
'use strict';
const _ = require('lodash');
const registerUploadMiddleware = require('./middlewares/upload');
/**
@ -7,9 +8,67 @@ const registerUploadMiddleware = require('./middlewares/upload');
* @param {{ strapi: import('@strapi/strapi').Strapi }}
*/
module.exports = async ({ strapi }) => {
strapi.plugin('upload').provider = createProvider(strapi.config.get('plugin.upload', {}));
await registerUploadMiddleware({ strapi });
if (strapi.plugin('graphql')) {
require('./graphql')({ strapi });
}
};
const createProvider = (config) => {
const { providerOptions, actionOptions = {} } = config;
const providerName = _.toLower(config.provider);
let provider;
let modulePath;
try {
modulePath = require.resolve(`@strapi/provider-upload-${providerName}`);
} catch (error) {
if (error.code === 'MODULE_NOT_FOUND') {
modulePath = providerName;
} else {
throw error;
}
}
try {
provider = require(modulePath);
} catch (err) {
throw new Error(`Could not load upload provider "${providerName}".`);
}
const providerInstance = provider.init(providerOptions);
if (!providerInstance.delete) {
throw new Error(`The upload provider "${providerName}" doesn't implement the delete method.`);
}
if (!providerInstance.upload && !providerInstance.uploadStream) {
throw new Error(
`The upload provider "${providerName}" doesn't implement the uploadStream nor the upload method.`
);
}
if (!providerInstance.uploadStream) {
process.emitWarning(
`The upload provider "${providerName}" doesn't implement the uploadStream function. Strapi will fallback on the upload method. Some performance issues may occur.`
);
}
const wrappedProvider = _.mapValues(providerInstance, (method, methodName) => {
return async function (file, options = actionOptions[methodName]) {
return providerInstance[methodName](file, options);
};
});
return Object.assign(Object.create(baseProvider), wrappedProvider);
};
const baseProvider = {
extend(obj) {
Object.assign(this, obj);
},
};