nestjs-telegraf/lib/explorers/telegraf-scene.explorer.ts

import { Inject, Injectable, OnModuleInit } from '@nestjs/common';
import { DiscoveryService } from '@nestjs/core';
import { MetadataScanner } from '@nestjs/core/metadata-scanner';
import { InstanceWrapper } from '@nestjs/core/injector/instance-wrapper';
import { BaseScene as Scene, Stage, Telegraf } from 'telegraf';
import { TelegrafMetadataAccessor } from '../telegraf.metadata-accessor';

@Injectable()
export class TelegrafSceneExplorer implements OnModuleInit {
  private readonly stage = new Stage([]);

  constructor(
    @Inject(Telegraf)
    private readonly telegraf: Telegraf<never>,
    private readonly discoveryService: DiscoveryService,
    private readonly metadataAccessor: TelegrafMetadataAccessor,
    private readonly metadataScanner: MetadataScanner,
  ) {
    this.telegraf.use(this.stage.middleware());
  }

  onModuleInit(): void {
    this.explore();
  }

  private explore(): void {
    const sceneClasses = this.filterSceneClasses();

    sceneClasses.forEach((wrapper) => {
      const { instance } = wrapper;

      const sceneId = this.metadataAccessor.getSceneMetadata(
        instance.constructor,
      );
      const scene = new Scene(sceneId);
      this.stage.register(scene);

      const prototype = Object.getPrototypeOf(instance);
      this.metadataScanner.scanFromPrototype(
        instance,
        prototype,
        (methodKey: string) =>
          this.registerIfListener(scene, instance, methodKey),
      );
    });
  }

  private filterSceneClasses(): InstanceWrapper[] {
    return this.discoveryService
      .getProviders()
      .filter((wrapper) => wrapper.instance)
      .filter((wrapper) =>
        this.metadataAccessor.isScene(wrapper.instance.constructor),
      );
  }

  private registerIfListener(
    scene: Scene<never>,
    instance: Record<string, Function>,
    methodKey: string,
  ): void {
    const methodRef = instance[methodKey];
    const middlewareFn = methodRef.bind(instance);

    const listenerMetadata = this.metadataAccessor.getListenerMetadata(
      methodRef,
    );
    if (!listenerMetadata) return;

    const { method, args } = listenerMetadata;
    (scene[method] as any)(...args, middlewareFn);
  }
}