232 lines
5.2 KiB
C++
232 lines
5.2 KiB
C++
#include "config.h"
|
|
|
|
#include <nsIFactory.h>
|
|
|
|
#include "sugar-download-manager.h"
|
|
|
|
#include "GeckoDownload.h"
|
|
|
|
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;
|
|
aTarget->GetPath (mTargetFileName);
|
|
mMIMEInfo = aMIMEInfo;
|
|
mTempFile = aTempFile;
|
|
// mCancelable = aCancelable; Just a reminder for when we implement cancelling downloads.
|
|
|
|
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);
|
|
|
|
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;
|
|
}
|