From b5a9997b8265a28cc3ab817a63fabfdbe3e636e3 Mon Sep 17 00:00:00 2001
From: Evan Lin <evan.if.lin@linecorp.com>
Date: Thu, 25 Jan 2024 01:02:48 +0800
Subject: [PATCH] feat: refactor callback handler to handle query and response

- Add functionality to handle text messages in the callback handler
- Extract user ID from the event source
- Query the Notion database for results containing the user ID and the message text
- Return an error message if no results are found
- Marshal the results to JSON
- Send the JSON string to the GeminiChatComplete function
- Return the result of GeminiChatComplete to the user

Signed-off-by: Evan Lin <evan.if.lin@linecorp.com>
---
 bot.go | 40 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/bot.go b/bot.go
index fae808c..2cb2741 100644
--- a/bot.go
+++ b/bot.go
@@ -55,10 +55,46 @@ func callbackHandler(w http.ResponseWriter, r *http.Request) {
 		switch e := event.(type) {
 		case webhook.MessageEvent:
 			switch message := e.Message.(type) {
+
 			// Handle only on text message
 			case webhook.TextMessageContent:
-				// TODO: Will add LLM namecard search function here.
-				log.Printf("Got text message, ID: %s, text: %s", message.Id, message.Text)
+				// 取得用戶 ID
+				var uID string
+				switch source := e.Source.(type) {
+				case webhook.UserSource:
+					uID = source.UserId
+				case webhook.GroupSource:
+					uID = source.UserId
+				case webhook.RoomSource:
+					uID = source.UserId
+				}
+				//using test as keyword to query database
+				nDB := &NotionDB{
+					DatabaseID: os.Getenv("NOTION_DB_PAGEID"),
+					Token:      os.Getenv("NOTION_INTEGRATION_TOKEN"),
+				}
+
+				// Check email first before adding to database.
+				results, err := nDB.QueryDatabaseContains(uID, message.Text)
+				if err != nil || len(results) == 0 {
+					ret := "查不到資料，請重新輸入:" + err.Error()
+					if err := replyText(e.ReplyToken, ret); err != nil {
+						log.Print(err)
+					}
+					continue
+				}
+
+				// marshal results to json string
+				jsonData, err := json.Marshal(results)
+				if err != nil {
+					log.Println("Error parsing JSON:", err)
+				}
+
+				// send json string to gemini complete whole result.
+				ret := GeminiChatComplete("根據你的關鍵字，查詢到以下資料:" + string(jsonData))
+				if err := replyText(e.ReplyToken, ret); err != nil {
+					log.Print(err)
+				}
 
 			// Handle only on Sticker message
 			case webhook.StickerMessageContent: