Merge pull request #875 from atassis/feature/all-bots-decorator

feat: created all bots decorator

lib/decorators/core/index.ts

@@ -2,3 +2,4 @@ export * from './update.decorator';
 export * from './scene.decorator';
 export * from './wizard.decorator';
 export * from './inject-bot.decorator';
+export * from './inject-all-bots.decorator';

lib/decorators/core/inject-all-bots.decorator.ts

@@ -0,0 +1,9 @@
+import { Inject } from '@nestjs/common';
+import { Telegraf } from 'telegraf';
+
+import { getAllBotsToken } from '../../utils/get-all-bots-token.util';
+
+export type AllBotsMap = Map<string, Telegraf<any>>;
+
+export const InjectAllBots = (): ParameterDecorator =>
+  Inject(getAllBotsToken());

lib/decorators/scene/scene-enter.decorator.ts

@@ -1,6 +1,5 @@
 import { createListenerDecorator } from '../../utils';
 import { Scenes } from 'telegraf';
 
-export const SceneEnter = createListenerDecorator<Scenes.BaseScene<never>>(
-  'enter',
-);
+export const SceneEnter =
+  createListenerDecorator<Scenes.BaseScene<never>>('enter');

lib/decorators/scene/scene-leave.decorator.ts

@@ -1,6 +1,5 @@
 import { createListenerDecorator } from '../../utils';
 import { Scenes } from 'telegraf';
 
-export const SceneLeave = createListenerDecorator<Scenes.BaseScene<never>>(
-  'leave',
-);
+export const SceneLeave =
+  createListenerDecorator<Scenes.BaseScene<never>>('leave');

lib/execution-context/telegraf-arguments-host.ts

@@ -4,7 +4,8 @@ import { TgArgumentsHost } from './tg-arguments-host.interface';
 
 export class TelegrafArgumentsHost
   extends ExecutionContextHost
