store

type AsyncRequest struct {
	// Primary key
	RequestID string `dynamodbav:"RequestID" json:"requestId"`

	// Connection that created this request
	ConnectionID string `dynamodbav:"ConnectionID" json:"connectionId"`

	// Status tracking
	Status    RequestStatus `dynamodbav:"Status" json:"status"`
	CreatedAt time.Time     `dynamodbav:"CreatedAt" json:"createdAt"`

	// Request details
	Action  string                 `dynamodbav:"Action" json:"action"`
	Payload map[string]interface{} `dynamodbav:"Payload,omitempty" json:"payload,omitempty"`

	// Processing information
	ProcessingStarted *time.Time `dynamodbav:"ProcessingStarted,omitempty" json:"processingStarted,omitempty"`
	ProcessingEnded   *time.Time `dynamodbav:"ProcessingEnded,omitempty" json:"processingEnded,omitempty"`

	// Result or error
	Result map[string]interface{} `dynamodbav:"Result,omitempty" json:"result,omitempty"`
	Error  string                 `dynamodbav:"Error,omitempty" json:"error,omitempty"`

	// Progress tracking
	Progress        float64                `dynamodbav:"Progress" json:"progress"`
	ProgressMessage string                 `dynamodbav:"ProgressMessage,omitempty" json:"progressMessage,omitempty"`
	ProgressDetails map[string]interface{} `dynamodbav:"ProgressDetails,omitempty" json:"progressDetails,omitempty"`

	// Retry information
	RetryCount int       `dynamodbav:"RetryCount" json:"retryCount"`
	MaxRetries int       `dynamodbav:"MaxRetries" json:"maxRetries"`
	RetryAfter time.Time `dynamodbav:"RetryAfter,omitempty" json:"retryAfter,omitempty"`

	// User and tenant for querying
	UserID   string `dynamodbav:"UserID" json:"userId"`
	TenantID string `dynamodbav:"TenantID" json:"tenantId"`

	// TTL for automatic cleanup
	TTL int64 `dynamodbav:"TTL,omitempty" json:"ttl,omitempty"`
}

type Connection struct {
	// Primary key
	ConnectionID string `dynamodbav:"ConnectionID" json:"connectionId"`

	// User and tenant information for multi-tenancy
	UserID   string `dynamodbav:"UserID" json:"userId"`
	TenantID string `dynamodbav:"TenantID" json:"tenantId"`

	// WebSocket endpoint for sending messages
	Endpoint string `dynamodbav:"Endpoint" json:"endpoint"`

	// Timestamps
	ConnectedAt time.Time `dynamodbav:"ConnectedAt" json:"connectedAt"`
	LastPing    time.Time `dynamodbav:"LastPing" json:"lastPing"`

	// Metadata for storing additional information
	Metadata map[string]string `dynamodbav:"Metadata,omitempty" json:"metadata,omitempty"`

	// TTL for automatic cleanup
	TTL int64 `dynamodbav:"TTL,omitempty" json:"ttl,omitempty"`
}

type ConnectionStore interface {
	// Save creates or updates a connection
	Save(ctx context.Context, conn *Connection) error

	// Get retrieves a connection by ID
	Get(ctx context.Context, connectionID string) (*Connection, error)

	// Delete removes a connection
	Delete(ctx context.Context, connectionID string) error

	// ListByUser returns all connections for a user
	ListByUser(ctx context.Context, userID string) ([]*Connection, error)

	// ListByTenant returns all connections for a tenant
	ListByTenant(ctx context.Context, tenantID string) ([]*Connection, error)

	// UpdateLastPing updates the last ping timestamp
	UpdateLastPing(ctx context.Context, connectionID string) error

	// DeleteStale removes connections older than the specified time
	DeleteStale(ctx context.Context, before time.Time) error
}

type RequestQueue interface {
	// Enqueue adds a new request to the queue
	Enqueue(ctx context.Context, req *AsyncRequest) error

	// Dequeue retrieves and marks requests for processing
	// This is mainly for testing - in production, DynamoDB Streams handle this
	Dequeue(ctx context.Context, limit int) ([]*AsyncRequest, error)

	// UpdateStatus updates the status of a request
	UpdateStatus(ctx context.Context, requestID string, status RequestStatus, message string) error

	// UpdateProgress updates the progress of a request
	UpdateProgress(ctx context.Context, requestID string, progress float64, message string, details map[string]interface{}) error

	// CompleteRequest marks a request as completed with results
	CompleteRequest(ctx context.Context, requestID string, result map[string]interface{}) error

	// FailRequest marks a request as failed with an error
	FailRequest(ctx context.Context, requestID string, errMsg string) error

	// GetByConnection retrieves all requests for a connection
	GetByConnection(ctx context.Context, connectionID string, limit int) ([]*AsyncRequest, error)

	// GetByStatus retrieves requests by status
	GetByStatus(ctx context.Context, status RequestStatus, limit int) ([]*AsyncRequest, error)

	// Get retrieves a specific request
	Get(ctx context.Context, requestID string) (*AsyncRequest, error)

	// Delete removes a request
	Delete(ctx context.Context, requestID string) error
}

type RequestStatus string

type StoreError struct {
	Op      string // Operation that failed
	Table   string // DynamoDB table
	Key     string // Item key
	Err     error  // Underlying error
	Message string // Additional context
}

type Subscription struct {
	// Composite key: ConnectionID#RequestID
	SubscriptionID string `dynamodbav:"SubscriptionID" json:"subscriptionId"`

	// Individual components for querying
	ConnectionID string `dynamodbav:"ConnectionID" json:"connectionId"`
	RequestID    string `dynamodbav:"RequestID" json:"requestId"`

	// Subscription details
	EventTypes []string  `dynamodbav:"EventTypes,stringset" json:"eventTypes"`
	CreatedAt  time.Time `dynamodbav:"CreatedAt" json:"createdAt"`

	// TTL for automatic cleanup
	TTL int64 `dynamodbav:"TTL,omitempty" json:"ttl,omitempty"`
}

type SubscriptionStore interface {
	// Subscribe creates a subscription for progress updates
	Subscribe(ctx context.Context, sub *Subscription) error

	// Unsubscribe removes a subscription
	Unsubscribe(ctx context.Context, connectionID, requestID string) error

	// GetByConnection returns all subscriptions for a connection
	GetByConnection(ctx context.Context, connectionID string) ([]*Subscription, error)

	// GetByRequest returns all subscriptions for a request
	GetByRequest(ctx context.Context, requestID string) ([]*Subscription, error)

	// DeleteByConnection removes all subscriptions for a connection
	DeleteByConnection(ctx context.Context, connectionID string) error
}

type TableDefinition struct {
	TableName              string
	AttributeDefinitions   []types.AttributeDefinition
	KeySchema              []types.KeySchemaElement
	GlobalSecondaryIndexes []types.GlobalSecondaryIndex
	BillingMode            types.BillingMode
}

type ValidationError struct {
	Field   string
	Message string
}