ngork.exe http 80

package main

import (
	"fmt"
	"net/http"
	"wechat-oa/route"
)

func main() {
	fmt.Println("====== 微信公眾號(hào)服務(wù)器程序 ======")

	http.HandleFunc("/", route.WechatServer) // 處理微信服務(wù)器請(qǐng)求

	err := http.ListenAndServe(":80", nil)
	if err != nil {
		fmt.Println("wechat server start error", err)
	}
}

// 微信公眾號(hào)平臺(tái)配置
const (
	appId          = "xxxxxxxxxx" // 公眾號(hào)開發(fā)識(shí)別碼
	appSecret      = "xxxxxxxxxx" // 公眾號(hào)開發(fā)者密碼
	encodingAESKey = "xxxxxxxxxx" // 消息加密秘鑰
	token          = "xxxxxxxxxx" // 跟微信公眾平臺(tái)的token一樣即可
)

func WechatServer(w http.ResponseWriter, req *http.Request) {
	defer req.Body.Close()

	fmt.Println("服務(wù)器請(qǐng)求: ", req) //打印http的請(qǐng)求url
	// 1、驗(yàn)證消息是否來自微信服務(wù)器
	hashcode,signature,echostr := handleValid(req)
	if hashcode != signature { // 校驗(yàn)
		return
	}
	// 2、請(qǐng)求處理
	if req.Method == "GET" {
		_, _ = w.Write([]byte(echostr)) // 驗(yàn)證成功返回echostr
	} else if req.Method == "POST" {
		handleMsg(w,req)
	}else {
		_, _ = w.Write([]byte("error"))
	}
}

func handleValid(r *http.Request) (hashcode string,signature string,echostr string){
	//1.嘗試獲取4個(gè)字段
	nonce := r.URL.Query().Get("nonce")
	timestamp := r.URL.Query().Get("timestamp")
	signature = r.URL.Query().Get("signature")
	echostr = r.URL.Query().Get("echostr")
	//2. 賦值一個(gè)token

	//3.token，timestamp，nonce按字典排序的字符串list
	strs := sort.StringSlice{token, timestamp, nonce} // 使用本地的token生成校驗(yàn)
	sort.Strings(strs)
	str := ""
	for _, s := range strs {
		str += s
	}
	// 4. 哈希算法加密list得到hashcode
	h := sha1.New()
	h.Write([]byte(str))
	hashcode = fmt.Sprintf("%x", h.Sum(nil)) // h.Sum(nil) 做hash
	return
}

// 處理POST請(qǐng)求
// 微信服務(wù)器會(huì)將用戶發(fā)送的數(shù)據(jù)以POST請(qǐng)求的方式轉(zhuǎn)發(fā)給開發(fā)者服務(wù)器
func handleMsg(rw http.ResponseWriter, req *http.Request) {
	wc := wechat.NewWechat()
	//這里本地內(nèi)存保存access_token，也可選擇redis，memcache或者自定義cache
	memory := cache.NewMemory()
	cfg := &offConfig.Config{
		AppID:          appId,
		AppSecret:      appSecret,
		Token:          token,
		EncodingAESKey: encodingAESKey,
		Cache:          memory,
	}
	// 底層根據(jù)AppID和AppSecret獲取access_token并保存到cache中
	// access_token每?jī)蓚€(gè)小時(shí)刷新一次
	officialAccount := wc.GetOfficialAccount(cfg)
	// 傳入request和responseWriter
	server := officialAccount.GetServer(req, rw)
	// 設(shè)置接收消息的處理方法
	server.SetMessageHandler(controller.HandleMsg)
	//處理消息接收以及回復(fù)
	err := server.Serve()
	if err != nil {
		fmt.Println(err)
		return
	}
	//發(fā)送回復(fù)的消息
	_ = server.Send()
}

// 處理用戶發(fā)送的消息
func HandleMsg(msg *message.MixMessage) *message.Reply{
	fmt.Println("接收用戶消息:",msg.Content)
	switch msg.MsgType {
	case message.MsgTypeText: // 文本消息
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText(msg.Content),
		}
	case message.MsgTypeEvent: // 事件推送
		return service.HandleEventMsg(msg)
	default:
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText("感謝關(guān)注沐風(fēng)丶,更多功能正在開發(fā)中..."),
		}
	}
}

// 處理事件推送
func HandleEventMsg(msg *message.MixMessage) *message.Reply{
	switch msg.Event {
	case message.EventSubscribe: // 用戶訂閱
		fmt.Printf("%s 訂閱了你 \n",msg.FromUserName)
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText("感謝關(guān)注沐風(fēng)丶,(*￣︶￣)?你真好看"),
		}
	case message.EventUnsubscribe: // 用戶取消訂閱
		fmt.Printf("%s 取消了訂閱 \n",msg.FromUserName)
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText(""),
		}
	default:
		return &message.Reply{
			MsgType: message.MsgTypeText,
			MsgData: message.NewText("default"),
		}
	}
}