Inject widget dependencies if they are present

Author: dragonstyle

src/command/render/render.ts

@@ -51,6 +51,7 @@ import {
 } from "../../core/timing.ts";
 import { filesDirMediabagDir } from "./render-paths.ts";
 import { replaceNotebookPlaceholders } from "../../core/jupyter/jupyter-embed.ts";
+import { kIncludeAfterBody, kIncludeInHeader } from "../../config/constants.ts";
 
 export async function renderPandoc(
   file: ExecutedFile,
@@ -105,17 +106,33 @@ export async function renderPandoc(
   ensureDirSync(join(dirname(context.target.source), mediabagDir));
 
   // Process any placeholder for notebooks that have been injected
-  const markdown = await replaceNotebookPlaceholders(
+  const notebookResult = await replaceNotebookPlaceholders(
     format.pandoc.to || "html",
     context.target.source,
     context,
     context.options.flags || {},
     executeResult.markdown,
   );
 
+  // Map notebook includes to pandoc includes
+  const pandocIncludes: PandocIncludes = {
+    [kIncludeAfterBody]: notebookResult.includes?.afterBody
+      ? [notebookResult.includes?.afterBody]
+      : undefined,
+    [kIncludeInHeader]: notebookResult.includes?.inHeader
+      ? [notebookResult.includes?.inHeader]
+      : undefined,
+  };
+
+  // Inject dependencies
+  format.pandoc = mergePandocIncludes(
+    format.pandoc,
+    pandocIncludes,
+  );
+
   // pandoc options
   const pandocOptions: PandocOptions = {
-    markdown,
+    markdown: notebookResult.markdown,
     source: context.target.source,
     output: recipe.output,
     keepYaml: recipe.keepYaml,

src/core/jupyter/jupyter-embed.ts

@@ -23,6 +23,7 @@ import {
 } from "../../config/constants.ts";
 import {
   isHtmlCompatible,
+  isHtmlOutput,
   isIpynbOutput,
   isLatexOutput,
   isMarkdownOutput,
@@ -34,6 +35,11 @@ import { JupyterAssets, JupyterCellOutput } from "../jupyter/types.ts";
 
 import { dirname, extname } from "path/mod.ts";
 import { languages } from "../handlers/base.ts";
+import {
+  extractJupyterWidgetDependencies,
+  includesForJupyterWidgetDependencies,
+} from "./widgets.ts";
+import { globalTempContext } from "../temp.ts";
 
 export interface JupyterNotebookAddress {
   path: string;
@@ -133,6 +139,7 @@ export async function replaceNotebookPlaceholders(
   markdown: string,
 ) {
   let match = kPlaceholderRegex.exec(markdown);
+  let includes;
   while (match) {
     // Find and parse the placeholders
     const nbPath = match[1];
@@ -150,6 +157,23 @@ export async function replaceNotebookPlaceholders(
         to,
       );
 
+      const notebookIncludes = () => {
+        const notebook = jupyterFromFile(nbAddress.path);
+        const dependencies = isHtmlOutput(context.format.pandoc)
+          ? extractJupyterWidgetDependencies(notebook)
+          : undefined;
+        if (dependencies) {
+          const tempDir = globalTempContext().createDir();
+          return includesForJupyterWidgetDependencies(
+            [dependencies],
+            tempDir,
+          );
+        } else {
+          return undefined;
+        }
+      };
+      includes = notebookIncludes();
+
       // Render the notebook markdown
       const nbMarkdown = await notebookMarkdown(
         nbAddress,
@@ -166,7 +190,10 @@ export async function replaceNotebookPlaceholders(
     match = kPlaceholderRegex.exec(markdown);
   }
   kPlaceholderRegex.lastIndex = 0;
-  return markdown;
+  return {
+    includes,
+    markdown,
+  };
 }
 
 async function notebookMarkdown(