You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdenetwork/kopete/plugins/translator/translatorplugin.cpp

403 lines
13 KiB

/*
translatorplugin.cpp
Kopete Translator plugin
Copyright (c) 2001-2002 by Duncan Mac-Vicar Prett <duncan@kde.org>
Copyright (c) 2002-2004 by Olivier Goffart <ogoffart @ kde.org>
Kopete (c) 2002-2004 by the Kopete developers <kopete-devel@kde.org>
*************************************************************************
* *
* 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 <qapplication.h>
#include <qregexp.h>
#include <qsignal.h>
#include <qstring.h>
#include <kdebug.h>
#include <kaction.h>
#include <kgenericfactory.h>
#include <kglobal.h>
#include <kconfig.h>
#include <kdeversion.h>
#include <kaboutdata.h>
#include "kopetemetacontact.h"
#include "kopetecontactlist.h"
#include "kopetemessagemanagerfactory.h"
#include "kopetecontact.h"
#include "translatorplugin.h"
#include "translatordialog.h"
#include "translatorguiclient.h"
#include "translatorlanguages.h"
typedef KGenericFactory<TranslatorPlugin> TranslatorPluginFactory;
#if KDE_IS_VERSION(3,2,90)
static const KAboutData aboutdata("kopete_translator", I18N_NOOP("Translator") , "1.0" );
K_EXPORT_COMPONENT_FACTORY( kopete_translator, TranslatorPluginFactory( &aboutdata ) )
#else
K_EXPORT_COMPONENT_FACTORY( kopete_translator, TranslatorPluginFactory( "kopete_translator" ) )
#endif
TranslatorPlugin::TranslatorPlugin( QObject *parent, const char *name, const QStringList & /* args */ )
: Kopete::Plugin( TranslatorPluginFactory::instance(), parent, name )
{
kdDebug( 14308 ) << k_funcinfo << endl;
if ( pluginStatic_ )
kdWarning( 14308 ) << k_funcinfo << "Translator already initialized" << endl;
else
pluginStatic_ = this;
m_languages = new TranslatorLanguages;
connect( Kopete::ChatSessionManager::self(), SIGNAL( aboutToDisplay( Kopete::Message & ) ),
this, SLOT( slotIncomingMessage( Kopete::Message & ) ) );
connect( Kopete::ChatSessionManager::self(), SIGNAL( aboutToSend( Kopete::Message & ) ),
this, SLOT( slotOutgoingMessage( Kopete::Message & ) ) );
connect( Kopete::ChatSessionManager::self(), SIGNAL( chatSessionCreated( Kopete::ChatSession * ) ),
this, SLOT( slotNewKMM( Kopete::ChatSession * ) ) );
QStringList keys;
QMap<QString, QString> m = m_languages->languagesMap();
for ( int k = 0; k <= m_languages->numLanguages(); k++ )
keys << m[ m_languages->languageKey( k ) ];
m_actionLanguage = new KSelectAction( i18n( "Set &Language" ), "locale", 0, actionCollection(), "contactLanguage" );
m_actionLanguage->setItems( keys );
connect( m_actionLanguage, SIGNAL( activated() ), this, SLOT(slotSetLanguage() ) );
connect( Kopete::ContactList::self(), SIGNAL( metaContactSelected( bool ) ), this, SLOT( slotSelectionChanged( bool ) ) );
setXMLFile( "translatorui.rc" );
//Add GUI action to all already existing kmm (if the plugin is launched when kopete already rining)
QValueList<Kopete::ChatSession*> sessions = Kopete::ChatSessionManager::self()->sessions();
for (QValueListIterator<Kopete::ChatSession*> it= sessions.begin(); it!=sessions.end() ; ++it)
slotNewKMM( *it );
loadSettings();
connect( this, SIGNAL( settingsChanged() ), this, SLOT( loadSettings() ) );
}
TranslatorPlugin::~TranslatorPlugin()
{
kdDebug( 14308 ) << k_funcinfo << endl;
pluginStatic_ = 0L;
}
TranslatorPlugin* TranslatorPlugin::plugin()
{
return pluginStatic_;
}
TranslatorPlugin* TranslatorPlugin::pluginStatic_ = 0L;
void TranslatorPlugin::loadSettings()
{
KConfig *config = KGlobal::config();
int mode = 0;
config->setGroup( "Translator Plugin" );
m_myLang = m_languages->languageKey( config->readNumEntry( "myLang" , 0 ) );
m_service = m_languages->serviceKey( config->readNumEntry( "Service", 0 ) );
if ( config->readBoolEntry( "IncomingDontTranslate", true ) )
mode = 0;
else if ( config->readBoolEntry( "IncomingShowOriginal", false ) )
mode = 1;
else if ( config->readBoolEntry( "IncomingTranslate", false ) )
mode = 2;
m_incomingMode = mode;
if ( config->readBoolEntry( "OutgoingDontTranslate", true ) )
mode = 0;
else if ( config->readBoolEntry( "OutgoingShowOriginal", false ) )
mode = 1;
else if ( config->readBoolEntry( "OutgoingTranslate", false ) )
mode = 2;
else if ( config->readBoolEntry( "OutgoingAsk", false ) )
mode = 3;
m_outgoingMode = mode;
}
void TranslatorPlugin::slotSelectionChanged( bool b )
{
m_actionLanguage->setEnabled( b );
if ( !b )
return;
Kopete::MetaContact *m = Kopete::ContactList::self()->selectedMetaContacts().first();
if( !m )
return;
QString languageKey = m->pluginData( this, "languageKey" );
if ( !languageKey.isEmpty() && languageKey != "null" )
m_actionLanguage->setCurrentItem( m_languages->languageIndex( languageKey ) );
else
m_actionLanguage->setCurrentItem( m_languages->languageIndex( "null" ) );
}
void TranslatorPlugin::slotNewKMM( Kopete::ChatSession *KMM )
{
new TranslatorGUIClient( KMM );
}
void TranslatorPlugin::slotIncomingMessage( Kopete::Message &msg )
{
if ( m_incomingMode == DontTranslate )
return;
QString src_lang;
QString dst_lang;
if ( ( msg.direction() == Kopete::Message::Inbound ) && !msg.plainBody().isEmpty() )
{
Kopete::MetaContact *from = msg.from()->metaContact();
if ( !from )
{
// kdDebug( 14308 ) << k_funcinfo << "No metaContact for source contact" << endl;
return;
}
src_lang = from->pluginData( this, "languageKey" );
if( src_lang.isEmpty() || src_lang == "null" )
{
// kdDebug( 14308 ) << k_funcinfo << "Cannot determine src Metacontact language (" << from->displayName() << ")" << endl;
return;
}
dst_lang = m_myLang;
sendTranslation( msg, translateMessage( msg.plainBody(), src_lang, dst_lang ) );
}
}
void TranslatorPlugin::slotOutgoingMessage( Kopete::Message &msg )
{
if ( m_outgoingMode == DontTranslate )
return;
QString src_lang;
QString dst_lang;
if ( ( msg.direction() == Kopete::Message::Outbound ) && ( !msg.plainBody().isEmpty() ) )
{
src_lang = m_myLang;
// Sad, we have to consider only the first To: metacontact
Kopete::MetaContact *to = msg.to().first()->metaContact();
if ( !to )
{
// kdDebug( 14308 ) << k_funcinfo << "No metaContact for first contact" << endl;
return;
}
dst_lang = to->pluginData( this, "languageKey" );
if ( dst_lang.isEmpty() || dst_lang == "null" )
{
// kdDebug( 14308 ) << k_funcinfo << "Cannot determine dst Metacontact language (" << to->displayName() << ")" << endl;
return;
}
sendTranslation( msg, translateMessage( msg.plainBody(), src_lang, dst_lang ) );
}
}
void TranslatorPlugin::translateMessage( const QString &msg, const QString &from, const QString &to, QObject *obj, const char* slot )
{
QSignal completeSignal;
completeSignal.connect( obj, slot );
QString result = translateMessage( msg, from, to );
if(!result.isNull())
{
completeSignal.setValue( result );
completeSignal.activate();
}
}
QString TranslatorPlugin::translateMessage( const QString &msg, const QString &from, const QString &to )
{
if ( from == to )
{
kdDebug( 14308 ) << k_funcinfo << "Src and Dst languages are the same" << endl;
return QString::null;
}
// We search for src_dst
if(! m_languages->supported( m_service ).contains( from + "_" + to ) )
{
kdDebug( 14308 ) << k_funcinfo << from << "_" << to << " is not supported by service " << m_service << endl;
return QString::null;
}
if ( m_service == "babelfish" )
return babelTranslateMessage( msg ,from, to );
else if ( m_service == "google" )
return googleTranslateMessage( msg ,from, to );
else
return QString::null;
}
QString TranslatorPlugin::googleTranslateMessage( const QString &msg, const QString &from, const QString &to )
{
KURL translatorURL ( "http://translate.google.com/translate_t");
QString body = KURL::encode_string( msg );
QString lp = from + "|" + to;
QCString postData = QString( "text=" + body + "&langpair=" + lp ).utf8();
QString gurl = "http://translate.google.com/translate_t?text=" + body +"&langpair=" + lp;
kdDebug(14308) << k_funcinfo << " URL: " << gurl << endl;
KURL geturl ( gurl );
KIO::TransferJob *job = KIO::get( geturl, false, true );
//job = KIO::http_post( translatorURL, postData, true );
//job->addMetaData( "content-type", "application/x-www-form-urlencoded" );
//job->addMetaData( "referrer", "http://www.google.com" );
QObject::connect( job, SIGNAL( data( KIO::Job *, const QByteArray & ) ), this, SLOT( slotDataReceived( KIO::Job *, const QByteArray & ) ) );
QObject::connect( job, SIGNAL( result( KIO::Job * ) ), this, SLOT( slotJobDone( KIO::Job * ) ) );
// KIO is async and we use a sync API, so use the processEvents hack to work around that
// FIXME: We need to make the libkopete API async to get rid of this processEvents.
// It often causes crashes in the code. - Martijn
while ( !m_completed[ job ] )
qApp->processEvents();
QString data = QString::fromLatin1( m_data[ job ] );
// After hacks, we need to clean
m_data.remove( job );
m_completed.remove( job );
// kdDebug( 14308 ) << k_funcinfo << "Google response:"<< endl << data << endl;
QRegExp re( "<textarea name=q rows=5 cols=45 wrap=PHYSICAL>(.*)</textarea>" );
re.setMinimal( true );
re.search( data );
return re.cap( 1 );
}
QString TranslatorPlugin::babelTranslateMessage( const QString &msg, const QString &from, const QString &to )
{
QString body = KURL::encode_string( msg);
QString lp = from + "_" + to;
QString gurl = "http://babelfish.altavista.com/babelfish/tr?enc=utf8&doit=done&tt=urltext&urltext=" + body + "&lp=" + lp;
KURL geturl ( gurl );
kdDebug( 14308 ) << k_funcinfo << "URL: " << gurl << endl;
KIO::TransferJob *job = KIO::get( geturl, false, true );
QObject::connect( job, SIGNAL( data( KIO::Job *, const QByteArray & ) ), this, SLOT( slotDataReceived( KIO::Job *, const QByteArray & ) ) );
QObject::connect( job, SIGNAL( result( KIO::Job * ) ), this, SLOT( slotJobDone( KIO::Job * ) ) );
// KIO is async and we use a sync API, so use the processEvents hack to work around that
// FIXME: We need to make the libkopete API async to get rid of this processEvents.
// It often causes crashes in the code. - Martijn
while ( !m_completed[ job ] )
qApp->processEvents();
QString data = QString::fromUtf8( m_data[ job ] );
// After hacks, we need to clean
m_data.remove( job );
m_completed.remove( job );
//kdDebug( 14308 ) << k_funcinfo << "Babelfish response: " << endl << data << endl;
QRegExp re( "<Div style=padding:10px; lang=..>(.*)</div" );
re.setMinimal( true );
re.search( data );
return re.cap( 1 );
}
void TranslatorPlugin::sendTranslation( Kopete::Message &msg, const QString &translated )
{
if ( translated.isEmpty() )
{
kdWarning( 14308 ) << k_funcinfo << "Translated text is empty" << endl;
return;
}
TranslateMode mode = DontTranslate;
switch ( msg.direction() )
{
case Kopete::Message::Outbound:
mode = TranslateMode( m_outgoingMode );
break;
case Kopete::Message::Inbound:
mode = TranslateMode( m_incomingMode );
break;
default:
kdWarning( 14308 ) << k_funcinfo << "Can't determine if it is an incoming or outgoing message" << endl;
};
switch ( mode )
{
case JustTranslate:
msg.setBody( translated, msg.format() );
break;
case ShowOriginal:
msg.setBody( i18n( "%2\nAuto Translated: %1" ).arg( translated, msg.plainBody() ), msg.format() );
break;
case ShowDialog:
{
TranslatorDialog *d = new TranslatorDialog( translated );
d->exec();
msg.setBody( d->translatedText(), msg.format() );
delete d;
break;
}
case DontTranslate:
default:
//do nothing
break;
};
}
void TranslatorPlugin::slotDataReceived ( KIO::Job *job, const QByteArray &data )
{
m_data[ job ] += QCString( data, data.size() + 1 );
}
void TranslatorPlugin::slotJobDone ( KIO::Job *job )
{
m_completed[ job ] = true;
QObject::disconnect( job, SIGNAL( data( KIO::Job *, const QByteArray & ) ), this, SLOT( slotDataReceived( KIO::Job *, const QByteArray & ) ) );
QObject::disconnect( job, SIGNAL( result( KIO::Job * ) ), this, SLOT( slotJobDone( KIO::Job * ) ) );
}
void TranslatorPlugin::slotSetLanguage()
{
Kopete::MetaContact *m = Kopete::ContactList::self()->selectedMetaContacts().first();
if( m && m_actionLanguage )
m->setPluginData( this, "languageKey", m_languages->languageKey( m_actionLanguage->currentItem() ) );
}
#include "translatorplugin.moc"
// vim: set noet ts=4 sts=4 sw=4: