setup context in call backs

radiantspace · Mar 2, 2024 · 0e6cf70 · 0e6cf70
1 parent 6bef860
commit 0e6cf70
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/backend/app/telegram/telegram.go b/backend/app/telegram/telegram.go
@@ -320,8 +320,7 @@ func handleCallbackQuery(bot *telego.Bot, callbackQuery telego.CallbackQuery) {
 func handleCommandsInCallbackQuery(callbackQuery telego.CallbackQuery) {
 	chat := callbackQuery.Message.GetChat()
 	chatIDString := fmt.Sprint(chat.ID)
-	ctx := context.WithValue(context.Background(), models.UserContext{}, chatIDString)
-	ctx = context.WithValue(ctx, models.ClientContext{}, "telegram")
+	_, ctx, _, _ := lib.SetupUserAndContext(chatIDString, "telegram", chatIDString)
 	message := telego.Message{
 		Chat: telego.Chat{ID: chat.ID},
 		Text: "/" + callbackQuery.Data,
@@ -337,8 +336,7 @@ func handleEngineSwitchCallbackQuery(callbackQuery telego.CallbackQuery) {
 		chatID = chat.ID
 	}
 	chatIDString := fmt.Sprint(chatID)
-	ctx := context.WithValue(context.Background(), models.UserContext{}, chatIDString)
-	ctx = context.WithValue(ctx, models.ClientContext{}, "telegram")
+	_, ctx, _, _ := lib.SetupUserAndContext(chatIDString, "telegram", chatIDString)
 	currentEngine := redis.GetChatEngine(chatIDString)
 	if callbackQuery.Data == string(currentEngine) {
 		err := BOT.AnswerCallbackQuery(&telego.AnswerCallbackQueryParams{