liubin/icebergs
1
0
Fork 0
forked from x/icebergs
Code Pull Requests Activity

add disk

Browse Source
This commit is contained in:
IT 老营长 @云轩领航-创始人 2023-11-09 09:10:00 +08:00
parent 338b56fcf8
commit 798dfd62aa
4 changed files with 122 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
core/chat/oauth/client.go
View File

@ -21,12 +21,15 @@ const (
GRANT_URL = "grant_url"
TOKEN_URL = "token_url"
USERS_URL = "users_url"
USER_KEY = "user_key"
NICK_KEY = "nick_key"
REDIRECT_URI = "redirect_uri"
RESPONSE_TYPE = "response_type"
AUTHORIZATION_CODE = "authorization_code"
GRANT_TYPE = "grant_type"
STATE = "state"
SCOPE = "scope"
CODE = "code"
API_PREFIX = "api_prefix"
@ -38,7 +41,7 @@ const (
type Client struct {
ice.Hash
short string `data:"domain,client_id"`
field string `data:"time,hash,domain,client_id,client_secret,oauth_url,grant_url,token_url,users_url,api_prefix,token_prefix"`
field string `data:"time,hash,domain,client_id,client_secret,oauth_url,grant_url,token_url,users_url,scope,user_key,nick_key,api_prefix,token_prefix"`
sso string `name:"sso name* icon*@icon" help:"登录"`
auth string `name:"auth" help:"授权"`
user string `name:"user" help:"用户"`
@ -83,17 +86,20 @@ func (s Client) Inputs(m *ice.Message, arg ...string) {
}
}
func (s Client) Sso(m *ice.Message, arg ...string) {
m.Cmd(web.CHAT_HEADER, mdb.CREATE, m.Option(mdb.NAME), m.Option(mdb.ICON), kit.MergeURL2(m.Option(web.DOMAIN), m.Option(OAUTH_URL), m.OptionSimple(CLIENT_ID), REDIRECT_URI, s.RedirectURI(m), RESPONSE_TYPE, CODE, STATE, m.Option(mdb.HASH)))
m.Cmd(web.CHAT_HEADER, mdb.CREATE, m.Option(mdb.NAME), m.Option(mdb.ICON), s.OAuthURL(m))
}
func (s Client) Auth(m *ice.Message, arg ...string) {
m.Options(REDIRECT_URI, s.RedirectURI(m), RESPONSE_TYPE, CODE, STATE, m.Option(mdb.HASH))
m.ProcessOpen(kit.MergeURL2(m.Option(web.DOMAIN), m.Option(OAUTH_URL), m.OptionSimple(CLIENT_ID, REDIRECT_URI, RESPONSE_TYPE, STATE)))
m.ProcessOpen(s.OAuthURL(m))
}
func (s Client) User(m *ice.Message, arg ...string) {
if res := s.Get(m, m.Option(mdb.HASH), m.Option(USERS_URL)); res != nil {
m.Options(res).Cmd(aaa.USER, mdb.CREATE, aaa.USERNICK, m.Option("full_name"), m.OptionSimple(aaa.USERNAME), aaa.AVATAR, m.Option(aaa.AVATAR_URL),
aaa.USERROLE, kit.Select(aaa.VOID, aaa.TECH, m.Option("is_admin") == ice.TRUE), aaa.USERZONE, m.Option(web.DOMAIN),
m.OptionSimple(aaa.EMAIL, aaa.LANGUAGE, aaa.AVATAR_URL))
if res := s.Get(m, m.Option(mdb.HASH), m.Option(USERS_URL), arg...); res != nil {
m.Options(res).Cmd(aaa.USER, mdb.CREATE,
aaa.USERNICK, m.Option(kit.Select("full_name", m.Option(NICK_KEY))),
aaa.USERNAME, m.Option(aaa.USERNAME, m.Option(kit.Select(aaa.USERNAME, m.Option(USER_KEY)))),
aaa.AVATAR, m.Option(aaa.AVATAR_URL),
aaa.USERROLE, kit.Select(aaa.VOID, aaa.TECH, m.Option("is_admin") == ice.TRUE),
aaa.USERZONE, m.Option(web.DOMAIN),
m.OptionSimple(aaa.EMAIL, aaa.LANGUAGE))
}
}
func (s Client) Orgs(m *ice.Message, arg ...string) {}
@ -104,18 +110,15 @@ func (s Client) List(m *ice.Message, arg ...string) {
func init() { ice.ChatCtxCmd(Client{}) }
func (s Client) RedirectURI(m *ice.Message) string {
return strings.Split(web.MergeURL2(m.Message, web.ChatCmdPath(m.Message, m.PrefixKey(), ctx.ACTION, aaa.LOGIN)), web.QS)[0]
}
func (s Client) Login(m *ice.Message, arg ...string) {
if state, code := m.Option(STATE), m.Option(CODE); !m.Warn(state == "" || code == "") {
s.Hash.List(m.Spawn(), m.Option(mdb.HASH, state)).Table(func(value ice.Maps) { m.Options(value) })
m.Options(REDIRECT_URI, s.RedirectURI(m), GRANT_TYPE, AUTHORIZATION_CODE)
if res := s.Post(m, m.Option(mdb.HASH), m.Option(GRANT_URL), m.OptionSimple(CLIENT_ID, CLIENT_SECRET, REDIRECT_URI, CODE, GRANT_TYPE)...); !m.Warn(res == nil) {
kit.Value(res, EXPIRES_IN, m.Time(kit.Format("%vs", kit.Value(res, EXPIRES_IN))))
s.Modify(m, kit.Simple(res)...)
m.Options(GRANT_TYPE, AUTHORIZATION_CODE, REDIRECT_URI, s.RedirectURI(m))
if res := s.Post(m, m.Option(mdb.HASH), m.Option(GRANT_URL), m.OptionSimple(GRANT_TYPE, CODE, CLIENT_ID, CLIENT_SECRET, REDIRECT_URI)...); !m.Warn(res == nil) {
kit.Value(res, EXPIRES_IN, m.Time(kit.Format("%vs", kit.Int(kit.Value(res, EXPIRES_IN)))))
m.Options(res)
if s.User(m); !m.Warn(m.Option(aaa.USERNAME) == "") && m.R != nil {
m.Cmd(aaa.USER, mdb.MODIFY, m.OptionSimple(aaa.USERNAME), kit.Simple(res))
web.RenderCookie(m.Message, aaa.SessCreate(m.Message, m.Option(aaa.USERNAME)))
m.ProcessHistory()
} else {
@ -124,6 +127,29 @@ func (s Client) Login(m *ice.Message, arg ...string) {
}
}
}
func (s Client) Login2(m *ice.Message, arg ...string) {
if state, code := m.Option(STATE), m.Option(CODE); !m.Warn(state == "" || code == "") {
s.Hash.List(m.Spawn(), m.Option(mdb.HASH, state)).Table(func(value ice.Maps) { m.Options(value) })
m.Options(GRANT_TYPE, AUTHORIZATION_CODE, REDIRECT_URI, s.RedirectURI(m))
if res := s.Get(m, m.Option(mdb.HASH), m.Option(GRANT_URL), m.OptionSimple(GRANT_TYPE, CODE, CLIENT_ID, CLIENT_SECRET, REDIRECT_URI)...); !m.Warn(res == nil) {
kit.Value(res, EXPIRES_IN, m.Time(kit.Format("%vs", kit.Int(kit.Value(res, EXPIRES_IN)))))
m.Options(res)
if s.User(m); !m.Warn(m.Option(aaa.USERNAME) == "") && m.R != nil {
m.Cmd(aaa.USER, mdb.MODIFY, m.OptionSimple(aaa.USERNAME), kit.Simple(res))
web.RenderCookie(m.Message, aaa.SessCreate(m.Message, m.Option(aaa.USERNAME)))
m.ProcessHistory()
} else {
m.ProcessClose()
}
}
}
}
func (s Client) OAuthURL(m *ice.Message) string {
return kit.MergeURL2(m.Option(web.DOMAIN), m.Option(OAUTH_URL), RESPONSE_TYPE, CODE, m.OptionSimple(CLIENT_ID), REDIRECT_URI, s.RedirectURI(m), m.OptionSimple(SCOPE), STATE, m.Option(mdb.HASH))
}
func (s Client) RedirectURI(m *ice.Message) string {
return strings.Split(web.MergeURL2(m.Message, web.ChatCmdPath(m.Message, m.PrefixKey(), ctx.ACTION, aaa.LOGIN)), web.QS)[0]
}
func (s Client) Get(m *ice.Message, hash, api string, arg ...string) ice.Any {
return web.SpideGet(m.Message, s.request(m, hash, api, arg...))
@ -137,10 +163,22 @@ func (s Client) Post(m *ice.Message, hash, api string, arg ...string) ice.Any {
func (s Client) Delete(m *ice.Message, hash, api string, arg ...string) ice.Any {
return web.SpideDelete(m.Message, s.request(m, hash, api, arg...))
}
func (s Client) Save(m *ice.Message, hash, file, api string, arg ...string) ice.Any {
args := s.request(m, hash, api, arg...)
web.Toast(m.Message, "process", file, "-1", 0)
return web.SpideSave(m.Message, file, kit.MergeURL(args[0], args[1:]), func(count int, total int, value int) {
web.Toast(m.Message, kit.Format("%s/%s", kit.FmtSize(count), kit.FmtSize(total)), file, "-1", value)
})
}
func (s Client) request(m *ice.Message, hash, api string, arg ...string) []string {
msg := s.Hash.List(m.Spawn(), hash)
kit.If(msg.Append(ACCESS_TOKEN), func(p string) {
kit.If(m.Option(ACCESS_TOKEN) == "" && m.Option(ice.MSG_USERNAME) != "", func() {
msg := m.Cmd(aaa.USER, m.Option(ice.MSG_USERNAME))
m.Option(ACCESS_TOKEN, msg.Append(ACCESS_TOKEN))
})
kit.If(m.Option(ACCESS_TOKEN), func(p string) {
m.Options(web.SPIDE_HEADER, ice.Maps{web.Authorization: msg.Append(TOKEN_PREFIX) + lex.SP + p})
arg = append(arg, ACCESS_TOKEN, p)
})
kit.If(api == "", func() { api = path.Join(msg.Append(API_PREFIX), m.ActionKey()) })
return kit.Simple(kit.MergeURL2(msg.Append(web.DOMAIN), api), arg)

62
misc/disk/client.go Normal file
View File

@ -0,0 +1,62 @@
package disk
import (
"time"
"shylinux.com/x/ice"
"shylinux.com/x/icebergs/base/mdb"
"shylinux.com/x/icebergs/base/nfs"
"shylinux.com/x/icebergs/base/web"
"shylinux.com/x/icebergs/core/chat/oauth"
kit "shylinux.com/x/toolkits"
)
type Client struct {
oauth.Client
list string `name:"list hash path orgs:text repo:text auto" help:"仓库" icon:"gitea.png"`
}
func init() {
oauth.Inputs["baidu"] = map[string]string{
oauth.OAUTH_URL: "/oauth/2.0/authorize",
oauth.GRANT_URL: "/oauth/2.0/token",
oauth.TOKEN_URL: "/oauth/2.0/token",
oauth.USERS_URL: "https://pan.baidu.com/rest/2.0/xpan/nas?method=uinfo",
oauth.NICK_KEY: "baidu_name",
oauth.USER_KEY: "uk",
oauth.SCOPE: "basic,netdisk",
oauth.API_PREFIX: "/api/v1/",
oauth.TOKEN_PREFIX: "",
}
}
func (s Client) Login(m *ice.Message, arg ...string) {
s.Client.Login2(m, arg...)
}
func (s Client) Show(m *ice.Message, arg ...string) {
res := s.Client.Get(m, m.Option(mdb.HASH), "https://pan.baidu.com/rest/2.0/xpan/multimedia?method=filemetas", "dlink", "1", "fsids", kit.Format("%v", []string{m.Option(mdb.ID)}))
// 1 视频、2 音频、3 图片、4 文档、5 应用、6 其他、7 种子
p := "usr/local/disk/" + m.Option(mdb.ID)
s.Save(m, m.Option(mdb.HASH), p, kit.Format(kit.Value(res, "list.0.dlink")))
web.ToastSuccess(m.Message)
switch kit.Format(kit.Value(res, "list.0.category")) {
case "4":
m.Cmdy(nfs.CAT, p)
}
}
func (s Client) List(m *ice.Message, arg ...string) {
if len(arg) == 0 {
s.Client.List(m, arg...)
return
}
res := s.Client.Get(m, arg[0], "https://pan.baidu.com/rest/2.0/xpan/file?method=list", "dir", kit.Select("", arg, 1))
kit.For(kit.Value(res, mdb.LIST), func(value ice.Map) {
m.Push(mdb.TIME, time.Unix(kit.Int64(value["server_mtime"]), 0))
m.Push(nfs.PATH, kit.Format(value[nfs.PATH])+kit.Select("", nfs.PS, kit.Format(value["isdir"]) == "1"))
m.Push(nfs.SIZE, kit.FmtSize(kit.Int(value[nfs.SIZE])))
m.Push(mdb.ID, value["fs_id"])
})
m.PushAction("show")
m.Echo(kit.Formats(res))
}
func init() { ice.WikiCtxCmd(Client{}) }

5
misc/disk/client.shy Normal file
View File

@ -0,0 +1,5 @@
section "client"
refer `
后台 https://pan.baidu.com/union/console/applist
文档 https://pan.baidu.com/union/doc/al0rwqzzl
`

1
misc/disk/disk.go Normal file
View File

@ -0,0 +1 @@
package disk