gitea/routers/user/social.go

100 lines
2.4 KiB
Go
Raw Normal View History

2014-04-01 20:39:04 -06:00
// 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-01 20:39:04 -06:00
package user
import (
"encoding/json"
2014-04-13 16:12:07 -06:00
"errors"
2014-04-12 09:19:17 -06:00
"fmt"
2014-04-09 10:07:57 -06:00
"net/url"
"strings"
2014-04-01 20:39:04 -06:00
2014-04-12 09:19:17 -06:00
"github.com/go-martini/martini"
2014-04-12 23:57:42 -06:00
"github.com/gogits/gogs/models"
2014-04-01 20:39:04 -06:00
"github.com/gogits/gogs/modules/log"
"github.com/gogits/gogs/modules/middleware"
2014-05-25 18:11:25 -06:00
"github.com/gogits/gogs/modules/setting"
2014-04-13 16:12:07 -06:00
"github.com/gogits/gogs/modules/social"
2014-04-01 20:39:04 -06:00
)
2014-04-09 10:07:57 -06:00
func extractPath(next string) string {
n, err := url.Parse(next)
if err != nil {
return "/"
}
return n.Path
}
2014-04-13 16:12:07 -06:00
func SocialSignIn(ctx *middleware.Context, params martini.Params) {
2014-05-25 18:11:25 -06:00
if setting.OauthService == nil {
2014-04-13 16:12:07 -06:00
ctx.Handle(404, "social.SocialSignIn(oauth service not enabled)", nil)
2014-04-09 10:07:57 -06:00
return
}
2014-04-13 16:12:07 -06:00
2014-04-12 09:19:17 -06:00
next := extractPath(ctx.Query("next"))
name := params["name"]
2014-04-13 16:12:07 -06:00
connect, ok := social.SocialMap[name]
2014-04-12 09:19:17 -06:00
if !ok {
2014-04-13 16:12:07 -06:00
ctx.Handle(404, "social.SocialSignIn(social login not enabled)", errors.New(name))
2014-04-12 09:19:17 -06:00
return
2014-04-09 10:07:57 -06:00
}
2014-04-13 16:12:07 -06:00
2014-04-09 10:07:57 -06:00
code := ctx.Query("code")
if code == "" {
// redirect to social login page
2014-05-25 18:11:25 -06:00
connect.SetRedirectUrl(strings.TrimSuffix(setting.AppUrl, "/") + ctx.Req.URL.Path)
2014-04-12 09:19:17 -06:00
ctx.Redirect(connect.AuthCodeURL(next))
return
}
2014-04-09 10:07:57 -06:00
// handle call back
2014-04-13 16:12:07 -06:00
tk, err := connect.Exchange(code)
2014-04-09 10:07:57 -06:00
if err != nil {
2014-04-13 16:12:07 -06:00
ctx.Handle(500, "social.SocialSignIn(Exchange)", err)
2014-04-12 09:19:17 -06:00
return
2014-04-09 10:07:57 -06:00
}
next = extractPath(ctx.Query("state"))
2014-04-13 16:12:07 -06:00
log.Trace("social.SocialSignIn(Got token)")
2014-04-09 10:07:57 -06:00
2014-04-12 11:15:19 -06:00
ui, err := connect.UserInfo(tk, ctx.Req.URL)
2014-04-12 09:19:17 -06:00
if err != nil {
2014-04-13 16:12:07 -06:00
ctx.Handle(500, fmt.Sprintf("social.SocialSignIn(get info from %s)", name), err)
return
}
2014-04-13 16:12:07 -06:00
log.Info("social.SocialSignIn(social login): %s", ui)
2014-04-12 09:19:17 -06:00
oa, err := models.GetOauth2(ui.Identity)
2014-04-08 10:26:12 -06:00
switch err {
case nil:
ctx.Session.Set("userId", oa.User.Id)
ctx.Session.Set("userName", oa.User.Name)
2014-04-13 16:12:07 -06:00
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 {
2014-04-13 16:12:07 -06:00
log.Error("social.SocialSignIn(add oauth2): %v", err) // 501
return
}
2014-04-13 16:12:07 -06:00
case models.ErrOauth2NotAssociated:
2014-04-12 09:19:17 -06:00
next = "/user/sign_up"
2014-04-08 10:31:09 -06:00
default:
2014-04-13 16:12:07 -06:00
ctx.Handle(500, "social.SocialSignIn(GetOauth2)", err)
2014-04-08 10:31:09 -06:00
return
}
2014-04-13 16:12:07 -06:00
2014-04-08 10:26:12 -06:00
ctx.Session.Set("socialId", oa.Id)
2014-04-12 09:19:17 -06:00
ctx.Session.Set("socialName", ui.Name)
ctx.Session.Set("socialEmail", ui.Email)
2014-04-13 16:12:07 -06:00
log.Trace("social.SocialSignIn(social ID): %v", oa.Id)
2014-04-09 10:07:57 -06:00
ctx.Redirect(next)
2014-04-01 20:39:04 -06:00
}