manager

type Connection struct {
	Id                      string      `dynamodbav:"id"`
	Domain                  string      `dynamodbav:"domain"`
	Stage                   string      `dynamodbav:"stage"`
	Context                 string      `dynamodbav:"context"`
	IsInitialized           bool        `dynamodbav:"isInitialized"`
	CreatedAt               *time.Time  `dynamodbav:"createdAt"`
	WebsocketConnectContext interface{} `dynamodbav:"websocketConnectContext"`
	ConnectContext          interface{} `dynamodbav:"connectContext"`
	Ttl                     int64       `dynamodbav:"ttl"`
}

type ConnectionManager interface {
	OnConnect(newConnection *Connection) error
	OnDisconnect(connectionID string) error
	Get(connectionID string) (*Connection, error)
	Init(connectionID string, connectContext interface{}) error
	Terminate(connectionID string) error
	Hydrate(connectionId string) error
}

type ContextKey int

type Event struct {
	ID      string      `dynamodbav:"id"`
	Key     string      `dynamodbav:"key"`
	Payload interface{} `dynamodbav:"payload"`
}

type EventManager interface {
	Add(newEvent *Event) error
	Remove(eventId string) error
}

type Local struct {
	Enable    bool
	OnPublish *func(key string, payload map[string]interface{})
}

type Manager struct {
	Schema             *graphql.Schema
	Connection         ConnectionManager
	Subscription       SubscriptionManager
	Event              EventManager
	OnWebsocketConnect *func(*events.APIGatewayWebsocketProxyRequest) interface{}
	OnConnect          *func(*events.APIGatewayWebsocketProxyRequest) interface{}
	OnDisconnect       *func(*Connection)
}

type Message struct {
	OperationID string `json:"id,omitempty"`
	Type        string `json:"type"`
	Payload     struct {
		Query         string                 `json:"query,omitempty"`
		Variables     map[string]interface{} `json:"variables,omitempty"`
		OperationName string                 `json:"operationName,omitempty"`
	} `json:"payload,omitempty"`
}

type SetManagerArgs struct {
	Schema             *graphql.Schema
	Connection         ConnectionManager
	Subscription       SubscriptionManager
	Event              EventManager
	OnWebsocketConnect *func(*events.APIGatewayWebsocketProxyRequest) interface{}
	OnConnect          *func(*events.APIGatewayWebsocketProxyRequest) interface{}
	OnDisconnect       *func(*Connection)
}

type Subscription struct {
	Event         string                 `dynamodbav:"event"`
	ConnectionID  string                 `dynamodbav:"connectionId"`
	OperationID   string                 `dynamodbav:"operationId"`
	Query         string                 `dynamodbav:"query"`
	Variables     map[string]interface{} `dynamodbav:"variables"`
	OperationName string                 `dynamodbav:"operationName"`
	Ttl           int64                  `dynamodbav:"ttl"`
}

type SubscriptionManager interface {
	Start(subscription *Subscription) error
	Stop(connectionID string, operationID string) error
	ListByEvents(eventKey string, from *string) (*SubscriptionResponse, error)
}

type SubscriptionResponse struct {
	Items *[]Subscription
	Next  *string
}

type WSlambdaContext struct {
	OperationID             string
	ConnectionID            string
	DomainName              string
	Stage                   string
	RequestString           string
	Variables               map[string]interface{}
	OperationName           string
	WebsocketConnectContext interface{}
	ConnectContext          interface{}
}