tdg

type Config struct {
	// MaxTestRetries is the maximum test generation attempts.
	// When exceeded, TDG returns ErrMaxTestRetries.
	// Default: 3
	MaxTestRetries int

	// MaxFixRetries is the maximum fix generation attempts.
	// When exceeded, TDG returns ErrMaxFixRetries.
	// Default: 5
	MaxFixRetries int

	// MaxRegressionFixes is the maximum regression fix attempts.
	// When exceeded, TDG returns ErrMaxRegressionFixes.
	// Default: 3
	MaxRegressionFixes int

	// TestTimeout is the timeout for a single test execution.
	// Default: 30s
	TestTimeout time.Duration

	// SuiteTimeout is the timeout for running the full test suite.
	// Default: 5m
	SuiteTimeout time.Duration

	// TotalTimeout is the timeout for the entire TDG session.
	// Default: 10m
	TotalTimeout time.Duration

	// MaxOutputBytes is the maximum test output to capture.
	// Output beyond this is truncated.
	// Default: 65536 (64KB)
	MaxOutputBytes int

	// EnableLintCheck enables lint checking on generated code (CB-25).
	// Default: true
	EnableLintCheck bool

	// WorkingDir overrides the working directory for test execution.
	// If empty, uses the project root from the request.
	WorkingDir string
}

type Context struct {
	// SessionID uniquely identifies this TDG session.
	SessionID string

	// State is the current TDG state.
	State State

	// Request is the original request.
	Request *Request

	// ReproducerTest is the generated reproducer test.
	ReproducerTest *TestCase

	// ProposedPatches are the current proposed code changes.
	ProposedPatches []*Patch

	// AppliedPatches are patches that have been applied to disk.
	AppliedPatches []*Patch

	// TestRetries is the number of test generation attempts.
	TestRetries int

	// FixRetries is the number of fix generation attempts.
	FixRetries int

	// RegressionFixes is the number of regression fix attempts.
	RegressionFixes int

	// LastTestOutput is the output from the last test execution.
	LastTestOutput string

	// LastError is the last error encountered.
	LastError error

	// StartTime is when the TDG session started (Unix milliseconds UTC).
	StartTime int64

	// Metrics tracks execution metrics.
	Metrics *Metrics
}

type ContextAssembler interface {
	// AssembleContext gathers relevant code for a query.
	AssembleContext(ctx context.Context, query string, tokenBudget int) (string, error)
}

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

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

type HistoryEntry struct {
	// Timestamp is when this entry was recorded (Unix milliseconds UTC).
	Timestamp int64 `json:"timestamp"`

	// State is the TDG state at this step.
	State State `json:"state"`

	// Action describes what happened.
	Action string `json:"action"`

	// Details contains additional information.
	Details string `json:"details,omitempty"`

	// Duration is how long this step took.
	Duration time.Duration `json:"duration,omitempty"`

	// Error contains any error from this step.
	Error string `json:"error,omitempty"`
}

type LLMClient interface {
	// Generate produces text from a prompt.
	Generate(ctx context.Context, prompt string) (string, error)

	// GenerateWithSystem produces text with a system prompt.
	GenerateWithSystem(ctx context.Context, system, prompt string) (string, error)
}

type LanguageConfig struct {
	// Language is the language identifier (e.g., "go", "python").
	Language string

	// TestCommand is the test runner command (e.g., "go", "pytest").
	TestCommand string

	// TestArgs are arguments for running a single test.
	// Use {name} as placeholder for test name, {file} for file path.
	TestArgs []string

	// SuiteArgs are arguments for running the full test suite.
	// Use {package} as placeholder for package/directory path.
	SuiteArgs []string

	// TestFilePattern is the glob pattern for test files.
	TestFilePattern string

	// TestNameFlag is the flag for specifying test name (e.g., "-run").
	TestNameFlag string

	// Extensions are file extensions for this language.
	Extensions []string
}

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

type Metrics struct {
	// TestAttempts is the number of test generation attempts.
	TestAttempts int `json:"test_attempts"`

	// FixAttempts is the number of fix generation attempts.
	FixAttempts int `json:"fix_attempts"`

	// RegressionFixes is the number of regression fix attempts.
	RegressionFixes int `json:"regression_fixes"`

	// TestsRun is the total number of test executions.
	TestsRun int `json:"tests_run"`

	// LLMCalls is the number of LLM API calls.
	LLMCalls int `json:"llm_calls"`

	// TotalTestDuration is the cumulative time spent running tests.
	TotalTestDuration time.Duration `json:"total_test_duration"`

	// ContextTokens is the total context tokens assembled.
	ContextTokens int `json:"context_tokens"`
}

