strapi/packages/strapi-plugin-upload/config/functions/bootstrap.js

'use strict';

/**
 * An asynchronous bootstrap function that runs before
 * your application gets started.
 *
 * This gives you an opportunity to set up your data model,
 * run jobs, or perform some special logic.
 */

const path = require('path');
const _ = require('lodash');
const fs = require('fs');

module.exports = async cb => {
  const Model = strapi.plugins.upload.models.file;

  const hasTable = await strapi.connections[Model.connection].schema.hasTable(Model.tableName || Model.collectionName);

  if (!hasTable) {
    const quote = Model.client === 'pg' ? '"' : '`';

    strapi.log.warn(`
⚠️  TABLE \`upload_file\` DOESN'T EXIST
⚠️  TABLE \`upload_file_morph\` DOESN'T EXIST

CREATE TABLE ${quote}${Model.tableName || Model.collectionName}${quote} (
  id ${Model.client === 'pg' ? 'SERIAL' : 'INT AUTO_INCREMENT'} NOT NULL PRIMARY KEY,
  name text,
  hash text,
  ext text,
  mime text,
  size text,
  url text,
  updated_at ${Model.client === 'pg' ? 'timestamp with time zone' : 'timestamp'},
  created_at ${Model.client === 'pg' ? 'timestamp with time zone' : 'timestamp'}
);

CREATE TABLE ${quote}upload_file_morph${quote} (
  id ${Model.client === 'pg' ? 'SERIAL' : 'INT AUTO_INCREMENT'} NOT NULL PRIMARY KEY,
  upload_file_id  ${Model.client === 'pg' ? 'integer' : 'int'},
  related_id  ${Model.client === 'pg' ? 'integer' : 'int'},
  related_type text,
  field text
);

1️⃣  EXECUTE THE FOLLOWING SQL QUERY

2️⃣  RESTART YOUR SERVER
    `);

    strapi.stop();
  }

  // set plugin store
  const pluginStore = strapi.store({
    environment: strapi.config.environment,
    type: 'plugin',
    name: 'upload'
  });

  fs.readdir(path.join(strapi.config.appPath, 'plugins', 'upload', 'node_modules'), async (err, node_modules) => {
    // get all upload provider
    const uploads = _.filter(node_modules, (node_module) => {
      return _.startsWith(node_module, ('strapi-upload'));
    });

    strapi.plugins.upload.config.providers = [];

    // mount all providers to get configs
    _.forEach(uploads, (node_module) => {
      strapi.plugins.upload.config.providers.push(
        require(path.join(`${strapi.config.appPath}/plugins/upload/node_modules/${node_module}`))
      );
    });

    try {
      // if provider config not exit set one by default
      const config = await pluginStore.get({key: 'provider'});

      if (!config) {
        const provider = _.find(strapi.plugins.upload.config.providers, {provider: 'local'});

        const value = _.assign({}, provider, {
          enabled: true,
          // by default limit size to 1 GB
          sizeLimit: 1000000
        });

        await pluginStore.set({key: 'provider', value});
      }
    } catch (err) {
      strapi.log.error(`Can't load ${config.provider} upload provider.`);
      strapi.log.warn(`Please install strapi-upload-${config.provider} --save in ${path.join(strapi.config.appPath, 'plugins', 'upload')} folder.`);
      strapi.stop();
    }

    cb();
  });
};