configurapi-runner-ws 1.17.0 → 1.19.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "configurapi-runner-ws",
-  "version": "1.17.0",
+  "version": "1.19.0",
   "description": "Websocket runner for configurapi.",
   "bin": {
     "configurapi-runner-ws": "src/app.mjs"

package/src/index.js

@@ -64,9 +64,8 @@ module.exports = class HttpRunner extends events.EventEmitter
 
             if(connectionId && connections.has(connectionId))
             {
-                let wsResponse = new Response(e.payload, 200, {...e.request.headers, 'connection-id': connectionId})
-                await wsAdapter.write(connections.get(connectionId).ws, wsResponse)
-                e.response.statusCode = 204
+                e.response.headers = { ... (e.request.headers || {}), 'connection-id': connectionId }
+                await wsAdapter.write(connections.get(connectionId).ws, e.payload);
             }
             else
             {
@@ -296,7 +295,7 @@ module.exports = class HttpRunner extends events.EventEmitter
             await this.service.process(event);
 
             serverResponse.writeHead(200, event.response.headers);
-            serverResponse.end(JSON.stringify(event.response));
+            serverResponse.end(JSON.stringify(event.response.body));
         }
         catch (error) 
         {

package/src/wsAdapter.js

@@ -3,6 +3,7 @@ const { WebSocket } = require('ws');
 const StreamResponse = require('configurapi-handler-ws').StreamResponse;
 const Response = require('configurapi').Response;
 const { randomUUID } = require("node:crypto");
+const WsFrame = require('configurapi-handler-ws').WsFrame;
 
 module.exports = {
     write: async function(ws, message)
@@ -72,9 +73,11 @@ module.exports = {
                         }
 
                         const headers = { 'message-id': streamId, 'x-stream': 'chunk', 'x-stream-id': streamId, 'Content-Type': 'text/plain' };
-                        const response = new Response(bucket, 200, headers);
+                        
+                        const chunkResponse = new Response(bucket, 200, { });
+                        const frame = new WsFrame(chunkResponse, headers);
 
-                        ws.send(JSON.stringify(response), (err) =>
+                        ws.send(JSON.stringify(frame), (err) =>
                         {
                             if (err) { cleanup(); return safeReject(err); }
                         });
@@ -140,9 +143,11 @@ module.exports = {
                         if (ws.readyState === WebSocket.OPEN) 
                         {
                             const headers = { 'message-id': streamId, 'x-stream-id': streamId, 'x-stream': 'done' };
-                            const response = new Response('', 204, headers);
                             
-                            ws.send(JSON.stringify(response), (err) =>
+                            const doneResponse = new Response('', 204);
+                            const frame = new WsFrame(doneResponse, headers);
+
+                            ws.send(JSON.stringify(frame), (err) =>
                             {
                                 cleanup();
                                 return err ? safeReject(err) : safeResolve();

package/src/wsPostbackFunction.js

@@ -2,17 +2,10 @@ const ErrorResponse = require('configurapi').ErrorResponse;
 
 async function wsPostbackFunction(endpoint, response) 
 {
-    let message = response;
-
-    if(message?.statusCode && message?.headers)
-    {
-        message = message.body;
-    }
-
     let resp = await fetch(endpoint, {
         method: 'POST',
-        headers: response.headers || {},
-        body: typeof message === 'object' ? JSON.stringify(message) : message
+        headers: {...response.headers, 'content-type': 'application/json'},
+        body: typeof response === 'string' ? response: JSON.stringify(response)
     });
 
     if(resp.status >= 400)