/* ScummVM - Graphic Adventure Engine
 *
 * ScummVM is the legal property of its developers, whose names
 * are too numerous to list here. Please refer to the COPYRIGHT
 * file distributed with this source distribution.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 */

/*
 * This file is based on WME Lite.
 * http://dead-code.org/redir.php?target=wmelite
 * Copyright (c) 2011 Jan Nedoma
 */

#include "engines/wintermute/BGame.h"
#include "engines/wintermute/scriptables/ScStack.h"
#include "engines/wintermute/scriptables/ScValue.h"
#include "engines/wintermute/utils.h"
#include "engines/wintermute/scriptables/SXString.h"
#include "engines/wintermute/scriptables/SXArray.h"
#include "engines/wintermute/StringUtil.h"

namespace WinterMute {

IMPLEMENT_PERSISTENT(CSXString, false)

//////////////////////////////////////////////////////////////////////////
CSXString::CSXString(CBGame *inGame, CScStack *Stack): CBScriptable(inGame) {
	_string = NULL;
	_capacity = 0;

	Stack->CorrectParams(1);
	CScValue *Val = Stack->Pop();

	if (Val->IsInt()) {
		_capacity = MAX(0, Val->GetInt());
		if (_capacity > 0) {
			_string = new char[_capacity];
			memset(_string, 0, _capacity);
		}
	} else {
		SetStringVal(Val->GetString());
	}

	if (_capacity == 0) SetStringVal("");
}


//////////////////////////////////////////////////////////////////////////
CSXString::~CSXString() {
	if (_string) delete [] _string;
}


//////////////////////////////////////////////////////////////////////////
void CSXString::SetStringVal(const char *Val) {
	int Len = strlen(Val);
	if (Len >= _capacity) {
		_capacity = Len + 1;
		delete[] _string;
		_string = NULL;
		_string = new char[_capacity];
		memset(_string, 0, _capacity);
	}
	strcpy(_string, Val);
}


//////////////////////////////////////////////////////////////////////////
const char *CSXString::ScToString() {
	if (_string) return _string;
	else return "[null string]";
}


//////////////////////////////////////////////////////////////////////////
void CSXString::ScSetString(const char *Val) {
	SetStringVal(Val);
}


//////////////////////////////////////////////////////////////////////////
HRESULT CSXString::ScCallMethod(CScScript *Script, CScStack *Stack, CScStack *ThisStack, const char *Name) {
	//////////////////////////////////////////////////////////////////////////
	// Substring
	//////////////////////////////////////////////////////////////////////////
	if (strcmp(Name, "Substring") == 0) {
		Stack->CorrectParams(2);
		int start = Stack->Pop()->GetInt();
		int end   = Stack->Pop()->GetInt();

		if (end < start) CBUtils::Swap(&start, &end);

		//try {
		WideString str;
		if (Game->_textEncoding == TEXT_UTF8)
			str = StringUtil::Utf8ToWide(_string);
		else
			str = StringUtil::AnsiToWide(_string);

		//WideString subStr = str.substr(start, end - start + 1);
		WideString subStr(str.c_str() + start, end - start + 1);

		if (Game->_textEncoding == TEXT_UTF8)
			Stack->PushString(StringUtil::WideToUtf8(subStr).c_str());
		else
			Stack->PushString(StringUtil::WideToAnsi(subStr).c_str());
		//  } catch (std::exception &) {
		//      Stack->PushNULL();
		//  }

		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// Substr
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "Substr") == 0) {
		Stack->CorrectParams(2);
		int start = Stack->Pop()->GetInt();

		CScValue *val = Stack->Pop();
		int len = val->GetInt();

		if (!val->IsNULL() && len <= 0) {
			Stack->PushString("");
			return S_OK;
		}

		if (val->IsNULL()) len = strlen(_string) - start;

//		try {
		WideString str;
		if (Game->_textEncoding == TEXT_UTF8)
			str = StringUtil::Utf8ToWide(_string);
		else
			str = StringUtil::AnsiToWide(_string);

//			WideString subStr = str.substr(start, len);
		WideString subStr(str.c_str() + start, len);

		if (Game->_textEncoding == TEXT_UTF8)
			Stack->PushString(StringUtil::WideToUtf8(subStr).c_str());
		else
			Stack->PushString(StringUtil::WideToAnsi(subStr).c_str());
//		} catch (std::exception &) {
//			Stack->PushNULL();
//		}

		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// ToUpperCase
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "ToUpperCase") == 0) {
		Stack->CorrectParams(0);

		WideString str;
		if (Game->_textEncoding == TEXT_UTF8)
			str = StringUtil::Utf8ToWide(_string);
		else
			str = StringUtil::AnsiToWide(_string);

		StringUtil::ToUpperCase(str);

		if (Game->_textEncoding == TEXT_UTF8)
			Stack->PushString(StringUtil::WideToUtf8(str).c_str());
		else
			Stack->PushString(StringUtil::WideToAnsi(str).c_str());

		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// ToLowerCase
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "ToLowerCase") == 0) {
		Stack->CorrectParams(0);

		WideString str;
		if (Game->_textEncoding == TEXT_UTF8)
			str = StringUtil::Utf8ToWide(_string);
		else
			str = StringUtil::AnsiToWide(_string);

		StringUtil::ToLowerCase(str);

		if (Game->_textEncoding == TEXT_UTF8)
			Stack->PushString(StringUtil::WideToUtf8(str).c_str());
		else
			Stack->PushString(StringUtil::WideToAnsi(str).c_str());

		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// IndexOf
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "IndexOf") == 0) {
		Stack->CorrectParams(2);

		const char *strToFind = Stack->Pop()->GetString();
		int index = Stack->Pop()->GetInt();

		WideString str;
		if (Game->_textEncoding == TEXT_UTF8)
			str = StringUtil::Utf8ToWide(_string);
		else
			str = StringUtil::AnsiToWide(_string);

		WideString toFind;
		if (Game->_textEncoding == TEXT_UTF8)
			toFind = StringUtil::Utf8ToWide(strToFind);
		else
			toFind = StringUtil::AnsiToWide(strToFind);

		int indexOf = StringUtil::IndexOf(str, toFind, index);
		Stack->PushInt(indexOf);

		return S_OK;
	}

	//////////////////////////////////////////////////////////////////////////
	// Split
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "Split") == 0) {
		Stack->CorrectParams(1);
		CScValue *Val = Stack->Pop();
		char Separators[MAX_PATH] = ",";
		if (!Val->IsNULL()) strcpy(Separators, Val->GetString());

		CSXArray *Array = new CSXArray(Game);
		if (!Array) {
			Stack->PushNULL();
			return S_OK;
		}


		WideString str;
		if (Game->_textEncoding == TEXT_UTF8)
			str = StringUtil::Utf8ToWide(_string);
		else
			str = StringUtil::AnsiToWide(_string);

		WideString delims;
		if (Game->_textEncoding == TEXT_UTF8)
			delims = StringUtil::Utf8ToWide(Separators);
		else
			delims = StringUtil::AnsiToWide(Separators);

		Common::Array<WideString> parts;


		size_t start, pos;
		start = 0;
		do {
			pos = StringUtil::IndexOf(Common::String(str.c_str() + start), delims, start);
			//pos = str.find_first_of(delims, start);
			if (pos == start) {
				start = pos + 1;
			} else if (pos == str.size()) {
				parts.push_back(Common::String(str.c_str() + start));
				break;
			} else {
				parts.push_back(Common::String(str.c_str() + start, pos - start));
				start = pos + 1;
			}
			//start = str.find_first_not_of(delims, start);
			start = StringUtil::LastIndexOf(Common::String(str.c_str() + start), delims, start) + 1;

		} while (pos != str.size());

		for (Common::Array<WideString>::iterator it = parts.begin(); it != parts.end(); ++it) {
			WideString &part = (*it);

			if (Game->_textEncoding == TEXT_UTF8)
				Val = new CScValue(Game, StringUtil::WideToUtf8(part).c_str());
			else
				Val = new CScValue(Game, StringUtil::WideToAnsi(part).c_str());

			Array->Push(Val);
			delete[] Val;
			Val = NULL;
		}

		Stack->PushNative(Array, false);
		return S_OK;
	}

	else return E_FAIL;
}


