models

type AST struct {
	Language    string        `json:"language"`
	Definitions []*Definition `json:"definitions"`
	Imports     []string      `json:"imports"`
	Package     string        `json:"package,omitempty"`
}

type Definition struct {
	Name       string  `json:"name"`
	Signature  string  `json:"signature"`
	Body       string  `json:"body"`
	StartLine  int     `json:"start_line"`
	EndLine    int     `json:"end_line"`
	IsMethod   bool    `json:"is_method"`
	ClassName  string  `json:"class_name,omitempty"`
	Parameters []Param `json:"parameters,omitempty"`
	ReturnType string  `json:"return_type,omitempty"`
	Docstring  string  `json:"docstring,omitempty"`
}

type GeneratedTest struct {
	TestName      string   `json:"test_name"`
	TestCode      string   `json:"test_code"`
	Imports       []string `json:"imports"`
	CoverageLines []string `json:"coverage_lines,omitempty"`
	EdgeCases     []string `json:"edge_cases_covered,omitempty"`
	Dependencies  []string `json:"mocked_dependencies,omitempty"`
}

type GenerationResult struct {
	SourceFile      *SourceFile `json:"source_file"`
	TestCode        string      `json:"test_code,omitempty"`
	TestPath        string      `json:"test_path,omitempty"`
	FunctionsTested []string    `json:"functions_tested,omitempty"`
	TestCount       int         `json:"test_count"`
	Error           error       `json:"-"`
	ErrorMessage    string      `json:"error,omitempty"`
}

type Param struct {
	Name string `json:"name"`
	Type string `json:"type,omitempty"`
}

type SourceFile struct {
	Path      string   `json:"path"`
	Language  string   `json:"language"`
	Framework string   `json:"framework,omitempty"`
	Content   string   `json:"-"` // Not serialized
	LineCount int      `json:"line_count"`
	Functions []string `json:"functions,omitempty"`
}

type TestResults struct {
	ExitCode     int      `json:"exit_code"`
	Output       string   `json:"output"`
	Coverage     float64  `json:"coverage_percent,omitempty"`
	PassedCount  int      `json:"passed"`
	FailedCount  int      `json:"failed"`
	SkippedCount int      `json:"skipped"`
	Duration     float64  `json:"duration_seconds"`
	Errors       []string `json:"errors,omitempty"`
}

type UsageMetrics struct {
	RunID          string  `json:"run_id"`
	Timestamp      string  `json:"timestamp"`
	TotalFiles     int     `json:"total_files"`
	TokensInput    int     `json:"tokens_input"`
	TokensOutput   int     `json:"tokens_output"`
	TokensCached   int     `json:"tokens_cached"`
	CacheHitRate   float64 `json:"cache_hit_rate"`
	TotalCostUSD   float64 `json:"total_cost_usd"`
	ExecutionTimeS float64 `json:"execution_time_seconds"`
	SuccessCount   int     `json:"success_count"`
	ErrorCount     int     `json:"error_count"`
}