From 251cba56d9103c9ac38f9b98b69ee953ce23111e Mon Sep 17 00:00:00 2001
From: Melonai <einebeere@gmail.com>
Date: Wed, 14 Jul 2021 19:36:29 +0200
Subject: Started better typing for message handler

---
 assets/src/network/channel/share_connection.ts | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

(limited to 'assets/src/network/channel/share_connection.ts')

diff --git a/assets/src/network/channel/share_connection.ts b/assets/src/network/channel/share_connection.ts
index 415cb7f..1bb4f14 100644
--- a/assets/src/network/channel/share_connection.ts
+++ b/assets/src/network/channel/share_connection.ts
@@ -7,7 +7,7 @@ import {
     start,
     updateState,
 } from "./connection";
-import type { UnregisterHandler } from "./messages/handler";
+import type { Unregister } from "./messages/event_handler";
 import type {
     NewRequestMessage,
     RequestCancelledMessage,
@@ -38,12 +38,18 @@ function onNewRequest(message: NewRequestMessage) {
     const request = newIncomingRequest(token);
     requests.addRequest(request);
 
-    onWithToken("request_cancelled", token, onRequestCancelled);
+    const unregister = onWithToken(
+        "request_cancelled",
+        token,
+        (message: RequestCancelledMessage) => {
+            onRequestCancelled(message, unregister);
+        }
+    );
 }
 
 function onRequestCancelled(
     message: RequestCancelledMessage,
-    unregister: UnregisterHandler
+    unregister: Unregister
 ) {
     const token = message.token;
     requests.removeRequest(token);
-- 
cgit 1.4.1