types

type BlockCountComposite struct {
	NodePosition int64 `db:"node_position"`
	RangeStart   any   `db:"range_start"`
	RangeEnd     any   `db:"range_end"`
	Count        int64 `db:"cnt"`
}

type BlockCountSimple struct {
	NodePosition int64 `db:"node_position"`
	RangeStart   any   `db:"range_start"`
	RangeEnd     any   `db:"range_end"`
	Count        int64 `db:"count"`
}

type BlockRange struct {
	NodePosition int64 `db:"node_position"`
	RangeStart   []any `db:"range_start"`
	RangeEnd     []any `db:"range_end"`
}

type BlockRangeWithCount struct {
	NodePosition int64 `db:"node_position"`
	RangeStart   []any `db:"range_start"`
	RangeEnd     []any `db:"range_end"`
	Count        int64 `db:"count"`
}

type ClusterConfig struct {
	JSONVersion string `json:"json_version"`
	ClusterName string `json:"cluster_name"`
	LogLevel    string `json:"log_level"`
	UpdateDate  string `json:"update_date"`
	PGEdge      struct {
		PGVersion int         `json:"pg_version"`
		AutoStart string      `json:"auto_start"`
		Spock     SpockConfig `json:"spock"`
		Databases []Database  `json:"databases"`
	} `json:"pgedge"`
	NodeGroups []NodeGroup `json:"node_groups"`
}

type ColumnType struct {
	ColumnName string `db:"column_name"`
	DataType   string `db:"data_type"`
}

type Database struct {
	DBName      string `json:"db_name"`
	DBUser      string `json:"db_user"`
	DBPassword  string `json:"db_password"`
	SSLMode     string `json:"ssl_mode"`
	SSLCert     string `json:"ssl_cert"`
	SSLKey      string `json:"ssl_key"`
	SSLRootCert string `json:"ssl_root_cert"`
}

type DerivedFields struct {
	ClusterNodes     []map[string]any
	Schema           string
	Table            string
	Key              []string
	Cols             []string
	ConnParams       []string
	Database         Database
	NodeList         []string
	HostMap          map[string]string
	TableList        []string
	ColTypes         map[string]map[string]string
	SimplePrimaryKey bool
	PKeyTypes        map[string]string
}

type DiffByNodePair struct {
	Rows map[string][]OrderedMap `json:"rows"` // Keyed by actual node name e.g. "n1", "n2"
}

type DiffOutput struct {
	NodeDiffs map[string]DiffByNodePair `json:"diffs"` // Key: "nodeA/nodeB" (sorted names)
	Summary   DiffSummary               `json:"summary"`
}

type DiffSummary struct {
	Schema                string         `json:"schema"`
	Table                 string         `json:"table"`
	TableFilter           string         `json:"table_filter,omitempty"`
	Nodes                 []string       `json:"nodes"`
	BlockSize             int            `json:"block_size"`
	CompareUnitSize       int            `json:"compare_unit_size"`
	ConcurrencyFactor     int            `json:"concurrency_factor"`
	MaxDiffRows           int64          `json:"max_diff_rows"`
	StartTime             string         `json:"start_time"`
	EndTime               string         `json:"end_time"`
	TimeTaken             string         `json:"time_taken"`
	DiffRowsCount         map[string]int `json:"diff_rows_count"` // Key: "nodeA/nodeB", Value: count of differing rows
	DiffRowLimitReached   bool           `json:"diff_row_limit_reached"`
	TotalRowsChecked      int64          `json:"total_rows_checked"` // Estimated
	InitialRangesCount    int            `json:"initial_ranges_count"`
	MismatchedRangesCount int            `json:"mismatched_ranges_count"`
	PrimaryKey            []string       `json:"primary_key"`
	EffectiveFilter       string         `json:"effective_filter,omitempty"`
	AgainstOrigin         string         `json:"against_origin,omitempty"`
	AgainstOriginResolved string         `json:"against_origin_resolved,omitempty"`
	Until                 string         `json:"until,omitempty"`
	OriginOnly            bool           `json:"origin_only,omitempty"`
}

type KVPair struct {
	Key   string
	Value any
}

type LeafRange struct {
	RangeStart []any `db:"range_start"`
	RangeEnd   []any `db:"range_end"`
}

type MtreeMetadata struct {
	SchemaName  string    `db:"schema_name"`
	TableName   string    `db:"table_name"`
	TotalRows   int64     `db:"total_rows"`
	BlockSize   int       `db:"block_size"`
	NumBlocks   int       `db:"num_blocks"`
	IsComposite bool      `db:"is_composite"`
	LastUpdated time.Time `db:"last_updated"`
}

