/*
  Copyright (c) <2007-2012> <Barbara Philippot - Olivier Courtin>

  Permission is hereby granted, free of charge, to any person obtaining a copy
  of this software and associated documentation files (the "Software"), to deal
  in the Software without restriction, including without limitation the rights
  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  copies of the Software, and to permit persons to whom the Software is
  furnished to do so, subject to the following conditions:

  The above copyright notice and this permission notice shall be included in
  all copies or substantial portions of the Software.

  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  IN THE SOFTWARE.
*/


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>

#include "../ows/ows.h"


/*
 * Return expression matching the comparison operators following :
 * PropertyIsEqualTo, PropertyIsNotEqualTo, PropertyIsLessThan
 * PropertyIsGreaterThan, PropertyIsLessThanOrEqualTo, PropertyIsGreaterThanOrEqualTo
 */
static buffer *fe_binary_comparison_op(ows * o, buffer * typename, filter_encoding * fe, xmlNodePtr n)
{
  buffer *tmp, *type, *name;
  xmlChar *matchcase;
  bool bool_type = false;
  bool sensitive_case = true;

  assert(o && typename && fe && n);

  tmp = buffer_init();
  name = buffer_init();

  buffer_add_str(name, (char *) n->name);

  /* By default, comparison is case sensitive */
  matchcase = xmlGetProp(n, (xmlChar *) "matchCase");
  if (matchcase && !strcmp((char *) matchcase, "false")) sensitive_case = false;
  xmlFree(matchcase);

  n = n->children;

  while (n->type != XML_ELEMENT_NODE) n = n->next; /* eat spaces */

  /* If comparison are explicitly not case sensitive */
  if (!sensitive_case) buffer_add_str(fe->sql, "lower(");

  tmp = fe_expression(o, typename, fe, tmp, n);
  if (fe->error_code) {
    buffer_free(tmp);
    buffer_free(name);
    return fe->sql;
  }

  buffer_copy(fe->sql, tmp);

  if (buffer_cmp(name, "PropertyIsEqualTo") || buffer_cmp(name, "PropertyIsNotEqualTo")) {
    /* remove brackets (if any) and quotation marks */
    if (tmp->buf[0] == '(') {
      if (tmp->use < 3) {
        buffer_free(tmp);
        buffer_free(name);
        fe->error_code = FE_ERROR_FILTER;
        return fe->sql;
      }
      buffer_pop(tmp, 2);
      buffer_shift(tmp, 2);
    } else {
      if (tmp->use < 1) {
        buffer_free(tmp);
        buffer_free(name);
        fe->error_code = FE_ERROR_FILTER;
        return fe->sql;
      }
      buffer_pop(tmp, 1);
      buffer_shift(tmp, 1);
    }

    type = ows_psql_type(o, ows_layer_prefix_to_uri(o->layers, typename), tmp);

    if (buffer_cmp(type, "bool")) bool_type = true;
  }

  if (!sensitive_case) buffer_add_str(fe->sql, ")");

  if (buffer_cmp(name, "PropertyIsEqualTo"))
    buffer_add_str(fe->sql, " = ");

  if (buffer_cmp(name, "PropertyIsNotEqualTo"))
    buffer_add_str(fe->sql, " != ");

  if (buffer_cmp(name, "PropertyIsLessThan"))
    buffer_add_str(fe->sql, " < ");

  if (buffer_cmp(name, "PropertyIsGreaterThan"))
    buffer_add_str(fe->sql, " > ");

  if (buffer_cmp(name, "PropertyIsLessThanOrEqualTo"))
    buffer_add_str(fe->sql, " <= ");

  if (buffer_cmp(name, "PropertyIsGreaterThanOrEqualTo"))
    buffer_add_str(fe->sql, " >= ");

  buffer_empty(tmp);

  n = n->next;

  while (n->type != XML_ELEMENT_NODE) n = n->next;

  if (!sensitive_case) buffer_add_str(fe->sql, "lower(");

  tmp = fe_expression(o, typename, fe, tmp, n);
  if (fe->error_code) {
    buffer_free(tmp);
    buffer_free(name);
    return fe->sql;
  }

  /* If property is a boolean, XML content transformation */
  if (bool_type) {
    if (buffer_cmp(tmp, "'1'")) buffer_add_str(fe->sql, "'t'");
    if (buffer_cmp(tmp, "'0'")) buffer_add_str(fe->sql, "'f'");
  } else buffer_copy(fe->sql, tmp);

  if (!sensitive_case) buffer_add_str(fe->sql, ")");

  buffer_free(tmp);
  buffer_free(name);

  return fe->sql;
}


/*
 * String comparison operator with pattern matching
 * FIXME : remains a problem when escaping \* -> \%
 */
