package arn import ( "fmt" "time" "github.com/aerogo/nano" ) // Notification represents a user-associated notification. type Notification struct { UserID string `json:"userId"` Created string `json:"created"` Seen string `json:"seen"` hasID PushNotification } // User retrieves the user the notification was sent to. func (notification *Notification) User() *User { user, _ := GetUser(notification.UserID) return user } // CreatedTime returns the created date as a time object. func (notification *Notification) CreatedTime() time.Time { t, _ := time.Parse(time.RFC3339, notification.Created) return t } // String returns a string representation of the notification. func (notification *Notification) String() string { return fmt.Sprintf("[%s] %s", notification.Type, notification.Title) } // NewNotification creates a new notification. func NewNotification(userID UserID, pushNotification *PushNotification) *Notification { return &Notification{ hasID: hasID{ ID: GenerateID("Notification"), }, UserID: userID, Created: DateTimeUTC(), Seen: "", PushNotification: *pushNotification, } } // GetNotification ... func GetNotification(id string) (*Notification, error) { obj, err := DB.Get("Notification", id) if err != nil { return nil, err } return obj.(*Notification), nil } // StreamNotifications returns a stream of all notifications. func StreamNotifications() <-chan *Notification { channel := make(chan *Notification, nano.ChannelBufferSize) go func() { for obj := range DB.All("Notification") { channel <- obj.(*Notification) } close(channel) }() return channel } // AllNotifications returns a slice of all notifications. func AllNotifications() ([]*Notification, error) { all := make([]*Notification, 0, DB.Collection("Notification").Count()) for obj := range StreamNotifications() { all = append(all, obj) } return all, nil }