HashMap<String, String> parseAttributes(const String& string, bool& attrsOK)
{
using namespace EA::Internal::XML;
AttributeParseState state;
state.gotAttributes = false;
IXmlReader* pXmlReader = EA::XMLWrapper::CreateXMLReaderWrapperInterface();
if(!pXmlReader)
return state.attributes;
const String parseString = "<?xml version=\"1.0\"?><attrs " + string + " />";
pXmlReader->PushInputBuffer(parseString.characters(), parseString.length() * sizeof(char16_t), EA::Internal::XML::kReadEncodingUTF16, NULL, false);
bool bSuccess = true;
while(bSuccess && pXmlReader->Read())
{
const NodeType nodeType = pXmlReader->GetNodeType();
if (nodeType == EA::Internal::XML::Element)
attributesStartElementNsHandler(state, pXmlReader);
}
EA::XMLWrapper::DestroyXMLReaderWrapperInterface(pXmlReader);
attrsOK = state.gotAttributes;
return state.attributes;
}
HRESULT ReadValue(LPCWSTR path, LPCWSTR section, LPCWSTR key, std::wstring &strxmlval, LPCWSTR defval)
{
IXmlReader *pReader = NULL;
IStream *pFileStream = NULL;
HRESULT hr;
XmlNodeType nodeType;
LPCWSTR pwszLocalName;
LPCWSTR pwszAttributeName;
LPCWSTR pwszAttributeValue;
int sequence = 0;
strxmlval = defval;
hr = CreateStreamReader(path, &pReader, &pFileStream);
EXIT_NOT_S_OK(hr);
while(pReader->Read(&nodeType) == S_OK)
{
switch(nodeType)
{
case XmlNodeType_Element:
hr = pReader->GetLocalName(&pwszLocalName, NULL);
EXIT_NOT_S_OK(hr);
switch(sequence)
{
case 0:
if(wcscmp(TagRoot, pwszLocalName) == 0)
{
sequence = 1;
}
break;
case 1:
if(wcscmp(TagSection, pwszLocalName) == 0)
{
sequence = 2;
}
break;
case 3:
if(wcscmp(TagKey, pwszLocalName) == 0)
{
sequence = 4;
}
break;
default:
break;
}
for(hr = pReader->MoveToFirstAttribute(); hr == S_OK; hr = pReader->MoveToNextAttribute())
{
hr = pReader->GetLocalName(&pwszAttributeName, NULL);
EXIT_NOT_S_OK(hr);
hr = pReader->GetValue(&pwszAttributeValue, NULL);
EXIT_NOT_S_OK(hr);
switch(sequence)
{
case 2:
if(wcscmp(AttributeName, pwszAttributeName) == 0 && wcscmp(section, pwszAttributeValue) == 0)
{
sequence = 3;
}
break;
case 4:
if(wcscmp(AttributeName, pwszAttributeName) == 0 && wcscmp(key, pwszAttributeValue) == 0)
{
sequence = 5;
}
break;
case 5:
if(wcscmp(AttributeValue, pwszAttributeName) == 0)
{
strxmlval.assign(pwszAttributeValue);
goto exit;
}
break;
default:
break;
}
}
break;
case XmlNodeType_EndElement:
hr = pReader->GetLocalName(&pwszLocalName, NULL);
EXIT_NOT_S_OK(hr);
switch(sequence)
{
case 1:
if(wcscmp(TagRoot, pwszLocalName) == 0)
{
goto exit;
}
break;
case 2:
if(wcscmp(TagSection, pwszLocalName) == 0)
{
sequence = 1;
}
break;
case 3:
case 4:
case 5:
if(wcscmp(TagSection, pwszLocalName) == 0)
{
goto exit;
}
break;
default:
break;
}
break;
default:
break;
}
}
NOT_S_OK:
exit:
CloseStreamReader(pReader, pFileStream);
return hr;
}
HRESULT ReadList(LPCWSTR path, LPCWSTR section, APPDATAXMLLIST &list)
{
HRESULT hr;
IXmlReader *pReader = NULL;
IStream *pFileStream = NULL;
XmlNodeType nodeType;
LPCWSTR pwszLocalName;
LPCWSTR pwszAttributeName;
LPCWSTR pwszAttributeValue;
int sequence = 0;
APPDATAXMLATTR attr;
APPDATAXMLROW row;
hr = CreateStreamReader(path, &pReader, &pFileStream);
EXIT_NOT_S_OK(hr);
while(pReader->Read(&nodeType) == S_OK)
{
switch(nodeType)
{
case XmlNodeType_Element:
hr = pReader->GetLocalName(&pwszLocalName, NULL);
EXIT_NOT_S_OK(hr);
switch(sequence)
{
case 0:
if(wcscmp(TagRoot, pwszLocalName) == 0)
{
sequence = 1;
}
break;
case 1:
if(wcscmp(TagSection, pwszLocalName) == 0)
{
sequence = 2;
}
break;
case 3:
if(wcscmp(TagList, pwszLocalName) == 0)
{
sequence = 4;
}
break;
case 4:
if(wcscmp(TagRow, pwszLocalName) == 0)
{
sequence = 5;
row.clear();
}
break;
default:
break;
}
for(hr = pReader->MoveToFirstAttribute(); hr == S_OK; hr = pReader->MoveToNextAttribute())
{
hr = pReader->GetLocalName(&pwszAttributeName, NULL);
EXIT_NOT_S_OK(hr);
hr = pReader->GetValue(&pwszAttributeValue, NULL);
EXIT_NOT_S_OK(hr);
switch(sequence)
{
case 2:
if(wcscmp(TagSection, pwszLocalName) == 0 &&
wcscmp(AttributeName, pwszAttributeName) == 0 && wcscmp(section, pwszAttributeValue) == 0)
{
sequence = 3;
}
break;
case 5:
if(wcscmp(TagRow, pwszLocalName) == 0)
{
attr.first = pwszAttributeName;
attr.second = pwszAttributeValue;
row.push_back(attr);
}
break;
default:
break;
}
}
switch(sequence)
{
case 2:
sequence = 1;
break;
case 5:
list.push_back(row);
row.clear();
break;
default:
break;
}
break;
case XmlNodeType_EndElement:
hr = pReader->GetLocalName(&pwszLocalName, NULL);
EXIT_NOT_S_OK(hr);
switch(sequence)
{
case 1:
if(wcscmp(TagRoot, pwszLocalName) == 0)
{
goto exit;
}
break;
case 2:
if(wcscmp(TagSection, pwszLocalName) == 0)
{
sequence = 1;
}
break;
case 3:
if(wcscmp(TagSection, pwszLocalName) == 0)
{
goto exit;
}
break;
case 4:
if(wcscmp(TagSection, pwszLocalName) == 0 || wcscmp(TagList, pwszLocalName) == 0)
{
goto exit;
}
break;
case 5:
if(wcscmp(TagList, pwszLocalName) == 0)
{
goto exit;
}
break;
default:
break;
}
break;
default:
break;
}
}
NOT_S_OK:
exit:
CloseStreamReader(pReader, pFileStream);
return hr;
}
/*
Modify this to avoid temp file I/O
*/
bool CXMLParser::extractData( wstring filename)
{
bool bRet = false;
HRESULT hr = S_OK;
IXmlReader *pReader = nullptr;
IStream *pStream = nullptr;
XmlNodeType nodeType;
const TCHAR* pwszName;
const TCHAR* pwszValue;
if (FAILED(hr = SHCreateStreamOnFile(filename.c_str(), STGM_READ, &pStream)))
{
return bRet;
}
if (!GetDataContainer())
{
if (FAILED(hr = CreateXmlReader(__uuidof(IXmlReader), (void**)&pReader, nullptr)))
{
return bRet;
}
if (FAILED(hr = pReader->SetInput(pStream)))
{
return bRet;
}
bool inKey = false;
wstring csKeyFoundStr;
while (!pReader->IsEOF())
{
pReader->Read(&nodeType);
switch (nodeType)
{
case XmlNodeType_Element:
if (S_OK == pReader->GetLocalName(&pwszName, nullptr))
{
if (true == isNodeInKeys(pwszName))
{
inKey = true;
csKeyFoundStr = pwszName;
}
}
break;
case XmlNodeType_CDATA:
case XmlNodeType_Text:
{
if (true == inKey)
{
if (S_OK == pReader->GetValue(&pwszValue, nullptr))
{
GetDataContainer()->setkeyValue(csKeyFoundStr, pwszValue);
inKey = false;
bRet = true;
}
}
}
break;
default:
break;
}
}
}
return bRet;
}