uriloader/exthandler/win/nsOSHelperAppService.h

   1 /* -*- Mode: C++; tab-width: 3; indent-tabs-mode: nil; c-basic-offset: 2 -*-
   2  * ***** BEGIN LICENSE BLOCK *****
   3  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   4  *
   5  * The contents of this file are subject to the Mozilla Public License Version
   6  * 1.1 (the "License"); you may not use this file except in compliance with
   7  * the License. You may obtain a copy of the License at
   8  * http://www.mozilla.org/MPL/
   9  *
  10  * Software distributed under the License is distributed on an "AS IS" basis,
  11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  12  * for the specific language governing rights and limitations under the
  13  * License.
  14  *
  15  * The Original Code is the Mozilla browser.
  16  *
  17  * The Initial Developer of the Original Code is
  18  * Netscape Communications, Inc.
  19  * Portions created by the Initial Developer are Copyright (C) 1999
  20  * the Initial Developer. All Rights Reserved.
  21  *
  22  * Contributor(s):
  23  *   Scott MacGregor <mscott@netscape.com>
  24  *
  25  * Alternatively, the contents of this file may be used under the terms of
  26  * either of the GNU General Public License Version 2 or later (the "GPL"),
  27  * or the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  28  * in which case the provisions of the GPL or the LGPL are applicable instead
  29  * of those above. If you wish to allow use of your version of this file only
  30  * under the terms of either the GPL or the LGPL, and not to allow others to
  31  * use your version of this file under the terms of the MPL, indicate your
  32  * decision by deleting the provisions above and replace them with the notice
  33  * and other provisions required by the GPL or the LGPL. If you do not delete
  34  * the provisions above, a recipient may use your version of this file under
  35  * the terms of any one of the MPL, the GPL or the LGPL.
  36  *
  37  * ***** END LICENSE BLOCK ***** */
  38
  39 #ifndef nsOSHelperAppService_h__
  40 #define nsOSHelperAppService_h__
  41
  42 // The OS helper app service is a subclass of nsExternalHelperAppService and is implemented on each
  43 // platform. It contains platform specific code for finding helper applications for a given mime type
  44 // in addition to launching those applications.
  45
  46 #include "nsExternalHelperAppService.h"
  47 #include "nsCExternalHandlerService.h"
  48 #include "nsCOMPtr.h"
  49 #include <windows.h>
  50
  51 #ifdef _WIN32_WINNT
  52 #undef _WIN32_WINNT
  53 #endif
  54 #define _WIN32_WINNT 0x0600
  55 #define INITGUID
  56 #include <shlobj.h>
  57
  58 class nsMIMEInfoWin;
  59
  60 class nsOSHelperAppService : public nsExternalHelperAppService
  61 {
  62 public:
  63   nsOSHelperAppService();
  64   virtual ~nsOSHelperAppService();
  65
  66   // override nsIExternalProtocolService methods
  67   nsresult OSProtocolHandlerExists(const char * aProtocolScheme, PRBool * aHandlerExists);
  68   nsresult LoadUriInternal(nsIURI * aURL);
  69   NS_IMETHOD GetApplicationDescription(const nsACString& aScheme, nsAString& _retval);
  70
  71   // method overrides for windows registry look up steps....
  72   already_AddRefed<nsIMIMEInfo> GetMIMEInfoFromOS(const nsACString& aMIMEType, const nsACString& aFileExt, PRBool *aFound);
  73   NS_IMETHOD GetProtocolHandlerInfoFromOS(const nsACString &aScheme,
  74                                           PRBool *found,
  75                                           nsIHandlerInfo **_retval);
  76
  77   /** Get the string value of a registry value and store it in result.
  78    * @return PR_TRUE on success, PR_FALSE on failure
  79    */
  80   static PRBool GetValueString(HKEY hKey, const PRUnichar* pValueName, nsAString& result);
  81
  82   // Removes registry command handler parameters, quotes, and expands environment strings.
  83   static PRBool CleanupCmdHandlerPath(nsAString& aCommandHandler);
  84
  85 protected:
  86   nsresult GetDefaultAppInfo(const nsAString& aTypeName, nsAString& aDefaultDescription, nsIFile** aDefaultApplication);
  87   // Lookup a mime info by extension, using an optional type hint
  88   already_AddRefed<nsMIMEInfoWin> GetByExtension(const nsAFlatString& aFileExt, const char *aTypeHint = nsnull);
  89   nsresult FindOSMimeInfoForType(const char * aMimeContentType, nsIURI * aURI, char ** aFileExtension, nsIMIMEInfo ** aMIMEInfo);
  90
  91   static nsresult GetMIMEInfoFromRegistry(const nsAFlatString& fileType, nsIMIMEInfo *pInfo);
  92   /// Looks up the type for the extension aExt and compares it to aType
  93   static PRBool typeFromExtEquals(const PRUnichar* aExt, const char *aType);
  94
  95 private:
  96 #if !defined(MOZ_DISABLE_VISTA_SDK_REQUIREMENTS)
  97   IApplicationAssociationRegistration* mAppAssoc;
  98 #endif
  99 };
 100
 101 #endif // nsOSHelperAppService_h__