Initial Commit

common/MACRO.CPP

@@ -0,0 +1,155 @@
+#include <common/macro.hpp>
+#include <common/systime.hpp>
+#include <common/diskinfo.hpp>
+#include <common/regkey.hpp>
+
+WORD SymbolTable::isInSymbols(char symbol)
+{
+	for(int i=0;i<mCurrentSymbols;i++)if(symbol==*((LPSTR)mSymbolTable+i))return TRUE;
+	return FALSE;
+}
+
+Macro::Macro(void)
+: mUnsetString("UNSET")
+{
+}
+
+Macro::~Macro()
+{
+}
+
+void Macro::processEmbeddedMacro(String &someString)
+{
+	mInputString=someString;
+	mlpPtrString=mInputString;
+
+	while(mlpPtrString&&0!=*mlpPtrString)
+	{
+		switch(*mlpPtrString)
+		{
+			case '$' :
+				expandEmbeddedMacro();
+				break;
+			default :
+				++mlpPtrString;
+		}
+	}
+	someString=mInputString;
+}
+
+void Macro::expandEmbeddedMacro(void)
+{
+	String macroString;
+	String tempString;
+	Block<String> macroStrings;
+	char *startPosition=mlpPtrString;
+	WORD offsetPosition;
+
+	pushSymbol(EnvMacroMid);
+	pushSymbol(EnvMacroEnd);
+	pushSymbol(EnvMacroSync);
+	pushSymbol(EnvMacroStart);
+	if(!expect(EnvMacroStart))return;
+	if(!expect(EnvMacroSync))return;
+	popSymbol(2);
+	extractString(macroString);
+	macroStrings.insert(&macroString);
+	popSymbol(1);
+	while(isInSymbols(*mlpPtrString))
+	{
+		pushSymbol(EnvMacroEnd);
+		if(!expect(*mlpPtrString))return;
+		extractString(macroString);
+		macroStrings.insert(&macroString);
+		popSymbol(1);
+	}
+	popSymbol(1);
+	pushSymbol(EnvMacroEnd);
+	if(!expect(*mlpPtrString))return;
+	popSymbol(1);
+	*startPosition=0;
+	tempString=mInputString;
+	macroString=locateFirstEnvironmentString(macroStrings);
+	tempString+=macroString;
+	offsetPosition=::strlen(mInputString)+::strlen(macroString);
+	tempString+=mlpPtrString;
+	mInputString=tempString;
+	mlpPtrString=((LPSTR)mInputString+offsetPosition);
+}
+
+void Macro::extractString(String &macroString)
+{
+	String tempString;
+	BYTE counter(0);
+
+	while(0!=*mlpPtrString&&!isInSymbols(*mlpPtrString))
+  {
+		*((LPSTR)tempString+counter++)=*mlpPtrString;
+		mlpPtrString++;
+	}
+	*((LPSTR)tempString+counter)=0;
+	macroString=tempString;
+}
+
+WORD Macro::expect(char symbol)
+{
+	if(*mlpPtrString==symbol)
+	{
+		mlpPtrString++;
+		return TRUE;
+	}
+	while(0!=*mlpPtrString&&!isInSymbols(*mlpPtrString))mlpPtrString++;
+	return FALSE;
+}
+
+String Macro::locateFirstEnvironmentString(Block<String> &varStrings)
+{
+	size_t size(varStrings.size());
+	char *envPtr;
+
+	for(int i=0;i<size;i++)
+  {
+		if(0!=(envPtr=::getenv(varStrings[i])))return String(envPtr);
+		else if(expandFunction(varStrings[i]))return varStrings[i];
+    else if(expandLiteral(varStrings[i]))return varStrings[i];
+	}
+  return mUnsetString;
+}
+
+WORD Macro::expandLiteral(String &literalString)const
+{
+	String workString;
+
+	if(literalString.isNull())return FALSE;
+	workString=literalString.betweenString('"','"');
+	if(workString.isNull())return FALSE;
+	literalString=workString;
+  return TRUE;
+}
+
+WORD Macro::expandFunction(String &atFunction)const
+{
+	WORD returnCode(FALSE);
+
+	if(atFunction==String("@TIME"))
+	{
+	  SystemTime systemTime;
+		atFunction=systemTime.toString();
+		returnCode=TRUE;
+	}
+	else if(atFunction==String("@PWD"))
+	{
+	  DiskInfo diskInfo;
+		diskInfo.getCurrentDirectory(atFunction);
+		returnCode=TRUE;
+	}
+	else if(atFunction==String("@PROGRAMFILESDIR"))
+	{
+	  RegKey regKey(RegKey::LocalMachine);
+		regKey.openKey("Software\\Microsoft\\Windows\\CurrentVersion");
+		regKey.queryValue("ProgramFilesDir",atFunction);
+		if(atFunction.isNull())atFunction="c:\\program files";
+		returnCode=TRUE;
+	}
+	return returnCode;
+}