type NodeChild struct {
	NodeLevel    int    `db:"node_level"`
	NodePosition int64  `db:"node_position"`
	NodeHash     []byte `db:"node_hash"`
}

type NodeGroup struct {
	SSH struct {
		OSUser     string `json:"os_user"`
		PrivateKey string `json:"private_key"`
	} `json:"ssh"`
	Name      string `json:"name"`
	IsActive  string `json:"is_active"`
	PublicIP  string `json:"public_ip"`
	PrivateIP string `json:"private_ip"`
	Port      string `json:"port"`
	Path      string `json:"path"`
}

type NodePairDiff struct {
	Node1OnlyRows []map[string]any
	Node2OnlyRows []map[string]any
	ModifiedRows  []struct {
		Pkey      string
		Node1Data map[string]any
		Node2Data map[string]any
	}
}

type OrderedMap []KVPair

type PkeyColumnType struct {
	ColumnName string
	DataType   string
}

type PkeyOffset struct {
	RangeStart []any
	RangeEnd   []any
}

type RepSetInfo struct {
	SetName string   `json:"set_name"`
	Tables  []string `json:"tables"`
}

type ReplicationSetDiff struct {
	TablePlacementDiffs []TableRepSetDiff `json:"table_placement_differences"`
}

type ReplicationSetPair struct {
	Name  string     `json:"name"`
	Node1 RepSetInfo `json:"node1"`
	Node2 RepSetInfo `json:"node2"`
}

type RootNode struct {
	NodePosition int64  `db:"node_position"`
	NodeHash     []byte `db:"node_hash"`
}

type SpockConfig struct {
	SpockVersion string `json:"spock_version"`
	AutoDDL      string `json:"auto_ddl"`
}

type SpockDiffDetail struct {
	Subscriptions   SubscriptionDiff   `json:"subscriptions"`
	ReplicationSets ReplicationSetDiff `json:"replication_sets"`
}

type SpockDiffOutput struct {
	SpockConfigs map[string]any           `json:"spock_config"`
	Diffs        map[string]SpockPairDiff `json:"diffs"`
}

type SpockNodeAndSubInfo struct {
	NodeID             int64    `db:"node_id"`
	NodeName           string   `db:"node_name"`
	Location           string   `db:"location"`
	Country            string   `db:"country"`
	SubID              int64    `db:"sub_id"`
	SubName            string   `db:"sub_name"`
	SubEnabled         bool     `db:"sub_enabled"`
	SubReplicationSets []string `db:"sub_replication_sets"`
}

type SpockPairDiff struct {
	Mismatch bool            `json:"mismatch"`
	Message  string          `json:"message"`
	Details  SpockDiffDetail `json:"details,omitempty"`
}

type SpockRepSetInfo struct {
	SetName string   `db:"set_name"`
	RelName []string `db:"relname"`
}

type SpockSubscription struct {
	SubName         string   `json:"sub_name"`
	SubEnabled      bool     `json:"sub_enabled"`
	ReplicationSets []string `json:"replication_sets"`
}

type SubscriptionDiff struct {
	MissingOnNode1 []string           `json:"missing_on_node1"`
	MissingOnNode2 []string           `json:"missing_on_node2"`
	Different      []SubscriptionPair `json:"different"`
}

type SubscriptionPair struct {
	Name  string            `json:"name"`
	Node1 SpockSubscription `json:"node1"`
	Node2 SpockSubscription `json:"node2"`
}

type TableRepSetDiff struct {
	TableName   string `json:"table_name"`
	Node1RepSet string `json:"node1_repset"`
	Node2RepSet string `json:"node2_repset"`
}

type Task struct {
	ClusterName string
	TaskID      string
	TaskType    string
	TaskStatus  string
	TaskContext string
	StartedAt   time.Time
	FinishedAt  time.Time
	TimeTaken   float64
}

type UserPrivileges struct {
	TableSelect            bool `db:"table_select"`
	TableCreate            bool `db:"table_create"`
	TableInsert            bool `db:"table_insert"`
	TableUpdate            bool `db:"table_update"`
	TableDelete            bool `db:"table_delete"`
	ColumnsSelect          bool `db:"columns_select"`
	TableConstraintsSelect bool `db:"table_constraints_select"`
	KeyColumnUsageSelect   bool `db:"key_column_usage_select"`
}