run on application context

src/koishi.service.ts

@@ -10,7 +10,7 @@ import {
   KoishiCommandInterceptorRegistration,
   KoishiModuleOptions,
 } from './utility/koishi.interfaces';
-import { Server } from 'http';
+import { createServer, Server } from 'http';
 import Koa from 'koa';
 import KoaBodyParser from 'koa-bodyparser';
 import { KoishiMetascanService } from './providers/koishi-metascan.service';
@@ -19,6 +19,7 @@ import { KoishiLoggerService } from './providers/koishi-logger.service';
 import { KoishiHttpDiscoveryService } from './koishi-http-discovery/koishi-http-discovery.service';
 import { Filter, ReplacedContext } from './utility/replaced-context';
 import { applySelector } from 'koishi-decorators';
+import WebSocket from 'ws';
 
 // eslint-disable-next-line @typescript-eslint/no-empty-function
 Router.prepare = () => {};
@@ -59,6 +60,18 @@ export class KoishiService
   private async setHttpServer() {
     const httpAdapter = this.httpDiscovery.getHttpAdapter();
     if (!httpAdapter) {
+      this.logger('app').info('No http adapters found from Nest application.');
+      this._httpServer = createServer(this._nestKoaTmpInstance.callback());
+      this._wsServer = new WebSocket.Server({
+        server: this._httpServer,
+      });
+
+      this._wsServer.on('connection', (socket, request) => {
+        for (const manager of this.router.wsStack) {
+          if (manager.accept(socket, request)) return;
+        }
+        socket.close();
+      });
       return;
     }
     const httpServer: Server = httpAdapter.getHttpServer();

tests/koishi-tests.spec.ts

@@ -39,7 +39,7 @@ class KoishiTestService {
   }
 }
 
-describe('HttpServer', () => {
+describe('Koishi in Nest.js', () => {
   let app: INestApplication;
   let koishiApp: KoishiService;
 

tests/pure-context.spec.ts

+import { KoishiService } from '../src/koishi.service';
+import { KoishiModule } from '../src/koishi.module';
+import { NestFactory } from '@nestjs/core';
+import { INestApplicationContext, Module } from '@nestjs/common';
+
+@Module({
+  imports: [KoishiModule.register({})],
+})
+class TestModule {}
+
+describe('Koishi in Nest.js context', () => {
+  let app: INestApplicationContext;
+  let koishiApp: KoishiService;
+
+  beforeEach(async () => {
+    app = await NestFactory.createApplicationContext(TestModule);
+    await app.init();
+    koishiApp = app.get(KoishiService);
+  });
+
+  it('should register http service', () => {
+    expect(koishiApp._httpServer).toBeDefined();
+  });
+});