wakapi/services/alias.go

package services

import (
	"errors"

	"github.com/jinzhu/gorm"
	"github.com/n1try/wakapi/models"
)

type AliasService struct {
	Config *models.Config
	Db     *gorm.DB
}

var userAliases map[string][]*models.Alias

func (srv *AliasService) LoadUserAliases(userId string) error {
	if userAliases == nil {
		userAliases = make(map[string][]*models.Alias)
	}

	var aliases []*models.Alias
	if err := srv.Db.
		Where(&models.Alias{UserID: userId}).
		Find(&aliases).Error; err != nil {
		return err
	}

	userAliases[userId] = aliases
	return nil
}

func (srv *AliasService) GetAliasOrDefault(userId string, summaryType uint8, value string) (string, error) {
	if userAliases, ok := userAliases[userId]; ok {
		for _, a := range userAliases {
			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[userId]; ok {
		return true
	}
	return false
}