/*
 * This file is a part of PikoTools
 * and is distributed under the 2-Clause BSD licence.
 * Author: Tomasz Sowa <t.sowa@ttmath.org>
 */

/*
 * Copyright (c) 2017-2022, Tomasz Sowa
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 */

#include "misc.h"
#include "inttostr.h"
#include "utf8/utf8.h"


namespace pt
{


void SetOverflow(bool * was_overflow, bool val)
{
	if( was_overflow )
		*was_overflow = val;
}



void esc_to_json_uformat(wchar_t val, Stream & out)
{
	char buf[10];
	size_t len;

	Toa((unsigned long)val, buf, sizeof(buf)/sizeof(char), 16, &len);

	out << "\\u";

	if( len < 4 )
	{
		for(size_t i=0 ; i < (4-len) ; ++i)
		{
			out << '0';
		}
	}

	out << buf;
}


/*
 * return true if the val character was escaped and put to the out stream
 * if the character is invalid for such a stream then only return true
 * but not put it to the stream
 */
bool try_esc_to_json(wchar_t val, Stream & out)
{
	bool status = false;

	if( val == '\r' )
	{
		out << '\\' << 'r';
		status = true;
	}
	else
	if( val == '\n' )
	{
		out << '\\' << 'n';
		status = true;
	}
	else
	if( val == '\t' )
	{
		out << '\\' << 't';
		status = true;
	}
	else
	if( val == 0x08 )
	{
		out << '\\' << 'b';
		status = true;
	}
	else
	if( val == 0x0c )
	{
		out << '\\' << 'f';
		status = true;
	}
	else
	if( val == '\\' )
	{
		out << '\\' << '\\';
		status = true;
	}
	else
	if( val == '"' )
	{
		out << '\\' << '\"';
		status = true;
	}
	else
	if( val < 32 )
	{
		esc_to_json_uformat(val, out);
		status = true;
	}

	return status;
}


void esc_to_json(wchar_t val, Stream & out)
{
	if( !try_esc_to_json(val, out) )
	{
		out << val;
	}
}


void esc_to_json(char val, Stream & out)
{
	if( !try_esc_to_json((wchar_t)(unsigned char)val, out) )
	{
		out << val;
	}
}



void esc_to_json(const char * c, pt::Stream & out)
{
	for(size_t i = 0 ; c[i] != 0 ; ++i)
	{
		esc_to_json(c[i], out);
	}
}


void esc_to_json(const char * c, std::size_t len, pt::Stream & out)
{
	for(size_t i = 0 ; i < len ; ++i)
	{
		esc_to_json(c[i], out);
	}
}


void esc_to_json(const wchar_t * c, pt::Stream & out)
{
	for(size_t i = 0 ; c[i] != 0 ; ++i)
	{
		esc_to_json(c[i], out);
	}
}


void esc_to_json(const wchar_t * c, size_t len, pt::Stream & out)
{
	for(size_t i = 0 ; i < len ; ++i)
	{
		esc_to_json(c[i], out);
	}
}


void esc_to_json(const std::string & in, Stream & out)
{
	esc_to_json(in.c_str(), in.size(), out);
}


void esc_to_json(const std::wstring & in, Stream & out)
{
	esc_to_json(in.c_str(), in.size(), out);
}



/*
 * return true if the val character was escaped and put to the out stream
 * if the character is invalid for such a stream then only return true
 * but not put it to the stream
 */
bool try_esc_to_xml(wchar_t val, Stream & out)
{
	bool status = false;

	if( val == 0 )
	{
		// null character is invalid in XML 1.0 and 1.1
		// https://en.wikipedia.org/wiki/Valid_characters_in_XML
		// return true but not put the char to the out stream
		status = true;
	}
	else
	if( val == '<')
	{
		out << "&lt;";
		status = true;
	}
	else
	if( val == '>')
	{
		out << "&gt;";
		status = true;
	}
	else
	if( val == '&')
	{
		out << "&amp;";
		status = true;
	}
	else
	if( val == '"')
	{
		out << "&quot;";
		status = true;
	}

	return status;
}


void esc_to_xml(wchar_t val, Stream & out)
{
	if( !try_esc_to_xml(val, out) )
	{
		out << val;
	}
}


void esc_to_xml(char val, Stream & out)
{
	if( !try_esc_to_xml((wchar_t)(unsigned char)val, out) )
	{
		out << val;
	}
}


void esc_to_xml(const char * c, pt::Stream & out)
{
	for(size_t i = 0 ; c[i] != 0 ; ++i)
	{
		esc_to_xml(c[i], out);
	}
}


void esc_to_xml(const char * c, std::size_t len, pt::Stream & out)
{
	for(size_t i = 0 ; i < len ; ++i)
	{
		esc_to_xml(c[i], out);
	}
}


void esc_to_xml(const wchar_t * c, pt::Stream & out)
{
	for(size_t i = 0 ; c[i] != 0 ; ++i)
	{
		esc_to_xml(c[i], out);
	}
}


void esc_to_xml(const wchar_t * c, size_t len, pt::Stream & out)
{
	for(size_t i = 0 ; i < len ; ++i)
	{
		esc_to_xml(c[i], out);
	}
}


void esc_to_xml(const std::string & in, Stream & out)
{
	esc_to_xml(in.c_str(), in.size(), out);
}


void esc_to_xml(const std::wstring & in, Stream & out)
{
	esc_to_xml(in.c_str(), in.size(), out);
}





/*
 * return true if the val character was escaped and put to the out stream
 * if the character is invalid for such a stream then only return true
 * but not put it to the stream
 */
bool try_esc_to_csv(wchar_t val, pt::Stream & out)
{
	bool status = false;

	if( val == 0 )
	{
		// null characters are invalid in text files
		// return true but not put to the out stream
		status = true;
	}
	else
	if( val == '"' )
	{
		out << "\"\"";
		status = true;
	}

	return status;
}


void esc_to_csv(wchar_t val, pt::Stream & out)
{
	if( !try_esc_to_csv(val, out) )
	{
		out << val;
	}
}


void esc_to_csv(char val, Stream & out)
{
	if( !try_esc_to_csv((wchar_t)(unsigned char)val, out) )
	{
		out << val;
	}
}


void esc_to_csv(const char * c, pt::Stream & out)
{
	for(size_t i = 0 ; c[i] != 0 ; ++i)
	{
		esc_to_csv(c[i], out);
	}
}


void esc_to_csv(const char * c, std::size_t len, pt::Stream & out)
{
	for(size_t i = 0 ; i < len ; ++i)
	{
		esc_to_csv(c[i], out);
	}
}


void esc_to_csv(const wchar_t * c, pt::Stream & out)
{
	for(size_t i = 0 ; c[i] != 0 ; ++i)
	{
		esc_to_csv(c[i], out);
	}
}


void esc_to_csv(const wchar_t * c, size_t len, pt::Stream & out)
{
	for(size_t i = 0 ; i < len ; ++i)
	{
		esc_to_csv(c[i], out);
	}
}


void esc_to_csv(const std::string & in, Stream & out)
{
	esc_to_csv(in.c_str(), in.size(), out);
}


void esc_to_csv(const std::wstring & in, Stream & out)
{
	esc_to_csv(in.c_str(), in.size(), out);
}




}