handler.go

package weixin

import (
	"github.com/arstd/log"
	"net/http"
)

// 各类消息处理器
var (
	RecvTextHandler       func(*RecvText) ReplyMsg
	RecvImageHandler      func(*RecvImage) ReplyMsg
	RecvVoiceHandler      func(*RecvVoice) ReplyMsg
	RecvVideoHandler      func(*RecvVideo) ReplyMsg
	RecvShortVideoHandler func(*RecvVideo) ReplyMsg
	RecvLocationHandler   func(*RecvLocation) ReplyMsg
	RecvLinkHandler       func(*RecvLink) ReplyMsg
)

// 各类事件处理器
var (
	EventSubscribeHandler             func(w http.ResponseWriter, r *http.Request, sub *EventSubscribe) ReplyMsg
	EventUnsubscribeHandler           func(*EventSubscribe) ReplyMsg
	EventLocationHandler              func(*EventLocation) ReplyMsg
	EventClickHandler                 func(*EventClick) ReplyMsg
	EventViewHandler                  func(*EventView) ReplyMsg
	EventTemplateSendJobFinishHandler func(*EventTemplateSendJobFinish) ReplyMsg

	EventScancodePushHandler    func(*EventScancodePush) ReplyMsg
	EventScancodeWaitmsgHandler func(*EventScancodeWaitmsg) ReplyMsg
	EventPicSysphotoHandler     func(*EventPicSysphoto) ReplyMsg
	EventPicPhotoOrAlbumHandler func(*EventPicPhotoOrAlbum) ReplyMsg
	EventPicWeixinHandler       func(*EventPicWeixin) ReplyMsg
	EventLocationSelectHandler  func(*EventLocationSelect) ReplyMsg

	EventQualificationVerifySuccessHandler func(*EventQualificationVerifySuccess) ReplyMsg // 资质认证成功
	EventQualificationVerifyFailHandler    func(*EventQualificationVerifyFail) ReplyMsg    // 资质认证失败
	EventNamingVerifySuccessHandler        func(*EventNamingVerifySuccess) ReplyMsg        // 名称认证成功（即命名成功）
	EventNamingVerifyFailHandler           func(*EventNamingVerifyFail) ReplyMsg           // 名称认证失败
	EventAnnualRenewHandler                func(*EventAnnualRenew) ReplyMsg                // 年审通知
	EventVerifyExpiredHandler              func(*EventVerifyExpired) ReplyMsg              // 认证过期失效通知
)

// RecvDefaultHandler 如果没有注册某类消息处理器，那么收到这类消息时，使用这个默认处理器
var RecvDefaultHandler = func(msg *Message) (reply ReplyMsg) {
	log.Debugf("unregistered receive message handler %s, use RecvDefaultHandler", msg.MsgType)
	return nil
}

// EventDefaultHandler 如果没有注册某类事件处理器，那么收到这类事件时，使用这个默认处理器
var EventDefaultHandler = func(msg *Message) (reply ReplyMsg) {
	log.Debugf("unregistered receive event handler %s, use EventDefaultHandler", msg.Event)
	return nil
}

// HandleMessage 处理各类消息
func HandleMessage(w http.ResponseWriter, r *http.Request, msg *Message) (ret ReplyMsg) {
	log.Debugf("process `%s` message", msg.MsgType)

	switch msg.MsgType {
	case MsgTypeText:
		if RecvTextHandler != nil {
			return RecvTextHandler(NewRecvText(msg))
		}
	case MsgTypeImage:
		if RecvImageHandler != nil {
			return RecvImageHandler(NewRecvImage(msg))
		}
	case MsgTypeVoice:
		if RecvVoiceHandler != nil {
			return RecvVoiceHandler(NewRecvVoice(msg))
		}
	case MsgTypeVideo:
		if RecvVideoHandler != nil {
			return RecvVideoHandler(NewRecvVideo(msg))
		}
	case MsgTypeShortVideo:
		if RecvShortVideoHandler != nil {
			return RecvShortVideoHandler(NewRecvVideo(msg))
		}
	case MsgTypeLocation:
		if RecvLocationHandler != nil {
			return RecvLocationHandler(NewRecvLocation(msg))
		}
	case MsgTypeLink:
		if RecvLinkHandler != nil {
			return RecvLinkHandler(NewRecvLink(msg))
		}
	case MsgTypeEvent:
		return HandleEvent(w,r,msg)
	default:
		log.Errorf("unexpected receive MsgType: %s", msg.MsgType)
		return nil
	}

	return RecvDefaultHandler(msg)
}

