feat(): add middleware support & refactor

2025-09-30 04:18:51 +03:00 · 2020-12-30 01:49:09 +03:00
parent 306f239603
commit 3e585859d0
35 changed files with 105 additions and 81 deletions
--- a/lib/explorers/telegraf-scene.explorer.ts
+++ b/lib/explorers/telegraf-scene.explorer.ts
@@ -0,0 +1,73 @@
+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,
+    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,
+    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);
+  }
+}
--- a/lib/explorers/telegraf-update.explorer.ts
+++ b/lib/explorers/telegraf-update.explorer.ts
@@ -0,0 +1,63 @@
+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 { Telegraf } from 'telegraf';
+import { TelegrafMetadataAccessor } from '../telegraf.metadata-accessor';
+
+@Injectable()
+export class TelegrafUpdateExplorer implements OnModuleInit {
+  constructor(
+    @Inject(Telegraf)
+    private readonly telegraf: Telegraf,
+    private readonly discoveryService: DiscoveryService,
+    private readonly metadataAccessor: TelegrafMetadataAccessor,
+    private readonly metadataScanner: MetadataScanner,
+  ) {}
+
+  onModuleInit(): void {
+    this.explore();
+  }
+
+  private explore(): void {
+    const updateClasses = this.filterUpdateClasses();
+
+    updateClasses.forEach((wrapper) => {
+      const { instance } = wrapper;
+
+      const prototype = Object.getPrototypeOf(instance);
+      this.metadataScanner.scanFromPrototype(
+        instance,
+        prototype,
+        (methodKey: string) => this.registerIfListener(instance, methodKey),
+      );
+    });
+  }
+
+  private filterUpdateClasses(): InstanceWrapper[] {
+    return this.discoveryService
+      .getProviders()
+      .filter((wrapper) => wrapper.instance)
+      .filter((wrapper) =>
+        this.metadataAccessor.isUpdate(wrapper.instance.constructor),
+      );
+  }
+
+  private registerIfListener(
+    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;
+    // NOTE: Use "any" to disable "Expected at least 1 arguments, but got 1 or more." error.
+    // Use telegraf instance for non-scene listeners
+    (this.telegraf[method] as any)(...args, middlewareFn);
+  }
+}