mirror of
https://github.com/coder/coder.git
synced 2026-06-03 04:58:23 +00:00
Spike Curtis
bddb808b25
Fixes all our Go file imports to match the preferred spec that we've _mostly_ been using. For example: ``` import ( "context" "time" "github.com/prometheus/client_golang/prometheus" "golang.org/x/xerrors" "gopkg.in/natefinch/lumberjack.v2" "cdr.dev/slog/v3" "github.com/coder/coder/v2/codersdk/agentsdk" "github.com/coder/serpent" ) ``` 3 groups: standard library, 3rd partly libs, Coder libs. This PR makes the change across the codebase. The PR in the stack above modifies our formatting to maintain this state of affairs, and is a separate PR so it's possible to review that one in detail.
105 lines
3.0 KiB
Go
105 lines
3.0 KiB
Go
package dispatch
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"text/template"
|
|
|
|
"github.com/google/uuid"
|
|
"golang.org/x/xerrors"
|
|
|
|
"cdr.dev/slog/v3"
|
|
"github.com/coder/coder/v2/coderd/database"
|
|
"github.com/coder/coder/v2/coderd/database/dbtime"
|
|
"github.com/coder/coder/v2/coderd/database/pubsub"
|
|
"github.com/coder/coder/v2/coderd/notifications/types"
|
|
coderdpubsub "github.com/coder/coder/v2/coderd/pubsub"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
)
|
|
|
|
type InboxStore interface {
|
|
InsertInboxNotification(ctx context.Context, arg database.InsertInboxNotificationParams) (database.InboxNotification, error)
|
|
}
|
|
|
|
// InboxHandler is responsible for dispatching notification messages to the Coder Inbox.
|
|
type InboxHandler struct {
|
|
log slog.Logger
|
|
store InboxStore
|
|
pubsub pubsub.Pubsub
|
|
}
|
|
|
|
func NewInboxHandler(log slog.Logger, store InboxStore, ps pubsub.Pubsub) *InboxHandler {
|
|
return &InboxHandler{log: log, store: store, pubsub: ps}
|
|
}
|
|
|
|
func (s *InboxHandler) Dispatcher(payload types.MessagePayload, titleTmpl, bodyTmpl string, _ template.FuncMap) (DeliveryFunc, error) {
|
|
return s.dispatch(payload, titleTmpl, bodyTmpl), nil
|
|
}
|
|
|
|
func (s *InboxHandler) dispatch(payload types.MessagePayload, title, body string) DeliveryFunc {
|
|
return func(ctx context.Context, msgID uuid.UUID) (bool, error) {
|
|
userID, err := uuid.Parse(payload.UserID)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("parse user ID: %w", err)
|
|
}
|
|
templateID, err := uuid.Parse(payload.NotificationTemplateID)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("parse template ID: %w", err)
|
|
}
|
|
|
|
actions, err := json.Marshal(payload.Actions)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("marshal actions: %w", err)
|
|
}
|
|
|
|
// nolint:exhaustruct
|
|
insertedNotif, err := s.store.InsertInboxNotification(ctx, database.InsertInboxNotificationParams{
|
|
ID: msgID,
|
|
UserID: userID,
|
|
TemplateID: templateID,
|
|
Targets: payload.Targets,
|
|
Title: title,
|
|
Content: body,
|
|
Actions: actions,
|
|
CreatedAt: dbtime.Now(),
|
|
})
|
|
if err != nil {
|
|
return false, xerrors.Errorf("insert inbox notification: %w", err)
|
|
}
|
|
|
|
event := coderdpubsub.InboxNotificationEvent{
|
|
Kind: coderdpubsub.InboxNotificationEventKindNew,
|
|
InboxNotification: codersdk.InboxNotification{
|
|
ID: msgID,
|
|
UserID: userID,
|
|
TemplateID: templateID,
|
|
Targets: payload.Targets,
|
|
Title: title,
|
|
Content: body,
|
|
Actions: func() []codersdk.InboxNotificationAction {
|
|
var actions []codersdk.InboxNotificationAction
|
|
err := json.Unmarshal(insertedNotif.Actions, &actions)
|
|
if err != nil {
|
|
return actions
|
|
}
|
|
return actions
|
|
}(),
|
|
ReadAt: nil, // notification just has been inserted
|
|
CreatedAt: insertedNotif.CreatedAt,
|
|
},
|
|
}
|
|
|
|
payload, err := json.Marshal(event)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("marshal event: %w", err)
|
|
}
|
|
|
|
err = s.pubsub.Publish(coderdpubsub.InboxNotificationForOwnerEventChannel(userID), payload)
|
|
if err != nil {
|
|
return false, xerrors.Errorf("publish event: %w", err)
|
|
}
|
|
|
|
return false, nil
|
|
}
|
|
}
|