Type-handling improvement

ZilaWS · Feb 20, 2024 · 061e161 · 061e161
1 parent 16d6842
commit 061e161
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -173,7 +173,8 @@ export class ZilaServer<T extends ZilaClient = ZilaClient> {
     return this._status;
   }
 
-  private _clients: Array<ZilaClient | T> = [];
+
+  private _clients: Array<T> = [];
 
   public get clients() {
     return this._clients;
@@ -312,7 +313,7 @@ export class ZilaServer<T extends ZilaClient = ZilaClient> {
         req.headers.cookie ? new Map(Object.entries(parseCookie(req.headers.cookie))) : new Map()
       );
 
-      this._clients.push(zilaSocket);
+      this._clients.push(zilaSocket as T);
 
       if (this.serverEvents.onClientConnect) {
         for (const cb of this.serverEvents.onClientConnect) {

diff --git a/test/index.test.ts b/test/index.test.ts
@@ -29,7 +29,7 @@ class MyClient extends ZilaClient {
 describe("Non-Secure", () => {
   let client: ZilaConnection;
   let server: ZilaServer<MyClient>;
-  let clientSocket: ZilaClient;
+  let clientSocket: MyClient;
 
   beforeAll(async () => {
     server = new ZilaServer<MyClient>({