sdk/nodejs: add send() and json() express-esque methods

lithdew · Jun 15, 2020 · 4f34ace · 4f34ace
1 parent 8fbccf1
commit 4f34ace
Show file tree

Hide file tree

Showing 3 changed files with 385 additions and 10 deletions.
diff --git a/nodejs/src/flatend.ts b/nodejs/src/flatend.ts
@@ -49,6 +49,15 @@ export class Context extends Duplex {
         return this;
     }
 
+    send(data: string | Buffer | Uint8Array) {
+        this.write(data);
+        this.end();
+    }
+
+    json(data: any) {
+        this.send(JSON.stringify(data));
+    }
+
     _writeHeader() {
         if (this._written) return;
 

diff --git a/nodejs/src/index.ts b/nodejs/src/index.ts
@@ -1,16 +1,10 @@
-import {Node, Context} from "./flatend";
+import {Context, Node} from "./flatend";
 import * as fs from "fs";
 
 async function main() {
     const node = new Node();
-    node.register("get_todos", (ctx: Context) => {
-        ctx.write("hello world");
-        ctx.end();
-    });
-    node.register("all_todos", (ctx: Context) => {
-        const stream = fs.createReadStream("tsconfig.json");
-        stream.pipe(ctx);
-    });
+    node.register("get_todos", (ctx: Context) => ctx.json(ctx.headers));
+    node.register("all_todos", (ctx: Context) => fs.createReadStream("tsconfig.json").pipe(ctx));
     node.register('pipe', (ctx: Context) => ctx.pipe(ctx));
     await node.dial("0.0.0.0:9000");
 }