Add support for arquero (#25)

Author: DonJayamanne

package-lock.json

@@ -389,6 +389,7 @@
         "@types/yargs": "^17.0.2",
         "@typescript-eslint/eslint-plugin": "^4.29.0",
         "@typescript-eslint/parser": "^4.29.0",
+        "arquero": "^4.8.4",
         "cache-loader": "^4.1.0",
         "concurrently": "^5.2.0",
         "copy-webpack-plugin": "^9.0.0",

package.json

@@ -14,6 +14,7 @@ import { register as registerTsNode } from './tsnode';
 import { noop } from '../coreUtils';
 import { createConsoleOutputCompletedMarker } from '../const';
 import { DanfoNodePlotter } from './extensions/danforPlotter';
+import { ArqueroFormatter } from './extensions/arqueroFormatter';
 // eslint-disable-next-line @typescript-eslint/no-var-requires
 const Module = require('module');
 
@@ -153,9 +154,6 @@ async function runCode(
     }
 }
 
-async function runCodeSilently(code: string | CodeObject): Promise<unknown> {
-    return runCode(code, 'silent');
-}
 async function replEvalCode(code, _context, _filename, _callback) {
     return runCode(code);
 }
@@ -303,15 +301,14 @@ Module._load = function (request: any, parent: any) {
         });
         return TensorflowJsVisualizer.initialize(tfCore, tfMath);
     }
-    if (request === '@tensorflow/tfjs-node') {
-        // injectCustomProgress();
-        // TensorFormatter.initialize();
-    }
 
     // eslint-disable-next-line prefer-rest-params
     const result = originalLoad.apply(this, arguments);
     if (request === 'danfojs-node') {
-        DanfoJsFormatter.initialize(runCodeSilently, result);
+        DanfoJsFormatter.initialize(result);
+    }
+    if (request === 'arquero') {
+        ArqueroFormatter.initialize(result);
     }
     return result;
 };

src/extension/server/codeExecution.ts

@@ -0,0 +1,13 @@
+export class ArqueroFormatter {
+    public static initialize(arquero: typeof import('arquero')) {
+        var table = arquero.table({ colA: ['a', 'b', 'c'], colB: [3, 4, 5] });
+        const proto = (table as any).__proto__ || (table as any).protptype;
+        if (!proto) {
+            return;
+        }
+        proto.print = function (opts) {
+            const { display } = require('node-kernel');
+            display.html(this.toHTML(opts));
+        };
+    }
+}

src/extension/server/extensions/arqueroFormatter.ts

@@ -20,8 +20,8 @@ export class DanfoJsFormatter {
         DanfoJsFormatter._instance = new DanfoJsFormatter();
         return DanfoJsFormatter._instance;
     }
-    public static initialize(codeRunner: (code: string) => Promise<unknown>, danfoModule: typeof dfd) {
-        DanfoJsFormatter.instance.inject(codeRunner, danfoModule);
+    public static initialize(danfoModule: typeof dfd) {
+        DanfoJsFormatter.instance.inject(danfoModule);
     }
     public canFormatAsDanfo(value: unknown) {
         if ((!value && typeof value !== 'object') || !this.danfoJs) {
@@ -68,7 +68,7 @@ export class DanfoJsFormatter {
             };
         }
     }
-    public inject(codeRunner: (code: string) => Promise<unknown>, danfoModule: typeof dfd) {
+    public inject(danfoModule: typeof dfd) {
         if (this.isLoaded || this.failedToInject) {
             return;
         }