feat: Add a function for migration configuration

fixtures/networks/net_2k.go

@@ -27,26 +27,25 @@ func Net2k() *NetworkConf {
 			MinVerifiedDealSize:     256,
 			PreCommitChallengeDelay: abi.ChainEpoch(10),
 			ForkUpgradeParam: &config.ForkUpgradeConfig{
-				UpgradeBreezeHeight:     -1,
-				UpgradeSmokeHeight:      -2,
-				UpgradeIgnitionHeight:   -3,
-				UpgradeRefuelHeight:     -4,
-				UpgradeTapeHeight:       -5,
-				UpgradeAssemblyHeight:   -6,
-				UpgradeLiftoffHeight:    -7,
-				UpgradeKumquatHeight:    -8,
-				UpgradeCalicoHeight:     -9,
-				UpgradePersianHeight:    -10,
-				UpgradeOrangeHeight:     -11,
-				UpgradeTrustHeight:      -13,
-				UpgradeNorwegianHeight:  -14,
-				UpgradeTurboHeight:      -15,
-				UpgradeHyperdriveHeight: -16,
-				UpgradeChocolateHeight:  -17,
-				UpgradeOhSnapHeight:     -18,
-
 				BreezeGasTampingDuration: 0,
-				UpgradeClausHeight:       -11,
+				UpgradeBreezeHeight:      -1,
+				UpgradeSmokeHeight:       -2,
+				UpgradeIgnitionHeight:    -3,
+				UpgradeRefuelHeight:      -4,
+				UpgradeTapeHeight:        -5,
+				UpgradeAssemblyHeight:    -6,
+				UpgradeLiftoffHeight:     -7,
+				UpgradeKumquatHeight:     -8,
+				UpgradeCalicoHeight:      -9,
+				UpgradePersianHeight:     -10,
+				UpgradeOrangeHeight:      -11,
+				UpgradeClausHeight:       -12,
+				UpgradeTrustHeight:       -13,
+				UpgradeNorwegianHeight:   -14,
+				UpgradeTurboHeight:       -15,
+				UpgradeHyperdriveHeight:  -16,
+				UpgradeChocolateHeight:   -17,
+				UpgradeOhSnapHeight:      -18,
 				UpgradeSkyrHeight:        -19,
 				UpgradeV17Height:         99999999999999,
 			},

pkg/migration/migrate.go

@@ -31,6 +31,7 @@ var versionMap = []versionInfo{
 	{version: 6, upgrade: Version6Upgrade},
 	{version: 7, upgrade: Version7Upgrade},
 	{version: 8, upgrade: Version8Upgrade},
+	{version: 9, upgrade: Version9Upgrade},
 }
 
 // TryToMigrate used to migrate data(db,config,file,etc) in local repo
@@ -121,7 +122,7 @@ func Version4Upgrade(repoPath string) (err error) {
 	return repo.WriteVersion(repoPath, 4)
 }
 
-//Version5Upgrade
+// Version5Upgrade
 func Version5Upgrade(repoPath string) (err error) {
 	var fsrRepo repo.Repo
 	if fsrRepo, err = repo.OpenFSRepo(repoPath, 4); err != nil {
@@ -159,7 +160,7 @@ func Version5Upgrade(repoPath string) (err error) {
 	return repo.WriteVersion(repoPath, 5)
 }
 
-//Version6Upgrade
+// Version6Upgrade
 func Version6Upgrade(repoPath string) (err error) {
 	var fsrRepo repo.Repo
 	if fsrRepo, err = repo.OpenFSRepo(repoPath, 5); err != nil {
@@ -197,7 +198,7 @@ func Version6Upgrade(repoPath string) (err error) {
 	return repo.WriteVersion(repoPath, 6)
 }
 
-//Version7Upgrade
+// Version7Upgrade
 func Version7Upgrade(repoPath string) (err error) {
 	var fsrRepo repo.Repo
 	if fsrRepo, err = repo.OpenFSRepo(repoPath, 6); err != nil {
@@ -259,7 +260,7 @@ func Version7Upgrade(repoPath string) (err error) {
 	return repo.WriteVersion(repoPath, 7)
 }
 
-//Version8Upgrade
+// Version8Upgrade
 func Version8Upgrade(repoPath string) (err error) {
 	var fsrRepo repo.Repo
 	if fsrRepo, err = repo.OpenFSRepo(repoPath, 7); err != nil {
@@ -281,10 +282,10 @@ func Version8Upgrade(repoPath string) (err error) {
 		cfg.NetworkParams.ForkUpgradeParam.UpgradeSkyrHeight = -19
 	case types.NetworkInterop:
 		cfg.NetworkParams.GenesisNetworkVersion = network.Version15
-		cfg.NetworkParams.ForkUpgradeParam.UpgradeSkyrHeight = 100
+		cfg.NetworkParams.ForkUpgradeParam.UpgradeSkyrHeight = -19
 	case types.NetworkButterfly:
 		cfg.NetworkParams.GenesisNetworkVersion = network.Version15
-		cfg.NetworkParams.ForkUpgradeParam.UpgradeSkyrHeight = 50
+		cfg.NetworkParams.ForkUpgradeParam.UpgradeSkyrHeight = -19
 	default:
 		return fsrRepo.Close()
 	}
@@ -299,3 +300,44 @@ func Version8Upgrade(repoPath string) (err error) {
 
 	return repo.WriteVersion(repoPath, 8)
 }
+
+// Version9Upgrade
+func Version9Upgrade(repoPath string) (err error) {
+	var fsrRepo repo.Repo
+	if fsrRepo, err = repo.OpenFSRepo(repoPath, 8); err != nil {
+		return
+	}
+	cfg := fsrRepo.Config()
+	switch cfg.NetworkParams.NetworkType {
+	case types.NetworkMainnet:
+		cfg.NetworkParams.GenesisNetworkVersion = network.Version0
+		cfg.NetworkParams.ForkUpgradeParam.UpgradeV17Height = 99999999999999
+	case types.Network2k:
+		cfg.NetworkParams.GenesisNetworkVersion = network.Version16
+		cfg.NetworkParams.ForkUpgradeParam.UpgradeV17Height = 99999999999999
+	case types.NetworkCalibnet:
+		cfg.NetworkParams.GenesisNetworkVersion = network.Version0
+		cfg.NetworkParams.ForkUpgradeParam.UpgradeV17Height = 99999999999999
+	case types.NetworkForce:
+		cfg.NetworkParams.GenesisNetworkVersion = network.Version16
+		cfg.NetworkParams.ForkUpgradeParam.UpgradeV17Height = 99999999999999
+	case types.NetworkInterop:
+		cfg.NetworkParams.GenesisNetworkVersion = network.Version16
+		cfg.NetworkParams.ForkUpgradeParam.UpgradeV17Height = 99999999999999
+	case types.NetworkButterfly:
+		cfg.NetworkParams.GenesisNetworkVersion = network.Version16
+		cfg.NetworkParams.ForkUpgradeParam.UpgradeV17Height = 99999999999999
+	default:
+		return fsrRepo.Close()
+	}
+
+	if err = fsrRepo.ReplaceConfig(cfg); err != nil {
+		return
+	}
+
+	if err = fsrRepo.Close(); err != nil {
+		return
+	}
+
+	return repo.WriteVersion(repoPath, 9)
+}

pkg/repo/fsrepo.go

@@ -26,7 +26,7 @@ import (
 )
 
 // Version is the version of repo schema that this code understands.
-const LatestVersion uint = 8
+const LatestVersion uint = 9
 
 const (
 	// apiFile is the filename containing the filecoin node's api address.
@@ -247,9 +247,6 @@ func (r *FSRepo) loadFromDisk() error {
 		return errors.Wrap(err, "failed to open paych datastore")
 	}
 
-	/*if err := r.openMarketDataStore(); err != nil {
-		return errors.Wrap(err, "failed to open market datastore")
-	}*/
 	return nil
 }
 
@@ -474,14 +471,6 @@ func (r *FSRepo) openPaychDataStore() error {
 	return nil
 }
 
-/*func (r *FSRepo) openMarketDataStore() error {
-	var err error
-	r.marketDs, err = badgerds.NewDatastore(filepath.Join(r.path, marketDatastoreProfix), badgerOptions())
-	if err != nil {
-		return err
-	}
-	return nil
-}*/
 func (r *FSRepo) openWalletDatastore() error {
 	// TODO: read wallet datastore info from config, use that to open it up
 	ds, err := badgerds.NewDatastore(filepath.Join(r.path, walletDatastorePrefix), badgerOptions())