From b98b1cd0e5d07fd37be54592edd362f443a145ad Mon Sep 17 00:00:00 2001
From: Melonai <einebeere@gmail.com>
Date: Mon, 14 Jun 2021 11:55:03 +0200
Subject: Make connection independent of type

---
 assets/src/network/channel/share.ts | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

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

diff --git a/assets/src/network/channel/share.ts b/assets/src/network/channel/share.ts
index 6dbf0cd..81942af 100644
--- a/assets/src/network/channel/share.ts
+++ b/assets/src/network/channel/share.ts
@@ -1,5 +1,11 @@
 import requests from "../../stores/requests";
-import { Connection, on, onWithToken } from "./connection";
+import {
+    Connection,
+    ConnectionState,
+    on,
+    onWithToken,
+    updateState,
+} from "./connection";
 import type { UnregisterHandler } from "./messages/handler";
 import type {
     NewRequestMessage,
@@ -8,6 +14,8 @@ import type {
 import { joinShareChannel } from "./socket";
 
 export async function startShare(connection: Connection) {
+    updateState(ConnectionState.CONNECTING_CHANNEL);
+
     const shareChannel = await joinShareChannel(
         connection.socket,
         connection.token
@@ -15,6 +23,8 @@ export async function startShare(connection: Connection) {
     connection.channel = shareChannel;
 
     on("new_request", onNewRequest);
+
+    updateState(ConnectionState.CONNECTED);
 }
 
 function onNewRequest(message: NewRequestMessage) {
-- 
cgit 1.4.1