Eli Ribble
41587c3694
I'm trying to see if this speeds up builds a bit. May not without a module boundary, but for now it's nice organization to have as the program grows.
85 lines
2.4 KiB
Go
85 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"time"
|
|
|
|
"github.com/Gleipnir-Technology/nidus-sync/db"
|
|
enums "github.com/Gleipnir-Technology/nidus-sync/db/enums"
|
|
"github.com/Gleipnir-Technology/nidus-sync/db/models"
|
|
"github.com/aarondl/opt/omit"
|
|
"github.com/aarondl/opt/omitnull"
|
|
)
|
|
|
|
var (
|
|
NotificationPathOauthReset string = "/oauth/refresh"
|
|
)
|
|
|
|
type Notification struct {
|
|
Link string
|
|
Message string
|
|
Time time.Time
|
|
Type string
|
|
}
|
|
|
|
// Clear all notifications for a given user with the given path
|
|
func clearNotificationsOauth(ctx context.Context, user *models.User) {
|
|
setter := models.NotificationSetter{
|
|
ResolvedAt: omitnull.From(time.Now()),
|
|
}
|
|
updater := models.Notifications.Update(
|
|
//models.SelectWhere.Notifications.Link.EQ(NotificationPathOauthReset),
|
|
models.UpdateWhere.Notifications.Link.EQ(NotificationPathOauthReset),
|
|
models.UpdateWhere.Notifications.UserID.EQ(user.ID),
|
|
setter.UpdateMod(),
|
|
)
|
|
updater.Exec(ctx, db.PGInstance.BobDB)
|
|
//user.UserNotifications(
|
|
//models.SelectWhere.Notifications.Link.EQ(NotificationPathOauthReset),
|
|
//).UpdateAll()
|
|
}
|
|
|
|
func notifyOauthInvalid(ctx context.Context, user *models.User) {
|
|
notificationSetter := models.NotificationSetter{
|
|
Created: omit.From(time.Now()),
|
|
Message: omit.From("Oauth token invalidated"),
|
|
Link: omit.From(NotificationPathOauthReset),
|
|
Type: omit.From(enums.NotificationtypeOauthTokenInvalidated),
|
|
}
|
|
err := user.InsertUserNotifications(ctx, db.PGInstance.BobDB, ¬ificationSetter)
|
|
if err != nil {
|
|
LogErrorTypeInfo(err)
|
|
slog.Error("Failed to insert new notification. Update this clause to detect duplicate inserts.", slog.String("err", err.Error()))
|
|
return
|
|
}
|
|
}
|
|
|
|
func notificationsForUser(ctx context.Context, u *models.User) ([]Notification, error) {
|
|
results := make([]Notification, 0)
|
|
notifications, err := u.UserNotifications(
|
|
models.SelectWhere.Notifications.ResolvedAt.IsNull(),
|
|
).All(ctx, db.PGInstance.BobDB)
|
|
if err != nil {
|
|
return results, fmt.Errorf("Failed to get notifications: %w", err)
|
|
}
|
|
for _, n := range notifications {
|
|
results = append(results, Notification{
|
|
Link: n.Link,
|
|
Message: n.Message,
|
|
Time: n.Created,
|
|
Type: notificationTypeName(n.Type),
|
|
})
|
|
}
|
|
return results, nil
|
|
}
|
|
|
|
func notificationTypeName(t enums.Notificationtype) string {
|
|
switch t {
|
|
case enums.NotificationtypeOauthTokenInvalidated:
|
|
return "alert"
|
|
default:
|
|
return "unknown-type"
|
|
}
|
|
}
|