api/upload/services/upload.js

upload.js

"use strict";

/**
 * Module dependencies
 */

// Node.js core.
const path = require("path");
const fs = require("fs");

// Public node modules.
const _ = require("lodash");

module.exports = {
  /**
   * Upload files.
   *
   * @param part   File extracted from the request body.
   * @param ctx
   * @returns {*}
   */

  upload: function*(part, ctx) {
    return new Promise(async function(resolve, reject) {
      // Init and format variables.
      const promises = [];
      const defaultUploadFolder = "public/upload";
      let stream;
      ctx = ctx || {};

      // Set the filename.
      const filename =
        Date.now().toString() +
        "-" +
        (_.kebabCase(part.filename) ||
          Math.floor(Math.random() * 1000000).toString());

      // Start uploading.
      stream = fs.createWriteStream(
        path.join(
          process.cwd(),
          strapi.api.upload.config.folder || defaultUploadFolder,
          filename
        )
      );
      part.pipe(stream);

      // Register the data of the file in the database.
      promises.push(
        strapi.orm.collections.upload.create(
          _.merge(part, {
            createdBy: ctx.user && ctx.user.id,
            originalFilenameFormatted: _.kebabCase(part.filename),
            originalFilename: part.filename || "",
            filename: filename
          })
        )
      );

      try {
        const uploadDescriptions = await promises;
        resolve(uploadDescriptions);
      } catch (err) {
        strapi.log.error(err);
        reject(err);
      }
    });
  }
};