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) Started() bool {
	return am.botConn != nil
}

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", GenReplyCQ(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
}