/
stakepool.go
37 lines (32 loc) · 1.09 KB
/
stakepool.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
// Copyright (c) 2016 The Decred developers
// Use of this source code is governed by an ISC
// license that can be found in the LICENSE file.
package wallet
import (
"github.com/decred/dcrd/dcrutil"
"github.com/decred/dcrwallet/errors"
"github.com/decred/dcrwallet/wallet/v2/udb"
"github.com/decred/dcrwallet/wallet/v2/walletdb"
)
// StakePoolUserInfo returns the stake pool user information for a user
// identified by their P2SH voting address.
func (w *Wallet) StakePoolUserInfo(userAddress dcrutil.Address) (*udb.StakePoolUser, error) {
const op errors.Op = "wallet.StakePoolUserInfo"
switch userAddress.(type) {
case *dcrutil.AddressPubKeyHash: // ok
case *dcrutil.AddressScriptHash: // ok
default:
return nil, errors.E(op, errors.Invalid, "address must be P2PKH or P2SH")
}
var user *udb.StakePoolUser
err := walletdb.View(w.db, func(tx walletdb.ReadTx) error {
stakemgrNs := tx.ReadBucket(wstakemgrNamespaceKey)
var err error
user, err = w.StakeMgr.StakePoolUserInfo(stakemgrNs, userAddress)
return err
})
if err != nil {
return nil, errors.E(op, err)
}
return user, nil
}