Merge pull request #39 from CodeVoyager/feat/reader-nav-single-type

feat: single type is a template for itself
VirtusLab-Open-Source · Mar 2, 2021 · 9227c04 · 9227c04
2 parents 7bfd441 + 2582d0c
commit 9227c04
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/services/utils/functions.js b/services/utils/functions.js
@@ -107,11 +107,25 @@ module.exports = {
               }),
       );
       const relatedResponseMap = responses.reduce((acc, curr) => ({ ...acc, ...curr }), {});
+      const singleTypes = new Set( 
+        contentTypes
+          .filter(x => x.isSingle)
+          .map(({ contentTypeName }) => contentTypeName)
+      );
 
       return (contentType, id) => {
         const template = get(relatedResponseMap[contentType].find(data => data.id === id), 'template');
-        const templateComponent = this.getTemplateComponentFromTemplate(template);
-        return get(templateComponent, 'options.templateName', TEMPLATE_DEFAULT);
+
+        if (template) {
+          const templateComponent = this.getTemplateComponentFromTemplate(template);
+          return get(templateComponent, 'options.templateName', TEMPLATE_DEFAULT);
+        }
+
+        if (singleTypes.has(contentType)) {
+          return contentType;
+        }
+
+        return TEMPLATE_DEFAULT;
       };
     },