// Dictionary.cpp: implementation of the CDictionary class.
//
//////////////////////////////////////////////////////////////////////

#include "uox3.h"
#include "Dictionary.h"
#include "speech.h"
#include "scriptc.h"
#include "ssection.h"
#include "StringUtility.hpp"

CDictionaryContainer *Dictionary;

const SI32 dictCANTOPEN			= -1;
//const SI32 dictDUPESECTION		= -2;
//const SI32 dictDUPEOPENBRACE	= -3;
//const SI32 dictNOOPENBRACE		= -4;


//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CDictionary::CDictionary() : IsValid( false )
{
	Language			= "ZRO";
	PathToDictionary	= "./dictionary.ZRO";
}

CDictionary::~CDictionary()
{
}

CDictionary::CDictionary( const std::string& filepath, const std::string& language ) : CDictionary()
{

	if( language.empty() )
		Language = "ZRO";
	else
		Language = language;

	//	Load the path
	if( filepath.empty() )
		PathToDictionary = "dictionary.ZRO";
	else
		PathToDictionary = filepath;
}

//const UI16 DICT_MAX_TEXTBUFFERSIZE = 1024;

SI32 CDictionary::LoadDictionary( void )
{
	SI32 count	= 0;
	IsValid		= false;
	if( FileExists( PathToDictionary ) )
	{
		Script *dictData = new Script( PathToDictionary, NUM_DEFS, false );
		if( dictData != nullptr )
		{
			ScriptSection *dictSect = nullptr;
			std::string tag, data;
			for( dictSect = dictData->FirstEntry(); dictSect != nullptr; dictSect = dictData->NextEntry() )
			{
				if( dictSect != nullptr )
				{
					// verify it is a dictionary entry
					if( dictData->EntryName().substr( 0, 10 ) != "DICTIONARY" )
					{
						continue;
					}
					for( tag = dictSect->First(); !dictSect->AtEnd(); tag = dictSect->Next() )
					{
						data					= dictSect->GrabData();
						if( data != "" )
						{
							Text2[static_cast<UI32>(std::stoul(tag, nullptr, 0))] = strutil::trim( strutil::removeTrailing( data, "//" ));
							++count;
						}
						else
						{
							Console.warning( strutil::format( "Entry with tag %s in %s dictionary has no value!", tag.c_str(), Language.c_str() ));
						}
					}
				}
			}
			delete dictData;
			dictData = nullptr;
		}
		IsValid = true;
		Console.print( " " );
		Console.MoveTo( 15 );
		Console << "Dictionary." << Language;
		Console.PrintSpecial( CGREEN, "loaded" );
	}
	else
	{
		count = dictCANTOPEN;
	}

	return count;
}

void CDictionary::ShowList( void )
{
	Console << "[Testing]" << myendl;
	std::map< UI32, std::string >::const_iterator toFind = Text2.begin();
	while( toFind != Text2.end() )
	{
		Console << (SI32)(toFind->first) << ") " << toFind->second << myendl;
		++toFind;
	}
}

std::string CDictionary::operator[]( SI32 Num )
{
	return GetEntry( Num );
}

void CDictionary::SetValid( const bool newVal )
{
	IsValid = newVal;
}

bool CDictionary::GetValid( void ) const
{
	return IsValid;
}

size_t CDictionary::NumberOfEntries( void ) const
{
	return Text2.size();
}

std::string CDictionary::GetEntry( const SI32 Num )
{
	std::string rvalue;
	if( IsValid )
	{
		std::map< UI32, std::string >::const_iterator toFind = Text2.find( Num );
		if( toFind != Text2.end() )
			rvalue = toFind->second;
		else
			Console.warning( strutil::format("Dictionary Reference %i not found in \"%s\"", Num, PathToDictionary.c_str() ));
	}
	return rvalue;
}

CDictionaryContainer::CDictionaryContainer() : defaultLang( ZERO )
{
	std::string buildName;
	for( UI16 i = (UI16)DL_DEFAULT; i < (UI16)DL_COUNT; ++i )
	{
		buildName = cwmWorldState->ServerData()->Directory( CSDDP_DICTIONARIES ) + "dictionary." + DistinctLanguageNames[i];
		dictList[i] = new CDictionary( buildName, DistinctLanguageNames[i] );
	}
}

CDictionaryContainer::CDictionaryContainer( const std::string& filepath ) : defaultLang( ZERO )
{
	std::string buildName;
	for( UI16 i = (UI16)DL_DEFAULT; i < (UI16)DL_COUNT; ++i )
	{
		buildName = filepath + "dictionary." + DistinctLanguageNames[i];
		dictList[i] = new CDictionary( buildName, DistinctLanguageNames[i] );
	}
}

CDictionaryContainer::~CDictionaryContainer()
{
	for( UI16 i = (UI16)DL_DEFAULT; i < (UI16)DL_COUNT; ++i )
		delete dictList[i];
}

SI32 CDictionaryContainer::LoadDictionary( void )
{
	SI32 rvalue = 0;
	for( UI16 i = (UI16)DL_DEFAULT; i < (UI16)DL_COUNT; ++i )
	{
		dictList[i]->LoadDictionary();
	}
	if( !dictList[LanguageCodesLang[ZERO]]->GetValid() )
	{
		Console.error( "Dictionary.ZRO is bad or nonexistant" );
		Shutdown( FATAL_UOX3_BAD_DEF_DICT );
		rvalue = -1;
	}
	return rvalue;
}

std::string CDictionaryContainer::operator[]( const SI32 Num )
{
	return GetEntry( Num );
}

std::string CDictionaryContainer::GetEntry( const SI32 Num, const UnicodeTypes toDisp )
{
	std::string RetVal;
	if( cwmWorldState->ServerData()->ServerLanguage() != DL_DEFAULT ) // defaultServerLang != DL_DEFAULT )
	{
		// If a default server language has been specified in uox.ini, force the use of that language
		RetVal = dictList[cwmWorldState->ServerData()->ServerLanguage()]->GetEntry( Num );
		if( RetVal.empty() ) // If the entry wasn't found, try the ZERO before we flat out return null
			RetVal = dictList[LanguageCodesLang[defaultLang]]->GetEntry( Num );

		return RetVal;
	}

	auto typetouse = toDisp ;
	if ((static_cast<SI32>(toDisp) < 0) || (static_cast<SI32>(toDisp)>= UnicodeTypes::TOTAL_LANGUAGES)){
		typetouse = ZERO ;
	}
	const DistinctLanguage mLanguage = LanguageCodesLang[typetouse];

	if( mLanguage < DL_COUNT )
		RetVal = dictList[mLanguage]->GetEntry( Num );
	if( RetVal.empty() && typetouse != defaultLang )//if we are using a diff language, and the entry wasn't found, try the ZERO before we flat out return null
		RetVal = dictList[LanguageCodesLang[defaultLang]]->GetEntry( Num );

	return RetVal;
}

void CDictionaryContainer::SetDefaultLang( const UnicodeTypes newType )
{
	if( dictList[LanguageCodesLang[newType]]->GetValid() )
		defaultLang = newType;
}