微信接入

zhima1234 · zhima1234 · commit b8561b0eafe9 · 2017-07-18T18:19:31.000+08:00
diff --git a/src/http/controller/routes.go b/src/http/controller/routes.go
@@ -34,5 +34,7 @@ func RegisterRoutes(g *echo.Group) {
 	new(OAuthController).RegisterRoute(g)
 	new(WebsocketController).RegisterRoute(g)
 
+	new(WechatController).RegisterRoute(g)
+
 	new(InstallController).RegisterRoute(g)
 }
diff --git a/src/http/controller/wechat.go b/src/http/controller/wechat.go
@@ -0,0 +1,49 @@
+// Copyright 2017 The StudyGolang Authors. All rights reserved.
+// Use of self source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+// http://studygolang.com
+// Author: polaris	polaris@studygolang.com
+
+package controller
+
+import (
+	"fmt"
+	"io/ioutil"
+	"logic"
+	"net/http"
+
+	"github.com/labstack/echo"
+)
+
+type WechatController struct{}
+
+// 注册路由
+func (self WechatController) RegisterRoute(g *echo.Group) {
+	g.Any("/wechat/autoreply", self.AutoReply)
+}
+
+func (self WechatController) AutoReply(ctx echo.Context) error {
+	fmt.Printf("form values:%+v\n", ctx.FormParams())
+	fmt.Printf("query values:%+v\n", ctx.QueryParams())
+
+	// 配置微信（不校验，直接返回成功）
+	if ctx.QueryParam("echostr") != "" {
+		return ctx.String(http.StatusOK, ctx.QueryParam("echostr"))
+	}
+
+	body, err := ioutil.ReadAll(ctx.Request().Body())
+	if err != nil {
+		return ctx.String(http.StatusOK, "failure")
+	}
+
+	if len(body) == 0 {
+		return ctx.String(http.StatusOK, "没有获取到消息内容")
+	}
+
+	respContent, err := logic.DefaultWechat.AutoReply(ctx, body)
+	if err != nil {
+		return ctx.String(http.StatusOK, "对不起服务错误！")
+	}
+
+	return ctx.String(http.StatusOK, respContent)
+}
diff --git a/src/logic/rank.go b/src/logic/rank.go
@@ -155,6 +155,10 @@ func (self RankLogic) FindDAURank(ctx context.Context, num int, ymds ...string)
 		weights = append(weights, weight)
 	}
 
