xiaoice.go

package main

import (
	"sync"

	wx "github.com/KevinGong2013/wechat"
)

type xiaoice struct {
	sync.Mutex
	un       string
	waitting []string
	bot      *wx.WeChat
}

func newXiaoice(wx *wx.WeChat) *xiaoice {
	x := &xiaoice{}
	x.bot = wx
	return x
}

func (x *xiaoice) autoReplay(msg wx.EventMsgData) {
	if msg.IsSentByMySelf {
		return
	}
	if msg.FromUserName == x.un { // 小冰发来的消息
		x.Lock()
		x.Unlock()

		count := len(x.waitting)
		if count == 0 {
			logger.Warnf(`msg Form xiaoice %s`, msg.Content)
			return
		}
		to := x.waitting[count-1]
		x.waitting = x.waitting[:count-1]

		if msg.IsMediaMsg {
			if path, err := x.bot.DownloadMedia(msg.MediaURL, msg.OriginalMsg[`MsgId`].(string)); err == nil {
				x.bot.SendFile(path, to)
			}
		} else {
			x.bot.SendTextMsg(msg.Content, to)
		}
	} else if !msg.IsSentByMySelf { // 转发别人的消息到小冰
		var err error
		if msg.IsMediaMsg {
			if path, e := x.bot.DownloadMedia(msg.MediaURL, msg.OriginalMsg[`MsgId`].(string)); e == nil {
				err = x.bot.SendFile(path, x.un)
			} else {
				err = e
			}
		} else {
			err = x.bot.SendTextMsg(msg.Content, x.un)
		}

		if err == nil {
			x.Lock()
			defer x.Unlock()
			x.waitting = append(x.waitting, msg.FromUserName)
		} else {
			logger.Error(err)
		}
	}
}