diff --git a/src/index.js b/src/index.js index 3895f2c..d469ed3 100644 --- a/src/index.js +++ b/src/index.js @@ -117,7 +117,12 @@ export default class Bot extends EventEmitter { * gets the update * @return {object} returns the bot object */ - command(command, listener) { + command(command, customMessage, listener) { + if (!listener) { + listener = customMessage; + customMessage = {}; + } + const regex = /[^\s]+/; const cmd = command.match(regex)[0].trim(); @@ -125,6 +130,7 @@ export default class Bot extends EventEmitter { this._userEvents.push({ pattern: new RegExp(`^/${cmd}`), parse: argumentParser.bind(null, command), + customMessage, listener }); @@ -211,9 +217,16 @@ export default class Bot extends EventEmitter { const bot = this; function* getAnswer() { + const customMessage = ev.customMessage; + for (const param of requiredParams) { - const ga = new Message().to(msg.chat.id) - .text(`Enter value for ${param}`); + let ga; + if (customMessage[param]) { + ga = new Message().to(msg.chat.id).text(customMessage[param]); + } else { + ga = new Message().to(msg.chat.id) + .text(`Enter value for ${param}`); + } yield bot.send(ga).then(answer => { args[param] = answer.text; });