rsssensor.cpp

/***************************************************************************
 *   Copyright (C) 2003 by Ralph M. Churchill                              *
 *   mrchucho@yahoo.com                                                    *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 ***************************************************************************/
 
#include "karambaapp.h"
#include "rsssensor.h"
#include <tqdom.h>
#include <tqregexp.h>
#include <kurl.h>
#include <tdeio/netaccess.h>
 
RssSensor::RssSensor( const TQString &src, int interval, const TQString &form, const TQString &enc)
    : Sensor(interval),
    source(src),
    format(form),
    encoding(enc)
 
{
    // Format:
    //  %t = title (DEFAULT)
    //  %d = desc
 
    if( !encoding.isEmpty() )
    {
        codec = TQTextCodec::codecForName( encoding.ascii() );
        if ( codec == 0)
            codec = TQTextCodec::codecForLocale();
    }
    else
        codec = TQTextCodec::codecForLocale();
}
 
RssSensor::~RssSensor()
{
}
 
void RssSensor::update()
{
    TQDomDocument doc;
    TQFile file;
    TQString tmpFile;
    bool OK = false;
 
    if(TDEIO::NetAccess::download(KURL(source), tmpFile, karambaApp->parentWindow()))
    {
        file.setName(tmpFile);
        if ( file.open(IO_ReadOnly | IO_Translate) )
        {
            if ( doc.setContent( &file ) )
            {
                OK = true;
            }
            else
            {
                tqDebug("Error on building DOM");
            }
        }
        else
        {
            tqDebug("Error opening file");
        }
    }
    else {
        tqDebug( "Error Downloading: %s", source.ascii());
    }
 
    if ( OK )
    {
        SensorParams *sp;
        Meter *meter;
 
        TQObjectListIt it( *objList );
        while (it != 0)
        {
            sp = (SensorParams*)(*it);
            meter = sp->getMeter();
 
            // this is a hack to force the
            // clickmap to reset its data lists
            meter->setValue(0);
 
            TQDomElement docElem = doc.documentElement();
            TQDomNode n = docElem.firstChild();
            if (!n.isNull())
            {
                TQDomNodeList links = docElem.elementsByTagName( "link" );
                TQDomNodeList displays;
                if ( format.contains( "%d", false ) > 0 )
                {
                    displays = docElem.elementsByTagName( "description" );
                }
                else
                {
                    displays = docElem.elementsByTagName( "title" );
                }
 
                TQRegExp rx("^http://", false );
                for (uint i=1; i < displays.count(); ++i )
                {
                    TQString dispTxt = displays.item( i ).toElement().text();
                    TQString linkTxt = links.item( i ).toElement().text();
                    if( (rx.search(dispTxt) == -1) && (rx.search(linkTxt) != -1) )
                    {
                        meter->setValue( dispTxt );
                        meter->setValue( linkTxt );
                    }
                    else
                    {
                        tqDebug("Skipping");
                    }
                }
            }
            else
            {
                tqDebug ("Document Node was null!!");
            }
 
            ++it;
        }
    }
    // Cleanup
    file.close();
    TDEIO::NetAccess::removeTempFile( tmpFile );
}
 
#include "rsssensor.moc"