bench-forgejo/routers/user/social.go

118 lines
2.8 KiB
Go
Raw Normal View History

2014-04-02 08:09:04 +05:30
// Copyright 2014 The Gogs Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
2014-04-02 08:09:04 +05:30
package user
import (
"encoding/json"
2014-04-12 20:49:17 +05:30
"fmt"
2014-04-09 21:37:57 +05:30
"net/url"
"strings"
2014-04-02 08:09:04 +05:30
"code.google.com/p/goauth2/oauth"
2014-04-12 20:49:17 +05:30
"github.com/go-martini/martini"
"github.com/gogits/gogs/models"
2014-04-09 21:37:57 +05:30
"github.com/gogits/gogs/modules/base"
2014-04-02 08:09:04 +05:30
"github.com/gogits/gogs/modules/log"
"github.com/gogits/gogs/modules/middleware"
2014-04-02 08:09:04 +05:30
)
2014-04-12 20:49:17 +05:30
type BasicUserInfo struct {
Identity string
Name string
Email string
}
2014-04-12 20:49:17 +05:30
type SocialConnector interface {
Type() int
SetRedirectUrl(string)
2014-04-12 22:45:19 +05:30
UserInfo(*oauth.Token, *url.URL) (*BasicUserInfo, error)
2014-04-12 20:49:17 +05:30
AuthCodeURL(string) string
Exchange(string) (*oauth.Token, error)
}
2014-04-02 08:09:04 +05:30
2014-04-09 21:37:57 +05:30
func extractPath(next string) string {
n, err := url.Parse(next)
if err != nil {
return "/"
}
return n.Path
}
2014-04-12 20:49:17 +05:30
var (
SocialBaseUrl = "/user/login"
SocialMap = make(map[string]SocialConnector)
)
2014-04-12 07:12:09 +05:30
2014-04-12 20:49:17 +05:30
// github && google && ...
func SocialSignIn(params martini.Params, ctx *middleware.Context) {
if base.OauthService == nil || !base.OauthService.GitHub.Enabled {
ctx.Handle(404, "social login not enabled", nil)
2014-04-09 21:37:57 +05:30
return
}
2014-04-12 20:49:17 +05:30
next := extractPath(ctx.Query("next"))
name := params["name"]
connect, ok := SocialMap[name]
if !ok {
ctx.Handle(404, "social login", nil)
return
2014-04-09 21:37:57 +05:30
}
code := ctx.Query("code")
if code == "" {
// redirect to social login page
2014-04-12 22:45:19 +05:30
connect.SetRedirectUrl(strings.TrimSuffix(base.AppUrl, "/") + ctx.Req.URL.Host + ctx.Req.URL.Path)
2014-04-12 20:49:17 +05:30
ctx.Redirect(connect.AuthCodeURL(next))
return
}
2014-04-09 21:37:57 +05:30
// handle call back
2014-04-12 22:45:19 +05:30
tk, err := connect.Exchange(code) // exchange for token
2014-04-09 21:37:57 +05:30
if err != nil {
log.Error("oauth2 handle callback error: %v", err)
2014-04-12 20:49:17 +05:30
ctx.Handle(500, "exchange code error", nil)
return
2014-04-09 21:37:57 +05:30
}
next = extractPath(ctx.Query("state"))
2014-04-12 20:49:17 +05:30
log.Trace("success get token")
2014-04-09 21:37:57 +05:30
2014-04-12 22:45:19 +05:30
ui, err := connect.UserInfo(tk, ctx.Req.URL)
2014-04-12 20:49:17 +05:30
if err != nil {
ctx.Handle(500, fmt.Sprintf("get infomation from %s error: %v", name, err), nil)
log.Error("social connect error: %s", err)
return
}
2014-04-12 20:49:17 +05:30
log.Info("social login: %s", ui)
oa, err := models.GetOauth2(ui.Identity)
2014-04-08 21:56:12 +05:30
switch err {
case nil:
ctx.Session.Set("userId", oa.User.Id)
ctx.Session.Set("userName", oa.User.Name)
case models.ErrOauth2RecordNotExists:
oa = &models.Oauth2{}
2014-04-12 20:49:17 +05:30
raw, _ := json.Marshal(tk) // json encode
oa.Token = string(raw)
2014-04-11 22:31:30 +05:30
oa.Uid = -1
2014-04-12 20:49:17 +05:30
oa.Type = connect.Type()
oa.Identity = ui.Identity
log.Trace("oa: %v", oa)
if err = models.AddOauth2(oa); err != nil {
2014-04-09 21:37:57 +05:30
log.Error("add oauth2 %v", err) // 501
return
}
2014-04-08 21:56:12 +05:30
case models.ErrOauth2NotAssociatedWithUser:
2014-04-12 20:49:17 +05:30
next = "/user/sign_up"
2014-04-08 22:01:09 +05:30
default:
2014-04-12 20:49:17 +05:30
log.Error("other error: %v", err)
ctx.Handle(500, err.Error(), nil)
2014-04-08 22:01:09 +05:30
return
}
2014-04-08 21:56:12 +05:30
ctx.Session.Set("socialId", oa.Id)
2014-04-12 20:49:17 +05:30
ctx.Session.Set("socialName", ui.Name)
ctx.Session.Set("socialEmail", ui.Email)
log.Trace("socialId: %v", oa.Id)
2014-04-09 21:37:57 +05:30
ctx.Redirect(next)
2014-04-02 08:09:04 +05:30
}