weseek__growi/apps/app/src/server/service/import/overwrite-params/pages.ts

import { PageGrant } from '@growi/core';

const mongoose = require('mongoose');

// eslint-disable-next-line no-unused-vars
const ImportOptionForPages = require('~/models/admin/import-option-for-pages');

const { ObjectId } = mongoose.Types;

class PageOverwriteParamsFactory {

  /**
   * generate overwrite params object
   * @param {string} operatorUserId
   * @param {ImportOptionForPages} option
   * @return {import('~/server/service/import').OverwriteParams}
   *  key: property name
   *  value: any value or a function `(value, { document, schema, propertyName }) => { return newValue }`
   */
  static generate(operatorUserId, option) {
    const params = {};

    if (option.isOverwriteAuthorWithCurrentUser) {
      const userId = ObjectId(operatorUserId);
      params.creator = userId;
      params.lastUpdateUser = userId;
    }

    params.grant = (value, { document, schema, propertyName }) => {
      if (option.makePublicForGrant2 && value === 2) {
        return PageGrant.GRANT_PUBLIC;
      }
      if (option.makePublicForGrant4 && value === 4) {
        return PageGrant.GRANT_PUBLIC;
      }
      if (option.makePublicForGrant5 && value === 5) {
        return PageGrant.GRANT_PUBLIC;
      }
      return value;
    };

    params.parent = (value, { document, schema, propertyName }) => {
      return null;
    };

    params.descendantCount = (value, { document, schema, propertyName }) => {
      return 0;
    };

    if (option.initPageMetadatas) {
      params.liker = [];
      params.seenUsers = [];
      params.commentCount = 0;
      params.extended = {};
    }

    return params;
  }

}

module.exports = (operatorUserId, option) => PageOverwriteParamsFactory.generate(operatorUserId, option);