fix: make some improvements

index.d.ts

@@ -0,0 +1 @@
+export * from './src/lib/index'

package.json

@@ -73,6 +73,7 @@
     "@types/debug": "^4.1.7",
     "@types/socket.io": "^3.0.2",
     "@types/uri-templates": "^0.1.31",
+    "@types/ws": "^8.5.3",
     "@typescript-eslint/eslint-plugin": "^5.9.0",
     "@typescript-eslint/parser": "^5.9.0",
     "all-contributors-cli": "^6.14.2",

src/adapters/ws/index.ts

@@ -93,15 +93,17 @@ class WebSocketsAdapter extends Adapter {
             return
           }
         }
-      }
-
-      if (servers.has(pathname)) {
-        servers.get(pathname).handleUpgrade(request, socket, head, (ws) => {
-          servers.get(pathname).emit('server:connection:open', ws, request)
-
-          ws.on('message', (payload) => {
-            const msg = this._createMessage(pathname, payload)
-            this.emit('message', msg, ws)
+
+        if (servers.has(pathname)) {
+          servers.get(pathname).handleUpgrade(request, socket, head, (ws) => {
+            servers.get(pathname).emit('connect', ws, request)
+
+            ws.on('message', (payload) => {
+              const msg = this._createMessage(pathname, payload)
+              this.emit('message', msg, ws)
+            })
+
+            this.emit('server:connection:open', { name: this.name(), adapter: this, connection: ws, channel: pathname, request })
           })
 
           this.emit('connect', { name: this.name(), adapter: this, connection: ws, channel: pathname })

src/index.ts

@@ -67,7 +67,7 @@ export default async function GleeAppInitializer () {
     if (channel.hasPublish()) {
       const operationId = channel.publish().json('operationId')
       if (operationId) {
-        const schema = channel.publish().message().payload().json()
+        const schema = {oneOf: channel.publish().messages().map(message => message.payload().json())} as any;
         app.use(channelName, validate(schema), (event, next) => {
           triggerFunction({
             app,
@@ -78,7 +78,7 @@ export default async function GleeAppInitializer () {
       }
     }
     if (channel.hasSubscribe()) {
-      const schema = channel.subscribe().message().payload().json()
+      const schema = {oneOf: channel.subscribe().messages().map(message => message.payload().json())} as any;
       app.useOutbound(channelName, validate(schema), json2string)
     }
   })

src/lib/configs.ts

@@ -44,9 +44,7 @@ async function loadConfigsFromFile() {
     ASYNCAPI_FILE_PATH = projectConfigs.ASYNCAPI_FILE_PATH ? path.resolve(GLEE_DIR, projectConfigs.ASYNCAPI_FILE_PATH) : ASYNCAPI_FILE_PATH
     return projectConfigs
   } catch (e) {
-    if (e.code !== 'ERR_MODULE_NOT_FOUND') {
       return console.error(e)
-    }
   }
 }
 

src/lib/lifecycleEvents.ts

@@ -79,7 +79,7 @@ export async function run(lifecycleEvent: string, params: GleeFunctionEvent) {
     const responses = await Promise.all(handlers.map(info => info.fn(params)))
 
     responses.forEach(res => {
-      if (res.send) {
+      if (res?.send) {
         res.send.forEach((event: GleeFunctionReturnSend) => {
           try {
             params.glee.send(new GleeMessage({