mirror of https://github.com/muety/wakapi.git
50 lines
995 B
Go
50 lines
995 B
Go
package services
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/muety/wakapi/models"
|
|
)
|
|
|
|
type AliasService struct {
|
|
Config *models.Config
|
|
Db *gorm.DB
|
|
}
|
|
|
|
var userAliases sync.Map
|
|
|
|
func (srv *AliasService) Init() {}
|
|
|
|
func (srv *AliasService) LoadUserAliases(userId string) error {
|
|
var aliases []*models.Alias
|
|
if err := srv.Db.
|
|
Where(&models.Alias{UserID: userId}).
|
|
Find(&aliases).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
userAliases.Store(userId, aliases)
|
|
return nil
|
|
}
|
|
|
|
func (srv *AliasService) GetAliasOrDefault(userId string, summaryType uint8, value string) (string, error) {
|
|
if ua, ok := userAliases.Load(userId); ok {
|
|
for _, a := range ua.([]*models.Alias) {
|
|
if a.Type == summaryType && a.Value == value {
|
|
return a.Key, nil
|
|
}
|
|
}
|
|
return value, nil
|
|
}
|
|
return "", errors.New("user aliases not initialized")
|
|
}
|
|
|
|
func (src *AliasService) IsInitialized(userId string) bool {
|
|
if _, ok := userAliases.Load(userId); ok {
|
|
return true
|
|
}
|
|
return false
|
|
}
|