#include "mxstring.h"

#include "decomp.h"

#include <stdlib.h>
#include <string.h>

DECOMP_SIZE_ASSERT(MxString, 0x10)

// FUNCTION: LEGO1 0x100ae200
// FUNCTION: BETA10 0x1012c110
MxString::MxString()
{
	// Set string to one char in length and set that char to null terminator
	this->m_data = new char[1];
	this->m_data[0] = 0;
	this->m_length = 0;
}

// FUNCTION: LEGO1 0x100ae2a0
// FUNCTION: BETA10 0x1012c1a1
MxString::MxString(const MxString& p_str)
{
	this->m_length = p_str.m_length;
	this->m_data = new char[this->m_length + 1];
	strcpy(this->m_data, p_str.m_data);
}

// FUNCTION: LEGO1 0x100ae350
// FUNCTION: BETA10 0x1012c24f
MxString::MxString(const char* p_str)
{
	if (p_str) {
		this->m_length = strlen(p_str);
		this->m_data = new char[this->m_length + 1];
		strcpy(this->m_data, p_str);
	}
	else {
		this->m_data = new char[1];
		this->m_data[0] = 0;
		this->m_length = 0;
	}
}

// FUNCTION: BETA10 0x1012c330
MxString::MxString(const char* p_str, MxU16 p_maxlen)
{
	if (p_str) {
		if (strlen(p_str) <= p_maxlen) {
			this->m_length = strlen(p_str);
		}
		else {
			this->m_length = p_maxlen;
		}

		// Basically strncpy
		this->m_data = new char[this->m_length + 1];
		memcpy(this->m_data, p_str, this->m_length);
		this->m_data[this->m_length] = '\0';
	}
	else {
		this->m_data = new char[1];
		this->m_data[0] = 0;
		this->m_length = 0;
	}
}

// FUNCTION: LEGO1 0x100ae420
// FUNCTION: BETA10 0x1012c45b
MxString::~MxString()
{
	delete[] this->m_data;
}

// FUNCTION: BETA10 0x1012c4de
void MxString::Reverse()
{
	char* start = this->m_data;
	char* end = this->m_data + this->m_length - 1;

	while (start < end) {
		CharSwap(start, end);
		start++;
		end--;
	}
}

// FUNCTION: LEGO1 0x100ae490
// FUNCTION: BETA10 0x1012c537
void MxString::ToUpperCase()
{
	strupr(this->m_data);
}

// FUNCTION: LEGO1 0x100ae4a0
// FUNCTION: BETA10 0x1012c55c
void MxString::ToLowerCase()
{
	strlwr(this->m_data);
}

// FUNCTION: LEGO1 0x100ae4b0
// FUNCTION: BETA10 0x1012c581
MxString& MxString::operator=(const MxString& p_str)
{
	if (this->m_data != p_str.m_data) {
		delete[] this->m_data;
		this->m_length = p_str.m_length;
		this->m_data = new char[this->m_length + 1];
		strcpy(this->m_data, p_str.m_data);
	}

	return *this;
}

// FUNCTION: LEGO1 0x100ae510
// FUNCTION: BETA10 0x1012c606
const MxString& MxString::operator=(const char* p_str)
{
	if (this->m_data != p_str) {
		delete[] this->m_data;
		this->m_length = strlen(p_str);
		this->m_data = new char[this->m_length + 1];
		strcpy(this->m_data, p_str);
	}

	return *this;
}

// FUNCTION: BETA10 0x1012c68a
MxString MxString::operator+(const MxString& p_str) const
{
	MxString tmp;
	delete[] tmp.m_data;

	tmp.m_length = p_str.m_length + this->m_length;
	tmp.m_data = new char[tmp.m_length + 1];

	strcpy(tmp.m_data, this->m_data);
	strcpy(tmp.m_data + this->m_length, p_str.m_data);

	return MxString(tmp);
}

// Return type is intentionally just MxString, not MxString&.
// This forces MSVC to add $ReturnUdt$ to the stack for 100% match.
// FUNCTION: LEGO1 0x100ae580
// FUNCTION: BETA10 0x1012c78d
MxString MxString::operator+(const char* p_str) const
{
	// MxString constructor allocates 1 byte for m_data, so free that first
	MxString tmp;
	delete[] tmp.m_data;

	tmp.m_length = strlen(p_str) + this->m_length;
	tmp.m_data = new char[tmp.m_length + 1];

	strcpy(tmp.m_data, this->m_data);
	strcpy(tmp.m_data + this->m_length, p_str);

	return MxString(tmp);
}

// FUNCTION: LEGO1 0x100ae690
// FUNCTION: BETA10 0x1012c92f
MxString& MxString::operator+=(const char* p_str)
{
	int newlen = this->m_length + strlen(p_str);

	char* tmp = new char[newlen + 1];
	strcpy(tmp, this->m_data);
	strcpy(tmp + this->m_length, p_str);

	delete[] this->m_data;
	this->m_data = tmp;
	this->m_length = newlen;

	return *this;
}

// FUNCTION: BETA10 0x1012ca10
void MxString::CharSwap(char* p_a, char* p_b)
{
	char t = *p_a;
	*p_a = *p_b;
	*p_b = t;
}