]> git.sur5r.net Git - minitube/blob - src/ytuser.cpp
Imported Upstream version 2.1.3
[minitube] / src / ytuser.cpp
1 /* $BEGIN_LICENSE
2
3 This file is part of Minitube.
4 Copyright 2009, Flavio Tordini <flavio.tordini@gmail.com>
5
6 Minitube is free software: you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation, either version 3 of the License, or
9 (at your option) any later version.
10
11 Minitube is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 GNU General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with Minitube.  If not, see <http://www.gnu.org/licenses/>.
18
19 $END_LICENSE */
20
21 #include "ytuser.h"
22 #include "networkaccess.h"
23 #include "database.h"
24 #include <QtSql>
25
26 namespace The {
27 NetworkAccess* http();
28 }
29
30 YTUser::YTUser(QString userId, QObject *parent) : QObject(parent),
31     id(0),
32     userId(userId),
33     loadingThumbnail(false),
34     notifyCount(0),
35     checked(0),
36     watched(0),
37     loaded(0),
38     loading(false) { }
39
40 QHash<QString, YTUser*> YTUser::cache;
41
42 YTUser* YTUser::forId(QString userId) {
43     if (userId.isEmpty()) return 0;
44
45     if (cache.contains(userId))
46         return cache.value(userId);
47
48     QSqlDatabase db = Database::instance().getConnection();
49     QSqlQuery query(db);
50     query.prepare("select id,name,description,thumb_url,notify_count,watched,checked,loaded "
51                   "from subscriptions where user_id=?");
52     query.bindValue(0, userId);
53     bool success = query.exec();
54     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
55
56     YTUser* user = 0;
57     if (query.next()) {
58         user = new YTUser(userId);
59         user->id = query.value(0).toInt();
60         user->displayName = query.value(1).toString();
61         user->description = query.value(2).toString();
62         user->thumbnailUrl = query.value(3).toString();
63         user->notifyCount = query.value(4).toInt();
64         user->watched = query.value(5).toUInt();
65         user->checked = query.value(6).toUInt();
66         user->loaded = query.value(7).toUInt();
67         user->thumbnail = QPixmap(user->getThumbnailLocation());
68         user->maybeLoadfromAPI();
69         cache.insert(userId, user);
70     }
71
72     return user;
73 }
74
75 void YTUser::maybeLoadfromAPI() {
76     if (loading) return;
77     if (userId.isEmpty()) return;
78
79     uint now = QDateTime::currentDateTime().toTime_t();
80     static const int refreshInterval = 60 * 60 * 24 * 10;
81     if (loaded > now - refreshInterval) return;
82
83     loading = true;
84
85     QUrl url("http://gdata.youtube.com/feeds/api/users/" + userId);
86     url.addQueryItem("v", "2");
87     QObject *reply = The::http()->get(url);
88     connect(reply, SIGNAL(data(QByteArray)), SLOT(parseResponse(QByteArray)));
89     connect(reply, SIGNAL(error(QNetworkReply*)), SLOT(requestError(QNetworkReply*)));
90 }
91
92 void YTUser::parseResponse(QByteArray bytes) {
93     QXmlStreamReader xml(bytes);
94     xml.readNextStartElement();
95     if (xml.name() == QLatin1String("entry"))
96         while(xml.readNextStartElement()) {
97             const QStringRef n = xml.name();
98             if (n == QLatin1String("summary"))
99                 description = xml.readElementText().simplified();
100             else if (n == QLatin1String("title"))
101                 displayName = xml.readElementText();
102             else if (n == QLatin1String("thumbnail")) {
103                 thumbnailUrl = xml.attributes().value("url").toString();
104                 xml.skipCurrentElement();
105             } else if (n == QLatin1String("username"))
106                 userName = xml.readElementText();
107             else xml.skipCurrentElement();
108         }
109
110     if (xml.hasError()) {
111         emit error(xml.errorString());
112         qWarning() << xml.errorString();
113     }
114
115     emit infoLoaded();
116     storeInfo();
117     loading = false;
118 }
119
120 void YTUser::loadThumbnail() {
121     if (loadingThumbnail) return;
122     if (thumbnailUrl.isEmpty()) return;
123     loadingThumbnail = true;
124
125 #ifdef Q_WS_WIN
126     thumbnailUrl.replace(QLatin1String("https://"), QLatin1String("http://"));
127 #endif
128
129     QUrl url(thumbnailUrl);
130     QObject *reply = The::http()->get(url);
131     connect(reply, SIGNAL(data(QByteArray)), SLOT(storeThumbnail(QByteArray)));
132     connect(reply, SIGNAL(error(QNetworkReply*)), SLOT(requestError(QNetworkReply*)));
133 }
134
135 const QString & YTUser::getThumbnailDir() {
136     static const QString thumbDir = QDesktopServices::storageLocation(
137                 QDesktopServices::DataLocation) + "/channels/";
138     return thumbDir;
139 }
140
141 QString YTUser::getThumbnailLocation() {
142     return getThumbnailDir() + userId;
143 }
144
145 void YTUser::storeThumbnail(QByteArray bytes) {
146     thumbnail.loadFromData(bytes);
147     static const int maxWidth = 88;
148
149     QDir dir;
150     dir.mkpath(getThumbnailDir());
151
152     if (thumbnail.width() > maxWidth) {
153         thumbnail = thumbnail.scaledToWidth(maxWidth, Qt::SmoothTransformation);
154         thumbnail.save(getThumbnailLocation(), "JPG");
155     } else {
156         QFile file(getThumbnailLocation());
157         if (!file.open(QIODevice::WriteOnly))
158             qWarning() << "Error opening file for writing" << file.fileName();
159         QDataStream stream(&file);
160         stream.writeRawData(bytes.constData(), bytes.size());
161     }
162
163     emit thumbnailLoaded();
164     loadingThumbnail = false;
165 }
166
167 void YTUser::requestError(QNetworkReply *reply) {
168     emit error(reply->errorString());
169     qWarning() << reply->errorString();
170     loading = false;
171     loadingThumbnail = false;
172 }
173
174 void YTUser::storeInfo() {
175     if (userId.isEmpty()) return;
176     QSqlDatabase db = Database::instance().getConnection();
177     QSqlQuery query(db);
178     query.prepare("update subscriptions set "
179                   "user_name=?, name=?, description=?, thumb_url=?, loaded=? "
180                   "where user_id=?");
181     qDebug() << userName;
182     query.bindValue(0, userName);
183     query.bindValue(1, displayName);
184     query.bindValue(2, description);
185     query.bindValue(3, thumbnailUrl);
186     query.bindValue(4, QDateTime::currentDateTime().toTime_t());
187     query.bindValue(5, userId);
188     bool success = query.exec();
189     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
190
191     loadThumbnail();
192 }
193
194 void YTUser::subscribe(QString userId) {
195     if (userId.isEmpty()) return;
196
197     uint now = QDateTime::currentDateTime().toTime_t();
198
199     QSqlDatabase db = Database::instance().getConnection();
200     QSqlQuery query(db);
201     query.prepare("insert into subscriptions "
202                   "(user_id,added,watched,checked,views,notify_count)"
203                   " values (?,?,?,0,0,0)");
204     query.bindValue(0, userId);
205     query.bindValue(1, now);
206     query.bindValue(2, now);
207     bool success = query.exec();
208     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
209
210     // This will call maybeLoadFromApi
211     YTUser::forId(userId);
212 }
213
214 void YTUser::unsubscribe(QString userId) {
215     if (userId.isEmpty()) return;
216     QSqlDatabase db = Database::instance().getConnection();
217     QSqlQuery query(db);
218     query.prepare("delete from subscriptions where user_id=?");
219     query.bindValue(0, userId);
220     bool success = query.exec();
221     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
222
223     query = QSqlQuery(db);
224     query.prepare("delete from subscriptions_videos where user_id=?");
225     query.bindValue(0, userId);
226     success = query.exec();
227     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
228
229     YTUser *user = cache.take(userId);
230     if (user) user->deleteLater();
231 }
232
233 bool YTUser::isSubscribed(QString userId) {
234     if (!Database::exists()) return false;
235     if (userId.isEmpty()) return false;
236     QSqlDatabase db = Database::instance().getConnection();
237     QSqlQuery query(db);
238     query.prepare("select count(*) from subscriptions where user_id=?");
239     query.bindValue(0, userId);
240     bool success = query.exec();
241     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
242     if (query.next())
243         return query.value(0).toInt() > 0;
244     return false;
245 }
246
247 void YTUser::updateChecked() {
248     if (userId.isEmpty()) return;
249
250     uint now = QDateTime::currentDateTime().toTime_t();
251     checked = now;
252
253     QSqlDatabase db = Database::instance().getConnection();
254     QSqlQuery query(db);
255     query.prepare("update subscriptions set checked=? where user_id=?");
256     query.bindValue(0, QDateTime::currentDateTime().toTime_t());
257     query.bindValue(1, userId);
258     bool success = query.exec();
259     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
260 }
261
262 void YTUser::updateWatched() {
263     if (userId.isEmpty()) return;
264
265     uint now = QDateTime::currentDateTime().toTime_t();
266     watched = now;
267     notifyCount = 0;
268
269     QSqlDatabase db = Database::instance().getConnection();
270     QSqlQuery query(db);
271     query.prepare("update subscriptions set watched=?, notify_count=0, views=views+1 where user_id=?");
272     query.bindValue(0, now);
273     query.bindValue(1, userId);
274     bool success = query.exec();
275     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
276 }
277
278 void YTUser::storeNotifyCount(int count) {
279     notifyCount = count;
280
281     QSqlDatabase db = Database::instance().getConnection();
282     QSqlQuery query(db);
283     query.prepare("update subscriptions set notify_count=? where user_id=?");
284     query.bindValue(0, count);
285     query.bindValue(1, userId);
286     bool success = query.exec();
287     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
288 }
289
290 bool YTUser::updateNotifyCount() {
291     QSqlDatabase db = Database::instance().getConnection();
292     QSqlQuery query(db);
293     query.prepare("select count(*) from subscriptions_videos "
294                   "where channel_id=? and added>? and published>? and watched=0");
295     query.bindValue(0, id);
296     query.bindValue(1, watched);
297     query.bindValue(2, watched);
298     bool success = query.exec();
299     if (!success) qWarning() << query.lastQuery() << query.lastError().text();
300     if (!query.next()) {
301         qWarning() << __PRETTY_FUNCTION__ << "Count failed";
302         return false;
303     }
304     int count = query.value(0).toInt();
305     storeNotifyCount(count);
306     return count != notifyCount;
307 }