add perf to shortcodes

Elderjs · May 11, 2022 · 3dd174e · 3dd174e
1 parent 01ec0aa
commit 3dd174e
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/src/hooks/index.ts b/src/hooks/index.ts
@@ -57,8 +57,10 @@ const hooks: Array<HookOptions> = [
       layoutHtml,
       shortcodes,
       allRequests,
+      perf,
     }) => {
       const ShortcodeParser = prepareShortcodeParser({
+        perf,
         shortcodes,
         helpers,
         data,
@@ -215,7 +217,7 @@ const hooks: Array<HookOptions> = [
           console.log(`${Math.round(timings.slice(-1)[0].duration * 10) / 10}ms: \t ${request.permalink}`);
           displayPerfTimings([...timings]);
         } else {
-          console.log(request.permalink);
+          console.log(request.req.path);
         }
       }
     },

diff --git a/src/utils/__tests__/prepareShortcodeParser.spec.ts b/src/utils/__tests__/prepareShortcodeParser.spec.ts
@@ -20,6 +20,10 @@ jest.mock('@elderjs/shortcodes', () => (opts) => new ShortcodeParser(opts));
 jest.mock('../createReadOnlyProxy');
 
 const args = {
+  perf: {
+    start: () => '',
+    end: () => '',
+  },
   helpers: {},
   data: {},
   request: {},

diff --git a/src/utils/prepareShortcodeParser.ts b/src/utils/prepareShortcodeParser.ts
@@ -13,6 +13,7 @@ function prepareShortcodeParser({
   cssStack,
   headStack,
   customJsStack,
+  perf,
 }) {
   const { openPattern, closePattern } = settings.shortcodes;
   const shortcodeParser = ShortcodeParser({ openPattern, closePattern });
@@ -26,7 +27,11 @@ function prepareShortcodeParser({
       );
 
     shortcodeParser.add(shortcode.shortcode, async (props, content) => {
+      perf.start(shortcode.shortcode);
+      console.log('hit', shortcode, perf);
+
       const shortcodeResponse = await shortcode.run({
+        perf,
         props,
         content,
         plugin: shortcode.plugin,
@@ -86,9 +91,11 @@ function prepareShortcodeParser({
             string: head,
           });
         }
+        perf.end(shortcode.shortcode);
         return html || '';
       }
 
+      perf.end(shortcode.shortcode);
       return shortcodeResponse || '';
     });
   });