#include "toonz/namebuilder.h"
#include "tconvert.h"

//-------------------------------------------------------------------

NameBuilder *NameBuilder::getBuilder(std::wstring levelName) {
  if (levelName == L"")
    return new NameCreator();
  else
    return new NameModifier(levelName);
}

//-------------------------------------------------------------------

std::wstring NameCreator::getNext() {
  if (m_s.empty()) {
    m_s.push_back(0);
    return L"A";
  }
  int i = 0;
  int n = m_s.size();
  while (i < n) {
    m_s[i]++;
    if (m_s[i] <= 'Z' - 'A') break;
    m_s[i] = 0;
    i++;
  }
  if (i >= n) {
    n++;
    m_s.push_back(0);
  }
  std::wstring s;
  for (i = n - 1; i >= 0; i--) s.append(1, (wchar_t)(L'A' + m_s[i]));

#ifdef _WIN32
  std::vector<std::wstring> invalidNames{L"AUX", L"COM", L"CON",
                                         L"LPT", L"NUL", L"PRN"};
  // If we're an invalid combination, let's check the next one
  if (std::find(invalidNames.begin(), invalidNames.end(), s) !=
      invalidNames.end())
    return getNext();
#endif

  return s;
}

//-------------------------------------------------------------------

NameModifier::NameModifier(std::wstring name) : m_nameBase(name), m_index(0) {
  int j = name.find_last_not_of(L"0123456789");
  if (j != (int)std::wstring::npos && j + 1 < (int)name.length() &&
      name[j] == '_') {
    m_index    = std::stoi(name.substr(j + 1));
    m_nameBase = name.substr(0, j);
  }
}

//-------------------------------------------------------------------

std::wstring NameModifier::getNext() {
  int index = m_index++;
  if (index < 1)
    return m_nameBase;
  else
    return m_nameBase + L"_" + std::to_wstring(index);
}