implement inplace context

package.json

@@ -69,7 +69,7 @@
         "pretty-quick": "^2.0.1",
         "ts-jest": "^24.0.2",
         "ts-node": "^8.3.0",
-        "typedraft": "0.2.2"
+        "typedraft": "0.0.0-31fb921"
     },
     "husky": {
         "hooks": {

src/code-object/module.tsx

@@ -13,7 +13,7 @@ export type Draft = Array<ExportClassCode | MethodCode | LocalContext | InlineCo
  * As we are only interested in the draft part of a module, then we need a way to return this "view" of module code.
  */
 <ModuleCode /> +
-    function ToDraft(this: ModuleCode) {
+    function ToDraft(this: ModuleCode & IModuleCode) {
         /**
          * refresh and update bindings because DSL only transforms code
          */
@@ -23,8 +23,7 @@ export type Draft = Array<ExportClassCode | MethodCode | LocalContext | InlineCo
          * traverse file and set path
          */
         let draft: Draft = [];
-        traverse<{ _module: ModuleCode }>(
-            this.m_File,
+        this.Traverse<{ _module: ModuleCode }>(
             {
                 Program(path) {
                     this._module.m_Path = path;
@@ -43,7 +42,6 @@ export type Draft = Array<ExportClassCode | MethodCode | LocalContext | InlineCo
                     }
                 },
             },
-            null,
             { _module: this }
         );
 
@@ -57,6 +55,16 @@ export type Draft = Array<ExportClassCode | MethodCode | LocalContext | InlineCo
         return draft;
     };
 
+export interface IModuleCode {
+    Traverse<S>(visitor: TraverseOptions<S>, state: S): void;
+}
+
+<ModuleCode /> +
+    function Traverse<S>(this: ModuleCode, visitor: TraverseOptions<S>, state: S = null) {
+        this.m_File = ToFile(ToString(this.m_File));
+        traverse<S>(this.m_File, visitor, null, state);
+    };
+
 /**
  * ## Create draft
  */
@@ -136,5 +144,5 @@ import { ExportClassCode } from "./export-class";
 import { MethodCode } from "./method";
 import { LocalContext } from "./local-context";
 import { ToFile, ToString } from "../common/utility";
-import traverse, { NodePath, Node } from "@babel/traverse";
+import traverse, { NodePath, Node, TraverseOptions } from "@babel/traverse";
 import { InlineContext } from "./inline-context";

src/core/transcriber.tsx

@@ -92,11 +92,13 @@ export class Transcriber {
     };
 
 export interface IDSL {
-    Transcribe(
+    Transcribe?(
         block: Array<Statement>,
         path?: NodePath<FunctionDeclaration> | NodePath<BlockStatement>
     ): Array<Statement>;
 
+    InplaceTranscribe?(literal: NodePath<TemplateLiteral>): Expression;
+
     m_Merge?: boolean;
 }
 
@@ -135,7 +137,13 @@ export type TraverseLocalContextCallback = (context: LocalContext, name: string)
 export type TraverseInlineContextCallback = (context: InlineContext) => void;
 export type TraverseMethodCallback = (methods: Array<MethodCode>, class_name: string) => void;
 
-import { Statement, FunctionDeclaration, BlockStatement } from "@babel/types";
+import {
+    Statement,
+    FunctionDeclaration,
+    BlockStatement,
+    Expression,
+    TemplateLiteral,
+} from "@babel/types";
 import { NodePath } from "@babel/traverse";
 
 import { ToString } from "../common/utility";

src/plug-in/draft-plugin-inplace-context.tsx

@@ -0,0 +1,43 @@
+import { ITranscriber } from "../core/transcriber";
+import { NodePath, Node } from "@babel/core";
+import { StringLiteral, Identifier } from "@babel/types";
+
+export class InplaceContextPlugin {
+    m_Transcriber: ITranscriber;
+}
+
+<InplaceContextPlugin /> +
+    function constructor(this: InplaceContextPlugin, transcriber: ITranscriber) {
+        this.m_Transcriber = transcriber;
+    };
+
+<InplaceContextPlugin /> +
+    function Transcribe(this: InplaceContextPlugin) {
+        const transcriber = this.m_Transcriber;
+        transcriber.m_Module.Traverse({
+            TaggedTemplateExpression(path) {
+                const tag = path.get("tag");
+                if (!tag.isCallExpression()) {
+                    return;
+                }
+
+                const callee = tag.get("callee") as NodePath<Identifier>;
+                const callee_name = callee.node.name as string;
+                if (callee_name !== "context" && callee_name !== "Λ") {
+                    return;
+                }
+
+                const [arg] = (tag.get("arguments") as Array<NodePath<Node>>) as [
+                    NodePath<StringLiteral>
+                ];
+                const dsl_name = arg.node.value;
+
+                const dsl = transcriber.m_DSLMap.get(dsl_name);
+                if (!dsl) {
+                    return;
+                }
+
+                path.replaceWith(dsl.InplaceTranscribe(path.get("quasi")));
+            },
+        });
+    };

test/plug-in/__snapshots__/inplace-context.test.ts.snap

@@ -0,0 +1,23 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`not inplace context no dsl 1`] = `
+"const temp = Λ<number>(\\"math\\")\`
+                1+2
+            \`;"
+`;
+
+exports[`not inplace context not call expression 1`] = `
+"context<number>\`
+                1+2
+            \`;"
+`;
+
+exports[`not inplace context not context or Λ 1`] = `
+"whatever<number>(\\"math\\")\`
+                1+2
+            \`;"
+`;
+
+exports[`use inplace context as expression 1`] = `"const temp = \\"3\\";"`;
+
+exports[`use inplace context as expression statement 1`] = `"\\"3\\";"`;

test/plug-in/inplace-context.test.ts

@@ -0,0 +1,84 @@
+import { MakeDefaultTranscriber, IDSL } from "../../src";
+import { InplaceContextPlugin } from "../../src/plug-in/draft-plugin-inplace-context";
+import { NodePath } from "@babel/core";
+import { TemplateLiteral, stringLiteral } from "@babel/types";
+
+class Math implements IDSL {
+    InplaceTranscribe(literal: NodePath<TemplateLiteral>) {
+        return stringLiteral("3");
+    }
+}
+
+test("use inplace context as expression", () => {
+    //
+    const code = `
+        const temp = Λ<number>("math")\`
+            1+2
+        \`;
+    `;
+
+    const transcriber = MakeDefaultTranscriber(code);
+    transcriber.m_Plugins = [new InplaceContextPlugin(transcriber)];
+    transcriber.m_DSLMap.set("math", new Math());
+    const result = transcriber.Transcribe();
+    expect(result).toMatchSnapshot();
+});
+
+test("use inplace context as expression statement", () => {
+    //
+    const code = `
+        context<number>("math")\`
+            1+2
+        \`;
+    `;
+
+    const transcriber = MakeDefaultTranscriber(code);
+    transcriber.m_Plugins = [new InplaceContextPlugin(transcriber)];
+    transcriber.m_DSLMap.set("math", new Math());
+    const result = transcriber.Transcribe();
+    expect(result).toMatchSnapshot();
+});
+
+describe("not inplace context", () => {
+    test("not call expression", () => {
+        //
+        const code = `
+            context<number>\`
+                1+2
+            \`;
+        `;
+
+        const transcriber = MakeDefaultTranscriber(code);
+        transcriber.m_Plugins = [new InplaceContextPlugin(transcriber)];
+        const result = transcriber.Transcribe();
+        expect(result).toMatchSnapshot();
+    });
+
+    test("not context or Λ", () => {
+        //
+        const code = `
+            whatever<number>("math")\`
+                1+2
+            \`;
+        `;
+
+        const transcriber = MakeDefaultTranscriber(code);
+        transcriber.m_Plugins = [new InplaceContextPlugin(transcriber)];
+        const result = transcriber.Transcribe();
+        expect(result).toMatchSnapshot();
+    });
+
+    test("no dsl", () => {
+        //
+        const code = `
+            const temp = Λ<number>("math")\`
+                1+2
+            \`;
+        `;
+
+        const transcriber = MakeDefaultTranscriber(code);
+        transcriber.m_Plugins = [new InplaceContextPlugin(transcriber)];
+        const result = transcriber.Transcribe();
+        expect(result).toMatchSnapshot();
+    });
+});