kinesis

type ConfigProvider interface {
	GetKinesisSettings() Settings
}

type Consumer struct {
	ConsumerCreationTimestamp time.Time `json:"consumerCreationTimestamp"`
	ConsumerName              string    `json:"consumerName"`
	ConsumerARN               string    `json:"consumerARN"`
	ConsumerStatus            string    `json:"consumerStatus"`
	StreamARN                 string    `json:"streamARN"`
}

type CreateStreamInput struct {
	StreamName string
	Region     string
	AccountID  string
	StreamMode string
	ShardCount int
}

type DecreaseStreamRetentionPeriodInput struct {
	StreamName           string
	RetentionPeriodHours int
}

type DeleteResourcePolicyInput struct {
	ResourceARN string
}

type DeleteStreamInput struct {
	StreamName string
}

type DeregisterStreamConsumerInput struct {
	StreamARN    string
	ConsumerARN  string
	ConsumerName string
}

type DescribeAccountSettingsOutput struct {
	ShardLimit               int
	OnDemandStreamCount      int
	OnDemandStreamCountLimit int
}

type DescribeStreamConsumerInput struct {
	StreamARN    string
	ConsumerARN  string
	ConsumerName string
}

type DescribeStreamConsumerOutput struct {
	ConsumerDescription Consumer
}

type DescribeStreamInput struct {
	StreamName string
	// ExclusiveStartShardID resumes shard pagination after the given shard ID.
	ExclusiveStartShardID string
	// Limit caps the number of ShardDescription entries returned (AWS default
	// 100, max 10000). Zero means "use the AWS default".
	Limit int
}

type DescribeStreamOutput struct {
	StreamCreationTimestamp time.Time
	StreamName              string
	StreamARN               string
	StreamStatus            string
	EncryptionType          string
	StreamMode              string
	KeyID                   string
	Shards                  []ShardDescription
	EnhancedMonitoring      []string
	RetentionPeriodHours    int
	// HasMoreShards indicates the shard list was truncated by Limit and more
	// shards can be fetched with a follow-up call using ExclusiveStartShardID.
	HasMoreShards bool
}

type DisableEnhancedMonitoringInput struct {
	StreamName        string
	ShardLevelMetrics []string
}

type DisableEnhancedMonitoringOutput struct {
	StreamName               string
	CurrentShardLevelMetrics []string
	DesiredShardLevelMetrics []string
}

type EnableEnhancedMonitoringInput struct {
	StreamName        string
	ShardLevelMetrics []string
}

type EnableEnhancedMonitoringOutput struct {
	StreamName               string
	CurrentShardLevelMetrics []string
	DesiredShardLevelMetrics []string
}

type GetRecordResult struct {
	ApproximateArrivalTimestamp time.Time
	PartitionKey                string
	SequenceNumber              string
	Data                        []byte
}

type GetRecordsInput struct {
	ShardIterator string
	Limit         int
}

type GetRecordsOutput struct {
	NextShardIterator  string
	Records            []GetRecordResult
	MillisBehindLatest int64
}

type GetResourcePolicyInput struct {
	ResourceARN string
}

type GetResourcePolicyOutput struct {
	Policy string
}

type GetShardIteratorInput struct {
	Timestamp              time.Time
	StreamName             string
	ShardID                string
	ShardIteratorType      string
	StartingSequenceNumber string
}

type GetShardIteratorOutput struct {
	ShardIterator string
}

