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)
	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"`
}