mirror of
https://github.com/gravitl/netmaker.git
synced 2026-04-23 00:17:10 +08:00
move api tokens models to schema pkg (#3434)
This commit is contained in:
@@ -1,60 +0,0 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
"github.com/gravitl/netmaker/db"
|
||||
)
|
||||
|
||||
// accessTokenTableName - access tokens table
|
||||
const accessTokenTableName = "user_access_tokens"
|
||||
|
||||
// UserAccessToken - token used to access netmaker
|
||||
type UserAccessToken struct {
|
||||
ID string `gorm:"id,primary_key" json:"id"`
|
||||
Name string `gorm:"name" json:"name"`
|
||||
UserName string `gorm:"user_name" json:"user_name"`
|
||||
ExpiresAt time.Time `gorm:"expires_at" json:"expires_at"`
|
||||
LastUsed time.Time `gorm:"last_used" json:"last_used"`
|
||||
CreatedBy string `gorm:"created_by" json:"created_by"`
|
||||
CreatedAt time.Time `gorm:"created_at" json:"created_at"`
|
||||
}
|
||||
|
||||
func (a *UserAccessToken) Table() string {
|
||||
return accessTokenTableName
|
||||
}
|
||||
|
||||
func (a *UserAccessToken) Get() error {
|
||||
return db.FromContext(context.TODO()).Table(a.Table()).First(&a).Where("id = ?", a.ID).Error
|
||||
}
|
||||
|
||||
func (a *UserAccessToken) Update() error {
|
||||
return db.FromContext(context.TODO()).Table(a.Table()).Where("id = ?", a.ID).Updates(&a).Error
|
||||
}
|
||||
|
||||
func (a *UserAccessToken) Create() error {
|
||||
return db.FromContext(context.TODO()).Table(a.Table()).Create(&a).Error
|
||||
}
|
||||
|
||||
func (a *UserAccessToken) List() (ats []UserAccessToken, err error) {
|
||||
err = db.FromContext(context.TODO()).Table(a.Table()).Find(&ats).Error
|
||||
return
|
||||
}
|
||||
|
||||
func (a *UserAccessToken) ListByUser() (ats []UserAccessToken) {
|
||||
db.FromContext(context.TODO()).Table(a.Table()).Where("user_name = ?", a.UserName).Find(&ats)
|
||||
if ats == nil {
|
||||
ats = []UserAccessToken{}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func (a *UserAccessToken) Delete() error {
|
||||
return db.FromContext(context.TODO()).Table(a.Table()).Where("id = ?", a.ID).Delete(&a).Error
|
||||
}
|
||||
|
||||
func (a *UserAccessToken) DeleteAllUserTokens() error {
|
||||
return db.FromContext(context.TODO()).Table(a.Table()).Where("user_name = ? OR created_by = ?", a.UserName, a.UserName).Delete(&a).Error
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user