1 files changed, 142 insertions, 0 deletions
diff --git a/browser/GeckoContentHandler.cpp b/browser/GeckoContentHandler.cpp
new file mode 100644
index 0000000..8b6c9fb
--- /dev/null
+++ b/browser/GeckoContentHandler.cpp
@@ -0,0 +1,142 @@
+#include <nsCExternalHandlerService.h>
+#include <nsIFile.h>
+#include <nsIFactory.h>
+#include <nsILocalFile.h>
+#include <nsStringAPI.h>
+
+#include <nsComponentManagerUtils.h>
+
+#include "GeckoContentHandler.h"
+
+class GeckoContentHandler : public nsIHelperAppLauncherDialog
+{
+public:
+    NS_DECL_ISUPPORTS
+    NS_DECL_NSIHELPERAPPLAUNCHERDIALOG
+
+    GeckoContentHandler();
+    virtual ~GeckoContentHandler();
+};
+
+GeckoContentHandler::GeckoContentHandler()
+{
+
+}
+
+GeckoContentHandler::~GeckoContentHandler()
+{
+
+}
+
+NS_IMPL_ISUPPORTS1(GeckoContentHandler, nsIHelperAppLauncherDialog)
+
+NS_IMETHODIMP
+GeckoContentHandler::Show (nsIHelperAppLauncher *aLauncher,
+						   nsISupports *aContext,
+						   PRUint32 aReason)
+{
+	aLauncher->SaveToDisk(NULL, PR_FALSE);
+
+	return NS_OK;
+}
+
+#include <glib.h>
+NS_IMETHODIMP
+GeckoContentHandler::PromptForSaveToFile (nsIHelperAppLauncher *aLauncher,			    
+										  nsISupports *aWindowContext,
+										  const PRUnichar *aDefaultFile,
+										  const PRUnichar *aSuggestedFileExtension,
+										  nsILocalFile **_retval)
+{
+	char *filename = NULL;
+	nsCString defaultFile;
+
+	NS_UTF16ToCString(nsString(aDefaultFile), NS_CSTRING_ENCODING_UTF8, defaultFile);
+
+	nsCOMPtr <nsILocalFile> destFile(do_CreateInstance("@mozilla.org/file/local;1"));
+	NS_ENSURE_TRUE(destFile, NS_ERROR_FAILURE);
+
+	const char * suggested = defaultFile.get();
+	if (strlen(suggested) > 0) {
+		filename = g_build_path("/", g_get_tmp_dir (), suggested, NULL);
+	} else {
+		filename = tempnam(NULL, NULL);
+	}
+
+	if (filename == NULL)
+		return NS_ERROR_OUT_OF_MEMORY;
+
+	destFile->InitWithNativePath(nsCString(filename));
+	g_free(filename);
+
+	NS_IF_ADDREF(*_retval = destFile);
+	return NS_OK;
+}
+
+//*****************************************************************************
+// GeckoContentHandlerFactory
+//*****************************************************************************
+
+class GeckoContentHandlerFactory : public nsIFactory {
+public:
+  NS_DECL_ISUPPORTS
+  NS_DECL_NSIFACTORY
+
+  GeckoContentHandlerFactory();
+  virtual ~GeckoContentHandlerFactory();
+};
+
+//*****************************************************************************
+
+NS_IMPL_ISUPPORTS1(GeckoContentHandlerFactory, nsIFactory)
+
+GeckoContentHandlerFactory::GeckoContentHandlerFactory() {
+}
+
+GeckoContentHandlerFactory::~GeckoContentHandlerFactory() {
+}
+
+NS_IMETHODIMP
+GeckoContentHandlerFactory::CreateInstance(nsISupports *aOuter,
+                                           const nsIID & aIID,
+                                           void **aResult)
+{
+    NS_ENSURE_ARG_POINTER(aResult);
+
+    *aResult = NULL;
+    GeckoContentHandler *inst = new GeckoContentHandler;
+    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
+GeckoContentHandlerFactory::LockFactory(PRBool lock)
+{
+    return NS_OK;
+}
+
+//*****************************************************************************
+
+nsresult
+NS_NewGeckoContentHandlerFactory(nsIFactory** aFactory)
+{
+    NS_ENSURE_ARG_POINTER(aFactory);
+    *aFactory = nsnull;
+
+    GeckoContentHandlerFactory *result = new GeckoContentHandlerFactory;
+    if (!result)
+        return NS_ERROR_OUT_OF_MEMORY;
+
+    NS_ADDREF(result);
+    *aFactory = result;
+
+    return NS_OK;
+}