// Code generated by sqlc. DO NOT EDIT. // versions: // sqlc v1.31.1 // source: notifications.sql package dbq import ( "context" "github.com/jackc/pgx/v5/pgtype" ) const countUnreadNotifications = `-- name: CountUnreadNotifications :one SELECT count(*) FROM user_notifications WHERE user_id = $1 AND read_at IS NULL ` func (q *Queries) CountUnreadNotifications(ctx context.Context, userID pgtype.UUID) (int64, error) { row := q.db.QueryRow(ctx, countUnreadNotifications, userID) var count int64 err := row.Scan(&count) return count, err } const insertNotification = `-- name: InsertNotification :one INSERT INTO user_notifications (user_id, kind, payload) VALUES ($1, $2, $3) RETURNING id ` type InsertNotificationParams struct { UserID pgtype.UUID Kind string Payload []byte } // M489 notifications inbox (#726). Rows are written only through // internal/notifications.Notifier, which decides recipients and honours each // recipient's inbox preference before it reaches these. func (q *Queries) InsertNotification(ctx context.Context, arg InsertNotificationParams) (pgtype.UUID, error) { row := q.db.QueryRow(ctx, insertNotification, arg.UserID, arg.Kind, arg.Payload) var id pgtype.UUID err := row.Scan(&id) return id, err } const listAdminUserIDs = `-- name: ListAdminUserIDs :many SELECT id FROM users WHERE is_admin = true ORDER BY created_at, id ` func (q *Queries) ListAdminUserIDs(ctx context.Context) ([]pgtype.UUID, error) { rows, err := q.db.Query(ctx, listAdminUserIDs) if err != nil { return nil, err } defer rows.Close() var items []pgtype.UUID for rows.Next() { var id pgtype.UUID if err := rows.Scan(&id); err != nil { return nil, err } items = append(items, id) } if err := rows.Err(); err != nil { return nil, err } return items, nil } const listNotificationPrefsForKind = `-- name: ListNotificationPrefsForKind :many SELECT user_id, inbox, phone, email FROM user_notification_prefs WHERE kind = $1 AND user_id = ANY($2::uuid[]) ` type ListNotificationPrefsForKindParams struct { Kind string UserIds []pgtype.UUID } type ListNotificationPrefsForKindRow struct { UserID pgtype.UUID Inbox bool Phone bool Email bool } // The stored prefs of one kind for a set of recipients. A recipient with no // row has the kind's defaults. func (q *Queries) ListNotificationPrefsForKind(ctx context.Context, arg ListNotificationPrefsForKindParams) ([]ListNotificationPrefsForKindRow, error) { rows, err := q.db.Query(ctx, listNotificationPrefsForKind, arg.Kind, arg.UserIds) if err != nil { return nil, err } defer rows.Close() var items []ListNotificationPrefsForKindRow for rows.Next() { var i ListNotificationPrefsForKindRow if err := rows.Scan( &i.UserID, &i.Inbox, &i.Phone, &i.Email, ); err != nil { return nil, err } items = append(items, i) } if err := rows.Err(); err != nil { return nil, err } return items, nil } const listNotificationPrefsForUser = `-- name: ListNotificationPrefsForUser :many SELECT kind, inbox, phone, email FROM user_notification_prefs WHERE user_id = $1 ` type ListNotificationPrefsForUserRow struct { Kind string Inbox bool Phone bool Email bool } func (q *Queries) ListNotificationPrefsForUser(ctx context.Context, userID pgtype.UUID) ([]ListNotificationPrefsForUserRow, error) { rows, err := q.db.Query(ctx, listNotificationPrefsForUser, userID) if err != nil { return nil, err } defer rows.Close() var items []ListNotificationPrefsForUserRow for rows.Next() { var i ListNotificationPrefsForUserRow if err := rows.Scan( &i.Kind, &i.Inbox, &i.Phone, &i.Email, ); err != nil { return nil, err } items = append(items, i) } if err := rows.Err(); err != nil { return nil, err } return items, nil } const listNotifications = `-- name: ListNotifications :many SELECT id, kind, payload, created_at, read_at FROM user_notifications WHERE user_id = $1 AND ($2::timestamptz IS NULL OR (created_at, id) < ($2::timestamptz, $3::uuid)) ORDER BY created_at DESC, id DESC LIMIT $4 ` type ListNotificationsParams struct { UserID pgtype.UUID BeforeCreatedAt pgtype.Timestamptz BeforeID pgtype.UUID PageLimit int32 } type ListNotificationsRow struct { ID pgtype.UUID Kind string Payload []byte CreatedAt pgtype.Timestamptz ReadAt pgtype.Timestamptz } // Newest first, keyset-paged on (created_at, id). Pass both cursor halves // from the last row of the previous page, or neither for the first page. func (q *Queries) ListNotifications(ctx context.Context, arg ListNotificationsParams) ([]ListNotificationsRow, error) { rows, err := q.db.Query(ctx, listNotifications, arg.UserID, arg.BeforeCreatedAt, arg.BeforeID, arg.PageLimit, ) if err != nil { return nil, err } defer rows.Close() var items []ListNotificationsRow for rows.Next() { var i ListNotificationsRow if err := rows.Scan( &i.ID, &i.Kind, &i.Payload, &i.CreatedAt, &i.ReadAt, ); err != nil { return nil, err } items = append(items, i) } if err := rows.Err(); err != nil { return nil, err } return items, nil } const markAllNotificationsRead = `-- name: MarkAllNotificationsRead :execrows UPDATE user_notifications SET read_at = now() WHERE user_id = $1 AND read_at IS NULL AND ($2::timestamptz IS NULL OR created_at <= $2) ` type MarkAllNotificationsReadParams struct { UserID pgtype.UUID UpTo pgtype.Timestamptz } // up_to, when set, limits it to what existed when the user asked: a "mark // all read" queued offline and replayed later must not mark notices that // arrived in between, which the user never saw. A coalesced row updated // since then carries a newer created_at, so it stays unread too. func (q *Queries) MarkAllNotificationsRead(ctx context.Context, arg MarkAllNotificationsReadParams) (int64, error) { result, err := q.db.Exec(ctx, markAllNotificationsRead, arg.UserID, arg.UpTo) if err != nil { return 0, err } return result.RowsAffected(), nil } const markNotificationRead = `-- name: MarkNotificationRead :execrows UPDATE user_notifications SET read_at = COALESCE(read_at, now()) WHERE id = $1 AND user_id = $2 ` type MarkNotificationReadParams struct { ID pgtype.UUID UserID pgtype.UUID } // Scoped to the owner: another user's id matches no row. Marking an already // read row keeps its original read_at and still matches, so a repeat is not // mistaken for "not yours". func (q *Queries) MarkNotificationRead(ctx context.Context, arg MarkNotificationReadParams) (int64, error) { result, err := q.db.Exec(ctx, markNotificationRead, arg.ID, arg.UserID) if err != nil { return 0, err } return result.RowsAffected(), nil } const trimNotifications = `-- name: TrimNotifications :execrows DELETE FROM user_notifications WHERE (read_at IS NOT NULL AND read_at < $1) OR created_at < $2 ` type TrimNotificationsParams struct { ReadCutoff pgtype.Timestamptz AnyCutoff pgtype.Timestamptz } // Retention: read rows go after read_cutoff, and anything at all after // any_cutoff, so an inbox nobody opens doesn't grow without bound either. func (q *Queries) TrimNotifications(ctx context.Context, arg TrimNotificationsParams) (int64, error) { result, err := q.db.Exec(ctx, trimNotifications, arg.ReadCutoff, arg.AnyCutoff) if err != nil { return 0, err } return result.RowsAffected(), nil } const upsertCoalescedNotification = `-- name: UpsertCoalescedNotification :one INSERT INTO user_notifications (user_id, kind, payload, coalesce_key) VALUES ($1, $2, $3, $4) ON CONFLICT (user_id, coalesce_key) WHERE read_at IS NULL AND coalesce_key IS NOT NULL DO UPDATE SET payload = CASE WHEN $5::boolean THEN EXCLUDED.payload || jsonb_build_object('count', COALESCE((user_notifications.payload->>'count')::bigint, 0) + COALESCE((EXCLUDED.payload->>'count')::bigint, 0)) ELSE EXCLUDED.payload END, created_at = now(), emailed_at = NULL RETURNING id ` type UpsertCoalescedNotificationParams struct { UserID pgtype.UUID Kind string Payload []byte CoalesceKey *string SumCount bool } // One unread row per (user, coalesce_key). A new event while that row is // unread updates it in place and moves it back to the top of the inbox. // // sum_count: the payload's `count` adds to the unread row's count rather than // replacing it. For kinds whose event is "N more happened" (tracks marked // missing). Kinds whose event states the whole current total (pending // duplicate groups) pass false and the payload simply replaces. // // emailed_at is cleared so the newer state goes out in the next batch; an // emailed-but-unread row that keeps growing is news the user hasn't seen. func (q *Queries) UpsertCoalescedNotification(ctx context.Context, arg UpsertCoalescedNotificationParams) (pgtype.UUID, error) { row := q.db.QueryRow(ctx, upsertCoalescedNotification, arg.UserID, arg.Kind, arg.Payload, arg.CoalesceKey, arg.SumCount, ) var id pgtype.UUID err := row.Scan(&id) return id, err } const upsertNotificationPref = `-- name: UpsertNotificationPref :exec INSERT INTO user_notification_prefs (user_id, kind, inbox, phone, email) VALUES ($1, $2, $3, $4, $5) ON CONFLICT (user_id, kind) DO UPDATE SET inbox = EXCLUDED.inbox, phone = EXCLUDED.phone, email = EXCLUDED.email, updated_at = now() ` type UpsertNotificationPrefParams struct { UserID pgtype.UUID Kind string Inbox bool Phone bool Email bool } func (q *Queries) UpsertNotificationPref(ctx context.Context, arg UpsertNotificationPrefParams) error { _, err := q.db.Exec(ctx, upsertNotificationPref, arg.UserID, arg.Kind, arg.Inbox, arg.Phone, arg.Email, ) return err }