cmd

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

type CacheEntry struct {
	Table            string    `json:"table"`
	Partition        string    `json:"partition"`
	RowCount         int64     `json:"rowCount"`
	CountTime        time.Time `json:"countTime"`
	FileSize         int64     `json:"fileSize"`
	UncompressedSize int64     `json:"uncompressedSize"`
	FileMD5          string    `json:"fileMD5"`
	MultipartETag    string    `json:"multipartETag,omitempty"` // S3 multipart ETag for files >100MB
	FileTime         time.Time `json:"fileTime"`
	S3Key            string    `json:"s3Key"`
	S3Uploaded       bool      `json:"s3Uploaded"`
	S3UploadTime     time.Time `json:"s3UploadTime"`
	LastError        string    `json:"lastError"`
	ErrorTime        time.Time `json:"errorTime"`
	ProcessStartTime time.Time `json:"processStartTime,omitempty"` // When processing started for this job
}

type CacheResponse struct {
	Tables    []TableCache `json:"tables"`
	Timestamp time.Time    `json:"timestamp"`
}

type CacheScope struct {
	Command    string
	Table      string
	OutputPath string
}

type ColumnInfo struct {
	Name     string
	DataType string
	UDTName  string // PostgreSQL user-defined type name (e.g., int4, varchar, timestamp)
}

type ColumnTypeMismatch struct {
	ColumnName  string `json:"column_name"`
	Source1Type string `json:"source1_type"`
	Source2Type string `json:"source2_type"`
}

type CompareConfig struct {
	Mode            string // schema-only, data-only, schema-and-data
	DataCompareType string // row-count, row-by-row, sample
	SampleSize      int
	Tables          []string // Empty = all tables
	OutputFormat    string   // text, json
	OutputFile      string
	Debug           bool
	DryRun          bool
}

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

type ComparisonResult struct {
	Schema *SchemaComparisonResult `json:"schema,omitempty"`
	Data   *DataComparisonResult   `json:"data,omitempty"`
}

type ComparisonSource struct {
	Type         string // "db" or "s3"
	Database     DatabaseConfig
	S3           S3Config
	SchemaPath   string // S3 path for schemas
	DataPath     string // S3 path for data
	SchemaSource string // pg_dump, inferred, auto
}

type Config struct {
	Debug                     bool
	LogFormat                 string
	DryRun                    bool
	Workers                   int
	SkipCount                 bool
	CacheViewer               bool
	ViewerPort                int
	ChunkSize                 int  // Number of rows to process in each chunk (streaming mode)
	IncludeNonPartitionTables bool // Include regular tables matching partition naming pattern
	Database                  DatabaseConfig
	S3                        S3Config
	Table                     string
	StartDate                 string
	EndDate                   string
	OutputDuration            string
	OutputFormat              string
	Compression               string
	CompressionLevel          int
	DateColumn                string
	DumpMode                  string // pg_dump mode: schema-only, data-only, schema-and-data
	CacheScope                CacheScope
}

type DataComparisonResult struct {
	RowCountDiffs map[string]*RowCountDiff `json:"row_count_diffs,omitempty"`
	RowByRowDiffs map[string]*RowByRowDiff `json:"row_by_row_diffs,omitempty"`
	SampleDiffs   map[string]*SampleDiff   `json:"sample_diffs,omitempty"`
}

type DatabaseConfig struct {
	Host             string
	Port             int
	User             string
	Password         string
	Name             string
	SSLMode          string
	StatementTimeout int // Statement timeout in seconds (0 = no timeout, default 300)
	MaxRetries       int // Maximum number of retry attempts for failed queries (default 3)
	RetryDelay       int // Delay in seconds between retry attempts (default 5)
}

type GitHubRelease struct {
	TagName     string    `json:"tag_name"`
	Name        string    `json:"name"`
	PublishedAt time.Time `json:"published_at"`
	HTMLURL     string    `json:"html_url"`
}

type LogMessage struct {
	Timestamp string `json:"timestamp"`
	Level     string `json:"level"`
	Message   string `json:"message"`
}

type PartitionCache struct {
	Entries map[string]PartitionCacheEntry `json:"entries"`
}

type PartitionCacheEntry struct {
	// Row count information
	RowCount  int64     `json:"row_count"`
	CountTime time.Time `json:"count_time"`

	// File metadata (stored after processing)
	FileSize         int64     `json:"file_size,omitempty"`         // Compressed size
	UncompressedSize int64     `json:"uncompressed_size,omitempty"` // Original size
	FileMD5          string    `json:"file_md5,omitempty"`
	MultipartETag    string    `json:"multipart_etag,omitempty"` // S3 multipart ETag for files >100MB
	FileTime         time.Time `json:"file_time,omitempty"`

	// S3 information
	S3Key        string    `json:"s3_key,omitempty"`
	S3Uploaded   bool      `json:"s3_uploaded,omitempty"`
	S3UploadTime time.Time `json:"s3_upload_time,omitempty"`

	// Error tracking
	LastError string    `json:"last_error,omitempty"`
	ErrorTime time.Time `json:"error_time,omitempty"`

	// Processing time tracking
	ProcessStartTime time.Time `json:"process_start_time,omitempty"` // When processing started for this job
}

