Provide a back link to the notebook

Author: dragonstyle

src/core/handlers/embed.ts

@@ -27,21 +27,20 @@ interface EmbedHandler {
   ): Promise<{
     handled: boolean;
     markdownFragments: EitherString[];
+    resources?: string[];
   }>;
 }
 
 const kHandlers: EmbedHandler[] = [
   {
     name: "Jupyter Notebook Embed",
-    handle(filename: string, handlerContext: LanguageCellHandlerContext) {
+    handle(filename: string, _handlerContext: LanguageCellHandlerContext) {
       const markdownFragments: EitherString[] = [];
 
       // Resolve the filename into a path
-      const path = handlerContext.resolvePath(filename);
-
-      const notebookAddress = parseNotebookAddress(path);
+      const notebookAddress = parseNotebookAddress(filename);
       if (notebookAddress) {
-        const placeHolder = notebookMarkdownPlaceholder(path, {
+        const placeHolder = notebookMarkdownPlaceholder(filename, {
           echo: false,
           warning: false,
           asis: true,
@@ -51,6 +50,9 @@ const kHandlers: EmbedHandler[] = [
         return Promise.resolve({
           handled: true,
           markdownFragments,
+          resources: [
+            notebookAddress.path,
+          ],
         });
       } else {
         return Promise.resolve({
@@ -87,6 +89,12 @@ const embedHandler: LanguageHandler = {
       const result = await handler.handle(filename, handlerContext);
       if (result.handled) {
         textFragments.push(...result.markdownFragments);
+        if (result.resources) {
+          result.resources.forEach((res) => {
+            handlerContext.addResource(res);
+          });
+        }
+
         break;
       }
     }

src/core/handlers/types.ts

@@ -118,7 +118,7 @@ export interface LanguageCellHandlerContext {
     resources?: [string, string][];
   }): Promise<string[]>;
 
-  addResource: (name: string, contents: string) => void;
+  addResource: (fileName: string) => void;
   addInclude: (content: string, where: PandocIncludeType) => void;
   addHtmlDependency: (
     dep: FormatDependency,

src/core/jupyter/jupyter-embed.ts

@@ -185,6 +185,15 @@ async function notebookMarkdown(
     options,
   );
 
+  const notebookMarkdown = (cells: JupyterCellOutput[]) => {
+    const markdown = ["", `:::{notebook="${nbAddress.path}"}`];
+    markdown.push("");
+    markdown.push(cells.map((cell) => cell.markdown).join(""));
+    markdown.push("");
+    markdown.push(":::");
+    return markdown.join("\n");
+  };
+
   if (nbAddress.ids) {
     // If cellIds are present, filter the notebook to only include
     // those cells (cellIds can eiher be an explicitly set cellId, a label in the
@@ -199,7 +208,7 @@ async function notebookMarkdown(
         return cell;
       }
     });
-    return theCells.map((cell) => cell.markdown).join("");
+    return notebookMarkdown(theCells);
   } else if (nbAddress.indexes) {
     // Filter and sort based upon cell indexes
     const theCells = nbAddress.indexes.map((idx) => {
@@ -210,11 +219,11 @@ async function notebookMarkdown(
       }
       return cellOutputs[idx];
     });
-    return theCells.map((cell) => cell.markdown).join("");
+    return notebookMarkdown(theCells);
   } else {
     // Return all the cell outputs as there is no addtional
     // specification of cells
-    return cellOutputs.map((cell) => cell.markdown).join("");
+    return notebookMarkdown(cellOutputs);
   }
 }