protocol

type CommandRequest struct {
	RequestInfo

	Collection NamespacedCollection
	Command    string
	Args       bson.M
}

type DeleteRequest struct {
	RequestInfo

	Collection NamespacedCollection
	Deletes    []DeleteTarget
}

type DeleteTarget struct {
	Selector bson.M
	Limit    int
}

type ErrorCode int

type FindAndDeleteRequest struct {
	RequestInfo

	Collection NamespacedCollection

	// Query for matching the document to update
	Query bson.M

	// Optional sort order in case multiple documents match the query. Only
	// the first document will be affected by this operation.
	Sort bson.M

	// An optional selector for the fields in the returned document.
	FieldSelector bson.M
}

type FindAndUpdateRequest struct {
	RequestInfo

	Collection NamespacedCollection

	// Query for matching the document to update
	Query bson.M

	// Optional sort order in case multiple documents match the query. Only
	// the first document will be affected by this operation.
	Sort bson.M

	Update       bson.M
	ArrayFilters []bson.M

	// Create the document if missing.
	Upsert bool

	// If true, return back the updated document; otherwise return the
	// original document before applying the update.
	ReturnUpdatedDoc bool

	// An optional selector for the fields in the returned document.
	FieldSelector bson.M
}

type GetMoreRequest struct {
	RequestInfo

	Collection  NamespacedCollection
	NumToReturn int32
	CursorID    int64
}

type InsertFlag uint32

type InsertRequest struct {
	RequestInfo

	Collection NamespacedCollection
	Flags      InsertFlag
	Inserts    []bson.M
}

type KillCursorsRequest struct {
	RequestInfo

	CursorIDs []int64
}

type NamespacedCollection struct {
	Database   string
	Collection string
}

type QueryFlag uint32

type QueryRequest struct {
	RequestInfo

	Collection    NamespacedCollection
	Flags         QueryFlag
	NumToSkip     int32
	NumToReturn   int32
	Query         bson.M
	Sort          bson.M
	FieldSelector bson.M
}

type RPCHeader struct {
	MessageLength int32
	RequestID     int32
	ResponseTo    int32
	Opcode        int32
}

type ReplyType uint8

type Request interface {
	// Opcode returns the opcode identifying this request type.
	Opcode() int32

	// GetType returns a string representation of this request type.
	GetType() RequestType

	// GetReplyType returns the type of reply expected for this request.
	GetReplyType() ReplyType

	// RequestID returns the unique request ID for an incoming request.
	RequestID() int32
}

type RequestInfo struct {
	// The standard RPC header used by all request and responses.
	Header RPCHeader

	// The type of this request.
	RequestType RequestType

	// The type of expected reply for this request. Depending on the request
	// opcode, the reply:
	//   - can be omitted (e.g. OP_INSERT/UPDATE/DELETE/KILL_CURSORS)
	//   - uses the OP_REPLY format (OP_QUERY, OP_GETMORE)
	//   - uses the new OP_MSG format (for requests using OP_MSG envelopes).
	ReplyType ReplyType
}

type RequestType string

type Response struct {
	Flags        ResponseFlag
	CursorID     int64
	StartingFrom int32
	Documents    []bson.M
}

type ResponseFlag uint32

type ServerError struct {
	Msg  string
	Code ErrorCode
}

type UnknownRequest struct {
	RequestInfo

	// The raw payload of the captured request (sans header)
	Payload []byte
}

type UpdateFlag uint32

type UpdateRequest struct {
	RequestInfo

	Collection NamespacedCollection
	Updates    []UpdateTarget
}

type UpdateTarget struct {
	Selector     bson.M
	Update       bson.M
	ArrayFilters []bson.M
	Flags        UpdateFlag
}