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

XslLocator::XslLocator() : d ( new XslLocatorData() )
{
    d->parser = p;
    XML_SetUserData ( p, d.get() );
    XML_SetProcessingInstructionHandler ( p, processingInstructionHandler );
}

XslLocator::~XslLocator()
{}

void XMLCALL XslLocator::processingInstructionHandler (
    void *userData,
    const XML_Char *target,
    const XML_Char *data )
{
    XslLocatorData *d;
    d = ( XslLocatorData * ) userData;

    if ( strcmp ( target, "xml-stylesheet" ) )
        return;

    char *value, *iterator;
    value = strstr ( ( const char * ) data, "href=" );
    if ( !value || strlen ( value ) < 7 )
        return;
    value += 6;
    iterator = value;
    for ( iterator = value; *iterator; ++iterator )
    {
        if ( *iterator == '"' || *iterator == '\'' )
        {
            *iterator = '\0';
            break;
        }
    }

    d->xslLocation = value;
    XML_StopParser ( d->parser, false );
}


std::string XslLocator::getXslLocation()
{
    return d->xslLocation;
}