yujunjun/strapi
1
0
Fork 0
mirror of https://github.com/strapi/strapi.git synced 2025-09-02 05:13:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

use map to recursively clean relation attributes

Browse Source
This commit is contained in:
Mark Kaylor 2021-10-08 12:06:16 -04:00
parent e8d691e9bd
commit 67c61a0638
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
packages/plugins/documentation/server/utils/clean-schema-attributes.js
View File

@ -11,7 +11,7 @@ const getSchemaData = require('./get-schema-data');
* @returns Attributes using OpenAPI acceptable data types
*/
const cleanSchemaAttributes = attributes => {
const cleanSchemaAttributes = (attributes, typeMap = new Map()) => {
const attributesCopy = _.cloneDeep(attributes);
for (const prop in attributesCopy) {
@ -21,6 +21,9 @@ const cleanSchemaAttributes = attributes => {
}
switch (attribute.type) {
case 'password':
case 'email':
case 'date':
case 'datetime': {
attributesCopy[prop] = { type: 'string' };
break;
@ -66,12 +69,22 @@ const cleanSchemaAttributes = attributes => {
break;
}
case 'relation': {
// TODO: Sanitize relation attributes and list them in the schema
const isListOfEntities = attribute.relation.includes('ToMany');
if (!attribute.target || typeMap.has(attribute.target)) {
attributesCopy[prop] = {
type: 'object',
properties: { data: getSchemaData(isListOfEntities, {}) },
};
break;
}
typeMap.set(attribute.target, true);
const targetAttributes = strapi.contentType(attribute.target).attributes;
attributesCopy[prop] = {
type: 'object',
properties: {
data: getSchemaData(isListOfEntities, {}),
data: getSchemaData(isListOfEntities, cleanSchemaAttributes(targetAttributes, typeMap)),
},
};