From f5f53c6af4199660e452b241d71f6b71299be56e Mon Sep 17 00:00:00 2001
From: tim <tim@dexorder.trade>
Date: Wed, 26 Feb 2025 17:28:23 -0400
Subject: [PATCH] re-establish subscriptions after ws disconnect

---
 src/blockchain/ohlcs.js   | 10 ++++++++++
 src/components/Status.vue |  1 -
 src/socket.js             |  7 ++++++-
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/blockchain/ohlcs.js b/src/blockchain/ohlcs.js
index 462349b..b5f9574 100644
--- a/src/blockchain/ohlcs.js
+++ b/src/blockchain/ohlcs.js
@@ -17,6 +17,16 @@ export function subOHLC( chainId, pool, period ) {
 }
 
 
+export function refreshOHLCSubs() {
+    const keys = []
+    for (const key of Object.keys(ohlcSubCounts)) {
+        const [chainId, pool, period] = key.split('|')
+        keys.push(`${pool}|${period}`)
+    }
+    socket.emit('subOHLCs', chainId, keys)
+}
+
+
 export function unsubOHLC( chainId, pool, period ) {
     const key = `${pool}|${period}`
     const ckey = `${chainId}|${key}`
diff --git a/src/components/Status.vue b/src/components/Status.vue
index 9948f6f..646a755 100644
--- a/src/components/Status.vue
+++ b/src/components/Status.vue
@@ -318,7 +318,6 @@ const orders = computed(()=>{
       st.id = `${vault}|${index}`
       st.index = parseInt(index)
       // st.startTime = timestampString(st.startTime)
-      console.log('starttime', st.startTime)
 /*
       o.tranches = o.tranches.map((tranche)=>{
         const t = {...tranche}
diff --git a/src/socket.js b/src/socket.js
index 5cd481f..6228177 100644
--- a/src/socket.js
+++ b/src/socket.js
@@ -4,12 +4,17 @@ import {flushOrders} from "@/blockchain/wallet.js";
 import {parseElaboratedOrderStatus} from "@/blockchain/orderlib.js";
 import { DataFeed } from "./charts/datafeed";
 import {notifyFillEvent} from "@/notify.js";
+import {refreshOHLCSubs} from "@/blockchain/ohlcs.js";
 
 export const socket = io(import.meta.env.VITE_WS_URL || undefined, {transports: ["websocket"]})
 
 socket.on('connect', () => {
     console.log(new Date(), 'ws connected')
-    useStore().connected = true
+    const s = useStore();
+    s.connected = true
+    if (s.chainId && s.account)
+        socket.emit('address', s.chainId, s.account)
+    refreshOHLCSubs()
 })
 
 socket.on('disconnect', () => {