Merge pull request #304 from blinklabs-io/fix/ledger-address-stake-address

fix: generate stake address for script address

Author: agaffney

ledger/common.go

@@ -251,8 +251,8 @@ func (a *Address) MarshalCBOR() ([]byte, error) {
 }
 
 // StakeAddress returns a new Address with only the stake key portion. This will return nil if the address is not a payment/staking key pair
-func (a *Address) StakeAddress() *Address {
-	if a.addressType != addressTypeKeyKey {
+func (a Address) StakeAddress() *Address {
+	if a.addressType != addressTypeKeyKey && a.addressType != addressTypeScriptKey {
 		return nil
 	}
 	newAddr := &Address{

ledger/common_test.go

@@ -152,6 +152,11 @@ func TestAddressStakeAddress(t *testing.T) {
 			address:              "addr1q9h4f2vhh5vnqgnsejan3psw6mj3a504fxlqm2eh3262qufesdvfs83ulr22vprsv9mwnt0vgkfwxlflxkns32twqzdqjpq2na",
 			expectedStakeAddress: "stake1uyucxkycrc70349xq3cxzahf4hkytyhr05lntfcg49hqpxsqlrayk",
 		},
+		// Script address
+		{
+			address:              "addr1z8snz7c4974vzdpxu65ruphl3zjdvtxw8strf2c2tmqnxz2j2c79gy9l76sdg0xwhd7r0c0kna0tycz4y5s6mlenh8pq0xmsha",
+			expectedStakeAddress: "stake1u9f9v0z5zzlldgx58n8tklphu8mf7h4jvp2j2gddluemnssjfnkzz",
+		},
 	}
 	for _, testDef := range testDefs {
 		addr, err := NewAddress(testDef.address)