const (
	FeatureEnabled  = feature.Enabled
	FeatureDisabled = feature.Disabled
)

const (
	IndexTypeGlobal = IndexType(iota)
	IndexTypeGlobalAsync
)

const (
	ChangefeedStateUnspecified = ChangefeedState(Ydb_Table.ChangefeedDescription_STATE_UNSPECIFIED)
	ChangefeedStateEnabled     = ChangefeedState(Ydb_Table.ChangefeedDescription_STATE_ENABLED)
	ChangefeedStateDisabled    = ChangefeedState(Ydb_Table.ChangefeedDescription_STATE_DISABLED)
)

const (
	ChangefeedModeUnspecified     = ChangefeedMode(Ydb_Table.ChangefeedMode_MODE_UNSPECIFIED)
	ChangefeedModeKeysOnly        = ChangefeedMode(Ydb_Table.ChangefeedMode_MODE_KEYS_ONLY)
	ChangefeedModeUpdates         = ChangefeedMode(Ydb_Table.ChangefeedMode_MODE_UPDATES)
	ChangefeedModeNewImage        = ChangefeedMode(Ydb_Table.ChangefeedMode_MODE_NEW_IMAGE)
	ChangefeedModeOldImage        = ChangefeedMode(Ydb_Table.ChangefeedMode_MODE_OLD_IMAGE)
	ChangefeedModeNewAndOldImages = ChangefeedMode(Ydb_Table.ChangefeedMode_MODE_NEW_AND_OLD_IMAGES)
)

const (
	ChangefeedFormatUnspecified         = ChangefeedFormat(Ydb_Table.ChangefeedFormat_FORMAT_UNSPECIFIED)
	ChangefeedFormatJSON                = ChangefeedFormat(Ydb_Table.ChangefeedFormat_FORMAT_JSON)
	ChangefeedFormatDynamoDBStreamsJSON = ChangefeedFormat(Ydb_Table.ChangefeedFormat_FORMAT_DYNAMODB_STREAMS_JSON)
)

const (
	StoreTypeUnspecified = StoreType(Ydb_Table.StoreType_STORE_TYPE_UNSPECIFIED)
	StoreTypeRow         = StoreType(Ydb_Table.StoreType_STORE_TYPE_ROW)
	StoreTypeColumn      = StoreType(Ydb_Table.StoreType_STORE_TYPE_COLUMN)
)

const (
	ExecuteScanQueryStatsTypeNone = iota
	ExecuteScanQueryStatsTypeBasic
	ExecuteScanQueryStatsTypeFull
)

type AlterTableDesc Ydb_Table.AlterTableRequest

type AlterTableOption interface {
	ApplyAlterTableOption(desc *AlterTableDesc, a *allocator.Allocator)
}

type BulkUpsertOption interface {
	ApplyBulkUpsertOption() []grpc.CallOption
}

type CachingPolicyDescription struct {
	Name   string
	Labels map[string]string
}

type CachingPolicyOption func(*cachingPolicy)

type ChangefeedDescription struct {
	Name             string
	Mode             ChangefeedMode
	Format           ChangefeedFormat
	State            ChangefeedState
	VirtualTimestamp bool
}

type ChangefeedFormat int

type ChangefeedMode int

type ChangefeedState int

type Column struct {
	Name   string
	Type   types.Type
	Family string
}

type ColumnFamily struct {
	Name         string
	Data         StoragePool
	Compression  ColumnFamilyCompression
	KeepInMemory FeatureFlag
}

type ColumnFamilyCompression byte

const (
	ColumnFamilyCompressionUnknown ColumnFamilyCompression = iota
	ColumnFamilyCompressionNone
	ColumnFamilyCompressionLZ4
)

type CommitTransactionDesc Ydb_Table.CommitTransactionRequest

type CommitTransactionOption func(*CommitTransactionDesc)

type CompactionPolicyDescription struct {
	Name   string
	Labels map[string]string
}

type CompactionPolicyOption func(*compactionPolicy)

type CopyTableDesc Ydb_Table.CopyTableRequest

type CopyTableOption func(*CopyTableDesc)

type CopyTablesDesc Ydb_Table.CopyTablesRequest

type CopyTablesOption func(*CopyTablesDesc)

type CreateTableDesc Ydb_Table.CreateTableRequest

type CreateTableOption interface {
	ApplyCreateTableOption(d *CreateTableDesc, a *allocator.Allocator)
}

type DescribeTableDesc Ydb_Table.DescribeTableRequest

type DescribeTableOption func(d *DescribeTableDesc)

type Description struct {
	Name                 string
	Columns              []Column
	PrimaryKey           []string
	KeyRanges            []KeyRange
	Stats                *TableStats
	ColumnFamilies       []ColumnFamily
	Attributes           map[string]string
	ReadReplicaSettings  ReadReplicasSettings
	StorageSettings      StorageSettings
	KeyBloomFilter       FeatureFlag
	PartitioningSettings PartitioningSettings
	Indexes              []IndexDescription
	TimeToLiveSettings   *TimeToLiveSettings
	Changefeeds          []ChangefeedDescription
	Tiering              string
	StoreType            StoreType
}

type DropTableDesc Ydb_Table.DropTableRequest

type DropTableOption interface {
	ApplyDropTableOption(desc *DropTableDesc)
}

type ExecuteDataQueryDesc struct {
	*Ydb_Table.ExecuteDataQueryRequest

	IgnoreTruncated bool
}

type ExecuteDataQueryOption interface {
	ApplyExecuteDataQueryOption(d *ExecuteDataQueryDesc, a *allocator.Allocator) []grpc.CallOption
}

type ExecuteScanQueryDesc Ydb_Table.ExecuteScanQueryRequest

