feat: Implementation for initalize(), declareFunction(), functionCall(), registerHook() and triggerHook() requests

Author: thebongy

src/main.ts

@@ -1,6 +1,6 @@
-import { Protocol, Dependency } from "./protocol/protocol";
+import { Protocol, Dependency, FnArgs } from "./protocol/protocol";
 
-class Gotham {
+export class Gotham {
     moduleId: string;
     protocol: Protocol
     constructor() {
@@ -11,4 +11,28 @@ class Gotham {
             this.protocol.initialize(moduleId, version, deps),
         );
     }
+
+    async declareFunction(fnName: string, fn: Function) {
+        return await this.protocol.sendRequest(
+            this.protocol.declareFunction(fnName, fn)
+        );
+    }
+
+    async functionCall(fnNmame: string, args: FnArgs) {
+        return await this.protocol.sendRequest(
+            this.protocol.functionCall(fnNmame, args)
+        );
+    }
+
+    async registerHook(hook: string, cb: Function) {
+        return await this.protocol.sendRequest(
+            this.protocol.registerHook(hook, cb)
+        );
+    }
+
+    async triggerHook(hook: string) {
+        return await this.protocol.sendRequest(
+            this.protocol.triggerHook(hook)
+        );
+    }
 }

src/protocol/protocol.ts

@@ -6,7 +6,7 @@ export interface Dependency {
     [type: string]: string;
 }
 
-interface FnArgs {
+export interface FnArgs {
     [type: string]: any;
 }
 
@@ -115,7 +115,7 @@ export class Protocol {
         }
     }
 
-    registerHook(moduleId: string, hook: string, cb: Function): RegisterHookRequest {
+    registerHook(hook: string, cb: Function): RegisterHookRequest {
         if (this.hookListeners[hook]) {
             this.hookListeners[hook] = [cb];
         } else {
@@ -124,7 +124,7 @@ export class Protocol {
         return {
             requestId: this.generateRequestId(),
             type: 'registerHook',
-            hook: `${moduleId}-${hook}`,
+            hook: `${this.moduleId}-${hook}`,
         }
     }
 
@@ -162,9 +162,9 @@ export class Protocol {
         }
     }
 
-    functionCall(functionName: string, ...args: FnArgs[]) {
+    functionCall(functionName: string, args: FnArgs) {
         if (this.functions[functionName]) {
-            this.functions[functionName]();
+            this.functions[functionName](...args);
         }
     }