Skip to content

Commit 4aa291d

Browse files
authored
Merge pull request #6 from berty/fix/HasByAddress-crypto-Address
fix: HasByAddress should take crypto.Address, like GetByAddress
2 parents 1f680a2 + b20bcc2 commit 4aa291d

File tree

3 files changed

+8
-12
lines changed

3 files changed

+8
-12
lines changed

Diff for: tm2/pkg/crypto/keys/keybase.go

+5-9
Original file line numberDiff line numberDiff line change
@@ -170,11 +170,11 @@ func (kb dbKeybase) List() ([]Info, error) {
170170

171171
// HasByNameOrAddress checks if a key with the name or bech32 string address is in the keybase.
172172
func (kb dbKeybase) HasByNameOrAddress(nameOrBech32 string) (bool, error) {
173-
has, err := kb.HasByAddress(nameOrBech32)
173+
address, err := crypto.AddressFromBech32(nameOrBech32)
174174
if err != nil {
175175
return kb.HasByName(nameOrBech32)
176176
} else {
177-
return has, nil
177+
return kb.HasByAddress(address)
178178
}
179179
}
180180

@@ -183,13 +183,9 @@ func (kb dbKeybase) HasByName(name string) (bool, error) {
183183
return kb.db.Has(infoKey(name)), nil
184184
}
185185

186-
// HasByAddress checks if a key with the bech32 string address is in the keybase.
187-
func (kb dbKeybase) HasByAddress(bech32Address string) (bool, error) {
188-
addr, err := crypto.AddressFromBech32(bech32Address)
189-
if err != nil {
190-
return false, err
191-
}
192-
return kb.db.Has(addrKey(addr)), nil
186+
// HasByAddress checks if a key with the address is in the keybase.
187+
func (kb dbKeybase) HasByAddress(address crypto.Address) (bool, error) {
188+
return kb.db.Has(addrKey(address)), nil
193189
}
194190

195191
// Get returns the public information about one key.

Diff for: tm2/pkg/crypto/keys/lazy_keybase.go

+2-2
Original file line numberDiff line numberDiff line change
@@ -57,14 +57,14 @@ func (lkb lazyKeybase) HasByName(name string) (bool, error) {
5757
return NewDBKeybase(db).HasByName(name)
5858
}
5959

60-
func (lkb lazyKeybase) HasByAddress(bech32Address string) (bool, error) {
60+
func (lkb lazyKeybase) HasByAddress(address crypto.Address) (bool, error) {
6161
db, err := db.NewDB(lkb.name, dbBackend, lkb.dir)
6262
if err != nil {
6363
return false, err
6464
}
6565
defer db.Close()
6666

67-
return NewDBKeybase(db).HasByAddress(bech32Address)
67+
return NewDBKeybase(db).HasByAddress(address)
6868
}
6969

7070
func (lkb lazyKeybase) GetByNameOrAddress(nameOrBech32 string) (Info, error) {

Diff for: tm2/pkg/crypto/keys/types.go

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ type Keybase interface {
1515
List() ([]Info, error)
1616
HasByNameOrAddress(nameOrBech32 string) (bool, error)
1717
HasByName(name string) (bool, error)
18-
HasByAddress(bech32Address string) (bool, error)
18+
HasByAddress(address crypto.Address) (bool, error)
1919
GetByNameOrAddress(nameOrBech32 string) (Info, error)
2020
GetByName(name string) (Info, error)
2121
GetByAddress(address crypto.Address) (Info, error)

0 commit comments

Comments
 (0)