#include <BAN/String.h>
#include <BAN/StringView.h>
#include <BAN/Vector.h>

#include <string.h>

namespace BAN
{

	StringView::StringView()
	{ }

	StringView::StringView(const String& other)
		: StringView(other.Data(), other.Size())
	{ }

	StringView::StringView(const char* string, size_type len)
	{
		if (len == size_type(-1))
			len = strlen(string);
		m_data = string;
		m_size = len;
	}

	char StringView::operator[](size_type index) const
	{
		ASSERT(index < m_size);
		return m_data[index];
	}

	bool StringView::operator==(const String& other) const
	{
		if (m_size != other.Size())
			return false;
		return memcmp(m_data, other.Data(), m_size) == 0;
	}

	bool StringView::operator==(StringView other) const
	{
		if (m_size != other.m_size)
			return false;
		return memcmp(m_data, other.m_data, m_size) == 0;
	}

	bool StringView::operator==(const char* other) const
	{
		if (memcmp(m_data, other, m_size))
			return false;
		return other[m_size] == '\0';
	}

	StringView StringView::Substring(size_type index, size_type len) const
	{
		ASSERT(index <= m_size);
		if (len == size_type(-1))
			len = m_size - index;
		ASSERT(len <= m_size - index); // weird order to avoid overflow
		StringView result;
		result.m_data = m_data + index;
		result.m_size = len;
		return result;
	}

	ErrorOr<Vector<StringView>> StringView::Split(char delim, bool allow_empties)
	{
		// FIXME: Won't work while multithreading
		static char s_delim = delim;
		return Split([](char c){ return c == s_delim; }, allow_empties);
	}

	ErrorOr<Vector<StringView>> StringView::Split(bool(*comp)(char), bool allow_empties)
	{
		size_type count = 0;
		{
			size_type start = 0;
			for (size_type i = 0; i < m_size; i++)
			{
				if (comp(m_data[i]))
				{
					if (allow_empties || start != i)
						count++;
					start = i + 1;
				}
			}
			if (start != m_size)
				count++;
		}

		Vector<StringView> result;
		TRY(result.Reserve(count));

		size_type start = 0;
		for (size_type i = 0; i < m_size; i++)
		{
			if (comp(m_data[i]))
			{
				if (allow_empties || start != i)
					TRY(result.PushBack(this->Substring(start, i - start)));
				start = i + 1;
			}
		}
		if (start != m_size)
			TRY(result.PushBack(this->Substring(start)));
		return result;
	}

	StringView::size_type StringView::Count(char ch) const
	{
		size_type result = 0;
		for (size_type i = 0; i < m_size; i++)
			if (m_data[i] == ch)
				result++;
		return result;
	}

	bool StringView::Empty() const
	{
		return m_size == 0;
	}

	StringView::size_type StringView::Size() const
	{
		return m_size;
	}
	
	const char* StringView::Data() const
	{
		return m_data;
	}

}