type Handler struct {
	Backend StorageBackend

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

type InMemoryBackend struct {
	OnStreamPurged func(string)
	// contains filtered or unexported fields
}

type IncreaseStreamRetentionPeriodInput struct {
	StreamName           string
	RetentionPeriodHours int
}

type Janitor struct {
	Backend *InMemoryBackend

	Interval time.Duration
	// TaskTimeout bounds each individual janitor task. When non-zero, each task
	// runs with a child context that expires after this duration, preventing a
	// stalled operation from blocking the janitor loop indefinitely.
	TaskTimeout time.Duration
	// contains filtered or unexported fields
}

type ListShardsInput struct {
	StreamName            string
	NextToken             string
	ExclusiveStartShardID string
	// ShardFilter controls which shards are returned.
	// Supported values: "FROM_TRIM_HORIZON" (all shards including closed),
	// "AT_LATEST" (open shards only), "AFTER_SHARD_ID", "AT_TIMESTAMP", "FROM_TIMESTAMP".
	// Empty string defaults to open shards only.
	ShardFilter        string
	ShardFilterType    string
	ShardFilterShardID string
	MaxResults         int
}

type ListShardsOutput struct {
	NextToken string
	Shards    []ShardDescription
}

type ListStreamConsumersInput struct {
	StreamARN  string
	NextToken  string
	MaxResults int
}

type ListStreamConsumersOutput struct {
	NextToken string
	Consumers []Consumer
}

type ListStreamsInput struct {
	NextToken                string
	ExclusiveStartStreamName string
	Limit                    int
}

type ListStreamsOutput struct {
	NextToken      string
	StreamNames    []string
	HasMoreStreams bool
}

type ListTagsForResourceInput struct {
	ResourceARN string
}

type ListTagsForResourceOutput struct {
	Tags map[string]string
}

type MergeShardsInput struct {
	StreamName           string
	StreamARN            string
	ShardToMerge         string
	AdjacentShardToMerge string
}

type Provider struct{}

type PutRecordInput struct {
	StreamName      string
	PartitionKey    string
	ExplicitHashKey string
	Data            []byte
}

type PutRecordOutput struct {
	ShardID        string
	SequenceNumber string
	EncryptionType string
}

type PutRecordsEntry struct {
	PartitionKey    string
	ExplicitHashKey string
	Data            []byte
}

type PutRecordsInput struct {
	StreamName string
	Records    []PutRecordsEntry
}

type PutRecordsOutput struct {
	Records           []PutRecordsResultEntry
	FailedRecordCount int
}

type PutRecordsResultEntry struct {
	ShardID        string
	SequenceNumber string
	ErrorCode      string
	ErrorMessage   string
}

type PutResourcePolicyInput struct {
	ResourceARN string
	Policy      string
}

type Record struct {
	ApproximateArrivalTimestamp time.Time `json:"approximateArrivalTimestamp"`
	PartitionKey                string    `json:"partitionKey"`
	SequenceNumber              string    `json:"sequenceNumber"`
	Data                        []byte    `json:"data"`
}

type RegisterStreamConsumerInput struct {
	StreamARN    string
	ConsumerName string
}

type RegisterStreamConsumerOutput struct {
	Consumer Consumer
}

type Settings struct {
	JanitorInterval time.Duration `json:"janitor_interval" env:"KINESIS_JANITOR_INTERVAL" default:"1m" help:"Janitor tick interval."` //nolint:lll // Kong struct tag makes this line long
}

type Shard struct {
	ID                    string       `json:"id"`
	HashKeyRangeStart     string       `json:"hashKeyRangeStart"`
	HashKeyRangeEnd       string       `json:"hashKeyRangeEnd"`
	ParentShardID         string       `json:"parentShardId,omitempty"`
	AdjacentParentShardID string       `json:"adjacentParentShardId,omitempty"`
	Records               shardRecords `json:"records"`
	NextSeq               uint64       `json:"nextSeq"`
	Closed                bool         `json:"closed,omitempty"`
}

type ShardDescription struct {
	ShardID                  string
	HashKeyRangeStart        string
	HashKeyRangeEnd          string
	SequenceNumberRangeStart string
	SequenceNumberRangeEnd   string
	ParentShardID            string
	AdjacentParentShardID    string
	Closed                   bool
}

type ShardIterator struct {
	CreatedAt      time.Time `json:"CreatedAt"`
	StreamName     string    `json:"StreamName"`
	ShardID        string    `json:"ShardID"`
	SequenceNumber string    `json:"SequenceNumber"`
	Region         string    `json:"Region"`
	Position       int       `json:"Position"`
}

type SplitShardInput struct {
	StreamName         string
	StreamARN          string
	ShardToSplit       string
	NewStartingHashKey string
}

type StartStreamEncryptionInput struct {
	StreamName     string
	StreamARN      string
	EncryptionType string
	KeyID          string
}

type StartingPosition struct {
	Timestamp      *time.Time `json:"Timestamp,omitempty"`
	Type           string     `json:"Type"`
	SequenceNumber string     `json:"SequenceNumber,omitempty"`
}

type StopStreamEncryptionInput struct {
	StreamName     string
	StreamARN      string
	EncryptionType string
	KeyID          string
}

type StorageBackend interface {
	CreateStream(ctx context.Context, input *CreateStreamInput) error
	DeleteStream(ctx context.Context, input *DeleteStreamInput) error
	DescribeStream(ctx context.Context, input *DescribeStreamInput) (*DescribeStreamOutput, error)
	ListStreams(ctx context.Context, input *ListStreamsInput) (*ListStreamsOutput, error)
	PutRecord(ctx context.Context, input *PutRecordInput) (*PutRecordOutput, error)
	PutRecords(ctx context.Context, input *PutRecordsInput) (*PutRecordsOutput, error)
	GetShardIterator(ctx context.Context, input *GetShardIteratorInput) (*GetShardIteratorOutput, error)
	GetRecords(ctx context.Context, input *GetRecordsInput) (*GetRecordsOutput, error)
	ListShards(ctx context.Context, input *ListShardsInput) (*ListShardsOutput, error)
	RegisterStreamConsumer(
		ctx context.Context,
		input *RegisterStreamConsumerInput,
	) (*RegisterStreamConsumerOutput, error)
	DescribeStreamConsumer(
		ctx context.Context,
		input *DescribeStreamConsumerInput,
	) (*DescribeStreamConsumerOutput, error)
	ListStreamConsumers(ctx context.Context, input *ListStreamConsumersInput) (*ListStreamConsumersOutput, error)
	DeregisterStreamConsumer(ctx context.Context, input *DeregisterStreamConsumerInput) error
	SubscribeToShard(ctx context.Context, input *SubscribeToShardInput) (*SubscribeToShardOutput, error)
	UpdateShardCount(ctx context.Context, input *UpdateShardCountInput) (*UpdateShardCountOutput, error)
	EnableEnhancedMonitoring(
		ctx context.Context,
		input *EnableEnhancedMonitoringInput,
	) (*EnableEnhancedMonitoringOutput, error)
	DisableEnhancedMonitoring(
		ctx context.Context,
		input *DisableEnhancedMonitoringInput,
	) (*DisableEnhancedMonitoringOutput, error)
	IncreaseStreamRetentionPeriod(ctx context.Context, input *IncreaseStreamRetentionPeriodInput) error
	DecreaseStreamRetentionPeriod(ctx context.Context, input *DecreaseStreamRetentionPeriodInput) error
	MergeShards(ctx context.Context, input *MergeShardsInput) error
	SplitShard(ctx context.Context, input *SplitShardInput) error
	StartStreamEncryption(ctx context.Context, input *StartStreamEncryptionInput) error
	StopStreamEncryption(ctx context.Context, input *StopStreamEncryptionInput) error
	DeleteResourcePolicy(ctx context.Context, input *DeleteResourcePolicyInput) error
	GetResourcePolicy(ctx context.Context, input *GetResourcePolicyInput) (*GetResourcePolicyOutput, error)
	PutResourcePolicy(ctx context.Context, input *PutResourcePolicyInput) error
	ListTagsForResource(ctx context.Context, input *ListTagsForResourceInput) (*ListTagsForResourceOutput, error)
	TagResource(ctx context.Context, input *TagResourceInput) error
	UntagResource(ctx context.Context, input *UntagResourceInput) error
	UpdateStreamMode(ctx context.Context, input *UpdateStreamModeInput) error
	UpdateAccountSettings(ctx context.Context, input *UpdateAccountSettingsInput) error
	UpdateMaxRecordSize(ctx context.Context, input *UpdateMaxRecordSizeInput) error
	UpdateStreamWarmThroughput(ctx context.Context, input *UpdateStreamWarmThroughputInput) error
	DescribeAccountSettings(ctx context.Context) (*DescribeAccountSettingsOutput, error)
	CountOpenShards(ctx context.Context) int
	ListAll(ctx context.Context) []StreamInfo
}

type Stream struct {
	CreatedAt time.Time `json:"createdAt"`

	Tags      *tags.Tags           `json:"tags,omitempty"`
	Consumers map[string]*Consumer `json:"consumers,omitempty"`
	Name      string               `json:"name"`
	ARN       string               `json:"arn"`
	// Region is the AWS region this stream lives in. It is the second half of
	// the composite key (see streamKey in backend.go) that keeps same-named
	// streams in different regions isolated inside the single flat
	// store.Table[Stream] — the region-nested map it replaced used the
	// region as an outer map key instead of a field on Stream itself.
	Region             string   `json:"region,omitempty"`
	Status             string   `json:"status"`
	EncryptionType     string   `json:"encryptionType,omitempty"`
	KeyID              string   `json:"keyId,omitempty"`
	StreamMode         string   `json:"streamMode,omitempty"`
	Shards             []*Shard `json:"shards"`
	EnhancedMonitoring []string `json:"enhancedMonitoring,omitempty"`
	RetentionPeriod    int      `json:"retentionPeriod"`
	// MaxRecordSizeBytes is the per-record data payload size limit for this stream.
	// Defaults to defaultMaxRecordSizeBytes (1 MiB); updatable via UpdateMaxRecordSize.
	MaxRecordSizeBytes int `json:"maxRecordSizeBytes,omitempty"`
	// contains filtered or unexported fields
}

type StreamInfo struct {
	Name       string
	ARN        string
	Status     string
	ShardCount int
}

type StreamModeDetails struct {
	StreamMode string
}

type SubscribeToShardEvent struct {
	ContinuationSequenceNumber string
	Records                    []GetRecordResult
	MillisBehindLatest         int64
}

type SubscribeToShardInput struct {
	ConsumerARN      string
	ShardID          string
	StartingPosition StartingPosition
}

type SubscribeToShardOutput struct {
	Event SubscribeToShardEvent
}

type TagResourceInput struct {
	Tags        map[string]string
	ResourceARN string
}

type UntagResourceInput struct {
	ResourceARN string
	TagKeys     []string
}

type UpdateAccountSettingsInput struct {
	// OnDemandStreamCountLimit sets the account-level limit for ON_DEMAND streams.
	OnDemandStreamCountLimit int
}

type UpdateMaxRecordSizeInput struct {
	StreamName         string
	StreamARN          string
	MaxRecordSizeBytes int
}

type UpdateShardCountInput struct {
	StreamName       string
	ScalingType      string
	TargetShardCount int
}

type UpdateShardCountOutput struct {
	StreamName        string
	CurrentShardCount int
	TargetShardCount  int
}

type UpdateStreamModeInput struct {
	StreamARN         string
	StreamModeDetails StreamModeDetails
}

type UpdateStreamWarmThroughputInput struct {
	StreamName         string
	StreamARN          string
	WriteCapacityUnits int64
	ReadCapacityUnits  int64
}