bb-cascades-oauth/oauth/kqoauthauthreplyserver.cpp

/**
 * KQOAuth - An OAuth authentication library for Qt.
 *
 * Author: Johan Paul (johan.paul@d-pointer.com)
 *         http://www.d-pointer.com
 *
 *  This library is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU Lesser General Public
 *  License as published by the Free Software Foundation; either
 *  version 2.1 of the License, or (at your option) any later version.
 *
 *  KQOAuth is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *
 *  In addition, as a special exception, KQOauth provides you certain additional
 *  rights. These rights are described in the Nokia Qt LGPL Exception
 *  version 1.1, included in the file LGPL_EXCEPTION.txt in this package. 
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with KQOAuth.  If not, see <http://www.gnu.org/licenses/>
 */
#include <QTcpSocket>
#include <QStringList>
#include <QUrl>
#include <QFile>
#include <QDir>
#include <QFileInfoList>

#include "kqoauthauthreplyserver.h"
#include "kqoauthauthreplyserver_p.h"

KQOAuthAuthReplyServerPrivate::KQOAuthAuthReplyServerPrivate(KQOAuthAuthReplyServer *parent):
    q_ptr(parent)
{

}

KQOAuthAuthReplyServerPrivate::~KQOAuthAuthReplyServerPrivate()
{

}

void KQOAuthAuthReplyServerPrivate::onIncomingConnection() {
    Q_Q(KQOAuthAuthReplyServer);
    qDebug() << "Incoming Connection";
    socket = q->nextPendingConnection();
    connect(socket, SIGNAL(readyRead()),
            this, SLOT(onBytesReady()), Qt::UniqueConnection);
}

void KQOAuthAuthReplyServerPrivate::onBytesReady() {
    Q_Q(KQOAuthAuthReplyServer);
    qDebug() << "Socket peer host address: " << socket->peerAddress();
    QByteArray reply;
    QByteArray content;

	 QByteArray data = socket->readAll();
	 qDebug()<< "Query Data: " << data;
	 QMultiMap<QString, QString> queryParams = parseQueryParams(&data);
	 if(queryParams.size() == 0 && !handlingRedirect) { //assume theres a hash and do the redirect hack
		 handlingRedirect = true;
		 content.append("<HTML><head><script type=\"text/javascript\">var str='http://'+window.location.host + '?' + window.location.hash.substring(1); window.location=str;</script></head><h1>Account authorized, go ahead back to the tumblr app and start your experience!</h1></HTML>");
	 } else {
		 handlingRedirect = false;
		QFile file("app/native/assets/" + localFile);
		QString fileData;
		 if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {
			 qDebug() << "file worked";
			 QTextStream in(&file);
			 while (!in.atEnd()) {
				 fileData += in.readLine();
			 }
			 file.close();
		 }
		 if(fileData.isEmpty()) {
			 content.append("<HTML><h1>Account linked, go ahead back to the app and check the status!</h1></HTML>");
		 } else {
			 content.append(fileData);
		 }
	 }

    reply.append("HTTP/1.0 200 OK \r\n");
    reply.append("Content-Type: text/html; charset=\"utf-8\"\r\n");
    reply.append(QString("Content-Length: %1\r\n").arg(content.size()));
    reply.append("\r\n");
    reply.append(content);
    socket->write(reply);

    if(!handlingRedirect) {
    	socket->disconnectFromHost();
    	q->close();
    	emit q->verificationReceived(queryParams);
    }
}

QMultiMap<QString, QString> KQOAuthAuthReplyServerPrivate::parseQueryParams(QByteArray *data) {
    QString splitGetLine = QString(*data).split("\r\n").first();   // Retrieve the first line with query params.
    splitGetLine.remove("GET ");                                   // Clean the line from GET
    splitGetLine.remove("HTTP/1.1");                               // From HTTP
    splitGetLine.remove("\r\n");                                   // And from rest.
    splitGetLine.prepend("http://localhost");                      // Now, make it a URL

    QUrl getTokenUrl(splitGetLine);
    QList< QPair<QString, QString> > tokens = getTokenUrl.queryItems();  // Ask QUrl to do our work.

    QMultiMap<QString, QString> queryParams;
    QPair<QString, QString> tokenPair;
    foreach (tokenPair, tokens) {
        queryParams.insert(tokenPair.first.trimmed(), tokenPair.second.trimmed());
    }

    return queryParams;
}


