-
Notifications
You must be signed in to change notification settings - Fork 1
/
utils.go
79 lines (67 loc) · 1.57 KB
/
utils.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
package wallet
import (
"crypto/rand"
"encoding/json"
"os"
"path/filepath"
"github.com/tyler-smith/go-bip39"
"github.com/zenon-network/go-zenon/node"
zwallet "github.com/zenon-network/go-zenon/wallet"
)
func WriteKeyFile(ks *zwallet.KeyStore, name, password string) error {
kf, err := ks.Encrypt(password)
if err != nil {
return err
}
if name == "" {
name = kf.BaseAddress.String()
}
kf.Path = filepath.Join(node.DefaultDataDir(), "wallet", name)
keyFileJson, err := json.MarshalIndent(kf, "", " ")
if err != nil {
return err
}
return os.WriteFile(kf.Path, keyFileJson, 0644)
}
func ReadKeyFile(name, password, path string) (*zwallet.KeyStore, error) {
var newPath string
if path == "" {
newPath = filepath.Join(node.DefaultDataDir(), "wallet", name)
} else {
newPath = path
}
kf, err := zwallet.ReadKeyFile(newPath)
if err != nil {
return nil, err
}
ks, err := kf.Decrypt(password)
if err != nil {
return nil, err
}
return ks, nil
}
func NewKeyStoreFromEntropy(entropy []byte) (*zwallet.KeyStore, error) {
mnemonic, err := bip39.NewMnemonic(entropy)
if err != nil {
return nil, err
}
ks := &zwallet.KeyStore{
Entropy: entropy,
Seed: bip39.NewSeed(mnemonic, ""),
Mnemonic: mnemonic,
}
// setup base address
if _, kp, err := ks.DeriveForIndexPath(0); err == nil {
ks.BaseAddress = kp.Address
} else {
return nil, err
}
return ks, nil
}
func NewKeyStore() (*zwallet.KeyStore, error) {
entropy := make([]byte, 32)
if _, err := rand.Read(entropy); err != nil {
return nil, err
}
return NewKeyStoreFromEntropy(entropy)
}