implement logic to build genesis state for the sequencer

adapters/vm2core/vm2core.go

@@ -5,6 +5,7 @@
 	"slices"
 
 	"github.com/NethermindEth/juno/core"
+	"github.com/NethermindEth/juno/core/felt"
 	"github.com/NethermindEth/juno/utils"
 	"github.com/NethermindEth/juno/vm"
 	"github.com/ethereum/go-ethereum/common"
@@ -39,3 +40,50 @@
 	})
 	return utils.Map(events, AdaptOrderedEvent)
 }
+
+func AdaptStateDiff(fromStateDiff *vm.StateDiff) core.StateDiff {
+	var result core.StateDiff
+	if fromStateDiff == nil {
+		return result
+	}
+
+	// Preallocate all maps with known sizes from fromStateDiff
+	result = core.StateDiff{
+		StorageDiffs:      make(map[felt.Felt]map[felt.Felt]*felt.Felt, len(fromStateDiff.StorageDiffs)),
+		Nonces:            make(map[felt.Felt]*felt.Felt, len(fromStateDiff.Nonces)),
+		DeployedContracts: make(map[felt.Felt]*felt.Felt, len(fromStateDiff.DeployedContracts)),
+		DeclaredV0Classes: make([]*felt.Felt, len(fromStateDiff.DeprecatedDeclaredClasses)),
+		DeclaredV1Classes: make(map[felt.Felt]*felt.Felt, len(fromStateDiff.DeclaredClasses)),
+		ReplacedClasses:   make(map[felt.Felt]*felt.Felt, len(fromStateDiff.ReplacedClasses)),
+	}
+
+	for _, sd := range fromStateDiff.StorageDiffs {
+		entries := make(map[felt.Felt]*felt.Felt, len(sd.StorageEntries))
+		for _, entry := range sd.StorageEntries {
+			val := entry.Value
+			entries[entry.Key] = &val
+		}
+		result.StorageDiffs[sd.Address] = entries
+	}
+
+	for _, nonce := range fromStateDiff.Nonces {
+		newNonce := nonce.Nonce
+		result.Nonces[nonce.ContractAddress] = &newNonce
+	}
+
+	for _, dc := range fromStateDiff.DeployedContracts {
+		ch := dc.ClassHash
+		result.DeployedContracts[dc.Address] = &ch
+	}
+
+	for _, dc := range fromStateDiff.DeclaredClasses {
+		cch := dc.CompiledClassHash
+		result.DeclaredV1Classes[dc.ClassHash] = &cch
+	}
+
+	for _, rc := range fromStateDiff.ReplacedClasses {
+		ch := rc.ClassHash
+		result.ReplacedClasses[rc.ContractAddress] = &ch
+	}
+	return result
+}

core/state_update.go

@@ -24,6 +24,31 @@
 	ReplacedClasses   map[felt.Felt]*felt.Felt               // addr -> class hash
 }
 
+func (d *StateDiff) MergeStateDiffs(incoming *StateDiff) {
+	mergeMaps := func(oldMap, newMap map[felt.Felt]*felt.Felt) {
+		for key, value := range newMap {
+			oldMap[key] = value
+		}
+	}
+
+	mergeStorageDiffs := func(oldMap, newMap map[felt.Felt]map[felt.Felt]*felt.Felt) {
+		for addr, newAddrStorage := range newMap {
+			if oldAddrStorage, exists := oldMap[addr]; exists {
+				mergeMaps(oldAddrStorage, newAddrStorage)
+			} else {
+				oldMap[addr] = newAddrStorage
+			}
+		}
+	}
+
+	mergeStorageDiffs(d.StorageDiffs, incoming.StorageDiffs)
+	mergeMaps(d.Nonces, incoming.Nonces)
+	mergeMaps(d.DeployedContracts, incoming.DeployedContracts)
+	mergeMaps(d.DeclaredV1Classes, incoming.DeclaredV1Classes)
+	mergeMaps(d.ReplacedClasses, incoming.ReplacedClasses)
+	d.DeclaredV0Classes = append(d.DeclaredV0Classes, incoming.DeclaredV0Classes...)
+}
+
 func (d *StateDiff) Length() uint64 {
 	var length int
 
@@ -39,6 +64,24 @@
 	return uint64(length)
 }
 
+func (d *StateDiff) Merge(incoming *StateDiff) {
+	mergeStorageDiffs := func(oldMap, newMap map[felt.Felt]map[felt.Felt]*felt.Felt) {
+		for addr, newAddrStorage := range newMap {
+			if oldAddrStorage, exists := oldMap[addr]; exists {
+				maps.Copy(oldAddrStorage, newAddrStorage)
+			} else {
+				oldMap[addr] = newAddrStorage
+			}
+		}
+	}
+	maps.Copy(d.Nonces, incoming.Nonces)
+	maps.Copy(d.DeployedContracts, incoming.DeployedContracts)
+	maps.Copy(d.DeclaredV1Classes, incoming.DeclaredV1Classes)
+	maps.Copy(d.ReplacedClasses, incoming.ReplacedClasses)
+	mergeStorageDiffs(d.StorageDiffs, incoming.StorageDiffs)
+	d.DeclaredV0Classes = append(d.DeclaredV0Classes, incoming.DeclaredV0Classes...)
+}
+
 var starknetStateDiff0 = new(felt.Felt).SetBytes([]byte("STARKNET_STATE_DIFF0"))
 
 func (d *StateDiff) Hash() *felt.Felt {
@@ -199,3 +242,14 @@
 		digest.Update(&addr, nonces[addr])
 	}
 }
+
+func EmptyStateDiff() StateDiff {
+	return StateDiff{
+		StorageDiffs:      make(map[felt.Felt]map[felt.Felt]*felt.Felt),
+		Nonces:            make(map[felt.Felt]*felt.Felt),
+		DeployedContracts: make(map[felt.Felt]*felt.Felt),
+		DeclaredV0Classes: make([]*felt.Felt, 0),
+		DeclaredV1Classes: make(map[felt.Felt]*felt.Felt),
+		ReplacedClasses:   make(map[felt.Felt]*felt.Felt),
+	}
+}