2007-03-22 10:30:00 -07:00
|
|
|
/* ***** BEGIN LICENSE BLOCK *****
|
|
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
|
|
*
|
|
|
|
* 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 toolkit.
|
|
|
|
*
|
|
|
|
* The Initial Developer of the Original Code is
|
|
|
|
* Benjamin Smedberg <benjamin@smedbergs.us>
|
|
|
|
*
|
|
|
|
* Portions created by the Initial Developer are Copyright (C) 2004
|
|
|
|
* the Initial Developer. All Rights Reserved.
|
|
|
|
*
|
|
|
|
* Contributor(s):
|
|
|
|
*
|
|
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
|
|
* the provisions above, a recipient may use your version of this file under
|
|
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
|
|
*
|
|
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
|
|
|
|
#include "nsICommandLineRunner.h"
|
|
|
|
|
|
|
|
#include "nsICategoryManager.h"
|
|
|
|
#include "nsICommandLineHandler.h"
|
2007-07-10 22:59:56 -07:00
|
|
|
#include "nsICommandLineValidator.h"
|
2010-07-20 06:29:38 -07:00
|
|
|
#include "nsIConsoleService.h"
|
2007-03-22 10:30:00 -07:00
|
|
|
#include "nsIClassInfoImpl.h"
|
|
|
|
#include "nsIDOMWindow.h"
|
|
|
|
#include "nsIFile.h"
|
|
|
|
#include "nsISimpleEnumerator.h"
|
|
|
|
#include "nsIStringEnumerator.h"
|
|
|
|
|
|
|
|
#include "nsCOMPtr.h"
|
2010-06-10 11:11:40 -07:00
|
|
|
#include "mozilla/ModuleUtils.h"
|
2007-03-22 10:30:00 -07:00
|
|
|
#include "nsISupportsImpl.h"
|
|
|
|
#include "nsNativeCharsetUtils.h"
|
|
|
|
#include "nsNetUtil.h"
|
|
|
|
#include "nsUnicharUtils.h"
|
2009-01-18 12:14:14 -08:00
|
|
|
#include "nsTArray.h"
|
2010-07-20 06:29:38 -07:00
|
|
|
#include "nsTextFormatter.h"
|
2007-03-22 10:30:00 -07:00
|
|
|
#include "nsXPCOMCID.h"
|
|
|
|
#include "plstr.h"
|
|
|
|
|
2010-04-27 05:57:46 -07:00
|
|
|
#ifdef MOZ_WIDGET_COCOA
|
2009-03-19 10:41:13 -07:00
|
|
|
#include <CoreFoundation/CoreFoundation.h>
|
2007-03-22 10:30:00 -07:00
|
|
|
#include "nsILocalFileMac.h"
|
|
|
|
#elif defined(XP_WIN)
|
|
|
|
#include <windows.h>
|
|
|
|
#include <shlobj.h>
|
|
|
|
#elif defined(XP_UNIX)
|
|
|
|
#include <unistd.h>
|
|
|
|
#elif defined(XP_OS2)
|
|
|
|
#include <os2.h>
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef DEBUG_bsmedberg
|
|
|
|
#define DEBUG_COMMANDLINE
|
|
|
|
#endif
|
|
|
|
|
2010-06-22 09:59:57 -07:00
|
|
|
#define NS_COMMANDLINE_CID \
|
|
|
|
{ 0x23bcc750, 0xdc20, 0x460b, { 0xb2, 0xd4, 0x74, 0xd8, 0xf5, 0x8d, 0x36, 0x15 } }
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
class nsCommandLine : public nsICommandLineRunner
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
NS_DECL_ISUPPORTS
|
|
|
|
NS_DECL_NSICOMMANDLINE
|
|
|
|
NS_DECL_NSICOMMANDLINERUNNER
|
|
|
|
|
|
|
|
nsCommandLine();
|
|
|
|
|
|
|
|
protected:
|
|
|
|
~nsCommandLine() { }
|
|
|
|
|
2007-07-10 22:59:56 -07:00
|
|
|
typedef nsresult (*EnumerateHandlersCallback)(nsICommandLineHandler* aHandler,
|
|
|
|
nsICommandLine* aThis,
|
|
|
|
void *aClosure);
|
|
|
|
typedef nsresult (*EnumerateValidatorsCallback)(nsICommandLineValidator* aValidator,
|
2007-03-22 10:30:00 -07:00
|
|
|
nsICommandLine* aThis,
|
|
|
|
void *aClosure);
|
|
|
|
|
|
|
|
void appendArg(const char* arg);
|
2008-04-18 13:43:49 -07:00
|
|
|
void resolveShortcutURL(nsILocalFile* aFile, nsACString& outURL);
|
2007-07-10 22:59:56 -07:00
|
|
|
nsresult EnumerateHandlers(EnumerateHandlersCallback aCallback, void *aClosure);
|
|
|
|
nsresult EnumerateValidators(EnumerateValidatorsCallback aCallback, void *aClosure);
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2009-01-18 12:14:14 -08:00
|
|
|
nsTArray<nsString> mArgs;
|
|
|
|
PRUint32 mState;
|
|
|
|
nsCOMPtr<nsIFile> mWorkingDir;
|
|
|
|
nsCOMPtr<nsIDOMWindow> mWindowContext;
|
|
|
|
PRBool mPreventDefault;
|
2007-03-22 10:30:00 -07:00
|
|
|
};
|
|
|
|
|
|
|
|
nsCommandLine::nsCommandLine() :
|
|
|
|
mState(STATE_INITIAL_LAUNCH),
|
|
|
|
mPreventDefault(PR_FALSE)
|
|
|
|
{
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2010-06-22 09:59:57 -07:00
|
|
|
NS_IMPL_CLASSINFO(nsCommandLine, NULL, 0, NS_COMMANDLINE_CID)
|
2007-03-22 10:30:00 -07:00
|
|
|
NS_IMPL_ISUPPORTS2_CI(nsCommandLine,
|
|
|
|
nsICommandLine,
|
|
|
|
nsICommandLineRunner)
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::GetLength(PRInt32 *aResult)
|
|
|
|
{
|
2009-01-18 12:14:14 -08:00
|
|
|
*aResult = PRInt32(mArgs.Length());
|
2007-03-22 10:30:00 -07:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::GetArgument(PRInt32 aIndex, nsAString& aResult)
|
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_MIN(aIndex, 0);
|
2010-04-29 00:14:00 -07:00
|
|
|
NS_ENSURE_ARG_MAX(PRUint32(aIndex), mArgs.Length());
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2009-01-18 12:14:14 -08:00
|
|
|
aResult = mArgs[aIndex];
|
2007-03-22 10:30:00 -07:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::FindFlag(const nsAString& aFlag, PRBool aCaseSensitive, PRInt32 *aResult)
|
|
|
|
{
|
|
|
|
NS_ENSURE_ARG(!aFlag.IsEmpty());
|
|
|
|
|
|
|
|
nsDefaultStringComparator caseCmp;
|
|
|
|
nsCaseInsensitiveStringComparator caseICmp;
|
|
|
|
nsStringComparator& c = aCaseSensitive ?
|
2007-07-08 00:08:04 -07:00
|
|
|
static_cast<nsStringComparator&>(caseCmp) :
|
|
|
|
static_cast<nsStringComparator&>(caseICmp);
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2009-01-18 12:14:14 -08:00
|
|
|
for (PRUint32 f = 0; f < mArgs.Length(); f++) {
|
|
|
|
const nsString &arg = mArgs[f];
|
2007-03-22 10:30:00 -07:00
|
|
|
|
|
|
|
if (arg.Length() >= 2 && arg.First() == PRUnichar('-')) {
|
|
|
|
if (aFlag.Equals(Substring(arg, 1), c)) {
|
|
|
|
*aResult = f;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
*aResult = -1;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::RemoveArguments(PRInt32 aStart, PRInt32 aEnd)
|
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_MIN(aStart, 0);
|
2010-04-29 00:14:00 -07:00
|
|
|
NS_ENSURE_ARG_MAX(PRUint32(aEnd) + 1, mArgs.Length());
|
2007-03-22 10:30:00 -07:00
|
|
|
|
|
|
|
for (PRInt32 i = aEnd; i >= aStart; --i) {
|
2009-01-18 12:14:14 -08:00
|
|
|
mArgs.RemoveElementAt(i);
|
2007-03-22 10:30:00 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::HandleFlag(const nsAString& aFlag, PRBool aCaseSensitive,
|
|
|
|
PRBool *aResult)
|
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
PRInt32 found;
|
|
|
|
rv = FindFlag(aFlag, aCaseSensitive, &found);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
if (found == -1) {
|
|
|
|
*aResult = PR_FALSE;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
*aResult = PR_TRUE;
|
|
|
|
RemoveArguments(found, found);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::HandleFlagWithParam(const nsAString& aFlag, PRBool aCaseSensitive,
|
|
|
|
nsAString& aResult)
|
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
PRInt32 found;
|
|
|
|
rv = FindFlag(aFlag, aCaseSensitive, &found);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
if (found == -1) {
|
|
|
|
aResult.SetIsVoid(PR_TRUE);
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2009-01-18 12:14:14 -08:00
|
|
|
if (found == PRInt32(mArgs.Length()) - 1) {
|
2007-03-22 10:30:00 -07:00
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
}
|
|
|
|
|
|
|
|
++found;
|
|
|
|
|
2009-01-18 12:14:14 -08:00
|
|
|
if (mArgs[found].First() == '-') {
|
2007-03-22 10:30:00 -07:00
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
}
|
|
|
|
|
2009-01-18 12:14:14 -08:00
|
|
|
aResult = mArgs[found];
|
2007-03-22 10:30:00 -07:00
|
|
|
RemoveArguments(found - 1, found);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::GetState(PRUint32 *aResult)
|
|
|
|
{
|
|
|
|
*aResult = mState;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::GetPreventDefault(PRBool *aResult)
|
|
|
|
{
|
|
|
|
*aResult = mPreventDefault;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::SetPreventDefault(PRBool aValue)
|
|
|
|
{
|
|
|
|
mPreventDefault = aValue;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::GetWorkingDirectory(nsIFile* *aResult)
|
|
|
|
{
|
|
|
|
NS_ENSURE_TRUE(mWorkingDir, NS_ERROR_NOT_INITIALIZED);
|
|
|
|
|
|
|
|
NS_ADDREF(*aResult = mWorkingDir);
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::GetWindowContext(nsIDOMWindow* *aResult)
|
|
|
|
{
|
|
|
|
NS_IF_ADDREF(*aResult = mWindowContext);
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::SetWindowContext(nsIDOMWindow* aValue)
|
|
|
|
{
|
|
|
|
mWindowContext = aValue;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::ResolveFile(const nsAString& aArgument, nsIFile* *aResult)
|
|
|
|
{
|
|
|
|
NS_ENSURE_TRUE(mWorkingDir, NS_ERROR_NOT_INITIALIZED);
|
|
|
|
|
|
|
|
// This is some seriously screwed-up code. nsILocalFile.appendRelativeNativePath
|
|
|
|
// explicitly does not accept .. or . path parts, but that is exactly what we
|
|
|
|
// need here. So we hack around it.
|
|
|
|
|
|
|
|
nsresult rv;
|
|
|
|
|
2010-04-27 05:57:46 -07:00
|
|
|
#if defined(MOZ_WIDGET_COCOA)
|
2007-03-22 10:30:00 -07:00
|
|
|
nsCOMPtr<nsILocalFileMac> lfm (do_QueryInterface(mWorkingDir));
|
|
|
|
NS_ENSURE_TRUE(lfm, NS_ERROR_NO_INTERFACE);
|
|
|
|
|
|
|
|
nsCOMPtr<nsILocalFileMac> newfile (do_CreateInstance(NS_LOCAL_FILE_CONTRACTID));
|
|
|
|
NS_ENSURE_TRUE(newfile, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
|
|
|
|
CFURLRef baseurl;
|
|
|
|
rv = lfm->GetCFURL(&baseurl);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
nsCAutoString path;
|
|
|
|
NS_CopyUnicodeToNative(aArgument, path);
|
|
|
|
|
|
|
|
CFURLRef newurl =
|
|
|
|
CFURLCreateFromFileSystemRepresentationRelativeToBase(NULL, (const UInt8*) path.get(),
|
|
|
|
path.Length(),
|
|
|
|
true, baseurl);
|
|
|
|
|
|
|
|
CFRelease(baseurl);
|
|
|
|
|
|
|
|
rv = newfile->InitWithCFURL(newurl);
|
|
|
|
CFRelease(newurl);
|
|
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
|
|
|
|
NS_ADDREF(*aResult = newfile);
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
#elif defined(XP_UNIX)
|
|
|
|
nsCOMPtr<nsILocalFile> lf (do_CreateInstance(NS_LOCAL_FILE_CONTRACTID));
|
|
|
|
NS_ENSURE_TRUE(lf, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
|
|
|
|
if (aArgument.First() == '/') {
|
|
|
|
// absolute path
|
|
|
|
rv = lf->InitWithPath(aArgument);
|
|
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
|
|
|
|
NS_ADDREF(*aResult = lf);
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
nsCAutoString nativeArg;
|
|
|
|
NS_CopyUnicodeToNative(aArgument, nativeArg);
|
|
|
|
|
|
|
|
nsCAutoString newpath;
|
|
|
|
mWorkingDir->GetNativePath(newpath);
|
|
|
|
|
|
|
|
newpath.Append('/');
|
|
|
|
newpath.Append(nativeArg);
|
|
|
|
|
|
|
|
rv = lf->InitWithNativePath(newpath);
|
|
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
|
|
|
|
rv = lf->Normalize();
|
|
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
|
|
|
|
NS_ADDREF(*aResult = lf);
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
#elif defined(XP_WIN32)
|
|
|
|
nsCOMPtr<nsILocalFile> lf (do_CreateInstance(NS_LOCAL_FILE_CONTRACTID));
|
|
|
|
NS_ENSURE_TRUE(lf, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
|
|
|
|
rv = lf->InitWithPath(aArgument);
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
// If it's a relative path, the Init is *going* to fail. We use string magic and
|
|
|
|
// win32 _fullpath. Note that paths of the form "\Relative\To\CurDrive" are
|
|
|
|
// going to fail, and I haven't figured out a way to work around this without
|
|
|
|
// the PathCombine() function, which is not available in plain win95/nt4
|
|
|
|
|
2007-12-31 07:15:43 -08:00
|
|
|
nsAutoString fullPath;
|
|
|
|
mWorkingDir->GetPath(fullPath);
|
2007-03-22 10:30:00 -07:00
|
|
|
|
|
|
|
fullPath.Append('\\');
|
2007-12-31 07:15:43 -08:00
|
|
|
fullPath.Append(aArgument);
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2007-12-31 07:15:43 -08:00
|
|
|
WCHAR pathBuf[MAX_PATH];
|
|
|
|
if (!_wfullpath(pathBuf, fullPath.get(), MAX_PATH))
|
2007-03-22 10:30:00 -07:00
|
|
|
return NS_ERROR_FAILURE;
|
|
|
|
|
2007-12-31 07:15:43 -08:00
|
|
|
rv = lf->InitWithPath(nsDependentString(pathBuf));
|
2007-03-22 10:30:00 -07:00
|
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
}
|
|
|
|
NS_ADDREF(*aResult = lf);
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
#elif defined(XP_OS2)
|
|
|
|
nsCOMPtr<nsILocalFile> lf (do_CreateInstance(NS_LOCAL_FILE_CONTRACTID));
|
|
|
|
NS_ENSURE_TRUE(lf, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
|
|
|
|
rv = lf->InitWithPath(aArgument);
|
|
|
|
if (NS_FAILED(rv)) {
|
|
|
|
|
|
|
|
nsCAutoString fullPath;
|
|
|
|
mWorkingDir->GetNativePath(fullPath);
|
|
|
|
|
|
|
|
nsCAutoString carg;
|
|
|
|
NS_CopyUnicodeToNative(aArgument, carg);
|
|
|
|
|
|
|
|
fullPath.Append('\\');
|
|
|
|
fullPath.Append(carg);
|
|
|
|
|
|
|
|
char pathBuf[CCHMAXPATH];
|
|
|
|
if (DosQueryPathInfo(fullPath.get(), FIL_QUERYFULLNAME, pathBuf, sizeof(pathBuf)))
|
|
|
|
return NS_ERROR_FAILURE;
|
|
|
|
|
|
|
|
rv = lf->InitWithNativePath(nsDependentCString(pathBuf));
|
|
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
}
|
|
|
|
NS_ADDREF(*aResult = lf);
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
#else
|
|
|
|
#error Need platform-specific logic here.
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::ResolveURI(const nsAString& aArgument, nsIURI* *aResult)
|
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
// First, we try to init the argument as an absolute file path. If this doesn't
|
|
|
|
// work, it is an absolute or relative URI.
|
|
|
|
|
|
|
|
nsCOMPtr<nsIIOService> io = do_GetIOService();
|
|
|
|
NS_ENSURE_TRUE(io, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
|
2008-04-18 13:43:49 -07:00
|
|
|
nsCOMPtr<nsIURI> workingDirURI;
|
|
|
|
if (mWorkingDir) {
|
|
|
|
io->NewFileURI(mWorkingDir, getter_AddRefs(workingDirURI));
|
|
|
|
}
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
nsCOMPtr<nsILocalFile> lf (do_CreateInstance(NS_LOCAL_FILE_CONTRACTID));
|
|
|
|
rv = lf->InitWithPath(aArgument);
|
|
|
|
if (NS_SUCCEEDED(rv)) {
|
|
|
|
lf->Normalize();
|
2008-04-18 13:43:49 -07:00
|
|
|
nsCAutoString url;
|
|
|
|
// Try to resolve the url for .url files.
|
|
|
|
resolveShortcutURL(lf, url);
|
|
|
|
if (!url.IsEmpty()) {
|
|
|
|
return io->NewURI(url,
|
|
|
|
nsnull,
|
|
|
|
workingDirURI,
|
|
|
|
aResult);
|
|
|
|
}
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2008-04-18 13:43:49 -07:00
|
|
|
return io->NewFileURI(lf, aResult);
|
2008-04-17 16:05:06 -07:00
|
|
|
}
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
return io->NewURI(NS_ConvertUTF16toUTF8(aArgument),
|
|
|
|
nsnull,
|
|
|
|
workingDirURI,
|
|
|
|
aResult);
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
nsCommandLine::appendArg(const char* arg)
|
|
|
|
{
|
|
|
|
#ifdef DEBUG_COMMANDLINE
|
|
|
|
printf("Adding XP arg: %s\n", arg);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
nsAutoString warg;
|
2007-12-31 07:15:43 -08:00
|
|
|
#ifdef XP_WIN
|
|
|
|
CopyUTF8toUTF16(nsDependentCString(arg), warg);
|
|
|
|
#else
|
2007-03-22 10:30:00 -07:00
|
|
|
NS_CopyNativeToUnicode(nsDependentCString(arg), warg);
|
2007-12-31 07:15:43 -08:00
|
|
|
#endif
|
2007-03-22 10:30:00 -07:00
|
|
|
|
2009-01-18 12:14:14 -08:00
|
|
|
mArgs.AppendElement(warg);
|
2007-03-22 10:30:00 -07:00
|
|
|
}
|
|
|
|
|
2008-04-18 13:43:49 -07:00
|
|
|
void
|
|
|
|
nsCommandLine::resolveShortcutURL(nsILocalFile* aFile, nsACString& outURL)
|
|
|
|
{
|
|
|
|
nsCOMPtr<nsIFileProtocolHandler> fph;
|
|
|
|
nsresult rv = NS_GetFileProtocolHandler(getter_AddRefs(fph));
|
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return;
|
|
|
|
|
|
|
|
nsCOMPtr<nsIURI> uri;
|
|
|
|
rv = fph->ReadURLFile(aFile, getter_AddRefs(uri));
|
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return;
|
|
|
|
|
|
|
|
uri->GetSpec(outURL);
|
|
|
|
}
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::Init(PRInt32 argc, char** argv, nsIFile* aWorkingDir,
|
|
|
|
PRUint32 aState)
|
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_MIN(aState, 0);
|
|
|
|
NS_ENSURE_ARG_MAX(aState, 2);
|
|
|
|
|
|
|
|
PRInt32 i;
|
|
|
|
|
|
|
|
mWorkingDir = aWorkingDir;
|
|
|
|
|
|
|
|
// skip argv[0], we don't want it
|
|
|
|
for (i = 1; i < argc; ++i) {
|
|
|
|
const char* curarg = argv[i];
|
|
|
|
|
|
|
|
#ifdef DEBUG_COMMANDLINE
|
|
|
|
printf("Testing native arg %i: '%s'\n", i, curarg);
|
|
|
|
#endif
|
|
|
|
#if defined(XP_WIN) || defined(XP_OS2)
|
|
|
|
if (*curarg == '/') {
|
|
|
|
char* dup = PL_strdup(curarg);
|
|
|
|
if (!dup) return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
|
|
|
|
*dup = '-';
|
|
|
|
char* colon = PL_strchr(dup, ':');
|
|
|
|
if (colon) {
|
|
|
|
*colon = '\0';
|
|
|
|
appendArg(dup);
|
|
|
|
appendArg(colon+1);
|
|
|
|
} else {
|
|
|
|
appendArg(dup);
|
|
|
|
}
|
|
|
|
PL_strfree(dup);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
#ifdef XP_UNIX
|
|
|
|
if (*curarg == '-' &&
|
|
|
|
*(curarg+1) == '-') {
|
|
|
|
++curarg;
|
|
|
|
|
|
|
|
char* dup = PL_strdup(curarg);
|
|
|
|
if (!dup) return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
|
|
|
|
char* eq = PL_strchr(dup, '=');
|
|
|
|
if (eq) {
|
|
|
|
*eq = '\0';
|
|
|
|
appendArg(dup);
|
|
|
|
appendArg(eq + 1);
|
|
|
|
} else {
|
|
|
|
appendArg(dup);
|
|
|
|
}
|
|
|
|
PL_strfree(dup);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
appendArg(curarg);
|
|
|
|
}
|
|
|
|
|
|
|
|
mState = aState;
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2010-07-20 06:29:38 -07:00
|
|
|
static void
|
2010-11-22 12:11:04 -08:00
|
|
|
LogConsoleMessage(const PRUnichar* fmt, ...)
|
2010-07-20 06:29:38 -07:00
|
|
|
{
|
|
|
|
va_list args;
|
|
|
|
va_start(args, fmt);
|
2010-11-22 12:11:04 -08:00
|
|
|
PRUnichar* msg = nsTextFormatter::vsmprintf(fmt, args);
|
2010-07-20 06:29:38 -07:00
|
|
|
va_end(args);
|
|
|
|
|
|
|
|
nsCOMPtr<nsIConsoleService> cs = do_GetService("@mozilla.org/consoleservice;1");
|
|
|
|
if (cs)
|
|
|
|
cs->LogStringMessage(msg);
|
|
|
|
|
|
|
|
NS_Free(msg);
|
|
|
|
}
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
nsresult
|
2007-07-10 22:59:56 -07:00
|
|
|
nsCommandLine::EnumerateHandlers(EnumerateHandlersCallback aCallback, void *aClosure)
|
2007-03-22 10:30:00 -07:00
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
nsCOMPtr<nsICategoryManager> catman
|
|
|
|
(do_GetService(NS_CATEGORYMANAGER_CONTRACTID));
|
|
|
|
NS_ENSURE_TRUE(catman, NS_ERROR_UNEXPECTED);
|
|
|
|
|
|
|
|
nsCOMPtr<nsISimpleEnumerator> entenum;
|
|
|
|
rv = catman->EnumerateCategory("command-line-handler",
|
|
|
|
getter_AddRefs(entenum));
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
nsCOMPtr<nsIUTF8StringEnumerator> strenum (do_QueryInterface(entenum));
|
|
|
|
NS_ENSURE_TRUE(strenum, NS_ERROR_UNEXPECTED);
|
|
|
|
|
|
|
|
nsCAutoString entry;
|
|
|
|
PRBool hasMore;
|
|
|
|
while (NS_SUCCEEDED(strenum->HasMore(&hasMore)) && hasMore) {
|
|
|
|
strenum->GetNext(entry);
|
|
|
|
|
2010-09-16 09:50:37 -07:00
|
|
|
nsCString contractID;
|
2007-03-22 10:30:00 -07:00
|
|
|
rv = catman->GetCategoryEntry("command-line-handler",
|
|
|
|
entry.get(),
|
|
|
|
getter_Copies(contractID));
|
2010-09-16 09:50:37 -07:00
|
|
|
if (NS_FAILED(rv))
|
2007-03-22 10:30:00 -07:00
|
|
|
continue;
|
|
|
|
|
|
|
|
nsCOMPtr<nsICommandLineHandler> clh(do_GetService(contractID.get()));
|
2010-07-20 06:29:38 -07:00
|
|
|
if (!clh) {
|
2010-11-22 12:11:04 -08:00
|
|
|
LogConsoleMessage(NS_LITERAL_STRING("Contract ID '%s' was registered as a command line handler for entry '%s', but could not be created.").get(),
|
2010-09-16 09:50:37 -07:00
|
|
|
contractID.get(), entry.get());
|
2007-03-22 10:30:00 -07:00
|
|
|
continue;
|
2010-07-20 06:29:38 -07:00
|
|
|
}
|
2007-03-22 10:30:00 -07:00
|
|
|
|
|
|
|
rv = (aCallback)(clh, this, aClosure);
|
|
|
|
if (rv == NS_ERROR_ABORT)
|
|
|
|
break;
|
|
|
|
|
|
|
|
rv = NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
2007-07-10 22:59:56 -07:00
|
|
|
nsresult
|
|
|
|
nsCommandLine::EnumerateValidators(EnumerateValidatorsCallback aCallback, void *aClosure)
|
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
nsCOMPtr<nsICategoryManager> catman
|
|
|
|
(do_GetService(NS_CATEGORYMANAGER_CONTRACTID));
|
|
|
|
NS_ENSURE_TRUE(catman, NS_ERROR_UNEXPECTED);
|
|
|
|
|
|
|
|
nsCOMPtr<nsISimpleEnumerator> entenum;
|
|
|
|
rv = catman->EnumerateCategory("command-line-validator",
|
|
|
|
getter_AddRefs(entenum));
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
nsCOMPtr<nsIUTF8StringEnumerator> strenum (do_QueryInterface(entenum));
|
|
|
|
NS_ENSURE_TRUE(strenum, NS_ERROR_UNEXPECTED);
|
|
|
|
|
|
|
|
nsCAutoString entry;
|
|
|
|
PRBool hasMore;
|
|
|
|
while (NS_SUCCEEDED(strenum->HasMore(&hasMore)) && hasMore) {
|
|
|
|
strenum->GetNext(entry);
|
|
|
|
|
|
|
|
nsXPIDLCString contractID;
|
|
|
|
rv = catman->GetCategoryEntry("command-line-validator",
|
|
|
|
entry.get(),
|
|
|
|
getter_Copies(contractID));
|
|
|
|
if (!contractID)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
nsCOMPtr<nsICommandLineValidator> clv(do_GetService(contractID.get()));
|
|
|
|
if (!clv)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
rv = (aCallback)(clv, this, aClosure);
|
|
|
|
if (rv == NS_ERROR_ABORT)
|
|
|
|
break;
|
|
|
|
|
|
|
|
rv = NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
return rv;
|
|
|
|
}
|
|
|
|
|
|
|
|
static nsresult
|
|
|
|
EnumValidate(nsICommandLineValidator* aValidator, nsICommandLine* aThis, void*)
|
|
|
|
{
|
|
|
|
return aValidator->Validate(aThis);
|
|
|
|
}
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
static nsresult
|
|
|
|
EnumRun(nsICommandLineHandler* aHandler, nsICommandLine* aThis, void*)
|
|
|
|
{
|
|
|
|
return aHandler->Handle(aThis);
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::Run()
|
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
2007-07-10 22:59:56 -07:00
|
|
|
rv = EnumerateValidators(EnumValidate, nsnull);
|
|
|
|
if (rv == NS_ERROR_ABORT)
|
|
|
|
return rv;
|
|
|
|
|
2007-03-22 10:30:00 -07:00
|
|
|
rv = EnumerateHandlers(EnumRun, nsnull);
|
|
|
|
if (rv == NS_ERROR_ABORT)
|
|
|
|
return rv;
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
static nsresult
|
|
|
|
EnumHelp(nsICommandLineHandler* aHandler, nsICommandLine* aThis, void* aClosure)
|
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
nsCString text;
|
|
|
|
rv = aHandler->GetHelpInfo(text);
|
|
|
|
if (NS_SUCCEEDED(rv)) {
|
|
|
|
NS_ASSERTION(text.Length() == 0 || text.Last() == '\n',
|
|
|
|
"Help text from command line handlers should end in a newline.");
|
|
|
|
|
2007-07-08 00:08:04 -07:00
|
|
|
nsACString* totalText = reinterpret_cast<nsACString*>(aClosure);
|
2007-03-22 10:30:00 -07:00
|
|
|
totalText->Append(text);
|
|
|
|
}
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP
|
|
|
|
nsCommandLine::GetHelpText(nsACString& aResult)
|
|
|
|
{
|
|
|
|
EnumerateHandlers(EnumHelp, &aResult);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_GENERIC_FACTORY_CONSTRUCTOR(nsCommandLine)
|
|
|
|
|
2010-06-10 11:11:40 -07:00
|
|
|
NS_DEFINE_NAMED_CID(NS_COMMANDLINE_CID);
|
|
|
|
|
|
|
|
static const mozilla::Module::CIDEntry kCommandLineCIDs[] = {
|
|
|
|
{ &kNS_COMMANDLINE_CID, false, NULL, nsCommandLineConstructor },
|
|
|
|
{ NULL }
|
|
|
|
};
|
|
|
|
|
|
|
|
static const mozilla::Module::ContractIDEntry kCommandLineContracts[] = {
|
|
|
|
{ "@mozilla.org/toolkit/command-line;1", &kNS_COMMANDLINE_CID },
|
|
|
|
{ NULL }
|
|
|
|
};
|
|
|
|
|
|
|
|
static const mozilla::Module kCommandLineModule = {
|
|
|
|
mozilla::Module::kVersion,
|
|
|
|
kCommandLineCIDs,
|
|
|
|
kCommandLineContracts
|
2007-03-22 10:30:00 -07:00
|
|
|
};
|
|
|
|
|
2010-06-10 11:11:40 -07:00
|
|
|
NSMODULE_DEFN(CommandLineModule) = &kCommandLineModule;
|