feat: created genesis pools from shelly genesis

Signed-off-by: Jenita <[email protected]>

Author: Jenita

ledger/common/certs.go

@@ -19,7 +19,6 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"math/big"
 	"net"
 
 	"github.com/blinklabs-io/gouroboros/cbor"
@@ -383,7 +382,7 @@ type PoolRegistrationCertificate struct {
 	VrfKeyHash           VrfKeyHash    `json:"vrfKeyHash"`
 	Pledge               uint64        `json:"pledge"`
 	Cost                 uint64        `json:"cost"`
-	Margin               cbor.Rat      `json:"margin"`
+	Margin               GenesisRat    `json:"margin"`
 	RewardAccount        AddrKeyHash   `json:"rewardAccount"`
 	PoolOwners           []AddrKeyHash `json:"poolOwners"`
 	Relays               []PoolRelay   `json:"relays"`
@@ -406,7 +405,6 @@ func (p *PoolRegistrationCertificate) UnmarshalJSON(data []byte) error {
 			Ipv6     *net.IP `json:"ipv6,omitempty"`
 			Hostname *string `json:"hostname,omitempty"`
 		} `json:"relays"`
-
 		PoolMetadata *PoolMetadata `json:"poolMetadata,omitempty"`
 	}
 
@@ -432,35 +430,18 @@ func (p *PoolRegistrationCertificate) UnmarshalJSON(data []byte) error {
 
 	// Handle margin field
 	if len(tmp.Margin) > 0 {
-		var marginValue interface{}
-		if err := json.Unmarshal(tmp.Margin, &marginValue); err != nil {
+		if err := p.Margin.UnmarshalJSON(tmp.Margin); err != nil {
 			return fmt.Errorf("failed to unmarshal margin: %w", err)
 		}
-
-		switch v := marginValue.(type) {
-		case float64:
-			p.Margin.Rat = new(big.Rat).SetFloat64(v)
-		case []interface{}:
-			if len(v) == 2 {
-				if num, ok := v[0].(float64); ok {
-					if den, ok := v[1].(float64); ok && den != 0 {
-						p.Margin.Rat = new(big.Rat).SetFrac64(int64(num), int64(den))
-					}
-				}
-			}
-		}
 	}
 
 	// Handle reward account
 	if len(tmp.RewardAccount) > 0 {
-		type credential struct {
-			KeyHash string `json:"key hash"`
-		}
-		type rewardAccount struct {
-			Credential credential `json:"credential"`
+		var ra struct {
+			Credential struct {
+				KeyHash string `json:"key hash"`
+			} `json:"credential"`
 		}
-
-		var ra rewardAccount
 		if err := json.Unmarshal(tmp.RewardAccount, &ra); err != nil {
 			return fmt.Errorf("failed to unmarshal reward account: %w", err)
 		}
@@ -470,30 +451,29 @@ func (p *PoolRegistrationCertificate) UnmarshalJSON(data []byte) error {
 			if err != nil {
 				return fmt.Errorf("failed to decode reward account key hash: %w", err)
 			}
-			if len(hashBytes) != 28 {
-				return fmt.Errorf("invalid key hash length: expected 28, got %d", len(hashBytes))
+			if len(hashBytes) != AddressHashSize {
+				return fmt.Errorf("invalid key hash length: expected %d, got %d", AddressHashSize, len(hashBytes))
 			}
-			var hash Blake2b224
-			copy(hash[:], hashBytes)
-			p.RewardAccount = AddrKeyHash(hash)
+			p.RewardAccount = AddrKeyHash(NewBlake2b224(hashBytes))
 		}
 	}
 
-	// Convert string fields to binary types
+	// Convert operator key
 	if tmp.Operator != "" {
 		opBytes, err := hex.DecodeString(tmp.Operator)
 		if err != nil {
 			return fmt.Errorf("invalid operator key: %w", err)
 		}
-		p.Operator = PoolKeyHash(Blake2b224(opBytes))
+		p.Operator = PoolKeyHash(NewBlake2b224(opBytes))
 	}
 
+	// Convert VRF key hash
 	if tmp.VrfKeyHash != "" {
 		vrfBytes, err := hex.DecodeString(tmp.VrfKeyHash)
 		if err != nil {
 			return fmt.Errorf("invalid VRF key hash: %w", err)
 		}
-		p.VrfKeyHash = VrfKeyHash(Blake2b256(vrfBytes))
+		p.VrfKeyHash = VrfKeyHash(NewBlake2b256(vrfBytes))
 	}
 
 	// Convert pool owners
@@ -504,7 +484,7 @@ func (p *PoolRegistrationCertificate) UnmarshalJSON(data []byte) error {
 			if err != nil {
 				return fmt.Errorf("invalid pool owner key: %w", err)
 			}
-			owners[i] = AddrKeyHash(Blake2b224(ownerBytes))
+			owners[i] = AddrKeyHash(NewBlake2b224(ownerBytes))
 		}
 		p.PoolOwners = owners
 	}

ledger/shelley/genesis.go

@@ -22,12 +22,49 @@ import (
 	"math/big"
 	"os"
 	"reflect"
+	"sync"
 	"time"
 
 	"github.com/blinklabs-io/gouroboros/cbor"
 	"github.com/blinklabs-io/gouroboros/ledger/common"
 )
 
+// Network stake address headers
+var (
+	stakeHeaderRegistry = map[string]struct {
+		Base, Script byte
+	}{
+		"Mainnet": {0xE0, 0xE1},
+		"Testnet": {0xF0, 0xF1},
+	}
+	stakeHeaderMutex sync.RWMutex
+)
+
+// RegisterStakeHeaders allows runtime registration of network address headers
+func RegisterStakeHeaders(networkId string, baseHeader, scriptHeader byte) {
+	stakeHeaderMutex.Lock()
+	defer stakeHeaderMutex.Unlock()
+	stakeHeaderRegistry[networkId] = struct{ Base, Script byte }{
+		Base:   baseHeader,
+		Script: scriptHeader,
+	}
+}
+
+func getStakeAddressHeader(networkId string, isScript bool) (byte, error) {
+	stakeHeaderMutex.RLock()
+	defer stakeHeaderMutex.RUnlock()
+
+	headers, exists := stakeHeaderRegistry[networkId]
+	if !exists {
+		return 0, errors.New("network not registered in stake header registry")
+	}
+
+	if isScript {
+		return headers.Script, nil
+	}
+	return headers.Base, nil
+}
+
 type ShelleyGenesis struct {
 	cbor.StructAsArray
 	SystemStart        time.Time                    `json:"systemStart"`
@@ -232,7 +269,6 @@ func (g *ShelleyGenesis) GenesisUtxos() ([]common.Utxo, error) {
 	return ret, nil
 }
 
-// InitialPools returns all pools and their delegators from the genesis data
 func (g *ShelleyGenesis) InitialPools() (map[string]common.PoolRegistrationCertificate, map[string][]common.Address, error) {
 	pools := make(map[string]common.PoolRegistrationCertificate)
 	poolStake := make(map[string][]common.Address)
@@ -241,6 +277,11 @@ func (g *ShelleyGenesis) InitialPools() (map[string]common.PoolRegistrationCerti
 		return pools, poolStake, nil
 	}
 
+	headerByte, err := getStakeAddressHeader(g.NetworkId, true)
+	if err != nil {
+		return nil, nil, errors.New("failed to get stake address header")
+	}
+
 	for stakeAddr, poolId := range g.Staking.Stake {
 		if len(stakeAddr) != 56 {
 			return nil, nil, errors.New("invalid stake address length")
@@ -251,7 +292,7 @@ func (g *ShelleyGenesis) InitialPools() (map[string]common.PoolRegistrationCerti
 			return nil, nil, errors.New("failed to decode stake key")
 		}
 
-		stakeAddrBytes := append([]byte{0xE1}, stakeKeyBytes...)
+		stakeAddrBytes := append([]byte{headerByte}, stakeKeyBytes...)
 		addr, err := common.NewAddressFromBytes(stakeAddrBytes)
 		if err != nil {
 			return nil, nil, errors.New("failed to create stake address")
@@ -286,7 +327,6 @@ func (g *ShelleyGenesis) InitialPools() (map[string]common.PoolRegistrationCerti
 	return pools, poolStake, nil
 }
 
-// PoolById returns a specific pool by its ID along with its delegators
 func (g *ShelleyGenesis) PoolById(poolId string) (*common.PoolRegistrationCertificate, []common.Address, error) {
 	if len(poolId) != 56 {
 		return nil, nil, errors.New("invalid pool ID length")
@@ -297,9 +337,9 @@ func (g *ShelleyGenesis) PoolById(poolId string) (*common.PoolRegistrationCertif
 		return nil, nil, errors.New("pool not found")
 	}
 
-	operatorBytes, err := hex.DecodeString(poolId)
+	headerByte, err := getStakeAddressHeader(g.NetworkId, true)
 	if err != nil {
-		return nil, nil, errors.New("failed to decode pool operator key")
+		return nil, nil, errors.New("failed to get stake address header")
 	}
 
 	var delegators []common.Address
@@ -314,7 +354,7 @@ func (g *ShelleyGenesis) PoolById(poolId string) (*common.PoolRegistrationCertif
 				return nil, nil, errors.New("failed to decode stake key")
 			}
 
-			stakeAddrBytes := append([]byte{0xE1}, stakeKeyBytes...)
+			stakeAddrBytes := append([]byte{headerByte}, stakeKeyBytes...)
 			addr, err := common.NewAddressFromBytes(stakeAddrBytes)
 			if err != nil {
 				return nil, nil, errors.New("failed to create stake address")
@@ -324,6 +364,11 @@ func (g *ShelleyGenesis) PoolById(poolId string) (*common.PoolRegistrationCertif
 		}
 	}
 
+	operatorBytes, err := hex.DecodeString(poolId)
+	if err != nil {
+		return nil, nil, errors.New("failed to decode pool operator key")
+	}
+
 	return &common.PoolRegistrationCertificate{
 		Operator:      common.Blake2b224(operatorBytes),
 		VrfKeyHash:    pool.VrfKeyHash,

ledger/shelley/genesis_test.go

@@ -27,6 +27,12 @@ import (
 	"github.com/blinklabs-io/gouroboros/ledger/shelley"
 )
 
+func init() {
+	// Initialize network headers for tests
+	shelley.RegisterStakeHeaders("Mainnet", 0xE0, 0xE1)
+	shelley.RegisterStakeHeaders("Testnet", 0xF0, 0xF1)
+}
+
 const shelleyGenesisConfig = `
 {
   "activeSlotsCoeff": 0.05,
@@ -256,6 +262,7 @@ func TestGenesisStaking(t *testing.T) {
 	const testGenesis = `{
         "systemStart": "2017-09-23T21:44:51Z",
         "networkMagic": 764824073,
+        "networkId": "Testnet",
         "staking": {
             "pools": {
                 "0aedc455785463235311c990f68742c9043cd79af09ab31c2ba5e195": {
@@ -327,8 +334,8 @@ func TestGenesisStaking(t *testing.T) {
 		} else {
 			// Extract stake key from address
 			addrBytes, _ := delegs[0].Bytes()
-			if len(addrBytes) != 29 || addrBytes[0] != 0xE1 {
-				t.Error("Delegator address is not in expected stake address format")
+			if len(addrBytes) != 29 || addrBytes[0] != 0xF1 { // Testnet script stake address
+				t.Errorf("Delegator address is not in expected stake address format: got %x", addrBytes[0])
 			} else {
 				stakeKey := hex.EncodeToString(addrBytes[1:])
 				if stakeKey != expectedStakeKey {
@@ -364,8 +371,8 @@ func TestGenesisStaking(t *testing.T) {
 		} else {
 			// Extract stake key from address
 			addrBytes, _ := delegators[0].Bytes()
-			if len(addrBytes) != 29 || addrBytes[0] != 0xE1 {
-				t.Error("Delegator address is not in expected stake address format")
+			if len(addrBytes) != 29 || addrBytes[0] != 0xF1 { // Testnet script stake address
+				t.Errorf("Delegator address is not in expected stake address format: got %x", addrBytes[0])
 			} else {
 				stakeKey := hex.EncodeToString(addrBytes[1:])
 				if stakeKey != expectedStakeKey {