关键词回复写库

polaris1119 · polaris1119 · commit 97cc7ba882bd · 2020-07-13T23:30:34.000+08:00
diff --git a/config/db.sql b/config/db.sql
@@ -827,3 +827,16 @@ CREATE TABLE `wechat_user` (
   KEY `uid` (`uid`),
   KEY `updated_at` (`updated_at`)
 ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='微信用户绑定表';
+
+CREATE TABLE `wechat_auto_reply` (
+  `id` int(11) unsigned NOT NULL AUTO_INCREMENT COMMENT '自增id',
+  `typ` tinyint unsigned NOT NULL DEFAULT 0 COMMENT '回复类型：0-关键词回复；1-收到消息未找到回复；2-被关注回复',
+  `word` varchar(15) NOT NULL DEFAULT '' COMMENT '关键词', 
+  `msg_type` varchar(15) NOT NULL DEFAULT '' COMMENT '回复消息类型，和微信对应',
+  `content` varchar(255) NOT NULL DEFAULT '' COMMENT '要回复的内容',
+  `created_at` timestamp NOT NULL DEFAULT '2020-07-13 14:38:09',
+  `updated_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+  PRIMARY KEY (`id`),
+  KEY `word` (`word`),
+  KEY `updated_at` (`updated_at`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='微信自动回复';
diff --git a/http/http.go b/http/http.go
@@ -260,6 +260,10 @@ func Render(ctx echo.Context, contentTpl string, data map[string]interface{}) er
 		return err
 	}
 
+	if strings.Contains(ctx.Request().UserAgent(), "miniProgram") {
+		data["min_program"] = true
+	}
+
 	data["pos_ad"] = logic.DefaultAd.FindAll(context.EchoContext(ctx), ctx.Path())
 	data["cur_time"] = times.Format("Y-m-d H:i:s")
 	data["path"] = ctx.Path()
diff --git a/logic/wechat.go b/logic/wechat.go
@@ -176,6 +176,7 @@ func (self WechatLogic) AutoReply(ctx context.Context, reqData []byte) (*model.W
 		} else if strings.Contains(wechatMsg.Content, "图书") || strings.Contains(wechatMsg.Content, "book") {
 			return self.bookContent(ctx, wechatMsg)
 		} else {
+			// 用户获取验证码用
 			user := DefaultUser.FindOne(ctx, "username", wechatMsg.Content)
 			if user.Uid > 0 {
 				var content string
@@ -188,13 +189,28 @@ func (self WechatLogic) AutoReply(ctx context.Context, reqData []byte) (*model.W
 				return self.wechatResponse(ctx, content, wechatMsg)
 			}
 
+			// 关键词回复
+			autoReply := &model.WechatAutoReply{}
+			MasterDB.Where("word LIKE ?", "%"+wechatMsg.Content+"%").Get(autoReply)
+			if autoReply.Id != 0 {
+				wechatMsg.MsgType = autoReply.MsgType
+				return self.wechatResponse(ctx, autoReply.Content, wechatMsg)
+			}
+
 			return self.searchContent(ctx, wechatMsg)
 		}
 	case model.WeMsgTypeEvent:
 		switch wechatMsg.Event {
 		case model.WeEventSubscribe:
 			wechatMsg.MsgType = model.WeMsgTypeText
 			welcomeText := strings.ReplaceAll(config.ConfigFile.MustValue("wechat", "subscribe"), "\\n", "\n")
+
+			autoReply := &model.WechatAutoReply{}
+			_, err = MasterDB.Where("typ=?", model.AutoReplyTypSubscribe).Get(autoReply)
+			if err == nil {
+				welcomeText = autoReply.Content
+			}
+
 			return self.wechatResponse(ctx, welcomeText, wechatMsg)
 		}
 	}
diff --git a/model/wechat.go b/model/wechat.go
@@ -22,3 +22,20 @@ type WechatUser struct {
 	CreatedAt  time.Time
 	UpdatedAt  time.Time `xorm:"<-"`
 }
+
+const (
+	AutoReplyTypWord      = iota // 关键词回复
+	AutoReplyTypNotFound         // 收到消息（未命中关键词且未搜索到）
+	AutoReplyTypSubscribe        // 被关注回复
+)
+
+// WechatAutoReply 微信自动回复
+type WechatAutoReply struct {
+	Id        int `xorm:"pk autoincr"`
+	Typ       uint8
+	Word      string
+	MsgType   string
+	Content   string
+	CreatedAt time.Time
+	UpdatedAt time.Time `xorm:"<-"`
+}
diff --git a/template/common/layout.html b/template/common/layout.html
@@ -249,7 +249,7 @@
 
 	<script type="text/javascript" src="{{.static_domain}}/static/dist/js/sidebar.min.js"></script>
 
-	{{if .is_pro}}
+	{{if and .is_pro (not .min_program)}}
 		<!-- 统计分析、广告脚本等 -->
 		{{include "common/analytics.html" .}}
 	{{end}}
