chain_spec.go

package commander

import (
	"os"

	"github.com/Worldcoin/hubble-commander/models"
	"github.com/Worldcoin/hubble-commander/utils/ref"
	"gopkg.in/yaml.v2"
)

func ReadChainSpecFile(path string) (*models.ChainSpec, error) {
	yamlFile, err := os.ReadFile(path)
	if err != nil {
		return nil, err
	}

	var chainSpec models.ChainSpec
	err = yaml.Unmarshal(yamlFile, &chainSpec)
	if err != nil {
		return nil, err
	}

	return &chainSpec, nil
}

func GenerateChainSpec(chainState *models.ChainState) (*string, error) {
	chainSpec := makeChainSpec(chainState)

	yamlChainSpec, err := yaml.Marshal(chainSpec)
	if err != nil {
		return nil, err
	}

	return ref.String(string(yamlChainSpec)), nil
}

func makeChainSpec(chainState *models.ChainState) models.ChainSpec {
	return models.ChainSpec{
		ChainID:                        chainState.ChainID,
		AccountRegistry:                chainState.AccountRegistry,
		AccountRegistryDeploymentBlock: chainState.AccountRegistryDeploymentBlock,
		TokenRegistry:                  chainState.TokenRegistry,
		SpokeRegistry:                  chainState.SpokeRegistry,
		DepositManager:                 chainState.DepositManager,
		WithdrawManager:                chainState.WithdrawManager,
		Rollup:                         chainState.Rollup,
		GenesisAccounts:                chainState.GenesisAccounts,
	}
}

func newChainStateFromChainSpec(chainSpec *models.ChainSpec) *models.ChainState {
	return &models.ChainState{
		ChainID:                        chainSpec.ChainID,
		AccountRegistry:                chainSpec.AccountRegistry,
		AccountRegistryDeploymentBlock: chainSpec.AccountRegistryDeploymentBlock,
		TokenRegistry:                  chainSpec.TokenRegistry,
		SpokeRegistry:                  chainSpec.SpokeRegistry,
		DepositManager:                 chainSpec.DepositManager,
		WithdrawManager:                chainSpec.WithdrawManager,
		Rollup:                         chainSpec.Rollup,
		GenesisAccounts:                chainSpec.GenesisAccounts,
	}
}