//////////////////////////////////////////////////////////////////////////
CScValue *CSXString::ScGetProperty(const char *Name) {
	_scValue->SetNULL();

	//////////////////////////////////////////////////////////////////////////
	// Type (RO)
	//////////////////////////////////////////////////////////////////////////
	if (strcmp(Name, "Type") == 0) {
		_scValue->SetString("string");
		return _scValue;
	}
	//////////////////////////////////////////////////////////////////////////
	// Length (RO)
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "Length") == 0) {
		if (Game->_textEncoding == TEXT_UTF8) {
			WideString wstr = StringUtil::Utf8ToWide(_string);
			_scValue->SetInt(wstr.size());
		} else
			_scValue->SetInt(strlen(_string));

		return _scValue;
	}
	//////////////////////////////////////////////////////////////////////////
	// Capacity
	//////////////////////////////////////////////////////////////////////////
	else if (strcmp(Name, "Capacity") == 0) {
		_scValue->SetInt(_capacity);
		return _scValue;
	}

	else return _scValue;
}


//////////////////////////////////////////////////////////////////////////
HRESULT CSXString::ScSetProperty(const char *Name, CScValue *Value) {
	//////////////////////////////////////////////////////////////////////////
	// Capacity
	//////////////////////////////////////////////////////////////////////////
	if (strcmp(Name, "Capacity") == 0) {
		int NewCap = Value->GetInt();
		if (NewCap < strlen(_string) + 1) Game->LOG(0, "Warning: cannot lower string capacity");
		else if (NewCap != _capacity) {
			char *NewStr = new char[NewCap];
			if (NewStr) {
				memset(NewStr, 0, NewCap);
				strcpy(NewStr, _string);
				delete[] _string;
				_string = NewStr;
				_capacity = NewCap;
			}
		}
		return S_OK;
	}

	else return E_FAIL;
}


//////////////////////////////////////////////////////////////////////////
HRESULT CSXString::Persist(CBPersistMgr *PersistMgr) {

	CBScriptable::Persist(PersistMgr);

	PersistMgr->Transfer(TMEMBER(_capacity));

	if (PersistMgr->_saving) {
		if (_capacity > 0) PersistMgr->PutBytes((byte *)_string, _capacity);
	} else {
		if (_capacity > 0) {
			_string = new char[_capacity];
			PersistMgr->GetBytes((byte *)_string, _capacity);
		} else _string = NULL;
	}

	return S_OK;
}


//////////////////////////////////////////////////////////////////////////
int CSXString::ScCompare(CBScriptable *Val) {
	return strcmp(_string, ((CSXString *)Val)->_string);
}

} // end of namespace WinterMute