status-im/status-go

package account

import (
    "time"

    "github.com/ethereum/go-ethereum/accounts"

    "github.com/status-im/status-go/account/generator"
    "github.com/status-im/status-go/rpc"
)

// GethManager represents account manager interface.
type GethManager struct {
    *DefaultManager

    gethAccManager *accounts.Manager
}

// NewGethManager returns new node account manager.
func NewGethManager() *GethManager {
    m := &GethManager{}
    m.DefaultManager = &DefaultManager{accountsGenerator: generator.New(m)}
    return m
}

func (m *GethManager) SetRPCClient(rpcClient *rpc.Client, rpcTimeout time.Duration) {
    m.DefaultManager.rpcClient = rpcClient
    m.DefaultManager.rpcTimeout = rpcTimeout
}

// InitKeystore sets key manager and key store.
func (m *GethManager) InitKeystore(keydir string) error {
    m.mu.Lock()
    defer m.mu.Unlock()

    var err error
    m.gethAccManager, err = makeAccountManager(keydir)
    if err != nil {
        return err
    }

    m.keystore, err = makeKeyStore(m.gethAccManager)
    m.Keydir = keydir
    return err
}

func (m *GethManager) GetManager() *accounts.Manager {
    m.mu.RLock()
    defer m.mu.RUnlock()
    return m.gethAccManager
}