2019-11-13 19:25:57 +01:00

167 lines
4.2 KiB
JavaScript

'use strict';
/**
* Module dependencies
*/
// Node.js core.
const fs = require('fs');
const path = require('path');
// Public node modules.
const _ = require('lodash');
const pluralize = require('pluralize');
const slugify = require('@sindresorhus/slugify');
const toSlug = str => slugify(str, { separator: '-' });
/**
* This `before` function is run before generating targets.
* Validate, configure defaults, get extra dependencies, etc.
*
* @param {Object} scope
* @param {Function} cb
*/
module.exports = (scope, cb) => {
if (!scope.rootPath || !scope.id) {
return cb.invalid('Usage: `$ strapi generate:api apiName`');
}
// Check `api` and `plugin` parameters
const parent = scope.args.api || scope.args.plugin;
// Format `id`.
const name = toSlug(scope.id);
const environment = process.env.NODE_ENV || 'development';
// `scope.args` are the raw command line arguments.
_.defaults(scope, {
idPluralized: pluralize(name),
parentId: _.isEmpty(parent) ? undefined : _.trim(_.deburr(parent)),
globalID: name,
});
// Take another pass to take advantage of the defaults absorbed in previous passes.
_.defaults(scope, {
filename: `${name}.js`,
filenameSettings: `${name}.settings.json`,
folderPrefix: !scope.args.api && scope.args.plugin ? 'extensions' : 'api',
folderName: scope.parentId || name,
});
// Humanize output.
_.defaults(scope, {
humanizeId: _.camelCase(scope.id).toLowerCase(),
humanizedPath: `./${scope.folderPrefix}/${scope.folderName}`,
});
// Validate optional attribute arguments.
const invalidAttributes = [];
if (_.isPlainObject(scope.args.attributes)) {
scope.attributes = scope.args.attributes;
} else {
// Map attributes and split them for CLI.
scope.attributes = scope.args.attributes.map(attribute => {
if (_.isString(attribute)) {
const parts = attribute.split(':');
parts[1] = parts[1] || 'string';
// Handle invalid attributes.
if (!parts[1] || !parts[0]) {
invalidAttributes.push(
'Error: Invalid attribute notation `' + attribute + '`.'
);
return;
}
return {
name: _.trim(_.deburr(parts[0].toLowerCase())),
params: {
type: _.trim(_.deburr(parts[1].toLowerCase())),
},
};
} else {
return _.has(attribute, 'params.type') ? attribute : undefined;
}
});
scope.attributes = _.compact(scope.attributes);
// Handle invalid action arguments.
// Send back invalidActions.
if (invalidAttributes.length) {
return cb.invalid(invalidAttributes);
}
// Make sure there aren't duplicates.
if (
_(scope.attributes.map(attribute => attribute.name))
.uniq()
.valueOf().length !== scope.attributes.length
) {
return cb.invalid('Duplicate attributes not allowed!');
}
// Render some stringified code from the action template
// and make it available in our scope for use later on.
scope.attributes = scope.attributes.reduce((acc, attribute) => {
acc[attribute.name] = attribute.params;
return acc;
}, {});
}
// Set collectionName
scope.collectionName = _.has(scope.args, 'collectionName')
? scope.args.collectionName
: pluralize(name);
// Set description
scope.description = _.has(scope.args, 'description')
? scope.args.description
: '';
// Get default connection
try {
scope.connection =
_.get(scope.args, 'connection') ||
JSON.parse(
fs.readFileSync(
path.resolve(
scope.rootPath,
'config',
'environments',
environment,
'database.json'
)
)
).defaultConnection ||
'';
} catch (err) {
return cb.invalid(err);
}
scope.schema = JSON.stringify(
{
connection: scope.connection,
collectionName: scope.collectionName,
info: {
name: scope.args.name || scope.id,
description: scope.description,
},
options: {
increments: true,
timestamps: true,
comment: '',
},
attributes: scope.attributes,
},
null,
2
);
// Trigger callback with no error to proceed.
return cb.success();
};