enable delay suggestions (#23)

PKuebler · Sep 2, 2020 · c488478 · c488478
1 parent 4a7380b
commit c488478
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 4 deletions.
diff --git a/cmd/bot.go b/cmd/bot.go
@@ -161,11 +161,16 @@ func BotCommand(ctx context.Context, cmd *cobra.Command, args []string) {
 				if tctx.IsRemoveSuggestions() {
 					reply.ReplyMarkup = tgbotapi.NewRemoveKeyboard(true)
 				} else if tctx.IsSuggestions() {
-					buttons := []tgbotapi.KeyboardButton{}
+					buttons := [][]tgbotapi.KeyboardButton{}
 					for _, button := range tctx.Suggestions() {
-						buttons = append(buttons, tgbotapi.NewKeyboardButton(button.Label))
+						buttons = append(
+							buttons,
+							tgbotapi.NewKeyboardButtonRow(
+								tgbotapi.NewKeyboardButton(button.Label),
+							),
+						)
 					}
-					keyboard := tgbotapi.NewReplyKeyboard(buttons)
+					keyboard := tgbotapi.NewReplyKeyboard(buttons...)
 					reply.ReplyMarkup = keyboard
 				}
 

diff --git a/pkg/interface/telegram/editdelay.go b/pkg/interface/telegram/editdelay.go
@@ -30,5 +30,16 @@ func (t *TelegramService) EditDelay(ctx telegramconversation.TContext) telegramc
 	}
 
 	ctx.SetStatePayload(alarm.GetID())
-	return ctx.SendWithState(fmt.Sprintf("Ab wie viel Abweichung von `%s > %s` soll bescheid gesagt werden? z.B. 1h5min oder 10m", alarm.GetTrainName(), alarm.GetFinalDestinationName()), "savedelay")
+	ctx.ChangeState("savedelay")
+
+	txt := fmt.Sprintf("Ab wie viel Abweichung von `%s > %s` soll bescheid gesagt werden? z.B. 1h5min oder 10m", alarm.GetTrainName(), alarm.GetFinalDestinationName())
+
+	buttons := []telegramconversation.TButton{}
+	units := []string{"1m", "5m", "10m", "15m", "20m", "30m", "1h", "1h30m"}
+	for _, unit := range units {
+		button := telegramconversation.NewTButton(unit, unit)
+		buttons = append(buttons, button)
+	}
+
+	return ctx.SendWithSuggestions(txt, buttons, 1)
 }