mirror of
https://github.com/coder/coder.git
synced 2025-07-03 16:13:58 +00:00
82e37732ea
This PR aims to allow clients to use different format for the title and content of inbox notifications - on the watch endpoint. This solution will help to have it working and formatted differently on VSCode, WebUI ... [Related to this issue](https://github.com/coder/internal/issues/523)
107 lines
3.0 KiB
Go
107 lines
3.0 KiB
Go
package dispatch
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"text/template"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"cdr.dev/slog"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"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
|
|
}
|
|
}
|