/*
  Licensed to the Apache Software Foundation (ASF) under one
  or more contributor license agreements.  See the NOTICE file
  distributed with this work for additional information
  regarding copyright ownership.  The ASF licenses this file
  to you 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.
*/

#include "dse.h"

#include "dse_date_range.hpp"
#include "dse_line_string.hpp"
#include "dse_point.hpp"
#include "dse_polygon.hpp"

#include "statement.hpp"

using namespace datastax::internal::enterprise;

extern "C" {

CassError cass_statement_bind_dse_point(CassStatement* statement, size_t index, cass_double_t x,
                                        cass_double_t y) {
  Bytes bytes = encode_point(x, y);
  return cass_statement_bind_custom(statement, index, DSE_POINT_TYPE, bytes.data(), bytes.size());
}

CassError cass_statement_bind_dse_point_by_name(CassStatement* statement, const char* name,
                                                cass_double_t x, cass_double_t y) {
  return cass_statement_bind_dse_point_by_name_n(statement, name, SAFE_STRLEN(name), x, y);
}

CassError cass_statement_bind_dse_point_by_name_n(CassStatement* statement, const char* name,
                                                  size_t name_length, cass_double_t x,
                                                  cass_double_t y) {
  Bytes bytes = encode_point(x, y);
  return cass_statement_bind_custom_by_name_n(statement, name, name_length, DSE_POINT_TYPE,
                                              sizeof(DSE_POINT_TYPE) - 1, bytes.data(),
                                              bytes.size());
}

CassError cass_statement_bind_dse_line_string(CassStatement* statement, size_t index,
                                              const DseLineString* line_string) {
  return cass_statement_bind_custom(statement, index, DSE_LINE_STRING_TYPE,
                                    line_string->bytes().data(), line_string->bytes().size());
}

CassError cass_statement_bind_dse_line_string_by_name(CassStatement* statement, const char* name,
                                                      const DseLineString* line_string) {
  return cass_statement_bind_custom_by_name(statement, name, DSE_LINE_STRING_TYPE,
                                            line_string->bytes().data(),
                                            line_string->bytes().size());
}

CassError cass_statement_bind_dse_line_string_by_name_n(CassStatement* statement, const char* name,
                                                        size_t name_length,
                                                        const DseLineString* line_string) {
  return cass_statement_bind_custom_by_name_n(
      statement, name, name_length, DSE_LINE_STRING_TYPE, sizeof(DSE_LINE_STRING_TYPE) - 1,
      line_string->bytes().data(), line_string->bytes().size());
}

CassError cass_statement_bind_dse_polygon(CassStatement* statement, size_t index,
                                          const DsePolygon* polygon) {
  return cass_statement_bind_custom(statement, index, DSE_POLYGON_TYPE, polygon->bytes().data(),
                                    polygon->bytes().size());
}

CassError cass_statement_bind_dse_polygon_by_name(CassStatement* statement, const char* name,
                                                  const DsePolygon* polygon) {
  return cass_statement_bind_custom_by_name(statement, name, DSE_POLYGON_TYPE,
                                            polygon->bytes().data(), polygon->bytes().size());
}

CassError cass_statement_bind_dse_polygon_by_name_n(CassStatement* statement, const char* name,
                                                    size_t name_length, const DsePolygon* polygon) {
  return cass_statement_bind_custom_by_name_n(statement, name, name_length, DSE_POLYGON_TYPE,
                                              sizeof(DSE_POLYGON_TYPE) - 1, polygon->bytes().data(),
                                              polygon->bytes().size());
}

CassError cass_statement_bind_dse_date_range(CassStatement* statement, size_t index,
                                             const DseDateRange* range) {
  Bytes bytes = encode_date_range(range);
  return cass_statement_bind_custom(statement, index, DSE_DATE_RANGE_TYPE, bytes.data(),
                                    bytes.size());
}

CassError cass_statement_bind_dse_date_range_by_name(CassStatement* statement, const char* name,
                                                     const DseDateRange* range) {
  return cass_statement_bind_dse_date_range_by_name_n(statement, name, SAFE_STRLEN(name), range);
}

CassError cass_statement_bind_dse_date_range_by_name_n(CassStatement* statement, const char* name,
                                                       size_t name_length,
                                                       const DseDateRange* range) {
  Bytes bytes = encode_date_range(range);
  return cass_statement_bind_custom_by_name_n(statement, name, name_length, DSE_DATE_RANGE_TYPE,
                                              sizeof(DSE_DATE_RANGE_TYPE) - 1, bytes.data(),
                                              bytes.size());
}

CassError cass_statement_set_execute_as_n(CassStatement* statement, const char* name,
                                          size_t name_length) {
  statement->set_custom_payload("ProxyExecute", reinterpret_cast<const uint8_t*>(name),
                                name_length);
  return CASS_OK;
}

CassError cass_statement_set_execute_as(CassStatement* statement, const char* name) {
  return cass_statement_set_execute_as_n(statement, name, SAFE_STRLEN(name));
}

} // extern "C"