-  implements TgArgumentsHost {
+  implements TgArgumentsHost
+{
   static create(context: ArgumentsHost): TelegrafArgumentsHost {
     const type = context.getType();
     const tgContext = new TelegrafArgumentsHost(context.getArgs());

lib/execution-context/telegraf-execution-context.ts

@@ -6,7 +6,8 @@ export type TelegrafContextType = 'telegraf' | ContextType;
 
 export class TelegrafExecutionContext
   extends ExecutionContextHost
-  implements TgArgumentsHost {
+  implements TgArgumentsHost
+{
   static create(context: ExecutionContext): TelegrafExecutionContext {
     const type = context.getType();
     const tgContext = new TelegrafExecutionContext(

lib/interfaces/telegraf-options.interface.ts

@@ -11,7 +11,9 @@ export interface TelegrafModuleOptions {
 }
 
 export interface TelegrafOptionsFactory {
-  createTelegrafOptions(): Promise<TelegrafModuleOptions> | TelegrafModuleOptions;
+  createTelegrafOptions():
+    | Promise<TelegrafModuleOptions>
+    | TelegrafModuleOptions;
 }
 
 export interface TelegrafModuleAsyncOptions

lib/services/base-explorer.service.ts

@@ -10,8 +10,7 @@ export class BaseExplorerService {
     if (!include || isEmpty(include)) {
       return [...modulesContainer.values()];
     }
-    const whitelisted = this.includeWhitelisted(modulesContainer, include);
-    return whitelisted;
+    return this.includeWhitelisted(modulesContainer, include);
   }
 
   includeWhitelisted(

lib/services/listeners-explorer.service.ts

@@ -162,7 +162,7 @@ export class ListenersExplorerService
         {},
       );
 
-    const steps = Object.values(group).map((stepsMetadata) => {
+    wizard.steps = Object.values(group).map((stepsMetadata) => {
       const composer = new Composer();
       stepsMetadata.forEach((stepMethod) => {
         this.registerIfListener(
@@ -175,8 +175,6 @@ export class ListenersExplorerService
       });
       return composer.middleware();
     });
-
-    wizard.steps = steps;
   }
 
   private registerIfListener(
@@ -187,7 +185,8 @@ export class ListenersExplorerService
     defaultMetadata?: ListenerMetadata[],
   ): void {
     const methodRef = prototype[methodName];
-    const metadata = this.metadataAccessor.getListenerMetadata(methodRef) || defaultMetadata;
+    const metadata =
+      this.metadataAccessor.getListenerMetadata(methodRef) || defaultMetadata;
     if (!metadata || metadata.length < 1) {
       return undefined;
     }
@@ -222,7 +221,7 @@ export class ListenersExplorerService
     methodName: string,
   ) {
     const paramsFactory = this.telegrafParamsFactory;
-    const resolverCallback = this.externalContextCreator.create<
+    return this.externalContextCreator.create<
       Record<number, ParamMetadata>,
       TelegrafContextType
     >(
@@ -236,6 +235,5 @@ export class ListenersExplorerService
       undefined,
       'telegraf',
     );
-    return resolverCallback;
   }
 }

lib/telegraf-all-bots.provider.ts

@@ -0,0 +1,10 @@
+import { Provider } from '@nestjs/common';
+import { Telegraf } from 'telegraf';
+import { TELEGRAF_ALL_BOTS } from './telegraf.constants';
+
+export const allBotsMap = new Map<string, Telegraf<any>>();
+
+export const telegrafAllBotsProvider: Provider = {
+  provide: TELEGRAF_ALL_BOTS,
+  useValue: allBotsMap,
+};

lib/telegraf-core.module.ts

@@ -19,6 +19,10 @@ import {
 } from './telegraf.constants';
 import { ListenersExplorerService, MetadataAccessorService } from './services';
 import { telegrafStageProvider } from './stage.provider';
+import {
+  allBotsMap,
+  telegrafAllBotsProvider,
+} from './telegraf-all-bots.provider';
 import { createBotFactory, getBotToken } from './utils';
 
 @Global()
@@ -43,7 +47,11 @@ export class TelegrafCoreModule implements OnApplicationShutdown {
 
     const telegrafBotProvider: Provider = {
       provide: telegrafBotName,
-      useFactory: async () => await createBotFactory(options),
+      useFactory: async () => {
+        const bot = await createBotFactory(options);
+        allBotsMap.set(telegrafBotName, bot);
+        return bot;
+      },
     };
 
     return {
@@ -56,11 +64,13 @@ export class TelegrafCoreModule implements OnApplicationShutdown {
         telegrafStageProvider,
         telegrafBotNameProvider,
         telegrafBotProvider,
+        telegrafAllBotsProvider,
       ],
       exports: [
         telegrafStageProvider,
         telegrafBotNameProvider,
         telegrafBotProvider,
+        telegrafAllBotsProvider,
       ],
     };
   }
@@ -77,8 +87,11 @@ export class TelegrafCoreModule implements OnApplicationShutdown {
 
     const telegrafBotProvider: Provider = {
       provide: telegrafBotName,
-      useFactory: async (options: TelegrafModuleOptions) =>
-        await createBotFactory(options),
+      useFactory: async (options: TelegrafModuleOptions) => {
+        const bot = await createBotFactory(options);
+        allBotsMap.set(telegrafBotName, bot);
+        return bot;
+      },
       inject: [TELEGRAF_MODULE_OPTIONS],
     };
 
@@ -91,11 +104,13 @@ export class TelegrafCoreModule implements OnApplicationShutdown {
         telegrafStageProvider,
         telegrafBotNameProvider,
         telegrafBotProvider,
+        telegrafAllBotsProvider,
       ],
       exports: [
         telegrafStageProvider,
         telegrafBotNameProvider,
         telegrafBotProvider,
+        telegrafAllBotsProvider,
       ],
     };
   }

lib/telegraf.constants.ts

@@ -12,3 +12,5 @@ export const WIZARD_STEP_METADATA = 'WIZARD_STEP_METADATA';
 export const PARAM_ARGS_METADATA = ROUTE_ARGS_METADATA;
 
 export const TELEGRAF_STAGE = 'TelegrafStage';
+
+export const TELEGRAF_ALL_BOTS = 'TELEGRAF_ALL_BOTS';

lib/types/index.ts

@@ -1,4 +1,4 @@
-import { Composer, Middleware, Telegraf } from 'telegraf';
+import { Composer, Middleware } from 'telegraf';
 
 export type Filter<T extends any[], F> = T extends []
   ? []
@@ -14,5 +14,5 @@ export type OnlyFunctionPropertyNames<T> = {
 
 export type ComposerMethodArgs<
   T extends Composer<never>,
-  U extends OnlyFunctionPropertyNames<T> = OnlyFunctionPropertyNames<T>
+  U extends OnlyFunctionPropertyNames<T> = OnlyFunctionPropertyNames<T>,
 > = Filter<Parameters<T[U]>, Middleware<never>>;

lib/utils/create-listener-decorator.util.ts

@@ -19,8 +19,9 @@ export function createListenerDecorator<
           args,
         } as ListenerMetadata,
       ];
-          
-      const previousValue = Reflect.getMetadata(LISTENERS_METADATA, descriptor.value) || [];
+
+      const previousValue =
+        Reflect.getMetadata(LISTENERS_METADATA, descriptor.value) || [];
       const value = [...previousValue, ...metadata];
       Reflect.defineMetadata(LISTENERS_METADATA, value, descriptor.value);
       return descriptor;

lib/utils/get-all-bots-token.util.ts

@@ -0,0 +1,3 @@
+import { TELEGRAF_ALL_BOTS } from '../telegraf.constants';
+
+export const getAllBotsToken = (): string => TELEGRAF_ALL_BOTS;

lib/utils/param-decorator.util.ts

@@ -5,8 +5,10 @@ import { PARAM_ARGS_METADATA } from '../telegraf.constants';
 
 export type ParamData = object | string | number;
 
-export const createTelegrafParamDecorator = (paramtype: TelegrafParamtype) => {
-  return (data?: ParamData): ParameterDecorator => (target, key, index) => {
+export const createTelegrafParamDecorator =
+  (paramtype: TelegrafParamtype) =>
+  (data?: ParamData): ParameterDecorator =>
+  (target, key, index) => {
     const args =
       Reflect.getMetadata(PARAM_ARGS_METADATA, target.constructor, key) || {};
     Reflect.defineMetadata(
@@ -16,16 +18,16 @@ export const createTelegrafParamDecorator = (paramtype: TelegrafParamtype) => {
       key,
     );
   };
-};
 
-export const createTelegrafPipesParamDecorator = (
-  paramtype: TelegrafParamtype,
-) => (
-  data?: any,
-  ...pipes: (Type<PipeTransform> | PipeTransform)[]
-): ParameterDecorator => (target, key, index) => {
-  addPipesMetadata(paramtype, data, pipes, target, key, index);
-};
+export const createTelegrafPipesParamDecorator =
+  (paramtype: TelegrafParamtype) =>
+  (
+    data?: any,
+    ...pipes: (Type<PipeTransform> | PipeTransform)[]
+  ): ParameterDecorator =>
+  (target, key, index) => {
+    addPipesMetadata(paramtype, data, pipes, target, key, index);
+  };
 
 export const addPipesMetadata = (
   paramtype: TelegrafParamtype,

package-lock.json

@@ -5062,8 +5062,7 @@
       "version": "5.3.2",
       "resolved": "https://registry.npmjs.org/acorn-jsx/-/acorn-jsx-5.3.2.tgz",
       "integrity": "sha512-rq9s+JNhf0IChjtDXxllJ7g41oZk5SlXtp0LHwyA5cejwn7vKmKp4pPri6YEePv2PU65sAsegbXtIinmDFDXgQ==",
-      "dev": true,
-      "requires": {}
+      "dev": true
     },
     "acorn-walk": {
       "version": "8.2.0",
@@ -5620,8 +5619,7 @@
       "version": "8.5.0",
       "resolved": "https://registry.npmjs.org/eslint-config-prettier/-/eslint-config-prettier-8.5.0.tgz",
       "integrity": "sha512-obmWKLUNCnhtQRKc+tmnYuQl0pFU1ibYJQ5BGhTVB08bHe9wC8qUeG7c08dj9XX+AuPj1YSGSQIHl1pnDHZR0Q==",
-      "dev": true,
-      "requires": {}
+      "dev": true
     },
     "eslint-plugin-prettier": {
       "version": "4.2.1",