feat(node-runtime): add query to ctx (#814)

josecleiton · web-flow · commit bd408ed00a0b · 2024-03-26T14:48:12.000-03:00
diff --git a/packages/node-runtime/src/context.ts b/packages/node-runtime/src/context.ts
@@ -14,6 +14,7 @@ export interface ContextRequest {
   };
   extra: Record<string, unknown>;
   headers: IncomingHttpHeaders;
+  query: string | null;
   id: string;
   ip: string;
   name: string;
diff --git a/packages/node-runtime/src/http-server.ts b/packages/node-runtime/src/http-server.ts
@@ -520,6 +520,7 @@ export class SdkgenHttpServer<ExtraContextT = unknown> {
               headers: req.headers,
               id: randomBytes(16).toString("hex"),
               ip,
+              query,
               name: op.name,
               version: 3,
             };
@@ -769,7 +770,7 @@ export class SdkgenHttpServer<ExtraContextT = unknown> {
       return;
     }
 
-    const request = this.parseRequest(req, body.toString(), clientIp);
+    const request = this.parseRequest(req, body.toString(), clientIp, query);
 
     if (!request) {
       this.writeReply(
@@ -798,14 +799,14 @@ export class SdkgenHttpServer<ExtraContextT = unknown> {
     writeReply(ctx, await executeRequest(ctx, this.apiConfig));
   }
 
-  private parseRequest(req: IncomingMessage, body: string, ip: string): ContextRequest | null {
+  private parseRequest(req: IncomingMessage, body: string, ip: string, query: string | null): ContextRequest | null {
     switch (this.identifyRequestVersion(req, body)) {
       case 1:
-        return this.parseRequestV1(req, body, ip);
+        return this.parseRequestV1(req, body, ip, query);
       case 2:
-        return this.parseRequestV2(req, body, ip);
+        return this.parseRequestV2(req, body, ip, query);
       case 3:
-        return this.parseRequestV3(req, body, ip);
+        return this.parseRequestV3(req, body, ip, query);
       default:
         throw new Error("Failed to understand request");
     }
@@ -826,7 +827,7 @@ export class SdkgenHttpServer<ExtraContextT = unknown> {
   }
 
   // Old Sdkgen format
-  private parseRequestV1(req: IncomingMessage, body: string, ip: string): ContextRequest {
+  private parseRequestV1(req: IncomingMessage, body: string, ip: string, query: string | null): ContextRequest {
     const parsed = decode(
       {
         Request: {
@@ -871,14 +872,15 @@ export class SdkgenHttpServer<ExtraContextT = unknown> {
       files: [],
       headers: req.headers,
       id: `${deviceId}-${parsed.id}`,
+      query,
       ip,
       name: parsed.name,
       version: 1,
     };
   }
 
   // Maxima sdkgen format
-  private parseRequestV2(req: IncomingMessage, body: string, ip: string): ContextRequest {
+  private parseRequestV2(req: IncomingMessage, body: string, ip: string, query: string | null): ContextRequest {
     const parsed = decode(
       {
         Request: {
@@ -926,14 +928,15 @@ export class SdkgenHttpServer<ExtraContextT = unknown> {
       files: [],
       headers: req.headers,
       id: `${parsed.deviceId}-${parsed.requestId ?? randomBytes(16).toString("hex")}`,
+      query,
       ip,
       name: parsed.name,
       version: 2,
     };
   }
 
   // New sdkgen format
-  private parseRequestV3(req: IncomingMessage, body: string, ip: string): ContextRequest {
+  private parseRequestV3(req: IncomingMessage, body: string, ip: string, query: string | null): ContextRequest {
     const parsed = decode(
       {
         DeviceInfo: {
@@ -989,6 +992,7 @@ export class SdkgenHttpServer<ExtraContextT = unknown> {
       headers: req.headers,
       id: `${deviceId}-${parsed.requestId ?? randomBytes(16).toString("hex")}`,
       ip,
+      query,
       name: parsed.name,
       version: 3,
     };
diff --git a/packages/node-runtime/src/test-wrapper.ts b/packages/node-runtime/src/test-wrapper.ts
@@ -39,6 +39,7 @@ export function apiTestWrapper<ExtraContextT, ApiT extends BaseApiConfig<ExtraCo
           ip: partialCtx.request?.ip ?? "0.0.0.0",
           name: functionName,
           version: 3,
+          query: null,
         },
         response: {
           headers: new Map(),
