Jupiter/Jupiter/HTTP_QueryString.h

/**
 * Copyright (C) 2016 Jessica James.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
 * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
 * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 *
 * Written by Jessica James <jessica.aj@outlook.com>
 */

#if !defined _HTTP_QUERYSTRING_H_HEADER
#define _HTTP_QUERYSTRING_H_HEADER

#include "String.h"
#include "INIFile.h"

/**
 * @file HTTP_QueryString.h
 * @brief Provides parsing for HTTP Query Strings.
 */

namespace Jupiter
{
	namespace HTTP
	{
		/**
		* @brief Provides parsing for HTTP Query Strings.
		*/
		class QueryString : public Jupiter::StringS
		{
		public:
			QueryString() = delete;
			inline QueryString(const Jupiter::ReadableString &query_string) : QueryString(query_string.ptr(), query_string.size()) {}
			inline QueryString(const char *ptr, size_t str_size);
		};

		/**
		* @brief Provides parsing for HTML form responses.
		* Note: This is essentially the same thing as QueryString, except key/value pairs are pared into 'table'.
		*/
		class HTMLFormResponse : public Jupiter::StringS
		{
		public:
			HTMLFormResponse() = delete;
			inline HTMLFormResponse(const Jupiter::ReadableString &query_string) : HTMLFormResponse(query_string.ptr(), query_string.size()) {}
			inline HTMLFormResponse(const char *ptr, size_t str_size);
			Jupiter::INIFile::Section table;
		};
	}
}

/** Implementation */
inline Jupiter::HTTP::QueryString::QueryString(const char *ptr, size_t str_size) : Jupiter::StringS(str_size)
{
	if (str_size < 3) // not enough room for "%XX", therefore no escaped characters to parse
	{
		Jupiter::StringType::length = str_size;
		switch (str_size)
		{
		case 2:
			str[1] = ptr[1];
		case 1:
			*str = *ptr;
		case 0:
			// nothing to copy
			return;
		}
	}

	const char *end = ptr + str_size - 2;
	char *buf = str;
	int val;

	while (ptr != end)
	{
		if (*ptr == '%')
		{
			if ((val = Jupiter_getHex(*++ptr)) != -1)
			{
				*buf = static_cast<uint8_t>(val) << 4;
				if ((val = Jupiter_getHex(*++ptr)) != -1)
					*buf |= val;

				++buf, ++ptr;
				if (ptr > end)
				{
					if (ptr == end + 1) // copy 1 remaining character
					{
						*buf = *ptr;
						++buf;
					}
					Jupiter::StringType::length = buf - Jupiter::StringType::str;
					return;
				}
			}
		}
		else // Copy character
		{
			*buf = *ptr;
			++buf, ++ptr;
		}
	}

	// copy last 2 characters
	*buf = *ptr;
	*++buf = *++ptr;
	Jupiter::StringType::length = buf + 1 - str;
}

inline Jupiter::HTTP::HTMLFormResponse::HTMLFormResponse(const char *ptr, size_t str_size) : Jupiter::StringS(str_size)
{
	if (str_size < 3) // not enough room for "%XX", therefore no escaped characters to parse
	{
		Jupiter::StringType::length = str_size;
		switch (str_size)
		{
		case 2:
			str[1] = ptr[1];
		case 1:
			*str = *ptr;
		case 0:
			// nothing to copy
			return;
		}
	}

	const char *end = ptr + str_size - 2;
	char *buf = str;
	const char *token_start = buf;
	int val;
	Jupiter::ReferenceString key;

	while (ptr != end)
	{
		if (*ptr == '%')
		{
			if ((val = Jupiter_getHex(*++ptr)) != -1)
			{
				*buf = static_cast<uint8_t>(val) << 4;
				if ((val = Jupiter_getHex(*++ptr)) != -1)
					*buf |= val;

				++buf, ++ptr;
				if (ptr > end)
				{
					if (ptr == end + 1) // copy 1 remaining character
					{
						*buf = *ptr;
						++buf;
					}
					Jupiter::StringType::length = buf - Jupiter::StringType::str;
					return;
				}
			}
		}
		else if (*ptr == '&') // End of key/value, start of key
		{
			if (key.isNotEmpty()) // A key was already set; end of value
				Jupiter::HTTP::HTMLFormResponse::table.set(key, Jupiter::ReferenceString(token_start, buf - token_start));

			key.erase();
			++buf, ++ptr;
			token_start = buf;
		}
		else if (*ptr == '=') // End of key, start of value
		{
			key.set(token_start, buf - token_start);
			++buf, ++ptr;
			token_start = buf;
		}
		else // Copy character
		{
			*buf = *ptr;
			++buf, ++ptr;
		}
	}

	// copy last 2 characters
	*buf = *ptr;

	if (*buf == '=') // End of key, start of value
	{
		key.set(token_start, ++buf - token_start);
		*buf = *++ptr;
		Jupiter::HTTP::HTMLFormResponse::table.set(key, Jupiter::ReferenceString(ptr, 1));
	}
	else
		*++buf = *++ptr;

	if (key.isNotEmpty()) // A key was already set; end of value
		Jupiter::HTTP::HTMLFormResponse::table.set(key, Jupiter::ReferenceString(token_start, buf - token_start + 1));

	Jupiter::StringType::length = buf + 1 - str;
}

#endif // _HTTP_QUERYSTRING_H_HEADER