diff --git a/.npmignore b/.npmignore index 267876c..130fdb1 100644 --- a/.npmignore +++ b/.npmignore @@ -1,5 +1,6 @@ # source lib +index.ts package-lock.json tsconfig.json .prettierrc diff --git a/lib/decorators/core/inject-bot.decorator.ts b/lib/decorators/core/inject-bot.decorator.ts index d010fbe..73299e8 100644 --- a/lib/decorators/core/inject-bot.decorator.ts +++ b/lib/decorators/core/inject-bot.decorator.ts @@ -1,4 +1,5 @@ import { Inject } from '@nestjs/common'; -import { Telegraf } from 'telegraf'; +import { getBotToken } from '../../utils'; -export const InjectBot = (): ParameterDecorator => Inject(Telegraf); +export const InjectBot = (name?: string): ParameterDecorator => + Inject(getBotToken(name)); diff --git a/lib/explorers/telegraf-scene.explorer.ts b/lib/explorers/telegraf-scene.explorer.ts deleted file mode 100644 index 2ab2c49..0000000 --- a/lib/explorers/telegraf-scene.explorer.ts +++ /dev/null @@ -1,73 +0,0 @@ -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, - 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); - } -} diff --git a/lib/explorers/telegraf-update.explorer.ts b/lib/explorers/telegraf-update.explorer.ts deleted file mode 100644 index a848806..0000000 --- a/lib/explorers/telegraf-update.explorer.ts +++ /dev/null @@ -1,63 +0,0 @@ -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, - 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); - } -} diff --git a/lib/helpers/create-scene-listener-decorator.helper.ts b/lib/helpers/create-scene-listener-decorator.helper.ts index c997ffc..b089ebf 100644 --- a/lib/helpers/create-scene-listener-decorator.helper.ts +++ b/lib/helpers/create-scene-listener-decorator.helper.ts @@ -1,6 +1,6 @@ import { SetMetadata } from '@nestjs/common'; import { BaseScene as Scene } from 'telegraf'; -import { ComposerMethodArgs, SceneMethods } from '../telegraf.types'; +import { ComposerMethodArgs, SceneMethods } from '../types'; import { UPDATE_LISTENER_METADATA } from '../telegraf.constants'; import { ListenerMetadata } from '../interfaces'; diff --git a/lib/helpers/create-update-listener-decorator.helper.ts b/lib/helpers/create-update-listener-decorator.helper.ts index a27f85b..28343e2 100644 --- a/lib/helpers/create-update-listener-decorator.helper.ts +++ b/lib/helpers/create-update-listener-decorator.helper.ts @@ -1,6 +1,6 @@ import { SetMetadata } from '@nestjs/common'; import { Composer } from 'telegraf'; -import { ComposerMethodArgs, UpdateMethods } from '../telegraf.types'; +import { ComposerMethodArgs, UpdateMethods } from '../types'; import { UPDATE_LISTENER_METADATA } from '../telegraf.constants'; import { ListenerMetadata } from '../interfaces'; diff --git a/lib/index.ts b/lib/index.ts index da17d36..ace41d2 100644 --- a/lib/index.ts +++ b/lib/index.ts @@ -1,5 +1,15 @@ +export * as Composer from 'telegraf/composer'; +export * as Markup from 'telegraf/markup'; +export * as BaseScene from 'telegraf/scenes/base'; +export * as session from 'telegraf/session'; +export * as Stage from 'telegraf/stage'; +export * as WizardScene from 'telegraf/scenes/wizard'; +export * as Extra from 'telegraf/extra'; + export * from './decorators'; export * from './interfaces'; export * from './helpers'; +export * from './utils'; export * from './telegraf.module'; -export * from './telegraf.types'; +export * from './types'; +export { Telegraf } from 'telegraf'; diff --git a/lib/interfaces/index.ts b/lib/interfaces/index.ts index ef12fd5..6be0df9 100644 --- a/lib/interfaces/index.ts +++ b/lib/interfaces/index.ts @@ -1,2 +1,3 @@ export * from './telegraf-options.interface'; export * from './listener-metadata.interface'; +export * from './update-metadata.interface'; diff --git a/lib/interfaces/telegraf-options.interface.ts b/lib/interfaces/telegraf-options.interface.ts index 09ffb2c..cfdb594 100644 --- a/lib/interfaces/telegraf-options.interface.ts +++ b/lib/interfaces/telegraf-options.interface.ts @@ -1,9 +1,9 @@ import { ModuleMetadata, Type } from '@nestjs/common/interfaces'; import { Middleware, Context } from 'telegraf'; import { + TelegrafOptions, LaunchPollingOptions, LaunchWebhookOptions, - TelegrafOptions, } from 'telegraf/typings/telegraf'; export interface TelegrafModuleOptions { @@ -13,7 +13,9 @@ export interface TelegrafModuleOptions { polling?: LaunchPollingOptions; webhook?: LaunchWebhookOptions; }; - middlewares?: Middleware[]; + botName?: string; + include?: Function[]; + middlewares?: ReadonlyArray>; } export interface TelegrafOptionsFactory { @@ -22,6 +24,7 @@ export interface TelegrafOptionsFactory { export interface TelegrafModuleAsyncOptions extends Pick { + botName?: string; useExisting?: Type; useClass?: Type; useFactory?: ( diff --git a/lib/interfaces/update-metadata.interface.ts b/lib/interfaces/update-metadata.interface.ts new file mode 100644 index 0000000..4c2ba0e --- /dev/null +++ b/lib/interfaces/update-metadata.interface.ts @@ -0,0 +1,6 @@ +export interface UpdateMetadata { + name: string; + type: string; + methodName: string; + callback?: Function | Record; +} diff --git a/lib/services/base-explorer.service.ts b/lib/services/base-explorer.service.ts new file mode 100644 index 0000000..eff7c82 --- /dev/null +++ b/lib/services/base-explorer.service.ts @@ -0,0 +1,59 @@ +import { InstanceWrapper } from '@nestjs/core/injector/instance-wrapper'; +import { Module } from '@nestjs/core/injector/module'; +import { flattenDeep, groupBy, identity, isEmpty, mapValues } from 'lodash'; +import { UpdateMetadata } from '../interfaces'; + +export class BaseExplorerService { + getModules( + modulesContainer: Map, + include: Function[], + ): Module[] { + if (!include || isEmpty(include)) { + return [...modulesContainer.values()]; + } + const whitelisted = this.includeWhitelisted(modulesContainer, include); + return whitelisted; + } + + includeWhitelisted( + modulesContainer: Map, + include: Function[], + ): Module[] { + const modules = [...modulesContainer.values()]; + return modules.filter(({ metatype }) => + include.some((item) => item === metatype), + ); + } + + flatMap( + modules: Module[], + callback: (instance: InstanceWrapper, moduleRef: Module) => T | T[], + ): T[] { + const invokeMap = () => { + return modules.map((moduleRef) => { + const providers = [...moduleRef.providers.values()]; + return providers.map((wrapper) => callback(wrapper, moduleRef)); + }); + }; + return flattenDeep(invokeMap()).filter(identity); + } + + groupMetadata(resolvers: UpdateMetadata[]) { + const groupByType = groupBy( + resolvers, + (metadata: UpdateMetadata) => metadata.type, + ); + const groupedMetadata = mapValues( + groupByType, + (resolversArr: UpdateMetadata[]) => + resolversArr.reduce( + (prev, curr) => ({ + ...prev, + [curr.name]: curr.callback, + }), + {}, + ), + ); + return groupedMetadata; + } +} diff --git a/lib/services/index.ts b/lib/services/index.ts new file mode 100644 index 0000000..9859475 --- /dev/null +++ b/lib/services/index.ts @@ -0,0 +1,4 @@ +export * from './updates-explorer.service'; +export * from './metadata-accessor.service'; +export * from './scenes-explorer.service'; +export * from './updates-explorer.service'; diff --git a/lib/telegraf.metadata-accessor.ts b/lib/services/metadata-accessor.service.ts similarity index 65% rename from lib/telegraf.metadata-accessor.ts rename to lib/services/metadata-accessor.service.ts index 5c4394d..3e7e044 100644 --- a/lib/telegraf.metadata-accessor.ts +++ b/lib/services/metadata-accessor.service.ts @@ -4,18 +4,31 @@ import { SCENE_METADATA, UPDATE_LISTENER_METADATA, UPDATE_METADATA, -} from './telegraf.constants'; -import { ListenerMetadata } from './interfaces'; +} from '../telegraf.constants'; +import { ListenerMetadata } from '../interfaces'; @Injectable() -export class TelegrafMetadataAccessor { +export class MetadataAccessorService { constructor(private readonly reflector: Reflector) {} isUpdate(target: Function): boolean { + if (!target) { + return false; + } return !!this.reflector.get(UPDATE_METADATA, target); } + isUpdateListener(target: Function) { + if (!target) { + return false; + } + return !!this.reflector.get(UPDATE_LISTENER_METADATA, target); + } + isScene(target: Function): boolean { + if (!target) { + return false; + } return !!this.reflector.get(SCENE_METADATA, target); } diff --git a/lib/services/scenes-explorer.service.ts b/lib/services/scenes-explorer.service.ts new file mode 100644 index 0000000..be3d47d --- /dev/null +++ b/lib/services/scenes-explorer.service.ts @@ -0,0 +1,98 @@ +import { Inject, Injectable, OnModuleInit } from '@nestjs/common'; +import { DiscoveryService, ModuleRef, ModulesContainer } 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 { MetadataAccessorService } from './metadata-accessor.service'; +import { BaseExplorerService } from './base-explorer.service'; +import { + TELEGRAF_BOT_NAME, + TELEGRAF_MODULE_OPTIONS, +} from '../telegraf.constants'; +import { TelegrafModuleOptions } from '../interfaces'; +import { Module } from '@nestjs/core/injector/module'; + +@Injectable() +export class ScenesExplorerService + extends BaseExplorerService + implements OnModuleInit { + private readonly stage = new Stage([]); + + constructor( + @Inject(TELEGRAF_BOT_NAME) + private readonly botName: string, + @Inject(TELEGRAF_MODULE_OPTIONS) + private readonly telegrafModuleOptions: TelegrafModuleOptions, + private readonly moduleRef: ModuleRef, + private readonly discoveryService: DiscoveryService, + private readonly metadataAccessor: MetadataAccessorService, + private readonly metadataScanner: MetadataScanner, + private readonly modulesContainer: ModulesContainer, + ) { + super(); + } + + private bot: Telegraf; + + onModuleInit(): void { + this.bot = this.moduleRef.get>(this.botName, { + strict: false, + }); + this.bot.use(this.stage.middleware()); + this.explore(); + } + + private explore(): void { + const modules = this.getModules( + this.modulesContainer, + this.telegrafModuleOptions.include || [], + ); + const scenes = this.flatMap(modules, (instance, moduleRef) => + this.applyScenes(instance, moduleRef), + ); + } + + private applyScenes(wrapper: InstanceWrapper, moduleRef: Module) { + const { instance } = wrapper; + if (!instance) { + return undefined; + } + const prototype = Object.getPrototypeOf(instance); + + const providers: InstanceWrapper[] = this.discoveryService.getProviders(); + const sceneProviders: InstanceWrapper[] = providers.filter( + (wrapper: InstanceWrapper) => + this.metadataAccessor.isScene(wrapper.metatype), + ); + + sceneProviders.forEach((wrapper) => { + const { instance } = wrapper; + if (!instance) { + return undefined; + } + + const sceneId = this.metadataAccessor.getSceneMetadata( + instance.constructor, + ); + + const scene = new Scene(sceneId); + this.stage.register(scene); + + this.metadataScanner.scanFromPrototype( + instance, + prototype, + (methodKey: string) => { + const methodRef = instance[methodKey]; + if (this.metadataAccessor.isUpdateListener(methodRef)) { + const metadata = this.metadataAccessor.getListenerMetadata( + methodRef, + ); + const middlewareFn = methodRef.bind(instance); + const { method, args } = metadata; + (scene[method] as any)(...args, middlewareFn); + } + }, + ); + }); + } +} diff --git a/lib/services/updates-explorer.service.ts b/lib/services/updates-explorer.service.ts new file mode 100644 index 0000000..09fa250 --- /dev/null +++ b/lib/services/updates-explorer.service.ts @@ -0,0 +1,83 @@ +import { Inject, Injectable, OnModuleInit } from '@nestjs/common'; +import { DiscoveryService, ModuleRef, ModulesContainer } from '@nestjs/core'; +import { InstanceWrapper } from '@nestjs/core/injector/instance-wrapper'; +import { MetadataScanner } from '@nestjs/core/metadata-scanner'; +import { MetadataAccessorService } from './metadata-accessor.service'; +import { + TELEGRAF_BOT_NAME, + TELEGRAF_MODULE_OPTIONS, +} from '../telegraf.constants'; +import { Telegraf } from 'telegraf'; +import { TelegrafModuleOptions } from '../interfaces'; +import { BaseExplorerService } from './base-explorer.service'; +import { Module } from '@nestjs/core/injector/module'; + +@Injectable() +export class UpdatesExplorerService + extends BaseExplorerService + implements OnModuleInit { + constructor( + @Inject(TELEGRAF_BOT_NAME) + private readonly botName: string, + @Inject(TELEGRAF_MODULE_OPTIONS) + private readonly telegrafModuleOptions: TelegrafModuleOptions, + private readonly moduleRef: ModuleRef, + private readonly discoveryService: DiscoveryService, + private readonly metadataAccessor: MetadataAccessorService, + private readonly metadataScanner: MetadataScanner, + private readonly modulesContainer: ModulesContainer, + ) { + super(); + } + + private bot: Telegraf; + + onModuleInit(): void { + this.bot = this.moduleRef.get>(this.botName, { + strict: false, + }); + this.explore(); + } + + explore() { + const modules = this.getModules( + this.modulesContainer, + this.telegrafModuleOptions.include || [], + ); + const updates = this.flatMap(modules, (instance, moduleRef) => + this.applyUpdates(instance, moduleRef), + ); + } + + private applyUpdates(wrapper: InstanceWrapper, moduleRef: Module) { + const { instance } = wrapper; + if (!instance) { + return undefined; + } + const prototype = Object.getPrototypeOf(instance); + + const providers: InstanceWrapper[] = this.discoveryService.getProviders(); + const updateProviders: InstanceWrapper[] = providers.filter( + (wrapper: InstanceWrapper) => + this.metadataAccessor.isUpdate(wrapper.metatype), + ); + + updateProviders.forEach((wrapper: InstanceWrapper) => { + const { instance } = wrapper; + if (!instance) { + return undefined; + } + this.metadataScanner.scanFromPrototype(instance, prototype, (name) => { + const methodRef = instance[name]; + if (this.metadataAccessor.isUpdateListener(methodRef)) { + const metadata = this.metadataAccessor.getListenerMetadata(methodRef); + const middlewareFn = methodRef.bind(instance); + const { method, args } = metadata; + // NOTE: Use "any" to disable "Expected at least 1 arguments, but got 1 or more." error. + // Use telegraf instance for non-scene listeners + (this.bot[method] as any)(...args, middlewareFn); + } + }); + }); + } +} diff --git a/lib/telegraf-core.module.ts b/lib/telegraf-core.module.ts new file mode 100644 index 0000000..f33d5cc --- /dev/null +++ b/lib/telegraf-core.module.ts @@ -0,0 +1,166 @@ +import { DiscoveryModule, ModuleRef } from '@nestjs/core'; +import { + Module, + DynamicModule, + Provider, + Type, + Global, + Inject, + OnApplicationShutdown, + Logger, +} from '@nestjs/common'; +import { + TelegrafModuleOptions, + TelegrafModuleAsyncOptions, + TelegrafOptionsFactory, +} from './interfaces'; +import { + TELEGRAF_BOT_NAME, + TELEGRAF_MODULE_OPTIONS, +} from './telegraf.constants'; +import { + MetadataAccessorService, + ScenesExplorerService, + UpdatesExplorerService, +} from './services'; +import { getBotToken } from './utils'; +import { Telegraf } from 'telegraf'; +import { defer } from 'rxjs'; + +@Global() +@Module({ + imports: [DiscoveryModule], + providers: [ + UpdatesExplorerService, + ScenesExplorerService, + MetadataAccessorService, + ], +}) +export class TelegrafCoreModule implements OnApplicationShutdown { + private static logger = new Logger(TelegrafCoreModule.name); + + constructor( + @Inject(TELEGRAF_BOT_NAME) private readonly botName: string, + private readonly moduleRef: ModuleRef, + ) {} + + public static forRoot(options: TelegrafModuleOptions): DynamicModule { + const telegrafBotName = getBotToken(options.botName); + + const telegrafBotProvider = { + provide: telegrafBotName, + useFactory: async (): Promise => + await defer(async () => { + const bot = new Telegraf(options.token); + this.applyBotMiddlewares(bot, options.middlewares); + await bot.launch(options.launchOptions); + return bot; + }).toPromise(), + }; + + return { + module: TelegrafCoreModule, + providers: [ + { + provide: TELEGRAF_MODULE_OPTIONS, + useValue: options, + }, + { + provide: TELEGRAF_BOT_NAME, + useValue: telegrafBotName, + }, + telegrafBotProvider, + ], + exports: [telegrafBotProvider], + }; + } + + public static forRootAsync( + options: TelegrafModuleAsyncOptions, + ): DynamicModule { + const telegrafBotName = getBotToken(options.botName); + + const telegrafBotProvider = { + provide: telegrafBotName, + useFactory: async ( + telegrafModuleOptions: TelegrafModuleOptions, + ): Promise => { + const { botName, ...telegrafOptions } = telegrafModuleOptions; + + return await defer(async () => { + const bot = new Telegraf(telegrafOptions.token); + this.applyBotMiddlewares(bot, telegrafOptions.middlewares); + await bot.launch(telegrafOptions.launchOptions); + return bot; + }).toPromise(); + }, + inject: [TELEGRAF_MODULE_OPTIONS], + }; + + const asyncProviders = this.createAsyncProviders(options); + return { + module: TelegrafCoreModule, + imports: options.imports, + providers: [ + ...asyncProviders, + { + provide: TELEGRAF_BOT_NAME, + useValue: telegrafBotName, + }, + telegrafBotProvider, + ], + exports: [telegrafBotProvider], + }; + } + + private static applyBotMiddlewares(bot, middlewares) { + if (middlewares) { + middlewares.forEach((middleware) => { + bot.use(middleware); + }); + } + } + + private static createAsyncProviders( + options: TelegrafModuleAsyncOptions, + ): Provider[] { + if (options.useExisting || options.useFactory) { + return [this.createAsyncOptionsProvider(options)]; + } + const useClass = options.useClass as Type; + return [ + this.createAsyncOptionsProvider(options), + { + provide: useClass, + useClass, + }, + ]; + } + + private static createAsyncOptionsProvider( + options: TelegrafModuleAsyncOptions, + ): Provider { + if (options.useFactory) { + return { + provide: TELEGRAF_MODULE_OPTIONS, + useFactory: options.useFactory, + inject: options.inject || [], + }; + } + // `as Type` is a workaround for microsoft/TypeScript#31603 + const inject = [ + (options.useClass || options.useExisting) as Type, + ]; + return { + provide: TELEGRAF_MODULE_OPTIONS, + useFactory: async (optionsFactory: TelegrafOptionsFactory) => + await optionsFactory.createTelegrafOptions(), + inject, + }; + } + + async onApplicationShutdown(): Promise { + const bot = this.moduleRef.get(this.botName); + bot && (await bot.stop()); + } +} diff --git a/lib/telegraf.constants.ts b/lib/telegraf.constants.ts index 1790594..3ac37e3 100644 --- a/lib/telegraf.constants.ts +++ b/lib/telegraf.constants.ts @@ -1,4 +1,6 @@ export const TELEGRAF_MODULE_OPTIONS = 'TELEGRAF_MODULE_OPTIONS'; +export const TELEGRAF_BOT_NAME = 'TELEGRAF_BOT_NAME'; +export const DEFAULT_BOT_NAME = 'DEFAULT_BOT_NAME'; export const UPDATE_METADATA = 'UPDATE_METADATA'; export const UPDATE_LISTENER_METADATA = 'UPDATE_LISTENER_METADATA'; diff --git a/lib/telegraf.module.ts b/lib/telegraf.module.ts index 4e9f958..fb3cdeb 100644 --- a/lib/telegraf.module.ts +++ b/lib/telegraf.module.ts @@ -1,106 +1,27 @@ -import { DiscoveryModule, ModuleRef } from '@nestjs/core'; +import { Module, DynamicModule } from '@nestjs/common'; +import { TelegrafCoreModule } from './telegraf-core.module'; import { - DynamicModule, - Inject, - Module, - OnApplicationBootstrap, - OnApplicationShutdown, - Provider, -} from '@nestjs/common'; -import { Telegraf } from 'telegraf'; -import { - TelegrafModuleAsyncOptions, TelegrafModuleOptions, - TelegrafOptionsFactory, + TelegrafModuleAsyncOptions, } from './interfaces'; -import { TELEGRAF_MODULE_OPTIONS } from './telegraf.constants'; -import { TelegrafMetadataAccessor } from './telegraf.metadata-accessor'; -import { TelegrafUpdateExplorer } from './explorers/telegraf-update.explorer'; -import { TelegrafSceneExplorer } from './explorers/telegraf-scene.explorer'; -import { createProviders, TelegrafProvider } from './telegraf.providers'; - -@Module({ - imports: [DiscoveryModule], - providers: [ - TelegrafMetadataAccessor, - TelegrafSceneExplorer, - TelegrafUpdateExplorer, - ], -}) -export class TelegrafModule - implements OnApplicationBootstrap, OnApplicationShutdown { - constructor( - @Inject(TELEGRAF_MODULE_OPTIONS) - private readonly options: TelegrafModuleOptions, - private readonly moduleRef: ModuleRef, - ) {} - - async onApplicationBootstrap(): Promise { - const { launchOptions } = this.options; - const telegraf = this.moduleRef.get(Telegraf); - await telegraf.launch(launchOptions); - } - - async onApplicationShutdown(): Promise { - const telegraf = this.moduleRef.get(Telegraf); - await telegraf.stop(); - } +@Module({}) +export class TelegrafModule { public static forRoot(options: TelegrafModuleOptions): DynamicModule { - const providers = [...createProviders(options), TelegrafProvider]; - return { module: TelegrafModule, - providers, - exports: providers, + imports: [TelegrafCoreModule.forRoot(options)], + exports: [TelegrafCoreModule], }; } public static forRootAsync( options: TelegrafModuleAsyncOptions, ): DynamicModule { - const providers = [...this.createAsyncProviders(options), TelegrafProvider]; - return { module: TelegrafModule, - imports: options.imports || [], - providers, - exports: providers, - }; - } - - private static createAsyncProviders( - options: TelegrafModuleAsyncOptions, - ): Provider[] { - if (options.useExisting || options.useFactory) { - return [this.createAsyncOptionsProvider(options)]; - } - - return [ - this.createAsyncOptionsProvider(options), - { - provide: options.useClass, - useClass: options.useClass, - }, - ]; - } - - private static createAsyncOptionsProvider( - options: TelegrafModuleAsyncOptions, - ): Provider { - if (options.useFactory) { - return { - provide: TELEGRAF_MODULE_OPTIONS, - useFactory: options.useFactory, - inject: options.inject || [], - }; - } - - return { - provide: TELEGRAF_MODULE_OPTIONS, - useFactory: async (optionsFactory: TelegrafOptionsFactory) => - await optionsFactory.createTelegrafOptions(), - inject: [options.useExisting || options.useClass], + imports: [TelegrafCoreModule.forRootAsync(options)], + exports: [TelegrafCoreModule], }; } } diff --git a/lib/telegraf.providers.ts b/lib/telegraf.providers.ts deleted file mode 100644 index b13de54..0000000 --- a/lib/telegraf.providers.ts +++ /dev/null @@ -1,25 +0,0 @@ -import { Provider } from '@nestjs/common'; -import { Telegraf } from 'telegraf'; -import { TELEGRAF_MODULE_OPTIONS } from './telegraf.constants'; -import { TelegrafModuleOptions } from './interfaces'; - -export const TelegrafProvider = { - provide: Telegraf, - useFactory: (options: TelegrafModuleOptions) => { - const telegraf = new Telegraf(options.token, options.options); - if (options.middlewares?.length > 0) { - telegraf.use(...options.middlewares); - } - return telegraf; - }, - inject: [TELEGRAF_MODULE_OPTIONS], -}; - -export function createProviders(options: TelegrafModuleOptions): Provider[] { - return [ - { - provide: TELEGRAF_MODULE_OPTIONS, - useValue: options, - }, - ]; -} diff --git a/lib/telegraf.types.ts b/lib/types/index.ts similarity index 100% rename from lib/telegraf.types.ts rename to lib/types/index.ts diff --git a/lib/utils/get-bot-token.util.ts b/lib/utils/get-bot-token.util.ts new file mode 100644 index 0000000..a8458e8 --- /dev/null +++ b/lib/utils/get-bot-token.util.ts @@ -0,0 +1,7 @@ +import { DEFAULT_BOT_NAME } from '../telegraf.constants'; + +export function getBotToken(name?: string) { + return name && name !== DEFAULT_BOT_NAME + ? `${name}_BOT_NAME` + : DEFAULT_BOT_NAME; +} diff --git a/lib/utils/index.ts b/lib/utils/index.ts new file mode 100644 index 0000000..9dc7d1d --- /dev/null +++ b/lib/utils/index.ts @@ -0,0 +1 @@ +export * from './get-bot-token.util'; diff --git a/package-lock.json b/package-lock.json index 7490ee2..acbfe97 100644 --- a/package-lock.json +++ b/package-lock.json @@ -13,6 +13,7 @@ "devDependencies": { "@nestjs/common": "7.6.5", "@nestjs/core": "7.6.5", + "@types/lodash": "^4.14.167", "@typescript-eslint/eslint-plugin": "4.11.1", "@typescript-eslint/parser": "4.11.1", "eslint": "7.17.0", @@ -20,6 +21,7 @@ "eslint-plugin-import": "2.22.1", "husky": "4.3.6", "lint-staged": "10.5.3", + "lodash": "^4.17.20", "prettier": "2.2.1", "reflect-metadata": "0.1.13", "rxjs": "6.6.3", @@ -292,6 +294,12 @@ "integrity": "sha1-7ihweulOEdK4J7y+UnC86n8+ce4=", "dev": true }, + "node_modules/@types/lodash": { + "version": "4.14.167", + "resolved": "https://registry.npmjs.org/@types/lodash/-/lodash-4.14.167.tgz", + "integrity": "sha512-w7tQPjARrvdeBkX/Rwg95S592JwxqOjmms3zWQ0XZgSyxSLdzWaYH3vErBhdVS/lRBX7F8aBYcYJYTr5TMGOzw==", + "dev": true + }, "node_modules/@types/parse-json": { "version": "4.0.0", "resolved": "https://registry.npmjs.org/@types/parse-json/-/parse-json-4.0.0.tgz", @@ -3880,6 +3888,12 @@ "integrity": "sha1-7ihweulOEdK4J7y+UnC86n8+ce4=", "dev": true }, + "@types/lodash": { + "version": "4.14.167", + "resolved": "https://registry.npmjs.org/@types/lodash/-/lodash-4.14.167.tgz", + "integrity": "sha512-w7tQPjARrvdeBkX/Rwg95S592JwxqOjmms3zWQ0XZgSyxSLdzWaYH3vErBhdVS/lRBX7F8aBYcYJYTr5TMGOzw==", + "dev": true + }, "@types/parse-json": { "version": "4.0.0", "resolved": "https://registry.npmjs.org/@types/parse-json/-/parse-json-4.0.0.tgz", diff --git a/package.json b/package.json index 401fb88..4bb4b75 100644 --- a/package.json +++ b/package.json @@ -37,6 +37,7 @@ "devDependencies": { "@nestjs/common": "7.6.5", "@nestjs/core": "7.6.5", + "@types/lodash": "^4.14.167", "@typescript-eslint/eslint-plugin": "4.11.1", "@typescript-eslint/parser": "4.11.1", "eslint": "7.17.0", @@ -44,6 +45,7 @@ "eslint-plugin-import": "2.22.1", "husky": "4.3.6", "lint-staged": "10.5.3", + "lodash": "^4.17.20", "prettier": "2.2.1", "reflect-metadata": "0.1.13", "rxjs": "6.6.3", diff --git a/website/docs/bot-injection.md b/website/docs/bot-injection.md index 8aa67df..316b1e9 100644 --- a/website/docs/bot-injection.md +++ b/website/docs/bot-injection.md @@ -8,10 +8,10 @@ slug: /bot-injection At times you may need to access the native `Telegraf` instance. For example, you may want to connect stage middleware. You can inject the Telegraf by using the `@InjectBot()` decorator as follows: ```typescript import { Injectable } from '@nestjs/common'; -import { InjectBot, TelegrafProvider } from 'nestjs-telegraf'; +import { InjectBot, TelegrafProvider, Context } from 'nestjs-telegraf'; @Injectable() export class BotSettingsService { - constructor(@InjectBot() private bot: TelegrafProvider) {} + constructor(@InjectBot() private bot: TelegrafProvider) {} } ``` diff --git a/website/docs/error-handling.md b/website/docs/error-handling.md new file mode 100644 index 0000000..a684e81 --- /dev/null +++ b/website/docs/error-handling.md @@ -0,0 +1,31 @@ +--- +id: error-handling +title: Error handling +sidebar_label: Error handling +slug: /error-handling +--- + +By default, `nestjs-telegraf` catches all errors using the `Logger` built into NestJS. + +Use can disable global errors catch with `disableGlobalCatch`: +```typescript +TelegrafModule.forRoot({ + disableGlobalCatch: true, +}), +``` + +After that you can override errors handling with bot instance `catch` function. + +```typescript +import { Injectable } from '@nestjs/common'; +import { InjectBot, TelegrafProvider, Context } from 'nestjs-telegraf'; + +@Injectable() +export class BotSettingsService { + constructor(@InjectBot() private bot: TelegrafProvider) { + this.bot.catch((err, ctx) => { + console.log(`Ooops, encountered an error for ${ctx.updateType}`, err); + }); + } +} +``` diff --git a/website/docs/getting-updates.md b/website/docs/getting-updates.md index df6e31a..a42d3fe 100644 --- a/website/docs/getting-updates.md +++ b/website/docs/getting-updates.md @@ -11,23 +11,26 @@ By default, the bot receives updates using long-polling and requires no addition ## Webhooks -If you want to configure a telegram bot webhook, you need to get a middleware from `TelegrafProvider` for connect it in your `main.ts` file. +If you want to configure a telegram bot webhook, you need to get a middleware via `getBotToken` helper in your `main.ts` file. To access it, you must use the `app.get()` method, followed by the provider reference: ```typescript -const telegrafProvider = app.get('TelegrafProvider'); +import { getBotToken } from 'nestjs-telegraf'; + +// ... +const bot = app.get(getBotToken()); ``` Now you can connect middleware: ```typescript -app.use(telegrafProvider.webhookCallback('/secret-path')); +app.use(bot.webhookCallback('/secret-path')); ``` The last step is to specify launchOptions in `forRoot` method: ```typescript TelegrafModule.forRootAsync({ - imports: [ConfigModule.forFeature(telegrafModuleConfig)], - useFactory: async (configService: ConfigService) => ({ + imports: [ConfigModule], + useFactory: (configService: ConfigService) => ({ token: configService.get('TELEGRAM_BOT_TOKEN'), launchOptions: { webhook: { diff --git a/website/docs/middlewares.md b/website/docs/middlewares.md new file mode 100644 index 0000000..b16387a --- /dev/null +++ b/website/docs/middlewares.md @@ -0,0 +1,13 @@ +--- +id: middlewares +title: Middlewares +sidebar_label: Middlewares +slug: /middlewares +--- + +`nestjs-telegraf` has support of the Telegraf middleware packages. To use an existing middleware package, simply import it and add it to the middlewares array: +```typescript +TelegrafModule.forRoot({ + middlewares: [session()], +}), +``` diff --git a/website/docs/multiple-bots.md b/website/docs/multiple-bots.md new file mode 100644 index 0000000..7500a6f --- /dev/null +++ b/website/docs/multiple-bots.md @@ -0,0 +1,74 @@ +--- +id: multiple-bots +title: Multiple bots +sidebar_label: Multiple bots +slug: /multiple-bots +--- + +In some cases, you may need to run multiple bots at the same time. This can also be achieved with this module. To work with multiple bots, first create the bots. In this case, bot naming becomes mandatory. +```typescript +import { Module } from '@nestjs/common'; +import { ConfigModule } from '@nestjs/config'; +import { TelegrafModule } from 'nestjs-telegraf'; + +@Module({ + imports: [ + ConfigModule.forRoot(), + TelegrafModule.forRootAsync({ + imports: [ConfigModule], + botName: 'cat', + useFactory: (configService: ConfigService) => ({ + token: configService.get('CAT_BOT_TOKEN'), + }), + inject: [ConfigService], + }), + TelegrafModule.forRootAsync({ + imports: [ConfigModule.forFeature(telegrafModuleConfig)], + botName: 'dog', + useFactory: async (configService: ConfigService) => ({ + token: configService.get('DOG_BOT_TOKEN'), + }), + inject: [ConfigService], + }), + ], +}) +export class AppModule {} +``` + +:::caution +Please note that you shouldn't have multiple bots without a name, or with the same name, otherwise they will get overridden. +::: + +You can also inject the `Bot` for a given bot: +```typescript +import { Injectable } from '@nestjs/common'; +import { InjectBot, Telegraf, Context } from 'nestjs-telegraf'; + +@Injectable() +export class EchoService { + constructor(@InjectBot('cat') private catBot: Telegraf) {} +} +``` +To inject a given `Bot` to a custom provider (for example, factory provider), use the `getBotToken()` function passing the name of the bot as an argument. +```typescript +{ + provide: CatsService, + useFactory: (catBot: Telegraf) => { + return new CatsService(catBot); + }, + inject: [getBotToken('cat')], +} +``` +Another useful feature of the `nestjs-telegraf` module is the ability to choose which modules should handle updates for each launched bot. By default, module searches for handlers throughout the whole app. To limit this scan to only a subset of modules, use the include property. + +```typescript +TelegrafModule.forRootAsync({ + imports: [ConfigModule], + botName: 'cat', + useFactory: (configService: ConfigService) => ({ + token: configService.get('CAT_BOT_TOKEN'), + include: [CatsModule], + }), + inject: [ConfigService], +}), +``` diff --git a/website/sidebars.js b/website/sidebars.js index 9912645..ea88714 100644 --- a/website/sidebars.js +++ b/website/sidebars.js @@ -6,6 +6,9 @@ module.exports = { 'telegraf-methods', 'bot-injection', 'async-configuration', + 'multiple-bots', + 'middlewares', + 'error-handling', ], Extras: ['extras/standalone-applications'], 'API Reference': ['api-reference/decorators'],