forked from Mirrors/gomuks
78 lines
2.1 KiB
Go
78 lines
2.1 KiB
Go
// Copyright (c) 2025 Tulir Asokan
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
package database
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"go.mau.fi/util/dbutil"
|
|
"go.mau.fi/util/jsontime"
|
|
)
|
|
|
|
const (
|
|
getNonExpiredPushTargets = `
|
|
SELECT device_id, type, data, encryption, expiration
|
|
FROM push_registration
|
|
WHERE expiration > $1
|
|
`
|
|
putPushRegistration = `
|
|
INSERT INTO push_registration (device_id, type, data, encryption, expiration)
|
|
VALUES ($1, $2, $3, $4, $5)
|
|
ON CONFLICT (device_id) DO UPDATE SET
|
|
type = EXCLUDED.type,
|
|
data = EXCLUDED.data,
|
|
encryption = EXCLUDED.encryption,
|
|
expiration = EXCLUDED.expiration
|
|
`
|
|
)
|
|
|
|
type PushRegistrationQuery struct {
|
|
*dbutil.QueryHelper[*PushRegistration]
|
|
}
|
|
|
|
func (prq *PushRegistrationQuery) Put(ctx context.Context, reg *PushRegistration) error {
|
|
return prq.Exec(ctx, putPushRegistration, reg.sqlVariables()...)
|
|
}
|
|
|
|
func (seq *PushRegistrationQuery) GetAll(ctx context.Context) ([]*PushRegistration, error) {
|
|
return seq.QueryMany(ctx, getNonExpiredPushTargets, time.Now().Unix())
|
|
}
|
|
|
|
type PushType string
|
|
|
|
const (
|
|
PushTypeFCM PushType = "fcm"
|
|
)
|
|
|
|
type EncryptionKey struct {
|
|
Key []byte `json:"key,omitempty"`
|
|
}
|
|
|
|
type PushRegistration struct {
|
|
DeviceID string `json:"device_id"`
|
|
Type PushType `json:"type"`
|
|
Data json.RawMessage `json:"data"`
|
|
Encryption EncryptionKey `json:"encryption"`
|
|
Expiration jsontime.Unix `json:"expiration"`
|
|
}
|
|
|
|
func (pe *PushRegistration) Scan(row dbutil.Scannable) (*PushRegistration, error) {
|
|
err := row.Scan(&pe.DeviceID, &pe.Type, (*[]byte)(&pe.Data), dbutil.JSON{Data: &pe.Encryption}, &pe.Expiration)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return pe, nil
|
|
}
|
|
|
|
func (pe *PushRegistration) sqlVariables() []any {
|
|
if pe.Expiration.IsZero() {
|
|
pe.Expiration = jsontime.U(time.Now().Add(7 * 24 * time.Hour))
|
|
}
|
|
return []interface{}{pe.DeviceID, pe.Type, unsafeJSONString(pe.Data), dbutil.JSON{Data: &pe.Encryption}, pe.Expiration}
|
|
}
|