package account

import (
	"time"

	"go.uber.org/zap"

	"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(logger *zap.Logger) *GethManager {
	m := &GethManager{}
	m.DefaultManager = &DefaultManager{
		accountsGenerator: generator.New(m),
		logger:            logger,
	}
	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
}