package oauth import ( "path" "strings" "shylinux.com/x/ice" "shylinux.com/x/icebergs/base/aaa" "shylinux.com/x/icebergs/base/ctx" "shylinux.com/x/icebergs/base/lex" "shylinux.com/x/icebergs/base/mdb" "shylinux.com/x/icebergs/base/web" kit "shylinux.com/x/toolkits" ) const ( CLIENT_ID = "client_id" CLIENT_SECRET = "client_secret" OAUTH_URL = "oauth_url" GRANT_URL = "grant_url" TOKEN_URL = "token_url" USERS_URL = "users_url" REDIRECT_URI = "redirect_uri" RESPONSE_TYPE = "response_type" AUTHORIZATION_CODE = "authorization_code" GRANT_TYPE = "grant_type" STATE = "state" CODE = "code" API_PREFIX = "api_prefix" TOKEN_PREFIX = "token_prefix" ACCESS_TOKEN = "access_token" EXPIRES_IN = "expires_in" ) 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"` sso string `name:"sso name* icon*@icon" help:"登录"` auth string `name:"auth" help:"授权"` user string `name:"user" help:"用户"` orgs string `name:"orgs" help:"组织"` repo string `name:"repo" help:"资源"` list string `name:"list hash auto" help:"授权" icon:"oauth.png"` login string `name:"login" role:"void"` } var Inputs = []map[string]string{} func init() { Inputs = append(Inputs, map[string]string{ OAUTH_URL: "/login/oauth/authorize", GRANT_URL: "/login/oauth/access_token", TOKEN_URL: "/login/oauth/access_token", USERS_URL: "/api/v1/user", API_PREFIX: "/api/v1/", TOKEN_PREFIX: "token", }) } func (s Client) Inputs(m *ice.Message, arg ...string) { switch m.Option(ctx.ACTION) { case web.SSO: switch arg[0] { case mdb.NAME: case mdb.ICON: } } switch s.Hash.Inputs(m, arg...); arg[0] { case web.DOMAIN: m.Cmdy(web.SPIDE, mdb.INPUTS, arg) default: for _, input := range Inputs { if value, ok := input[arg[0]]; ok { m.Push(arg[0], value) } } } } func (s Client) Sso(m *ice.Message, arg ...string) { mdb.Conf(m, web.CHAT_HEADER, kit.Keym(web.SSO, m.Option(mdb.NAME), web.URL), 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))) mdb.Conf(m, web.CHAT_HEADER, kit.Keym(web.SSO, m.Option(mdb.NAME), mdb.ICON), m.Option(mdb.ICON)) } 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))) } 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)) } } func (s Client) Orgs(m *ice.Message, arg ...string) {} func (s Client) Repo(m *ice.Message, arg ...string) {} func (s Client) List(m *ice.Message, arg ...string) { s.Hash.List(m, arg...).PushAction(s.User, s.Auth, s.Sso, s.Remove).EchoScript(s.RedirectURI(m)) } func init() { ice.ChatCtxCmd(Client{}) } func (s Client) RedirectURI(m *ice.Message) string { return strings.Split(web.MergeURL2(m, 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(res) if s.User(m); !m.Warn(m.Option(aaa.USERNAME) == "") && m.R != nil { web.RenderCookie(m.Message, aaa.SessCreate(m.Message, m.Option(aaa.USERNAME))) m.ProcessHistory() } else { m.ProcessClose() } } } } 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...)) } func (s Client) Put(m *ice.Message, hash, api string, arg ...string) ice.Any { return web.SpidePut(m.Message, s.request(m, hash, api, arg...)) } func (s Client) Post(m *ice.Message, hash, api string, arg ...string) ice.Any { return web.SpidePost(m.Message, s.request(m, hash, api, arg...)) } 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) 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) { m.Options(web.SPIDE_HEADER, ice.Maps{web.Authorization: msg.Append(TOKEN_PREFIX) + lex.SP + 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) }