init
[qstardict] / qstardict / dictcore.cpp
1 /*****************************************************************************
2  * dictcore.cpp - QStarDict, a StarDict clone written using Qt               *
3  * Copyright (C) 2008 Alexander Rodin                                        *
4  *                                                                           *
5  * This program 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 2 of the License, or         *
8  * (at your option) any later version.                                       *
9  *                                                                           *
10  * This program 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 along   *
16  * with this program; if not, write to the Free Software Foundation, Inc.,   *
17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.               *
18  *****************************************************************************/
19
20 #include "dictcore.h"
21
22 #include <QApplication>
23 #include <QFileInfoList>
24 #include <QFileInfo>
25 #include <QDir>
26 #include <QRegExp>
27 #include <QSettings>
28 #include <QDebug>
29 #include <QPluginLoader>
30 #include "../plugins/dictplugin.h"
31
32 namespace QStarDict
33 {
34
35 DictCore::DictCore(QObject *parent)
36     : QObject(parent)
37 {
38     loadSettings();
39 }
40
41 DictCore::~DictCore()
42 {
43     saveSettings();
44     foreach (QPluginLoader *loader, m_plugins)
45     {
46         delete loader->instance();
47         delete loader;
48     }
49 }
50
51 bool DictCore::isTranslatable(const QString &word)
52 {
53     for (QList<Dictionary>::const_iterator i = m_loadedDicts.begin(); i != m_loadedDicts.end(); ++i) 
54     {
55         if (! m_plugins.contains(i->plugin()))
56             continue;
57         if (qobject_cast<DictPlugin*>(m_plugins[i->plugin()]->instance())->isTranslatable(i->name(), word))
58             return true;
59     }
60     return false;
61 }
62
63 QString DictCore::translate(const QString &word)
64 {
65     QString simplifiedWord = word.simplified();
66     QString result;
67     for (QList<Dictionary>::const_iterator i = m_loadedDicts.begin(); i != m_loadedDicts.end(); ++i)
68     {
69         if (! m_plugins.contains(i->plugin()))
70             continue;
71         DictPlugin *plugin = qobject_cast<DictPlugin*>(m_plugins[i->plugin()]->instance());
72         if (! plugin->isTranslatable(i->name(), simplifiedWord))
73             continue;
74         DictPlugin::Translation translation = plugin->translate(i->name(), simplifiedWord);
75         result += "<p>\n"
76             "<font class=\"dict_name\">" + translation.dictName() + "</font><br>\n"
77             "<font class=\"title\">" + translation.title() + "</font><br>\n"
78             + translation.translation() + "</p>\n";
79     }
80     return result;
81 }
82
83 QStringList DictCore::findSimilarWords(const QString &word)
84 {
85     QString simplifiedWord = word.simplified();
86     QStringList result;
87     for (QList<Dictionary>::const_iterator i = m_loadedDicts.begin(); i != m_loadedDicts.end(); ++i)
88     {
89         if (! m_plugins.contains(i->plugin()))
90             continue;
91         DictPlugin *plugin = qobject_cast<DictPlugin*>(m_plugins[i->plugin()]->instance());
92         if (! plugin->features().testFlag(DictPlugin::SearchSimilar))
93             continue;
94         QStringList similar = plugin->findSimilarWords(i->name(), simplifiedWord);
95         for (QStringList::const_iterator j = similar.begin(); j != similar.end(); ++j)
96             if (! result.contains(*j, Qt::CaseSensitive))
97                 result << *j;
98     }
99     return result;
100 }
101
102 QStringList DictCore::availablePlugins() const
103 {
104     QStringList result;
105 #ifdef Q_WS_X11
106     QFileInfoList files = QDir(QSTARDICT_PLUGINS_DIR).entryInfoList(QStringList("lib*.so"),
107                   QDir::Files | QDir::NoDotAndDotDot);
108     for (QFileInfoList::const_iterator i = files.begin(); i != files.end(); ++i)
109         result << i->fileName().mid(3, i->fileName().length() - 6);
110 #elif defined Q_WS_WIN
111     QFileInfoList files = QDir(QSTARDICT_PLUGINS_DIR).entryInfoList(QStringList("*0.dll"),
112                   QDir::Files | QDir::NoDotAndDotDot);
113     for (QFileInfoList::const_iterator i = files.begin(); i != files.end(); ++i)
114         result << i->fileName().left(i->fileName().length() - 5);
115 #elif defined Q_WS_MAC
116     QStringList macFilters;
117     // various Qt versions...
118     macFilters << "*.dylib" << "*.bundle" << "*.so";
119     QString binPath = QCoreApplication::applicationDirPath();
120     // navigate through mac's bundle tree structore
121     QDir d(binPath + "/../lib/");
122     QFileInfoList files = d.entryInfoList(macFilters,
123                                           QDir::Files | QDir::NoDotAndDotDot);
124     for (QFileInfoList::const_iterator i = files.begin(); i != files.end(); ++i) {
125         result << i->fileName();
126     }
127 #else
128 #error "Function DictCore::availablePlugins() is not implemented on this platform"
129 #endif
130     return result;
131 }
132
133 void DictCore::setLoadedPlugins(const QStringList &loadedPlugins)
134 {
135     for (QHash <QString, QPluginLoader*>::iterator i = m_plugins.begin(); i != m_plugins.end(); ++i)
136     {
137         delete (*i)->instance();
138         delete *i;
139     }
140     m_plugins.clear();
141
142     for (QStringList::const_iterator i = loadedPlugins.begin(); i != loadedPlugins.end(); ++i)
143     {
144 #ifdef Q_WS_X11
145         QString pluginFilename = static_cast<QString>(QSTARDICT_PLUGINS_DIR) + "/" "lib" + *i + ".so";
146 #elif defined Q_WS_WIN
147         QString pluginFilename = static_cast<QString>(QSTARDICT_PLUGINS_DIR) + "/" + *i + "0.dll";
148 #elif defined Q_WS_MAC
149         // here we need to follow mac's bundle tree...
150         QString pluginFilename = QDir(QCoreApplication::applicationDirPath()+"/../lib/"+*i).absolutePath();
151 #else
152 #error "Function DictCore::setLoadedPlugins(const QStringList &loadedPlugins) is not available on this platform"
153 #endif
154         QPluginLoader *plugin = new QPluginLoader(pluginFilename);
155         if (! plugin->load())
156         {
157             qWarning() << plugin->errorString();
158             delete plugin;
159             continue;
160         }
161         m_plugins[*i] = plugin;
162     }
163 }
164
165 QList<DictCore::Dictionary> DictCore::availableDicts() const
166 {
167     QList<Dictionary> result;
168     for (QHash<QString, QPluginLoader*>::const_iterator i = m_plugins.begin(); i != m_plugins.end(); ++i)
169     {
170         DictPlugin *plugin = qobject_cast<DictPlugin*>((*i)->instance());
171         QStringList dicts = plugin->availableDicts();
172         for (QStringList::const_iterator j = dicts.begin(); j != dicts.end(); ++j)
173             result << Dictionary(i.key(), *j);
174     }
175     return result;
176 }
177
178 void DictCore::setLoadedDicts(const QList<Dictionary> &loadedDicts)
179 {
180     QHash<QString, QStringList> dicts;
181     for (QList<Dictionary>::const_iterator i = loadedDicts.begin(); i != loadedDicts.end(); ++i)
182         dicts[i->plugin()] << i->name();
183     for (QHash<QString, QStringList>::const_iterator i = dicts.begin(); i != dicts.end(); ++i)
184     {
185         if (! m_plugins.contains(i.key()))
186             continue;
187         DictPlugin *plugin = qobject_cast<DictPlugin*>(m_plugins[i.key()]->instance());
188         plugin->setLoadedDicts(*i);
189         dicts[i.key()] = plugin->loadedDicts();
190     }
191     m_loadedDicts.clear();
192     for (QList<Dictionary>::const_iterator i = loadedDicts.begin(); i != loadedDicts.end(); ++i)
193         if (dicts.contains(i->plugin()) && dicts[i->plugin()].contains(i->name()))
194             m_loadedDicts << *i;
195 }
196
197 void DictCore::saveSettings()
198 {
199     QSettings config;
200     config.setValue("DictCore/loadedPlugins", loadedPlugins());
201     QStringList rawDictsList;
202     for (QList<Dictionary>::const_iterator i = m_loadedDicts.begin(); i != m_loadedDicts.end(); ++i)
203         rawDictsList << i->plugin() << i->name();
204     config.setValue("DictCore/loadedDicts", rawDictsList);
205 }
206
207 void DictCore::loadSettings()
208 {
209     QSettings config;
210     setLoadedPlugins(config.value("DictCore/loadedPlugins", availablePlugins()).toStringList());
211     QStringList rawDictsList = config.value("DictCore/loadedDicts").toStringList();
212     if (rawDictsList.isEmpty())
213         setLoadedDicts(availableDicts());
214     else
215     {
216         QList<Dictionary> dicts;
217         for (QStringList::const_iterator i = rawDictsList.begin(); i != rawDictsList.end(); i += 2)
218             dicts << Dictionary(*i, *(i + 1));
219         setLoadedDicts(dicts);
220     }
221 }
222
223 void DictCore::reloadDicts()
224 {
225     QList<Dictionary> loaded;
226     for (QHash<QString, QPluginLoader*>::const_iterator i = m_plugins.begin(); i != m_plugins.end(); ++i)
227     {
228         DictPlugin *plugin = qobject_cast<DictPlugin*>((*i)->instance());
229         plugin->setLoadedDicts(plugin->loadedDicts());
230         QStringList loadedNames = plugin->loadedDicts();
231         for (QStringList::const_iterator j = loadedNames.begin(); j != loadedNames.end(); ++j)
232             loaded << Dictionary(i.key(), *j);
233     }
234     QList<Dictionary> oldLoaded = m_loadedDicts;
235     m_loadedDicts.clear();
236     for (QList<Dictionary>::iterator i = oldLoaded.begin(); i != oldLoaded.end(); ++i)
237         if (loaded.contains(*i))
238             m_loadedDicts << *i;
239 }
240
241 }
242
243 // vim: tabstop=4 softtabstop=4 shiftwidth=4 expandtab cindent textwidth=120 formatoptions=tc
244