KQOAuthAuthReplyServer::KQOAuthAuthReplyServer(QObject *parent) :
    QTcpServer(parent),
    d_ptr( new KQOAuthAuthReplyServerPrivate(this) )
{
    Q_D(KQOAuthAuthReplyServer);

    connect(this, SIGNAL(newConnection()),
            d, SLOT(onIncomingConnection()));
}

KQOAuthAuthReplyServer::~KQOAuthAuthReplyServer()
{
    delete d_ptr;
}


void KQOAuthAuthReplyServer::setSuccessHtmlFile(QString filePath) {
	Q_D(KQOAuthAuthReplyServer);
	d->localFile = filePath;
}
Update 2012-03-23 03:40:35 +00:00			`/**`
			`* KQOAuth - An OAuth authentication library for Qt.`
			`*`
			`* Author: Johan Paul (johan.paul@d-pointer.com)`
			`* http://www.d-pointer.com`
			`*`
Update to license on the files, and inclusion of LICENSE files 2012-10-12 19:36:01 +00:00			`* This library is free software; you can redistribute it and/or`
			`* modify it under the terms of the GNU Lesser General Public`
			`* License as published by the Free Software Foundation; either`
			`* version 2.1 of the License, or (at your option) any later version.`
Update 2012-03-23 03:40:35 +00:00			`*`
			`* KQOAuth is distributed in the hope that it will be useful,`
			`* but WITHOUT ANY WARRANTY; without even the implied warranty of`
			`* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the`
			`* GNU Lesser General Public License for more details.`
			`*`
Update to license on the files, and inclusion of LICENSE files 2012-10-12 19:36:01 +00:00			`* In addition, as a special exception, KQOauth provides you certain additional`
			`* rights. These rights are described in the Nokia Qt LGPL Exception`
			`* version 1.1, included in the file LGPL_EXCEPTION.txt in this package.`
			`*`
Update 2012-03-23 03:40:35 +00:00			`* You should have received a copy of the GNU Lesser General Public License`
Update to license on the files, and inclusion of LICENSE files 2012-10-12 19:36:01 +00:00			`* along with KQOAuth. If not, see <http://www.gnu.org/licenses/>`
Update 2012-03-23 03:40:35 +00:00			`*/`
			`#include <QTcpSocket>`
			`#include <QStringList>`
			`#include <QUrl>`
Oauth2 updates 2012-04-11 18:01:21 +00:00			`#include <QFile>`
Custom HTML success page fix 2012-04-19 06:48:41 +00:00			`#include <QDir>`
			`#include <QFileInfoList>`
Update 2012-03-23 03:40:35 +00:00
			`#include "kqoauthauthreplyserver.h"`
			`#include "kqoauthauthreplyserver_p.h"`

			`KQOAuthAuthReplyServerPrivate::KQOAuthAuthReplyServerPrivate(KQOAuthAuthReplyServer *parent):`
			`q_ptr(parent)`
			`{`

			`}`

			`KQOAuthAuthReplyServerPrivate::~KQOAuthAuthReplyServerPrivate()`
			`{`

			`}`

			`void KQOAuthAuthReplyServerPrivate::onIncomingConnection() {`
			`Q_Q(KQOAuthAuthReplyServer);`
Oauth2 updates 2012-04-11 18:01:21 +00:00			`qDebug() << "Incoming Connection";`
Update 2012-03-23 03:40:35 +00:00			`socket = q->nextPendingConnection();`
			`connect(socket, SIGNAL(readyRead()),`
			`this, SLOT(onBytesReady()), Qt::UniqueConnection);`
			`}`

			`void KQOAuthAuthReplyServerPrivate::onBytesReady() {`
			`Q_Q(KQOAuthAuthReplyServer);`
