web/forgejo
10
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

add id for oauth2

Browse source
This commit is contained in:
skyblue 2014-04-09 00:26:12 +08:00
parent 22feddf804
commit d4565483e6
3 changed files with 33 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

25
models/oauth2.go
View file

@ -1,6 +1,9 @@
package models
import "fmt"
import (
"errors"
"fmt"
)
// OT: Oauth2 Type
const (
@ -9,12 +12,18 @@ const (
OT_TWITTER
)
var (
ErrOauth2RecordNotExists = errors.New("not exists oauth2 record")
ErrOauth2NotAssociatedWithUser = errors.New("not associated with user")
)
type Oauth2 struct {
Uid int64 `xorm:"pk"` // userId
Id int64
Uid int64 `xorm:"pk"` // userId
User *User `xorm:"-"`
Type int `xorm:"pk unique(oauth)"` // twitter,github,google...
Identity string `xorm:"pk unique(oauth)"` // id..
Token string `xorm:"VARCHAR(200) not null"`
//RefreshTime time.Time `xorm:"created"`
}
func AddOauth2(oa *Oauth2) (err error) {
@ -24,8 +33,8 @@ func AddOauth2(oa *Oauth2) (err error) {
return nil
}
func GetOauth2User(identity string) (u *User, err error) {
oa := &Oauth2{}
func GetOauth2(identity string) (oa *Oauth2, err error) {
oa = &Oauth2{}
oa.Identity = identity
exists, err := orm.Get(oa)
if err != nil {
@ -35,5 +44,9 @@ func GetOauth2User(identity string) (u *User, err error) {
err = fmt.Errorf("not exists oauth2: %s", identity)
return
}
return GetUserById(oa.Uid)
if oa.Uid == 0 {
return oa, ErrOauth2NotAssociatedWithUser
}
oa.User, err = GetUserById(oa.Uid)
return
}