import { DefaultLogger, WS_KEY_MAP, WebsocketClient } from '../src';

// or
// import { DefaultLogger, WS_KEY_MAP, WebsocketClient } from 'bybit-api';

(async () => {
  const logger = {
    ...DefaultLogger,
    silly: (...params) => console.log('silly', ...params),
  };

  const wsClient = new WebsocketClient(
    {
      // key: key,
      // secret: secret,
      // market: 'linear',
      // market: 'inverse',
      // market: 'spot',
      // market: 'spotv3',
      // market: 'usdcOption',
      // market: 'usdcPerp',
      market: 'unifiedPerp',
      // market: 'unifiedOption',
    },
    logger
  );

  wsClient.on('update', (data) => {
    console.log('raw message received ', JSON.stringify(data));
    // console.log('raw message received ', JSON.stringify(data, null, 2));
  });

  wsClient.on('open', (data) => {
    console.log('connection opened open:', data.wsKey);

    // if (data.wsKey === WS_KEY_MAP.spotPublic) {
    //   // Spot public, but not recommended - use spotv3 client instead
    //   // The old spot websockets dont automatically resubscribe if they disconnect
    //   // wsClient.subscribePublicSpotTrades('BTCUSDT');
    //   // wsClient.subscribePublicSpotTradingPair('BTCUSDT');
    //   // wsClient.subscribePublicSpotV1Kline('BTCUSDT', '1m');
    //   // wsClient.subscribePublicSpotOrderbook('BTCUSDT', 'full');
    // }
  });
  wsClient.on('response', (data) => {
    console.log('log response: ', JSON.stringify(data, null, 2));
  });
  wsClient.on('reconnect', ({ wsKey }) => {
    console.log('ws automatically reconnecting.... ', wsKey);
  });
  wsClient.on('reconnected', (data) => {
    console.log('ws has reconnected ', data?.wsKey);
  });

  // Inverse
  // wsClient.subscribe('trade');

  // Linear
  // wsClient.subscribe('trade.BTCUSDT');

  // Spot V3
  // wsClient.subscribe('trade.BTCUSDT');
  // Or an array of topics
  // wsClient.subscribe(['trade.BTCUSDT', 'trade.LTCUSDT']);

  // usdc options
  // wsClient.subscribe([
  //   `recenttrades.BTC`,
  //   `recenttrades.ETH`,
  //   `recenttrades.SOL`,
  // ]);

  // usdc perps
  // wsClient.subscribe('trade.BTCPERP');

  // unified perps
  wsClient.subscribe('publicTrade.BTCUSDT');

  // For spot v1 (the old, deprecated client), request public connection first then send required topics on 'open'
  // Not necessary for spot v3
  // wsClient.connectPublic();

  // To unsubscribe from topics:
  // setTimeout(() => {
  //   console.log('unsubscribing');
  //   wsClient.unsubscribe('trade.BTCUSDT');
  // }, 5 * 1000);

  // Topics are tracked per websocket type
  // Get a list of subscribed topics (e.g. for public v3 spot topics)
  const publicSpotTopics = wsClient
    .getWsStore()
    .getTopics(WS_KEY_MAP.spotV3Public);

  console.log('public spot topics: ', publicSpotTopics);

  const privateSpotTopics = wsClient
    .getWsStore()
    .getTopics(WS_KEY_MAP.spotV3Private);
  console.log('private spot topics: ', publicSpotTopics);
})();