/**

 * Relational pipes

 * Copyright © 2018 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

 *

 * 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 <istream>

#include <xercesc/sax/InputSource.hpp>

#include <xercesc/util/PlatformUtils.hpp>

#include <xercesc/util/BinInputStream.hpp>

namespace relpipe {

namespace in {

namespace xml {

class StreamInputSource : public xercesc::InputSource {

private:

	std::istream& input;

	class StreamBinInputStream : public xercesc::BinInputStream {

	private:

		std::istream& input;

	public:

		StreamBinInputStream(std::istream& input) : BinInputStream(), input(input) {

		}

		XMLFilePos curPos() const override {

			return input.tellg();

		}

		const XMLCh* getContentType() const override {

			return nullptr;

		}

		XMLSize_t readBytes(XMLByte * const toFill, const XMLSize_t maxToRead) override {

			input.read((char*) toFill, maxToRead);

			return input.gcount();

		}

	};

public:

	StreamInputSource(std::istream& input, xercesc::MemoryManager * const manager = xercesc::XMLPlatformUtils::fgMemoryManager) : InputSource(manager), input(input) {

	}

	virtual ~StreamInputSource() {

	}

	xercesc::BinInputStream* makeStream() const override {

		// TODO: avoid multiple calls, input can be used only once 

		return new StreamBinInputStream(input);

	}

};

}

}

}