Oauth2 updates 2012-04-11 18:01:21 +00:00			`qDebug() << "Socket peer host address: " << socket->peerAddress();`
Update 2012-03-23 03:40:35 +00:00			`QByteArray reply;`
			`QByteArray content;`
Oauth2 updates 2012-04-11 18:01:21 +00:00
			`QByteArray data = socket->readAll();`
			`qDebug()<< "Query Data: " << data;`
			`QMultiMap<QString, QString> queryParams = parseQueryParams(&data);`
			`if(queryParams.size() == 0 && !handlingRedirect) { //assume theres a hash and do the redirect hack`
			`handlingRedirect = true;`
			`content.append("<HTML><head><script type=\"text/javascript\">var str='http://'+window.location.host + '?' + window.location.hash.substring(1); window.location=str;</script></head><h1>Account authorized, go ahead back to the tumblr app and start your experience!</h1></HTML>");`
			`} else {`
			`handlingRedirect = false;`
Custom HTML success page fix 2012-04-19 06:48:41 +00:00			`QFile file("app/native/assets/" + localFile);`
			`QString fileData;`
			`if (file.open(QIODevice::ReadOnly \| QIODevice::Text)) {`
			`qDebug() << "file worked";`
			`QTextStream in(&file);`
			`while (!in.atEnd()) {`
			`fileData += in.readLine();`
			`}`
			`file.close();`
			`}`
			`if(fileData.isEmpty()) {`
			`content.append("<HTML><h1>Account linked, go ahead back to the app and check the status!</h1></HTML>");`
			`} else {`
			`content.append(fileData);`
			`}`
Oauth2 updates 2012-04-11 18:01:21 +00:00			`}`
Update 2012-03-23 03:40:35 +00:00
			`reply.append("HTTP/1.0 200 OK \r\n");`
			`reply.append("Content-Type: text/html; charset=\"utf-8\"\r\n");`
			`reply.append(QString("Content-Length: %1\r\n").arg(content.size()));`
			`reply.append("\r\n");`
			`reply.append(content);`
			`socket->write(reply);`

Oauth2 updates 2012-04-11 18:01:21 +00:00			`if(!handlingRedirect) {`
			`socket->disconnectFromHost();`
			`q->close();`
			`emit q->verificationReceived(queryParams);`
			`}`
Update 2012-03-23 03:40:35 +00:00			`}`

			`QMultiMap<QString, QString> KQOAuthAuthReplyServerPrivate::parseQueryParams(QByteArray *data) {`
			`QString splitGetLine = QString(*data).split("\r\n").first(); // Retrieve the first line with query params.`
			`splitGetLine.remove("GET "); // Clean the line from GET`
			`splitGetLine.remove("HTTP/1.1"); // From HTTP`
			`splitGetLine.remove("\r\n"); // And from rest.`
			`splitGetLine.prepend("http://localhost"); // Now, make it a URL`

			`QUrl getTokenUrl(splitGetLine);`
			`QList< QPair<QString, QString> > tokens = getTokenUrl.queryItems(); // Ask QUrl to do our work.`

			`QMultiMap<QString, QString> queryParams;`
			`QPair<QString, QString> tokenPair;`
			`foreach (tokenPair, tokens) {`
			`queryParams.insert(tokenPair.first.trimmed(), tokenPair.second.trimmed());`
			`}`

			`return queryParams;`
			`}`



			`KQOAuthAuthReplyServer::KQOAuthAuthReplyServer(QObject *parent) :`
			`QTcpServer(parent),`
			`d_ptr( new KQOAuthAuthReplyServerPrivate(this) )`
			`{`
			`Q_D(KQOAuthAuthReplyServer);`

			`connect(this, SIGNAL(newConnection()),`
			`d, SLOT(onIncomingConnection()));`
			`}`

			`KQOAuthAuthReplyServer::~KQOAuthAuthReplyServer()`
			`{`
			`delete d_ptr;`
			`}`


			`void KQOAuthAuthReplyServer::setSuccessHtmlFile(QString filePath) {`
			`Q_D(KQOAuthAuthReplyServer);`
			`d->localFile = filePath;`
			`}`