type ExecuteScanQueryOption interface {
	ApplyExecuteScanQueryOption(d *ExecuteScanQueryDesc) []grpc.CallOption
}

type ExecuteScanQueryRequestMode byte

const (
	ExecuteScanQueryRequestModeExec ExecuteScanQueryRequestMode = iota
	ExecuteScanQueryRequestModeExplain
)

type ExecuteScanQueryStatsType uint32

type ExecuteSchemeQueryDesc Ydb_Table.ExecuteSchemeQueryRequest

type ExecuteSchemeQueryOption func(*ExecuteSchemeQueryDesc)

type ExecutionPolicyDescription struct {
	Name   string
	Labels map[string]string
}

type ExecutionPolicyOption func(*executionPolicy)

type FeatureFlag = feature.Flag

type IndexDescription struct {
	Name         string
	IndexColumns []string
	DataColumns  []string
	Status       Ydb_Table.TableIndexDescription_Status
	Type         IndexType
}

type IndexOption interface {
	ApplyIndexOption(d *indexDesc)
}

type IndexType uint8

type KeyRange struct {
	From value.Value
	To   value.Value
}

type PartitionStats struct {
	RowsEstimate uint64
	StoreSize    uint64
	LeaderNodeID uint32
}

type PartitioningMode byte

const (
	PartitioningUnknown PartitioningMode = iota
	PartitioningDisabled
	PartitioningAutoSplit
	PartitioningAutoSplitMerge
)

type PartitioningPolicyDescription struct {
	Name   string
	Labels map[string]string
}

type PartitioningPolicyOption func(*partitioningPolicy, *allocator.Allocator)

type PartitioningSettings struct {
	PartitioningBySize FeatureFlag
	PartitionSizeMb    uint64
	PartitioningByLoad FeatureFlag
	MinPartitionsCount uint64
	MaxPartitionsCount uint64
}

type PartitioningSettingsOption interface {
	ApplyPartitioningSettingsOption(settings *ydbPartitioningSettings)
}

type Partitions interface {
	CreateTableOption
	// contains filtered or unexported methods
}

type ProfileOption interface {
	ApplyProfileOption(p *profile, a *allocator.Allocator)
}

type QueryCachePolicyOption func(*queryCachePolicy, *allocator.Allocator)

type ReadReplicasSettings struct {
	Type  ReadReplicasType
	Count uint64
}

type ReadReplicasType byte

const (
	ReadReplicasPerAzReadReplicas ReadReplicasType = iota
	ReadReplicasAnyAzReadReplicas
)

type ReadRowsDesc Ydb_Table.ReadRowsRequest

type ReadRowsOption interface {
	ApplyReadRowsOption(desc *ReadRowsDesc, a *allocator.Allocator)
}

type ReadTableDesc Ydb_Table.ReadTableRequest

type ReadTableOption interface {
	ApplyReadTableOption(desc *ReadTableDesc, a *allocator.Allocator)
}

type RenameTablesDesc Ydb_Table.RenameTablesRequest

type RenameTablesOption func(desc *RenameTablesDesc)

type ReplicationPolicyDescription struct {
	Name   string
	Labels map[string]string
}

type ReplicationPolicyOption func(*replicationPolicy)

type StoragePolicyDescription struct {
	Name   string
	Labels map[string]string
}

type StoragePolicyOption func(*storagePolicy)

type StoragePool struct {
	Media string
}

type StorageSettings struct {
	TableCommitLog0    StoragePool
	TableCommitLog1    StoragePool
	External           StoragePool
	StoreExternalBlobs FeatureFlag
}

type StoreType int

type TTLSettings struct {
	DateTimeColumn string
	TTLSeconds     uint32
}

type TableOptionsDescription struct {
	TableProfilePresets       []TableProfileDescription
	StoragePolicyPresets      []StoragePolicyDescription
	CompactionPolicyPresets   []CompactionPolicyDescription
	PartitioningPolicyPresets []PartitioningPolicyDescription
	ExecutionPolicyPresets    []ExecutionPolicyDescription
	ReplicationPolicyPresets  []ReplicationPolicyDescription
	CachingPolicyPresets      []CachingPolicyDescription
}

type TableProfileDescription struct {
	Name   string
	Labels map[string]string

	DefaultStoragePolicy      string
	DefaultCompactionPolicy   string
	DefaultPartitioningPolicy string
	DefaultExecutionPolicy    string
	DefaultReplicationPolicy  string
	DefaultCachingPolicy      string

	AllowedStoragePolicies      []string
	AllowedCompactionPolicies   []string
	AllowedPartitioningPolicies []string
	AllowedExecutionPolicies    []string
	AllowedReplicationPolicies  []string
	AllowedCachingPolicies      []string
}

type TableStats struct {
	PartitionStats   []PartitionStats
	RowsEstimate     uint64
	StoreSize        uint64
	Partitions       uint64
	CreationTime     time.Time
	ModificationTime time.Time
}

type TimeToLiveMode byte

const (
	TimeToLiveModeDateType TimeToLiveMode = iota
	TimeToLiveModeValueSinceUnixEpoch
)

type TimeToLiveSettings struct {
	ColumnName string

	// Mode specified mode
	Mode TimeToLiveMode

	// ExpireAfterSeconds specified expiration in seconds
	ExpireAfterSeconds uint32

	// ColumnUnit valid with Mode = TimeToLiveModeValueSinceUnixEpoch
	ColumnUnit *TimeToLiveUnit
}

type TimeToLiveUnit int32

const (
	TimeToLiveUnitUnspecified TimeToLiveUnit = iota
	TimeToLiveUnitSeconds
	TimeToLiveUnitMilliseconds
	TimeToLiveUnitMicroseconds
	TimeToLiveUnitNanoseconds
)