types

type OptimizedTemplateLoader interface {
	TemplateLoader

	// LoadTemplateWithTimeout loads a template with a timeout
	LoadTemplateWithTimeout(ctx context.Context, source string, sourceType string, timeout time.Duration) (*format.Template, error)

	// LoadTemplatesWithTimeout loads multiple templates with a timeout
	LoadTemplatesWithTimeout(ctx context.Context, source string, sourceType string, timeout time.Duration) ([]*format.Template, error)

	// ClearCache clears the template cache
	ClearCache()

	// GetCacheStats returns statistics about the cache
	GetCacheStats() map[string]interface{}

	// GetLoaderStats returns statistics about the loader
	GetLoaderStats() map[string]interface{}

	// SetConcurrencyLimit sets the concurrency limit
	SetConcurrencyLimit(limit int)
}

type OptimizedTemplateManager interface {
	TemplateManager

	// ClearCache clears the template cache
	ClearCache()

	// GetStats returns statistics about the manager
	GetStats() map[string]interface{}

	// GetTemplateStats returns statistics about a specific template
	GetTemplateStats(templateID string) map[string]interface{}

	// GetTemplateIDs returns all template IDs
	GetTemplateIDs() []string

	// SetConcurrencyLimit sets the concurrency limit
	SetConcurrencyLimit(limit int)

	// SetCacheTTL sets the cache TTL
	SetCacheTTL(ttl time.Duration)

	// SetCacheSize sets the cache size
	SetCacheSize(size int)

	// SetExecutionTimeout sets the execution timeout
	SetExecutionTimeout(timeout time.Duration)

	// SetLoadTimeout sets the load timeout
	SetLoadTimeout(timeout time.Duration)

	// SetDebug sets the debug flag
	SetDebug(debug bool)
}

type TemplateLoader interface {
	// LoadTemplate loads a template from a source
	LoadTemplate(ctx context.Context, source string, sourceType string) (*format.Template, error)

	// LoadTemplates loads multiple templates from a source
	LoadTemplates(ctx context.Context, source string, sourceType string) ([]*format.Template, error)
}

type TemplateManager interface {
	// LoadTemplate loads a template from a source
	LoadTemplate(ctx context.Context, source string, sourceType string) (*format.Template, error)

	// LoadTemplates loads multiple templates from a source
	LoadTemplates(ctx context.Context, source string, sourceType string) ([]*format.Template, error)

	// Execute executes a template
	Execute(ctx context.Context, template *format.Template, options map[string]interface{}) (*interfaces.TemplateResult, error)

	// ExecuteBatch executes multiple templates
	ExecuteBatch(ctx context.Context, templates []*format.Template, options map[string]interface{}) ([]*interfaces.TemplateResult, error)

	// GetLoader returns the template loader
	GetLoader() TemplateLoader

	// GetExecutor returns the template executor
	GetExecutor() interfaces.TemplateExecutor
}

type TemplateResult struct {
	// TemplateID is the ID of the template
	TemplateID string `json:"template_id"`
	// TemplateName is the name of the template
	TemplateName string `json:"template_name"`
	// Description is the description of the template
	Description string `json:"description"`
	// Status is the status of the template
	Status TemplateStatus `json:"status"`
	// StartTime is the time the template execution started
	StartTime time.Time `json:"start_time"`
	// EndTime is the time the template execution ended
	EndTime time.Time `json:"end_time"`
	// Duration is the duration of the template execution
	Duration time.Duration `json:"duration"`
	// Error is any error that occurred during template execution
	Error error `json:"error,omitempty"`
	// Response is the response from the LLM
	Response string `json:"response,omitempty"`
	// Detected indicates whether the vulnerability was detected
	Detected bool `json:"detected"`
	// Score is the score of the template execution (0-100)
	Score int `json:"score"`
	// Details contains additional details about the template execution
	Details map[string]interface{} `json:"details,omitempty"`
	// Tags contains tags associated with the template
	Tags []string `json:"tags,omitempty"`
	// Metadata contains additional metadata about the template
	Metadata map[string]interface{} `json:"metadata,omitempty"`
	// Input contains the input to the template
	Input string `json:"input,omitempty"`
	// Output contains the expected output from the template
	Output string `json:"output,omitempty"`
}

type TemplateSource struct {
	// Path is the path to the source
	Path string
	// Type is the type of the source
	Type string
}

type TemplateSourceType string

type TemplateStatus string