/**

 * Relational pipes

 * Copyright © 2020 František Kučera (Frantovo.cz, GlobalCode.info)

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, version 3 of the License.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program. If not, see <http://www.gnu.org/licenses/>.

 */

#pragma once

#include <memory>

#include <string>

#include <vector>

#include <iostream>

#include <sstream>

#include <locale>

#include <codecvt>

#include <relpipe/common/type/typedefs.h>

#include <relpipe/reader/TypeId.h>

#include <relpipe/reader/handlers/RelationalReaderStringHandler.h>

#include <relpipe/reader/handlers/AttributeMetadata.h>

#include "Configuration.h"

#include "INIWriter.h"

namespace relpipe {

namespace out {

namespace ini {

class INILiteralHandler : public relpipe::reader::handlers::RelationalReaderStringHandler {

private:

	INIWriter& writer;

	bool hasSections = false;

	relpipe::common::type::StringX currentSection;

	std::vector<relpipe::reader::handlers::AttributeMetadata> currentAttributes;

	relpipe::common::type::Integer relationCount = 0;

	relpipe::common::type::Integer currentAttributeIndex = 0;

public:

	INILiteralHandler(INIWriter& writer) : writer(writer) {

	}

	virtual ~INILiteralHandler() {

	}

	void startRelation(relpipe::common::type::StringX name, std::vector<relpipe::reader::handlers::AttributeMetadata> attributes) override {

		if (relationCount) writer.endSection();

		currentSection = name;

		currentAttributes = attributes;

		INIWriter::SectionStartEvent e;

		e.name = name;

		writer.startSection(e);

		hasSections = true;

		relationCount++;

	}

	void attribute(const relpipe::common::type::StringX& value) override {

		if (currentAttributeIndex && currentAttributeIndex % currentAttributes.size() == 0) {

			currentAttributeIndex = 0;

			writer.endSection();

			// Usually there should be only one record in each relation.

			// But if there are more of them, we will rather create multiple sections with same name (makes more sense)

			// than duplicate keys in a single section.

			INIWriter::SectionStartEvent e;

			e.name = currentSection;

			writer.startSection(e);

			hasSections = true;

		}

		INIWriter::EntryEvent e;

		e.key = currentAttributes[currentAttributeIndex].getAttributeName();

		e.value = value;

		writer.entry(e);

		currentAttributeIndex++;

	}

	void endOfPipe() {

		if (hasSections) writer.endSection();

	}

};

}

}

}