type Option func(*Config)

type Patch struct {
	// FilePath is the file to modify.
	FilePath string `json:"file_path"`

	// OldContent is the original file content (for rollback).
	OldContent string `json:"old_content,omitempty"`

	// NewContent is the new file content.
	NewContent string `json:"new_content"`

	// Applied indicates if the patch has been applied to disk.
	Applied bool `json:"applied"`
}

type Request struct {
	// BugDescription describes the bug to fix or feature to implement.
	BugDescription string `json:"bug_description"`

	// ProjectRoot is the root directory of the project.
	ProjectRoot string `json:"project_root"`

	// Language is the programming language (go, python, typescript).
	Language string `json:"language"`

	// GraphID is an optional existing Trace graph ID.
	GraphID string `json:"graph_id,omitempty"`

	// TargetFile is an optional hint for which file contains the bug.
	TargetFile string `json:"target_file,omitempty"`

	// TargetFunction is an optional hint for which function has the bug.
	TargetFunction string `json:"target_function,omitempty"`
}

type Result struct {
	// Success indicates if TDG completed successfully.
	Success bool `json:"success"`

	// State is the final TDG state.
	State State `json:"final_state"`

	// ReproducerTest is the generated test that reproduces the bug.
	ReproducerTest *TestCase `json:"reproducer_test,omitempty"`

	// AppliedPatches are the code changes that were applied.
	AppliedPatches []*Patch `json:"applied_patches,omitempty"`

	// TestResults is the final test execution result.
	TestResults *TestResult `json:"test_results,omitempty"`

	// RegressionResults is the full suite test result.
	RegressionResults *TestResult `json:"regression_results,omitempty"`

	// Error contains the error message if TDG failed.
	Error string `json:"error,omitempty"`

	// Duration is how long the TDG session took.
	Duration time.Duration `json:"duration"`

	// Metrics contains execution metrics.
	Metrics *Metrics `json:"metrics"`
}

type RetryExhaustedError struct {
	// Phase is the TDG phase where retries were exhausted.
	Phase string

	// Attempts is the number of attempts made.
	Attempts int

	// MaxAttempts is the configured maximum attempts.
	MaxAttempts int

	// LastError is the error from the last attempt.
	LastError error
}

type State string

type StateTransitionError struct {
	From State
	To   State
}

type TestCase struct {
	// Name is the test function name (e.g., TestValidateToken_NilClaims).
	Name string `json:"name"`

	// FilePath is the path where the test file should be written.
	FilePath string `json:"file_path"`

	// Content is the full test file content.
	Content string `json:"content"`

	// Language is the programming language.
	Language string `json:"language"`

	// TestType is "unit" or "integration".
	TestType string `json:"test_type"`

	// PackagePath is the package/module path for running tests.
	PackagePath string `json:"package_path,omitempty"`
}

type TestExecutionError struct {
	// TestName is the name of the test that failed.
	TestName string

	// FilePath is the path to the test file.
	FilePath string

	// ExitCode is the exit code from the test process.
	ExitCode int

	// Output is the captured stdout/stderr.
	Output string

	// Cause is the underlying error if any.
	Cause error
}

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

type TestOutputParser func(output []byte) (passed bool, failedTests []string)

type TestResult struct {
	// Passed indicates if all tests passed.
	Passed bool `json:"passed"`

	// Output is the captured stdout/stderr.
	Output string `json:"output"`

	// Duration is how long test execution took.
	Duration time.Duration `json:"duration"`

	// ExitCode is the exit code from the test process.
	ExitCode int `json:"exit_code"`

	// FailedTests lists the names of failed tests.
	FailedTests []string `json:"failed_tests,omitempty"`

	// PassedTests lists the names of passed tests.
	PassedTests []string `json:"passed_tests,omitempty"`

	// TotalTests is the total number of tests run.
	TotalTests int `json:"total_tests"`

	// Truncated indicates if output was truncated due to size limits.
	Truncated bool `json:"truncated"`

	// TimedOut indicates if execution was killed due to timeout.
	TimedOut bool `json:"timed_out"`
}

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