87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
|
package walletaccounts
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
"github.com/ethereum/go-ethereum/event"
|
||
|
|
"github.com/ethereum/go-ethereum/log"
|
||
|
|
"github.com/status-im/status-go/multiaccounts/accounts"
|
||
|
|
"github.com/status-im/status-go/services/accounts/accountsevent"
|
||
|
|
"github.com/status-im/status-go/services/wallet/async"
|
||
|
|
)
|
||
|
|
|
||
|
|
type AccountsChangeCb func(changedAddresses []common.Address, eventType accountsevent.EventType, currentAddresses []common.Address)
|
||
|
|
|
||
|
|
// Watcher executes a given callback whenever an account gets added/removed
|
||
|
|
type Watcher struct {
|
||
|
|
accountsDB *accounts.Database
|
||
|
|
accountFeed *event.Feed
|
||
|
|
group *async.Group
|
||
|
|
callback AccountsChangeCb
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewWatcher(accountsDB *accounts.Database, accountFeed *event.Feed, callback AccountsChangeCb) *Watcher {
|
||
|
|
return &Watcher{
|
||
|
|
accountsDB: accountsDB,
|
||
|
|
accountFeed: accountFeed,
|
||
|
|
callback: callback,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *Watcher) Start() {
|
||
|
|
if w.group != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
w.group = async.NewGroup(context.Background())
|
||
|
|
w.group.Add(func(ctx context.Context) error {
|
||
|
|
return watch(ctx, w.accountsDB, w.accountFeed, w.callback)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *Watcher) Stop() {
|
||
|
|
if w.group != nil {
|
||
|
|
w.group.Stop()
|
||
|
|
w.group.Wait()
|
||
|
|
w.group = nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func onAccountsChange(accountsDB *accounts.Database, callback AccountsChangeCb, changedAddresses []common.Address, eventType accountsevent.EventType) {
|
||
|
|
currentEthAddresses, err := accountsDB.GetWalletAddresses()
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
log.Error("failed getting wallet addresses", "error", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
currentAddresses := make([]common.Address, 0, len(currentEthAddresses))
|
||
|
|
for _, ethAddress := range currentEthAddresses {
|
||
|
|
currentAddresses = append(currentAddresses, common.Address(ethAddress))
|
||
|
|
}
|
||
|
|
|
||
|
|
if callback != nil {
|
||
|
|
callback(changedAddresses, eventType, currentAddresses)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func watch(ctx context.Context, accountsDB *accounts.Database, accountFeed *event.Feed, callback AccountsChangeCb) error {
|
||
|
|
ch := make(chan accountsevent.Event, 1)
|
||
|
|
sub := accountFeed.Subscribe(ch)
|
||
|
|
defer sub.Unsubscribe()
|
||
|
|
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case <-ctx.Done():
|
||
|
|
return nil
|
||
|
|
case err := <-sub.Err():
|
||
|
|
if err != nil {
|
||
|
|
log.Error("accounts watcher subscription failed", "error", err)
|
||
|
|
}
|
||
|
|
case ev := <-ch:
|
||
|
|
onAccountsChange(accountsDB, callback, ev.Accounts, ev.Type)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|