forgejo-federation/models/oauth2.go

107 lines
2.4 KiB
Go
Raw Normal View History

2014-04-10 23:50:58 +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.
package models
2014-04-11 22:31:30 +05:30
import (
"errors"
2014-08-10 05:55:02 +05:30
"time"
2014-04-11 22:31:30 +05:30
)
2014-04-10 23:50:58 +05:30
2014-06-21 10:21:41 +05:30
type OauthType int
2014-04-10 23:50:58 +05:30
const (
2014-06-21 10:21:41 +05:30
GITHUB OauthType = iota + 1
GOOGLE
TWITTER
QQ
WEIBO
BITBUCKET
FACEBOOK
2014-04-10 23:50:58 +05:30
)
var (
2014-04-14 03:42:07 +05:30
ErrOauth2RecordNotExist = errors.New("OAuth2 record does not exist")
ErrOauth2NotAssociated = errors.New("OAuth2 is not associated with user")
2014-04-10 23:50:58 +05:30
)
type Oauth2 struct {
2014-08-10 05:55:02 +05:30
Id int64
Uid int64 `xorm:"unique(s)"` // userId
User *User `xorm:"-"`
Type int `xorm:"unique(s) unique(oauth)"` // twitter,github,google...
Identity string `xorm:"unique(s) unique(oauth)"` // id..
Token string `xorm:"TEXT not null"`
Created time.Time `xorm:"CREATED"`
Updated time.Time
HasRecentActivity bool `xorm:"-"`
2014-04-10 23:50:58 +05:30
}
2014-04-11 22:31:30 +05:30
func BindUserOauth2(userId, oauthId int64) error {
2014-06-21 10:21:41 +05:30
_, err := x.Id(oauthId).Update(&Oauth2{Uid: userId})
2014-04-11 22:31:30 +05:30
return err
}
2014-04-14 03:42:07 +05:30
func AddOauth2(oa *Oauth2) error {
2014-06-21 10:21:41 +05:30
_, err := x.Insert(oa)
2014-04-14 03:42:07 +05:30
return err
2014-04-10 23:50:58 +05:30
}
func GetOauth2(identity string) (oa *Oauth2, err error) {
oa = &Oauth2{Identity: identity}
2014-06-21 10:21:41 +05:30
isExist, err := x.Get(oa)
2014-04-10 23:50:58 +05:30
if err != nil {
return
} else if !isExist {
2014-04-14 03:42:07 +05:30
return nil, ErrOauth2RecordNotExist
2014-04-12 20:49:17 +05:30
} else if oa.Uid == -1 {
2014-04-14 03:42:07 +05:30
return oa, ErrOauth2NotAssociated
2014-04-10 23:50:58 +05:30
}
oa.User, err = GetUserById(oa.Uid)
return oa, err
}
2014-04-11 22:31:30 +05:30
func GetOauth2ById(id int64) (oa *Oauth2, err error) {
oa = new(Oauth2)
2014-06-21 10:21:41 +05:30
has, err := x.Id(id).Get(oa)
2014-04-11 22:31:30 +05:30
if err != nil {
return nil, err
2014-04-14 03:42:07 +05:30
} else if !has {
return nil, ErrOauth2RecordNotExist
2014-04-11 22:31:30 +05:30
}
return oa, nil
}
2014-04-14 06:30:12 +05:30
2014-08-10 05:55:02 +05:30
// UpdateOauth2 updates given OAuth2.
func UpdateOauth2(oa *Oauth2) error {
_, err := x.Id(oa.Id).AllCols().Update(oa)
return err
}
2014-04-14 06:30:12 +05:30
// GetOauthByUserId returns list of oauthes that are releated to given user.
2014-08-10 05:55:02 +05:30
func GetOauthByUserId(uid int64) ([]*Oauth2, error) {
socials := make([]*Oauth2, 0, 5)
err := x.Find(&socials, Oauth2{Uid: uid})
if err != nil {
return nil, err
}
for _, social := range socials {
social.HasRecentActivity = social.Updated.Add(7 * 24 * time.Hour).After(time.Now())
}
return socials, err
2014-04-14 06:30:12 +05:30
}
2014-05-06 01:51:43 +05:30
// DeleteOauth2ById deletes a oauth2 by ID.
func DeleteOauth2ById(id int64) error {
2014-06-21 10:21:41 +05:30
_, err := x.Delete(&Oauth2{Id: id})
2014-05-06 01:51:43 +05:30
return err
}
2014-05-06 23:17:47 +05:30
// CleanUnbindOauth deletes all unbind OAuthes.
func CleanUnbindOauth() error {
2014-06-21 10:21:41 +05:30
_, err := x.Delete(&Oauth2{Uid: -1})
2014-05-06 23:17:47 +05:30
return err
}