#include "iterator.hpp"

#include "collection_iterator.hpp"

#include "map_iterator.hpp"

#include "result_iterator.hpp"

#include "row_iterator.hpp"

#include "types.hpp"

extern "C" {

CassIteratorType cass_iterator_type(CassIterator* iterator) {

return iterator->type();

}

CassIterator* cass_iterator_from_result(const CassResult* result) {

return CassIterator::to(new cass::ResultIterator(result));

}

CassIterator* cass_iterator_from_row(const CassRow* row) {

return CassIterator::to(new cass::RowIterator(row));

}

CassIterator* cass_iterator_from_collection(const CassValue* value) {

if (!cass_value_is_collection(value)) {

return NULL;

}

return CassIterator::to(new cass::CollectionIterator(value));

}

CassIterator* cass_iterator_from_map(const CassValue* value) {

if (value->type() != CASS_VALUE_TYPE_MAP) {

return NULL;

}

return CassIterator::to(new cass::MapIterator(value));

}

void cass_iterator_free(CassIterator* iterator) {

delete iterator->from();

}

cass_bool_t cass_iterator_next(CassIterator* iterator) {

return static_cast<cass_bool_t>(iterator->from()->next());

}

const CassRow* cass_iterator_get_row(CassIterator* iterator) {

if (iterator->type() != CASS_ITERATOR_TYPE_RESULT) {

return NULL;

}

return CassRow::to(

static_cast<cass::ResultIterator*>(

iterator->from())->row());

}

const CassValue* cass_iterator_get_column(CassIterator* iterator) {

if (iterator->type() != CASS_ITERATOR_TYPE_ROW) {

return NULL;

}

return CassValue::to(

static_cast<cass::RowIterator*>(

iterator->from())->column());

}

const CassValue* cass_iterator_get_value(CassIterator* iterator) {

if (iterator->type() != CASS_ITERATOR_TYPE_COLLECTION) {

return NULL;

}

return CassValue::to(

static_cast<cass::CollectionIterator*>(

iterator->from())->value());

}

const CassValue* cass_iterator_get_map_key(CassIterator* iterator) {

if (iterator->type() != CASS_ITERATOR_TYPE_MAP) {

return NULL;

}

return CassValue::to(

static_cast<cass::MapIterator*>(

iterator->from())->key());

}

const CassValue* cass_iterator_get_map_value(CassIterator* iterator) {

if (iterator->type() != CASS_ITERATOR_TYPE_MAP) {

return NULL;

}

return CassValue::to(

static_cast<cass::MapIterator*>(

iterator->from())->value());

}

} // extern "C"