Files
relayer/closedrelay/save.go
2022-07-24 11:48:34 -03:00

57 lines
1.6 KiB
Go

package main
import (
"encoding/json"
"fmt"
"strings"
"time"
"github.com/fiatjaf/go-nostr"
)
func (b *ClosedRelay) SaveEvent(evt *nostr.Event) error {
// disallow anything from non-authorized pubkeys
for _, pubkey := range b.AuthorizedPubkeys {
if pubkey == evt.PubKey {
goto save
}
}
return fmt.Errorf("event from '%s' not allowed here", evt.PubKey)
save:
// react to different kinds of events
if evt.Kind == nostr.KindSetMetadata || evt.Kind == nostr.KindContactList || (10000 <= evt.Kind && evt.Kind < 20000) {
// delete past events from this user
b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = $2`, evt.PubKey, evt.Kind)
} else if evt.Kind == nostr.KindRecommendServer {
// delete past recommend_server events equal to this one
b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = $2 AND content = $3`,
evt.PubKey, evt.Kind, evt.Content)
} else {
// do not delete any, this is a closed relay so we trust everybody to not spam
}
// insert
tagsj, _ := json.Marshal(evt.Tags)
_, err := b.DB.Exec(`
INSERT INTO event (id, pubkey, created_at, kind, tags, content, sig)
VALUES ($1, $2, $3, $4, $5, $6, $7)
`, evt.ID, evt.PubKey, evt.CreatedAt.Unix(), evt.Kind, tagsj, evt.Content, evt.Sig)
if err != nil {
if strings.Index(err.Error(), "UNIQUE") != -1 {
// already exists
return nil
}
return fmt.Errorf("failed to save event %s: %w", evt.ID, err)
}
// delete ephemeral events after a minute
go func() {
time.Sleep(75 * time.Second)
b.DB.Exec("DELETE FROM event WHERE id = $1", evt.ID)
}()
return nil
}