Started better typing for message handler

1 files changed, 9 insertions, 3 deletions

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);