mirror of https://github.com/gogits/gogs.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
2.7 KiB
107 lines
2.7 KiB
// 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. |
|
|
|
package user |
|
|
|
import ( |
|
"encoding/json" |
|
"errors" |
|
"fmt" |
|
"net/url" |
|
"strings" |
|
"time" |
|
|
|
"github.com/gogits/gogs/models" |
|
"github.com/gogits/gogs/modules/log" |
|
"github.com/gogits/gogs/modules/middleware" |
|
"github.com/gogits/gogs/modules/setting" |
|
"github.com/gogits/gogs/modules/social" |
|
) |
|
|
|
func extractPath(next string) string { |
|
n, err := url.Parse(next) |
|
if err != nil { |
|
return setting.AppRootSubUrl + "/" |
|
} |
|
return n.Path |
|
} |
|
|
|
func SocialSignIn(ctx *middleware.Context) { |
|
if setting.OauthService == nil { |
|
ctx.Handle(404, "social.SocialSignIn(oauth service not enabled)", nil) |
|
return |
|
} |
|
|
|
next := extractPath(ctx.Query("next")) |
|
name := ctx.Params(":name") |
|
connect, ok := social.SocialMap[name] |
|
if !ok { |
|
ctx.Handle(404, "social.SocialSignIn(social login not enabled)", errors.New(name)) |
|
return |
|
} |
|
appUrl := strings.TrimSuffix(setting.AppUrl, "/") |
|
if name == "weibo" { |
|
appUrl = strings.Replace(appUrl, "localhost", "127.0.0.1", 1) |
|
} |
|
|
|
code := ctx.Query("code") |
|
if code == "" { |
|
// redirect to social login page |
|
connect.SetRedirectUrl(appUrl + ctx.Req.URL.Path) |
|
ctx.Redirect(connect.AuthCodeURL(next)) |
|
return |
|
} |
|
|
|
// handle call back |
|
tk, err := connect.Exchange(code) |
|
if err != nil { |
|
ctx.Handle(500, "social.SocialSignIn(Exchange)", err) |
|
return |
|
} |
|
next = extractPath(ctx.Query("state")) |
|
log.Trace("social.SocialSignIn(Got token)") |
|
|
|
ui, err := connect.UserInfo(tk, ctx.Req.URL) |
|
if err != nil { |
|
ctx.Handle(500, fmt.Sprintf("social.SocialSignIn(get info from %s)", name), err) |
|
return |
|
} |
|
log.Info("social.SocialSignIn(social login): %s", ui) |
|
|
|
oa, err := models.GetOauth2(ui.Identity) |
|
switch err { |
|
case nil: |
|
ctx.Session.Set("uid", oa.User.Id) |
|
ctx.Session.Set("uname", oa.User.Name) |
|
case models.ErrOauth2RecordNotExist: |
|
raw, _ := json.Marshal(tk) |
|
oa = &models.Oauth2{ |
|
Uid: -1, |
|
Type: connect.Type(), |
|
Identity: ui.Identity, |
|
Token: string(raw), |
|
} |
|
log.Trace("social.SocialSignIn(oa): %v", oa) |
|
if err = models.AddOauth2(oa); err != nil { |
|
log.Error(4, "social.SocialSignIn(add oauth2): %v", err) // 501 |
|
return |
|
} |
|
case models.ErrOauth2NotAssociated: |
|
next = setting.AppRootSubUrl + "/user/sign_up" |
|
default: |
|
ctx.Handle(500, "social.SocialSignIn(GetOauth2)", err) |
|
return |
|
} |
|
|
|
oa.Updated = time.Now() |
|
if err = models.UpdateOauth2(oa); err != nil { |
|
log.Error(4, "UpdateOauth2: %v", err) |
|
} |
|
|
|
ctx.Session.Set("socialId", oa.Id) |
|
ctx.Session.Set("socialName", ui.Name) |
|
ctx.Session.Set("socialEmail", ui.Email) |
|
log.Trace("social.SocialSignIn(social ID): %v", oa.Id) |
|
ctx.Redirect(next) |
|
}
|
|
|