sugarlabs/sugar-toolkit-gtk3
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

If we receive an application/octet-stream, check the mime type by looking at its extension.

Browse Source
This commit is contained in:
Tomeu 2007-03-30 14:32:07 +02:00
parent e179dbae14
commit 891624888f
1 changed files with 53 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
lib/src/GeckoDownload.cpp
View File

@ -3,11 +3,15 @@
#include <nsIFactory.h> #include <nsIFactory.h>
#include <nsIFile.h> #include <nsIFile.h>
#include <nsIFileURL.h> #include <nsIFileURL.h>
#include <nsServiceManagerUtils.h>
#include <nsIMIMEService.h>
#include "sugar-download-manager.h" #include "sugar-download-manager.h"
#include "GeckoDownload.h" #include "GeckoDownload.h"
#define APPLICATION_OCTET_STREAM "application/octet-stream"
class GeckoDownload : public nsITransfer class GeckoDownload : public nsITransfer
{ {
public: public:
@ -62,41 +66,63 @@ GeckoDownload::Init (nsIURI *aSource,
NS_ENSURE_SUCCESS(rv, NS_ERROR_FAILURE); NS_ENSURE_SUCCESS(rv, NS_ERROR_FAILURE);
file->GetNativePath (mTargetFileName); file->GetNativePath (mTargetFileName);
return NS_OK;
} }
NS_IMETHODIMP NS_IMETHODIMP
GeckoDownload::OnStateChange (nsIWebProgress *aWebProgress, GeckoDownload::OnStateChange(nsIWebProgress *aWebProgress,
nsIRequest *aRequest, nsIRequest *aRequest,
PRUint32 aStateFlags, PRUint32 aStateFlags,
nsresult aStatus) nsresult aStatus)
{ {
SugarDownloadManager *download_manager = sugar_get_download_manager (); SugarDownloadManager *download_manager = sugar_get_download_manager();
if (aStateFlags == STATE_START) { if(aStateFlags == STATE_START) {
nsCString url; nsCString url;
nsCString mimeType; nsCString mimeType;
mMIMEInfo->GetMIMEType (mimeType); mMIMEInfo->GetMIMEType(mimeType);
mSource->GetSpec (url); mSource->GetSpec(url);
sugar_download_manager_download_started (download_manager, /* If the file is application/octet-stream, look up a better mime type
url.get (), from the extension. */
mimeType.get (), if(mimeType.Equals(APPLICATION_OCTET_STREAM)) {
mTargetFileName.get ()); nsresult rv;
nsCOMPtr<nsIMIMEService> mimeService(
do_GetService("@mozilla.org/mime;1", &rv));
NS_ENSURE_SUCCESS(rv, rv);
} else if (aStateFlags == STATE_STOP) { const char *fileExt = strrchr(mTargetFileName.get(), '.');
if(fileExt) {
nsCString contentType;
if (NS_SUCCEEDED (aStatus)) { mimeService->GetTypeFromExtension(nsCString(fileExt),
sugar_download_manager_download_completed (download_manager, contentType);
mTargetFileName.get ()); if(!contentType.IsEmpty()) {
} else { mimeType = contentType;
sugar_download_manager_download_cancelled (download_manager, }
mTargetFileName.get ()); }
} }
}
return NS_OK; 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 NS_IMETHODIMP