121 lines
2.8 KiB
TypeScript
121 lines
2.8 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any */
|
|
import { parseResolution, getNextBarTime, socketUrl } from './helpers'
|
|
|
|
let subscriptionItem: any = {}
|
|
|
|
// Create WebSocket connection.
|
|
const socket = new WebSocket(socketUrl, 'echo-protocol')
|
|
|
|
// Connection opened
|
|
socket.addEventListener('open', (_event) => {
|
|
console.log('[socket] Connected birdeye')
|
|
})
|
|
|
|
// Listen for messages
|
|
socket.addEventListener('message', (msg) => {
|
|
const data = JSON.parse(msg.data)
|
|
|
|
if (data.type !== 'BASE_QUOTE_PRICE_DATA') return console.warn(data)
|
|
|
|
const currTime = data.data.unixTime * 1000
|
|
const lastBar = subscriptionItem.lastBar
|
|
|
|
if (
|
|
data.data.baseAddress !== subscriptionItem.baseAddress ||
|
|
data.data.quoteAddress !== subscriptionItem.quoteAddress
|
|
)
|
|
return
|
|
|
|
const resolution = subscriptionItem.resolution
|
|
const nextBarTime = getNextBarTime(lastBar, resolution)
|
|
|
|
let bar
|
|
if (currTime >= nextBarTime) {
|
|
bar = {
|
|
time: nextBarTime,
|
|
open: data.data.o,
|
|
high: data.data.h,
|
|
low: data.data.l,
|
|
close: data.data.c,
|
|
volume: data.data.v,
|
|
}
|
|
} else {
|
|
bar = {
|
|
...lastBar,
|
|
high: Math.max(lastBar.high, data.data.h),
|
|
low: Math.min(lastBar.low, data.data.l),
|
|
close: data.data.c,
|
|
volume: data.data.v,
|
|
}
|
|
}
|
|
|
|
subscriptionItem.lastBar = bar
|
|
subscriptionItem.callback(bar)
|
|
})
|
|
|
|
export function subscribeOnStream(
|
|
symbolInfo: any,
|
|
resolution: any,
|
|
onRealtimeCallback: any,
|
|
subscriberUID: any,
|
|
onResetCacheNeededCallback: any,
|
|
lastBar: any,
|
|
) {
|
|
subscriptionItem = {
|
|
resolution,
|
|
lastBar,
|
|
callback: onRealtimeCallback,
|
|
baseAddress: symbolInfo.base_token,
|
|
quoteAddress: symbolInfo.quote_token,
|
|
}
|
|
|
|
const msg = {
|
|
type: 'SUBSCRIBE_BASE_QUOTE_PRICE',
|
|
data: {
|
|
chartType: parseResolution(resolution),
|
|
baseAddress: symbolInfo.base_token,
|
|
quoteAddress: symbolInfo.quote_token,
|
|
},
|
|
}
|
|
|
|
if (!isOpen(socket)) {
|
|
console.warn('Socket Closed')
|
|
socket.addEventListener('open', (_event) => {
|
|
if (!msg.data.baseAddress || msg.data.quoteAddress) return
|
|
socket.send(JSON.stringify(msg))
|
|
})
|
|
return
|
|
}
|
|
console.warn('[subscribeBars birdeye]')
|
|
if (msg.data.baseAddress && msg.data.quoteAddress) {
|
|
socket.send(JSON.stringify(msg))
|
|
}
|
|
}
|
|
|
|
export function unsubscribeFromStream() {
|
|
const msg = {
|
|
type: 'UNSUBSCRIBE_BASE_QUOTE_PRICE',
|
|
}
|
|
|
|
if (!isOpen(socket)) {
|
|
console.warn('Socket Closed')
|
|
return
|
|
}
|
|
console.warn('[unsubscribeBars birdeye]')
|
|
socket.send(JSON.stringify(msg))
|
|
}
|
|
|
|
export function closeSocket() {
|
|
if (!isOpen(socket)) {
|
|
console.warn('Socket Closed birdeye')
|
|
return
|
|
}
|
|
console.warn('[closeSocket birdeye]')
|
|
socket.close()
|
|
}
|
|
|
|
export function isOpen(ws?: WebSocket) {
|
|
const sock = ws || socket
|
|
return sock.readyState === sock.OPEN
|
|
}
|