executor

type AggregatedStats struct {
	TotalTime        int64              `json:"time_total"`
	GasUsedTotal     uint64             `json:"gas_used_total"`
	GasUsedTimeTotal int64              `json:"gas_used_time_total"`
	Succeeded        int                `json:"success"`
	Failed           int                `json:"fail"`
	TotalMsgs        int                `json:"msg_count"`
	ResourceTotals   *ResourceTotals    `json:"resource_totals,omitempty"`
	MethodStats      *MethodsAggregated `json:"method_stats"`
}

type BlockLogCollector interface {
	RegisterBlockHash(testName, blockHash string)
}

type Config struct {
	Source                          *config.SourceConfig
	Filter                          string
	CacheDir                        string
	ResultsDir                      string
	ResultsOwner                    *fsutil.OwnerConfig // Optional file ownership for results directory
	SystemResourceCollectionEnabled bool                // Enable system resource collection (cgroups/Docker Stats)
	GitHubToken                     string              // Optional GitHub token for API-based artifact downloads
}

type EESTSource struct {
	// contains filtered or unexported fields
}

type EESTSourceInfo struct {
	GitHubRepo     string `json:"github_repo,omitempty"`
	GitHubRelease  string `json:"github_release,omitempty"`
	FixturesURL    string `json:"fixtures_url,omitempty"`
	GenesisURL     string `json:"genesis_url,omitempty"`
	FixturesSubdir string `json:"fixtures_subdir,omitempty"`
	// Artifact fields (alternative to releases).
	FixturesArtifactName  string `json:"fixtures_artifact_name,omitempty"`
	GenesisArtifactName   string `json:"genesis_artifact_name,omitempty"`
	FixturesArtifactRunID string `json:"fixtures_artifact_run_id,omitempty"`
	GenesisArtifactRunID  string `json:"genesis_artifact_run_id,omitempty"`
	// Local source fields.
	LocalFixturesDir     string `json:"local_fixtures_dir,omitempty"`
	LocalGenesisDir      string `json:"local_genesis_dir,omitempty"`
	LocalFixturesTarball string `json:"local_fixtures_tarball,omitempty"`
	LocalGenesisTarball  string `json:"local_genesis_tarball,omitempty"`
}

type ExecuteOptions struct {
	EngineEndpoint                string
	JWT                           string
	ResultsDir                    string
	Filter                        string
	ContainerID                   string                                // Container ID for stats collection.
	DockerClient                  *client.Client                        // Docker client for fallback stats reader.
	DropMemoryCaches              string                                // "tests", "steps", or "" (disabled).
	DropCachesPath                string                                // Path to drop_caches file (default: /proc/sys/vm/drop_caches).
	RollbackStrategy              string                                // "rpc-debug-setHead" or "" (disabled).
	RPCEndpoint                   string                                // RPC endpoint for rollback calls (e.g. http://host:port).
	ClientRPCRollbackSpec         *clientpkg.RPCRollbackSpec            // Client-specific rollback method and param format.
	Tests                         []*TestWithSteps                      // Optional subset of tests to run (nil = run all).
	BlockLogCollector             BlockLogCollector                     // Optional collector for capturing block logs from client.
	RetryNewPayloadsSyncingConfig *config.RetryNewPayloadsSyncingConfig // Retry config for SYNCING responses.
	PostTestRPCCalls              []config.PostTestRPCCall              // Arbitrary RPC calls to execute after the test step.
}

type ExecutionResult struct {
	TotalTests        int
	Passed            int
	Failed            int
	TotalDuration     time.Duration
	StatsReaderType   string // "cgroupv2", "dockerstats", or empty if not available
	ContainerDied     bool   // true if container exited during execution
	TerminationReason string // reason for early termination, if any
}

