iw4x-client/src/Utils/CSV.cpp

#include "STDInclude.hpp"

namespace Utils
{
	CSV::CSV(std::string file, bool isFile, bool allowComments)
	{
		CSV::Parse(file, isFile, allowComments);
	}

	CSV::~CSV()
	{
		for (auto row : CSV::DataMap)
		{
			for (auto entry : row)
			{
				entry.clear();
			}

			row.clear();
		}

		CSV::DataMap.clear();
	}

	int CSV::GetRows()
	{
		return CSV::DataMap.size();
	}

	int CSV::GetColumns(size_t row)
	{
		if (CSV::DataMap.size() > row)
		{
			return CSV::DataMap[row].size();
		}

		return 0;
	}

	int CSV::GetColumns()
	{
		int count = 0;

		for (int i = 0; i < CSV::GetRows(); ++i)
		{
			count = std::max(CSV::GetColumns(i), count);
		}

		return count;
	}

	std::string CSV::GetElementAt(size_t row, size_t column)
	{
		if (CSV::DataMap.size() > row)
		{
			auto _row = CSV::DataMap[row];

			if (_row.size() > column)
			{
				return _row[column];
			}
		}

		return "";
	}

	void CSV::Parse(std::string file, bool isFile, bool allowComments)
	{
		std::string buffer;

		if (isFile)
		{
			if (!Utils::FileExists(file)) return;
			buffer = Utils::ReadFile(file);
		}
		else
		{
			buffer = file;
		}
		
		if (!buffer.empty())
		{
			auto rows = Utils::Explode(buffer, '\n');

			for (auto row : rows)
			{
				CSV::ParseRow(row, allowComments);
			}
		}
	}

	void CSV::ParseRow(std::string row, bool allowComments)
	{
		bool isString = false;
		std::string element;
		std::vector<std::string> _row;
		char tempStr = 0;

		for (unsigned int i = 0; i < row.size(); ++i)
		{
			if (row[i] == ',' && !isString) // FLush entry
			{
				_row.push_back(element);
				element.clear();
				continue;
			}
			else if (row[i] == '"') // Start/Terminate string
			{
				isString = !isString;
				continue;
			}
			else if (i < (row.size() - 1) && row[i] == '\\' &&row[i + 1] == '"' && isString) // Handle quotes in strings as \"
			{
				tempStr = '"';
				++i;
			}
			else if (!isString && (row[i] == '\n' || row[i] == '\x0D' || row[i] == '\x0A' || row[i] == '\t'))
			{
				//++i;
				continue;
			}
			else if (!isString && row[i] == '#' && allowComments) // Skip comments. I know CSVs usually don't have comments, but in this case it's useful
			{
				return;
			}
			else
			{
				tempStr = row[i];
			}

			element.append(&tempStr, 1);
		}

		// Push last element
		_row.push_back(element);

		if (_row.size() == 0 || (_row.size() == 1 && !_row[0].size())) // Skip empty rows
		{
			return;
		}

		DataMap.push_back(_row);
	}
}