From 0e7ba94f8bdf3d9611b423f01f0141a9cffda2c7 Mon Sep 17 00:00:00 2001
From: shoopea <shoopea@siteop.biz>
Date: Sat, 18 May 2019 16:12:29 +0800
Subject: [PATCH] test

---
 bot.go |  5 ++++-
 sql.go | 17 +++++++++++++++++
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/bot.go b/bot.go
index 197d653..508d4d6 100644
--- a/bot.go
+++ b/bot.go
@@ -177,8 +177,11 @@ func botMsgDump(m *tb.Message) {
 	r := regexp.MustCompile("^[0-9]+$")
 	if r.MatchString(m.Payload) {
 		objId, _ := strconv.ParseInt(m.Payload, 10, 64)
-		objSubTypeId, err := getObjSubTypeId(objId)
+		objTypeId, err := getObjTypeId(objId)
+		logOnError(err, "botMsgDump : getObjSubTypeId")
 		if err != nil {
+			res = `Error retrieving the message`
+		} else if objTypeId != objTypeMessage {
 			res = `This is not a message reference`
 		} else {
 			cwm, err := getMsg(objId)
diff --git a/sql.go b/sql.go
index e563ab4..4dc3d68 100644
--- a/sql.go
+++ b/sql.go
@@ -1391,6 +1391,23 @@ func getMsg(objId int64) (*ChatWarsMessage, error) {
 	return m, nil
 }
 
+func getObjTypeId(objId int64) (int64, error) {
+	var objTypeId int64
+
+	stmt, err := db.Prepare(`SELECT o.obj_type_id FROM obj o WHERE o.id = ?`)
+	if err != nil {
+		return 0, err
+	}
+	defer stmt.Close()
+
+	err = stmt.QueryRow(objId).Scan(&objTypeId)
+	if err != nil {
+		return 0, err
+	}
+
+	return objTypeId, nil
+}
+
 func getObjSubTypeId(objId int64) (int64, error) {
 	var objSubTypeId int64