// HandleEvent 处理各类事件
func HandleEvent(w http.ResponseWriter, r *http.Request, msg *Message) (reply ReplyMsg) {
	log.Debugf("process `%s` event", msg.MsgType)

	switch msg.Event {
	case EventTypeSubscribe:
		if EventSubscribeHandler != nil {
			return EventSubscribeHandler(w,r, NewEventSubscribe(msg))
		}
	case EventTypeUnsubscribe:
		if EventUnsubscribeHandler != nil {
			return EventUnsubscribeHandler(NewEventSubscribe(msg))
		}
	case EventTypeLocation:
		if EventLocationHandler != nil {
			return EventLocationHandler(NewEventLocation(msg))
		}
	case EventTypeClick:
		if EventClickHandler != nil {
			return EventClickHandler(NewEventClick(msg))
		}
	case EventTypeView:
		if EventViewHandler != nil {
			return EventViewHandler(NewEventView(msg))
		}
	case EventTypeTemplateSendJobFinish:
		if EventTemplateSendJobFinishHandler != nil {
			return EventTemplateSendJobFinishHandler(NewEventTemplateSendJobFinish(msg))
		}
	case EventTypeScancodePush:
		if EventScancodePushHandler != nil {
			return EventScancodePushHandler(NewEventScancodePush(msg))
		}
	case EventTypeScancodeWaitmsg:
		if EventScancodeWaitmsgHandler != nil {
			return EventScancodeWaitmsgHandler(NewEventScancodeWaitmsg(msg))
		}
	case EventTypePicSysphoto:
		if EventPicSysphotoHandler != nil {
			return EventPicSysphotoHandler(NewEventPicSysphoto(msg))
		}
	case EventTypePicPhotoOrAlbum:
		if EventPicPhotoOrAlbumHandler != nil {
			return EventPicPhotoOrAlbumHandler(NewEventPicPhotoOrAlbum(msg))
		}
	case EventTypePicWeixin:
		if EventPicWeixinHandler != nil {
			return EventPicWeixinHandler(NewEventPicWeixin(msg))
		}
	case EventTypeLocationSelect:
		if EventLocationSelectHandler != nil {
			return EventLocationSelectHandler(NewEventLocationSelect(msg))
		}
	case EventTypeQualificationVerifySuccess:
		if EventQualificationVerifySuccessHandler != nil {
			return EventQualificationVerifySuccessHandler(NewEventQualificationVerifySuccess(msg))
		}
	case EventTypeQualificationVerifyFail:
		if EventQualificationVerifyFailHandler != nil {
			return EventQualificationVerifyFailHandler(NewEventQualificationVerifyFail(msg))
		}
	case EventTypeNamingVerifySuccess:
		if EventNamingVerifySuccessHandler != nil {
			return EventNamingVerifySuccessHandler(NewEventNamingVerifySuccess(msg))
		}
	case EventTypeNamingVerifyFail:
		if EventNamingVerifyFailHandler != nil {
			return EventNamingVerifyFailHandler(NewEventNamingVerifyFail(msg))
		}
	case EventTypeAnnualRenew:
		if EventAnnualRenewHandler != nil {
			return EventAnnualRenewHandler(NewEventAnnualRenew(msg))
		}
	case EventTypeVerifyExpired:
		if EventVerifyExpiredHandler != nil {
			return EventVerifyExpiredHandler(NewEventVerifyExpired(msg))
		}
	default:
		log.Errorf("unexpected receive EventType: %s", msg.Event)
		return nil
	}

	return EventDefaultHandler(msg)
}