#include "c-icap.h"
#include "service.h"
#include "header.h"
#include "body.h"
#include "simple_api.h"
#include "debug.h"

int echo_init_service(service_extra_data_t * srv_xdata,
                      struct icap_server_conf *server_conf);
int echo_check_preview_handler(char *preview_data, int preview_data_len,
                               request_t *);
int echo_end_of_data_handler(request_t * req);
void *echo_init_request_data(service_module_t * serv, request_t * req);
int echo_io(char *rbuf, int *rlen, char *wbuf, int *wlen, int iseof,
            request_t * req);


CI_DECLARE_MOD_DATA service_module_t service = {
     "echo",                    /*Module name */
     "Echo demo service",       /*Module short description */
     ICAP_RESPMOD | ICAP_REQMOD,        /*Service type responce or request modification */
     echo_init_service,         /*init_service. */
     NULL,                      /*post_init_service */
     NULL,                      /*close_service */
     echo_init_request_data,    /*init_request_data. */
     (void (*)(void *)) ci_membuf_free, /*release request data */
     echo_check_preview_handler,
     echo_end_of_data_handler,
     echo_io,
     NULL,
     NULL
};


int echo_init_service(service_extra_data_t * srv_xdata,
                      struct icap_server_conf *server_conf)
{
     printf("Initialization of echo module......\n");
     ci_service_set_preview(srv_xdata, 1024);
     ci_service_enable_204(srv_xdata);
     ci_service_set_transfer_preview(srv_xdata, "*");
     return CI_OK;
}


void *echo_init_request_data(service_module_t * serv, request_t * req)
{

     if (ci_req_hasbody(req))
          return ci_membuf_new();
     return NULL; 
}



static int whattodo = 0;
int echo_check_preview_handler(char *preview_data, int preview_data_len,
                               request_t * req)
{
     int content_len;
     ci_membuf_t *data = ci_service_data(req);
     content_len = ci_content_lenght(req);
     ci_debug_printf(10, "We expect to read :%d body data\n", content_len);
     
     ci_req_unlock_data(req);   /*Icap server can send data before all body has received */
     if (!preview_data_len)
          return CI_MOD_CONTINUE;

     if (whattodo == 0) {
          whattodo = 1;
          ci_debug_printf(8, "Echo service will process the request\n");
          if (preview_data_len)
               ci_membuf_write(data, preview_data, preview_data_len,
                               ci_req_hasalldata(req));

          return CI_MOD_CONTINUE;
     }
     else {
          whattodo = 0;
          ci_debug_printf(8, "Allow 204...\n");
          return CI_MOD_ALLOW204;
     }
}



int echo_end_of_data_handler(request_t * req)
{
     ci_debug_printf(8, "End of data handler called\n");

     ci_membuf_t *data = ci_service_data(req);
     
     return CI_MOD_DONE;
}


int echo_io(char *rbuf, int *rlen, char *wbuf, int *wlen, int iseof,
            request_t * req)
{
     int ret;
     ci_membuf_t *data = ci_service_data(req);
     ret = CI_OK;

     if (wbuf && wlen) {
          *wlen = ci_membuf_write(data, wbuf, *wlen, iseof);
          if (*wlen < 0)
               ret = CI_ERROR;
     }
     else if (iseof)
          ci_membuf_write(data, NULL, 0, iseof);

     if (rbuf && rlen) {
          *rlen = ci_membuf_read(data, rbuf, *rlen);
     }

     return ret;
}