Something went wrong. Try again.
Monorepo for Tangled tangled.org
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330package deliberi
import ( "bytes" "context" "fmt" "html/template" "log/slog" "strings" gotemplate "text/template" "time"
"github.com/bluesky-social/indigo/atproto/syntax" "tangled.org/core/deliberi/config" deldb "tangled.org/core/deliberi/db" "tangled.org/core/deliberi/emailtemplate" "tangled.org/core/deliberi/mailer" "tangled.org/core/deliberi/models" "tangled.org/core/idresolver")
const digestTextTmpl = `Hi {{.RecipientHandle}},
You have {{.Count}} new notification(s) on Tangled:
{{range .Groups}}- {{wrap 70 " " .Header}}{{if .EntityRef}} {{wrap 70 " " .EntityRef}}{{end}} {{.URL}}{{end}}{{if .HasMore}}View more notifications: {{.NotificationsURL}}{{end}}---Manage notifications: {{.SettingsURL}}`
const digestBodyHTMLTmpl = `<p style="font-size:16px;padding:16px 0;">Hi {{.RecipientHandle}},</p><p style="font-size:16px;padding:0 0 16px 0;margin:0;"> You have {{.Count}} new notification{{if gt .Count 1}}s{{end}}:</p><table width="100%" cellpadding="0" cellspacing="0" role="presentation"> {{range .Groups}} <tr> <td style="padding:12px 0;border-bottom:1px solid #eaeaea;"> <a href="{{.URL}}" style="text-decoration:none;color:inherit;display:block;"> <table cellpadding="0" cellspacing="0" width="100%" role="presentation"> <tr> <td style="width:20px;vertical-align:top;padding-top:1px;"> <img src="{{.IconURL}}" width="16" height="16" alt="" style="display:block;border:0;"> </td> <td style="padding-left:8px;"> <p style="margin:0 0 3px 0;font-size:14px;color:#374151;">{{.HeaderHTML}}</p> {{if .EntityRef}}<p style="margin:0;font-size:13px;color:#6b7280;">{{.EntityRef}}</p>{{end}} </td> </tr> </table> </a> </td> </tr> {{end}}</table>{{if .HasMore}}<p style="font-size:14px;padding:16px 0 0 0;margin:0;"> <a href="{{.NotificationsURL}}" style="color:#111827;text-decoration:underline;font-weight:400;">View more notifications</a></p>{{end}}<p style="font-size:16px;padding:32px 0 16px 0;text-align:center;"> <a href="{{.SettingsURL}}" style="color:#111827;text-decoration:underline;font-weight:400;">Manage notification settings</a></p>`
const digestMaxGroups = 10
type digestGroup struct { IconURL string Header string HeaderHTML template.HTML EntityRef string URL string}
type digestData struct { RecipientHandle string Count int Groups []digestGroup HasMore bool NotificationsURL string SettingsURL string}
func notifHeader(n *models.Notification, actor, repo string) string { switch n.Type { case models.NotificationTypeIssueCreated: return actor + " opened an issue on " + repo case models.NotificationTypeIssueCommented: return actor + " commented on an issue on " + repo case models.NotificationTypeIssueClosed: return actor + " closed an issue on " + repo case models.NotificationTypeIssueReopen: return actor + " reopened an issue on " + repo case models.NotificationTypePullCreated: return actor + " created a PR on " + repo case models.NotificationTypePullCommented: return actor + " commented on a PR on " + repo case models.NotificationTypePullMerged: return actor + " merged a PR on " + repo case models.NotificationTypePullClosed: return actor + " closed a PR on " + repo case models.NotificationTypePullReopen: return actor + " reopened a PR on " + repo case models.NotificationTypeUserMentioned: if n.EntityAt != "" && syntax.ATURI(n.EntityAt).Collection().String() == "sh.tangled.repo.pull" { return actor + " mentioned you on a pull request in " + repo } return actor + " mentioned you on an issue in " + repo case models.NotificationTypeIssueAssigned: return actor + " assigned you to an issue on " + repo case models.NotificationTypeIssueUnassigned: return actor + " unassigned you from an issue on " + repo case models.NotificationTypePullAssigned: return actor + " assigned you to a PR on " + repo case models.NotificationTypePullUnassigned: return actor + " unassigned you from a PR on " + repo default: return actor + " updated " + repo }}
func notifEntityRef(n *models.Notification) string { return n.EntityTitle}
func wordwrap(width int, indent, text string) string { words := strings.Fields(text) if len(words) == 0 { return text } var b strings.Builder col := 0 for i, w := range words { if i == 0 { b.WriteString(w) col = len(w) continue } if col+1+len(w) > width { b.WriteString("\n" + indent) b.WriteString(w) col = len(indent) + len(w) } else { b.WriteByte(' ') b.WriteString(w) col += 1 + len(w) } } return b.String()}
type Dispatcher struct { db *deldb.DB sender *mailer.Sender baseURL string assetsURL string resolver *idresolver.Resolver logger *slog.Logger batchWait time.Duration interval time.Duration
textTmpl *gotemplate.Template htmlTmpl *template.Template}
func NewDispatcher(database *deldb.DB, sender *mailer.Sender, resend config.ResendConfig, baseURL string, resolver *idresolver.Resolver, logger *slog.Logger, dev bool) *Dispatcher { batchWait := 10 * time.Minute interval := 5 * time.Minute if dev { batchWait = 30 * time.Second interval = 15 * time.Second } return &Dispatcher{ db: database, sender: sender, baseURL: strings.TrimRight(baseURL, "/"), assetsURL: strings.TrimRight(resend.AssetsURL, "/") + "/", resolver: resolver, logger: logger, batchWait: batchWait, interval: interval, textTmpl: gotemplate.Must(gotemplate.New("digest-text").Funcs(gotemplate.FuncMap{"wrap": wordwrap}).Parse(digestTextTmpl)), htmlTmpl: template.Must(template.New("digest-body").Parse(digestBodyHTMLTmpl)), }}
func (d *Dispatcher) Start(ctx context.Context) { d.logger.Info("email dispatcher started", "interval", d.interval, "batchWait", d.batchWait) ticker := time.NewTicker(d.interval) defer ticker.Stop() for { select { case <-ticker.C: d.dispatch(ctx) case <-ctx.Done(): d.logger.Info("email dispatcher stopped") return } }}
func (d *Dispatcher) dispatch(ctx context.Context) { cutoff := time.Now().Add(-d.batchWait) recipients, err := deldb.GetPendingEmailDigestRecipients(d.db, cutoff) if err != nil { d.logger.Error("email dispatcher: failed to get recipients", "err", err) return } for _, did := range recipients { if err := d.sendDigest(ctx, did, cutoff); err != nil { d.logger.Error("email dispatcher: failed to send digest", "did", did, "err", err) } }}
func (d *Dispatcher) sendDigest(ctx context.Context, recipientDid string, cutoff time.Time) error { em, err := deldb.GetPrimaryEmail(d.db, recipientDid) if err != nil || !em.Verified { return nil }
notifs, err := deldb.GetPendingNotificationsForEmailDigest(d.db, recipientDid, cutoff) if err != nil { return fmt.Errorf("get pending notifications: %w", err) } if len(notifs) == 0 { return nil }
handle := recipientDid if id, err := d.resolver.ResolveIdent(ctx, recipientDid); err == nil && !id.Handle.IsInvalidHandle() { handle = id.Handle.String() }
subject, text, html, err := d.renderDigest(ctx, handle, notifs) if err != nil { return fmt.Errorf("render digest: %w", err) }
// collect ids before sending so notifications arriving mid-send are not // marked emailed. ids := make([]int64, len(notifs)) for i, n := range notifs { ids[i] = n.ID }
if err := d.sender.Send(em.Address, subject, text, html); err != nil { return fmt.Errorf("send email: %w", err) }
d.logger.Info("email dispatcher: digest sent", "did", recipientDid, "notifications", len(notifs))
if err := deldb.MarkEmailed(d.db, ids); err != nil { d.logger.Error("email dispatcher: failed to mark emailed", "did", recipientDid, "err", err) } return nil}
func (d *Dispatcher) renderDigest(ctx context.Context, recipientHandle string, notifs []*models.Notification) (subject, text, html string, err error) { count := len(notifs)
shown := notifs if len(shown) > digestMaxGroups { shown = shown[:digestMaxGroups] }
groups := make([]digestGroup, 0, len(shown)) for _, n := range shown { actorHandle := n.ActorDid if id, err2 := d.resolver.ResolveIdent(ctx, n.ActorDid); err2 == nil && !id.Handle.IsInvalidHandle() { actorHandle = id.Handle.String() }
repoStr := "" if n.RepoDid != "" { repoHandle := n.RepoDid if id, err2 := d.resolver.ResolveIdent(ctx, n.RepoDid); err2 == nil && !id.Handle.IsInvalidHandle() { repoHandle = id.Handle.String() } if n.RepoName != "" { repoStr = repoHandle + "/" + n.RepoName } else { repoStr = repoHandle } }
header := notifHeader(n, actorHandle, repoStr) headerHTML := template.HTML(strings.Replace(header, actorHandle, "<strong>"+actorHandle+"</strong>", 1)) groups = append(groups, digestGroup{ IconURL: d.assetsURL + n.Icon() + ".png", Header: header, HeaderHTML: headerHTML, EntityRef: notifEntityRef(n), URL: d.baseURL + n.URL(d.resolver), }) }
data := digestData{ RecipientHandle: recipientHandle, Count: count, Groups: groups, HasMore: count > digestMaxGroups, NotificationsURL: d.baseURL + "/notifications", SettingsURL: d.baseURL + "/settings/notifications", }
if count > digestMaxGroups { subject = fmt.Sprintf("[%s] %d+ notifications", recipientHandle, digestMaxGroups) } else { subject = fmt.Sprintf("[%s] %d notification(s)", recipientHandle, count) }
var textBuf bytes.Buffer if err = d.textTmpl.Execute(&textBuf, data); err != nil { return } text = textBuf.String()
var htmlBuf bytes.Buffer if err = d.htmlTmpl.Execute(&htmlBuf, data); err != nil { return } html = emailtemplate.Shell("Tangled notifications", htmlBuf.String(), d.baseURL) return}