'use strict'; /** * Module dependencies. */ var _cloneDeep = require('lodash.clonedeep'); var translateDataTypeKeys = require('./translate-data-type-keys'); /** * Export the modelHelper singleton. */ var modelHelper = module.exports = { /** * Given a class (from remotes.classes()), generate a model definition. * This is used to generate the schema at the top of many endpoints. * @param {Class} class Remote class. * @return {Object} Associated model definition. */ generateModelDefinition: function generateModelDefinition(aClass) { var def = aClass.ctor.definition; var name = def.name; var required = []; // Don't modify original properties. var properties = _cloneDeep(def.properties); // Iterate through each property in the model definition. // Types may be defined as constructors (e.g. String, Date, etc.), // or as strings; getPropType() will take care of the conversion. // See more on types: // https://github.com/wordnik/swagger-spec/blob/master/versions/1.2.md#431-primitives Object.keys(properties).forEach(function(key) { var prop = properties[key]; // Eke a type out of the constructors we were passed. prop = modelHelper.LDLPropToSwaggerDataType(prop); // Required props sit in a per-model array. if (prop.required || (prop.id && !prop.generated)) { required.push(key); } // Change mismatched keys. prop = translateDataTypeKeys(prop); // Assign this back to the properties object. properties[key] = prop; }); var out = {}; out[name] = { id: name, properties: properties, required: required }; return out; }, /** * Given a propType (which may be a function, string, or array), * get a string type. * @param {*} propType Prop type description. * @return {String} Prop type string. */ getPropType: function getPropType(propType) { if (typeof propType === 'function') { propType = propType.name.toLowerCase(); } else if(Array.isArray(propType)) { propType = 'array'; } return propType; }, // Converts a prop defined with the LDL spec to one conforming to the // Swagger spec. // https://github.com/wordnik/swagger-spec/blob/master/versions/1.2.md#431-primitives LDLPropToSwaggerDataType: function LDLPropToSwaggerDataType(prop) { var out = _cloneDeep(prop); out.type = modelHelper.getPropType(out.type); if (out.type === 'array') { var arrayProp = prop.type[0]; if (!arrayProp.type) arrayProp = {type: arrayProp}; out.items = modelHelper.LDLPropToSwaggerDataType(arrayProp); } if (out.type === 'date') { out.type = 'string'; out.format = 'date'; } else if (out.type === 'buffer') { out.type = 'string'; out.format = 'byte'; } else if (out.type === 'number') { out.format = 'double'; // Since all JS numbers are doubles } return out; } };