GEEKING

RobloxStudio/RobloxNetworkReply.cpp

@@ -0,0 +1,114 @@
+/**
+ * RobloxFollowRedirectionNetworkRequest.cpp
+ * Copyright (c) 2013 ROBLOX Corp. All Rights Reserved.
+ */
+
+#include "stdafx.h"
+#include "RobloxNetworkReply.h"
+
+// Qt headers
+#include <QTime>
+#include <QCoreApplication>
+#include <QEventLoop>
+
+// Roblox headers
+#include "Rbx/debug.h"
+
+// Roblox Studio headers
+#include "RobloxNetworkAccessManager.h"
+
+RobloxNetworkReply::RobloxNetworkReply(QNetworkReply* pReply, bool followRedirections)
+	: m_pReply(pReply)
+	, m_childReply(NULL)
+	, m_followRedirections(followRedirections)
+{
+	// Set up the new network reply to follow redirection or signal if it's done
+	connect(pReply, SIGNAL(finished()), this, SLOT(handleRedirection()));
+}
+RobloxNetworkReply::~RobloxNetworkReply()
+{
+	RBXASSERT(m_pReply->isFinished()); // Catch weird cases where request was aborted
+	delete m_pReply;
+}
+void RobloxNetworkReply::onChildFinished()
+{
+	Q_EMIT finished();
+}
+void RobloxNetworkReply::handleRedirection()
+{
+	RBXASSERT(m_pReply);
+	
+	if (!m_pReply) // TODO: What should we do here?
+		return;
+
+	QString redirectUrl = m_pReply->attribute(QNetworkRequest::RedirectionTargetAttribute).toString();
+	if (!redirectUrl.isEmpty() 
+		&& m_followRedirections 
+		&& !m_redirectUrls.contains(redirectUrl)) // Prevent infinite loops
+		// TODO: Detect cycles
+	{
+		QNetworkRequest request(redirectUrl);
+		RobloxNetworkReply* networkReply = RobloxNetworkAccessManager::Instance().get(request, true);
+		networkReply->setParent(this);
+		networkReply->m_redirectUrls.insert(redirectUrl);
+		m_childReply = networkReply;
+		
+		// When it signals it is done, we need to trigger this network reply to signal (fire all the way up)
+		connect(networkReply, SIGNAL(finished()), this, SLOT(onChildFinished()));
+	}
+	else
+		onChildFinished(); // Trigger our final finished
+}
+
+QNetworkReply* RobloxNetworkReply::getLastNetworkReply() const
+{
+	if (m_childReply)
+	{
+		RBXASSERT(m_followRedirections);  // will only have a m_childReply if followRedirections is true
+		return m_childReply->getLastNetworkReply();	 // Go to the last child reply
+	}
+	else
+		return m_pReply;
+}
+
+/*
+*  Returns true if the reply finished in time.  False if it timed out.
+*/
+bool RobloxNetworkReply::waitForFinished(int timeOutTime, int waitIntervalTime) const
+{
+	QNetworkReply* finalNetworkReply = getLastNetworkReply();
+	
+	RBXASSERT(finalNetworkReply);
+
+	if (!finalNetworkReply) // TODO: what to do here?
+		return true;
+
+	QTime time;
+	time.start();
+	while(!finalNetworkReply->isFinished()) // wait until we get all of the response
+	{
+		QCoreApplication::processEvents(QEventLoop::AllEvents, waitIntervalTime); 
+		if (time.elapsed() >= timeOutTime)
+		{
+			return false;
+		}
+		// Check to see if there have been any redirects since, and if so, start waiting for them
+		finalNetworkReply = getLastNetworkReply();
+	}
+	return true;
+}
+
+QByteArray RobloxNetworkReply::readAll()
+{
+	return getLastNetworkReply()->readAll();	
+}
+
+QNetworkReply::NetworkError RobloxNetworkReply::error() const
+{
+	return getLastNetworkReply()->error();
+}
+
+int RobloxNetworkReply::statusCode() const
+{
+	return getLastNetworkReply()->attribute(QNetworkRequest::HttpStatusCodeAttribute).toInt();
+}