support replyMessage in single/multiple options. closes #66

src/core/__tests__/bot.spec.ts

@@ -239,6 +239,61 @@ test('auto reply message', async () => {
   expect(onTalk).toBeCalledWith('Thanks', {}, 'session');
 });
 
+test('auto reply message for single choice', async () => {
+  const onTalk = jest.fn();
+  const rules = loadYaml(`
+  - message: Color
+    type: SingleChoice
+    replyMessage: Nice color!
+    options:
+      - label: red
+        replyMessage: Red! Nice!
+      - label: white
+        replyMessage: Really?
+  `);
+  const bot = new YveBot(rules, OPTS)
+    .on('talk', onTalk)
+    .start();
+
+  await sleep();
+  bot.hear('red');
+  await sleep();
+
+  expect(onTalk).toBeCalledWith('Color', rules[0], 'session');
+  expect(onTalk).toBeCalledWith('Red! Nice!', {}, 'session');
+  expect(onTalk).not.toBeCalledWith('Nice color!', {}, 'session');
+  expect(onTalk).not.toBeCalledWith('Really?', {}, 'session');
+});
+
+test('auto reply message for multiple choice', async () => {
+  const onTalk = jest.fn();
+  const rules = loadYaml(`
+  - message: Color
+    type: MultipleChoice
+    replyMessage: Nice color!
+    options:
+      - label: red
+        replyMessage: Red! Nice!
+      - label: white
+        replyMessage: Really?
+      - label: blue
+        replyMessage: Nooo!
+  `);
+  const bot = new YveBot(rules, OPTS)
+    .on('talk', onTalk)
+    .start();
+
+  await sleep();
+  bot.hear('red, white');
+  await sleep();
+
+  expect(onTalk).toBeCalledWith('Color', rules[0], 'session');
+  expect(onTalk).toBeCalledWith('Red! Nice!', {}, 'session');
+  expect(onTalk).not.toBeCalledWith('Nice color!', {}, 'session');
+  expect(onTalk).not.toBeCalledWith('Nooo!', {}, 'session');
+  expect(onTalk).not.toBeCalledWith('Really?', {}, 'session');
+});
+
 test('compiled template', async () => {
   const onTalk = jest.fn();
   const rules = loadYaml(`

src/core/controller.ts

@@ -8,7 +8,7 @@ async function validateAnswer(
   rule: IRule,
   bot: YveBot,
   executorIndex: number,
-) {
+): Promise<Answer | Answer[]> {
   const ruleValidators = rule.validators || [];
   const typeExecutors = bot.types[rule.type].executors || [];
   const currentTypeExecutor = typeExecutors[executorIndex] || {};
@@ -38,6 +38,26 @@ async function validateAnswer(
   return answers;
 }
 
+function getReplyMessage(rule: IRule, answers: Answer | Answer[]): string | null {
+  const { replyMessage } = rule;
+  if (!rule.options.length) {
+    return replyMessage;
+  }
+  let opt;
+  // multiple
+  if (answers instanceof Array) {
+    [opt = null] = answers
+      .map((a) => utils.findOptionByAnswer(rule.options, a))
+      .filter((o) => o.replyMessage) ;
+  }
+  // single
+  opt = utils.findOptionByAnswer(rule.options, answers);
+  if (opt && opt.replyMessage) {
+    return opt.replyMessage;
+  }
+  return replyMessage;
+}
+
 function compileMessage(bot: YveBot, message: string): string {
   const output = bot.store.output();
   const { indexes } = bot.controller;
@@ -242,9 +262,10 @@ export class Controller {
       bot.store.set(`output.${output}`, answer);
     }
 
-    if (rule.replyMessage) {
+    const replyMessage = getReplyMessage(rule, answer);
+    if (replyMessage) {
       const replyRule = Object.assign({}, bot.options.rule);
-      await this.sendMessage(rule.replyMessage, replyRule);
+      await this.sendMessage(replyMessage, replyRule);
     }
 
     // run post-actions

src/types.ts

@@ -36,6 +36,7 @@ export interface IRuleOption {
   label?: string;
   value?: string | number;
   synonyms?: string[];
+  replyMessage?: string;
   next?: RuleNext;
 }