import { TelegramClient } from 'sibdigital-messaging-apis/packages/messaging-api-telegram';

import Bot, { OnRequest } from '../bot/Bot';
import SessionStore from '../session/SessionStore';

import TelegramConnector, {
  TelegramConnectorOptions,
} from './TelegramConnector';
import TelegramContext from './TelegramContext';
import TelegramEvent from './TelegramEvent';
import { PollingOptions, TelegramRequestBody } from './TelegramTypes';

export default class TelegramBot extends Bot<
  TelegramRequestBody,
  TelegramClient,
  TelegramEvent,
  TelegramContext
> {
  _offset: number | null;

  _shouldGetUpdates: boolean;

  constructor({
    sessionStore,
    sync,
    onRequest,
    ...connectorOptions
  }: TelegramConnectorOptions & {
    sessionStore?: SessionStore;
    sync?: boolean;
    onRequest?: OnRequest;
  }) {
    const connector = new TelegramConnector(connectorOptions);
    super({ connector, sessionStore, sync, onRequest });

    this._offset = null;
    this._shouldGetUpdates = false;
  }

  get offset(): number | null {
    return this._offset;
  }

  async createLongPollingRuntime(options: PollingOptions = {}): Promise<void> {
    this._shouldGetUpdates = true;
    this._offset = options.offset || null;

    const handler = this.createRequestHandler();

    /* eslint-disable no-await-in-loop */
    while (this._shouldGetUpdates) {
      try {
        const params = this._offset
          ? {
              ...options,
              offset: this._offset,
            }
          : options;

        const updates = await (
          this.connector as TelegramConnector
        ).client.getUpdates(params);

        if (updates.length > 0) {
          for (let i = 0; i < updates.length; i++) {
            await handler(updates[i]);
          }

          const highestUpdateId = Math.max(
            ...updates.map((update) => update.updateId)
          );

          this._offset = highestUpdateId + 1;
        }
      } catch (err) {
        console.error(err);
      }
    }
    /* eslint-enable no-await-in-loop */
  }

  stop(): void {
    this._shouldGetUpdates = false;
  }
}