2009-10-06 21:42:45 +00:00
|
|
|
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
2012-05-21 11:12:37 +00:00
|
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
#include "JumpListItem.h"
|
|
|
|
|
|
|
|
#include <shellapi.h>
|
|
|
|
#include <propvarutil.h>
|
|
|
|
#include <propkey.h>
|
|
|
|
|
|
|
|
#include "nsIFile.h"
|
|
|
|
#include "nsNetUtil.h"
|
|
|
|
#include "nsCRT.h"
|
|
|
|
#include "nsNetCID.h"
|
|
|
|
#include "nsCExternalHandlerService.h"
|
2010-03-18 10:59:10 +00:00
|
|
|
#include "nsCycleCollectionParticipant.h"
|
2011-09-06 19:11:28 +00:00
|
|
|
#include "mozilla/Preferences.h"
|
2011-09-06 19:11:28 +00:00
|
|
|
#include "JumpListBuilder.h"
|
2012-01-11 02:09:41 +00:00
|
|
|
#include "WinUtils.h"
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
namespace mozilla {
|
|
|
|
namespace widget {
|
|
|
|
|
|
|
|
// ISUPPORTS Impl's
|
|
|
|
NS_IMPL_ISUPPORTS1(JumpListItem,
|
|
|
|
nsIJumpListItem)
|
|
|
|
|
|
|
|
NS_IMPL_ISUPPORTS_INHERITED1(JumpListSeparator,
|
|
|
|
JumpListItem,
|
|
|
|
nsIJumpListSeparator)
|
|
|
|
|
|
|
|
NS_IMPL_ISUPPORTS_INHERITED1(JumpListLink,
|
|
|
|
JumpListItem,
|
|
|
|
nsIJumpListLink)
|
|
|
|
|
2010-03-18 10:59:10 +00:00
|
|
|
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(JumpListShortcut)
|
|
|
|
NS_INTERFACE_MAP_ENTRY(nsIJumpListShortcut)
|
|
|
|
NS_INTERFACE_MAP_END_INHERITING(JumpListItem)
|
|
|
|
|
2013-01-12 12:40:33 +00:00
|
|
|
NS_IMPL_CYCLE_COLLECTION_1(JumpListShortcut, mHandlerApp)
|
2010-03-18 10:59:10 +00:00
|
|
|
|
|
|
|
NS_IMPL_CYCLE_COLLECTING_ADDREF(JumpListShortcut)
|
|
|
|
NS_IMPL_CYCLE_COLLECTING_RELEASE(JumpListShortcut)
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
/* attribute short type; */
|
2012-08-22 15:56:38 +00:00
|
|
|
NS_IMETHODIMP JumpListItem::GetType(int16_t *aType)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_POINTER(aType);
|
|
|
|
|
|
|
|
*aType = mItemType;
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* boolean equals(nsIJumpListItem item); */
|
2011-09-29 06:19:26 +00:00
|
|
|
NS_IMETHODIMP JumpListItem::Equals(nsIJumpListItem *aItem, bool *aResult)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_POINTER(aItem);
|
|
|
|
|
2011-10-02 02:16:19 +00:00
|
|
|
*aResult = false;
|
2009-10-06 21:42:45 +00:00
|
|
|
|
2012-08-22 15:56:38 +00:00
|
|
|
int16_t theType = nsIJumpListItem::JUMPLIST_ITEM_EMPTY;
|
2009-10-06 21:42:45 +00:00
|
|
|
if (NS_FAILED(aItem->GetType(&theType)))
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
// Make sure the types match.
|
|
|
|
if (Type() != theType)
|
|
|
|
return NS_OK;
|
|
|
|
|
2011-10-02 02:16:19 +00:00
|
|
|
*aResult = true;
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* link impl. */
|
|
|
|
|
|
|
|
/* attribute nsIURI uri; */
|
|
|
|
NS_IMETHODIMP JumpListLink::GetUri(nsIURI **aURI)
|
|
|
|
{
|
|
|
|
NS_IF_ADDREF(*aURI = mURI);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP JumpListLink::SetUri(nsIURI *aURI)
|
|
|
|
{
|
|
|
|
mURI = aURI;
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* attribute AString uriTitle; */
|
|
|
|
NS_IMETHODIMP JumpListLink::SetUriTitle(const nsAString &aUriTitle)
|
|
|
|
{
|
|
|
|
mUriTitle.Assign(aUriTitle);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP JumpListLink::GetUriTitle(nsAString& aUriTitle)
|
|
|
|
{
|
|
|
|
aUriTitle.Assign(mUriTitle);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* readonly attribute long uriHash; */
|
|
|
|
NS_IMETHODIMP JumpListLink::GetUriHash(nsACString& aUriHash)
|
|
|
|
{
|
|
|
|
if (!mURI)
|
|
|
|
return NS_ERROR_NOT_AVAILABLE;
|
|
|
|
|
2012-07-21 20:07:26 +00:00
|
|
|
return mozilla::widget::FaviconHelper::HashURI(mCryptoHash, mURI, aUriHash);
|
2009-10-06 21:42:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* boolean compareHash(in nsIURI uri); */
|
2011-09-29 06:19:26 +00:00
|
|
|
NS_IMETHODIMP JumpListLink::CompareHash(nsIURI *aUri, bool *aResult)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
if (!mURI) {
|
|
|
|
*aResult = !aUri;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_ENSURE_ARG_POINTER(aUri);
|
|
|
|
|
2012-09-02 02:35:17 +00:00
|
|
|
nsAutoCString hash1, hash2;
|
2009-10-06 21:42:45 +00:00
|
|
|
|
2012-07-21 20:07:26 +00:00
|
|
|
rv = mozilla::widget::FaviconHelper::HashURI(mCryptoHash, mURI, hash1);
|
2009-10-06 21:42:45 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2012-07-21 20:07:26 +00:00
|
|
|
rv = mozilla::widget::FaviconHelper::HashURI(mCryptoHash, aUri, hash2);
|
2009-10-06 21:42:45 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
*aResult = hash1.Equals(hash2);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* boolean equals(nsIJumpListItem item); */
|
2011-09-29 06:19:26 +00:00
|
|
|
NS_IMETHODIMP JumpListLink::Equals(nsIJumpListItem *aItem, bool *aResult)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_POINTER(aItem);
|
|
|
|
|
|
|
|
nsresult rv;
|
|
|
|
|
2011-10-02 02:16:19 +00:00
|
|
|
*aResult = false;
|
2009-10-06 21:42:45 +00:00
|
|
|
|
2012-08-22 15:56:38 +00:00
|
|
|
int16_t theType = nsIJumpListItem::JUMPLIST_ITEM_EMPTY;
|
2009-10-06 21:42:45 +00:00
|
|
|
if (NS_FAILED(aItem->GetType(&theType)))
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
// Make sure the types match.
|
|
|
|
if (Type() != theType)
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
nsCOMPtr<nsIJumpListLink> link = do_QueryInterface(aItem, &rv);
|
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return rv;
|
|
|
|
|
|
|
|
// Check the titles
|
|
|
|
nsAutoString title;
|
|
|
|
link->GetUriTitle(title);
|
|
|
|
if (!mUriTitle.Equals(title))
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
// Call the internal object's equals() method to check.
|
|
|
|
nsCOMPtr<nsIURI> theUri;
|
2011-09-29 06:19:26 +00:00
|
|
|
bool equals = false;
|
2009-10-06 21:42:45 +00:00
|
|
|
if (NS_SUCCEEDED(link->GetUri(getter_AddRefs(theUri)))) {
|
|
|
|
if (!theUri) {
|
|
|
|
if (!mURI)
|
2011-10-02 02:16:19 +00:00
|
|
|
*aResult = true;
|
2009-10-06 21:42:45 +00:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
if (NS_SUCCEEDED(theUri->Equals(mURI, &equals)) && equals) {
|
2011-10-02 02:16:19 +00:00
|
|
|
*aResult = true;
|
2009-10-06 21:42:45 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* shortcut impl. */
|
|
|
|
|
|
|
|
/* attribute nsILocalHandlerApp app; */
|
|
|
|
NS_IMETHODIMP JumpListShortcut::GetApp(nsILocalHandlerApp **aApp)
|
|
|
|
{
|
|
|
|
NS_IF_ADDREF(*aApp = mHandlerApp);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
NS_IMETHODIMP JumpListShortcut::SetApp(nsILocalHandlerApp *aApp)
|
|
|
|
{
|
|
|
|
mHandlerApp = aApp;
|
|
|
|
|
|
|
|
// Confirm the app is present on the system
|
|
|
|
if (!ExecutableExists(mHandlerApp))
|
|
|
|
return NS_ERROR_FILE_NOT_FOUND;
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* attribute long iconIndex; */
|
2012-08-22 15:56:38 +00:00
|
|
|
NS_IMETHODIMP JumpListShortcut::GetIconIndex(int32_t *aIconIndex)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_POINTER(aIconIndex);
|
|
|
|
|
|
|
|
*aIconIndex = mIconIndex;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2012-08-22 15:56:38 +00:00
|
|
|
NS_IMETHODIMP JumpListShortcut::SetIconIndex(int32_t aIconIndex)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
mIconIndex = aIconIndex;
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2011-09-06 19:11:28 +00:00
|
|
|
/* attribute long iconURI; */
|
2011-09-06 19:11:28 +00:00
|
|
|
NS_IMETHODIMP JumpListShortcut::GetFaviconPageUri(nsIURI **aFaviconPageURI)
|
2011-09-06 19:11:28 +00:00
|
|
|
{
|
2011-09-06 19:11:28 +00:00
|
|
|
NS_IF_ADDREF(*aFaviconPageURI = mFaviconPageURI);
|
2011-09-06 19:11:28 +00:00
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2011-09-06 19:11:28 +00:00
|
|
|
NS_IMETHODIMP JumpListShortcut::SetFaviconPageUri(nsIURI *aFaviconPageURI)
|
2011-09-06 19:11:28 +00:00
|
|
|
{
|
2011-09-06 19:11:28 +00:00
|
|
|
mFaviconPageURI = aFaviconPageURI;
|
2011-09-06 19:11:28 +00:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2009-10-06 21:42:45 +00:00
|
|
|
/* boolean equals(nsIJumpListItem item); */
|
2011-09-29 06:19:26 +00:00
|
|
|
NS_IMETHODIMP JumpListShortcut::Equals(nsIJumpListItem *aItem, bool *aResult)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_POINTER(aItem);
|
|
|
|
|
|
|
|
nsresult rv;
|
|
|
|
|
2011-10-02 02:16:19 +00:00
|
|
|
*aResult = false;
|
2009-10-06 21:42:45 +00:00
|
|
|
|
2012-08-22 15:56:38 +00:00
|
|
|
int16_t theType = nsIJumpListItem::JUMPLIST_ITEM_EMPTY;
|
2009-10-06 21:42:45 +00:00
|
|
|
if (NS_FAILED(aItem->GetType(&theType)))
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
// Make sure the types match.
|
|
|
|
if (Type() != theType)
|
|
|
|
return NS_OK;
|
|
|
|
|
|
|
|
nsCOMPtr<nsIJumpListShortcut> shortcut = do_QueryInterface(aItem, &rv);
|
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return rv;
|
|
|
|
|
|
|
|
// Check the icon index
|
2012-08-22 15:56:38 +00:00
|
|
|
//int32_t idx;
|
2009-10-06 21:42:45 +00:00
|
|
|
//shortcut->GetIconIndex(&idx);
|
|
|
|
//if (mIconIndex != idx)
|
|
|
|
// return NS_OK;
|
2011-09-06 19:11:28 +00:00
|
|
|
// No need to check the icon page URI either
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
// Call the internal object's equals() method to check.
|
|
|
|
nsCOMPtr<nsILocalHandlerApp> theApp;
|
2011-09-29 06:19:26 +00:00
|
|
|
bool equals = false;
|
2009-10-06 21:42:45 +00:00
|
|
|
if (NS_SUCCEEDED(shortcut->GetApp(getter_AddRefs(theApp)))) {
|
|
|
|
if (!theApp) {
|
|
|
|
if (!mHandlerApp)
|
2011-10-02 02:16:19 +00:00
|
|
|
*aResult = true;
|
2009-10-06 21:42:45 +00:00
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
if (NS_SUCCEEDED(theApp->Equals(mHandlerApp, &equals)) && equals) {
|
2011-10-02 02:16:19 +00:00
|
|
|
*aResult = true;
|
2009-10-06 21:42:45 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* internal helpers */
|
|
|
|
|
|
|
|
// (static) Creates a ShellLink that encapsulate a separator.
|
|
|
|
nsresult JumpListSeparator::GetSeparator(nsRefPtr<IShellLinkW>& aShellLink)
|
|
|
|
{
|
|
|
|
HRESULT hr;
|
|
|
|
IShellLinkW* psl;
|
|
|
|
|
|
|
|
// Create a IShellLink.
|
|
|
|
hr = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
|
|
|
|
IID_IShellLinkW, (LPVOID*)&psl);
|
|
|
|
if (FAILED(hr))
|
|
|
|
return NS_ERROR_UNEXPECTED;
|
|
|
|
|
2012-07-30 14:20:58 +00:00
|
|
|
IPropertyStore* pPropStore = nullptr;
|
2009-10-06 21:42:45 +00:00
|
|
|
hr = psl->QueryInterface(IID_IPropertyStore, (LPVOID*)&pPropStore);
|
|
|
|
if (FAILED(hr))
|
|
|
|
return NS_ERROR_UNEXPECTED;
|
|
|
|
|
|
|
|
PROPVARIANT pv;
|
|
|
|
InitPropVariantFromBoolean(TRUE, &pv);
|
|
|
|
|
|
|
|
pPropStore->SetValue(PKEY_AppUserModel_IsDestListSeparator, pv);
|
|
|
|
pPropStore->Commit();
|
|
|
|
pPropStore->Release();
|
|
|
|
|
|
|
|
PropVariantClear(&pv);
|
|
|
|
|
|
|
|
aShellLink = dont_AddRef(psl);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
// (static) Creates a ShellLink that encapsulate a shortcut to local apps.
|
2011-09-06 19:11:28 +00:00
|
|
|
nsresult JumpListShortcut::GetShellLink(nsCOMPtr<nsIJumpListItem>& item,
|
|
|
|
nsRefPtr<IShellLinkW>& aShellLink,
|
|
|
|
nsCOMPtr<nsIThread> &aIOThread)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
HRESULT hr;
|
|
|
|
IShellLinkW* psl;
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
// Shell links:
|
|
|
|
// http://msdn.microsoft.com/en-us/library/bb776891(VS.85).aspx
|
|
|
|
// http://msdn.microsoft.com/en-us/library/bb774950(VS.85).aspx
|
|
|
|
|
2012-08-22 15:56:38 +00:00
|
|
|
int16_t type;
|
2009-10-06 21:42:45 +00:00
|
|
|
if (NS_FAILED(item->GetType(&type)))
|
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
|
|
|
|
if (type != nsIJumpListItem::JUMPLIST_ITEM_SHORTCUT)
|
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
|
|
|
|
nsCOMPtr<nsIJumpListShortcut> shortcut = do_QueryInterface(item, &rv);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
nsCOMPtr<nsILocalHandlerApp> handlerApp;
|
|
|
|
rv = shortcut->GetApp(getter_AddRefs(handlerApp));
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
// Create a IShellLink
|
|
|
|
hr = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
|
|
|
|
IID_IShellLinkW, (LPVOID*)&psl);
|
|
|
|
if (FAILED(hr))
|
|
|
|
return NS_ERROR_UNEXPECTED;
|
|
|
|
|
|
|
|
// Retrieve the app path, title, description and optional command line args.
|
|
|
|
nsAutoString appPath, appTitle, appDescription, appArgs;
|
2012-08-22 15:56:38 +00:00
|
|
|
int32_t appIconIndex = 0;
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
// Path
|
|
|
|
nsCOMPtr<nsIFile> executable;
|
|
|
|
handlerApp->GetExecutable(getter_AddRefs(executable));
|
|
|
|
|
2012-06-06 02:08:30 +00:00
|
|
|
rv = executable->GetPath(appPath);
|
2009-10-06 21:42:45 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
// Command line parameters
|
2012-08-22 15:56:38 +00:00
|
|
|
uint32_t count = 0;
|
2009-10-06 21:42:45 +00:00
|
|
|
handlerApp->GetParameterCount(&count);
|
2012-08-22 15:56:38 +00:00
|
|
|
for (uint32_t idx = 0; idx < count; idx++) {
|
2009-10-06 21:42:45 +00:00
|
|
|
if (idx > 0)
|
|
|
|
appArgs.Append(NS_LITERAL_STRING(" "));
|
|
|
|
nsAutoString param;
|
|
|
|
rv = handlerApp->GetParameter(idx, param);
|
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return rv;
|
|
|
|
appArgs.Append(param);
|
|
|
|
}
|
|
|
|
|
|
|
|
handlerApp->GetName(appTitle);
|
|
|
|
handlerApp->GetDetailedDescription(appDescription);
|
2011-09-06 19:11:28 +00:00
|
|
|
|
2011-09-29 06:19:26 +00:00
|
|
|
bool useUriIcon = false; // if we want to use the URI icon
|
|
|
|
bool usedUriIcon = false; // if we did use the URI icon
|
2009-10-06 21:42:45 +00:00
|
|
|
shortcut->GetIconIndex(&appIconIndex);
|
2011-09-06 19:11:28 +00:00
|
|
|
|
|
|
|
nsCOMPtr<nsIURI> iconUri;
|
2011-09-06 19:11:28 +00:00
|
|
|
rv = shortcut->GetFaviconPageUri(getter_AddRefs(iconUri));
|
2011-09-06 19:11:28 +00:00
|
|
|
if (NS_SUCCEEDED(rv) && iconUri) {
|
2011-10-02 02:16:19 +00:00
|
|
|
useUriIcon = true;
|
2011-09-06 19:11:28 +00:00
|
|
|
}
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
// Store the title of the app
|
|
|
|
if (appTitle.Length() > 0) {
|
2012-07-30 14:20:58 +00:00
|
|
|
IPropertyStore* pPropStore = nullptr;
|
2009-10-06 21:42:45 +00:00
|
|
|
hr = psl->QueryInterface(IID_IPropertyStore, (LPVOID*)&pPropStore);
|
|
|
|
if (FAILED(hr))
|
|
|
|
return NS_ERROR_UNEXPECTED;
|
|
|
|
|
|
|
|
PROPVARIANT pv;
|
|
|
|
InitPropVariantFromString(appTitle.get(), &pv);
|
|
|
|
|
|
|
|
pPropStore->SetValue(PKEY_Title, pv);
|
|
|
|
pPropStore->Commit();
|
|
|
|
pPropStore->Release();
|
|
|
|
|
|
|
|
PropVariantClear(&pv);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Store the rest of the params
|
|
|
|
psl->SetPath(appPath.get());
|
|
|
|
psl->SetDescription(appDescription.get());
|
|
|
|
psl->SetArguments(appArgs.get());
|
2011-09-06 19:11:28 +00:00
|
|
|
|
|
|
|
if (useUriIcon) {
|
|
|
|
nsString icoFilePath;
|
2012-07-21 20:07:26 +00:00
|
|
|
rv = mozilla::widget::FaviconHelper::ObtainCachedIconFile(iconUri,
|
|
|
|
icoFilePath,
|
|
|
|
aIOThread,
|
|
|
|
false);
|
2011-09-06 19:11:28 +00:00
|
|
|
if (NS_SUCCEEDED(rv)) {
|
|
|
|
// Always use the first icon in the ICO file
|
|
|
|
// our encoded icon only has 1 resource
|
|
|
|
psl->SetIconLocation(icoFilePath.get(), 0);
|
2011-10-02 02:16:19 +00:00
|
|
|
usedUriIcon = true;
|
2011-09-06 19:11:28 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// We didn't use an ICO via URI so fall back to the app icon
|
|
|
|
if (!usedUriIcon) {
|
|
|
|
psl->SetIconLocation(appPath.get(), appIconIndex);
|
|
|
|
}
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
aShellLink = dont_AddRef(psl);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2011-09-06 19:11:28 +00:00
|
|
|
// If successful fills in the aSame parameter
|
|
|
|
// aSame will be true if the path is in our icon cache
|
|
|
|
static nsresult IsPathInOurIconCache(nsCOMPtr<nsIJumpListShortcut>& aShortcut,
|
2011-09-29 06:19:26 +00:00
|
|
|
PRUnichar *aPath, bool *aSame)
|
2011-09-06 19:11:28 +00:00
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_POINTER(aPath);
|
|
|
|
NS_ENSURE_ARG_POINTER(aSame);
|
|
|
|
|
2011-10-02 02:16:19 +00:00
|
|
|
*aSame = false;
|
2011-09-06 19:11:28 +00:00
|
|
|
|
|
|
|
// Construct the path of our jump list cache
|
|
|
|
nsCOMPtr<nsIFile> jumpListCache;
|
|
|
|
nsresult rv = NS_GetSpecialDirectory("ProfLDS", getter_AddRefs(jumpListCache));
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
2012-07-21 20:07:26 +00:00
|
|
|
rv = jumpListCache->AppendNative(nsDependentCString(FaviconHelper::kJumpListCacheDir));
|
2011-09-06 19:11:28 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
nsAutoString jumpListCachePath;
|
|
|
|
rv = jumpListCache->GetPath(jumpListCachePath);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
// Construct the parent path of the passed in path
|
2012-06-06 02:08:30 +00:00
|
|
|
nsCOMPtr<nsIFile> passedInFile = do_CreateInstance("@mozilla.org/file/local;1");
|
2011-09-06 19:11:28 +00:00
|
|
|
NS_ENSURE_TRUE(passedInFile, NS_ERROR_FAILURE);
|
|
|
|
nsAutoString passedInPath(aPath);
|
|
|
|
rv = passedInFile->InitWithPath(passedInPath);
|
|
|
|
nsCOMPtr<nsIFile> passedInParentFile;
|
|
|
|
passedInFile->GetParent(getter_AddRefs(passedInParentFile));
|
|
|
|
nsAutoString passedInParentPath;
|
|
|
|
rv = jumpListCache->GetPath(passedInParentPath);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
*aSame = jumpListCachePath.Equals(passedInParentPath);
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
2009-10-06 21:42:45 +00:00
|
|
|
// (static) For a given IShellLink, create and return a populated nsIJumpListShortcut.
|
|
|
|
nsresult JumpListShortcut::GetJumpListShortcut(IShellLinkW *pLink, nsCOMPtr<nsIJumpListShortcut>& aShortcut)
|
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_POINTER(pLink);
|
|
|
|
|
|
|
|
nsresult rv;
|
|
|
|
HRESULT hres;
|
|
|
|
|
|
|
|
nsCOMPtr<nsILocalHandlerApp> handlerApp =
|
|
|
|
do_CreateInstance(NS_LOCALHANDLERAPP_CONTRACTID, &rv);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
PRUnichar buf[MAX_PATH];
|
|
|
|
|
|
|
|
// Path
|
|
|
|
hres = pLink->GetPath((LPWSTR)&buf, MAX_PATH, NULL, SLGP_UNCPRIORITY);
|
|
|
|
if (FAILED(hres))
|
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
|
2012-06-06 02:08:30 +00:00
|
|
|
nsCOMPtr<nsIFile> file;
|
2009-10-06 21:42:45 +00:00
|
|
|
nsDependentString filepath(buf);
|
2011-10-02 02:16:19 +00:00
|
|
|
rv = NS_NewLocalFile(filepath, false, getter_AddRefs(file));
|
2009-10-06 21:42:45 +00:00
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
rv = handlerApp->SetExecutable(file);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
// Parameters
|
|
|
|
hres = pLink->GetArguments((LPWSTR)&buf, MAX_PATH);
|
|
|
|
if (SUCCEEDED(hres)) {
|
|
|
|
LPWSTR *arglist;
|
2012-08-22 15:56:38 +00:00
|
|
|
int32_t numArgs;
|
|
|
|
int32_t idx;
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
arglist = ::CommandLineToArgvW(buf, &numArgs);
|
|
|
|
if(arglist) {
|
|
|
|
for (idx = 0; idx < numArgs; idx++) {
|
|
|
|
// szArglist[i] is null terminated
|
|
|
|
nsDependentString arg(arglist[idx]);
|
|
|
|
handlerApp->AppendParameter(arg);
|
|
|
|
}
|
|
|
|
::LocalFree(arglist);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
rv = aShortcut->SetApp(handlerApp);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
// Icon index or file location
|
|
|
|
int iconIdx = 0;
|
|
|
|
hres = pLink->GetIconLocation((LPWSTR)&buf, MAX_PATH, &iconIdx);
|
|
|
|
if (SUCCEEDED(hres)) {
|
|
|
|
// XXX How do we handle converting local files to images here? Do we need to?
|
|
|
|
aShortcut->SetIconIndex(iconIdx);
|
2011-09-06 19:11:28 +00:00
|
|
|
|
|
|
|
// Obtain the local profile directory and construct the output icon file path
|
|
|
|
// We only set the Icon Uri if we're sure it was from our icon cache.
|
2011-09-29 06:19:26 +00:00
|
|
|
bool isInOurCache;
|
2011-09-06 19:11:28 +00:00
|
|
|
if (NS_SUCCEEDED(IsPathInOurIconCache(aShortcut, buf, &isInOurCache)) &&
|
|
|
|
isInOurCache) {
|
|
|
|
nsCOMPtr<nsIURI> iconUri;
|
|
|
|
nsAutoString path(buf);
|
|
|
|
rv = NS_NewURI(getter_AddRefs(iconUri), path);
|
|
|
|
if (NS_SUCCEEDED(rv)) {
|
2011-09-06 19:11:28 +00:00
|
|
|
aShortcut->SetFaviconPageUri(iconUri);
|
2011-09-06 19:11:28 +00:00
|
|
|
}
|
|
|
|
}
|
2009-10-06 21:42:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Do we need the title and description? Probably not since handler app doesn't compare
|
|
|
|
// these in equals.
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
// (static) ShellItems are used to encapsulate links to things. We currently only support URI links,
|
|
|
|
// but more support could be added, such as local file and directory links.
|
|
|
|
nsresult JumpListLink::GetShellItem(nsCOMPtr<nsIJumpListItem>& item, nsRefPtr<IShellItem2>& aShellItem)
|
|
|
|
{
|
2012-07-30 14:20:58 +00:00
|
|
|
IShellItem2 *psi = nullptr;
|
2009-10-06 21:42:45 +00:00
|
|
|
nsresult rv;
|
|
|
|
|
2012-08-22 15:56:38 +00:00
|
|
|
int16_t type;
|
2009-10-06 21:42:45 +00:00
|
|
|
if (NS_FAILED(item->GetType(&type)))
|
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
|
|
|
|
if (type != nsIJumpListItem::JUMPLIST_ITEM_LINK)
|
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
|
|
|
|
nsCOMPtr<nsIJumpListLink> link = do_QueryInterface(item, &rv);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
nsCOMPtr<nsIURI> uri;
|
|
|
|
rv = link->GetUri(getter_AddRefs(uri));
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
2012-09-02 02:35:17 +00:00
|
|
|
nsAutoCString spec;
|
2009-10-06 21:42:45 +00:00
|
|
|
rv = uri->GetSpec(spec);
|
|
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
|
|
|
|
// Create the IShellItem
|
2012-01-11 02:09:41 +00:00
|
|
|
if (FAILED(WinUtils::SHCreateItemFromParsingName(
|
|
|
|
NS_ConvertASCIItoUTF16(spec).get(), NULL, IID_PPV_ARGS(&psi)))) {
|
2009-10-06 21:42:45 +00:00
|
|
|
return NS_ERROR_INVALID_ARG;
|
2012-01-11 02:09:41 +00:00
|
|
|
}
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
// Set the title
|
|
|
|
nsAutoString linkTitle;
|
|
|
|
link->GetUriTitle(linkTitle);
|
|
|
|
|
2012-07-30 14:20:58 +00:00
|
|
|
IPropertyStore* pPropStore = nullptr;
|
2009-10-06 21:42:45 +00:00
|
|
|
HRESULT hres = psi->GetPropertyStore(GPS_DEFAULT, IID_IPropertyStore, (void**)&pPropStore);
|
|
|
|
if (FAILED(hres))
|
|
|
|
return NS_ERROR_UNEXPECTED;
|
|
|
|
|
|
|
|
PROPVARIANT pv;
|
|
|
|
InitPropVariantFromString(linkTitle.get(), &pv);
|
|
|
|
|
|
|
|
// May fail due to shell item access permissions.
|
|
|
|
pPropStore->SetValue(PKEY_ItemName, pv);
|
|
|
|
pPropStore->Commit();
|
|
|
|
pPropStore->Release();
|
|
|
|
|
|
|
|
PropVariantClear(&pv);
|
|
|
|
|
|
|
|
aShellItem = dont_AddRef(psi);
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
// (static) For a given IShellItem, create and return a populated nsIJumpListLink.
|
|
|
|
nsresult JumpListLink::GetJumpListLink(IShellItem *pItem, nsCOMPtr<nsIJumpListLink>& aLink)
|
|
|
|
{
|
|
|
|
NS_ENSURE_ARG_POINTER(pItem);
|
|
|
|
|
|
|
|
// We assume for now these are URI links, but through properties we could
|
|
|
|
// query and create other types.
|
|
|
|
nsresult rv;
|
|
|
|
LPWSTR lpstrName = NULL;
|
|
|
|
|
|
|
|
if (SUCCEEDED(pItem->GetDisplayName(SIGDN_URL, &lpstrName))) {
|
|
|
|
nsCOMPtr<nsIURI> uri;
|
|
|
|
nsAutoString spec(lpstrName);
|
|
|
|
|
|
|
|
rv = NS_NewURI(getter_AddRefs(uri), NS_ConvertUTF16toUTF8(spec));
|
|
|
|
if (NS_FAILED(rv))
|
|
|
|
return NS_ERROR_INVALID_ARG;
|
|
|
|
|
|
|
|
aLink->SetUri(uri);
|
|
|
|
|
|
|
|
::CoTaskMemFree(lpstrName);
|
|
|
|
}
|
|
|
|
|
|
|
|
return NS_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Confirm the app is on the system
|
2011-09-29 06:19:26 +00:00
|
|
|
bool JumpListShortcut::ExecutableExists(nsCOMPtr<nsILocalHandlerApp>& handlerApp)
|
2009-10-06 21:42:45 +00:00
|
|
|
{
|
|
|
|
nsresult rv;
|
|
|
|
|
|
|
|
if (!handlerApp)
|
2011-10-02 02:16:19 +00:00
|
|
|
return false;
|
2009-10-06 21:42:45 +00:00
|
|
|
|
|
|
|
nsCOMPtr<nsIFile> executable;
|
|
|
|
rv = handlerApp->GetExecutable(getter_AddRefs(executable));
|
|
|
|
if (NS_SUCCEEDED(rv) && executable) {
|
2011-09-29 06:19:26 +00:00
|
|
|
bool exists;
|
2009-10-06 21:42:45 +00:00
|
|
|
executable->Exists(&exists);
|
|
|
|
return exists;
|
|
|
|
}
|
2011-10-02 02:16:19 +00:00
|
|
|
return false;
|
2009-10-06 21:42:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace widget
|
|
|
|
} // namespace mozilla
|
|
|
|
|