Make connection independent of type

1 files changed, 5 insertions, 1 deletions

diff --git a/assets/src/network/channel/request.ts b/assets/src/network/channel/request.ts
index 4a1c91f..693e408 100644
--- a/assets/src/network/channel/request.ts
+++ b/assets/src/network/channel/request.ts
@@ -1,10 +1,12 @@
 import getShareToken from "../../utils/getShareToken";
 import { answer } from "../transfer/request";
-import { Connection, on } from "./connection";
+import { Connection, ConnectionState, on, updateState } from "./connection";
 import type { RequestAcceptedMessage } from "./messages/messages";
 import { joinRequestChannel } from "./socket";
 
 export async function startRequest(connection: Connection) {
+    updateState(ConnectionState.CONNECTING_CHANNEL);
+
     const requestChannel = await joinRequestChannel(
         connection.socket,
         connection.token,
@@ -13,6 +15,8 @@ export async function startRequest(connection: Connection) {
     connection.channel = requestChannel;
 
     on("request_accepted", onRequestAccepted);
+
+    updateState(ConnectionState.CONNECTED);
 }
 
 async function onRequestAccepted(message: RequestAcceptedMessage) {