feat: add temaplate param

dylandepass · Jan 30, 2023 · 237f45a · 237f45a
1 parent 2884e49
commit 237f45a
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/template.ts b/src/template.ts
@@ -21,16 +21,24 @@
  * @returns A fully decorated element for rendering in storybook
  */
 function prepare(loadPage: any, args: any, parameters: any, main: any, content: HTMLElement | Element, decorate: any) {
-  const { selector, index } = parameters;
+  const { selector, index, template } = parameters;
   const { sectionClasses, blockClasses } = args;
   const section = document.createElement('div');
 
   const node = content.querySelectorAll(selector).item(index) || content.querySelector(selector);
 
+  if(!decorate) {
+    return;
+  }
+
   main.appendChild(section);
 
   section.innerHTML = parameters.root ? node.parentNode.innerHTML : node.outerHTML;
 
+  if(template) {
+    decorate(parameters.autoBlock ? main.querySelector('.block') : section.querySelector(selector));
+  }
+
   loadPage();
 
   if (sectionClasses) {
@@ -51,7 +59,7 @@ function prepare(loadPage: any, args: any, parameters: any, main: any, content:
     });
   }
 
-  if(decorate) {
+  if(decorate && !template) {
     decorate(parameters.autoBlock ? main.querySelector('.block') : section.querySelector(selector));
   }