kinesis

type ChildShard struct {
	ShardID           string
	HashKeyRangeStart string
	HashKeyRangeEnd   string
	ParentShards      []string
}

type ConfigProvider interface {
	GetKinesisSettings() Settings
}

type Consumer struct {
	ConsumerCreationTimestamp time.Time         `json:"consumerCreationTimestamp"`
	Tags                      map[string]string `json:"tags,omitempty"`
	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
	// MaxRecordSizeInKiB mirrors CreateStreamInput's own field of the same
	// name (kinesis@v1.46.4 api_op_CreateStream.go:101-103), not just
	// UpdateMaxRecordSize's. Zero means "not specified" -- CreateStream keeps
	// defaultMaxRecordSizeBytes, same as an omitted request member.
	MaxRecordSizeInKiB int
	// WarmThroughputMiBps mirrors CreateStreamInput's own field of the same
	// name (kinesis@v1.46.4 api_op_CreateStream.go:119-121). Zero means "not
	// specified".
	WarmThroughputMiBps int
}

type DecreaseStreamRetentionPeriodInput struct {
	StreamName           string
	RetentionPeriodHours int
}

type DeleteResourcePolicyInput struct {
	ResourceARN string
}

type DeleteStreamInput struct {
	StreamName string
	// EnforceConsumerDeletion mirrors the real DeleteStreamInput field: unset
	// or false with registered consumers fails the call with
	// ResourceInUseException instead of deleting the stream.
	EnforceConsumerDeletion bool
}

type DeregisterStreamConsumerInput struct {
	StreamARN    string
	ConsumerARN  string
	ConsumerName string
}

type DescribeAccountSettingsOutput struct {
	MinimumThroughputBillingCommitment MinimumThroughputBillingCommitmentOutput
}

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
	// MaxRecordSizeBytes and WarmThroughputMiBps mirror the same-named Stream
	// fields (see UpdateMaxRecordSize/UpdateStreamWarmThroughput). Real
	// StreamDescriptionSummary carries both (MaxRecordSizeInKiB/WarmThroughput);
	// StreamDescription (DescribeStream) does not, so only
	// handleDescribeStreamSummary reads these.
	MaxRecordSizeBytes  int
	WarmThroughputMiBps int
}

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
	ChildShards        []ChildShard
	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 KMSKeyValidator interface {
	// ValidateKMSKey resolves keyID against the KMS backend and returns nil if
	// the key exists and is usable, or one of ErrKMSNotFound/ErrKMSDisabled/
	// ErrKMSInvalidState (kinesis sentinels) describing why it is not.
	ValidateKMSKey(ctx context.Context, keyID string) error
}

type ListShardsInput struct {
	ShardFilterTimestamp  *time.Time
	StreamName            string
	NextToken             string
	ExclusiveStartShardID string
	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 MinimumThroughputBillingCommitmentInput struct {
	// Status is required: minimumThroughputBillingCommitmentEnabled or
	// minimumThroughputBillingCommitmentDisabled.
	Status string
}

type MinimumThroughputBillingCommitmentOutput struct {
	EarliestAllowedEndAt time.Time `json:"earliestAllowedEndAt"`
	EndedAt              time.Time `json:"endedAt"`
	StartedAt            time.Time `json:"startedAt"`
	Status               string    `json:"status"`
}

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 {
	Tags         map[string]string
	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 {
	// StartedAt is when this shard became open (stream creation for the initial
	// shard set, or reshard time for shards born from SplitShard/MergeShards/
	// UpdateShardCount/UpdateStreamMode). Used by ListShards' AT_TIMESTAMP/
	// FROM_TIMESTAMP/AT_TRIM_HORIZON ShardFilter to bound shard lineage by time.
	StartedAt time.Time `json:"startedAt"`
	// ClosedAt is when this shard was closed (zero if still open). Populated
	// alongside Closed by closeShard. omitempty has no effect on a struct
	// field like time.Time, so it is intentionally omitted here.
	ClosedAt              time.Time    `json:"closedAt"`
	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) (*UpdateAccountSettingsOutput, error)
	UpdateMaxRecordSize(ctx context.Context, input *UpdateMaxRecordSizeInput) error
	UpdateStreamWarmThroughput(
		ctx context.Context,
		input *UpdateStreamWarmThroughputInput,
	) (*UpdateStreamWarmThroughputOutput, error)
	DescribeAccountSettings(ctx context.Context) (*DescribeAccountSettingsOutput, error)
	CountOpenShards(ctx context.Context) int
	CountOnDemandStreams(ctx context.Context) int
	OnDemandStreamCountLimit(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
	// (wire unit is MaxRecordSizeInKiB; converted to bytes on write via bytesPerKiB).
	MaxRecordSizeBytes int `json:"maxRecordSizeBytes,omitempty"`
	// WarmThroughputMiBps is the stream's current UpdateStreamWarmThroughput
	// setting. Applied synchronously (this backend has no UPDATING transient
	// state), so Current and Target always match on read -- see
	// UpdateStreamWarmThroughputOutput and PARITY.md.
	WarmThroughputMiBps int `json:"warmThroughputMiBps,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 TaggedEntry struct {
	Tags map[string]string
	ARN  string
}

type UntagResourceInput struct {
	ResourceARN string
	TagKeys     []string
}

type UpdateAccountSettingsInput struct {
	// MinimumThroughputBillingCommitment is required.
	MinimumThroughputBillingCommitment *MinimumThroughputBillingCommitmentInput
}

type UpdateAccountSettingsOutput struct {
	MinimumThroughputBillingCommitment MinimumThroughputBillingCommitmentOutput
}

type UpdateMaxRecordSizeInput struct {
	StreamARN string
	// MaxRecordSizeInKiB is required; wire unit is KiB, not bytes.
	MaxRecordSizeInKiB 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
	// WarmThroughputMiBps mirrors UpdateStreamModeInput's own field
	// (kinesis@v1.46.4 api_op_UpdateStreamMode.go, "only valid when the
	// stream mode is being updated to on-demand"). Zero means "not
	// specified".
	WarmThroughputMiBps int
}

type UpdateStreamWarmThroughputInput struct {
	StreamName string
	StreamARN  string
	// WarmThroughputMiBps is required (api_op_UpdateStreamWarmThroughput.go:63-70).
	WarmThroughputMiBps int
}

type UpdateStreamWarmThroughputOutput struct {
	StreamARN      string
	StreamName     string
	WarmThroughput WarmThroughputObject
}

type WarmThroughputObject struct {
	CurrentMiBps int
	TargetMiBps  int
}