forked from x/ContextOS
359 lines
12 KiB
Go
359 lines
12 KiB
Go
package chat
|
|
|
|
import (
|
|
"contexts/ctx"
|
|
"contexts/web"
|
|
"examples/chat"
|
|
"toolkit"
|
|
|
|
"encoding/json"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func get(m *ctx.Message, arg ...string) *ctx.Message {
|
|
m.Option("temp_expire", -1)
|
|
m.Option("format", "object")
|
|
m.Cmdy("web.get", "feishu", arg, "temp", "data")
|
|
return m
|
|
}
|
|
func post(m *ctx.Message, arg ...string) *ctx.Message {
|
|
m.Option("temp_expire", -1)
|
|
m.Option("format", "object")
|
|
m.Cmdy("web.get", "method", "POST", "feishu", arg,
|
|
"content_type", "application/json",
|
|
"temp", "data",
|
|
)
|
|
return m
|
|
}
|
|
func parse(m *ctx.Message) {
|
|
data := m.Optionv("content_data")
|
|
if data == nil {
|
|
json.NewDecoder(m.Optionv("request").(*http.Request).Body).Decode(&data)
|
|
m.Optionv("content_data", data)
|
|
|
|
switch d := data.(type) {
|
|
case map[string]interface{}:
|
|
for k, v := range d {
|
|
switch d := v.(type) {
|
|
case map[string]interface{}:
|
|
for k, v := range d {
|
|
m.Add("option", k, v)
|
|
}
|
|
default:
|
|
for _, v := range kit.Trans(v) {
|
|
m.Add("option", "msg."+k, v)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if kit.Map(kit.Chain(data, "action.value"), "", func(key string, value string) {
|
|
m.Add("option", key, value)
|
|
}) != nil {
|
|
m.Option("msg.type", "event_click")
|
|
}
|
|
m.Log("info", "msg: %v", kit.Formats(data))
|
|
}
|
|
|
|
var Index = &ctx.Context{Name: "feishu", Help: "飞书",
|
|
Caches: map[string]*ctx.Cache{},
|
|
Configs: map[string]*ctx.Config{
|
|
"app": &ctx.Config{Name: "app", Value: map[string]interface{}{}, Help: "服务配置"},
|
|
"user": &ctx.Config{Name: "user", Value: map[string]interface{}{}, Help: "服务配置"},
|
|
},
|
|
Commands: map[string]*ctx.Command{
|
|
"app": &ctx.Command{Name: "app login|token bot", Help: "应用", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) (e error) {
|
|
if len(arg) == 0 {
|
|
m.Confm("app", func(key string, value map[string]interface{}) {
|
|
m.Push("key", key)
|
|
m.Push("id", value["id"])
|
|
})
|
|
m.Table()
|
|
return
|
|
}
|
|
|
|
switch arg[0] {
|
|
case "login":
|
|
m.Confv("app", arg[1], map[string]interface{}{"id": arg[2], "mm": arg[3]})
|
|
case "token":
|
|
if now := time.Now().Unix(); !m.Confs("app", []string{arg[1], "token"}) || int64(m.Confi("app", []string{arg[1], "expire"})) < now {
|
|
post(m, "auth/v3/tenant_access_token/internal/", "app_id", m.Conf("app", []string{arg[1], "id"}),
|
|
"app_secret", m.Conf("app", []string{arg[1], "mm"}))
|
|
m.Conf("app", []string{arg[1], "token"}, m.Append("tenant_access_token"))
|
|
m.Conf("app", []string{arg[1], "expire"}, int64(m.Appendi("expire"))+now)
|
|
m.Set("result")
|
|
}
|
|
m.Echo(m.Conf("app", []string{arg[1], "token"}))
|
|
}
|
|
return
|
|
}},
|
|
"ship": &ctx.Command{Name: "ship", Help: "组织", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) (e error) {
|
|
data := kit.UnMarshal(m.Cmdx("web.get", "feishu", "contact/v1/scope/get/",
|
|
"headers", "Authorization", "Bearer "+m.Cmdx(".app", "token", "bot"),
|
|
))
|
|
kit.Map(kit.Chain(data, "data.authed_departments"), "", func(index int, value string) {
|
|
m.Push("type", "ship")
|
|
m.Push("value", value)
|
|
})
|
|
kit.Map(kit.Chain(data, "data.authed_open_ids"), "", func(index int, value string) {
|
|
m.Push("type", "user")
|
|
m.Push("value", value)
|
|
})
|
|
m.Table()
|
|
return
|
|
}},
|
|
"group": &ctx.Command{Name: "group", Help: "群组", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) (e error) {
|
|
if len(arg) == 0 {
|
|
kit.Map(kit.Chain(kit.UnMarshal(m.Cmdx("web.get", "feishu", "chat/v4/list", "headers", "Authorization", "Bearer "+m.Cmdx(".app", "token", "bot"))),
|
|
"data.groups"), "", func(index int, value map[string]interface{}) {
|
|
m.Push("key", value["chat_id"])
|
|
m.Push("name", value["name"])
|
|
m.Push("avatar", value["avatar"])
|
|
})
|
|
m.Table()
|
|
}
|
|
kit.Map(kit.Chain(kit.UnMarshal(m.Cmdx("web.get", "feishu", "chat/v4/info", "headers", "Authorization", "Bearer "+m.Cmdx(".app", "token", "bot"))),
|
|
"data.members"), "", func(index int, value map[string]interface{}) {
|
|
m.Push("key", value["open_id"])
|
|
})
|
|
m.Table()
|
|
return
|
|
}},
|
|
"user": &ctx.Command{Name: "user code|email|mobile", Help: "用户", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) (e error) {
|
|
switch arg[0] {
|
|
case "code":
|
|
post(m, "/connect/qrconnect/oauth2/access_token/",
|
|
"app_secret", m.Conf("app", []string{"bot", "mm"}), "app_id", m.Conf("app", []string{"bot", "id"}),
|
|
"grant_type", "authorization_code", "code", arg[1],
|
|
)
|
|
msg := get(m.Spawn(), "/connect/qrconnect/oauth2/user_info/",
|
|
"headers", "Authorization", "Bearer "+m.Append("access_token"),
|
|
)
|
|
m.Confv("user", m.Append("open_id"), map[string]interface{}{
|
|
"name": m.Append("name"),
|
|
"en_name": m.Append("en_name"),
|
|
"avatar_url": m.Append("avatar_url"),
|
|
"access_token": m.Append("access_token"),
|
|
"token_type": m.Append("token_type"),
|
|
"expire": int64(m.Appendi("expire")) + time.Now().Unix(),
|
|
"refresh_token": m.Append("refresh_token"),
|
|
"tenant_key": m.Append("tenant_key"),
|
|
|
|
"email": msg.Append("Email"),
|
|
"mobile": msg.Append("Mobile"),
|
|
"status": msg.Append("status"),
|
|
"employee": msg.Append("EmployeeID"),
|
|
})
|
|
|
|
default:
|
|
us := []string{}
|
|
ps := []string{}
|
|
for i := 0; i < len(arg); i++ {
|
|
us = append(us, kit.Select("mobiles", "emails", strings.Contains(arg[i], "@")), arg[i])
|
|
ps = append(ps, kit.Select("mobile", "email", strings.Contains(arg[i], "@"))+"_users")
|
|
}
|
|
|
|
data := kit.UnMarshal(m.Cmdx("web.get", "feishu", "user/v1/batch_get_id", us,
|
|
"headers", "Authorization", "Bearer "+m.Cmdx(".app", "token", "bot")))
|
|
|
|
for i, v := range ps {
|
|
m.Append(arg[i], kit.Chain(data, []string{"data", v, arg[i], "0", "open_id"}))
|
|
}
|
|
m.Table()
|
|
}
|
|
return
|
|
}},
|
|
"send": &ctx.Command{Name: "send [chat_id|open_id|user_id|email] who [menu] [title] text", Help: "消息", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) (e error) {
|
|
var form = map[string]interface{}{"content": map[string]interface{}{}}
|
|
switch arg[0] {
|
|
case "chat_id", "open_id", "user_id", "email":
|
|
form[arg[0]], arg = arg[1], arg[2:]
|
|
default:
|
|
form["chat_id"], arg = arg[0], arg[1:]
|
|
}
|
|
|
|
switch arg[0] {
|
|
case "menu":
|
|
elements := []interface{}{}
|
|
elements = append(elements, map[string]interface{}{
|
|
"tag": "div",
|
|
"text": map[string]interface{}{
|
|
"tag": "plain_text",
|
|
"content": arg[2],
|
|
},
|
|
})
|
|
|
|
actions := []interface{}{}
|
|
for i := 3; i < len(arg); i++ {
|
|
button := map[string]interface{}{
|
|
"tag": "button", "text": map[string]interface{}{
|
|
"tag": "plain_text", "content": arg[i],
|
|
},
|
|
"type": "default",
|
|
}
|
|
|
|
switch arg[i+1] {
|
|
case "confirm":
|
|
button[arg[i+1]], i = map[string]interface{}{
|
|
"title": map[string]interface{}{"tag": "lark_md", "content": arg[i+2]},
|
|
"text": map[string]interface{}{"tag": "lark_md", "content": arg[i+3]},
|
|
}, i+3
|
|
case "value":
|
|
button[arg[i+1]], i = map[string]interface{}{
|
|
arg[i+2]: arg[i+3],
|
|
}, i+3
|
|
case "url":
|
|
button[arg[i+1]], i = arg[i+2], i+2
|
|
default:
|
|
button["value"], i = map[string]interface{}{
|
|
arg[i+1]: arg[i+2],
|
|
}, i+2
|
|
}
|
|
|
|
actions = append(actions, button)
|
|
}
|
|
elements = append(elements, map[string]interface{}{
|
|
"tag": "action", "actions": actions,
|
|
})
|
|
|
|
kit.Chain(form, "msg_type", "interactive")
|
|
kit.Chain(form, "card", map[string]interface{}{
|
|
"config": map[string]interface{}{
|
|
"wide_screen_mode": true,
|
|
// "title": map[string]interface{}{
|
|
// "tag": "lark_md", "content": arg[1],
|
|
// },
|
|
},
|
|
"header": map[string]interface{}{
|
|
"title": map[string]interface{}{
|
|
"tag": "lark_md", "content": arg[1],
|
|
},
|
|
},
|
|
"elements": elements,
|
|
})
|
|
default:
|
|
switch len(arg) {
|
|
case 0:
|
|
case 1:
|
|
kit.Chain(form, "msg_type", "text")
|
|
kit.Chain(form, "content.text", arg[0])
|
|
default:
|
|
content := []interface{}{}
|
|
line := []interface{}{}
|
|
for _, v := range arg[1:] {
|
|
if v == "\n" {
|
|
content, line = append(content, line), []interface{}{}
|
|
continue
|
|
}
|
|
line = append(line, map[string]interface{}{
|
|
"tag": "text", "text": v,
|
|
})
|
|
}
|
|
content = append(content, line)
|
|
|
|
kit.Chain(form, "msg_type", "post")
|
|
kit.Chain(form, "content.post", map[string]interface{}{
|
|
"zh_cn": map[string]interface{}{
|
|
"title": arg[0],
|
|
"content": content,
|
|
},
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
m.Cmdy("web.get", "method", "POST", "feishu", "message/v4/send/",
|
|
"headers", "Authorization", "Bearer "+m.Cmdx(".app", "token", "bot"),
|
|
"content_data", kit.Formats(form), "content_type", "application/json",
|
|
"temp", "data", "data.message_id",
|
|
)
|
|
return
|
|
}},
|
|
"/msg": &ctx.Command{Name: "/msg", Help: "消息", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) (e error) {
|
|
parse(m)
|
|
|
|
switch m.Option("msg.type") {
|
|
case "url_verification":
|
|
m.Echo(kit.Format(map[string]interface{}{"challenge": m.Option("challenge")}))
|
|
|
|
case "event_callback":
|
|
switch m.Option("type") {
|
|
case "chat_disband":
|
|
case "p2p_chat_create":
|
|
if m.Options("open_chat_id") {
|
|
m.Cmdy(".send", m.Option("open_chat_id"), "我们做朋友吧~")
|
|
}
|
|
case "add_bot":
|
|
if m.Options("open_chat_id") {
|
|
m.Cmdy(".send", m.Option("open_chat_id"), "我来也~")
|
|
}
|
|
default:
|
|
if m.Options("open_chat_id") {
|
|
m.Option("username", m.Option("open_id"))
|
|
m.Option("sessid", m.Cmdx("aaa.user", "session", "select"))
|
|
m.Cmd("ssh._check", "work", "create", m.Option("username"))
|
|
m.Goshy([]string{m.Option("text_without_at_bot")}, 0, nil, func(msg *ctx.Message) {
|
|
m.Cmdy(".send", m.Option("open_chat_id"), kit.Select("你好", strings.Join(msg.Meta["result"], "")))
|
|
})
|
|
}
|
|
}
|
|
case "event_click":
|
|
m.Echo(kit.Format(map[string]interface{}{
|
|
"header": map[string]interface{}{
|
|
"title": map[string]interface{}{
|
|
"tag": "lark_md", "content": "haha",
|
|
},
|
|
},
|
|
"elements": []interface{}{
|
|
map[string]interface{}{
|
|
"tag": "action",
|
|
"actions": []interface{}{
|
|
map[string]interface{}{
|
|
"tag": "button",
|
|
"type": "default",
|
|
"text": map[string]interface{}{
|
|
"tag": "plain_text",
|
|
"content": m.Time(),
|
|
},
|
|
"value": map[string]interface{}{
|
|
"hi": "hello",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}))
|
|
}
|
|
return
|
|
}},
|
|
"/sso": &ctx.Command{Name: "/sso", Help: "消息", Hand: func(m *ctx.Message, c *ctx.Context, key string, arg ...string) (e error) {
|
|
if m.Options("code") {
|
|
m.Option("username", m.Cmd(".user", "code", m.Option("code")).Append("open_id"))
|
|
m.Option("sessid", m.Cmdx("aaa.user", "session", "select"))
|
|
m.Cmd("ssh._check", "work", "create", m.Option("username"))
|
|
|
|
w := m.Optionv("response").(http.ResponseWriter)
|
|
r := m.Optionv("request").(*http.Request)
|
|
web.Cookie(m, w, r)
|
|
m.Append("redirect", m.Cmdx("web.spide", "serve", "merge", m.Option("index_path")), "code", "")
|
|
return
|
|
}
|
|
|
|
if !m.Options("sessid") || !m.Options("username") {
|
|
m.Append("redirect", m.Cmdx("web.spide", "feishu", "merge", "/connect/qrconnect/page/sso/",
|
|
"redirect_uri", m.Cmdx("web.spide", "serve", "merge", m.Option("index_path")),
|
|
"app_id", m.Conf("app", "bot.id"), "state", "ok"))
|
|
return
|
|
}
|
|
m.Cmd("/render")
|
|
return
|
|
}},
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
chat.Index.Register(Index, &web.WEB{Context: Index})
|
|
}
|