redshiftdata

type ClusterEngines interface {
	EngineLogin(region, clusterID string) (EngineLogin, bool)
}

type ColumnMetadata struct {
	Name            string `json:"name"`
	Label           string `json:"label"`
	TypeName        string `json:"typeName"`
	SchemaName      string `json:"schemaName"`
	TableName       string `json:"tableName"`
	Length          int32  `json:"length"`
	Nullable        int32  `json:"nullable"`
	Precision       int32  `json:"precision"`
	Scale           int32  `json:"scale"`
	IsCaseSensitive bool   `json:"isCaseSensitive"`
	IsCurrency      bool   `json:"isCurrency"`
	IsSigned        bool   `json:"isSigned"`
}

type ConfigProvider interface {
	GetRedshiftDataSettings() Settings
}

type EngineLogin struct {
	Addr     string
	User     string
	Password string
	DBName   string
}

type Field struct {
	BooleanValue *bool    `json:"booleanValue,omitempty"`
	DoubleValue  *float64 `json:"doubleValue,omitempty"`
	LongValue    *int64   `json:"longValue,omitempty"`
	StringValue  *string  `json:"stringValue,omitempty"`
	BlobValue    *string  `json:"blobValue,omitempty"`
	IsNull       bool     `json:"isNull,omitempty"`
}

type Handler struct {
	Backend StorageBackend

	AccountID string
	Region    string
	// contains filtered or unexported fields
}

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

type Janitor struct {
	Backend      *InMemoryBackend
	Interval     time.Duration
	StatementTTL time.Duration
	// TaskTimeout bounds each individual sweep task. When zero, each task runs
	// without a deadline. When non-zero, a child context with this timeout is
	// created for each sweep pass, preventing a stalled operation from blocking
	// the janitor loop indefinitely.
	TaskTimeout time.Duration
}

type ListSessionsFilter struct {
	RoleLevel         *bool
	ClusterIdentifier string
	WorkgroupName     string
	Database          string
	SessionID         string
	Status            string
	NextToken         string
	MaxResults        int
}

type ListStatementsFilter struct {
	RoleLevel         *bool
	ClusterIdentifier string
	WorkgroupName     string
	Database          string
	StatementName     string
	Status            string
	NextToken         string
	MaxResults        int
}

type ObjectStore interface {
	GetObject(ctx context.Context, bucket, key string) ([]byte, error)
	PutObject(ctx context.Context, bucket, key string, body []byte) error
	DeleteObject(ctx context.Context, bucket, key string) error
	// ListKeys returns every key under prefix in ascending order.
	ListKeys(ctx context.Context, bucket, prefix string) ([]string, error)
}

type Provider struct{}

type ResultSet struct {
	Columns []ColumnMetadata
	Rows    [][]Field
}

type SQLExecutor interface {
	ExecuteSQL(ctx context.Context, login EngineLogin, sqls []string, params []SQLParameter, atomic bool) []SQLOutcome
}

type SQLOutcome struct {
	Err    error
	Result *ResultSet
}

type SQLParameter struct {
	Name  string `json:"name"`
	Value string `json:"value"`
}

type SessionData struct {
	CreatedAt time.Time `json:"createdAt"`
	UpdatedAt time.Time `json:"updatedAt"`
	TTL       time.Time `json:"ttl"`

	SessionID          string `json:"sessionId"`
	ClusterIdentifier  string `json:"clusterIdentifier,omitempty"`
	WorkgroupName      string `json:"workgroupName,omitempty"`
	Database           string `json:"database,omitempty"`
	DBUser             string `json:"dbUser,omitempty"`
	Status             string `json:"status"`
	CurrentStatementID string `json:"-"`
	AliveSeconds       int32  `json:"aliveSeconds,omitempty"`
	// contains filtered or unexported fields
}

type Settings struct {
	JanitorInterval time.Duration `json:"janitor_interval" env:"REDSHIFTDATA_JANITOR_INTERVAL" default:"1m"  help:"Janitor tick interval."` //nolint:lll // Kong struct tag makes this line long
	StatementTTL    time.Duration ``                                                                                                        //nolint:lll // Kong struct tag makes this line long
	/* 134-byte string literal not displayed */
}

