keyer, nip17, nip44, nip59: this time is different!

This commit is contained in:
fiatjaf
2024-09-15 11:06:42 -03:00
parent f976296e01
commit db023e12e9
5 changed files with 67 additions and 55 deletions

View File

@@ -5,6 +5,7 @@ import (
"github.com/nbd-wtf/go-nostr"
"github.com/nbd-wtf/go-nostr/nip44"
"github.com/puzpuzpuz/xsync/v3"
)
// Keysigner is a signer that holds the private key in memory and can do all the operations instantly and easily.
@@ -12,34 +13,34 @@ type KeySigner struct {
sk string
pk string
conversationKeys map[string][32]byte
conversationKeys *xsync.MapOf[string, [32]byte]
}
func (ks KeySigner) SignEvent(ctx context.Context, evt *nostr.Event) error { return evt.Sign(ks.sk) }
func (ks KeySigner) GetPublicKey(ctx context.Context) string { return ks.pk }
func (ks KeySigner) Encrypt(ctx context.Context, plaintext string, recipient string) (string, error) {
ck, ok := ks.conversationKeys[recipient]
ck, ok := ks.conversationKeys.Load(recipient)
if !ok {
var err error
ck, err = nip44.GenerateConversationKey(recipient, ks.sk)
if err != nil {
return "", err
}
ks.conversationKeys[recipient] = ck
ks.conversationKeys.Store(recipient, ck)
}
return nip44.Encrypt(plaintext, ck)
}
func (ks KeySigner) Decrypt(ctx context.Context, base64ciphertext string, sender string) (string, error) {
ck, ok := ks.conversationKeys[sender]
ck, ok := ks.conversationKeys.Load(sender)
if !ok {
var err error
ck, err = nip44.GenerateConversationKey(sender, ks.sk)
if err != nil {
return "", err
}
ks.conversationKeys[sender] = ck
ks.conversationKeys.Store(sender, ck)
}
return nip44.Decrypt(base64ciphertext, ck)
}