2000-06-17 01:35:13 +04:00
|
|
|
/* -*- Mode: C++; tab-width: 3; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
|
|
|
*
|
|
|
|
* The contents of this file are subject to the Mozilla Public
|
|
|
|
* License Version 1.1 (the "License"); you may not use this file
|
|
|
|
* except in compliance with the License. You may obtain a copy of
|
|
|
|
* the License at http://www.mozilla.org/MPL/
|
|
|
|
*
|
|
|
|
* Software distributed under the License is distributed on an "AS
|
|
|
|
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
|
|
|
|
* implied. See the License for the specific language governing
|
|
|
|
* rights and limitations under the License.
|
|
|
|
*
|
|
|
|
* The Original Code is the Mozilla browser.
|
|
|
|
*
|
|
|
|
* The Initial Developer of the Original Code is Netscape
|
|
|
|
* Communications, Inc. Portions created by Netscape are
|
|
|
|
* Copyright (C) 1999, Mozilla. All Rights Reserved.
|
|
|
|
*
|
|
|
|
* Contributor(s):
|
|
|
|
* Scott MacGregor <mscott@netscape.com>
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef nsExternalHelperAppService_h__
|
|
|
|
#define nsExternalHelperAppService_h__
|
|
|
|
|
|
|
|
#include "nsIExternalHelperAppService.h"
|
2000-06-19 02:07:54 +04:00
|
|
|
#include "nsIExternalProtocolService.h"
|
2000-10-04 10:54:47 +04:00
|
|
|
#include "nsIURIContentListener.h"
|
|
|
|
#include "nsIWebProgressListener.h"
|
2001-05-04 01:24:36 +04:00
|
|
|
#include "nsIHelperAppLauncherDialog.h"
|
2000-06-19 02:07:54 +04:00
|
|
|
|
2000-07-07 05:48:05 +04:00
|
|
|
#include "nsIMIMEInfo.h"
|
2000-09-08 04:12:51 +04:00
|
|
|
#include "nsIMIMEService.h"
|
2000-06-17 01:35:13 +04:00
|
|
|
#include "nsIStreamListener.h"
|
|
|
|
#include "nsIFile.h"
|
2000-06-17 23:53:44 +04:00
|
|
|
#include "nsIFileStreams.h"
|
|
|
|
#include "nsIOutputStream.h"
|
2000-06-18 05:13:04 +04:00
|
|
|
#include "nsString.h"
|
2001-07-16 06:40:48 +04:00
|
|
|
#include "nsIInterfaceRequestor.h"
|
2001-09-06 01:28:38 +04:00
|
|
|
#include "nsIInterfaceRequestorUtils.h"
|
2001-07-16 06:40:48 +04:00
|
|
|
#include "nsILocalFile.h"
|
2003-01-18 05:15:14 +03:00
|
|
|
#include "nsIChannel.h"
|
2000-06-21 12:29:13 +04:00
|
|
|
|
|
|
|
#include "nsIRDFDataSource.h"
|
|
|
|
#include "nsIRDFResource.h"
|
2000-09-15 10:07:58 +04:00
|
|
|
#include "nsHashtable.h"
|
2000-06-17 01:35:13 +04:00
|
|
|
#include "nsCOMPtr.h"
|
2001-07-11 07:23:24 +04:00
|
|
|
#include "nsIObserver.h"
|
|
|
|
#include "nsISupportsArray.h"
|
2000-06-17 01:35:13 +04:00
|
|
|
|
2000-06-17 04:06:26 +04:00
|
|
|
class nsExternalAppHandler;
|
2000-06-21 12:29:13 +04:00
|
|
|
class nsIMIMEInfo;
|
|
|
|
class nsIRDFService;
|
2002-06-04 00:52:27 +04:00
|
|
|
class nsIDownload;
|
2000-06-17 04:06:26 +04:00
|
|
|
|
2000-09-08 04:12:51 +04:00
|
|
|
class nsExternalHelperAppService : public nsIExternalHelperAppService, public nsPIExternalAppLauncher,
|
2001-07-11 07:23:24 +04:00
|
|
|
public nsIExternalProtocolService, public nsIMIMEService, public nsIObserver
|
2000-06-17 01:35:13 +04:00
|
|
|
{
|
|
|
|
public:
|
|
|
|
NS_DECL_ISUPPORTS
|
|
|
|
NS_DECL_NSIEXTERNALHELPERAPPSERVICE
|
2000-06-18 03:04:53 +04:00
|
|
|
NS_DECL_NSPIEXTERNALAPPLAUNCHER
|
2000-06-19 02:07:54 +04:00
|
|
|
NS_DECL_NSIEXTERNALPROTOCOLSERVICE
|
2000-09-08 04:12:51 +04:00
|
|
|
NS_DECL_NSIMIMESERVICE
|
2001-07-11 07:23:24 +04:00
|
|
|
NS_DECL_NSIOBSERVER
|
2000-06-17 01:35:13 +04:00
|
|
|
|
|
|
|
nsExternalHelperAppService();
|
|
|
|
virtual ~nsExternalHelperAppService();
|
2000-09-08 04:12:51 +04:00
|
|
|
nsresult InitDataSource();
|
2000-06-17 01:35:13 +04:00
|
|
|
|
2000-06-21 12:29:13 +04:00
|
|
|
// CreateNewExternalHandler is implemented only by the base class...
|
2000-07-07 05:48:05 +04:00
|
|
|
// create an external app handler and binds it with a mime info object which represents
|
|
|
|
// how we want to dispose of this content
|
2000-06-18 05:13:04 +04:00
|
|
|
// aFileExtension --> the extension we need to append to our temp file INCLUDING the ".". i.e. .mp3
|
2000-07-11 01:21:10 +04:00
|
|
|
nsExternalAppHandler * CreateNewExternalHandler(nsIMIMEInfo * aMIMEInfo, const char * aFileExtension, nsISupports * aWindowContext);
|
2000-06-21 12:29:13 +04:00
|
|
|
|
2000-09-15 10:07:58 +04:00
|
|
|
// GetMIMEInfoForMimeTypeFromDS --> Given a content type, look up the user override information to
|
|
|
|
// see if we have a mime info object representing this content type. The user over ride information is contained
|
|
|
|
// in a in memory data source....
|
|
|
|
nsresult GetMIMEInfoForMimeTypeFromDS(const char * aContentType, nsIMIMEInfo ** aMIMEInfo);
|
2001-11-30 07:37:21 +03:00
|
|
|
|
|
|
|
// GetMIMEInfoForExtensionFromDS --> Given an extension, look up the user override information to
|
|
|
|
// see if we have a mime info object representing this extension. The user over ride information is contained
|
|
|
|
// in a in memory data source....
|
|
|
|
nsresult GetMIMEInfoForExtensionFromDS(const char * aFileExtension, nsIMIMEInfo ** aMIMEInfo);
|
2000-06-21 12:29:13 +04:00
|
|
|
|
2003-01-12 04:14:35 +03:00
|
|
|
// GetMIMEInfoForMimeTypeFromOS --> Given a content type, look up the system default information to
|
|
|
|
// see if we can create a mime info object for this content type.
|
|
|
|
virtual nsresult GetMIMEInfoForMimeTypeFromOS(const char * aContentType, nsIMIMEInfo ** aMIMEInfo);
|
|
|
|
|
|
|
|
// GetMIMEInfoForExtensionFromOS --> Given an extension, look up the system default information to
|
|
|
|
// see if we can create a mime info object for this extension.
|
|
|
|
virtual nsresult GetMIMEInfoForExtensionFromOS(const char * aFileExtension, nsIMIMEInfo ** aMIMEInfo);
|
|
|
|
|
2000-06-21 12:29:13 +04:00
|
|
|
// GetFileTokenForPath must be implemented by each platform.
|
|
|
|
// platformAppPath --> a platform specific path to an application that we got out of the
|
|
|
|
// rdf data source. This can be a mac file spec, a unix path or a windows path depending on the platform
|
|
|
|
// aFile --> an nsIFile representation of that platform application path.
|
|
|
|
virtual nsresult GetFileTokenForPath(const PRUnichar * platformAppPath, nsIFile ** aFile) = 0;
|
2003-01-12 04:14:35 +03:00
|
|
|
|
|
|
|
// helper routine used to test whether a given mime type is in our
|
|
|
|
// mimeTypes.rdf data source
|
|
|
|
PRBool MIMETypeIsInDataSource(const char * aContentType);
|
|
|
|
|
2000-06-17 01:35:13 +04:00
|
|
|
protected:
|
2000-06-21 12:29:13 +04:00
|
|
|
nsCOMPtr<nsIRDFDataSource> mOverRideDataSource;
|
|
|
|
|
|
|
|
nsCOMPtr<nsIRDFResource> kNC_Description;
|
|
|
|
nsCOMPtr<nsIRDFResource> kNC_Value;
|
|
|
|
nsCOMPtr<nsIRDFResource> kNC_FileExtensions;
|
|
|
|
nsCOMPtr<nsIRDFResource> kNC_Path;
|
2003-01-12 04:14:35 +03:00
|
|
|
nsCOMPtr<nsIRDFResource> kNC_UseSystemDefault;
|
2000-06-21 12:29:13 +04:00
|
|
|
nsCOMPtr<nsIRDFResource> kNC_SaveToDisk;
|
|
|
|
nsCOMPtr<nsIRDFResource> kNC_AlwaysAsk;
|
|
|
|
nsCOMPtr<nsIRDFResource> kNC_HandleInternal;
|
|
|
|
nsCOMPtr<nsIRDFResource> kNC_PrettyName;
|
|
|
|
|
2000-09-08 04:12:51 +04:00
|
|
|
PRBool mDataSourceInitialized;
|
|
|
|
|
2000-06-21 12:29:13 +04:00
|
|
|
// helper routines for digesting the data source and filling in a mime info object for a given
|
|
|
|
// content type inside that data source.
|
|
|
|
nsresult FillTopLevelProperties(const char * aContentType, nsIRDFResource * aContentTypeNodeResource,
|
|
|
|
nsIRDFService * aRDFService, nsIMIMEInfo * aMIMEInfo);
|
|
|
|
nsresult FillContentHandlerProperties(const char * aContentType, nsIRDFResource * aContentTypeNodeResource,
|
|
|
|
nsIRDFService * aRDFService, nsIMIMEInfo * aMIMEInfo);
|
|
|
|
|
|
|
|
// a small helper function which gets the target for a given source and property...QIs to a literal
|
|
|
|
// and returns a CONST ptr to the string value of that target
|
|
|
|
nsresult FillLiteralValueFromTarget(nsIRDFResource * aSource, nsIRDFResource * aProperty, const PRUnichar ** aLiteralValue);
|
2000-06-17 01:35:13 +04:00
|
|
|
|
2000-09-15 10:07:58 +04:00
|
|
|
// in addition to the in memory data source which stores the user over ride mime types, we also use a hash table
|
|
|
|
// for quick look ups of mime types...
|
|
|
|
nsHashtable *mMimeInfoCache; // used for fast access and multi index lookups
|
|
|
|
// used to add entries to the mime info cache
|
|
|
|
virtual nsresult AddMimeInfoToCache(nsIMIMEInfo * aMIMEInfo);
|
|
|
|
virtual nsresult AddDefaultMimeTypesToCache();
|
2001-05-04 01:24:36 +04:00
|
|
|
virtual nsresult GetMIMEInfoForMimeTypeFromExtras(const char * aContentType, nsIMIMEInfo ** aMIMEInfo );
|
2001-11-17 05:13:33 +03:00
|
|
|
virtual nsresult GetMIMEInfoForExtensionFromExtras(const char * aContentType, nsIMIMEInfo ** aMIMEInfo );
|
2000-09-15 10:07:58 +04:00
|
|
|
|
2001-07-11 07:23:24 +04:00
|
|
|
protected:
|
|
|
|
// functions related to the tempory file cleanup service provided by nsExternalHelperAppService
|
|
|
|
nsresult ExpungeTemporaryFiles();
|
|
|
|
nsCOMPtr<nsISupportsArray> mTemporaryFilesList;
|
2000-09-15 10:07:58 +04:00
|
|
|
};
|
|
|
|
|
|
|
|
// this is a small private struct used to help us initialize some
|
|
|
|
// default mime types.
|
|
|
|
struct nsDefaultMimeTypeEntry {
|
|
|
|
const char* mMimeType;
|
|
|
|
const char* mFileExtensions;
|
|
|
|
const char* mDescription;
|
|
|
|
PRUint32 mMactype;
|
|
|
|
PRUint32 mMacCreator;
|
2000-06-17 01:35:13 +04:00
|
|
|
};
|
|
|
|
|
2000-09-15 10:07:58 +04:00
|
|
|
|
2000-06-17 01:35:13 +04:00
|
|
|
// An external app handler is just a small little class that presents itself as
|
|
|
|
// a nsIStreamListener. It saves the incoming data into a temp file. The handler
|
|
|
|
// is bound to an application when it is created. When it receives an OnStopRequest
|
|
|
|
// it launches the application using the temp file it has stored the data into.
|
2000-06-17 06:34:44 +04:00
|
|
|
// we create a handler every time we have to process data using a helper app.
|
2000-06-17 01:35:13 +04:00
|
|
|
|
2000-06-17 23:53:44 +04:00
|
|
|
// we need to read the data out of the incoming stream into a buffer which we can then use
|
|
|
|
// to write the data into the output stream representing the temp file...
|
|
|
|
#define DATA_BUFFER_SIZE (4096*2)
|
|
|
|
|
2000-10-04 10:54:47 +04:00
|
|
|
class nsExternalAppHandler : public nsIStreamListener, public nsIHelperAppLauncher, public nsIURIContentListener,
|
2002-02-20 11:00:29 +03:00
|
|
|
public nsIInterfaceRequestor, public nsIObserver
|
2000-06-17 01:35:13 +04:00
|
|
|
{
|
|
|
|
public:
|
|
|
|
NS_DECL_ISUPPORTS
|
|
|
|
NS_DECL_NSISTREAMLISTENER
|
2001-04-10 10:01:08 +04:00
|
|
|
NS_DECL_NSIREQUESTOBSERVER
|
2000-07-07 05:48:05 +04:00
|
|
|
NS_DECL_NSIHELPERAPPLAUNCHER
|
2000-10-04 10:54:47 +04:00
|
|
|
NS_DECL_NSIURICONTENTLISTENER
|
|
|
|
NS_DECL_NSIINTERFACEREQUESTOR
|
2002-02-20 11:00:29 +03:00
|
|
|
NS_DECL_NSIOBSERVER
|
2000-06-17 01:35:13 +04:00
|
|
|
|
|
|
|
nsExternalAppHandler();
|
|
|
|
virtual ~nsExternalAppHandler();
|
|
|
|
|
2003-01-12 04:14:35 +03:00
|
|
|
virtual nsresult Init(nsIMIMEInfo * aMIMEInfo, const char * aFileExtension, nsISupports * aWindowContext, nsExternalHelperAppService *aHelperAppService);
|
2000-06-17 06:34:44 +04:00
|
|
|
|
2000-06-17 01:35:13 +04:00
|
|
|
protected:
|
|
|
|
nsCOMPtr<nsIFile> mTempFile;
|
2000-10-04 10:54:47 +04:00
|
|
|
nsCOMPtr<nsIURI> mSourceUrl;
|
2002-05-08 03:07:19 +04:00
|
|
|
nsString mTempFileExtension;
|
2000-07-07 05:48:05 +04:00
|
|
|
nsCOMPtr<nsIMIMEInfo> mMimeInfo;
|
2000-06-17 23:53:44 +04:00
|
|
|
nsCOMPtr<nsIOutputStream> mOutStream; // output stream to the temp file...
|
2000-07-11 01:21:10 +04:00
|
|
|
nsCOMPtr<nsISupports> mWindowContext;
|
2000-09-15 10:07:58 +04:00
|
|
|
// the following field is set if we were processing an http channel that had a content disposition header
|
|
|
|
// which specified the SUGGESTED file name we should present to the user in the save to disk dialog.
|
2002-05-08 03:07:19 +04:00
|
|
|
nsString mSuggestedFileName;
|
2000-06-17 01:35:13 +04:00
|
|
|
|
2000-07-07 05:48:05 +04:00
|
|
|
// the canceled flag is set if the user canceled the launching of this application before we finished
|
|
|
|
// saving the data to a temp file...
|
2002-06-04 00:52:27 +04:00
|
|
|
PRPackedBool mCanceled;
|
2000-07-07 05:48:05 +04:00
|
|
|
|
|
|
|
// have we received information from the user about how they want to dispose of this content...
|
2002-06-07 02:12:18 +04:00
|
|
|
PRPackedBool mReceivedDispositionInfo;
|
2002-06-04 00:52:27 +04:00
|
|
|
PRPackedBool mStopRequestIssued;
|
|
|
|
PRPackedBool mProgressListenerInitialized;
|
2002-09-04 02:00:22 +04:00
|
|
|
// This is set when handling something with "Content-Disposition: attachment"
|
|
|
|
PRPackedBool mHandlingAttachment;
|
2001-01-25 09:48:16 +03:00
|
|
|
PRInt64 mTimeDownloadStarted;
|
2002-02-20 11:00:29 +03:00
|
|
|
PRInt32 mContentLength;
|
|
|
|
PRInt32 mProgress; // Number of bytes received (for sending progress notifications).
|
2000-07-07 05:48:05 +04:00
|
|
|
|
|
|
|
// when we are told to save the temp file to disk (in a more permament location) before we are done
|
|
|
|
// writing the content to a temp file, then we need to remember the final destination until we are ready to
|
|
|
|
// use it.
|
|
|
|
nsCOMPtr<nsIFile> mFinalFileDestination;
|
|
|
|
|
2000-06-17 23:53:44 +04:00
|
|
|
char * mDataBuffer;
|
2000-07-07 05:48:05 +04:00
|
|
|
|
|
|
|
nsresult SetUpTempFile(nsIChannel * aChannel);
|
2000-10-04 10:54:47 +04:00
|
|
|
// when we download a helper app, we are going to retarget all load notifications into our own docloader
|
|
|
|
// and load group instead of using the window which initiated the load....RetargetLoadNotifications contains
|
|
|
|
// that information...
|
2001-02-21 23:38:08 +03:00
|
|
|
nsresult RetargetLoadNotifications(nsIRequest *request);
|
2002-06-04 00:52:27 +04:00
|
|
|
// if the user tells us how they want to dispose of the content and
|
|
|
|
// we still haven't finished downloading while they were deciding,
|
|
|
|
// then create a progress listener of some kind so they know
|
|
|
|
// what's going on...
|
|
|
|
nsresult CreateProgressListener();
|
2002-05-08 03:07:19 +04:00
|
|
|
nsresult PromptForSaveToFile(nsILocalFile ** aNewFile, const nsAFlatString &aDefaultFile, const nsAFlatString &aDefaultFileExt);
|
2000-10-04 10:54:47 +04:00
|
|
|
// if the passed in channel is an nsIHTTPChannel, we'll attempt to extract a suggested file name
|
2000-09-15 10:07:58 +04:00
|
|
|
// from the content disposition header...
|
|
|
|
void ExtractSuggestedFileNameFromChannel(nsIChannel * aChannel);
|
2000-11-01 11:06:19 +03:00
|
|
|
// after we're done prompting the user for any information, if the original channel had a refresh url associated
|
|
|
|
// with it (which might point to a "thank you for downloading" kind of page, then process that....It is safe
|
|
|
|
// to invoke this method multiple times. We'll clear mOriginalChannel after it's called and this ensures we won't
|
|
|
|
// call it again....
|
|
|
|
void ProcessAnyRefreshTags();
|
2001-01-25 09:48:16 +03:00
|
|
|
|
|
|
|
// an internal method used to actually move the temp file to the final destination
|
|
|
|
// once we done receiving data AND have showed the progress dialog.
|
|
|
|
nsresult MoveFile(nsIFile * aNewFileLocation);
|
|
|
|
// an internal method used to actually launch a helper app given the temp file
|
|
|
|
// once we are done receiving data AND have showed the progress dialog.
|
|
|
|
nsresult OpenWithApplication(nsIFile * aApplication);
|
2002-06-04 00:52:27 +04:00
|
|
|
|
2001-01-25 09:48:16 +03:00
|
|
|
// helper routine which peaks at the mime action specified by mMimeInfo
|
|
|
|
// and calls either MoveFile or OpenWithApplication
|
|
|
|
nsresult ExecuteDesiredAction();
|
2003-01-12 04:14:35 +03:00
|
|
|
// helper routine that searches a pref string for a given mime type
|
|
|
|
PRBool GetNeverAskFlagFromPref(const char * prefName, const char * aContentType);
|
2002-06-04 00:52:27 +04:00
|
|
|
|
|
|
|
// initialize an nsIDownload object for use as a progress object
|
|
|
|
nsresult InitializeDownload(nsIDownload*);
|
|
|
|
|
2002-04-24 04:09:52 +04:00
|
|
|
// helper routine to ensure mSuggestedFileName is "correct";
|
|
|
|
// the base class implementation ensures that mSuggestedFileName has
|
|
|
|
// mTempFileExtension as extension;
|
|
|
|
virtual void EnsureSuggestedFileName();
|
2002-05-17 04:31:53 +04:00
|
|
|
// utility function to send proper error notification to web progress listener
|
|
|
|
typedef enum { kReadError, kWriteError, kLaunchError } ErrorType;
|
|
|
|
void SendStatusChange(ErrorType type, nsresult aStatus, nsIRequest *aRequest, const nsAFlatString &path);
|
2000-10-04 10:54:47 +04:00
|
|
|
|
|
|
|
nsCOMPtr<nsISupports> mLoadCookie; // load cookie used by the uri loader when we fetch the url
|
|
|
|
nsCOMPtr<nsIWebProgressListener> mWebProgressListener;
|
2000-11-01 11:06:19 +03:00
|
|
|
nsCOMPtr<nsIChannel> mOriginalChannel; // in the case of a redirect, this will be the pre-redirect channel.
|
2001-05-04 01:24:36 +04:00
|
|
|
nsCOMPtr<nsIHelperAppLauncherDialog> mDialog;
|
2003-01-12 04:14:35 +03:00
|
|
|
nsExternalHelperAppService *mHelperAppService;
|
2000-06-17 01:35:13 +04:00
|
|
|
};
|
|
|
|
|
|
|
|
#endif // nsExternalHelperAppService_h__
|