2 #include <QApplication>
3 #include <QNetworkRequest>
4 #include <QNetworkReply>
5 #include <QNetworkCookie>
11 #include <qjson/parser.h>
17 #include "googlereader.h"
19 void Feed::updateSubscription(Feed *feed) {
21 sortid = feed->sortid;
22 firstitemmsec = feed->firstitemmsec;
23 cat_id = feed->cat_id;
24 cat_label = feed->cat_label;
25 subscription_updated = true;
28 void Feed::fetch(bool cont) {
29 QNetworkRequest request;
30 QByteArray ba = "http://www.google.com/reader/api/0/stream/contents/";
31 ba.append(QUrl::toPercentEncoding(id));
32 QUrl url = QUrl::fromEncoded(ba);
34 if(continuation != "" && cont)
35 url.addEncodedQueryItem("c", continuation.toUtf8());
37 if(!cont && updated) {
38 /* Add 1 to the timestamp, otherwise we get the latest item
39 * again. Also the order has to be reversed for this to work. */
40 url.addEncodedQueryItem("ot", QByteArray::number(updated + 1));
41 url.addEncodedQueryItem("r", "o");
44 request.setRawHeader("Authorization", reader->getAuth());
46 reply = reader->getManager()->get(request);
47 connect(reply, SIGNAL(finished()), SLOT(fetchFinished()));
50 void Feed::fetchFinished() {
52 qDebug() << "Download of" << reply->url() << "failed:" << qPrintable(reply->errorString());
58 QVariantMap result = parser.parse(reply->readAll(), &ok).toMap();
60 continuation = result["continuation"].toString();
61 updated = result["updated"].toUInt();
63 foreach(QVariant l, result["items"].toList()) {
64 QVariantMap e = l.toMap();
65 Entry *entry = new Entry();;
66 QString content, summary;
68 entry->id = e["id"].toString();
69 entry->published = QDateTime::fromTime_t(e["published"].toUInt());
70 entry->author = e["author"].toString();
71 entry->source = (e["origin"].toMap())["streamId"].toString();
72 foreach(QVariant a, e["alternate"].toList()) {
73 QVariantMap alt = a.toMap();
74 if(alt["type"].toString() == "text/html")
75 entry->link = alt["href"].toString();
78 content = (e["content"].toMap())["content"].toString();
79 summary = (e["summary"].toMap())["content"].toString();
80 if(content != "") entry->content = content; else entry->content = summary;
82 if(e["isReadStateLocked"].toBool())
83 entry->flags |= ENTRY_FLAG_LOCKED | ENTRY_FLAG_READ;
86 p.mainFrame()->setHtml(e["title"].toString());
87 entry->title = p.mainFrame()->toPlainText();
89 foreach(QVariant c, e["categories"].toList()) {
90 QString cat = c.toString();
91 if(cat.endsWith("/state/com.google/read"))
92 entry->flags |= ENTRY_FLAG_READ;
93 else if(cat.endsWith("/state/com.google/starred"))
94 entry->flags |= ENTRY_FLAG_STARRED;
95 else if(cat.endsWith("/state/com.google/broadcast"))
96 entry->flags |= ENTRY_FLAG_SHARED;
103 lastUpdated = QDateTime::currentDateTime();
105 emit updateFeedComplete();
107 reply->deleteLater();
110 GoogleReader::GoogleReader() {
111 connect(&manager, SIGNAL(finished(QNetworkReply*)),
112 SLOT(downloadFinished(QNetworkReply*)));
115 updateSubscriptionsPending = false;
116 updateUnreadPending = false;
119 login_url.setUrl("https://www.google.com/accounts/ClientLogin");
120 subscriptions_url.setUrl("http://www.google.com/reader/api/0/subscription/list?output=json");
121 unread_url.setUrl("http://www.google.com/reader/api/0/unread-count?output=json");
122 edittag_url.setUrl("http://www.google.com/reader/api/0/edit-tag?client=-");
123 token_url.setUrl("http://www.google.com/reader/api/0/token");
124 markallread_url.setUrl("http://www.google.com/reader/api/0/mark-all-as-read?client=-");
126 /* Add the virtual 'Starred items' feed */
127 Feed *feed = new Feed(this);
128 feed->id = "user/-/state/com.google/starred";
129 feed->title = "Starred items";
131 feeds.insert(feed->id, feed);
132 connect(feed, SIGNAL(allReadChanged()), SIGNAL(allReadChanged()));
134 /* Add the virtual 'Shared items' feed */
135 feed = new Feed(this);
136 feed->id = "user/-/state/com.google/broadcast";
137 feed->title = "Shared items";
139 feeds.insert(feed->id, feed);
140 connect(feed, SIGNAL(allReadChanged()), SIGNAL(allReadChanged()));
143 void GoogleReader::downloadFinished(QNetworkReply *reply) {
144 QUrl url = reply->url();
146 /* TODO: Instead of comparing against the url, use the signal from the
147 * QNetworkReply... */
149 if (reply->error()) {
150 qDebug() << "Download of" << url << "failed:" << qPrintable(reply->errorString());
151 if(url == login_url) {
153 emit loginFailed("Incorrect username or password");
155 else if(url == edittag_url)
159 else if(url == login_url) {
160 QByteArray data = reply->readAll();
161 data.remove(0, data.indexOf("Auth=", 0) + 5);
162 data.remove(data.indexOf("\n", 0), 1024);
164 auth.append("GoogleLogin auth=");
167 qDebug() << "Auth:" << auth;
173 /* TODO: Replace this with a proper state machine */
174 if(updateSubscriptionsPending) {
175 updateSubscriptionsPending = false;
176 updateSubscriptions();
179 else if(url == token_url) {
180 token = reply->readAll();
181 qDebug() << "token:" << token;
183 else if(url == subscriptions_url) {
184 parseSubscriptions(reply->readAll());
186 /* TODO: Replace this with a proper state machine */
187 if(updateUnreadPending) {
188 updateUnreadPending = false;
192 else if(url == unread_url) {
193 parseUnread(reply->readAll());
195 else if(url == edittag_url) {
196 QByteArray data = reply->readAll();
197 //qDebug() << "Result:" << data;
199 else if(url == markallread_url) {
200 QByteArray data = reply->readAll();
201 //qDebug() << "Result:" << data;
204 reply->deleteLater();
207 void GoogleReader::parseSubscriptions(QByteArray data) {
208 QJson::Parser parser;
210 QVariantMap result = parser.parse(data, &ok).toMap();
212 /* Clear the subscription updated flag */
213 QHash<QString, Feed *>::iterator i;
214 for(i = feeds.begin(); i != feeds.end(); ++i)
215 i.value()->subscription_updated = false;
217 foreach(QVariant l, result["subscriptions"].toList()) {
218 QVariantMap subscription = l.toMap();
219 Feed *feed = new Feed(this);
222 feed->id = subscription["id"].toString();
223 feed->title = subscription["title"].toString();
224 feed->sortid = subscription["sortid"].toString();
225 feed->firstitemmsec = subscription["firstitemmsec"].toString();
227 foreach(QVariant c, subscription["categories"].toList()) {
228 QVariantMap cat = c.toMap();
229 feed->cat_id = cat["id"].toString();
230 feed->cat_label = cat["label"].toString();
233 existing_feed = feeds.value(feed->id);
235 existing_feed->updateSubscription(feed);
237 feed = existing_feed;
241 feed->subscription_updated = true;
242 feeds.insert(feed->id, feed);
243 connect(feed, SIGNAL(allReadChanged()), SIGNAL(allReadChanged()));
247 /* Delete feeds no longer subscribed to */
248 for(i = feeds.begin(); i != feeds.end(); ++i) {
249 if(i.value()->subscription_updated == false && i.value()->special == 0) {
250 printf("DELETED: %s\n", i.value()->title.toLatin1().data());
255 lastUpdated = QDateTime::currentDateTime();
256 emit updateSubscriptionsComplete();
259 void GoogleReader::parseUnread(QByteArray data) {
260 QJson::Parser parser;
262 QVariantMap result = parser.parse(data, &ok).toMap();
264 foreach(QVariant l, result["unreadcounts"].toList()) {
265 QVariantMap unread = l.toMap();
266 QString id = unread["id"].toString();
267 int count = unread["count"].toInt();
268 ulong newestitem = unread["newestitem"].toUInt();
270 Feed *f = feeds.value(id);
273 f->newestitem = newestitem;
275 /* Not a good idea if it doesn't happen sequentially. */
276 /* Pre-fetch feeds with unread items */
277 /* f->fetch(false); */
281 lastUpdated = QDateTime::currentDateTime();
282 emit updateUnreadComplete();
285 void GoogleReader::clientLogin() {
292 QNetworkRequest request;
293 request.setUrl(login_url);
295 buffer.open(QBuffer::ReadWrite | QBuffer::Truncate);
296 buffer.write("Email=");
297 buffer.write(QUrl::toPercentEncoding(login));
298 buffer.write("&Passwd=");
299 buffer.write(QUrl::toPercentEncoding(passwd));
300 buffer.write("&service=reader&source=grms&continue=http://www.google.com");
303 //qDebug() << buffer.readAll();
306 manager.post(request, &buffer);
309 void GoogleReader::getToken() {
310 QNetworkRequest request;
311 request.setRawHeader("Authorization", auth);
312 request.setUrl(token_url);
313 manager.get(request);
316 void GoogleReader::updateSubscriptions() {
317 QNetworkRequest request;
319 if(updateSubscriptionsPending)
323 updateSubscriptionsPending = true;
328 request.setRawHeader("Authorization", auth);
329 request.setUrl(subscriptions_url);
330 manager.get(request);
333 void GoogleReader::updateUnread() {
334 QNetworkRequest request;
336 if(updateUnreadPending)
340 updateUnreadPending = true;
345 request.setRawHeader("Authorization", auth);
346 request.setUrl(unread_url);
347 manager.get(request);
350 static bool compareFeedItems(const Feed *f1, const Feed *f2) {
351 if(f1->special || f2->special)
352 return f1->special > f2->special;
354 if(f1->cat_label == f2->cat_label)
355 return f1->title.toLower() < f2->title.toLower();
357 if(f1->cat_label.length() == 0)
360 if(f2->cat_label.length() == 0)
363 return f1->cat_label.toLower() < f2->cat_label.toLower();
366 QList<Feed *> GoogleReader::getFeeds() {
367 QList<Feed *> list = feeds.values();
368 qSort(list.begin(), list.end(), compareFeedItems);
372 void Feed::addEntry(Entry *entry) {
373 entries.insert(entry->id, entry);
376 void Feed::delEntry(Entry *entry) {
377 entries.remove(entry->id);
380 void Feed::updateUnread(int i) {
381 bool allRead = (unread == 0);
384 if(unread <= 0) unread = 0;
386 if(allRead != (unread == 0))
387 emit allReadChanged();
390 void Feed::markRead() {
392 /* Mark all the remaining items read */
394 QNetworkRequest request;
395 request.setRawHeader("Authorization", reader->getAuth());
396 request.setUrl(reader->markallread_url);
398 buffer.open(QBuffer::ReadWrite | QBuffer::Truncate);
400 buffer.write(QUrl::toPercentEncoding(id));
401 //buffer.write("&ts=");
402 //buffer.write(QByteArray::number(oldest));
404 buffer.write(QUrl::toPercentEncoding(reader->token));
407 //qDebug() << buffer.readAll();
410 reader->manager.post(request, &buffer);
414 /* Go over all the entries and mark them read */
415 QHash<QString, Entry *>::iterator i;
416 for(i = entries.begin(); i != entries.end(); ++i)
417 i.value()->flags |= ENTRY_FLAG_READ | ENTRY_FLAG_LOCKED;
420 emit allReadChanged();
423 static bool compareEntryItems(const Entry *e1, const Entry *e2) {
424 return e1->published > e2->published;
427 QList<Entry *> Feed::getEntries() {
428 QList<Entry *> list = entries.values();
429 qSort(list.begin(), list.end(), compareEntryItems);
433 /* TODO: Remove the duplicate code in changing stated */
435 void Entry::markRead(bool mark_read) {
436 /* Check if the read flag differs from the requested state */
437 if(((flags & ENTRY_FLAG_READ) != 0) == mark_read)
440 /* Cannot mark an item unread if it's locked */
441 if((flags & ENTRY_FLAG_LOCKED) && !mark_read)
444 QNetworkRequest request;
445 request.setRawHeader("Authorization", feed->reader->getAuth());
446 request.setUrl(feed->reader->edittag_url);
448 postread.open(QBuffer::ReadWrite | QBuffer::Truncate);
449 postread.write("i=");
450 postread.write(QUrl::toPercentEncoding(id));
452 postread.write("&a=");
454 postread.write("&r=");
455 postread.write(QUrl::toPercentEncoding("user/-/state/com.google/read"));
456 postread.write("&ac=edit-tags&T=");
457 postread.write(QUrl::toPercentEncoding(feed->reader->token));
459 feed->reader->manager.post(request, &postread);
461 feed->updateUnread(mark_read ? -1 : 1);
464 flags |= ENTRY_FLAG_READ;
466 flags &= ~ENTRY_FLAG_READ;
469 void Entry::markStar(bool mark_star) {
470 /* Check if the starred flag differs from the requested state */
471 if(((flags & ENTRY_FLAG_STARRED) != 0) == mark_star)
474 QNetworkRequest request;
475 request.setRawHeader("Authorization", feed->reader->getAuth());
476 request.setUrl(feed->reader->edittag_url);
478 poststar.open(QBuffer::ReadWrite | QBuffer::Truncate);
479 poststar.write("i=");
480 poststar.write(QUrl::toPercentEncoding(id));
482 poststar.write("&a=");
484 poststar.write("&r=");
485 poststar.write(QUrl::toPercentEncoding("user/-/state/com.google/starred"));
486 poststar.write("&ac=edit-tags&T=");
487 poststar.write(QUrl::toPercentEncoding(feed->reader->token));
489 feed->reader->manager.post(request, &poststar);
491 Feed *starred = feed->reader->feeds.value("user/-/state/com.google/starred");
494 starred->addEntry(this);
495 flags |= ENTRY_FLAG_STARRED;
498 starred->delEntry(this);
499 flags &= ~ENTRY_FLAG_STARRED;
503 void Entry::markShared(bool mark_shared) {
504 /* Check if the shared flag differs from the requested state */
505 if(((flags & ENTRY_FLAG_SHARED) != 0) == mark_shared)
508 QNetworkRequest request;
509 request.setRawHeader("Authorization", feed->reader->getAuth());
510 request.setUrl(feed->reader->edittag_url);
512 postshared.open(QBuffer::ReadWrite | QBuffer::Truncate);
513 postshared.write("i=");
514 postshared.write(QUrl::toPercentEncoding(id));
516 postshared.write("&a=");
518 postshared.write("&r=");
519 postshared.write(QUrl::toPercentEncoding("user/-/state/com.google/broadcast"));
520 postshared.write("&ac=edit-tags&T=");
521 postshared.write(QUrl::toPercentEncoding(feed->reader->token));
523 feed->reader->manager.post(request, &postshared);
525 Feed *shared = feed->reader->feeds.value("user/-/state/com.google/broadcast");
528 shared->addEntry(this);
529 flags |= ENTRY_FLAG_SHARED;
532 shared->delEntry(this);
533 flags &= ~ENTRY_FLAG_SHARED;