hardfork test: implement advanced fork procedure

Author: glyh

src/app/hardfork_test/src/internal/hardfork/fork_config.go

@@ -43,7 +43,7 @@ func (t *HardforkTest) GetForkConfig(port int) ([]byte, error) {
 }
 
 // CreateRuntimeConfig creates the runtime configuration for the fork
-func (t *HardforkTest) CreateRuntimeConfig(forkGenesisTimestamp, forkConfigPath, configFile, baseConfigFile, forkHashesFile string) ([]byte, error) {
+func (t *HardforkTest) PatchRuntimeConfigLegacy(forkGenesisTimestamp, forkConfigPath, configFile, baseConfigFile, forkHashesFile string) ([]byte, error) {
 	cmd := exec.Command(filepath.Join(t.ScriptDir, "create_runtime_config.sh"))
 	cmd.Env = append(os.Environ(),
 		"GENESIS_TIMESTAMP="+forkGenesisTimestamp,

src/app/hardfork_test/src/internal/hardfork/ledger.go

@@ -64,19 +64,19 @@ func (t *HardforkTest) GenerateAndValidatePreforkLedgers(analysis *BlockAnalysis
 	return nil
 }
 
-// GenerateForkConfigAndLedgers does the following:
+// PatchForkConfigAndGenerateLedgersLegacy does the following:
 // 1. generate fork ledgers with runtime-genesis-ledger
 // 2. patch the genesis time & slot for fork config with create_runtime_config.sh
 // 3. perform some base sanity check on the fork config
-func (t *HardforkTest) GenerateForkConfigAndLedgers(analysis *BlockAnalysisResult, forkConfigPath, forkLedgersDir, forkHashesFile, configFile, preforkGenesisConfigFile string, forkGenesisTs, mainGenesisTs int64) error {
+func (t *HardforkTest) PatchForkConfigAndGenerateLedgersLegacy(analysis *BlockAnalysisResult, forkConfigPath, forkLedgersDir, forkHashesFile, configFile, preforkGenesisConfigFile string, forkGenesisTs, mainGenesisTs int64) error {
 	// Generate fork ledgers using fork network executable
 	if err := t.GenerateForkLedgers(t.Config.ForkRuntimeGenesisLedger, forkConfigPath, forkLedgersDir, forkHashesFile); err != nil {
 		return err
 	}
 
 	// Create runtime config
 	forkGenesisTimestamp := config.FormatTimestamp(forkGenesisTs)
-	runtimeConfigBytes, err := t.CreateRuntimeConfig(forkGenesisTimestamp, forkConfigPath, configFile, preforkGenesisConfigFile, forkHashesFile)
+	runtimeConfigBytes, err := t.PatchRuntimeConfigLegacy(forkGenesisTimestamp, forkConfigPath, configFile, preforkGenesisConfigFile, forkHashesFile)
 	if err != nil {
 		return err
 	}
@@ -100,5 +100,10 @@ func (t *HardforkTest) AdvancedGenerateHardForkConfig(configDir string) error {
 
 	cmd.Wait()
 
+	_, err := os.Stat(fmt.Sprintf("%s/activated"))
+	if err != nil {
+		return fmt.Errorf("failed to check on activated file for advanced generate fork config: %w", err)
+	}
+
 	return nil
 }

src/app/hardfork_test/src/internal/hardfork/phases.go

@@ -1,8 +1,13 @@
 package hardfork
 
 import (
+	"encoding/json"
 	"fmt"
+	"math"
 	"os"
+	"os/exec"
+	"path/filepath"
+	"strconv"
 	"time"
 )
 
@@ -113,7 +118,6 @@ func (t *HardforkTest) RunForkNetworkPhase(latestPreForkHeight int, forkData For
 
 func (t *HardforkTest) LegacyForkPhase(analysis *BlockAnalysisResult, forkConfigBytes []byte, mainGenesisTs int64) (*ForkData, error) {
 
-	// Define all localnet file paths
 	if err := os.MkdirAll("fork_data/prefork", 0755); err != nil {
 		return nil, err
 	}
@@ -150,11 +154,73 @@ func (t *HardforkTest) LegacyForkPhase(analysis *BlockAnalysisResult, forkConfig
 	{
 		preforkGenesisConfigFile := fmt.Sprintf("%s/daemon.json", t.Config.Root)
 		forkHashesFile := "fork_data/hf_ledger_hashes.json"
-		if err := t.GenerateForkConfigAndLedgers(analysis, preforkConfig, forkLedgersDir, forkHashesFile, postforkConfig, preforkGenesisConfigFile, forkGenesisTs, mainGenesisTs); err != nil {
+		if err := t.PatchForkConfigAndGenerateLedgersLegacy(analysis, preforkConfig, forkLedgersDir, forkHashesFile, postforkConfig, preforkGenesisConfigFile, forkGenesisTs, mainGenesisTs); err != nil {
 			return nil, err
 		}
 	}
 
 	return &ForkData{config: postforkConfig, ledgersDir: forkLedgersDir, genesis: forkGenesisTs}, nil
 
 }
+
+// Uses `mina advanced generate-hardfork-config CLI`
+func (t *HardforkTest) AdvancedForkPhase(analysis *BlockAnalysisResult, forkConfigBytes []byte, mainGenesisTs int64) (*ForkData, error) {
+
+	cwd := ""
+	var err error = nil
+	if cwd, err = os.Getwd(); err != nil {
+		return nil, err
+	}
+
+	forkDataPath := fmt.Sprintf("%s/fork_data", cwd)
+	if err := os.MkdirAll(forkDataPath, 0755); err != nil {
+		return nil, err
+	}
+
+	if err := t.AdvancedGenerateHardForkConfig(forkDataPath); err != nil {
+		return nil, err
+	}
+
+	configToPatch := fmt.Sprintf("%s/daemon.json", forkDataPath)
+
+	configJsonString, err := os.ReadFile(configToPatch)
+	if err != nil {
+		return nil, err
+	}
+
+	var configRaw map[string]map[string]string
+	json.Unmarshal(configJsonString, &configRaw)
+
+	preforkChainEndTs, err := time.Parse(time.RFC3339Nano, configRaw["genesis"]["genesis_state_timestamp"])
+	if err != nil {
+		return nil, err
+	}
+	roughForkGenesisTs := time.Now().Add(time.Duration(t.Config.ForkDelay) * time.Minute)
+	hardforkGenesisSecDelta := roughForkGenesisTs.Sub(preforkChainEndTs).Seconds()
+	hardforkGenesisSlotDelta := int(math.Ceil(hardforkGenesisSecDelta / float64(t.Config.MainSlot)))
+	forkSlotDuration := time.Duration(t.Config.MainSlot) * time.Second
+	forkGenesisTs := preforkChainEndTs.Add(time.Duration(hardforkGenesisSlotDelta) * forkSlotDuration).Unix()
+
+	cmd := exec.Command(filepath.Join(t.ScriptDir, "patch-runtime-config-advanced-gen-fork-config.sh"))
+
+	cmd.Env = append(os.Environ(),
+		"FORK_CONFIG_TO_PATCH="+configToPatch,
+		"PREFORK_SLOT_TIME_SEC="+strconv.Itoa(t.Config.MainSlot),
+		"HARDFORK_GENESIS_SLOT_DELTA="+strconv.Itoa(hardforkGenesisSlotDelta),
+	)
+	cmd.Stderr = os.Stderr
+
+	if err := cmd.Run(); err != nil {
+		return nil, fmt.Errorf("failed to run patch-runtime-config-advanced-gen-fork-config.sh: %w", err)
+	}
+
+	patchedConfigString, err := cmd.Output()
+
+	err = os.WriteFile(configToPatch, patchedConfigString, 0644)
+	if err != nil {
+		return nil, fmt.Errorf("failed to patch fork config: %w", err)
+	}
+
+	forkLedgersDir := fmt.Sprintf("%s/genesis", forkDataPath)
+	return &ForkData{config: configToPatch, ledgersDir: forkLedgersDir, genesis: forkGenesisTs}, nil
+}