182 lines
3.4 KiB
Go
182 lines
3.4 KiB
Go
package linemsg
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/url"
|
|
"os"
|
|
"path"
|
|
|
|
"git.trj.tw/golang/mtfosbot/model"
|
|
"git.trj.tw/golang/mtfosbot/module/apis/line"
|
|
"git.trj.tw/golang/mtfosbot/module/config"
|
|
"git.trj.tw/golang/mtfosbot/module/es"
|
|
lineobj "git.trj.tw/golang/mtfosbot/module/line-message/line-object"
|
|
msgcmd "git.trj.tw/golang/mtfosbot/module/message-command"
|
|
)
|
|
|
|
func messageType(e *lineobj.EventObject) {
|
|
msg := e.Message
|
|
mtype, ok := msg["type"]
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
if t, ok := mtype.(string); ok {
|
|
switch t {
|
|
case "text":
|
|
textMsg(e)
|
|
break
|
|
case "image":
|
|
imageMsg(e)
|
|
break
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func textMsg(e *lineobj.EventObject) {
|
|
msg := e.Message
|
|
mtxt, ok := msg["text"]
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
// group action
|
|
if e.Source.Type == "group" {
|
|
if txt, ok := mtxt.(string); ok {
|
|
msgcmd.ParseLineMsg(txt, e.ReplyToken, e.Source)
|
|
saveTextMsgToLog(txt, e.Source)
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func imageMsg(e *lineobj.EventObject) {
|
|
msg := e.Message
|
|
imgID, ok := msg["id"]
|
|
if !ok {
|
|
return
|
|
}
|
|
// group action
|
|
if e.Source.Type == "group" {
|
|
if id, ok := imgID.(string); ok {
|
|
saveImageMsgToLog(id, e.Source)
|
|
}
|
|
}
|
|
}
|
|
|
|
func getSourceUser(uid, gid string) (u *model.LineUser, err error) {
|
|
userData, err := model.GetLineUserByID(uid)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if userData == nil {
|
|
tmpu, err := line.GetUserInfo(uid, gid)
|
|
if err != nil || tmpu == nil {
|
|
return nil, err
|
|
}
|
|
userData = &model.LineUser{}
|
|
userData.ID = tmpu.UserID
|
|
userData.Name = tmpu.DisplayName
|
|
err = userData.Add()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return userData, nil
|
|
}
|
|
|
|
func saveTextMsgToLog(txt string, s *lineobj.SourceObject) {
|
|
u, err := getSourceUser(s.UserID, s.GroupID)
|
|
if err != nil || u == nil {
|
|
return
|
|
}
|
|
|
|
go saveLineMessageLogToES(s.GroupID, s.UserID, txt, "text")
|
|
model.AddLineMessageLog(s.GroupID, s.UserID, txt, "text")
|
|
}
|
|
|
|
func saveImageMsgToLog(id string, s *lineobj.SourceObject) {
|
|
u, err := getSourceUser(s.UserID, s.GroupID)
|
|
if err != nil || u == nil {
|
|
return
|
|
}
|
|
|
|
mime, err := line.GetContentHead(id)
|
|
if err != nil || len(mime) == 0 {
|
|
return
|
|
}
|
|
|
|
ext := ""
|
|
switch mime {
|
|
case "image/jpeg":
|
|
ext = ".jpg"
|
|
break
|
|
case "image/jpg":
|
|
ext = ".jpg"
|
|
break
|
|
case "image/png":
|
|
ext = ".png"
|
|
break
|
|
default:
|
|
return
|
|
}
|
|
|
|
conf := config.GetConf()
|
|
|
|
fname := fmt.Sprintf("log_%s%s", id, ext)
|
|
|
|
fullPath := path.Join(conf.LogImageRoot, fname)
|
|
|
|
w, err := os.Create(fullPath)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer w.Close()
|
|
|
|
err = line.DownloadContent(id, w)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
furl, err := url.Parse(conf.URL)
|
|
if err == nil {
|
|
furl, err = furl.Parse(fmt.Sprintf("/image/line_log_image/%s", fname))
|
|
if err == nil {
|
|
fname = furl.String()
|
|
}
|
|
}
|
|
|
|
go saveLineMessageLogToES(s.GroupID, s.UserID, fname, "image")
|
|
model.AddLineMessageLog(s.GroupID, s.UserID, fname, "image")
|
|
}
|
|
|
|
func saveLineMessageLogToES(gid, uid, content, msgType string) {
|
|
lineGroup, err := model.GetLineGroup(gid)
|
|
if err != nil {
|
|
log.Println("get line group error :: ", err)
|
|
return
|
|
}
|
|
lineUser, err := model.GetLineUserByID(uid)
|
|
if err != nil {
|
|
log.Println("get line user error :: ", err)
|
|
return
|
|
}
|
|
|
|
logMsg := make(map[string]interface{})
|
|
|
|
logMsg["message"] = content
|
|
logMsg["type"] = msgType
|
|
logMsg["group"] = lineGroup.ID
|
|
logMsg["group_name"] = lineGroup.Name
|
|
logMsg["user"] = lineUser.ID
|
|
logMsg["user_name"] = lineUser.Name
|
|
|
|
err = es.PutLog("log", logMsg)
|
|
if err != nil {
|
|
log.Println("put log fail :: ", err)
|
|
}
|
|
}
|