forgejo-federation/models/oauth2.go

50 lines
1.1 KiB
Go
Raw Normal View History

2014-04-09 23:41:24 +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-03 08:43:35 +05:30
package models
2014-04-08 22:01:09 +05:30
import "errors"
2014-04-03 08:43:35 +05:30
// OT: Oauth2 Type
const (
OT_GITHUB = iota + 1
OT_GOOGLE
OT_TWITTER
)
2014-04-08 21:56:12 +05:30
var (
ErrOauth2RecordNotExists = errors.New("not exists oauth2 record")
ErrOauth2NotAssociatedWithUser = errors.New("not associated with user")
)
2014-04-03 08:43:35 +05:30
type Oauth2 struct {
2014-04-08 21:56:12 +05:30
Id int64
2014-04-09 23:41:24 +05:30
Uid int64 // userId
2014-04-08 21:56:12 +05:30
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"`
}
func AddOauth2(oa *Oauth2) (err error) {
if _, err = orm.Insert(oa); err != nil {
return err
}
return nil
}
2014-04-08 21:56:12 +05:30
func GetOauth2(identity string) (oa *Oauth2, err error) {
2014-04-09 23:41:24 +05:30
oa = &Oauth2{Identity: identity}
isExist, err := orm.Get(oa)
if err != nil {
return
2014-04-09 23:41:24 +05:30
} else if !isExist {
2014-04-08 22:01:09 +05:30
return nil, ErrOauth2RecordNotExists
2014-04-09 23:41:24 +05:30
} else if oa.Uid == 0 {
2014-04-08 21:56:12 +05:30
return oa, ErrOauth2NotAssociatedWithUser
}
oa.User, err = GetUserById(oa.Uid)
return oa, err
2014-04-03 08:43:35 +05:30
}