import { DefaultLogger } from '../src';
import { WebsocketClient } from '../src/websocket-client';

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

(async () => {
  const logger = {
    ...DefaultLogger,
    // silly: () => {},
  };

  const key = process.env.API_KEY;
  const secret = process.env.API_SECRET;

  // USDT Perps:
  const market = 'linear';
  // Inverse Perp
  // const market = 'inverse';
  // const market = 'spot';

  // Note: the WebsocketClient defaults to testnet. Set `livenet: true` to use live markets.
  const wsClient = new WebsocketClient(
    {
      key: key,
      secret: secret,
      market: market,
      livenet: true,
      restOptions: {
        // enable_time_sync: true,
      },
    },
    logger
  );

  // wsClient.subscribePublicSpotOrderbook('test', 'full');

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

  wsClient.on('open', (data) => {
    console.log('connection opened open:', data.wsKey);
  });
  wsClient.on('response', (data) => {
    console.log('ws 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);
  });

  // subscribe to private endpoints
  wsClient.subscribe(['position', 'execution', 'order', 'wallet']);
})();