type Executor interface {
	Start(ctx context.Context) error
	Stop() error

	// ExecuteTests runs all tests against the specified endpoint.
	ExecuteTests(ctx context.Context, opts *ExecuteOptions) (*ExecutionResult, error)

	// GetSuiteHash returns the hash of the test suite.
	GetSuiteHash() string

	// GetTests returns the prepared test list. Returns nil if not yet prepared.
	GetTests() []*TestWithSteps

	// GetSource returns the underlying source, which can be used for genesis resolution.
	GetSource() Source
}

type GenesisGroup struct {
	GenesisHash string
	Tests       []*TestWithSteps
}

type GenesisGroupProvider interface {
	// GetGenesisGroups returns the genesis groups discovered from pre_alloc.
	// Returns nil if no pre_alloc directory exists (backward compatible).
	GetGenesisGroups() []*GenesisGroup
	// GetGenesisPathForGroup returns the genesis file path for a specific
	// genesis hash and client type.
	GetGenesisPathForGroup(genesisHash, clientType string) string
}

type GenesisProvider interface {
	// GetGenesisPath returns the path to the genesis file for a client type.
	// Returns an empty string if no genesis is available for the client.
	GetGenesisPath(clientType string) string
}

type GitSource struct {
	// contains filtered or unexported fields
}

type GitSourceInfo struct {
	Repo        string            `json:"repo"`
	Version     string            `json:"version"`
	SHA         string            `json:"sha"`
	PreRunSteps []string          `json:"pre_run_steps,omitempty"`
	Steps       *SourceStepsGlobs `json:"steps,omitempty"`
}

type Index struct {
	Generated int64         `json:"generated"`
	Entries   []*IndexEntry `json:"entries"`
}

type IndexEntry struct {
	RunID             string          `json:"run_id"`
	Timestamp         int64           `json:"timestamp"`
	TimestampEnd      int64           `json:"timestamp_end,omitempty"`
	SuiteHash         string          `json:"suite_hash,omitempty"`
	Instance          *IndexInstance  `json:"instance"`
	Tests             *IndexTestStats `json:"tests"`
	Status            string          `json:"status,omitempty"`
	TerminationReason string          `json:"termination_reason,omitempty"`
}

type IndexInstance struct {
	ID     string `json:"id"`
	Client string `json:"client"`
	Image  string `json:"image"`
}

type IndexObjectReader interface {
	// ListPrefixes lists immediate sub-prefixes under prefix.
	ListPrefixes(ctx context.Context, prefix string) ([]string, error)
	// GetObject returns the contents of key, or (nil, nil) when the
	// key does not exist.
	GetObject(ctx context.Context, key string) ([]byte, error)
}

type IndexStepStats struct {
	Success         int             `json:"success"`
	Fail            int             `json:"fail"`
	Duration        int64           `json:"duration"`
	GasUsed         uint64          `json:"gas_used"`
	GasUsedDuration int64           `json:"gas_used_duration"`
	ResourceTotals  *ResourceTotals `json:"resource_totals,omitempty"`
}

type IndexStepsStats struct {
	Setup   *IndexStepStats `json:"setup,omitempty"`
	Test    *IndexStepStats `json:"test,omitempty"`
	Cleanup *IndexStepStats `json:"cleanup,omitempty"`
}

type IndexTestStats struct {
	TestsTotal  int              `json:"tests_total"`
	TestsPassed int              `json:"tests_passed"`
	TestsFailed int              `json:"tests_failed"`
	Steps       *IndexStepsStats `json:"steps"`
}

type LocalSource struct {
	// contains filtered or unexported fields
}

type LocalSourceInfo struct {
	BaseDir     string            `json:"base_dir"`
	PreRunSteps []string          `json:"pre_run_steps,omitempty"`
	Steps       *SourceStepsGlobs `json:"steps,omitempty"`
}

type MethodResourceStats struct {
	CPUUsec        *MethodStats `json:"cpu_usec,omitempty"`
	DiskReadBytes  *MethodStats `json:"disk_read_bytes,omitempty"`
	DiskWriteBytes *MethodStats `json:"disk_write_bytes,omitempty"`
	DiskReadOps    *MethodStats `json:"disk_read_iops,omitempty"`
	DiskWriteOps   *MethodStats `json:"disk_write_iops,omitempty"`
}