type Statement struct {
	CreatedAt time.Time `json:"createdAt"`
	UpdatedAt time.Time `json:"updatedAt"`

	ID                      string             `json:"id"`
	SessionID               string             `json:"sessionID,omitempty"`
	WorkgroupName           string             `json:"workgroupName"`
	QueryString             string             `json:"queryString"`
	DBUser                  string             `json:"dbUser"`
	SecretARN               string             `json:"secretARN"`
	StatementName           string             `json:"statementName"`
	ResultFormat            string             `json:"resultFormat"`
	Owner                   string             `json:"owner,omitempty"`
	OwnerSession            string             `json:"ownerSession,omitempty"`
	ExecutionMode           string             `json:"executionMode,omitempty"`
	Error                   string             `json:"error"`
	Database                string             `json:"database"`
	ClusterIdentifier       string             `json:"clusterIdentifier"`
	Status                  string             `json:"status"`
	SubStatements           []SubStatementData `json:"subStatements,omitempty"`
	QueryStrings            []string           `json:"queryStrings"`
	Parameters              []SQLParameter     `json:"parameters,omitempty"`
	DurationMs              int64              `json:"durationMs"`
	ResultRows              int64              `json:"resultRows"`
	ResultSize              int64              `json:"resultSize"`
	SessionKeepAliveSeconds int32              `json:"sessionKeepAliveSeconds,omitempty"`
	HasResultSet            bool               `json:"hasResultSet"`
	IsBatchStatement        bool               `json:"isBatchStatement"`
	WithEvent               bool               `json:"withEvent"`
	EngineBacked            bool               `json:"engineBacked,omitempty"`
	// contains filtered or unexported fields
}

type StorageBackend interface {
	// Statement execution
	ExecuteStatement(
		ctx context.Context,
		sql, clusterIdentifier, workgroupName, database, dbUser, secretARN, statementName string,
		withEvent bool, resultFormat string,
		parameters []SQLParameter,
		sessionID string,
	) (*Statement, error)
	BatchExecuteStatement(
		ctx context.Context,
		sqls []string, clusterIdentifier, workgroupName, database, dbUser, secretARN, statementName string,
		withEvent bool, resultFormat string,
		parameters []SQLParameter,
		sessionID, executionMode string,
	) (*Statement, error)

	// Statement inspection
	DescribeStatement(ctx context.Context, id string) (*Statement, error)
	// EngineResult returns the result set of a statement run on a live engine; found is false otherwise.
	EngineResult(ctx context.Context, id string) (rs *ResultSet, found bool, err error)
	WaitForStatement(ctx context.Context, id string, wait time.Duration) (*Statement, error)
	CancelStatement(ctx context.Context, id string) error
	// ListStatements returns a page of statements and a next-token for pagination.
	ListStatements(ctx context.Context, filter ListStatementsFilter) (
		[]*Statement, string, error,
	)

	// Sessions
	SetStatementSessionKeepAlive(ctx context.Context, id string, seconds int32)
	// ListSessions returns a page of sessions -- derived from stored statements
	// that share a SessionID, not a separately stored resource -- and a
	// next-token for pagination.
	ListSessions(ctx context.Context, filter ListSessionsFilter) (
		[]*SessionData, string, error,
	)

	// Maintenance
	// EvictExpiredStatements removes terminal statements older than cutoff.
	// Returns the number of evicted statements.
	EvictExpiredStatements(cutoff time.Time) int

	// Lifecycle
	Reset()
	Region() string
	AccountID() string
	Snapshot(ctx context.Context) []byte
	Restore(ctx context.Context, data []byte) error
}

type SubStatementData struct {
	ID           string    `json:"id"`
	CreatedAt    time.Time `json:"createdAt"`
	UpdatedAt    time.Time `json:"updatedAt"`
	QueryString  string    `json:"queryString"`
	Status       string    `json:"status"`
	Error        string    `json:"error"`
	HasResultSet bool      `json:"hasResultSet"`
	ResultRows   int64     `json:"resultRows"`
	ResultSize   int64     `json:"resultSize"`
	DurationMs   int64     `json:"durationMs"`
}

type WorkgroupEngines interface {
	WorkgroupEngineLogin(region, name string) (EngineLogin, bool)
}

type WorkgroupResolver interface {
	WorkgroupStatus(region, name string) (status string, found bool)
}