79 lines
1.5 KiB
Go
79 lines
1.5 KiB
Go
package service
|
|
|
|
import (
|
|
"orbits-server/internal/server/api/assets"
|
|
"orbits-server/internal/server/database"
|
|
"orbits-server/internal/shared/security"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
const (
|
|
accessKeyLen = 32
|
|
)
|
|
|
|
type KeyService struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewKeyService(db *gorm.DB) *KeyService {
|
|
return &KeyService{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (s *KeyService) ListValidKeyHashes() ([]string, error) {
|
|
keyRecords, err := database.ListKeys(s.db)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
hashList := make([]string, 0, len(keyRecords))
|
|
for _, k := range keyRecords {
|
|
hashList = append(hashList, k.KeyHash)
|
|
}
|
|
|
|
return hashList, nil
|
|
}
|
|
|
|
func (s *KeyService) Create(name string, expiresAt time.Time) (assets.KeyResponse, error) {
|
|
keyContent := security.GenerateChars(accessKeyLen)
|
|
|
|
hash, err := security.HashKey(keyContent)
|
|
if err != nil {
|
|
return assets.KeyResponse{}, err
|
|
}
|
|
|
|
keyRecord := database.BuildKeyRecord(hash, name, expiresAt)
|
|
|
|
if err := database.CreateKey(s.db, &keyRecord); err != nil {
|
|
return assets.KeyResponse{}, err
|
|
}
|
|
|
|
keyResponse := assets.KeyResponse{
|
|
ID: keyRecord.ID,
|
|
MetaName: keyRecord.MetaName,
|
|
KeyName: keyRecord.KeyName,
|
|
KeyContent: keyContent,
|
|
CreatedAt: keyRecord.CreatedAt,
|
|
UpdatedAt: keyRecord.UpdatedAt,
|
|
ExpiresAt: keyRecord.ExpiresAt,
|
|
}
|
|
|
|
return keyResponse, nil
|
|
}
|
|
|
|
func (s *KeyService) DeleteByName(name string) error {
|
|
keyRecord, err := database.FindKeyByName(s.db, name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := database.DeleteKeyByID(s.db, keyRecord.ID); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|