54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package action
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"git.lxtend.com/qqbot/model"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func init() {
|
|
}
|
|
|
|
var ActionManager = actionManager{}
|
|
|
|
type actionManager struct {
|
|
botConn *websocket.Conn
|
|
}
|
|
|
|
func (am *actionManager) SetConn(conn *websocket.Conn) {
|
|
am.botConn = conn
|
|
}
|
|
|
|
func (am *actionManager) SendAction(action string) error {
|
|
return am.botConn.WriteMessage(websocket.TextMessage, []byte(action))
|
|
}
|
|
|
|
func (am *actionManager) SendMsg(reply model.Reply) error {
|
|
if reply.ReferOriginMsg {
|
|
reply.ReplyMsg = fmt.Sprintf("%s%s", GenReply(reply.FromMsg.OriginMsgId, "", 0, 0, 0), reply.ReplyMsg)
|
|
}
|
|
sendPkg := model.GenSendPkg(reply.FromMsg.UserId, reply.FromMsg.GroupInfo.GroupId, reply.ReplyMsg, false)
|
|
sendPkgJson, err := json.Marshal(sendPkg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err = am.botConn.WriteMessage(websocket.TextMessage, sendPkgJson); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (am *actionManager) DrawbackMsg(msgId int32) error {
|
|
withDrawPkg := model.GenDrawbackPkg(msgId)
|
|
withDrawPkgJson, err := json.Marshal(withDrawPkg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err = am.botConn.WriteMessage(websocket.TextMessage, withDrawPkgJson); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|