#pragma once

#include <string>
#include <iostream>
#include <vector>

#include "common.h"
#include "DataType.h"
#include "BooleanDataType.h"
#include "IntegerDataType.h"
#include "StringDataType.h"

using namespace std;

namespace rp_prototype {

class DataTypeCatalog {
private:
	BooleanDataType booleanType;
	IntegerDataType integerType;
	StringDataType stringType;
	vector<DataTypeBase*> types = {&booleanType, &integerType, &stringType};
public:

	integer_t toTypeId(const wstring typeCode) {
		for (DataTypeBase* dataType : types) if (dataType->supports(typeCode)) return dataType->getTypeId();
		throw RelpipeException(L"Unsupported data type: " + typeCode, EXIT_CODE_DATA_ERROR);
	}

	wstring toTypeCode(const integer_t typeId) {
		for (DataTypeBase* dataType : types) if (dataType->supports(typeId)) return dataType->getTypeCode();
		throw RelpipeException(L"Unsupported data type: " + typeId, EXIT_CODE_DATA_ERROR);
	}

	void writeString(ostream &output, const wstring &stringValue, const integer_t typeId) {
		for (DataTypeBase* dataType : types) if (dataType->supports(typeId)) return dataType->writeString(output, stringValue);
		throw RelpipeException(L"Unsupported data type: " + typeId, EXIT_CODE_DATA_ERROR);
	}

	wstring readString(istream &input, const integer_t typeId) {
		for (DataTypeBase* dataType : types) if (dataType->supports(typeId)) return dataType->readString(input);
		throw RelpipeException(L"Unsupported data type: " + typeId, EXIT_CODE_DATA_ERROR);
	}

};

}