const Questrade = require('questrade');

const qt = new Questrade('<your-api-key-here>');

// Connect to Questrade
const newKey = await qt.connect()
// Save newKey for next time

// Access your account here
const account = await qt.getPrimaryAccount()
const accounts = await qt.getAccounts()
const balances = await account.getBalances()

// Get Market quotes
const msft = await qt.getSymbol('MSFT')
const quote = await msft.getQuote()
const options = await msft.getOptionChain()

// ... etc. See the full documentation for all the calls you can make!

const account = await qt.getPrimaryAccount() // Account
const markets = await qt.getMarkets()
const accounts = await qt.getAccounts() // => [Account]

const balances = await account.getBalances()
const activities = await account.getActivities()
const orders = await qt.getOpenOrders() // [Order]
const orders = await qt.getOrders() // [Order]
const orders = await qt.getClosedOrders() // [Order]
const order = await qt.getOrder(orderId) // => Order
await qt.createOrder(newOrder)
await qt.updateOrder(orderId, newOrder)
await qt.removeOrder(orderId)
await qt.testOrder(order)

const symbol = await qt.getSymbol('MSFT') // => Symbol
const symbols = await qt.getSymbols(['MSFT', 'AAPL', 'BMO']) // => [Symbol]
const symbols = qt.searchSymbols('MS') // => [Symbol]

// Example fetching TSLA options
const tsla = await qt.getSymbol('tsla')
const chain = await tsla.getOptionChain()
const jan25 = chain['2025-01-17']
const jan25600 = jan25['600']
const quote = await jan25600.getQuote()

// ... connect to Questrade first!

// Websocket port changes by API and by symbol. So you have to get the port every time you need different data stream
var getWebSocketURL = function (symbolId, cb) {
  var webSocketURL;
  request({
    method: 'GET',
    url: qt.apiUrl + '/markets/quotes?ids=' + symbolId + '&stream=true&mode=WebSocket',
    auth: {
      bearer: qt.accessToken
    }
  }, function (err, http, body) {
    if (err) {
      cb(err, null);
    } else {
      response = JSON.parse(body);
      webSocketURL = qt.api_server.slice(0, -1) + ':' + response.streamPort + '/' + qt.apiVersion + '/markets/quotes?ids=' + symbolId + 'stream=true&mode=WebSocket';
      cb(null, webSocketURL);
    }
  });
}
getWebSocketURL('9291,8049', function (err, webSocketURL) { // BMO.TO & AAPL
  console.log(webSocketURL);
  const WebSocket = require('ws');
  const ws = new WebSocket(webSocketURL);

  ws.on('open', function open() {
    ws.send(qt.accessToken);
  });

  ws.on('message', function incoming(data) {
    console.log(data);
    // Do what you want with the data
  });

  // CLOSING WebSocket Connections otherwise will remain open
  process.on('exit', function () {
    if (ws) {
      console.log('CLOSE WebSocket');
      ws.close();
    }
  });

  //catches ctrl+c event
  process.on('SIGINT', function () {
    if (ws) {
      console.log('CLOSE WebSocket SIGINT');
      ws.close();
    }
  });

  //catches uncaught exceptions
  process.on('uncaughtException', function () {
    if (ws) {
      console.log('CLOSE WebSocket');
      ws.close();
    }
  });
});