/*
 * Copyright (c) 2011-2014, Intel Corporation
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation and/or
 * other materials provided with the distribution.
 *
 * 3. Neither the name of the copyright holder nor the names of its contributors
 * may be used to endorse or promote products derived from this software without
 * specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
#include "MappingData.h"
#include "Tokenizer.h"
#include "Utility.h"
#include <assert.h>

CMappingData::CMappingData()
{
}

bool CMappingData::fromXml(const CXmlElement& xmlElement, CXmlSerializingContext& serializingContext)
{
    assert(xmlElement.hasAttribute("Mapping"));

    std::string strMapping = xmlElement.getAttributeString("Mapping");

    Tokenizer mappingTok(strMapping, ",");

    std::string strMappingElement;

    while (!(strMappingElement = mappingTok.next()).empty()) {

        std::string::size_type iFistDelimiterOccurrence = strMappingElement.find_first_of(':');

        std::string strKey, strValue;

        if (iFistDelimiterOccurrence == std::string::npos) {

            // There is no delimiter in the mapping field,
            // it means that no value has been provided
            strKey = strMappingElement;
            strValue = "";

        } else {

            // Get mapping key
            strKey = strMappingElement.substr(0, iFistDelimiterOccurrence);

            // Get mapping value
            strValue = strMappingElement.substr(iFistDelimiterOccurrence + 1);

        }

        if (!addValue(strKey, strValue)) {

            serializingContext.setError("Duplicate Mapping data: Unable to process Mapping element key = " + strKey + ", value = " + strValue + " from XML element " + xmlElement.getPath());

            return false;
        }
    }
    return true;
}

bool CMappingData::getValue(const std::string& strkey, const std::string*& pStrValue) const
{
    KeyToValueMapConstIterator it = _keyToValueMap.find(strkey);

    if (it != _keyToValueMap.end()) {

        pStrValue = &it->second;

        return true;
    }
    return false;
}

std::string CMappingData::asString() const
{
    std::string strValue;

    CUtility::asString(_keyToValueMap, strValue, ", ", ":");

    return strValue;
}

bool CMappingData::addValue(const std::string& strkey, const std::string& strValue)
{
    if (_keyToValueMap.find(strkey) != _keyToValueMap.end()) {

        return false;
    }
    _keyToValueMap[strkey] = strValue;

    return true;
}

