#pragma once

#include <string>

#include <iostream>

#include <vector>

#include "../include/typedefs.h"

#include "../include/RelationalWriter.h"

#include "../include/TypeId.h"

#include "format.h"

#include "DataTypeWriterBase.h"

#include "types/BooleanDataTypeWriter.h"

#include "types/IntegerDataTypeWriter.h"

#include "types/StringDataTypeWriter.h"

namespace relpipe {

namespace writer {

class StreamRelationalWriter : public RelationalWriter {

private:

	std::ostream &output;

	types::BooleanDataTypeWriter booleanWriter;

	types::IntegerDataTypeWriter integerWriter;

	types::StringDataTypeWriter stringWriter;

	vector<DataTypeWriterBase*> writers = {&booleanWriter, &integerWriter, &stringWriter};

	/**

	 * count of columns in the current table

	 */

	integer_t columnCount;

	/**

	 * types of columns in the current table

	 */

		for (DataTypeWriterBase* writer : writers) if (writer->supports(typeId)) return writer->writeString(output, stringValue);

	}

public:

	StreamRelationalWriter(std::ostream &output) :

	output(output) {

	}

		for (DataTypeWriterBase* writer : writers) if (writer->supports(typeCode)) return writer->getTypeId();

		throw RelpipeWriterException(L"Unsupported data type: " + typeCode);

	}

		for (DataTypeWriterBase* writer : writers) if (writer->supports(typeId)) return writer->getTypeCode();

	}

	void startRelation(string_t name, std::vector<std::pair<string_t, TypeId> > attributes, boolean_t writeHeader) override {

		string_t tableName = name;

		columnCount = attributes.size();

		// Write table name and column count:

		integerWriter.writeValue(output, DATA_PART_START);

		stringWriter.writeValue(output, tableName);

		integerWriter.writeValue(output, columnCount);

		columnTypes.clear();

		columnTypes.resize(columnCount);

		// Write column names:

		for (size_t c = 0; c < columnCount; c++) {

			stringWriter.writeValue(output, columnName);

		}

		// Write column types:

		for (size_t c = 0; c < columnCount; c++) {

			columnTypes[c] = typeId;

		}

	}

	void writeRecord(std::vector<string_t> attributes) override {

		// FIXME: check vector size

		for (size_t c = 0; c < columnCount; c++) {

			// TODO: support multiple rows in a single method call?

			if (c % columnCount == 0) {

				integerWriter.writeValue(output, DATA_PART_ROW);

			}

			wstring stringValue = attributes[c];

			writeString(stringValue, typeId);

		}

	}

};

}

}