Merge pull request #12 from snehalyelmati/dev

Quick replies

Author: snehalyelmati

main.go

@@ -6,11 +6,12 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"regexp"
 	"strconv"
 	"strings"
 
 	fiber "github.com/gofiber/fiber/v2"
-	"github.com/snehalyelmati/telegram-bot-golang/models"
+	. "github.com/snehalyelmati/telegram-bot-golang/models"
 	DialogFlow "github.com/snehalyelmati/telegram-bot-golang/services"
 )
 
@@ -42,12 +43,14 @@ func main() {
 
 	app.Post(URI, func(c *fiber.Ctx) error {
 		l.Println("Post request")
-		reqBody := new(models.MessageReq)
+		reqBody := new(TelgramMessageReq)
 		json.Unmarshal(c.Body(), reqBody)
 		l.Printf("%v", reqBody)
 
+		// initialize stuff
 		inputMessage := reqBody.Message.Text
 		if inputMessage == "/start" {
+			// save the user data
 			inputMessage = "hi"
 		}
 
@@ -58,6 +61,12 @@ func main() {
 		// })
 
 		// get response from dialogflow
+
+		// sample format for text and quick replies:
+		// statement1
+		// statement2
+		// <qrType1, qrName1, qrText1><qrType2, qrName2, qrText2>...
+
 		sessionID := strconv.Itoa(reqBody.Message.Chat.ID)
 		response, queryResult, err := DialogFlow.DetectIntentText(PROJECT_ID, sessionID, inputMessage, LANGUAGE)
 		if err != nil {
@@ -77,6 +86,43 @@ func main() {
 				l.Println(err)
 			}
 
+			response := new(TelegramMessageRes)
+			response.ChatID = strconv.Itoa(reqBody.Message.Chat.ID)
+			response.ReplyMarkup.Keyboard = make([][]TelegramMessageResReplyMarkupKeyboard, 0)
+
+			// get all the Quick Replies from the text
+			// prepare a map with label, postBack and qrType
+			regEx := `<(?P<qrType>[\sa-zA-Z0-9]*),(?P<postBack>[\sa-zA-Z0-9]*),(?P<label>[\sa-zA-Z0-9]*)>`
+			quickReplies, statement := getParams(regEx, statement)
+			response.Text = statement
+			l.Println("Quick replies:", quickReplies)
+
+			// initialize keyboard obj
+			response.ReplyMarkup.Keyboard = append(response.ReplyMarkup.Keyboard, []TelegramMessageResReplyMarkupKeyboard{})
+
+			// append available quick replies
+			for _, qr := range quickReplies {
+				if qr["qrType"] == "OPT" {
+					// append to response.ReplyMarkup.Keyboard
+					response.ReplyMarkup.Keyboard = append(response.ReplyMarkup.Keyboard, []TelegramMessageResReplyMarkupKeyboard{{Text: qr["label"]}})
+				} else if qr["qrType"] == "SUGT" {
+					// TODO: add buttons for suggestions
+				}
+			}
+			l.Println("Response ReplyMarkup Keyboard:", response.ReplyMarkup.Keyboard)
+
+			if len(response.ReplyMarkup.Keyboard) > 0 {
+				response.ReplyMarkup.OneTimeKeyboard = true
+				response.ReplyMarkup.ResizeKeyboard = true
+			}
+
+			// prepare a json object from the response object
+			data, err = json.Marshal(response)
+			if err != nil {
+				l.Println(err)
+			}
+
+			// call the Telegram API
 			_, err = http.Post(TELEGRAM_API+"/sendMessage", "application/json", bytes.NewBuffer(data))
 			if err != nil {
 				l.Println(err)
@@ -89,3 +135,30 @@ func main() {
 	app.Listen(":3000")
 	l.Println("Server running on port 3000")
 }
+
+/**
+ * Parses string with the given regular expression and returns the
+ * group values defined in the expression.
+ *
+ */
+func getParams(regEx, str string) ([]map[string]string, string) {
+
+	var compRegEx = regexp.MustCompile(regEx)
+	matchArr := compRegEx.FindAllStringSubmatch(str, -1)
+	if len(matchArr) > 0 {
+		str = compRegEx.ReplaceAllString(str, "")
+	}
+
+	res := make([]map[string]string, len(matchArr)) // map array
+
+	for _, match := range matchArr {
+		paramsMap := make(map[string]string)
+		for i, name := range compRegEx.SubexpNames() {
+			if i > 0 && i <= len(match) {
+				paramsMap[name] = match[i]
+			}
+		}
+		res = append(res, paramsMap)
+	}
+	return res, str
+}

models/MessageReq.go renamed to models/TelegramMessageReq.go

@@ -1,6 +1,6 @@
 package models
 
-type MessageReq struct {
+type TelgramMessageReq struct {
 	UpdateID int `json:"update_id"`
 	Message  struct {
 		MessageID int `json:"message_id"`

models/TelegramMessageRes.go

@@ -0,0 +1,18 @@
+package models
+
+type TelegramMessageRes struct {
+	ChatID      string                        `json:"chat_id"`
+	Text        string                        `json:"text"`
+	ReplyMarkup TelegramMessageResReplyMarkup `json:"reply_markup"`
+}
+
+type TelegramMessageResReplyMarkup struct {
+	OneTimeKeyboard bool                                      `json:"one_time_keyboard"`
+	ResizeKeyboard  bool                                      `json:"resize_keyboard"`
+	Keyboard        [][]TelegramMessageResReplyMarkupKeyboard `json:"keyboard"`
+}
+
+type TelegramMessageResReplyMarkupKeyboard struct {
+	Text string `json:"text"`
+}
+