type MethodStats struct {
	Count int64 `json:"count"`
	Min   int64 `json:"min"`
	Max   int64 `json:"max"`
	P50   int64 `json:"p50"`
	P95   int64 `json:"p95"`
	P99   int64 `json:"p99"`
	Mean  int64 `json:"mean"`
	Last  int64 `json:"last"`
}

type MethodStatsFloat struct {
	Count int64   `json:"count"`
	Min   float64 `json:"min"`
	Max   float64 `json:"max"`
	P50   float64 `json:"p50"`
	P95   float64 `json:"p95"`
	P99   float64 `json:"p99"`
	Mean  float64 `json:"mean"`
	Last  float64 `json:"last"`
}

type MethodsAggregated struct {
	Times      map[string]*MethodStats         `json:"times"`
	MGasPerSec map[string]*MethodStatsFloat    `json:"mgas_s"`
	Resources  map[string]*MethodResourceStats `json:"resources,omitempty"`
}

type PostTestTemplateData struct {
	BlockHash      string // e.g. "0xabc..."
	BlockNumber    string // Decimal string, e.g. "1234"
	BlockNumberHex string // Hex with 0x prefix, e.g. "0x4d2"
}

type PreparedSource struct {
	BasePath    string
	PreRunSteps []*StepFile
	Tests       []*TestWithSteps
}

type ResourceDelta struct {
	MemoryDelta    int64  `json:"memory_delta_bytes"`
	MemoryAbsBytes uint64 `json:"memory_abs_bytes,omitempty"`
	CPUDeltaUsec   uint64 `json:"cpu_delta_usec"`
	DiskReadBytes  uint64 `json:"disk_read_bytes"`
	DiskWriteBytes uint64 `json:"disk_write_bytes"`
	DiskReadOps    uint64 `json:"disk_read_iops"`
	DiskWriteOps   uint64 `json:"disk_write_iops"`
}

type ResourceTotals struct {
	CPUUsec        uint64 `json:"cpu_usec"`
	MemoryDelta    int64  `json:"memory_delta_bytes"`
	MemoryBytes    uint64 `json:"memory_bytes,omitempty"`
	DiskReadBytes  uint64 `json:"disk_read_bytes"`
	DiskWriteBytes uint64 `json:"disk_write_bytes"`
	DiskReadIOPS   uint64 `json:"disk_read_iops"`
	DiskWriteIOPS  uint64 `json:"disk_write_iops"`
}

type ResultDetails struct {
	DurationNS []int64                `json:"duration_ns"`
	Status     []int                  `json:"status"`
	MGasPerSec map[int]float64        `json:"mgas_s"`
	GasUsed    map[int]uint64         `json:"gas_used"`
	Resources  map[int]*ResourceDelta `json:"resources,omitempty"`
	// OriginalTestName stores the original test name when using hashed filenames.
	OriginalTestName string `json:"original_test_name,omitempty"`
	// FilenameHash stores the truncated+hash filename when the original was too long.
	FilenameHash string `json:"filename_hash,omitempty"`
}

type RunDuration struct {
	ID       string                 `json:"id"`
	Client   string                 `json:"client"`
	GasUsed  uint64                 `json:"gas_used"`
	Time     int64                  `json:"time_ns"`
	RunStart int64                  `json:"run_start"`
	RunEnd   int64                  `json:"run_end,omitempty"`
	Steps    *RunDurationStepsStats `json:"steps,omitempty"`
}

type RunDurationStepStats struct {
	GasUsed uint64 `json:"gas_used"`
	Time    int64  `json:"time_ns"`
}

type RunDurationStepsStats struct {
	Setup   *RunDurationStepStats `json:"setup,omitempty"`
	Test    *RunDurationStepStats `json:"test,omitempty"`
	Cleanup *RunDurationStepStats `json:"cleanup,omitempty"`
}

