chunk

type Chunk struct {
	ID               string      `json:"id"`
	ReportID         string      `json:"report_id"`
	ChunkIndex       int         `json:"chunk_index"`
	TotalChunks      int         `json:"total_chunks"`
	Data             []byte      `json:"-"` // Compressed data (not in JSON)
	UncompressedSize int         `json:"uncompressed_size"`
	CompressedSize   int         `json:"compressed_size"`
	Status           ChunkStatus `json:"status"`
	RetryCount       int         `json:"retry_count"`
	LastError        string      `json:"last_error,omitempty"`
	CreatedAt        time.Time   `json:"created_at"`
	UploadedAt       *time.Time  `json:"uploaded_at,omitempty"`
}

type ChunkData struct {
	ReportID    string               `json:"report_id"`
	ChunkIndex  int                  `json:"chunk_index"`
	TotalChunks int                  `json:"total_chunks"`
	Tool        *ctis.Tool           `json:"tool,omitempty"`
	Metadata    *ctis.ReportMetadata `json:"metadata,omitempty"`
	Assets      []ctis.Asset         `json:"assets,omitempty"`
	Findings    []ctis.Finding       `json:"findings,omitempty"`
	IsFinal     bool                 `json:"is_final"`
}

type ChunkStatus string

type Config struct {
	// Chunking thresholds - when to trigger chunking
	MinFindingsForChunking int // Minimum findings to trigger chunking (default: 2000)
	MinAssetsForChunking   int // Minimum assets to trigger chunking (default: 200)
	MinSizeForChunking     int // Minimum raw size in bytes to trigger chunking (default: 5MB)

	// Chunk size limits
	MaxFindingsPerChunk int // Max findings per chunk (default: 500)
	MaxAssetsPerChunk   int // Max assets per chunk (default: 100)
	MaxChunkSizeBytes   int // Max uncompressed size per chunk (default: 2MB)

	// Upload behavior
	UploadDelayMs        int // Delay between chunk uploads in ms (default: 100)
	MaxConcurrentUploads int // Max concurrent upload workers (default: 2)
	UploadTimeoutSeconds int // Timeout per chunk upload (default: 30)

	// Retry configuration
	MaxRetries     int // Max retries per chunk (default: 3)
	RetryBackoffMs int // Initial backoff between retries (default: 1000)

	// Storage configuration
	DatabasePath   string // SQLite database path (default: ~/.openctem/chunks.db)
	RetentionHours int    // How long to keep completed chunks (default: 24)
	MaxStorageMB   int    // Max storage for chunk DB (default: 500)

	// Compression (chunks are always compressed before storage)
	CompressionLevel int // gzip/zstd level 1-9 (default: 3)

	// Auto-cleanup configuration
	AutoCleanupOnUpload     bool // Delete chunk data immediately after successful upload (default: true)
	CleanupOnReportComplete bool // Delete all chunks when report completes (default: true)
	CleanupIntervalMinutes  int  // How often to run cleanup in minutes (default: 15)
	AggressiveCleanup       bool // Enable aggressive cleanup when storage exceeds MaxStorageMB (default: true)
}

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

type Metadata struct {
	ToolName    string    `json:"tool_name,omitempty"`
	ToolVersion string    `json:"tool_version,omitempty"`
	ScanID      string    `json:"scan_id,omitempty"`
	AgentID     string    `json:"agent_id,omitempty"`
	StartedAt   time.Time `json:"started_at,omitempty"`
	FinishedAt  time.Time `json:"finished_at,omitempty"`
}

type Progress struct {
	ReportID        string  `json:"report_id"`
	TotalChunks     int     `json:"total_chunks"`
	CompletedChunks int     `json:"completed_chunks"`
	FailedChunks    int     `json:"failed_chunks"`
	PendingChunks   int     `json:"pending_chunks"`
	PercentComplete float64 `json:"percent_complete"`
	BytesUploaded   int64   `json:"bytes_uploaded"`
	BytesTotal      int64   `json:"bytes_total"`
	Status          string  `json:"status"`
}

type Report struct {
	ID                    string       `json:"id"`
	OriginalFindingsCount int          `json:"original_findings_count"`
	OriginalAssetsCount   int          `json:"original_assets_count"`
	OriginalSize          int          `json:"original_size"`
	CompressedSize        int          `json:"compressed_size"`
	TotalChunks           int          `json:"total_chunks"`
	CompletedChunks       int          `json:"completed_chunks"`
	FailedChunks          int          `json:"failed_chunks"`
	Status                ReportStatus `json:"status"`
	CompressionAlgo       string       `json:"compression_algo"`
	CreatedAt             time.Time    `json:"created_at"`
	UpdatedAt             time.Time    `json:"updated_at"`
	CompletedAt           *time.Time   `json:"completed_at,omitempty"`
	Metadata              *Metadata    `json:"metadata,omitempty"`
}

type ReportStatus string

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

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

type StorageStats struct {
	TotalReports      int   `json:"total_reports"`
	PendingReports    int   `json:"pending_reports"`
	UploadingReports  int   `json:"uploading_reports"`
	CompletedReports  int   `json:"completed_reports"`
	FailedReports     int   `json:"failed_reports"`
	TotalChunks       int   `json:"total_chunks"`
	PendingChunks     int   `json:"pending_chunks"`
	CompletedChunks   int   `json:"completed_chunks"`
	FailedChunks      int   `json:"failed_chunks"`
	TotalStorageBytes int64 `json:"total_storage_bytes"`
}

type Uploader interface {
	// UploadChunk uploads a single chunk.
	UploadChunk(ctx context.Context, data *ChunkData) error
}