const _ = require('lodash');
const { streamToBuffer } = require('../utils/file');

module.exports = {
  definition: `
    input FileInfoInput {
      name: String
      alternativeText: String
      caption: String
    }
  `,
  mutation: `
    upload(refId: ID, ref: String, field: String, source: String, file: Upload!): UploadFile!
    multipleUpload(refId: ID, ref: String, field: String, source: String, files: [Upload]!): [UploadFile]!
    updateFileInfo(id: ID!, info: FileInfoInput!): UploadFile!
  `,
  resolver: {
    Query: {
      file: false,
      files: {
        resolver: 'plugins::upload.upload.find',
      },
    },
    Mutation: {
      createFile: false,
      updateFile: false,
      deleteFile: false,
      upload: {
        description: 'Upload one file',
        resolverOf: 'plugins::upload.upload.upload',
        resolver: async (obj, { file: upload, ...fields }) => {
          const file = await formatFile(upload, fields);

          const uploadedFiles = await strapi.plugins.upload.services.upload.uploadFileAndPersist(
            file
          );

          // Return response.
          return uploadedFiles.length === 1 ? uploadedFiles[0] : uploadedFiles;
        },
      },
      multipleUpload: {
        description: 'Upload one file',
        resolverOf: 'plugins::upload.upload.upload',
        resolver: async (obj, { files: uploads, ...fields }) => {
          const files = await Promise.all(uploads.map(upload => formatFile(upload, fields)));

          const uploadService = strapi.plugins.upload.services.upload;

          return Promise.all(files.map(file => uploadService.uploadFileAndPersist(file)));
        },
      },
      updateFileInfo: {
        description: 'Update file information',
        resolverOf: 'plugins::upload.upload.upload',
        resolver: async (obj, { id, info }) => {
          return await strapi.plugins.upload.services.upload.updateFileInfo(id, info);
        },
      },
    },
  },
};

const formatFile = async (upload, metas) => {
  const { filename, mimetype, createReadStream } = await upload;

  const { optimize } = strapi.plugins.upload.services['image-manipulation'];
  const readBuffer = await streamToBuffer(createReadStream());

  const { buffer, info } = await optimize(readBuffer);

  const uploadService = strapi.plugins.upload.services.upload;
  const fileInfo = uploadService.formatFileInfo(
    {
      filename,
      type: mimetype,
      size: buffer.length,
    },
    {},
    metas
  );

  return _.assign(fileInfo, info, { buffer });
};