aboutsummaryrefslogtreecommitdiff
path: root/internal/db/token.go
diff options
context:
space:
mode:
authorUnknwon <u@gogs.io>2019-10-24 01:51:46 -0700
committerGitHub <noreply@github.com>2019-10-24 01:51:46 -0700
commit01c8df01ec0608f1f25b2f1444adabb98fa5ee8a (patch)
treef8a7e5dd8d2a8c51e1ce2cabb9d33571a93314dd /internal/db/token.go
parent613139e7bef81d3573e7988a47eb6765f3de347a (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.go102
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
+}