type RunResult struct {
	PreRunSteps map[string]*StepResult `json:"pre_run_steps,omitempty"`
	Tests       map[string]*TestEntry  `json:"tests"`
}

type Source interface {
	// Prepare ensures test files are available and returns the prepared source.
	Prepare(ctx context.Context) (*PreparedSource, error)
	// Cleanup removes any temporary resources.
	Cleanup() error
	// GetSourceInfo returns source information for the suite summary.
	GetSourceInfo() (*SuiteSource, error)
}

type SourceStepsGlobs struct {
	Setup   []string `json:"setup,omitempty"`
	Test    []string `json:"test,omitempty"`
	Cleanup []string `json:"cleanup,omitempty"`
}

type StepFile struct {
	Path     string       // Full absolute path (empty if using provider)
	Name     string       // Relative path from base or logical name
	Provider StepProvider // Optional provider for in-memory steps
}

type StepProvider interface {
	// Lines returns the JSON-RPC lines for this step.
	Lines() []string
	// Content returns the full content as bytes for hashing.
	Content() []byte
}

type StepResult struct {
	Aggregated *AggregatedStats `json:"aggregated"`
}

type StepType string

type StepsResult struct {
	Setup   *StepResult `json:"setup,omitempty"`
	Test    *StepResult `json:"test,omitempty"`
	Cleanup *StepResult `json:"cleanup,omitempty"`
}

type SuiteFile struct {
	OgPath string `json:"og_path"` // original relative path
}

type SuiteInfo struct {
	Hash        string       `json:"hash"`
	Source      *SuiteSource `json:"source"`
	Filter      string       `json:"filter,omitempty"`
	PreRunSteps []SuiteFile  `json:"pre_run_steps,omitempty"`
	Tests       []SuiteTest  `json:"tests"`
}

type SuiteSource struct {
	Git   *GitSourceInfo   `json:"git,omitempty"`
	Local *LocalSourceInfo `json:"local,omitempty"`
	EEST  *EESTSourceInfo  `json:"eest,omitempty"`
}

type SuiteStats map[string]*TestDurations

type SuiteTest struct {
	Name        string         `json:"name"`
	GenesisHash string         `json:"genesis,omitempty"`
	Setup       *SuiteFile     `json:"setup,omitempty"`
	Test        *SuiteFile     `json:"test,omitempty"`
	Cleanup     *SuiteFile     `json:"cleanup,omitempty"`
	EEST        *SuiteTestEEST `json:"eest,omitempty"`
}

type SuiteTestEEST struct {
	Info *eest.FixtureInfo `json:"info"`
}

type TestDurations struct {
	Durations []*RunDuration `json:"durations"`
}

type TestEntry struct {
	Dir          string       `json:"dir"`
	FilenameHash string       `json:"filename_hash,omitempty"`
	Steps        *StepsResult `json:"steps,omitempty"`
}

type TestResult struct {
	TestFile             string
	Responses            []string
	Times                []int64
	Statuses             []int // 0=success, 1=fail
	MGasPerSec           map[int]float64
	GasUsed              map[int]uint64
	Resources            map[int]*ResourceDelta
	MethodTimes          map[string][]int64
	MethodMGasPerSec     map[string][]float64
	MethodCPUUsec        map[string][]int64
	MethodDiskReadBytes  map[string][]int64
	MethodDiskWriteBytes map[string][]int64
	MethodDiskReadOps    map[string][]int64
	MethodDiskWriteOps   map[string][]int64
	Succeeded            int
	Failed               int
}

type TestWithSteps struct {
	Name        string            // Common test name (e.g., "abc.txt")
	Setup       *StepFile         // Optional setup step
	Test        *StepFile         // Optional test step
	Cleanup     *StepFile         // Optional cleanup step
	GenesisHash string            // Genesis hash from pre_alloc (empty if single-genesis)
	EESTInfo    *eest.FixtureInfo // EEST fixture metadata (nil for non-EEST sources)
}