-
Notifications
You must be signed in to change notification settings - Fork 142
/
trusted_network_manager.go
81 lines (67 loc) · 2.65 KB
/
trusted_network_manager.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
// Copyright 2020 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0
package testutil
import (
"context"
"errors"
"github.com/iotaledger/hive.go/runtime/event"
"github.com/iotaledger/wasp/packages/cryptolib"
"github.com/iotaledger/wasp/packages/peering"
)
type trustedNetworkManager struct {
data map[cryptolib.PublicKeyKey]trustedNetworkDataEntry
changeEvents *event.Event1[[]*peering.TrustedPeer]
}
type trustedNetworkDataEntry struct {
name string
key *cryptolib.PublicKey
peeringURL string
}
var _ peering.TrustedNetworkManager = &trustedNetworkManager{}
func NewTrustedNetworkManager() peering.TrustedNetworkManager {
return &trustedNetworkManager{
data: map[cryptolib.PublicKeyKey]trustedNetworkDataEntry{},
changeEvents: event.New1[[]*peering.TrustedPeer](),
}
}
// IsTrustedPeer implements the peering.TrustedNetworkManager interface.
func (tnm *trustedNetworkManager) IsTrustedPeer(pubKey *cryptolib.PublicKey) error {
if _, ok := tnm.data[pubKey.AsKey()]; ok {
return nil
}
return errors.New("not trusted")
}
// TrustPeer implements the peering.TrustedNetworkManager interface.
func (tnm *trustedNetworkManager) TrustPeer(name string, pubKey *cryptolib.PublicKey, peeringURL string) (*peering.TrustedPeer, error) {
if err := peering.ValidateTrustedPeerParams(name, pubKey, peeringURL); err != nil {
return nil, err
}
tnm.data[pubKey.AsKey()] = trustedNetworkDataEntry{name, pubKey, peeringURL}
tnm.changeEvents.Trigger(tnm.mustTrustedPeers())
return peering.NewTrustedPeer(name, pubKey, peeringURL), nil
}
// DistrustPeer implements the peering.TrustedNetworkManager interface.
func (tnm *trustedNetworkManager) DistrustPeer(pubKey *cryptolib.PublicKey) (*peering.TrustedPeer, error) {
delete(tnm.data, pubKey.AsKey())
tnm.changeEvents.Trigger(tnm.mustTrustedPeers())
return nil, nil
}
// TrustedPeers implements the peering.TrustedNetworkManager interface.
func (tnm *trustedNetworkManager) TrustedPeers() ([]*peering.TrustedPeer, error) {
return tnm.mustTrustedPeers(), nil
}
func (tnm *trustedNetworkManager) TrustedPeersByPubKeyOrName(pubKeysOrNames []string) ([]*peering.TrustedPeer, error) {
return peering.QueryByPubKeyOrName(tnm.mustTrustedPeers(), pubKeysOrNames)
}
func (tnm *trustedNetworkManager) mustTrustedPeers() []*peering.TrustedPeer {
res := []*peering.TrustedPeer{}
for _, v := range tnm.data {
res = append(res, peering.NewTrustedPeer(v.name, v.key, v.peeringURL))
}
return res
}
func (tnm *trustedNetworkManager) TrustedPeersListener(callback func([]*peering.TrustedPeer)) context.CancelFunc {
callback(tnm.mustTrustedPeers())
unhook := tnm.changeEvents.Hook(callback).Unhook
return unhook
}