/*
 * Copyright 2015 MongoDB, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef REQUEST_H
#define REQUEST_H

#include <bson.h>
#include <mongoc-buffer-private.h>

#include "mongoc.h"

#include "mongoc-rpc-private.h"

struct _mock_server_t;  /* forward declaration */

typedef struct _request_t
{
   uint8_t *data;
   size_t data_len;
   mongoc_rpc_t request_rpc;
   mongoc_opcode_t opcode;  /* copied from rpc for convenience */
   struct _mock_server_t *server;
   mongoc_stream_t *client;
   uint16_t client_port;
   bool is_command;
   char *command_name;
   char *as_str;
   mongoc_array_t docs;  /* array of bson_t pointers */
} request_t;


request_t *request_new (const mongoc_buffer_t *buffer,
                        int32_t msg_len,
                        struct _mock_server_t *server,
                        mongoc_stream_t *client,
                        uint16_t client_port);

const bson_t * request_get_doc (const request_t *request,
                                int n);

bool request_matches_flags (const request_t *request,
                            mongoc_query_flags_t flags);

bool request_matches_query (const request_t *request,
                            const char *ns,
                            mongoc_query_flags_t flags,
                            uint32_t skip,
                            uint32_t n_return,
                            const char *query_json,
                            const char *fields_json,
                            bool is_command);

bool request_matches_insert (const request_t *request,
                             const char *ns,
                             mongoc_insert_flags_t flags,
                             const char *doc_json);

bool request_matches_bulk_insert (const request_t *request,
                                  const char *ns,
                                  mongoc_insert_flags_t flags,
                                  int n);

bool request_matches_update (const request_t *request,
                             const char *ns,
                             mongoc_update_flags_t flags,
                             const char *selector_json,
                             const char *update_json);

bool request_matches_delete (const request_t *request,
                             const char *ns,
                             mongoc_remove_flags_t flags,
                             const char *selector_json);

bool request_matches_getmore (const request_t *request,
                              const char *ns,
                              uint32_t n_return,
                              int64_t cursor_id);

bool request_matches_kill_cursors (const request_t *request,
                                   int64_t cursor_id);

uint16_t request_get_server_port (request_t *request);

uint16_t request_get_client_port (request_t *request);

void request_destroy (request_t *request);

#endif //REQUEST_H
