diff --git a/main.js b/main.js index 09598bc..c99fa92 100644 --- a/main.js +++ b/main.js @@ -4,7 +4,7 @@ import {lookupToken} from "./token.js"; import {httpServer, io} from "./io.js"; import {ensureVault, loginAddress} from "./vault.js"; import {chainInfo, VAULT_INIT_CODE_HASH} from "./chain.js"; -import {subPools, unsubPools} from "./pool.js"; +import {subOHLCs, subPools, unsubOHLCs, unsubPools} from "./pool.js"; import {faucet} from "./faucet.js"; @@ -17,6 +17,8 @@ io.on("connection", (socket) => { socket.on('address', (chainId, address) => loginAddress(socket, chainId, address) ) socket.on('subPools', (chainId, addresses) => subPools(socket, chainId, addresses) ) socket.on('unsubPools', (chainId, addresses) => unsubPools(socket, chainId, addresses) ) + socket.on('subOHLCs', (chainId, poolPeriods) => subOHLCs(socket, chainId, poolPeriods) ) + socket.on('unsubOHLCs', (chainId, poolPeriods) => unsubOHLCs(socket, chainId, poolPeriods) ) socket.on('ensureVault', (chainId,owner,num) => ensureVault(socket, chainId, owner, num) ) if( 31337 in chainInfo && chainInfo[31337]?.mockenv || 1337 in chainInfo && chainInfo[1337]?.mockenv ) socket.on('faucet', async (chainId, owner) => await faucet(chainId, owner)) diff --git a/pool.js b/pool.js index abf5590..b5413a9 100644 --- a/pool.js +++ b/pool.js @@ -14,3 +14,21 @@ export function unsubPools( socket, chainId, addresses ) { console.log('left room', room) } } + + +export function subOHLCs( socket, chainId, poolPeriods) { + for(const key of poolPeriods) { + const room = `${chainId}|${key}`; + socket.join(room) + console.log('joined room', room) + } +} + +export function unsubOHLCs( socket, chainId, poolPeriods ) { + for(const key of poolPeriods) { + const room = `${chainId}|${key}`; + socket.leave(room) + console.log('left room', room) + } +} +