add to test - search with strip_accents
[mdictionary] / trunk / src / plugins / google / src / GooglePlugin.cpp
1 /*******************************************************************************
2
3     This file is part of mDictionary.
4
5     mDictionary is free software: you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation, either version 3 of the License, or
8     (at your option) any later version.
9
10     mDictionary is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License
16     along with mDictionary.  If not, see <http://www.gnu.org/licenses/>.
17
18     Copyright 2010 Comarch S.A.
19
20 *******************************************************************************/
21
22 /*! \file GooglePlugin.cpp
23     \author Jakub Jaszczynski <j.j.jaszczynski@gmail.com>
24 */
25
26 #include "GooglePlugin.h"
27 #include <QDebug>
28 #include "GoogleDictDialog.h"
29
30
31 GooglePlugin::GooglePlugin(QObject *parent): CommonDictInterface(parent),
32                     _langFrom(tr("")), _langTo(tr("")),_name(tr("")),
33                     _type(tr("google")), _infoNote(tr("")) {
34
35     stopped = false;
36     _connectionAccept = false;
37     _settings = new Settings();
38     _dictDialog = new GoogleDictDialog(this,this);
39
40     _icon = QIcon(":/icons/drawing.png");
41     languages=initLanguages();
42     _settings->setValue("type","google");
43     _settings->setValue("connectionAcepted","false");
44
45     http = new QHttp(this);
46     connect(http, SIGNAL(done(bool)), this, SLOT(done()));
47 }
48
49 GooglePlugin::~GooglePlugin() {
50     delete _settings;
51 }
52
53 QString GooglePlugin::langFrom() const {
54     return languages.key(_langFrom);
55 }
56 QString GooglePlugin::langTo() const {
57     return languages.key(_langTo);
58 }
59
60 QString GooglePlugin::name() const {
61     return QString("dictionary");
62 }
63
64 QString GooglePlugin::type() const {
65     return _type;
66 }
67
68 QString GooglePlugin::infoNote() const {
69     return _infoNote;
70 }
71
72 void GooglePlugin::setLangTo(QString langTo){
73     _langTo=langTo;
74 }
75
76 void GooglePlugin::setLangFrom(QString langFrom){
77     _langFrom=langFrom;
78 }
79
80 DictDialog* GooglePlugin::dictDialog() {
81     return _dictDialog;
82 }
83
84 bool GooglePlugin::isAvailable() const {
85     if(_connectionAccept)
86         return true;
87     else
88         return false;
89 }
90
91 void GooglePlugin::setConnectionAccept(QString connectionAcepted) {
92     if(connectionAcepted=="true")
93         _connectionAccept=true;
94     else
95         _connectionAccept=false;
96 }
97
98 bool GooglePlugin::isConnectionAccept() {
99     return _connectionAccept;
100 }
101
102 QString GooglePlugin::search(QString) {
103     /*function is not used on this plagin */
104     return QString("");
105 }
106
107 uint GooglePlugin::hash() const {
108     return _hash;
109 }
110
111 void GooglePlugin::setHash(uint) {
112     this->_hash=_hash;
113 }
114
115 Settings* GooglePlugin::settings() {
116     return _settings;
117 }
118
119 QIcon* GooglePlugin::icon() {
120     return &_icon;
121 }
122
123 CommonDictInterface* GooglePlugin::getNew(const Settings* settings) const
124 {
125     GooglePlugin *plugin = new GooglePlugin();
126     if(settings) {
127         plugin->setLangFrom(settings->value("langFrom"));
128         plugin->setLangTo(settings->value("langTo"));
129         plugin->setConnectionAccept(settings->value("connectionAccept"));
130         QStringList list = settings->keys();
131         foreach(QString key, list)
132             plugin->settings()->setValue(key, settings->value(key));
133     }
134     delete settings;
135     plugin->getDictionaryInfo();
136     return plugin;
137 }
138
139 void GooglePlugin::setSettings(Settings* settings) {
140     if(settings){
141         _langFrom=settings->value("langFrom");
142         _langTo=settings->value("langTo");
143         setConnectionAccept(settings->value("connectionAccept"));
144         QStringList list = settings->keys();
145         foreach(QString key, list)
146             _settings->setValue(key, settings->value(key));
147         delete settings;
148     }
149 }
150
151 Translation* GooglePlugin::getTranslationFor(QString key) {
152     return (new TranslationGoogle(key,"",_infoNote,this));
153 }
154
155 QList<Translation*> GooglePlugin::searchWordList(QString word, int ) {
156     QList<Translation*> translations;
157     if(isAvailable()) {
158         word.replace("*","");
159         word.replace("?","");
160         QString url = QString("/translate_a/t?client=t&sl=%1&tl=%2").arg(_langFrom,
161                                                                          _langTo);
162         QHttpRequestHeader head = QHttpRequestHeader("POST", url, 1,1);
163         head.setValue("Host", "www.google.pl");
164         head.setValue("User-Agent", "Mozilla/5.0");
165         head.setValue("Accept-Encoding", "deflate");
166         head.setContentLength(word.length());
167         head.setValue("Connection", "Close");
168
169         QByteArray data("text=");
170         data.append(word.toUtf8());
171         http->setHost("www.google.pl");
172
173         wait=true;
174         stopped=false;
175         http->request(head, data);
176         QString error("");
177         while(wait && (error=="" || error=="Unknown error") && !stopped)
178             error=http->errorString();
179
180         if(error!="" && error!="Unknown error") {
181             qDebug()<<error;
182             Q_EMIT notify(Notify::Warning,
183                     QString("GooglePlugin: %1").arg(error));
184         }
185         QString text = QString::fromUtf8(http->readAll());
186         text=jsonParse(text);
187         if(text!="") {
188             text=tr("<key>") + word + tr("</key>") + tr("<t>") + text + tr("</t>");
189             translations<<(new TranslationGoogle(word,text,_infoNote,this));
190         }
191     }
192     return translations;
193 }
194
195 QString GooglePlugin::jsonParse(QString result) {
196     int pos=0,pos2=0;
197     int index=0;
198     int size=0;
199     QString returnLang;
200     QString translation;
201     QString original;
202     QList<QString> partOfSpeach;
203     QList<QList<QString>* > words;
204
205     QStringList list1 = result.split("\"");
206     size=(list1.size()-1)/2;
207     if(size<=2)
208         return QString(""); // wrong format of data
209     translation=list1.at(1);
210     original=list1.at(3);
211     pos=result.indexOf("]");
212     pos=result.indexOf("]",pos+1);
213     pos++;
214     index=3;
215     if(result.at(pos+1)==QChar(','))
216         returnLang=list1.at(index*2+1);
217     while(result.indexOf("[",pos+1)!=-1){
218         partOfSpeach.append(list1.at(index*2+1));
219         pos2=result.indexOf("]",pos+1);
220         pos=result.indexOf("\"",pos+1);
221         pos=result.indexOf("\"",pos+1);
222         pos=result.indexOf("\"",pos+1);
223         pos=result.indexOf("\"",pos+1);
224         QList<QString> *list=new QList<QString>;
225         while(pos2>pos && pos2!=-1 && pos!= -1) {
226             index++;
227             if(size==index)
228                 return QString("");
229             list->append(list1.at(index*2+1));
230             pos=result.indexOf("\"",pos+1);
231             pos=result.indexOf("\"",pos+1);
232         }
233         words.append(list);
234         index++;
235         if(size==index)
236             return QString("");
237         pos=pos2+2;
238     }
239     if(words.size()!=partOfSpeach.size()) {
240         qDebug()<<"Error in Google Plugin (JsonParse)";
241         Q_EMIT notify(Notify::Warning,
242                 QString("GooglePlugin: can't parse Json"));
243         return QString("");
244     }
245     returnLang=list1.at(index*2+1);
246     if(partOfSpeach.size()==0){
247         if(translation==original){
248             return QString(""); // word don't exist";
249         }
250         else
251             return "\""+translation+"\"";
252     }
253     result=translation;
254     for(int i=0; i<words.size();i++) {
255         result+="<br/>";
256         result+="<pos>"+partOfSpeach.at(i)+"</pos>:";
257         for(int j=0; j<words.at(i)->size();j++)
258             result+=words.at(i)->at(j)+", ";
259         result.remove(result.size()-2,2);
260     }
261     return result;
262 }
263
264 void GooglePlugin::done() {
265     wait=false;
266 }
267
268 void GooglePlugin::stop() {
269     stopped=true;
270 }
271
272 QMap<QString, QString> GooglePlugin::initLanguages() {
273     QMap<QString, QString> languages;
274     languages["AFRIKAANS"] = "af";
275     languages["ALBANIAN"] = "sq";
276     languages["AMHARIC"] = "am";
277     languages["ARABIC"] = "ar";
278     languages["ARMENIAN"] = "hy";
279     languages["AZERBAIJANI"] = "az";
280     languages["BASQUE"] = "eu";
281     languages["BELARUSIAN"] = "be";
282     languages["BENGALI"] = "bn";
283     languages["BIHARI"] = "bh";
284     languages["BRETON"] = "br";
285     languages["BULGARIAN"] = "bg";
286     languages["BURMESE"] = "my";
287     languages["CATALAN"] = "ca";
288     languages["CHEROKEE"] = "chr";
289     languages["CHINESE"] = "zh";
290     languages["CHINESE_SIMPLIFIED"] = "zh-CN";
291     languages["CHINESE_TRADITIONAL"] = "zh-TW";
292     languages["CORSICAN"] = "co";
293     languages["CROATIAN"] = "hr";
294     languages["CZECH"] = "cs";
295     languages["DANISH"] = "da";
296     languages["DHIVEHI"] = "dv";
297     languages["DUTCH"] = "nl";
298     languages["ENGLISH"] = "en";
299     languages["ESPERANTO"] = "eo";
300     languages["ESTONIAN"] = "et";
301     languages["FAROESE"] = "fo";
302     languages["FILIPINO"] = "tl";
303     languages["FINNISH"] = "fi";
304     languages["FRENCH"] = "fr";
305     languages["FRISIAN"] = "fy";
306     languages["GALICIAN"] = "gl";
307     languages["GEORGIAN"] = "ka";
308     languages["GERMAN"] = "de";
309     languages["GREEK"] = "el";
310     languages["GUJARATI"] = "gu";
311     languages["HAITIAN_CREOLE"] = "ht";
312     languages["HEBREW"] = "iw";
313     languages["HINDI"] = "hi";
314     languages["HUNGARIAN"] = "hu";
315     languages["ICELANDIC"] = "is";
316     languages["INDONESIAN"] = "id";
317     languages["INUKTITUT"] = "iu";
318     languages["IRISH"] = "ga";
319     languages["ITALIAN"] = "it";
320     languages["JAPANESE"] = "ja";
321     languages["JAVANESE"] = "jw";
322     languages["KANNADA"] = "kn";
323     languages["KAZAKH"] = "kk";
324     languages["KHMER"] = "km";
325     languages["KOREAN"] = "ko";
326     languages["KURDISH"] = "ku";
327     languages["KYRGYZ"] = "ky";
328     languages["LAO"] = "lo";
329     languages["LATIN"] = "la";
330     languages["LATVIAN"] = "lv";
331     languages["LITHUANIAN"] = "lt";
332     languages["LUXEMBOURGISH"] = "lb";
333     languages["MACEDONIAN"] = "mk";
334     languages["MALAY"] = "ms";
335     languages["MALAYALAM"] = "ml";
336     languages["MALTESE"] = "mt";
337     languages["MAORI"] = "mi";
338     languages["MARATHI"] = "mr";
339     languages["MONGOLIAN"] = "mn";
340     languages["NEPALI"] = "ne";
341     languages["NORWEGIAN"] = "no";
342     languages["OCCITAN"] = "oc";
343     languages["ORIYA"] = "or";
344     languages["PASHTO"] = "ps";
345     languages["PERSIAN"] = "fa";
346     languages["POLISH"] = "pl";
347     languages["PORTUGUESE"] = "pt";
348     languages["PORTUGUESE_PORTUGAL"] = "pt-PT";
349     languages["PUNJABI"] = "pa";
350     languages["QUECHUA"] = "qu";
351     languages["ROMANIAN"] = "ro";
352     languages["RUSSIAN"] = "ru";
353     languages["SANSKRIT"] = "sa";
354     languages["SCOTS_GAELIC"] = "gd";
355     languages["SERBIAN"] = "sr";
356     languages["SINDHI"] = "sd";
357     languages["SINHALESE"] = "si";
358     languages["SLOVAK"] = "sk";
359     languages["SLOVENIAN"] = "sl";
360     languages["SPANISH"] = "es";
361     languages["SUNDANESE"] = "su";
362     languages["SWAHILI"] = "sw";
363     languages["SWEDISH"] = "sv";
364     languages["SYRIAC"] = "syr";
365     languages["TAJIK"] = "tg";
366     languages["TAMIL"] = "ta";
367     languages["TATAR"] = "tt";
368     languages["TELUGU"] = "te";
369     languages["THAI"] = "th";
370     languages["TIBETAN"] = "bo";
371     languages["TONGA"] = "to";
372     languages["TURKISH"] = "tr";
373     languages["UKRAINIAN"] = "uk";
374     languages["URDU"] = "ur";
375     languages["UZBEK"] = "uz";
376     languages["UIGHUR"] = "ug";
377     languages["VIETNAMESE"] = "vi";
378     languages["WELSH"] = "cy";
379     languages["YIDDISH"] = "yi";
380     languages["YORUBA"] = "yo";
381     languages["UNKNOWN"] = "";
382     return languages;
383 }
384
385 void GooglePlugin::getDictionaryInfo() {
386     QString fullLangFrom=languages.key(_langFrom);
387     QString fullLangTo=languages.key(_langTo);
388     _infoNote=" [" + fullLangFrom + "-";
389     _infoNote+= fullLangTo + "] (" + _type + ")";
390 }
391
392
393 Q_EXPORT_PLUGIN2(google, GooglePlugin)