pubsub

type Connection interface {
	ID() string

	PushData(id string, response *SubscriptionResponse)

	PushError(err error)
	PushSubscriptionError(id string, err []error)
}

type ConnectionSubscriptions map[string]*Subscription

type Consumer interface {
	Subscribe(topic string) error
	Unsubscribe()
}

type ConsumerConfig struct {
	NSQConfig   NSQConfig
	HandlerFunc CosumerHandleFunc
	Topic       string
	Channel     string
}

type Consumers interface {
	Schema(schema *graphql.Schema)

	Add(topic string, s *Subscription) error
	Remove(topic string, s *Subscription)
}

type CosumerHandleFunc func(m Message) error

type Engine interface {
	Producer() (Producer, error)
	Consumers() Consumers
	SubscriptionHandler() SubscriptionHandler

	Publish(topic string, payload interface{}) error

	Schema(schema *graphql.Schema)
}

type EventHandler struct {
	Subscribe   func(Connection, string, *SubscriptionRequest) []error
	Unsubscribe func(Connection, string)
	Close       func(Connection)
}

type Message struct {
	//nsq.Message
	Topic   string      `json:"topic,omitempty"` // topic to publish or consume, as well as subscription name
	Payload interface{} `json:"payload,omitempty"`
}

type NSQConfig struct {
	Consumer struct {
		Address string
		Config  *nsq.Config
	}
	Producer struct {
		Address string
		Config  *nsq.Config
	}
}

type Producer interface {
	Publish(topic string, payload interface{}) error
}

type PushResponseFunc func(*SubscriptionResponse)

type Subscription struct {
	ID            string
	Query         string
	Variables     map[string]interface{}
	OperationName string
	Conenction    Connection
	Push          PushResponseFunc
	Topic         string
	OperationID   string
}

type SubscriptionHandler interface {
	Subcribe(Connection, *Subscription) []error
	UnsubscribeAll(Connection)
	Unsubscribe(Connection, *Subscription)
	Subscriptions() Subscriptions

	Schema(schema *graphql.Schema)
}

type SubscriptionRequest struct {
	OperationName string                 `json:"operation_name,omitempty"`
	Query         string                 `json:"query,omitempty"`
	Variables     map[string]interface{} `json:"variables,omitempty"`
}

type SubscriptionResponse struct {
	Data   interface{} `json:"data"`
	Errors []error     `json:"errors"`
}

type Subscriptions map[Connection]ConnectionSubscriptions

type WebsocketMessage struct {
	ID      string      `json:"id,omitempty"`
	Type    string      `json:"type,omitempty"`
	Payload interface{} `json:"payload,omitempty"`
}