package services import ( "fmt" "github.com/emvi/logbuch" "github.com/leandro-lugaresi/hub" "github.com/muety/wakapi/config" "github.com/muety/wakapi/models" "github.com/muety/wakapi/repositories" "github.com/muety/wakapi/utils" "github.com/patrickmn/go-cache" uuid "github.com/satori/go.uuid" "time" ) type UserService struct { config *config.Config cache *cache.Cache eventBus *hub.Hub mailService IMailService repository repositories.IUserRepository } func NewUserService(mailService IMailService, userRepo repositories.IUserRepository) *UserService { srv := &UserService{ config: config.Get(), eventBus: config.EventBus(), cache: cache.New(1*time.Hour, 2*time.Hour), mailService: mailService, repository: userRepo, } sub1 := srv.eventBus.Subscribe(0, config.EventWakatimeFailure) go func(sub *hub.Subscription) { for m := range sub.Receiver { user := m.Fields[config.FieldUser].(*models.User) n := m.Fields[config.FieldPayload].(int) logbuch.Warn("resetting wakatime api key for user %s, because of too many failures (%d)", user.ID, n) if _, err := srv.SetWakatimeApiCredentials(user, "", ""); err != nil { logbuch.Error("failed to set wakatime api key for user %s", user.ID) } if user.Email != "" { if err := mailService.SendWakatimeFailureNotification(user, n); err != nil { logbuch.Error("failed to send wakatime failure notification mail to user %s", user.ID) } else { logbuch.Info("sent wakatime connection failure mail to %s", user.ID) } } } }(&sub1) return srv } func (srv *UserService) GetUserById(userId string) (*models.User, error) { if u, ok := srv.cache.Get(userId); ok { return u.(*models.User), nil } u, err := srv.repository.GetById(userId) if err != nil { return nil, err } srv.cache.Set(u.ID, u, cache.DefaultExpiration) return u, nil } func (srv *UserService) GetUserByKey(key string) (*models.User, error) { if u, ok := srv.cache.Get(key); ok { return u.(*models.User), nil } u, err := srv.repository.GetByApiKey(key) if err != nil { return nil, err } srv.cache.SetDefault(u.ID, u) return u, nil } func (srv *UserService) GetUserByEmail(email string) (*models.User, error) { return srv.repository.GetByEmail(email) } func (srv *UserService) GetUserByResetToken(resetToken string) (*models.User, error) { return srv.repository.GetByResetToken(resetToken) } func (srv *UserService) GetAll() ([]*models.User, error) { return srv.repository.GetAll() } func (srv *UserService) GetAllByReports(reportsEnabled bool) ([]*models.User, error) { return srv.repository.GetAllByReports(reportsEnabled) } func (srv *UserService) GetActive(exact bool) ([]*models.User, error) { minDate := time.Now().Add(-24 * time.Hour * time.Duration(srv.config.App.InactiveDays)) if !exact { minDate = utils.FloorDateHour(minDate) } cacheKey := fmt.Sprintf("%s--active", minDate.String()) if u, ok := srv.cache.Get(cacheKey); ok { return u.([]*models.User), nil } results, err := srv.repository.GetByLastActiveAfter(minDate) if err != nil { return nil, err } srv.cache.SetDefault(cacheKey, results) return results, nil } func (srv *UserService) Count() (int64, error) { return srv.repository.Count() } func (srv *UserService) CreateOrGet(signup *models.Signup, isAdmin bool) (*models.User, bool, error) { u := &models.User{ ID: signup.Username, ApiKey: uuid.NewV4().String(), Email: signup.Email, Location: signup.Location, Password: signup.Password, IsAdmin: isAdmin, } if hash, err := utils.HashBcrypt(u.Password, srv.config.Security.PasswordSalt); err != nil { return nil, false, err } else { u.Password = hash } return srv.repository.InsertOrGet(u) } func (srv *UserService) Update(user *models.User) (*models.User, error) { srv.cache.Flush() srv.notifyUpdate(user) return srv.repository.Update(user) } func (srv *UserService) ResetApiKey(user *models.User) (*models.User, error) { srv.cache.Flush() user.ApiKey = uuid.NewV4().String() return srv.Update(user) } func (srv *UserService) SetWakatimeApiCredentials(user *models.User, apiKey string, apiUrl string) (*models.User, error) { srv.cache.Flush() if apiKey != user.WakatimeApiKey { if u, err := srv.repository.UpdateField(user, "wakatime_api_key", apiKey); err != nil { return u, err } } if apiUrl != user.WakatimeApiUrl { return srv.repository.UpdateField(user, "wakatime_api_url", apiUrl) } return user, nil } func (srv *UserService) MigrateMd5Password(user *models.User, login *models.Login) (*models.User, error) { srv.cache.Flush() user.Password = login.Password if hash, err := utils.HashBcrypt(user.Password, srv.config.Security.PasswordSalt); err != nil { return nil, err } else { user.Password = hash } return srv.repository.UpdateField(user, "password", user.Password) } func (srv *UserService) GenerateResetToken(user *models.User) (*models.User, error) { return srv.repository.UpdateField(user, "reset_token", uuid.NewV4()) } func (srv *UserService) Delete(user *models.User) error { srv.cache.Flush() user.ReportsWeekly = false srv.notifyUpdate(user) return srv.repository.Delete(user) } func (srv *UserService) FlushCache() { srv.cache.Flush() } func (srv *UserService) notifyUpdate(user *models.User) { srv.eventBus.Publish(hub.Message{ Name: config.EventUserUpdate, Fields: map[string]interface{}{config.FieldPayload: user}, }) }