From 41c0d7a0c287ca18e84aa22658e30a34dd3a44f7 Mon Sep 17 00:00:00 2001
From: shylinux <shylinux@icloud.com>
Date: Sun, 12 Jan 2020 18:40:47 +0800
Subject: [PATCH] add mp

---
 misc/mp/mp.go    | 101 +++++++++++++++++++++++++++++++++++++++++++++++
 miss/src/main.go |   2 +
 2 files changed, 103 insertions(+)
 create mode 100644 misc/mp/mp.go

diff --git a/misc/mp/mp.go b/misc/mp/mp.go
new file mode 100644
index 00000000..c6e74632
--- /dev/null
+++ b/misc/mp/mp.go
@@ -0,0 +1,101 @@
+package mp
+
+import (
+	"github.com/shylinux/icebergs"
+	"github.com/shylinux/icebergs/base/web"
+	"github.com/shylinux/icebergs/core/chat"
+	"github.com/shylinux/toolkits"
+	"time"
+)
+
+var Index = &ice.Context{Name: "mp", Help: "小程序",
+	Caches: map[string]*ice.Cache{},
+	Configs: map[string]*ice.Config{
+		"login": {Name: "login", Help: "认证", Value: kit.Data(
+			"auth", "/sns/jscode2session?grant_type=authorization_code",
+			"weixin", "https://api.weixin.qq.com",
+			"appid", "", "appmm", "", "token", "",
+			"userrole", kit.Dict(),
+		)},
+	},
+	Commands: map[string]*ice.Command{
+		ice.ICE_INIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
+			m.Cmd(ice.CTX_CONFIG, "load", "mp.json")
+			m.Confm("login", "meta.userrole", func(key string, value string) {
+				m.Cmd(ice.AAA_ROLE, value, key)
+			})
+			m.Cmd(ice.WEB_SPIDE, "add", "weixin", m.Conf("login", "meta.weixin"))
+		}},
+		ice.ICE_EXIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
+			m.Cmd(ice.CTX_CONFIG, "save", "mp.json", "web.chat.mp.login")
+		}},
+		"/login/": {Name: "/login/", Help: "登录", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
+			switch arg[0] {
+			case "code":
+				msg := m.Cmd(ice.WEB_SPIDE, "weixin", "GET", m.Conf("login", "meta.auth"),
+					"js_code", m.Option("code"), "appid", m.Conf("login", "meta.appid"), "secret", m.Conf("login", "meta.appmm"))
+
+				if m.Richs(ice.AAA_USER, nil, msg.Append("openid"), nil) == nil {
+					// 创建用户
+					m.Rich(ice.AAA_USER, nil, kit.Dict(
+						"username", msg.Append("openid"),
+						"expires_in", time.Now().Unix()+kit.Int64(msg.Append("expires_in")),
+						"session_key", msg.Append("session_key"),
+						"usernode", m.Conf(ice.CLI_RUNTIME, "boot.hostname"),
+					))
+					m.Event(ice.USER_CREATE, msg.Append("openid"))
+				}
+
+				if m.Options(ice.MSG_SESSID) && m.Cmdx(ice.AAA_SESS, "check", m.Option(ice.MSG_SESSID)) == msg.Append("openid") {
+					// 复用会话
+					m.Echo(m.Option(ice.MSG_SESSID))
+				} else {
+					// 创建会话
+					role := m.Conf("login", kit.Keys("meta.userrole", msg.Append("openid")))
+					sessid := m.Rich(ice.AAA_SESS, nil, kit.Dict(
+						"username", msg.Append("openid"), "userrole", role,
+					))
+					m.Info("user: %s role: %s sess: %s", msg.Append("openid"), role, sessid)
+					m.Echo(msg.Option(ice.MSG_SESSID, sessid))
+				}
+
+			case "info":
+				m.Richs(ice.AAA_SESS, nil, m.Option(ice.MSG_SESSID), func(key string, value map[string]interface{}) {
+					m.Richs(ice.AAA_USER, nil, value["username"], func(key string, value map[string]interface{}) {
+						// 注册用户
+						value["gender"] = m.Option("gender")
+						value["avatar"] = m.Option("avatarUrl")
+						value["nickname"] = m.Option("nickName")
+						value["language"] = m.Option("language")
+						value["province"] = m.Option("province")
+						value["country"] = m.Option("country")
+						value["city"] = m.Option("city")
+					})
+				})
+
+			case "scan":
+				m.Echo(m.Option("scan")).Push("_output", "qrcode")
+
+			case "auth":
+				m.Log("fuck", "what %v", m.Option(ice.MSG_USERNAME))
+				m.Log("fuck", "what %v", m.Option(ice.MSG_SESSID))
+				if !m.Options(ice.MSG_USERNAME) || !m.Options(ice.MSG_SESSID) {
+					m.Echo("401").Push("_output", "status")
+					break
+				}
+				m.Cmd(ice.WEB_SPACE, "auth", m.Option("auth"), m.Option(ice.MSG_USERNAME), m.Option(ice.MSG_USERROLE))
+
+			case "cmds":
+				if arg = kit.Split(arg[1]); m.Right(arg) {
+					m.Hand = false
+					if m.Cmdy(arg); !m.Hand {
+						m.Set("result")
+						m.Cmdy(ice.CLI_SYSTEM, arg)
+					}
+				}
+			}
+		}},
+	},
+}
+
+func init() { chat.Index.Register(Index, &web.Frame{}) }
diff --git a/miss/src/main.go b/miss/src/main.go
index b9de0a2d..c6aeb888 100644
--- a/miss/src/main.go
+++ b/miss/src/main.go
@@ -5,6 +5,8 @@ import (
 	_ "github.com/shylinux/icebergs/base"
 	_ "github.com/shylinux/icebergs/core"
 	_ "github.com/shylinux/icebergs/misc"
+
+	_ "github.com/shylinux/icebergs/misc/mp"
 )
 
 func main() {