src/network/PeerClient.ts

@@ -25,6 +25,12 @@ export class PeerClient extends Client {
     this.connection.on("open", () => {
       this.emit("connected");
     });
+    this.connection.on("close", () => {
+      this.emit("disconnected");
+    });
+    this.connection.on("error", err => {
+      this.emit("error", err);
+    });
     this.connection.on("data", data => {
       this._received(data);
     });

src/store/network/actions.ts

@@ -13,9 +13,15 @@ const actions: ActionTree<NetworkState, AppState> = {
   connect({ commit }, options: ConnectOptions) {
     const client = new PeerClient(options.playerInfo, options._customPeer);
     commit("becomeClient", { peer: client });
-    client.once("connected", () => {
+    client.on("connected", () => {
       commit("connected");
     });
+    client.on("disconnected", () => {
+      commit("disconnected");
+    });
+    client.on("error", err => {
+      commit("connectionError", err);
+    });
     client.connect(options.serverID);
   }
 };

src/store/network/mutations.ts

@@ -23,6 +23,15 @@ const mutations: MutationTree<NetworkState> = {
 
   connected(state) {
     (state as ClientNetworkState).connectionStatus = "connected";
+  },
+
+  disconnected(state) {
+    (state as ClientNetworkState).connectionStatus = "disconnected";
+  },
+
+  connectionError(state, error) {
+    (state as ClientNetworkState).connectionStatus = "error";
+    (state as ClientNetworkState).connectionError = error;
   }
 };
 

src/store/network/types.ts

@@ -23,7 +23,7 @@ export interface NoNetworkState extends SharedNetworkState {
 
 export interface ClientNetworkState extends SharedNetworkState {
   peerType: "client";
-  connectionStatus: "connecting" | "connected" | "error";
+  connectionStatus: "connecting" | "connected" | "disconnected" | "error";
   connectionError?: Error;
   peer: PeerClient;
 }