prettier output tables

Author: markbaas

R/notebook.R

@@ -80,21 +80,27 @@ r$run(function() {
           )
         } else if (out$visible) {
           if (inherits(out$value, "data.frame")) {
-            text <- utils::capture.output(knitr::kable(out$value, format = "html"))
+            table <- head(out$value, 10)
+            list(
+              id = id,
+              type = "table",
+              result = list(
+                html = knitr::kable(table, format = "html"),
+                markdown = paste0(knitr::kable(table, format = "markdown"), collapse = "\n")
+              )
+            )
+          } else {
+            list(
+              id = id,
+              type = "text",
+              result = paste0(text, collapse = "\n")
+            )
           }
-
-          list(
-            id = id,
-            type = "text",
-            result = paste0(text, collapse = "\n"),
-            tempdir = plot_dir
-          )
         } else {
           list(
             id = id,
             type = "text",
-            result = "",
-            tempdir = plot_dir
+            result = ""
           )
         }
       }, error = function(e) {

src/notebook.ts

@@ -4,7 +4,6 @@ import { spawn, ChildProcess } from 'child_process';
 import { dirname } from 'path';
 import { inlineAll } from './inlineScripts';
 
-
 interface RKernelRequest {
   id: number;
   type: 'eval' | 'cancel';
@@ -339,7 +338,6 @@ export class RNotebookProvider implements vscode.NotebookContentProvider, vscode
   }
 
   async renderHtmlOutput(response) {
-
   	const html = (await vscode.workspace.fs.readFile(vscode.Uri.parse(response.result))).toString();
     const htmlDir = dirname(response.result)
     const htmlInline = await inlineAll(html, htmlDir)
@@ -358,6 +356,26 @@ export class RNotebookProvider implements vscode.NotebookContentProvider, vscode
     }
   }
 
+  async renderTableOutput(response) {
+    const html = `
+      <div id="chunk-table-${response.id}"></div>
+      <script>
+        $('#example').DataTable({
+          data: JSON.parse('${JSON.stringify(response.result)}')
+        })
+      </script>
+    `
+
+    return {
+      outputKind: vscode.CellOutputKind.Rich,
+      data: {
+        'text/html': response.result.html,
+        'text/markdown': response.result.markdown,
+        'text/plain': response.result.markdown
+      },
+    }
+  }
+
   async renderOutput(cell, response) {
 
     switch (response.type) {
@@ -382,6 +400,10 @@ export class RNotebookProvider implements vscode.NotebookContentProvider, vscode
         }];
         break;
       }
+      case 'table': {
+        cell.outputs = [await this.renderTableOutput(response)];
+        break;
+      }
       case 'error': {
         cell.metadata.runState = vscode.NotebookCellRunState.Error;
         cell.outputs = [{