-
Notifications
You must be signed in to change notification settings - Fork 2
/
key.go
106 lines (86 loc) · 2.14 KB
/
key.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package nostr
import (
"crypto/rand"
"encoding/hex"
"fmt"
"io"
"math/big"
"strings"
"github.com/btcsuite/btcd/btcec/v2"
"github.com/btcsuite/btcd/btcec/v2/schnorr"
"github.com/btcsuite/btcd/btcutil/bech32"
)
// NIP-01
func GeneratePrivateKey() string {
params := btcec.S256().Params()
one := new(big.Int).SetInt64(1)
b := make([]byte, params.BitSize/8+8)
if _, err := io.ReadFull(rand.Reader, b); err != nil {
return ""
}
k := new(big.Int).SetBytes(b)
n := new(big.Int).Sub(params.N, one)
k.Mod(k, n)
k.Add(k, one)
return hex.EncodeToString(k.Bytes())
}
// NIP-01
func GetPublicKey(sk string) (string, error) {
b, err := hex.DecodeString(sk)
if err != nil {
return "", err
}
_, pk := btcec.PrivKeyFromBytes(b)
return hex.EncodeToString(schnorr.SerializePubKey(pk)), nil
}
// NIP-01
func IsValidPublicKeyHex(pk string) bool {
if strings.ToLower(pk) != pk {
return false
}
dec, _ := hex.DecodeString(pk)
return len(dec) == 32
}
// NIP-19
func DecodeBech32(bech32string string) (value any, err error) {
prefix, bits5, err := bech32.DecodeNoLimit(bech32string)
if err != nil {
return nil, err
}
data, err := bech32.ConvertBits(bits5, 5, 8, false)
if err != nil {
return nil, fmt.Errorf("failed translating data into 8 bits: %s", err.Error())
}
switch prefix {
case "npub", "nsec":
if len(data) < 32 {
return nil, fmt.Errorf("data is less than 32 bytes (%d)", len(data))
}
return hex.EncodeToString(data[0:32]), nil
}
return data, fmt.Errorf("unknown tag %s", prefix)
}
// NIP-19
func EncodePrivateKey(privateKeyHex string) (string, error) {
b, err := hex.DecodeString(privateKeyHex)
if err != nil {
return "", fmt.Errorf("failed to decode private key hex: %w", err)
}
bits5, err := bech32.ConvertBits(b, 8, 5, true)
if err != nil {
return "", err
}
return bech32.Encode("nsec", bits5)
}
// NIP-19
func EncodePublicKey(publicKeyHex string) (string, error) {
b, err := hex.DecodeString(publicKeyHex)
if err != nil {
return "", fmt.Errorf("failed to decode public key hex: %w", err)
}
bits5, err := bech32.ConvertBits(b, 8, 5, true)
if err != nil {
return "", err
}
return bech32.Encode("npub", bits5)
}