#include <string>
#include <vector>
#include <stdexcept>
#include <expat.h>
#include "housestylereader.h"

using namespace std;

HouseStyleReader::HouseStyleReader (
    map<string, map<string, set<string> > > &m
    ) : WrapExpat ( true ), ud ( new HouseStyleReaderData() )
{
    ud->setState ( STATE_UNKNOWN );
    ud->depth = ud->cutoffDepth = 0;
    ud->filterActive = false;
    ud->filterMap = m;

    XML_SetUserData ( p, ud.get() );
    XML_SetElementHandler ( p, start, end );
    XML_SetCharacterDataHandler ( p, characterdata );
    XML_SetCdataSectionHandler ( p, cdatastart, cdataend );
}

HouseStyleReader::~HouseStyleReader()
{}

void HouseStyleReader::getNodeVector ( vector<pair<string, unsigned> > &v )
{
    v = ud->nodevector;
}

void XMLCALL HouseStyleReader::start ( void *data, const XML_Char *el, const XML_Char **attr )
{
    HouseStyleReaderData *ud;
    ud = ( HouseStyleReaderData * ) data;
    ud->setState ( STATE_ON_START );
    ++ ( ud->depth );

    if ( ud->textnode.size() )
    {
        if ( !ud->filterActive )
            ud->nodevector.push_back ( make_pair ( ud->textnode, ud->getCount() ) );
        ud->textnode = "";
    }

    if ( !ud->filterActive && ud->filterMap.find ( el ) != ud->filterMap.end() )
    {
        map<string, set<string> > attributeMap;
        attributeMap = ud->filterMap[el];

        // no attribute keys/values specified
        if ( attributeMap.empty() )
        {
            ud->filterActive = true;
            ud->cutoffDepth = ud->depth;
        }

        // examine attribute keys/values
        else
        {

            for ( ; *attr; attr += 2 )
            {
                char *key, *value;
                key = ( char * ) *attr;
                value = ( char * ) * ( attr + 1 );
                set<string> valueSet;
                valueSet = attributeMap[key];

                if ( attributeMap.find ( key ) != attributeMap.end() )
                {
                    if ( valueSet.count ( value ) )
                    {
                        ud->filterActive = true;
                        ud->cutoffDepth = ud->depth;
                    }
                }
            }
        }
        ud->cutoffDepth = ud->depth;
    }
}

void XMLCALL HouseStyleReader::end ( void *data, const XML_Char *el )
{
    HouseStyleReaderData *ud;
    ud = ( HouseStyleReaderData * ) data;

    ud->setState ( STATE_ON_END );
    -- ( ud->depth );
    if ( ud->textnode.size() )
    {
        if ( !ud->filterActive )
            ud->nodevector.push_back ( make_pair ( ud->textnode, ud->getCount() ) );
        ud->textnode = "";
    }
    if ( ud->filterActive && ud->depth < ud->cutoffDepth )
    {
        ud->cutoffDepth = 0;
        ud->filterActive = false;
    }
}

void XMLCALL HouseStyleReader::characterdata ( void *data, const XML_Char *s, int len )
{
    HouseStyleReaderData *ud;
    ud = ( HouseStyleReaderData * ) data;
    ud->textnode.append ( s, len );
}

void XMLCALL HouseStyleReader::cdatastart ( void *data )
{
    HouseStyleReaderData *ud;
    ud = ( HouseStyleReaderData * ) data;
    ud->setState ( STATE_ON_CDATA_START );

    if ( ud->textnode.size() )
    {
        if ( !ud->filterActive )
            ud->nodevector.push_back ( make_pair ( ud->textnode, ud->getCount() ) );
        ud->textnode = "";
    }
}

void XMLCALL HouseStyleReader::cdataend ( void *data )
{
    HouseStyleReaderData *ud;
    ud = ( HouseStyleReaderData * ) data;
    ud->setState ( STATE_ON_CDATA_END );

    if ( ud->textnode.size() )
    {
        if ( !ud->filterActive )
            ud->nodevector.push_back ( make_pair ( ud->textnode, ud->getCount() ) );
        ud->textnode = "";
    }
}