diff options
author | Unknwon <u@gogs.io> | 2019-10-24 01:51:46 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-10-24 01:51:46 -0700 |
commit | 01c8df01ec0608f1f25b2f1444adabb98fa5ee8a (patch) | |
tree | f8a7e5dd8d2a8c51e1ce2cabb9d33571a93314dd /internal/db/token.go | |
parent | 613139e7bef81d3573e7988a47eb6765f3de347a (diff) |
internal: move packages under this directory (#5836)
* Rename pkg -> internal
* Rename routes -> route
* Move route -> internal/route
* Rename models -> db
* Move db -> internal/db
* Fix route2 -> route
* Move cmd -> internal/cmd
* Bump version
Diffstat (limited to 'internal/db/token.go')
-rw-r--r-- | internal/db/token.go | 102 |
1 files changed, 102 insertions, 0 deletions
diff --git a/internal/db/token.go b/internal/db/token.go new file mode 100644 index 00000000..2e2f3492 --- /dev/null +++ b/internal/db/token.go @@ -0,0 +1,102 @@ +// 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 db + +import ( + "time" + + gouuid "github.com/satori/go.uuid" + "xorm.io/xorm" + + "gogs.io/gogs/internal/db/errors" + "gogs.io/gogs/internal/tool" +) + +// AccessToken represents a personal access token. +type AccessToken struct { + ID int64 + UID int64 `xorm:"INDEX"` + Name string + Sha1 string `xorm:"UNIQUE VARCHAR(40)"` + + Created time.Time `xorm:"-" json:"-"` + CreatedUnix int64 + Updated time.Time `xorm:"-" json:"-"` // Note: Updated must below Created for AfterSet. + UpdatedUnix int64 + HasRecentActivity bool `xorm:"-" json:"-"` + HasUsed bool `xorm:"-" json:"-"` +} + +func (t *AccessToken) BeforeInsert() { + t.CreatedUnix = time.Now().Unix() +} + +func (t *AccessToken) BeforeUpdate() { + t.UpdatedUnix = time.Now().Unix() +} + +func (t *AccessToken) AfterSet(colName string, _ xorm.Cell) { + switch colName { + case "created_unix": + t.Created = time.Unix(t.CreatedUnix, 0).Local() + case "updated_unix": + t.Updated = time.Unix(t.UpdatedUnix, 0).Local() + t.HasUsed = t.Updated.After(t.Created) + t.HasRecentActivity = t.Updated.Add(7 * 24 * time.Hour).After(time.Now()) + } +} + +// NewAccessToken creates new access token. +func NewAccessToken(t *AccessToken) error { + t.Sha1 = tool.SHA1(gouuid.NewV4().String()) + has, err := x.Get(&AccessToken{ + UID: t.UID, + Name: t.Name, + }) + if err != nil { + return err + } else if has { + return errors.AccessTokenNameAlreadyExist{t.Name} + } + + _, err = x.Insert(t) + return err +} + +// GetAccessTokenBySHA returns access token by given sha1. +func GetAccessTokenBySHA(sha string) (*AccessToken, error) { + if sha == "" { + return nil, ErrAccessTokenEmpty{} + } + t := &AccessToken{Sha1: sha} + has, err := x.Get(t) + if err != nil { + return nil, err + } else if !has { + return nil, ErrAccessTokenNotExist{sha} + } + return t, nil +} + +// ListAccessTokens returns a list of access tokens belongs to given user. +func ListAccessTokens(uid int64) ([]*AccessToken, error) { + tokens := make([]*AccessToken, 0, 5) + return tokens, x.Where("uid=?", uid).Desc("id").Find(&tokens) +} + +// UpdateAccessToken updates information of access token. +func UpdateAccessToken(t *AccessToken) error { + _, err := x.Id(t.ID).AllCols().Update(t) + return err +} + +// DeleteAccessTokenOfUserByID deletes access token by given ID. +func DeleteAccessTokenOfUserByID(userID, id int64) error { + _, err := x.Delete(&AccessToken{ + ID: id, + UID: userID, + }) + return err +} |