type PartitionInfo struct {
	TableName  string
	Date       time.Time
	RowCount   int64
	RangeStart time.Time
	RangeEnd   time.Time
}

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

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

type Phase int

type ProcessResult struct {
	Partition    PartitionInfo
	Compressed   bool
	Uploaded     bool
	Skipped      bool
	SkipReason   string
	Error        error
	BytesWritten int64
	Stage        string
	S3Key        string        // S3 object key for uploaded file
	StartTime    time.Time     // When partition processing started
	Duration     time.Duration // How long partition processing took
}

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

type RowByRowDiff struct {
	Source1TotalRows int64 `json:"source1_total_rows"`
	Source2TotalRows int64 `json:"source2_total_rows"`
	MatchingRows     int64 `json:"matching_rows"`
	MissingInSource2 int64 `json:"missing_in_source2"`
	ExtraInSource2   int64 `json:"extra_in_source2"`
}

type RowCountCache struct {
	Counts map[string]RowCountEntry `json:"counts"`
}

type RowCountDiff struct {
	Source1Count int64 `json:"source1_count"`
	Source2Count int64 `json:"source2_count"`
	Difference   int64 `json:"difference"`
}

type RowCountEntry struct {
	Count     int64     `json:"count"`
	Timestamp time.Time `json:"timestamp"`
}

type S3Config struct {
	Endpoint     string
	Bucket       string
	AccessKey    string
	SecretKey    string
	Region       string
	PathTemplate string
}

type S3File struct {
	Key                 string
	Size                int64
	LastModified        time.Time
	DetectedFormat      string
	DetectedCompression string
	Date                time.Time // Extracted from filename
}

type SampleDiff struct {
	Source1Sample []map[string]interface{} `json:"source1_sample,omitempty"`
	Source2Sample []map[string]interface{} `json:"source2_sample,omitempty"`
	Differences   []string                 `json:"differences"`
}

type SchemaComparisonResult struct {
	TablesOnlyInSource1 []string                    `json:"tables_only_in_source1"`
	TablesOnlyInSource2 []string                    `json:"tables_only_in_source2"`
	TableDiffs          map[string]*TableSchemaDiff `json:"table_diffs"`
}

type StatusResponse struct {
	ArchiverRunning bool      `json:"archiverRunning"`
	PID             int       `json:"pid,omitempty"`
	CurrentTask     *TaskInfo `json:"currentTask,omitempty"`
	Version         string    `json:"version"`
	UpdateAvailable bool      `json:"updateAvailable"`
	LatestVersion   string    `json:"latestVersion,omitempty"`
	ReleaseURL      string    `json:"releaseUrl,omitempty"`
	// Slice tracking fields
	CurrentSliceIndex int    `json:"currentSliceIndex,omitempty"`
	TotalSlices       int    `json:"totalSlices,omitempty"`
	CurrentSliceDate  string `json:"currentSliceDate,omitempty"`
	IsSlicing         bool   `json:"isSlicing"`
}

type TableCache struct {
	TableName string       `json:"tableName"`
	Entries   []CacheEntry `json:"entries"`
}

type TableSchema struct {
	TableName string
	Columns   []ColumnInfo
}

type TableSchemaDiff struct {
	ColumnsOnlyInSource1 []ColumnInfo         `json:"columns_only_in_source1"`
	ColumnsOnlyInSource2 []ColumnInfo         `json:"columns_only_in_source2"`
	TypeMismatches       []ColumnTypeMismatch `json:"type_mismatches"`
}

type TaskInfo struct {
	PID              int       `json:"pid"`
	StartTime        time.Time `json:"start_time"`
	Table            string    `json:"table"`
	StartDate        string    `json:"start_date"`
	EndDate          string    `json:"end_date"`
	CurrentTask      string    `json:"current_task"`
	CurrentPartition string    `json:"current_partition,omitempty"`
	CurrentStep      string    `json:"current_step,omitempty"`
	Progress         float64   `json:"progress"`
	TotalItems       int       `json:"total_items"`
	CompletedItems   int       `json:"completed_items"`
	LastUpdate       time.Time `json:"last_update"`
	// Slice tracking fields
	CurrentSliceIndex int    `json:"current_slice_index,omitempty"`
	TotalSlices       int    `json:"total_slices,omitempty"`
	CurrentSliceDate  string `json:"current_slice_date,omitempty"`
	// Partition and slice statistics
	TotalPartitions     int `json:"total_partitions,omitempty"`     // Total partitions discovered
	PartitionsCounted   int `json:"partitions_counted,omitempty"`   // Partitions that have been counted
	PartitionsProcessed int `json:"partitions_processed,omitempty"` // Partitions that have been processed
	SlicesProcessed     int `json:"slices_processed,omitempty"`     // Total slices processed across all partitions
}

type VersionCheckCache struct {
	UpdateAvailable bool      `json:"update_available"`
	LatestVersion   string    `json:"latest_version"`
	ReleaseURL      string    `json:"release_url"`
	Timestamp       time.Time `json:"timestamp"`
}

type VersionCheckResult struct {
	UpdateAvailable bool
	CurrentVersion  string
	LatestVersion   string
	ReleaseURL      string
	Error           error
}

type WSMessage struct {
	Type string      `json:"type"`
	Data interface{} `json:"data"`
}