#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;
}