import { When, Then } from '@cucumber/cucumber';
import { expect } from 'chai';

When('I subscribe to channel {string}', async function(channel) {
  // remember the channel we subscribed to
  this.channel = channel;
  let pubnub = this.getPubnub({
      publishKey: this.keyset.publishKey,
      subscribeKey: this.keyset.subscribeKey
    });

  let connectedResponse = new Promise<void>(((resolveConnected) => {
    this.subscribeResponse = new Promise<void>(((resolveSubscribe) => {
      pubnub.addListener({
        status: function(statusEvent) { console.log('status', statusEvent.category)
          // Once the SDK fires this event
          if (statusEvent.category === "PNConnectedCategory") {
            resolveConnected();
          }
        },
        message: (m) => {
          // remember the message received to compare and then resolve the promise
          this.message = m.message;
          resolveSubscribe();
        }
      });
    }));
  })); 

  pubnub.subscribe({ channels: [ this.channel ] });

  // return the promise so the next cucumber step waits for the sdk to return connected status
  return connectedResponse;
});

When('I publish the message {string} to channel {string}', async function(message, channel) {
  // ensure the channel we subscribed to is the same we publish to
  expect(channel).to.equal(this.channel);

  // returning the promise so the next cucumber step will wait for the publish to complete
  return this.getPubnub().publish({
    message: message,
    channel: channel
  });
});

Then('I receive the message in my subscribe response', async function() {
  // wait for the message to be received by the subscription and then
  // check the expected message matches the message received
  await this.subscribeResponse;
  expect('hello').to.equal(this.message);

  // allow the subscribe loop to continue and then clean up
  return this.delayCleanup();
});