runtime/kurifilter-plugins/fixhost/fixhosturifilter.cpp

   1 /*
   2     fixhostfilter.cpp
   3
   4     This file is part of the KDE project
   5     Copyright (C) 2007 Lubos Lunak <llunak@suse.cz>
   6
   7     This program is free software; you can redistribute it and/or modify
   8     it under the terms of the GNU General Public License as published by
   9     the Free Software Foundation; either version 2 of the License, or
  10     (at your option) any later version.
  11
  12     This program is distributed in the hope that it will be useful,
  13     but WITHOUT ANY WARRANTY; without even the implied warranty of
  14     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15     GNU General Public License for more details.
  16
  17     You should have received a copy of the GNU General Public License
  18     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include <config-runtime.h>
  22
  23 #include "fixhosturifilter.h"
  24
  25 #include <kdebug.h>
  26 #include <kurl.h>
  27 #include <k3resolver.h>
  28
  29 using namespace KNetwork;
  30
  31 /**
  32  * IMPORTANT: If you change anything here, please run the regression test
  33  * ../tests/kurifiltertest
  34  */
  35
  36 FixHostUriFilter::FixHostUriFilter( QObject *parent, const QVariantList & /*args*/ )
  37     : KUriFilterPlugin( "fixhosturifilter", parent )
  38 {
  39 }
  40
  41 bool FixHostUriFilter::filterUri( KUriFilterData& data ) const
  42 {
  43     KUrl url = data.uri();
  44     QString cmd = url.url();
  45
  46     kDebug() << "FixHostUriFilter::filterUri: " << url;
  47
  48     KUrl url2 = url;
  49     url2.setHost( "www." + url.host());
  50
  51     if(( url.protocol() == "http" || url.protocol() == "https" )
  52         && !url.host().startsWith( "www." ) && !exists( url ) && exists( url2 ))
  53     {
  54         setFilteredUri( data, url2 );
  55         setUriType( data, KUriFilterData::NetProtocol );
  56
  57         kDebug() << "FilteredUri: " << data.uri();
  58         return true;
  59     }
  60
  61     return false;
  62 }
  63
  64 bool FixHostUriFilter::exists( const KUrl& url )
  65 {
  66     KResolver resolver( url.host());
  67     resolver.setFamily( KResolver::InetFamily );
  68     return( resolver.start() && resolver.wait( 5000 ) && resolver.error() == KResolver::NoError );
  69 }
  70
  71 K_PLUGIN_FACTORY(FixHostUriFilterFactory, registerPlugin<FixHostUriFilter>();)
  72 K_EXPORT_PLUGIN(FixHostUriFilterFactory("kcmkurifilt"))
  73
  74 #include "fixhosturifilter.moc"