scan

type CreateScanConfigRequest struct {
	// Name is the name of the scan configuration
	Name string `json:"name" validate:"required"`
	// Description is a description of the scan configuration
	Description string `json:"description"`
	// Target is the target of the scan (e.g., a prompt, system, or model)
	Target string `json:"target" validate:"required"`
	// TargetType is the type of target (e.g., "prompt", "system", "model")
	TargetType string `json:"target_type" validate:"required"`
	// Templates is a list of template IDs to use for the scan
	Templates []string `json:"templates" validate:"required,min=1"`
	// Parameters is a map of parameters for the scan
	Parameters map[string]interface{} `json:"parameters"`
}

type CreateScanRequest struct {
	// ConfigID is the ID of the scan configuration
	ConfigID string `json:"config_id" validate:"required"`
}

type ErrorResponse struct {
	// Error is the error message
	Error string `json:"error"`
	// Code is the error code
	Code int `json:"code"`
}

type FilterParams struct {
	// Status filters results by status
	Status string `json:"status"`
	// Severity filters results by severity
	Severity string `json:"severity"`
	// StartDate filters results by start date
	StartDate string `json:"start_date"`
	// EndDate filters results by end date
	EndDate string `json:"end_date"`
	// Search is a search term to filter results
	Search string `json:"search"`
}

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

type ListResponse struct {
	// Pagination contains pagination information
	Pagination PaginationParams `json:"pagination"`
	// Data contains the response data
	Data interface{} `json:"data"`
}

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

type PaginationParams struct {
	// Page is the page number (1-based)
	Page int `json:"page"`
	// PageSize is the number of items per page
	PageSize int `json:"page_size"`
	// TotalItems is the total number of items
	TotalItems int `json:"total_items"`
	// TotalPages is the total number of pages
	TotalPages int `json:"total_pages"`
}

type Scan struct {
	// ID is the unique identifier for the scan
	ID string `json:"id"`
	// ConfigID is the ID of the scan configuration
	ConfigID string `json:"config_id"`
	// Status is the status of the scan
	Status ScanStatus `json:"status"`
	// StartTime is the time the scan started
	StartTime time.Time `json:"start_time"`
	// EndTime is the time the scan ended
	EndTime time.Time `json:"end_time,omitempty"`
	// Progress is the progress of the scan (0-100)
	Progress int `json:"progress"`
	// Error is the error message if the scan failed
	Error string `json:"error,omitempty"`
	// Results is the results of the scan
	Results []ScanResult `json:"results,omitempty"`
}

type ScanConfig struct {
	// ID is the unique identifier for the scan configuration
	ID string `json:"id"`
	// Name is the name of the scan configuration
	Name string `json:"name"`
	// Description is a description of the scan configuration
	Description string `json:"description"`
	// Target is the target of the scan (e.g., a prompt, system, or model)
	Target string `json:"target"`
	// TargetType is the type of target (e.g., "prompt", "system", "model")
	TargetType string `json:"target_type"`
	// Templates is a list of template IDs to use for the scan
	Templates []string `json:"templates"`
	// Parameters is a map of parameters for the scan
	Parameters map[string]interface{} `json:"parameters"`
	// CreatedAt is the time the scan configuration was created
	CreatedAt time.Time `json:"created_at"`
	// UpdatedAt is the time the scan configuration was last updated
	UpdatedAt time.Time `json:"updated_at"`
	// CreatedBy is the user who created the scan configuration
	CreatedBy string `json:"created_by"`
}

type ScanResult struct {
	// ID is the unique identifier for the scan result
	ID string `json:"id"`
	// ScanID is the ID of the scan
	ScanID string `json:"scan_id"`
	// TemplateID is the ID of the template that generated the result
	TemplateID string `json:"template_id"`
	// Severity is the severity of the finding
	Severity ScanSeverity `json:"severity"`
	// Title is the title of the finding
	Title string `json:"title"`
	// Description is a description of the finding
	Description string `json:"description"`
	// Details contains detailed information about the finding
	Details map[string]interface{} `json:"details"`
	// Timestamp is the time the result was generated
	Timestamp time.Time `json:"timestamp"`
}

type ScanSeverity string

type ScanStatus string

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

type Storage interface {
	// ScanConfig methods
	CreateScanConfig(ctx context.Context, config *ScanConfig) error
	GetScanConfig(ctx context.Context, id string) (*ScanConfig, error)
	UpdateScanConfig(ctx context.Context, config *ScanConfig) error
	DeleteScanConfig(ctx context.Context, id string) error
	ListScanConfigs(ctx context.Context, page, pageSize int, filters FilterParams) ([]*ScanConfig, int, error)

	// Scan methods
	CreateScan(ctx context.Context, scan *Scan) error
	GetScan(ctx context.Context, id string) (*Scan, error)
	UpdateScan(ctx context.Context, scan *Scan) error
	DeleteScan(ctx context.Context, id string) error
	ListScans(ctx context.Context, page, pageSize int, filters FilterParams) ([]*Scan, int, error)

	// ScanResult methods
	CreateScanResult(ctx context.Context, result *ScanResult) error
	GetScanResult(ctx context.Context, id string) (*ScanResult, error)
	UpdateScanResult(ctx context.Context, result *ScanResult) error
	DeleteScanResult(ctx context.Context, id string) error
	ListScanResults(ctx context.Context, scanID string, page, pageSize int, filters FilterParams) ([]*ScanResult, int, error)
}

type UpdateScanConfigRequest struct {
	// Name is the name of the scan configuration
	Name string `json:"name"`
	// Description is a description of the scan configuration
	Description string `json:"description"`
	// Target is the target of the scan (e.g., a prompt, system, or model)
	Target string `json:"target"`
	// TargetType is the type of target (e.g., "prompt", "system", "model")
	TargetType string `json:"target_type"`
	// Templates is a list of template IDs to use for the scan
	Templates []string `json:"templates"`
	// Parameters is a map of parameters for the scan
	Parameters map[string]interface{} `json:"parameters"`
}