import {
  Injectable,
  Inject,
  OnApplicationBootstrap,
  Logger,
  OnApplicationShutdown,
} from '@nestjs/common';
import { Telegraf } from 'telegraf';
import { Context, TelegrafModuleOptions } from './interfaces';
import { TELEGRAF_MODULE_OPTIONS } from './telegraf.constants';

@Injectable()
export class TelegrafProvider<C extends Context = Context>
  extends Telegraf<C>
  implements OnApplicationBootstrap, OnApplicationShutdown {
  private logger = new Logger('Telegraf');
  private readonly launchOptions;

  constructor(@Inject(TELEGRAF_MODULE_OPTIONS) options: TelegrafModuleOptions) {
    super(options.token, options.options);
    this.launchOptions = options.launchOptions;
  }

  async onApplicationBootstrap(): Promise<void> {
    this.catch(async (err, ctx) => {
      this.logger.error(
        `Encountered an error for ${ctx.updateType} update type`,
        err as string,
      );
    });

    await this.launch(this.launchOptions);
  }

  async onApplicationShutdown(): Promise<void> {
    await this.stop();
  }
}