static buffer *fe_property_is_like(ows * o, buffer * typename, filter_encoding * fe, xmlNodePtr n)
{
  xmlChar *content, *wildcard, *singlechar, *escape, *matchcase;
  buffer *pg_string;
  char *escaped;
  bool sensitive_case = true;

  assert(o && typename && fe && n);

  wildcard = xmlGetProp(n, (xmlChar *) "wildCard");
  singlechar = xmlGetProp(n, (xmlChar *) "singleChar");
  matchcase = xmlGetProp(n, (xmlChar *) "matchCase");

  if (ows_version_get(o->request->version) == 100)
    escape = xmlGetProp(n, (xmlChar *) "escape");
  else
    escape = xmlGetProp(n, (xmlChar *) "escapeChar");

  /* By default, comparison is case sensitive */
  if (matchcase && !strcmp((char *) matchcase, "false")) sensitive_case = false;

  n = n->children;

  while (n->type != XML_ELEMENT_NODE) n = n->next; /* eat spaces */

  /* If comparison are explicitly not case sensitive */
  if (!sensitive_case) buffer_add_str(fe->sql, "LOWER(");

  /* We need to cast as varchar at least for timestamp PostgreSQL data type */
  buffer_add_str(fe->sql, " CAST(\"");
  fe->sql = fe_property_name(o, typename, fe, fe->sql, n, false, true);
  buffer_add_str(fe->sql, "\" AS varchar)");

  if (!sensitive_case) {
    buffer_add_str(fe->sql, ") LIKE LOWER(E");
  } else {
    buffer_add_str(fe->sql, " LIKE E");
  }

  n = n->next;

  while (n->type != XML_ELEMENT_NODE) n = n->next; /* eat spaces */

  content = xmlNodeGetContent(n->children);

  pg_string = buffer_init();
  buffer_add_str(pg_string, (char *) content);

  /* Replace the wildcard,singlechar and escapechar */
  if ((char *) wildcard && (char *) singlechar && (char *) escape) {
    if (strlen((char *) escape))     pg_string = buffer_replace(pg_string, (char *) escape,     "\\\\");
    if (strlen((char *) wildcard))   pg_string = buffer_replace(pg_string, (char *) wildcard,   "%");
    if (strlen((char *) singlechar)) pg_string = buffer_replace(pg_string, (char *) singlechar, "_");
  } else fe->error_code = FE_ERROR_FILTER;

  buffer_add_str(fe->sql, "'");
  escaped = ows_psql_escape_string(o, pg_string->buf);
  if (escaped) {
    buffer_add_str(fe->sql, escaped);
    free(escaped);
  }
  buffer_add_str(fe->sql, "'");

  if (!sensitive_case) buffer_add_str(fe->sql, ")");

  xmlFree(content);
  xmlFree(wildcard);
  xmlFree(singlechar);
  xmlFree(matchcase);
  xmlFree(escape);
  buffer_free(pg_string);

  return fe->sql;
}


/*
 * Check if the value of a property is null
 */
static buffer *fe_property_is_null(ows * o, buffer * typename, filter_encoding * fe, xmlNodePtr n)
{
  assert(o && typename && fe && n);

  n = n->children;
  while (n->type != XML_ELEMENT_NODE) n = n->next; /* eat spaces */
  buffer_add(fe->sql, '"');
  fe->sql = fe_property_name(o, typename, fe, fe->sql, n, false, true);
  buffer_add_str(fe->sql, "\" isnull");

  return fe->sql;
}


/*
 * Check if property is between two boundary values
 */
static buffer *fe_property_is_between(ows * o, buffer * typename, filter_encoding * fe, xmlNodePtr n)
{
  buffer *tmp;

  assert(o && typename && fe && n);

  tmp = buffer_init();

  n = n->children;

  while (n->type != XML_ELEMENT_NODE) n = n->next;  /* eat spaces */

  tmp = fe_expression(o, typename, fe, tmp, n);

  buffer_copy(fe->sql, tmp);
  buffer_empty(tmp);

  buffer_add_str(fe->sql, " Between ");

  n = n->next;

  while (n->type != XML_ELEMENT_NODE) n = n->next; /* eat spaces */

  tmp = fe_expression(o, typename, fe, tmp, n->children);

  buffer_copy(fe->sql, tmp);
  buffer_empty(tmp);

  buffer_add_str(fe->sql, " And ");

  n = n->next;

  while (n->type != XML_ELEMENT_NODE) n = n->next; /* eat spaces */

  tmp = fe_expression(o, typename, fe, tmp, n->children);

  buffer_copy(fe->sql, tmp);
  buffer_free(tmp);

  return fe->sql;
}


/*
 * Check if the string is a comparison operator
 */
bool fe_is_comparison_op(char *name)
{
  assert(name);

  /* Case sensitive comparison as specified in GML standard */
  if (    !strcmp(name, "PropertyIsEqualTo")
       || !strcmp(name, "PropertyIsNotEqualTo")
       || !strcmp(name, "PropertyIsLessThan")
       || !strcmp(name, "PropertyIsGreaterThan")
       || !strcmp(name, "PropertyIsLessThanOrEqualTo")
       || !strcmp(name, "PropertyIsGreaterThanOrEqualTo")
       || !strcmp(name, "PropertyIsLike")
       || !strcmp(name, "PropertyIsNull")
       || !strcmp(name, "PropertyIsBetween"))
    return true;

  return false;
}


/*
 * Execute the matching comparison function
 * CAUTION : call fe_is_comparison_op before calling this function,
 */
buffer *fe_comparison_op(ows * o, buffer * typename, filter_encoding * fe, xmlNodePtr n)
{
  assert(o && typename && fe && n);

  /* Case sensitive comparison as specified in GML standard */
  if (    !strcmp((char *) n->name, "PropertyIsEqualTo")
       || !strcmp((char *) n->name, "PropertyIsNotEqualTo")
       || !strcmp((char *) n->name, "PropertyIsLessThan")
       || !strcmp((char *) n->name, "PropertyIsGreaterThan")
       || !strcmp((char *) n->name, "PropertyIsLessThanOrEqualTo")
       || !strcmp((char *) n->name, "PropertyIsGreaterThanOrEqualTo"))
    fe->sql = fe_binary_comparison_op(o, typename, fe, n);
  else if (!strcmp((char *) n->name, "PropertyIsLike"))
    fe->sql = fe_property_is_like(o, typename, fe, n);
  else if (!strcmp((char *) n->name, "PropertyIsNull"))
    fe->sql = fe_property_is_null(o, typename, fe, n);
  else if (!strcmp((char *) n->name, "PropertyIsBetween"))
    fe->sql = fe_property_is_between(o, typename, fe, n);
  else
    fe->error_code = FE_ERROR_FILTER;

  return fe->sql;
}