configurapi-runner-ws 1.19.0 → 1.20.1

package/package.json

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

package/src/index.js

@@ -5,7 +5,8 @@ const events = require('events');
 const Configurapi = require('configurapi');
 const { WebSocketServer } = require('ws');
 const { randomUUID } = require("node:crypto");
-const { Response, Route, Policy, PolicyHandlerLoader, LogLevel } = require('configurapi');
+const { Route, Policy, PolicyHandlerLoader, LogLevel } = require('configurapi');
+const { WsFrame } = require('configurapi-handler-ws');
 const oneliner = require('one-liner');
 const wsAdapter = require('./wsAdapter');
 const URL = require('url');
@@ -64,7 +65,9 @@ module.exports = class HttpRunner extends events.EventEmitter
 
             if(connectionId && connections.has(connectionId))
             {
-                e.response.headers = { ... (e.request.headers || {}), 'connection-id': connectionId }
+                e.payload.headers = e.payload.headers || {};
+                e.payload.headers['connection-id'] = connectionId;
+                
                 await wsAdapter.write(connections.get(connectionId).ws, e.payload);
             }
             else
@@ -239,7 +242,7 @@ module.exports = class HttpRunner extends events.EventEmitter
                     event.response.headers['Sec-WebSocket-Protocol'] = protocol;
                 }
 
-                await wsAdapter.write(ws, event.response); 
+                await wsAdapter.write(ws, new WsFrame(event.response)); 
             }
             else
             {
@@ -251,7 +254,7 @@ module.exports = class HttpRunner extends events.EventEmitter
                 }
                 event.response.headers['message-id'] = event.request.headers?.['message-id'];
 
-                await wsAdapter.write(ws, event.response); // Send response back to the caller.
+                await wsAdapter.write(ws, new WsFrame(event.response)); // Send response back to the caller.
             }
 
             return event.response;
@@ -268,7 +271,7 @@ module.exports = class HttpRunner extends events.EventEmitter
             
             try
             {
-                await wsAdapter.write(ws, response);
+                await wsAdapter.write(ws, new WsFrame(response));
             }
             catch(err)
             {