#pragma once
#include <string>
#include <vector>
#include <iostream>
#include <sstream>
#include <locale>
#include <codecvt>
#include <regex>
#include <relpipe/reader/typedefs.h>
#include <relpipe/reader/TypeId.h>
#include <relpipe/reader/handlers/RelationalReaderStringHandler.h>
#include <relpipe/reader/handlers/AttributeMetadata.h>
namespace relpipe {
namespace out {
namespace tabular {
using namespace relpipe::reader;
class XmlHandler : public handlers::RelationalReaderStringHadler {
private:
std::wstring_convert<std::codecvt_utf8<wchar_t>> convertor; // XML output will be always in UTF-8
const char* INDENT = "\t";
std::ostream &output;
std::vector<TypeId> columnTypes;
std::vector<string_t> columnTypeCodes;
std::vector<string_t> columnNames;
std::vector<integer_t> columnWidths;
std::vector<string_t> values; // all values are saved here and processed at the end of the relation
integer_t columnCount = 0;
const string_t escapeXmlText(const string_t &value, const char* color) {
std::wstringstream result;
for (auto & ch : value) {
switch (ch) {
// FIXME: xml escaping
case L'&': result << L"&";
break;
default: result << ch;
}
}
return result.str();
}
public:
XmlHandler(std::ostream& output) : output(output) {
}
void startRelation(string_t name, std::vector<handlers::AttributeMetadata> attributes) override {
output << convertor.to_bytes(name) << ": (XML)" << endl;
columnCount = attributes.size();
columnTypes.resize(columnCount);
columnTypeCodes.resize(columnCount);
columnNames.resize(columnCount);
columnWidths.resize(columnCount, 0);
for (int i = 0; i < attributes.size(); i++) {
columnNames[i] = attributes[i].getAttributeName();
columnTypes[i] = attributes[i].getTypeId();
columnTypeCodes[i] = attributes[i].getTypeName();
}
}
void attribute(const string_t& value) override {
integer_t i = values.size() % columnCount;
values.push_back(value);
columnWidths[i] = max(columnWidths[i], value.length());
}
void endOfPipe() {
}
};
}
}
}