98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package khatru
|
|
|
|
import (
|
|
"context"
|
|
|
|
"fiatjaf.com/nostr"
|
|
)
|
|
|
|
const (
|
|
wsKey = iota
|
|
subscriptionIdKey
|
|
nip86HeaderAuthKey
|
|
internalCallKey
|
|
)
|
|
|
|
func RequestAuth(ctx context.Context) {
|
|
ws := GetConnection(ctx)
|
|
ws.WriteJSON(nostr.AuthEnvelope{Challenge: &ws.Challenge})
|
|
}
|
|
|
|
func GetConnection(ctx context.Context) *WebSocket {
|
|
wsi := ctx.Value(wsKey)
|
|
if wsi != nil {
|
|
return wsi.(*WebSocket)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetAuthed returns the last pubkey to have authenticated. Returns false if no one has.
|
|
//
|
|
// In a NIP-86 context it returns the single pubkey that have authenticated for that specific method call.
|
|
func GetAuthed(ctx context.Context) (nostr.PubKey, bool) {
|
|
if conn := GetConnection(ctx); conn != nil {
|
|
total := len(conn.AuthedPublicKeys)
|
|
if total == 0 {
|
|
return nostr.ZeroPK, false
|
|
}
|
|
return conn.AuthedPublicKeys[total-1], true
|
|
}
|
|
if nip86Auth := ctx.Value(nip86HeaderAuthKey); nip86Auth != nil {
|
|
return nip86Auth.(nostr.PubKey), true
|
|
}
|
|
return nostr.ZeroPK, false
|
|
}
|
|
|
|
// GetAllAuthed returns all authenticated public keys.
|
|
//
|
|
// In a NIP-86 context it returns the single pubkey that authenticated for that method call.
|
|
func GetAllAuthed(ctx context.Context) []nostr.PubKey {
|
|
if conn := GetConnection(ctx); conn != nil {
|
|
return conn.AuthedPublicKeys
|
|
}
|
|
if nip86Auth := ctx.Value(nip86HeaderAuthKey); nip86Auth != nil {
|
|
return []nostr.PubKey{nip86Auth.(nostr.PubKey)}
|
|
}
|
|
return []nostr.PubKey{}
|
|
}
|
|
|
|
// IsAuthed checks if the given public key is among the multiple that may have potentially authenticated.
|
|
func IsAuthed(ctx context.Context, pubkey nostr.PubKey) bool {
|
|
if conn := GetConnection(ctx); conn != nil {
|
|
for _, pk := range conn.AuthedPublicKeys {
|
|
if pk == pubkey {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
|
|
if nip86Auth := ctx.Value(nip86HeaderAuthKey); nip86Auth != nil {
|
|
return nip86Auth.(nostr.PubKey) == pubkey
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
// IsInternalCall returns true when a call to QueryEvents, for example, is being made because of a deletion
|
|
// or expiration request.
|
|
func IsInternalCall(ctx context.Context) bool {
|
|
return ctx.Value(internalCallKey) != nil
|
|
}
|
|
|
|
func GetIP(ctx context.Context) string {
|
|
conn := GetConnection(ctx)
|
|
if conn == nil {
|
|
return ""
|
|
}
|
|
|
|
return GetIPFromRequest(conn.Request)
|
|
}
|
|
|
|
func GetSubscriptionID(ctx context.Context) string {
|
|
return ctx.Value(subscriptionIdKey).(string)
|
|
}
|
|
|
|
func SendNotice(ctx context.Context, msg string) {
|
|
GetConnection(ctx).WriteJSON(nostr.NoticeEnvelope(msg))
|
|
}
|