timescale

type AggregateFunction string

type ColumnAggregation struct {
	Function AggregateFunction
	Column   string
	Alias    string
}

type ColumnMetadata struct {
	Name         string
	Type         string
	Nullable     bool
	IsPrimaryKey bool
	IsIndexed    bool
	Description  string
}

type CompressionPolicy struct {
	Enabled       bool
	After         string // e.g., "7 days"
	OrderBy       string // Column to order by during compression
	SegmentBy     string // Column to segment by during compression
	CompressChunk bool   // Whether to manually compress chunks
}

type CompressionSettings struct {
	HypertableName      string
	SegmentBy           string
	OrderBy             string
	ChunkTimeInterval   string
	CompressionInterval string
	CompressionEnabled  bool
}

type ContinuousAggregateMetadata struct {
	ViewName           string
	ViewSchema         string
	MaterializedOnly   bool
	RefreshInterval    string
	RefreshLag         string
	RefreshStartOffset string
	RefreshEndOffset   string
	HypertableName     string
	HypertableSchema   string
	ViewDefinition     string
}

type ContinuousAggregateOptions struct {
	// Required parameters
	ViewName       string // Name of the continuous aggregate view to create
	SourceTable    string // Source table with raw data
	TimeColumn     string // Time column to bucket
	BucketInterval string // Time bucket interval (e.g., '1 hour', '1 day')

	// Optional parameters
	Aggregations      []ColumnAggregation // Aggregations to include in the view
	WhereCondition    string              // WHERE condition to filter source data
	WithData          bool                // Whether to materialize data immediately (WITH DATA)
	RefreshPolicy     bool                // Whether to add a refresh policy
	RefreshInterval   string              // Refresh interval (default: '1 day')
	RefreshLookback   string              // How far back to look when refreshing (default: '1 week')
	MaterializedOnly  bool                // Whether to materialize only (no real-time)
	CreateIfNotExists bool                // Whether to use IF NOT EXISTS
}

type ContinuousAggregatePolicyOptions struct {
	ViewName         string // Name of the continuous aggregate view
	Start            string // Start offset (e.g., '-2 days')
	End              string // End offset (e.g., 'now()')
	ScheduleInterval string // Execution interval (e.g., '1 hour')
}

type DB struct {
	db.Database // Embed standard Database interface
	// contains filtered or unexported fields
}

type DBConfig struct {
	// Inherit PostgreSQL config
	PostgresConfig db.Config

	// TimescaleDB-specific settings
	ChunkTimeInterval string             // Default chunk time interval (e.g., "7 days")
	RetentionPolicy   *RetentionPolicy   // Data retention configuration
	CompressionPolicy *CompressionPolicy // Compression configuration
	UseTimescaleDB    bool               // Enable TimescaleDB features (default: true)
}

type DownsampleOptions struct {
	SourceTable       string
	DestTable         string
	TimeColumn        string
	BucketInterval    string
	Aggregations      []ColumnAggregation
	WhereCondition    string
	CreateTable       bool
	ChunkTimeInterval string
}

type Hypertable struct {
	TableName          string
	SchemaName         string
	TimeColumn         string
	SpaceColumn        string
	NumDimensions      int
	CompressionEnabled bool
	RetentionEnabled   bool
}

type HypertableConfig struct {
	TableName          string
	TimeColumn         string
	ChunkTimeInterval  string
	PartitioningColumn string
	CreateIfNotExists  bool
	SpacePartitions    int  // Number of space partitions (for multi-dimensional partitioning)
	IfNotExists        bool // If true, don't error if table is already a hypertable
	MigrateData        bool // If true, migrate existing data to chunks
}

type HypertableMetadata struct {
	TableName         string
	SchemaName        string
	Owner             string
	NumDimensions     int
	TimeDimension     string
	TimeDimensionType string
	SpaceDimensions   []string
	ChunkTimeInterval string
	Compression       bool
	RetentionPolicy   bool
	TotalSize         string
	TotalRows         int64
	Chunks            int
}

type HypertableOption func(*HypertableConfig)

type RetentionPolicy struct {
	Enabled    bool
	Duration   string // e.g., "90 days"
	DropChunks bool   // Whether to physically drop chunks (vs logical deletion)
}

type RetentionSettings struct {
	HypertableName    string
	RetentionInterval string
	RetentionEnabled  bool
}

type TimeBucket struct {
	Interval string // e.g., '1 hour', '1 day', '1 month'
	Column   string // Time column to bucket
	Alias    string // Optional alias for the bucket column
}

type TimeRange struct {
	Start time.Time
	End   time.Time
}

type TimeSeriesQueryOptions struct {
	// Required parameters
	Table          string // The table to query
	TimeColumn     string // The time column
	BucketInterval string // Time bucket interval (e.g., '1 hour', '1 day')

	// Optional parameters
	BucketColumnName string              // Name for the bucket column (defaults to "time_bucket")
	SelectColumns    []string            // Additional columns to select
	Aggregations     []ColumnAggregation // Aggregations to perform
	WindowFunctions  []WindowFunction    // Window functions to apply
	StartTime        time.Time           // Start of time range
	EndTime          time.Time           // End of time range
	WhereCondition   string              // Additional WHERE conditions
	GroupByColumns   []string            // Additional GROUP BY columns
	OrderBy          string              // ORDER BY clause
	Limit            int                 // LIMIT clause
	Offset           int                 // OFFSET clause
}

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

type WindowFunction struct {
	Function    string // e.g. LAG, LEAD, ROW_NUMBER
	Expression  string // Expression to apply function to
	Alias       string // Result column name
	PartitionBy string // PARTITION BY column(s)
	OrderBy     string // ORDER BY column(s)
	Frame       string // Window frame specification
}