Passport/pkg/internal/services/statuses.go

94 lines
2.1 KiB
Go

package services
import (
"fmt"
"time"
"git.solsynth.dev/hydrogen/passport/pkg/internal/database"
"git.solsynth.dev/hydrogen/passport/pkg/internal/models"
"github.com/samber/lo"
)
var statusCache = make(map[uint]models.Status)
func GetStatus(uid uint) (models.Status, error) {
if status, ok := statusCache[uid]; ok {
if status.ClearAt != nil && status.ClearAt.Before(time.Now()) {
delete(statusCache, uid)
} else {
return status, nil
}
}
var status models.Status
if err := database.C.
Where("account_id = ?", uid).
Where("clear_at > ?", time.Now()).
First(&status).Error; err != nil {
return status, err
} else {
statusCache[uid] = status
}
return status, nil
}
func GetUserOnline(uid uint) bool {
return wsConn[uid] != nil && len(wsConn[uid]) > 0
}
func GetStatusDisturbable(uid uint) error {
status, err := GetStatus(uid)
isOnline := GetUserOnline(uid)
if isOnline && err != nil {
return nil
} else if err == nil && status.IsNoDisturb {
return fmt.Errorf("do not disturb")
} else {
return nil
}
}
func GetStatusOnline(uid uint) error {
status, err := GetStatus(uid)
isOnline := wsConn[uid] != nil && len(wsConn[uid]) > 0
if isOnline && err != nil {
return nil
} else if err == nil && status.IsInvisible {
return fmt.Errorf("invisible")
} else if !isOnline {
return fmt.Errorf("offline")
} else {
return nil
}
}
func NewStatus(user models.Account, status models.Status) (models.Status, error) {
if err := database.C.Save(&status).Error; err != nil {
return status, err
} else {
statusCache[user.ID] = status
}
return status, nil
}
func EditStatus(user models.Account, status models.Status) (models.Status, error) {
if err := database.C.Save(&status).Error; err != nil {
return status, err
} else {
statusCache[user.ID] = status
}
return status, nil
}
func ClearStatus(user models.Account) error {
if err := database.C.
Where("account_id = ?", user.ID).
Where("clear_at > ?", time.Now()).
Updates(models.Status{ClearAt: lo.ToPtr(time.Now())}).Error; err != nil {
return err
} else {
delete(statusCache, user.ID)
}
return nil
}