strapi/packages/strapi-generate-migrations/lib/builder/createTable.js

'use strict';

/**
 * Module dependencies
 */

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

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

/**
 * Create a table if it doesn't exist yet
 * and drop the table to reverse the migration.
 */

module.exports = (models, modelName) => {

  // Template: create the table for the `up` export if it doesn't exist.
  // This adds a `up` logic for the current model.
  // Then, every `up` logic of every model call the
  // `./builder/tables/createTableIfNotExists` template.
  const tplTableCreate = fs.readFileSync(path.resolve(__dirname, '..', '..', 'templates', 'builder', 'tables', 'createTableIfNotExists.template'), 'utf8');
  if (_.isEmpty(_.get(models[modelName], 'up.others'))) {
    _.set(models[modelName], 'up.others', _.unescape(_.template(tplTableCreate)({
      models,
      tableName: modelName,
      attributes: models[modelName].attributes,
      options: models[modelName].options
    })));
  } else {
    models[modelName].up.others += _.unescape(_.template(tplTableCreate)({
      models,
      tableName: modelName,
      attributes: models[modelName].attributes,
      options: models[modelName].options
    }));
  }

  // Template: drop the table for the `down` export.
  // This adds a `down` logic for the current model.
  // Then, every `down` logic of every model call the
  // `./builder/tables/dropTable` template.
  const tplTableDrop = fs.readFileSync(path.resolve(__dirname, '..', '..', 'templates', 'builder', 'tables', 'dropTable.template'), 'utf8');

  if (_.isEmpty(_.get(models[modelName], 'down.others'))) {
    _.set(models[modelName], 'down.others', _.unescape(_.template(tplTableDrop)({
      tableName: modelName
    })));
  } else {
    models[modelName].down.others += _.unescape(_.template(tplTableDrop)({
      tableName: modelName
    }));
  }
};