99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
package openapi
|
|
|
|
import (
|
|
"bytes"
|
|
"sync"
|
|
|
|
"realy.lol/context"
|
|
"realy.lol/event"
|
|
"realy.lol/filter"
|
|
"realy.lol/publish"
|
|
"realy.lol/publish/publisher"
|
|
"realy.lol/tag"
|
|
"realy.lol/typer"
|
|
)
|
|
|
|
const Type = "openapi"
|
|
|
|
// H is the control structure for a HTTP SSE subscription, including the filter, authed
|
|
// pubkey and a channel to send the events to.
|
|
type H struct {
|
|
// Ctx is the http.Request context of the subscriber, this enables garbage
|
|
// collecting the subscriptions from http.
|
|
Ctx context.T
|
|
// Receiver is a channel that the listener sends subscription events to for http
|
|
// subscribe endpoint.
|
|
Receiver event.C
|
|
// Pubkey is the pubkey authed to this subscription
|
|
Pubkey []byte
|
|
// Filter is the filter associated with the http subscription
|
|
Filter *filter.T
|
|
}
|
|
|
|
func init() {
|
|
publish.Register(NewPublisher())
|
|
}
|
|
|
|
func (h *H) Type() string { return Type }
|
|
|
|
// Map is a collection of H TTP subscriptions.
|
|
type Map map[*H]struct{}
|
|
|
|
type S struct {
|
|
// Map is the map of subscriptions from the http api.
|
|
Map
|
|
// HLock is the mutex that locks the Map.
|
|
Mx sync.Mutex
|
|
}
|
|
|
|
var _ publisher.I = &S{}
|
|
|
|
func NewPublisher() *S { return &S{Map: make(Map)} }
|
|
|
|
func (p *S) Type() string { return Type }
|
|
|
|
func (p *S) Receive(msg typer.T) {
|
|
if m, ok := msg.(*H); ok {
|
|
p.Mx.Lock()
|
|
p.Map[m] = struct{}{}
|
|
p.Mx.Unlock()
|
|
}
|
|
}
|
|
|
|
func (p *S) Deliver(ev *event.T) {
|
|
p.Mx.Lock()
|
|
var subs []*H
|
|
for sub := range p.Map {
|
|
// check if the subscription's subscriber is still alive
|
|
select {
|
|
case <-sub.Ctx.Done():
|
|
subs = append(subs, sub)
|
|
default:
|
|
}
|
|
}
|
|
for _, sub := range subs {
|
|
delete(p.Map, sub)
|
|
}
|
|
subs = subs[:0]
|
|
for sub := range p.Map {
|
|
// if the filter doesn't match, skip
|
|
if !sub.Filter.Matches(ev) {
|
|
continue
|
|
}
|
|
// if the filter is privileged and the user doesn't have matching auth, skip
|
|
if ev.Kind.IsPrivileged() {
|
|
ab := sub.Pubkey
|
|
var containsPubkey bool
|
|
if ev.Tags != nil {
|
|
containsPubkey = ev.Tags.ContainsAny([]byte{'p'}, tag.New(ab))
|
|
}
|
|
if !bytes.Equal(ev.Pubkey, ab) || containsPubkey {
|
|
continue
|
|
}
|
|
}
|
|
// send the event to the subscriber
|
|
sub.Receiver <- ev
|
|
}
|
|
p.Mx.Unlock()
|
|
}
|