#include "../../Include/RmlUi/Core/Stream.h"
#include <stdio.h>
#include <string.h>

namespace Rml {

const size_t READ_BLOCK_SIZE = 1024;

Stream::Stream()
{
	stream_mode = 0;
}

Stream::~Stream()
{
	Stream::Close();
}

void Stream::Close()
{
	stream_mode = 0;
}

int Stream::GetStreamMode() const
{
	return stream_mode;
}

const URL& Stream::GetSourceURL() const
{
	return url;
}

bool Stream::IsEOS() const
{
	return Tell() >= Length();
}

size_t Stream::Peek(void* buffer, size_t bytes) const
{
	size_t pos = Tell();
	size_t read = Read(buffer, bytes);
	Seek((long)pos, SEEK_SET);
	return read;
}

size_t Stream::Read(Stream* stream, size_t bytes) const
{
	byte buffer[READ_BLOCK_SIZE];
	size_t total_bytes_read = 0;
	while (total_bytes_read < bytes)
	{
		size_t bytes_read = this->Read(buffer, Math::Min(READ_BLOCK_SIZE, bytes - total_bytes_read));
		if (bytes_read < 1)
			return total_bytes_read;
		stream->Write(buffer, bytes_read);
		total_bytes_read += bytes_read;
	}
	return total_bytes_read;
}

size_t Stream::Read(String& string, size_t bytes) const
{
	size_t string_size = string.size();
	string.resize(string_size + bytes + 1);
	size_t read = Read(&string[string_size], bytes);
	string[string_size + read] = '\0';
	string.resize(string_size + read);
	return read;
}

size_t Stream::Write(const Stream* stream, size_t bytes)
{
	return stream->Read(this, bytes);
}

size_t Stream::Write(const char* string)
{
	return Write(string, strlen(string));
}

size_t Stream::Write(const String& string)
{
	return Write(string.c_str(), string.size());
}

size_t Stream::PushFront(const void* /*buffer*/, size_t /*bytes*/)
{
	RMLUI_ERRORMSG("No generic way to PushFront to a stream.");
	return false;
}

size_t Stream::PushBack(const void* buffer, size_t bytes)
{
	size_t pos = Tell();
	Seek(0, SEEK_END);
	size_t wrote = Write(buffer, bytes);
	Seek((long)pos, SEEK_SET);
	return wrote;
}

size_t Stream::PopFront(size_t /*bytes*/)
{
	RMLUI_ERRORMSG("No generic way to PopFront from a stream.");
	return 0;
}

size_t Stream::PopBack(size_t bytes)
{
	return Truncate(Length() - bytes);
}

void Stream::SetStreamDetails(const URL& _url, int _stream_mode)
{
	url = _url;
	stream_mode = _stream_mode;
}

} // namespace Rml