#include "mxstring.h"
#include "decomp.h"

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

DECOMP_SIZE_ASSERT(MxString, 0x10)

// OFFSET: LEGO1 0x100ae200
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;
}

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

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

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

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

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

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

  return *this;
}

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

  return *this;
}

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

  tmp.m_length = strlen(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, str);

  return MxString(tmp);
}

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

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

  return *this;
}
