refactor(telegram-driver): restrict event names

goodmind · Nov 20, 2016 · c2154a9 · c2154a9
1 parent 41f5273
commit c2154a9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/interfaces.ts b/src/interfaces.ts
@@ -4,6 +4,7 @@ import { TcombWebhookResponse, TcombRequest, TcombUpdate, TcombUpdatesState } fr
 export type Token = string
 export type GenericStream<T> = any
 export type DriverSink = TcombRequest | TcombWebhookResponse
+export type EventNames = 'message' | 'inline_query' | 'chosen_inline_result' | 'callback_query'
 
 export interface DriverOptions {
   webhook?: boolean

diff --git a/src/telegram-driver/telegram-driver.ts b/src/telegram-driver/telegram-driver.ts
@@ -8,7 +8,7 @@ import {
   DriverExecution,
   DriverSources,
   DriverSink,
-  Token
+  Token, EventNames
 } from '../interfaces'
 import { makeSources, makeUpdates, makeWebHook } from './sources'
 import { makeAPIRequest } from './api-request'
@@ -17,7 +17,7 @@ import { TcombUpdate, TcombUpdatesState, TcombWebhookResponse, TcombRequest } fr
 import { adapter, isWebhookResponse, convertStream } from '../helpers'
 
 function makeEventsSelector ({message, inlineQuery, chosenInlineResult, callbackQuery}: DriverSources) {
-  return function events (eventName: string): Observable<TcombUpdate> {
+  return function events (eventName: EventNames): Observable<TcombUpdate> {
     return $.case(() => eventName, {
       'message': message.share(),
       'inline_query': inlineQuery.share(),