+	if len(uids) == 0 {
+		return nil
+	}
+
 	userMap := DefaultUser.FindDAUUsers(ctx, uids)
 	users := make([]*model.User, len(userMap))
 	for i, uid := range uids {
diff --git a/src/logic/wechat.go b/src/logic/wechat.go
@@ -0,0 +1,125 @@
+// Copyright 2017 The StudyGolang Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+// http://studygolang.com
+// Author:polaris	polaris@studygolang.com
+
+package logic
+
+import (
+	"encoding/xml"
+	"fmt"
+	"model"
+	"strings"
+
+	"golang.org/x/net/context"
+
+	"github.com/polaris1119/config"
+)
+
+type WechatLogic struct{}
+
+var DefaultWechat = WechatLogic{}
+
+func (self WechatLogic) AutoReply(ctx context.Context, reqData []byte) (string, error) {
+	objLog := GetLogger(ctx)
+
+	wechatMsg := &model.WechatMsg{}
+	err := xml.Unmarshal(reqData, wechatMsg)
+	if err != nil {
+		objLog.Errorln("wechat autoreply xml unmarshal error:", err)
+		return "", err
+	}
+
+	switch wechatMsg.MsgType {
+	case model.WeMsgTypeText:
+		if strings.Contains(wechatMsg.Content, "晨读") {
+			return self.readingContent(ctx, wechatMsg)
+		} else {
+			return self.searchContent(ctx, wechatMsg)
+		}
+	case model.WeMsgTypeEvent:
+		switch wechatMsg.Event {
+		case model.WeEventSubscribe:
+			return config.ConfigFile.MustValue("wechat", "subscribe"), nil
+		}
+	}
+
+	return "", nil
+}
+
+func (self WechatLogic) readingContent(ctx context.Context, wechatMsg *model.WechatMsg) (string, error) {
+
+	var formatContent = func(reading *model.MorningReading) string {
+		if reading.Inner == 0 {
+			return fmt.Sprintf("%s\n%s", reading.Content, reading.Url)
+		}
+
+		host := "http://"
+		if WebsiteSetting.OnlyHttps {
+			host = "https://"
+		}
+		return fmt.Sprintf("%s\n%s%s/articles/%d", reading.Content, host, WebsiteSetting.Domain, reading.Inner)
+	}
+
+	var readings []*model.MorningReading
+	if wechatMsg.Content == "最新晨读" {
+		readings = DefaultReading.FindBy(ctx, 1, model.RtypeGo)
+		if len(readings) == 0 {
+			return "没有找到您要的内容", nil
+		}
+
+		return formatContent(readings[0]), nil
+	}
+
+	readings = DefaultReading.FindBy(ctx, 5, model.RtypeGo)
+
+	respContentSlice := make([]string, len(readings))
+	for i, reading := range readings {
+		respContentSlice[i] = formatContent(reading)
+	}
+
+	return strings.Join(respContentSlice, "\n\n"), nil
+}
+
+func (self WechatLogic) searchContent(ctx context.Context, wechatMsg *model.WechatMsg) (string, error) {
+	objLog := GetLogger(ctx)
+
+	respBody, err := DefaultSearcher.SearchByField("title", wechatMsg.Content, 0, 5)
+	if err != nil {
+		objLog.Errorln("wechat search by field error:", err)
+		return "", err
+	}
+
+	if respBody.NumFound == 0 {
+		return "没有找到您要的内容", nil
+	}
+
+	host := WebsiteSetting.Domain
+	if WebsiteSetting.OnlyHttps {
+		host = "https://" + host
+	} else {
+		host = "http://" + host
+	}
+
+	respContentSlice := make([]string, len(respBody.Docs))
+	for i, doc := range respBody.Docs {
+		url := ""
+
+		switch doc.Objtype {
+		case model.TypeTopic:
+			url = fmt.Sprintf("%s/topics/%d", host, doc.Objid)
+		case model.TypeArticle:
+			url = fmt.Sprintf("%s/articles/%d", host, doc.Objid)
+		case model.TypeProject:
+			url = fmt.Sprintf("%s/p/%d", host, doc.Objid)
+		case model.TypeWiki:
+			url = fmt.Sprintf("%s/wiki/%d", host, doc.Objid)
+		case model.TypeBook:
+			url = fmt.Sprintf("%s/book/%d", host, doc.Objid)
+		}
+		respContentSlice[i] = fmt.Sprintf("《%s》 %s", doc.Title, url)
+	}
+
+	return strings.Join(respContentSlice, "\n"), nil
+}
diff --git a/src/model/wechat_msg.go b/src/model/wechat_msg.go
@@ -0,0 +1,54 @@
+// Copyright 2017 The StudyGolang Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+// http://studygolang.com
+// Author: polaris	polaris@studygolang.com
+
+package model
+
+const (
+	WeMsgTypeText       = "text"
+	WeMsgTypeImage      = "image"
+	WeMsgTypeVoice      = "voice"
+	WeMsgTypeVideo      = "video"
+	WeMsgTypeShortVideo = "shortvideo"
+	WeMsgTypeLocation   = "location"
+	WeMsgTypeLink       = "link"
+	WeMsgTypeEvent      = "event"
+
+	WeEventSubscribe   = "subscribe"
+	WeEventUnsubscribe = "unsubscribe"
+)
+
+type WechatMsg struct {
+	ToUserName   string
+	FromUserName string
+	CreateTime   int
+	MsgType      string
+	Content      string
+	MsgId        int64
+
+	// 图片消息
+	PicUrl  string
+	MediaId string
+
+	// 音频消息
+	Format string
+
+	// 视频或短视频
+	ThumbMediaId string
+
+	// 地理位置消息
+	Location_X float64
+	Location_Y float64
+	Scale      int
+	Label      string
+
+	// 链接消息
+	Title       string
+	Description string
+	Url         string
+
+	// 事件
+	Event string
+}

Original file line number	Diff line number	Diff line change
`@@ -34,5 +34,7 @@ func RegisterRoutes(g *echo.Group) {`
`34`	`34`	`new(OAuthController).RegisterRoute(g)`
`35`	`35`	`new(WebsocketController).RegisterRoute(g)`
`36`	`36`
	`37`	`+ new(WechatController).RegisterRoute(g)`
	`38`	`+`
`37`	`39`	`new(InstallController).RegisterRoute(g)`
`38`	`40`	`}`