From 5343752c353a5abbaaef2a342614b32b7f0e0e8c Mon Sep 17 00:00:00 2001
From: Marco Pesenti Gritti <mpg@redhat.com>
Date: Sat, 07 Apr 2007 11:17:36 +0000
Subject: Make browser a modules inside sugar library

---
(limited to 'browser/GeckoDownload.cpp')

diff --git a/browser/GeckoDownload.cpp b/browser/GeckoDownload.cpp
new file mode 100644
index 0000000..6f099bd
--- /dev/null
+++ b/browser/GeckoDownload.cpp
@@ -0,0 +1,268 @@
+#include "config.h"
+
+#include <nsIFactory.h>
+#include <nsIFile.h>
+#include <nsIFileURL.h>
+#include <nsServiceManagerUtils.h>
+#include <nsIMIMEService.h>
+
+#include "sugar-download-manager.h"
+
+#include "GeckoDownload.h"
+
+#define APPLICATION_OCTET_STREAM "application/octet-stream"
+
+class GeckoDownload : public nsITransfer
+{
+public:
+    GeckoDownload();
+	virtual ~GeckoDownload();
+
+	NS_DECL_ISUPPORTS
+	NS_DECL_NSIWEBPROGRESSLISTENER
+	NS_DECL_NSIWEBPROGRESSLISTENER2
+	NS_DECL_NSITRANSFER
+
+protected:
+	nsIURI			*mSource;
+	nsCString		mTargetFileName;
+	nsIMIMEInfo		*mMIMEInfo;
+	nsILocalFile	*mTempFile;
+};
+
+GeckoDownload::GeckoDownload ()
+{
+}
+
+GeckoDownload::~GeckoDownload ()
+{
+}
+
+NS_IMPL_ISUPPORTS3 (GeckoDownload,
+					nsIWebProgressListener,
+					nsIWebProgressListener2,
+					nsITransfer)
+
+NS_IMETHODIMP
+GeckoDownload::Init (nsIURI *aSource,
+					 nsIURI *aTarget,
+					 const nsAString &aDisplayName,
+					 nsIMIMEInfo *aMIMEInfo,
+					 PRTime aStartTime,
+					 nsILocalFile *aTempFile,
+					 nsICancelable *aCancelable)
+{
+    mSource = aSource;
+    mMIMEInfo = aMIMEInfo;
+    mTempFile = aTempFile;
+
+    nsresult rv;
+
+    nsCOMPtr<nsIFileURL> fileURL = do_QueryInterface(aTarget);
+    NS_ENSURE_TRUE(fileURL, NS_ERROR_FAILURE);
+
+    nsCOMPtr<nsIFile> file;
+    rv = fileURL->GetFile(getter_AddRefs(file));
+    NS_ENSURE_SUCCESS(rv, NS_ERROR_FAILURE);
+
+    file->GetNativePath (mTargetFileName);
+
+    return NS_OK; 
+}
+
+NS_IMETHODIMP 
+GeckoDownload::OnStateChange(nsIWebProgress *aWebProgress,
+	                          nsIRequest *aRequest,
+                              PRUint32 aStateFlags,
+                              nsresult aStatus)
+{
+    SugarDownloadManager *download_manager = sugar_get_download_manager();
+
+    if(aStateFlags == STATE_START) {
+
+        nsCString url;
+        nsCString mimeType;
+
+        mMIMEInfo->GetMIMEType(mimeType);
+        mSource->GetSpec(url);
+
+#ifdef HAVE_GECKO_1_9
+        /* If the file is application/octet-stream, look up a better mime type
+           from the extension. */
+        if(mimeType.Equals(APPLICATION_OCTET_STREAM)) {
+            nsresult rv;
+            nsCOMPtr<nsIMIMEService> mimeService(
+                do_GetService("@mozilla.org/mime;1", &rv));
+            NS_ENSURE_SUCCESS(rv, rv);
+            
+            const char *fileExt = strrchr(mTargetFileName.get(), '.');
+            if(fileExt) {
+                nsCString contentType;
+
+                mimeService->GetTypeFromExtension(nsCString(fileExt),
+                                                   contentType);
+                if(!contentType.IsEmpty()) {
+                    mimeType = contentType;
+                }
+            }
+        }
+#endif
+
+        sugar_download_manager_download_started(download_manager,
+                                                 url.get(),
+                                                 mimeType.get(),
+                                                 mTargetFileName.get());
+
+    } else if(aStateFlags == STATE_STOP) {
+        
+        if(NS_SUCCEEDED(aStatus)) {
+            sugar_download_manager_download_completed(download_manager,
+                                                       mTargetFileName.get());
+        } else {
+            sugar_download_manager_download_cancelled(download_manager,
+                                                       mTargetFileName.get());
+        }
+    }
+
+    return NS_OK; 
+}
+
+NS_IMETHODIMP
+GeckoDownload::OnProgressChange (nsIWebProgress *aWebProgress,
+								 nsIRequest *aRequest,
+								 PRInt32 aCurSelfProgress,
+								 PRInt32 aMaxSelfProgress,
+								 PRInt32 aCurTotalProgress,
+								 PRInt32 aMaxTotalProgress)
+{
+	return OnProgressChange64 (aWebProgress,
+							   aRequest,
+							   aCurSelfProgress,
+							   aMaxSelfProgress,
+							   aCurTotalProgress,
+							   aMaxTotalProgress);
+}
+
+NS_IMETHODIMP
+GeckoDownload::OnProgressChange64 (nsIWebProgress *aWebProgress,
+								   nsIRequest *aRequest,
+								   PRInt64 aCurSelfProgress,
+								   PRInt64 aMaxSelfProgress,
+								   PRInt64 aCurTotalProgress,
+								   PRInt64 aMaxTotalProgress)
+{	
+	SugarDownloadManager *download_manager = sugar_get_download_manager ();
+	PRInt32 percentComplete =
+		(PRInt32)(((float)aCurSelfProgress / (float)aMaxSelfProgress) * 100.0);
+
+	sugar_download_manager_update_progress (download_manager,
+											mTargetFileName.get (),
+											percentComplete);
+
+	return NS_OK;
+}
+
+NS_IMETHODIMP
+GeckoDownload::OnLocationChange (nsIWebProgress *aWebProgress,
+								 nsIRequest *aRequest,
+								 nsIURI *location)
+{
+	return NS_OK;
+}
+
+NS_IMETHODIMP 
+GeckoDownload::OnStatusChange (nsIWebProgress *aWebProgress,
+							   nsIRequest *aRequest,
+							   nsresult aStatus, 
+							   const PRUnichar *aMessage)
+{
+	return NS_OK;
+}
+
+NS_IMETHODIMP 
+GeckoDownload::OnSecurityChange (nsIWebProgress *aWebProgress,
+								 nsIRequest *aRequest,
+								 PRUint32 state)
+{
+	return NS_OK;
+}
+
+#ifdef HAVE_GECKO_1_9
+
+NS_IMETHODIMP
+GeckoDownload::OnRefreshAttempted (nsIWebProgress *aWebProgress,
+                                   nsIURI *aRefreshURI,
+                                   PRInt32 aMillis,
+                                   PRBool aSameURI,
+                                   PRBool *_retval)
+{
+	return NS_OK;
+}
+
+#endif
+
+//*****************************************************************************
+// GeckoDownloadFactory
+//*****************************************************************************
+
+class GeckoDownloadFactory : public nsIFactory {
+public:
+    NS_DECL_ISUPPORTS
+    NS_DECL_NSIFACTORY
+
+    GeckoDownloadFactory();
+    virtual ~GeckoDownloadFactory();
+};
+
+//*****************************************************************************
+
+NS_IMPL_ISUPPORTS1(GeckoDownloadFactory, nsIFactory)
+
+GeckoDownloadFactory::GeckoDownloadFactory() {
+}
+
+GeckoDownloadFactory::~GeckoDownloadFactory() {
+}
+
+NS_IMETHODIMP
+GeckoDownloadFactory::CreateInstance(nsISupports *aOuter, const nsIID & aIID, void **aResult)
+{
+    NS_ENSURE_ARG_POINTER(aResult);
+
+    *aResult = NULL;
+    GeckoDownload *inst = new GeckoDownload;
+    if (!inst)
+        return NS_ERROR_OUT_OF_MEMORY;
+
+    nsresult rv = inst->QueryInterface(aIID, aResult);
+    if (rv != NS_OK) {
+        // We didn't get the right interface, so clean up
+        delete inst;
+    }
+
+    return rv;
+}
+
+NS_IMETHODIMP
+GeckoDownloadFactory::LockFactory(PRBool lock)
+{
+    return NS_OK;
+}
+
+//*****************************************************************************
+
+nsresult
+NS_NewGeckoDownloadFactory(nsIFactory** aFactory)
+{
+    NS_ENSURE_ARG_POINTER(aFactory);
+    *aFactory = nsnull;
+
+    GeckoDownloadFactory *result = new GeckoDownloadFactory;
+    if (!result)
+        return NS_ERROR_OUT_OF_MEMORY;
+
+    NS_ADDREF(result);
+    *aFactory = result;
+
+    return NS_OK;
+}
--
cgit v0.9.1