From 9c194ee3cb2a98a14bc56523cf224ad6f2b01328 Mon Sep 17 00:00:00 2001
From: Andrey Antukh <niwi@niwi.nz>
Date: Wed, 29 Jun 2022 14:39:56 +0200
Subject: [PATCH] :bug: Fix websocket unexpected exception on exportation
 module

A regression caused by the previous commit that refactos
the websockets API and its state management.
---
 frontend/src/app/main/data/websocket.cljs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/frontend/src/app/main/data/websocket.cljs b/frontend/src/app/main/data/websocket.cljs
index 3612770d41..a0b552393c 100644
--- a/frontend/src/app/main/data/websocket.cljs
+++ b/frontend/src/app/main/data/websocket.cljs
@@ -55,6 +55,7 @@
                        (rx/filter (ptk/type? ::initialize) stream))]
 
           (->> (rx/merge
+                (rx/of #(assoc % :ws-conn ws))
                 (->> (ws/get-rcv-stream ws)
                      (rx/filter ws/message-event?)
                      (rx/map :payload)
@@ -69,6 +70,10 @@
 (defn finalize
   []
   (ptk/reify ::finalize
+    ptk/UpdateEvent
+    (update [_ state]
+      (dissoc state :ws-conn))
+
     ptk/EffectEvent
     (effect [_ _ _]
       (l/trace :hint "event:finalize" :fn "effect")