#if !defined( INCLUDED_STREAM_STRINGSTREAM_H )
#define INCLUDED_STREAM_STRINGSTREAM_H

#include "itextstream.h"
#include "string/string.h"
#include <vector>


/// \brief A wrapper around a STL vector of char.
/// Maintains a null-terminated array of char.
/// Provides a limited STL-style interface to push and pop characters at the end of the string.
class StringBuffer
{
std::vector<char> m_string;
public:
StringBuffer(){
	m_string.push_back( '\0' );
}
explicit StringBuffer( std::size_t capacity ){
	m_string.reserve( capacity );
	m_string.push_back( '\0' );
}
explicit StringBuffer( const char* string ) : m_string( string, string + string_length( string ) + 1 ){
}

typedef std::vector<char>::iterator iterator;
typedef std::vector<char>::const_iterator const_iterator;

iterator begin(){
	return m_string.begin();
}
const_iterator begin() const {
	return m_string.begin();
}
iterator end(){
	return m_string.end() - 1;
}
const_iterator end() const {
	return m_string.end() - 1;
}

void push_back( char c ){
	m_string.insert( end(), c );
}
void pop_back(){
	m_string.erase( end() - 1 );
}
void push_range( const char* first, const char* last ){
	m_string.insert( end(), first, last );
}
void push_string( const char* string ){
	push_range( string, string + string_length( string ) );
}
char* c_str(){
	return &( *m_string.begin() );
}
const char* c_str() const {
	return &( *m_string.begin() );
}

char& back(){
	return *( end() - 1 );
}
const char& back() const {
	return *( end() - 1 );
}
bool empty() const {
	return m_string.size() == 1;
}
void clear(){
	m_string.clear();
	m_string.push_back( '\0' );
}
};

/// \brief A TextOutputStream which writes to a StringBuffer.
/// Similar to std::stringstream.
class StringOutputStream : public TextOutputStream
{
StringBuffer m_string;
public:
typedef StringBuffer::iterator iterator;
typedef StringBuffer::const_iterator const_iterator;

StringOutputStream(){
}
StringOutputStream( std::size_t capacity ) : m_string( capacity ){
}
std::size_t write( const char* buffer, std::size_t length ){
	m_string.push_range( buffer, buffer + length );
	return length;
}

iterator begin(){
	return m_string.begin();
}
const_iterator begin() const {
	return m_string.begin();
}
iterator end(){
	return m_string.end();
}
const_iterator end() const {
	return m_string.end();
}

bool empty() const {
	return m_string.empty();
}
char* c_str(){
	return m_string.c_str();
}
const char* c_str() const {
	return m_string.c_str();
}
void clear(){
	m_string.clear();
}
};

template<typename T>
inline StringOutputStream& operator<<( StringOutputStream& ostream, const T& t ){
	return ostream_write( ostream, t );
}


#endif