summaryrefslogtreecommitdiffstats
path: root/models/oauth2.go
blob: a17d4e30faffcef7add6088549e02dd5ea4c1c40 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package models

import "fmt"

// OT: Oauth2 Type
const (
	OT_GITHUB = iota + 1
	OT_GOOGLE
	OT_TWITTER
)

type Oauth2 struct {
	Uid      int64  `xorm:"pk"`               // userId
	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) {
	if _, err = orm.Insert(oa); err != nil {
		return err
	}
	return nil
}

func GetOauth2User(identity string) (u *User, err error) {
	oa := &Oauth2{}
	oa.Identity = identity
	exists, err := orm.Get(oa)
	if err != nil {
		return
	}
	if !exists {
		err = fmt.Errorf("not exists oauth2: %s", identity)
		return
	}
	return GetUserById(oa.Uid)
}