Mozilla/mozilla/content/base/src/nsContentAreaDragDrop.cpp
2004-01-09 23:54:21 +00:00

1591 lines
50 KiB
C++

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* ***** 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 Mozilla Communicator.
*
* The Initial Developer of the Original Code is
* Netscape Communications.
* Portions created by the Initial Developer are Copyright (C) 2002
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
* Mike Pinkerton <pinkerton@netscape.com>
*
* 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 "nsReadableUtils.h"
// Local Includes
#include "nsContentAreaDragDrop.h"
// Helper Classes
#include "nsString.h"
// Interfaces needed to be included
#include "nsIDOMNSUIEvent.h"
#include "nsIDOMUIEvent.h"
#include "nsISelection.h"
#include "nsIDOMNode.h"
#include "nsIDOMNodeList.h"
#include "nsIDOMEventReceiver.h"
#include "nsIDOMEvent.h"
#include "nsIDOMNSEvent.h"
#include "nsIDOMMouseEvent.h"
#include "nsIDOMAbstractView.h"
#include "nsIDOMWindow.h"
#include "nsIDOMDocument.h"
#include "nsIDOMDocumentRange.h"
#include "nsIDOMRange.h"
#include "nsIDocumentEncoder.h"
#include "nsIFormControl.h"
#include "nsISelectionPrivate.h"
#include "nsIDOMHTMLLinkElement.h"
#include "nsIDOMHTMLAreaElement.h"
#include "nsIDOMHTMLImageElement.h"
#include "nsIDOMHTMLAnchorElement.h"
#include "nsIDOMHTMLBodyElement.h"
#include "nsIDOMHTMLHtmlElement.h"
#include "nsITransferable.h"
#include "nsIDragService.h"
#include "nsIDragSession.h"
#include "nsComponentManagerUtils.h"
#include "nsXPCOM.h"
#include "nsISupportsPrimitives.h"
#include "nsIServiceManagerUtils.h"
#include "nsNetUtil.h"
#include "nsIFile.h"
#include "nsIWebNavigation.h"
#include "nsIClipboardDragDropHooks.h"
#include "nsIClipboardDragDropHookList.h"
#include "nsIDocShell.h"
#include "nsIContent.h"
#include "nsIImageLoadingContent.h"
#include "nsIXMLContent.h"
#include "nsINameSpaceManager.h"
#include "nsUnicharUtils.h"
#include "nsHTMLAtoms.h"
#include "nsIURL.h"
#include "nsIImage.h"
#include "nsIDocument.h"
#include "nsIPresShell.h"
#include "nsIPresContext.h"
#include "nsIFrame.h"
#include "nsLayoutAtoms.h"
#include "imgIContainer.h"
#include "imgIRequest.h"
#include "gfxIImageFrame.h"
#include "nsIInterfaceRequestor.h"
#include "nsIDocumentEncoder.h"
#include "nsRange.h"
#include "nsIWebBrowserPersist.h"
#include "nsEscape.h"
// private clipboard data flavors for html copy, used by editor when pasting
#define kHTMLContext "text/_moz_htmlcontext"
#define kHTMLInfo "text/_moz_htmlinfo"
NS_IMPL_ADDREF(nsContentAreaDragDrop)
NS_IMPL_RELEASE(nsContentAreaDragDrop)
NS_INTERFACE_MAP_BEGIN(nsContentAreaDragDrop)
NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsISupports, nsIDOMDragListener)
NS_INTERFACE_MAP_ENTRY_AMBIGUOUS(nsIDOMEventListener, nsIDOMDragListener)
NS_INTERFACE_MAP_ENTRY(nsIDOMDragListener)
NS_INTERFACE_MAP_ENTRY(nsIFlavorDataProvider)
NS_INTERFACE_MAP_ENTRY(nsIDragDropHandler)
NS_INTERFACE_MAP_END
class nsTransferableFactory
{
public:
static nsresult CreateFromEvent(nsIDOMEvent* inMouseEvent, nsIFlavorDataProvider *inFlavorDataProvider,
nsITransferable** outTrans);
protected:
nsTransferableFactory(nsIDOMEvent* inMouseEvent, nsIFlavorDataProvider *inFlavorDataProvider);
nsresult Produce(nsITransferable** outTrans);
private:
nsresult ConvertStringsToTransferable(nsITransferable** outTrans);
static nsresult GetDraggableSelectionData(nsISelection* inSelection, nsIDOMNode* inRealTargetNode,
nsIDOMNode **outImageOrLinkNode, PRBool* outDragSelectedText);
static void FindParentLinkNode(nsIDOMNode* inNode, nsIDOMNode** outParent);
static void GetAnchorURL(nsIDOMNode* inNode, nsAString& outURL);
static void GetNodeString(nsIDOMNode* inNode, nsAString & outNodeString);
static nsresult GetImageFromDOMNode(nsIDOMNode* inNode, nsIImage** outImage);
static void CreateLinkText(const nsAString& inURL, const nsAString & inText,
nsAString& outLinkText);
static void FindFirstAnchor(nsIDOMNode* inNode, nsIDOMNode** outAnchor);
enum serializationMode {serializeAsText, serializeAsHTML};
// if inNode is null, use the selection from the window
static nsresult SerializeNodeOrSelection(serializationMode inMode, PRUint32 inFlags,
nsIDOMWindow* inWindow, nsIDOMNode* inNode, nsAString& outResultString,
nsAString& outHTMLContext, nsAString& outHTMLInfo);
PRBool mInstanceAlreadyUsed;
nsCOMPtr<nsIDOMEvent> mMouseEvent;
nsCOMPtr<nsIFlavorDataProvider> mFlavorDataProvider;
nsString mUrlString;
nsString mImageSourceString;
nsString mTitleString;
nsString mHtmlString; // will be filled automatically if you fill urlstring
nsString mContextString;
nsString mInfoString;
PRBool mIsAnchor;
nsCOMPtr<nsIImage> mImage;
};
//
// nsContentAreaDragDrop ctor
//
nsContentAreaDragDrop::nsContentAreaDragDrop ( )
: mListenerInstalled(PR_FALSE), mNavigator(nsnull)
{
} // ctor
//
// ChromeTooltipListener dtor
//
nsContentAreaDragDrop::~nsContentAreaDragDrop ( )
{
RemoveDragListener();
} // dtor
NS_IMETHODIMP
nsContentAreaDragDrop::HookupTo(nsIDOMEventTarget *inAttachPoint, nsIWebNavigation* inNavigator)
{
NS_ASSERTION(inAttachPoint, "Can't hookup Drag Listeners to NULL receiver");
mEventReceiver = do_QueryInterface(inAttachPoint);
NS_ASSERTION(mEventReceiver, "Target doesn't implement nsIDOMEventReceiver as needed");
mNavigator = inNavigator;
return AddDragListener();
}
NS_IMETHODIMP
nsContentAreaDragDrop::Detach()
{
return RemoveDragListener();
}
//
// AddDragListener
//
// Subscribe to the events that will allow us to track drags.
//
nsresult
nsContentAreaDragDrop::AddDragListener()
{
nsresult rv = NS_ERROR_FAILURE;
if ( mEventReceiver ) {
nsIDOMDragListener *pListener = NS_STATIC_CAST(nsIDOMDragListener *, this);
rv = mEventReceiver->AddEventListenerByIID(pListener, NS_GET_IID(nsIDOMDragListener));
if (NS_SUCCEEDED(rv))
mListenerInstalled = PR_TRUE;
}
return rv;
}
//
// RemoveDragListener
//
// Unsubscribe from all the various drag events that we were listening to.
//
nsresult
nsContentAreaDragDrop::RemoveDragListener()
{
nsresult rv = NS_ERROR_FAILURE;
if (mEventReceiver) {
nsIDOMDragListener *pListener = NS_STATIC_CAST(nsIDOMDragListener *, this);
rv = mEventReceiver->RemoveEventListenerByIID(pListener, NS_GET_IID(nsIDOMDragListener));
if (NS_SUCCEEDED(rv))
mListenerInstalled = PR_FALSE;
mEventReceiver = nsnull;
}
return rv;
}
//
// DragEnter
//
// Called when an OS drag is in process and the mouse enters a gecko window. We
// don't care so much about dragEnters.
//
NS_IMETHODIMP
nsContentAreaDragDrop::DragEnter(nsIDOMEvent* aMouseEvent)
{
// nothing really to do here.
return NS_OK;
}
//
// DragOver
//
// Called when an OS drag is in process and the mouse is over a gecko window.
// The main purpose of this routine is to set the |canDrop| property on the
// drag session to false if we want to disallow the drop so that the OS can
// provide the appropriate feedback. All this does is show feedback, it
// doesn't actually cancel the drop; that comes later.
//
NS_IMETHODIMP
nsContentAreaDragDrop::DragOver(nsIDOMEvent* inEvent)
{
// first check that someone hasn't already handled this event
PRBool preventDefault = PR_TRUE;
nsCOMPtr<nsIDOMNSUIEvent> nsuiEvent(do_QueryInterface(inEvent));
if ( nsuiEvent )
nsuiEvent->GetPreventDefault(&preventDefault);
if ( preventDefault )
return NS_OK;
// if the drag originated w/in this content area, bail
// early. This avoids loading a URL dragged from the content
// area into the very same content area (which is almost never
// the desired action). This code is a bit too simplistic and
// may have problems with nested frames.
nsCOMPtr<nsIDragService> dragService(do_GetService("@mozilla.org/widget/dragservice;1"));
if ( !dragService )
return NS_ERROR_FAILURE;
nsCOMPtr<nsIDragSession> session;
dragService->GetCurrentSession(getter_AddRefs(session));
if ( session ) {
// if the client has provided an override callback, check if we
// the drop is allowed. If it allows it, we should still protect against
// dropping w/in the same document.
PRBool dropAllowed = PR_TRUE;
nsCOMPtr<nsISimpleEnumerator> enumerator;
GetHookEnumeratorFromEvent(inEvent, getter_AddRefs(enumerator));
if (enumerator) {
PRBool hasMoreHooks = PR_FALSE;
while (NS_SUCCEEDED(enumerator->HasMoreElements(&hasMoreHooks))
&& hasMoreHooks) {
nsCOMPtr<nsISupports> isupp;
if (NS_FAILED(enumerator->GetNext(getter_AddRefs(isupp))))
break;
nsCOMPtr<nsIClipboardDragDropHooks> override = do_QueryInterface(isupp);
if (override) {
#ifdef DEBUG
nsresult hookResult =
#endif
override->AllowDrop(inEvent, session, &dropAllowed);
NS_ASSERTION(NS_SUCCEEDED(hookResult), "hook failure in AllowDrop");
if (!dropAllowed)
break;
}
}
}
nsCOMPtr<nsIDOMDocument> sourceDoc;
session->GetSourceDocument(getter_AddRefs(sourceDoc));
nsCOMPtr<nsIDOMDocument> eventDoc;
GetEventDocument(inEvent, getter_AddRefs(eventDoc));
if ( sourceDoc == eventDoc )
dropAllowed = PR_FALSE;
session->SetCanDrop(dropAllowed);
}
return NS_OK;
}
//
// DragExit
//
// Called when an OS drag is in process and the mouse is over a gecko window. We
// don't care so much about dragExits.
//
NS_IMETHODIMP
nsContentAreaDragDrop::DragExit(nsIDOMEvent* aMouseEvent)
{
// nothing really to do here.
return NS_OK;
}
//
// ExtractURLFromData
//
// build up a url from whatever data we get from the OS. How we interpret the
// data depends on the flavor as it tells us the nsISupports* primitive type
// we have.
//
void
nsContentAreaDragDrop::ExtractURLFromData(const nsACString & inFlavor, nsISupports* inDataWrapper, PRUint32 inDataLen,
nsAString & outURL)
{
if ( !inDataWrapper )
return;
outURL.Truncate();
if ( inFlavor.Equals(kUnicodeMime) || inFlavor.Equals(kURLDataMime) ) {
// the data is regular unicode, just go with what we get. It may be a url, it
// may not be. *shrug*
nsCOMPtr<nsISupportsString> stringData(do_QueryInterface(inDataWrapper));
if ( stringData ) {
stringData->GetData(outURL);
}
}
else if ( inFlavor.Equals(kURLMime) ) {
// the data is an internet shortcut of the form <url>\n<title>. Strip
// out the url piece and return that.
nsCOMPtr<nsISupportsString> stringData(do_QueryInterface(inDataWrapper));
if ( stringData ) {
nsAutoString data;
stringData->GetData(data);
PRInt32 separator = data.FindChar('\n');
if ( separator >= 0 )
outURL = Substring(data, 0, separator);
else
outURL = data;
}
}
else if ( inFlavor.Equals(kFileMime) ) {
// the data is a file. Use the necko parsing utils to get a file:// url
// from the OS data.
nsCOMPtr<nsIFile> file(do_QueryInterface(inDataWrapper));
if ( file ) {
nsCAutoString url;
NS_GetURLSpecFromFile(file, url);
CopyUTF8toUTF16(url, outURL);
}
}
}
//
// DragDrop
//
// Called when an OS drag is in process and the mouse is released a gecko window.
// Extract the data from the OS and do something with it.
//
NS_IMETHODIMP
nsContentAreaDragDrop::DragDrop(nsIDOMEvent* inMouseEvent)
{
// if we don't have a nsIWebNavigation object to do anything with,
// just bail. The client will have to have another way to deal with it
if ( !mNavigator )
return NS_OK;
// check that someone hasn't already handled this event
PRBool preventDefault = PR_TRUE;
nsCOMPtr<nsIDOMNSUIEvent> nsuiEvent(do_QueryInterface(inMouseEvent));
if ( nsuiEvent )
nsuiEvent->GetPreventDefault(&preventDefault);
if ( preventDefault )
return NS_OK;
// pull the transferable out of the drag service. at the moment, we
// only care about the first item of the drag. We don't allow dropping
// multiple items into a content area.
nsCOMPtr<nsIDragService> dragService(do_GetService("@mozilla.org/widget/dragservice;1"));
if ( !dragService )
return NS_ERROR_FAILURE;
nsCOMPtr<nsIDragSession> session;
dragService->GetCurrentSession(getter_AddRefs(session));
if ( !session )
return NS_ERROR_FAILURE;
nsCOMPtr<nsITransferable> trans(do_CreateInstance("@mozilla.org/widget/transferable;1"));
if ( !trans )
return NS_ERROR_FAILURE;
// add the relevant flavors. order is important (highest fidelity to lowest)
trans->AddDataFlavor(kURLDataMime);
trans->AddDataFlavor(kURLMime);
trans->AddDataFlavor(kFileMime);
trans->AddDataFlavor(kUnicodeMime);
nsresult rv = session->GetData(trans, 0); // again, we only care about the first object
if ( NS_SUCCEEDED(rv) ) {
// if the client has provided an override callback, call it. It may
// still return that we should continue processing.
nsCOMPtr<nsISimpleEnumerator> enumerator;
GetHookEnumeratorFromEvent(inMouseEvent, getter_AddRefs(enumerator));
if (enumerator) {
PRBool actionCanceled = PR_TRUE;
PRBool hasMoreHooks = PR_FALSE;
while (NS_SUCCEEDED(enumerator->HasMoreElements(&hasMoreHooks))
&& hasMoreHooks) {
nsCOMPtr<nsISupports> isupp;
if (NS_FAILED(enumerator->GetNext(getter_AddRefs(isupp))))
break;
nsCOMPtr<nsIClipboardDragDropHooks> override = do_QueryInterface(isupp);
if (override) {
#ifdef DEBUG
nsresult hookResult =
#endif
override->OnPasteOrDrop(inMouseEvent, trans, &actionCanceled);
NS_ASSERTION(NS_SUCCEEDED(hookResult), "hook failure in OnPasteOrDrop");
if (!actionCanceled)
return NS_OK;
}
}
}
nsXPIDLCString flavor;
nsCOMPtr<nsISupports> dataWrapper;
PRUint32 dataLen = 0;
rv = trans->GetAnyTransferData(getter_Copies(flavor), getter_AddRefs(dataWrapper), &dataLen);
if ( NS_SUCCEEDED(rv) && dataLen > 0 ) {
// get the url from one of several possible formats
nsAutoString url;
ExtractURLFromData(flavor, dataWrapper, dataLen, url);
NS_ASSERTION(!url.IsEmpty(), "Didn't get anything we can use as a url");
// valid urls don't have spaces. bail if this does.
if ( url.IsEmpty() || url.FindChar(' ') >= 0 )
return NS_OK;
// ok, we have the url, load it.
mNavigator->LoadURI(url.get(), nsIWebNavigation::LOAD_FLAGS_NONE, nsnull, nsnull, nsnull);
}
}
return NS_OK;
}
//
// NormalizeSelection
//
void
nsContentAreaDragDrop::NormalizeSelection(nsIDOMNode* inBaseNode, nsISelection* inSelection)
{
nsCOMPtr<nsIDOMNode> parent;
inBaseNode->GetParentNode(getter_AddRefs(parent));
if ( !parent || !inSelection )
return;
nsCOMPtr<nsIDOMNodeList> childNodes;
parent->GetChildNodes(getter_AddRefs(childNodes));
if ( !childNodes )
return;
PRUint32 listLen = 0;
childNodes->GetLength(&listLen);
PRUint32 index = 0;
for ( ; index < listLen; ++index ) {
nsCOMPtr<nsIDOMNode> indexedNode;
childNodes->Item(index, getter_AddRefs(indexedNode));
if ( indexedNode == inBaseNode )
break;
}
if ( index >= listLen )
return;
// now make the selection contain all of |inBaseNode|'s siblings up to and including
// |inBaseNode|
inSelection->Collapse(parent, index);
inSelection->Extend(parent, index+1);
}
//
// GetEventDocument
//
// Get the DOM document associated with a given DOM event
//
void
nsContentAreaDragDrop::GetEventDocument(nsIDOMEvent* inEvent, nsIDOMDocument** outDocument)
{
if ( !outDocument )
return;
*outDocument = nsnull;
nsCOMPtr<nsIDOMUIEvent> uiEvent(do_QueryInterface(inEvent));
if ( uiEvent ) {
nsCOMPtr<nsIDOMAbstractView> view;
uiEvent->GetView(getter_AddRefs(view));
nsCOMPtr<nsIDOMWindow> window(do_QueryInterface(view));
if ( window )
window->GetDocument(outDocument);
}
}
nsresult
nsContentAreaDragDrop::CreateTransferable(nsIDOMEvent* inMouseEvent, nsITransferable** outTrans)
{
return nsTransferableFactory::CreateFromEvent(inMouseEvent,
NS_STATIC_CAST(nsIFlavorDataProvider*, this),
outTrans);
}
nsresult
nsContentAreaDragDrop::GetHookEnumeratorFromEvent(nsIDOMEvent* inEvent,
nsISimpleEnumerator **outEnumerator)
{
*outEnumerator = nsnull;
nsCOMPtr<nsIDOMDocument> domdoc;
GetEventDocument(inEvent, getter_AddRefs(domdoc));
nsCOMPtr<nsIDocument> doc = do_QueryInterface(domdoc);
NS_ENSURE_TRUE(doc, NS_ERROR_FAILURE);
nsCOMPtr<nsISupports> container = doc->GetContainer();
nsCOMPtr<nsIDocShell> docShell = do_QueryInterface(container);
NS_ENSURE_TRUE(docShell, NS_ERROR_FAILURE);
nsCOMPtr<nsIClipboardDragDropHookList> hookList = do_GetInterface(docShell);
NS_ENSURE_TRUE(hookList, NS_ERROR_FAILURE);
nsCOMPtr<nsISimpleEnumerator> enumerator;
hookList->GetHookEnumerator(getter_AddRefs(enumerator));
NS_ENSURE_TRUE(enumerator, NS_ERROR_FAILURE);
*outEnumerator = enumerator;
NS_ADDREF(*outEnumerator);
return NS_OK;
}
//
// DragGesture
//
// Determine if the user has started to drag something and kick off
// an OS-level drag if it's applicable
//
NS_IMETHODIMP
nsContentAreaDragDrop::DragGesture(nsIDOMEvent* inMouseEvent)
{
// first check that someone hasn't already handled this event
PRBool preventDefault = PR_TRUE;
nsCOMPtr<nsIDOMNSUIEvent> nsuiEvent(do_QueryInterface(inMouseEvent));
if ( nsuiEvent )
nsuiEvent->GetPreventDefault(&preventDefault);
if ( preventDefault )
return NS_OK;
// if the client has provided an override callback, check if we
// should continue
nsCOMPtr<nsISimpleEnumerator> enumerator;
GetHookEnumeratorFromEvent(inMouseEvent, getter_AddRefs(enumerator));
if (enumerator) {
PRBool allow = PR_TRUE;
PRBool hasMoreHooks = PR_FALSE;
while (NS_SUCCEEDED(enumerator->HasMoreElements(&hasMoreHooks))
&& hasMoreHooks) {
nsCOMPtr<nsISupports> isupp;
if (NS_FAILED(enumerator->GetNext(getter_AddRefs(isupp))))
break;
nsCOMPtr<nsIClipboardDragDropHooks> override = do_QueryInterface(isupp);
if (override) {
#ifdef DEBUG
nsresult hookResult =
#endif
override->AllowStartDrag(inMouseEvent, &allow);
NS_ASSERTION(NS_SUCCEEDED(hookResult), "hook failure in AllowStartDrag");
if (!allow)
return NS_OK;
}
}
}
nsCOMPtr<nsITransferable> trans;
nsresult rv = CreateTransferable(inMouseEvent, getter_AddRefs(trans));
if (NS_FAILED(rv))
return rv;
if ( trans ) {
// if the client has provided an override callback, let them manipulate
// the flavors or drag data
nsCOMPtr<nsISimpleEnumerator> enumerator;
GetHookEnumeratorFromEvent(inMouseEvent, getter_AddRefs(enumerator));
if (enumerator)
{
PRBool hasMoreHooks = PR_FALSE;
PRBool doContinueDrag = PR_TRUE;
while (NS_SUCCEEDED(enumerator->HasMoreElements(&hasMoreHooks))
&& hasMoreHooks)
{
nsCOMPtr<nsISupports> isupp;
if (NS_FAILED(enumerator->GetNext(getter_AddRefs(isupp))))
break;
nsCOMPtr<nsIClipboardDragDropHooks> override = do_QueryInterface(isupp);
if (override)
{
#ifdef DEBUG
nsresult hookResult =
#endif
override->OnCopyOrDrag(inMouseEvent, trans, &doContinueDrag);
NS_ASSERTION(NS_SUCCEEDED(hookResult), "hook failure in OnCopyOrDrag");
if (!doContinueDrag)
return NS_OK;
}
}
}
nsCOMPtr<nsISupportsArray> transArray(do_CreateInstance("@mozilla.org/supports-array;1"));
if ( !transArray )
return NS_ERROR_FAILURE;
transArray->InsertElementAt(trans, 0);
// kick off the drag
nsCOMPtr<nsIDOMEventTarget> target;
inMouseEvent->GetTarget(getter_AddRefs(target));
nsCOMPtr<nsIDOMNode> targetNode(do_QueryInterface(target));
nsCOMPtr<nsIDragService> dragService(do_GetService("@mozilla.org/widget/dragservice;1"));
if ( !dragService )
return NS_ERROR_FAILURE;
dragService->InvokeDragSession(targetNode, transArray, nsnull, nsIDragService::DRAGDROP_ACTION_COPY +
nsIDragService::DRAGDROP_ACTION_MOVE + nsIDragService::DRAGDROP_ACTION_LINK);
}
return NS_OK;
}
NS_IMETHODIMP
nsContentAreaDragDrop::HandleEvent(nsIDOMEvent *event)
{
return NS_OK;
}
#if 0
#pragma mark -
#endif
// SaveURIToFile
// used on platforms where it's possible to drag items (e.g. images)
// into the file system
nsresult
nsContentAreaDragDrop::SaveURIToFileInDirectory(nsAString& inSourceURIString, nsILocalFile* inDestDirectory, nsILocalFile** outFile)
{
*outFile = nsnull;
nsresult rv;
// clone it because it belongs to the drag data, so we shouldn't mess with it
nsCOMPtr<nsIFile> clonedFile;
rv = inDestDirectory->Clone(getter_AddRefs(clonedFile));
if (NS_FAILED(rv)) return rv;
nsCOMPtr<nsILocalFile> destFile = do_QueryInterface(clonedFile);
if (!destFile) return NS_ERROR_NO_INTERFACE;
nsCOMPtr<nsIURI> sourceURI;
rv = NS_NewURI(getter_AddRefs(sourceURI), inSourceURIString);
if (NS_FAILED(rv)) return NS_ERROR_FAILURE;
nsCOMPtr<nsIURL> sourceURL = do_QueryInterface(sourceURI);
if (!sourceURL) return NS_ERROR_NO_INTERFACE;
nsCAutoString fileName; // escaped, UTF-8
sourceURL->GetFileName(fileName);
if (fileName.IsEmpty())
return NS_ERROR_FAILURE; // this is an error; the URL must point to a file
NS_UnescapeURL(fileName);
NS_ConvertUTF8toUCS2 wideFileName(fileName);
// make the name safe for the filesystem
wideFileName.ReplaceChar(PRUnichar('/'), PRUnichar('_'));
wideFileName.ReplaceChar(PRUnichar('\\'), PRUnichar('_'));
wideFileName.ReplaceChar(PRUnichar(':'), PRUnichar('_'));
rv = destFile->Append(wideFileName);
if (NS_FAILED(rv)) return rv;
rv = destFile->CreateUnique(nsIFile::NORMAL_FILE_TYPE, 0600);
if (NS_FAILED(rv)) return rv;
// we rely on the fact that the WPB is refcounted by the channel etc,
// so we don't keep a ref to it. It will die when finished.
nsCOMPtr<nsIWebBrowserPersist> persist = do_CreateInstance("@mozilla.org/embedding/browser/nsWebBrowserPersist;1", &rv);
if (NS_FAILED(rv)) return rv;
nsCOMPtr<nsISupports> fileAsSupports = do_QueryInterface(destFile);
rv = persist->SaveURI(sourceURI, nsnull, nsnull, nsnull, nsnull, fileAsSupports);
if (NS_FAILED(rv)) return rv;
*outFile = destFile;
NS_ADDREF(*outFile);
return NS_OK;
}
// This is our nsIFlavorDataProvider callback. There are several assumptions here that
// make this work:
//
// 1. Someone put a kFilePromiseURLMime flavor into the transferable with the source
// URI of the file to save (as a string). We did that above.
//
// 2. Someone put a kFilePromiseDirectoryMime flavor into the transferable with
// an nsILocalFile for the directory we are to save in. That has to be done
// by platform-specific code (in widget), which gets the destination directory
// from OS-specific drag information.
//
NS_IMETHODIMP
nsContentAreaDragDrop::GetFlavorData(nsITransferable *aTransferable,
const char *aFlavor, nsISupports **aData, PRUint32 *aDataLen)
{
NS_ENSURE_ARG_POINTER(aData && aDataLen);
*aData = nsnull;
*aDataLen = 0;
nsresult rv = NS_ERROR_NOT_IMPLEMENTED;
if (strcmp(aFlavor, kFilePromiseMime) == 0)
{
// get the URI from the kFilePromiseURLMime flavor
NS_ENSURE_ARG(aTransferable);
nsCOMPtr<nsISupports> urlPrimitive;
PRUint32 dataSize = 0;
aTransferable->GetTransferData(kFilePromiseURLMime, getter_AddRefs(urlPrimitive), &dataSize);
nsCOMPtr<nsISupportsString> srcUrlPrimitive = do_QueryInterface(urlPrimitive);
if (!srcUrlPrimitive) return NS_ERROR_FAILURE;
nsAutoString sourceURLString;
srcUrlPrimitive->GetData(sourceURLString);
if (sourceURLString.IsEmpty())
return NS_ERROR_FAILURE;
// get the target directory from the kFilePromiseDirectoryMime flavor
nsCOMPtr<nsISupports> dirPrimitive;
dataSize = 0;
aTransferable->GetTransferData(kFilePromiseDirectoryMime, getter_AddRefs(dirPrimitive), &dataSize);
nsCOMPtr<nsILocalFile> destDirectory = do_QueryInterface(dirPrimitive);
if (!destDirectory) return NS_ERROR_FAILURE;
// now save the file
nsCOMPtr<nsILocalFile> destFile;
rv = SaveURIToFileInDirectory(sourceURLString, destDirectory, getter_AddRefs(destFile));
// send back an nsILocalFile
if (NS_SUCCEEDED(rv))
{
CallQueryInterface(destFile, aData);
*aDataLen = sizeof(nsILocalFile*);
}
}
return rv;
}
nsresult
nsTransferableFactory::CreateFromEvent(nsIDOMEvent* inMouseEvent, nsIFlavorDataProvider *inFlavorDataProvider,
nsITransferable** outTrans)
{
nsTransferableFactory factory(inMouseEvent, inFlavorDataProvider);
return factory.Produce(outTrans);
}
nsTransferableFactory::nsTransferableFactory(nsIDOMEvent* inMouseEvent, nsIFlavorDataProvider *inFlavorDataProvider)
:mInstanceAlreadyUsed(PR_FALSE),
mMouseEvent(inMouseEvent),
mFlavorDataProvider(inFlavorDataProvider)
{
}
//
// FindFirstAnchor
//
// A recursive routine that finds the first child link initiating anchor
//
void
nsTransferableFactory::FindFirstAnchor(nsIDOMNode* inNode, nsIDOMNode** outAnchor)
{
if ( !inNode && !outAnchor )
return;
*outAnchor = nsnull;
static NS_NAMED_LITERAL_STRING(simple, "simple");
nsCOMPtr<nsIDOMNode> curr = inNode;
while ( curr ) {
// check me (base case of recursion)
PRUint16 nodeType = 0;
curr->GetNodeType(&nodeType);
if ( nodeType == nsIDOMNode::ELEMENT_NODE ) {
// a?
nsCOMPtr<nsIDOMHTMLAnchorElement> a(do_QueryInterface(curr));
if (a) {
*outAnchor = curr;
NS_ADDREF(*outAnchor);
return;
}
// area?
nsCOMPtr<nsIDOMHTMLAreaElement> area(do_QueryInterface(curr));
if (area) {
*outAnchor = curr;
NS_ADDREF(*outAnchor);
return;
}
// Simple XLink?
nsCOMPtr<nsIContent> content(do_QueryInterface(curr));
NS_WARN_IF_FALSE(content, "DOM node is not content?");
if (!content)
return;
nsAutoString value;
content->GetAttr(kNameSpaceID_XLink, nsHTMLAtoms::type, value);
if (value.Equals(simple)) {
*outAnchor = curr;
NS_ADDREF(*outAnchor);
return;
}
}
// recursively check my children
nsCOMPtr<nsIDOMNode> firstChild;
curr->GetFirstChild(getter_AddRefs(firstChild));
FindFirstAnchor(firstChild, outAnchor);
if ( *outAnchor )
return;
// check my siblings
nsIDOMNode* temp = nsnull;
curr->GetNextSibling(&temp);
curr = dont_AddRef(temp);
}
}
//
// FindParentLinkNode
//
// Finds the parent with the given link tag starting at |inNode|. If it gets
// up to <body> or <html> or the top document w/out finding it, we
// stop looking and |outParent| will be null.
//
void
nsTransferableFactory::FindParentLinkNode(nsIDOMNode* inNode,
nsIDOMNode** outParent)
{
if ( !inNode || !outParent )
return;
*outParent = nsnull;
nsCOMPtr<nsIDOMNode> node(inNode); // to make refcounting easier
PRUint16 nodeType = 0;
inNode->GetNodeType(&nodeType);
if ( nodeType == nsIDOMNode::TEXT_NODE )
inNode->GetParentNode(getter_AddRefs(node));
static NS_NAMED_LITERAL_STRING(document, "#document");
static NS_NAMED_LITERAL_STRING(simple, "simple");
while ( node ) {
// (X)HTML body or html?
node->GetNodeType(&nodeType);
if ( nodeType == nsIDOMNode::ELEMENT_NODE ) {
// body?
nsCOMPtr<nsIDOMHTMLBodyElement> body(do_QueryInterface(node));
if (body) {
return;
}
// html?
nsCOMPtr<nsIDOMHTMLHtmlElement> html(do_QueryInterface(node));
if (html) {
return;
}
}
// Other document root?
nsAutoString localName;
node->GetLocalName(localName);
if ( localName.IsEmpty() )
return;
if ( localName.Equals(document, nsCaseInsensitiveStringComparator()) )
return; // XXX Check if #document always lower case
if ( nodeType == nsIDOMNode::ELEMENT_NODE ) {
PRBool found = PR_FALSE;
nsCOMPtr<nsIDOMHTMLAnchorElement> a(do_QueryInterface(node));
if (a) {
found = PR_TRUE;
} else {
nsCOMPtr<nsIContent> content(do_QueryInterface(node));
NS_WARN_IF_FALSE(content, "DOM node is not content?");
if (!content)
return;
nsAutoString value;
content->GetAttr(kNameSpaceID_XLink, nsHTMLAtoms::type, value);
if (value.Equals(simple)) {
found = PR_TRUE;
}
}
if (found) {
*outParent = node;
NS_ADDREF(*outParent);
return;
}
}
// keep going, up to parent
nsIDOMNode* temp;
node->GetParentNode(&temp);
node = dont_AddRef(temp);
}
}
//
// GetAnchorURL
//
// Get the url for this anchor. First try the href, and if that's empty,
// go for the name.
//
void
nsTransferableFactory::GetAnchorURL(nsIDOMNode* inNode, nsAString& outURL)
{
outURL.Truncate();
// a?
nsCOMPtr<nsIDOMHTMLAnchorElement> anchor(do_QueryInterface(inNode));
if ( anchor ) {
anchor->GetHref(outURL);
if ( outURL.IsEmpty() )
anchor->GetName(outURL);
} else {
// area?
nsCOMPtr<nsIDOMHTMLAreaElement> area(do_QueryInterface(inNode));
if ( area ) {
area->GetHref(outURL);
if ( outURL.IsEmpty() ) {
nsCOMPtr<nsIDOMHTMLElement> e(do_QueryInterface(inNode));
e->GetId(outURL);
}
} else {
// Try XLink next...
nsCOMPtr<nsIContent> content(do_QueryInterface(inNode));
nsAutoString value;
content->GetAttr(kNameSpaceID_XLink, nsHTMLAtoms::type, value);
if (value.Equals(NS_LITERAL_STRING("simple"))) {
content->GetAttr(kNameSpaceID_XLink, nsHTMLAtoms::href, value);
if (!value.IsEmpty()) {
nsCOMPtr<nsIURI> baseURI = content->GetBaseURI();
if (baseURI) {
nsCAutoString absoluteSpec;
baseURI->Resolve(NS_ConvertUCS2toUTF8(value), absoluteSpec);
CopyUTF8toUTF16(absoluteSpec, outURL);
}
}
} else {
// ... or just get the ID
nsCOMPtr<nsIXMLContent> xml(do_QueryInterface(inNode));
nsCOMPtr<nsIAtom> id;
if (xml && NS_SUCCEEDED(xml->GetID(getter_AddRefs(id))) && id) {
id->ToString(outURL);
}
}
}
}
}
//
// CreateLinkText
//
// Creates the html for an anchor in the form
// <a href="inURL">inText</a>
//
void
nsTransferableFactory::CreateLinkText(const nsAString& inURL, const nsAString & inText,
nsAString& outLinkText)
{
// use a temp var in case |inText| is the same string as |outLinkText| to
// avoid overwriting it while building up the string in pieces.
nsAutoString linkText(NS_LITERAL_STRING("<a href=\"") +
inURL +
NS_LITERAL_STRING("\">") +
inText +
NS_LITERAL_STRING("</a>") );
outLinkText = linkText;
}
//
// GetNodeString
//
// Gets the text associated with a node
//
void
nsTransferableFactory::GetNodeString(nsIDOMNode* inNode, nsAString & outNodeString)
{
outNodeString.Truncate();
// use a range to get the text-equivalent of the node
nsCOMPtr<nsIDOMDocument> doc;
inNode->GetOwnerDocument(getter_AddRefs(doc));
nsCOMPtr<nsIDOMDocumentRange> docRange(do_QueryInterface(doc));
if ( docRange ) {
nsCOMPtr<nsIDOMRange> range;
docRange->CreateRange(getter_AddRefs(range));
if ( range ) {
range->SelectNode(inNode);
range->ToString(outNodeString);
}
}
}
nsresult
nsTransferableFactory::Produce(nsITransferable** outTrans)
{
if (mInstanceAlreadyUsed)
return NS_ERROR_FAILURE;
if (!outTrans || !mMouseEvent || !mFlavorDataProvider)
return NS_ERROR_FAILURE;
mInstanceAlreadyUsed = PR_TRUE;
*outTrans = nsnull;
nsCOMPtr<nsIDOMWindow> window;
PRBool isAltKeyDown = PR_FALSE;
mIsAnchor = PR_FALSE;
{
nsCOMPtr<nsIDOMUIEvent> uiEvent(do_QueryInterface(mMouseEvent));
if (!uiEvent)
return NS_OK;
// find the selection to see what we could be dragging and if
// what we're dragging is in what is selected.
nsCOMPtr<nsIDOMAbstractView> view;
uiEvent->GetView(getter_AddRefs(view));
window = do_QueryInterface(view);
if (!window)
return NS_OK;
}
{
nsCOMPtr<nsIDOMMouseEvent> mouseEvent(do_QueryInterface(mMouseEvent));
if (mouseEvent)
mouseEvent->GetAltKey(&isAltKeyDown);
}
nsCOMPtr<nsISelection> selection;
window->GetSelection(getter_AddRefs(selection));
if (!selection)
return NS_OK;
nsCOMPtr<nsIDOMNode> nodeToSerialize; // if set, serialize the content under this node
PRBool useSelectedText = PR_FALSE;
{
PRBool haveSelectedContent = PR_FALSE;
nsCOMPtr<nsIDOMNode> parentLink; // possible parent link node
nsCOMPtr<nsIDOMNode> draggedNode;
{
nsCOMPtr<nsIDOMEventTarget> target;
mMouseEvent->GetTarget(getter_AddRefs(target));
// only drag form elements by using the alt key,
// otherwise buttons and select widgets are hard to use
nsCOMPtr<nsIFormControl> form(do_QueryInterface(target));
if (form && !isAltKeyDown)
return NS_OK;
draggedNode = do_QueryInterface(target);
}
nsCOMPtr<nsIDOMHTMLAreaElement> area; // client-side image map
nsCOMPtr<nsIDOMHTMLImageElement> image;
nsCOMPtr<nsIDOMHTMLAnchorElement> link;
{
// Get the real target and see if it is in the selection
nsCOMPtr<nsIDOMNode> realTargetNode;
{
nsCOMPtr<nsIDOMNSEvent> internalEvent = do_QueryInterface(mMouseEvent);
if (internalEvent)
{
nsCOMPtr<nsIDOMEventTarget> realTarget;
internalEvent->GetExplicitOriginalTarget(getter_AddRefs(realTarget));
realTargetNode = do_QueryInterface(realTarget);
}
}
{
nsCOMPtr<nsIDOMNode> selectedImageOrLinkNode;
GetDraggableSelectionData(selection, realTargetNode, getter_AddRefs(selectedImageOrLinkNode), &haveSelectedContent);
if (selectedImageOrLinkNode)
{
image = do_QueryInterface(selectedImageOrLinkNode);
link = do_QueryInterface(selectedImageOrLinkNode);
useSelectedText = !image && link;
}
else
{
// we're not using a selected element. Look for draggable elements
// under the mouse
// if the alt key is down, don't start a drag if we're in an anchor because
// we want to do selection.
FindParentLinkNode(draggedNode, getter_AddRefs(parentLink));
if (parentLink && isAltKeyDown)
return NS_OK;
area = do_QueryInterface(draggedNode);
image = do_QueryInterface(draggedNode);
link = do_QueryInterface(draggedNode);
if (haveSelectedContent)
useSelectedText = PR_TRUE;
}
}
}
{
nsCOMPtr<nsIDOMNode> linkNode; // set for linked images, and links
if (area)
{
// use the alt text (or, if missing, the href) as the title
area->GetAttribute(NS_LITERAL_STRING("alt"), mTitleString);
if (mTitleString.IsEmpty())
area->GetAttribute(NS_LITERAL_STRING("href"), mTitleString); // this can be a relative link
// we'll generate HTML like <a href="absurl">alt text</a>
mIsAnchor = PR_TRUE;
GetAnchorURL(area, mUrlString); // gives an absolute link
mHtmlString = NS_LITERAL_STRING("<a href=\"");
mHtmlString.Append(mUrlString);
mHtmlString.Append(NS_LITERAL_STRING("\">"));
mHtmlString.Append(mTitleString);
mHtmlString.Append(NS_LITERAL_STRING("</a>"));
}
else if (image)
{
mIsAnchor = PR_TRUE;
// grab the href as the url, use alt text as the title of the area if it's there.
// the drag data is the image tag and src attribute.
image->GetSrc(mUrlString);
image->GetAttribute(NS_LITERAL_STRING("alt"), mTitleString);
if (mTitleString.IsEmpty())
mTitleString = mUrlString;
// pass out the image source string
mImageSourceString = mUrlString;
// also grab the image data
GetImageFromDOMNode(draggedNode, getter_AddRefs(mImage));
if (parentLink)
{
// If we are dragging around an image in an anchor, then we
// are dragging the entire anchor
linkNode = parentLink;
nodeToSerialize = linkNode;
}
else
nodeToSerialize = draggedNode;
}
else if (link)
{
// set linkNode. The code below will handle this
linkNode = link; // XXX test this
GetNodeString(draggedNode, mTitleString);
}
else if (parentLink)
{
linkNode = parentLink;
nodeToSerialize = linkNode;
if (haveSelectedContent)
useSelectedText = PR_TRUE;
}
else if (!haveSelectedContent)
{
// nothing draggable
return NS_OK;
}
if (linkNode)
{
mIsAnchor = PR_TRUE;
GetAnchorURL(linkNode, mUrlString);
}
}
}
if (nodeToSerialize || useSelectedText)
{
// if we have selected text, use it in preference to the node
if (useSelectedText)
nodeToSerialize = nsnull;
SerializeNodeOrSelection(serializeAsHTML,
nsIDocumentEncoder::OutputAbsoluteLinks |
nsIDocumentEncoder::OutputEncodeW3CEntities,
window, nodeToSerialize,
mHtmlString, mContextString, mInfoString);
nsAutoString dummy1, dummy2;
SerializeNodeOrSelection(serializeAsText, 0,
window, nodeToSerialize,
mTitleString, dummy1, dummy2);
#ifdef CHANGE_SELECTION_ON_DRAG
// We used to change the selection to wrap the dragged node (mainly
// to work around now-fixed issues with dragging unselected elements).
// There is no reason to do this any more.
NormalizeSelection(selectionNormalizeNode, selection);
#endif
}
// default text value is the URL
if (mTitleString.IsEmpty())
mTitleString = mUrlString;
// if we haven't constructed a html version, make one now
if (mHtmlString.IsEmpty() && !mUrlString.IsEmpty())
CreateLinkText(mUrlString, mTitleString, mHtmlString);
return ConvertStringsToTransferable(outTrans);
}
nsresult
nsTransferableFactory::ConvertStringsToTransferable(nsITransferable** outTrans)
{
// now create the transferable and stuff data into it.
nsCOMPtr<nsITransferable> trans(do_CreateInstance("@mozilla.org/widget/transferable;1"));
if ( !trans )
return NS_ERROR_FAILURE;
// add a special flavor if we're an anchor to indicate that we have a URL
// in the drag data
if ( !mUrlString.IsEmpty() && mIsAnchor ) {
nsAutoString dragData ( mUrlString );
dragData += NS_LITERAL_STRING("\n");
dragData += mTitleString;
nsCOMPtr<nsISupportsString> urlPrimitive(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if ( !urlPrimitive )
return NS_ERROR_FAILURE;
urlPrimitive->SetData(dragData);
trans->SetTransferData(kURLMime, urlPrimitive, dragData.Length() * sizeof(PRUnichar));
nsCOMPtr<nsISupportsString> urlDataPrimitive(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if (!urlDataPrimitive)
return NS_ERROR_FAILURE;
urlDataPrimitive->SetData(mUrlString);
trans->SetTransferData(kURLDataMime, urlDataPrimitive, mUrlString.Length() * sizeof(PRUnichar));
nsCOMPtr<nsISupportsString> urlDescPrimitive(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if (!urlDescPrimitive)
return NS_ERROR_FAILURE;
urlDescPrimitive->SetData(mTitleString);
trans->SetTransferData(kURLDescriptionMime, urlDescPrimitive, mTitleString.Length() * sizeof(PRUnichar));
}
// add a special flavor, even if we don't have html context data
nsCOMPtr<nsISupportsString> context(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if ( !context )
return NS_ERROR_FAILURE;
nsAutoString contextData(mContextString);
context->SetData(contextData);
trans->SetTransferData(kHTMLContext, context, contextData.Length() * 2);
// add a special flavor if we have html info data
if ( !mInfoString.IsEmpty() ) {
nsCOMPtr<nsISupportsString> info(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if (!info)
return NS_ERROR_FAILURE;
nsAutoString infoData(mInfoString);
info->SetData(infoData);
trans->SetTransferData(kHTMLInfo, info, infoData.Length() * 2);
}
// add the full html
nsCOMPtr<nsISupportsString> htmlPrimitive(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if (!htmlPrimitive)
return NS_ERROR_FAILURE;
htmlPrimitive->SetData(mHtmlString);
trans->SetTransferData(kHTMLMime, htmlPrimitive, mHtmlString.Length() * sizeof(PRUnichar));
// add the plain (unicode) text. we use the url for text/unicode data if an anchor
// is being dragged, rather than the title text of the link or the alt text for
// an anchor image.
nsCOMPtr<nsISupportsString> textPrimitive(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if (!textPrimitive)
return NS_ERROR_FAILURE;
textPrimitive->SetData(mIsAnchor ? mUrlString : mTitleString);
trans->SetTransferData(kUnicodeMime, textPrimitive, (mIsAnchor ? mUrlString.Length() : mTitleString.Length()) * sizeof(PRUnichar));
// add image data, if present. For now, all we're going to do with this is turn it
// into a native data flavor, so indicate that with a new flavor so as not to confuse
// anyone who is really registered for image/gif or image/jpg.
if (mImage) {
nsCOMPtr<nsISupportsInterfacePointer> ptrPrimitive(do_CreateInstance(NS_SUPPORTS_INTERFACE_POINTER_CONTRACTID));
if ( !ptrPrimitive )
return NS_ERROR_FAILURE;
ptrPrimitive->SetData(mImage);
trans->SetTransferData(kNativeImageMime, ptrPrimitive, sizeof(nsIImage*));
// assume the image comes from a file, and add a file promise. We register ourselves
// as a nsIFlavorDataProvider, and will use the GetFlavorData callback to save the
// image to disk.
trans->SetTransferData(kFilePromiseMime, mFlavorDataProvider, nsITransferable::kFlavorHasDataProvider);
nsCOMPtr<nsISupportsString> imageUrlPrimitive(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if (!imageUrlPrimitive)
return NS_ERROR_FAILURE;
imageUrlPrimitive->SetData(mImageSourceString);
trans->SetTransferData(kFilePromiseURLMime, imageUrlPrimitive, mImageSourceString.Length() * sizeof(PRUnichar));
// if not an anchor, add the image url
if (!mIsAnchor)
{
nsCOMPtr<nsISupportsString> urlDataPrimitive(do_CreateInstance(NS_SUPPORTS_STRING_CONTRACTID));
if (!urlDataPrimitive)
return NS_ERROR_FAILURE;
urlDataPrimitive->SetData(mUrlString);
trans->SetTransferData(kURLDataMime, urlDataPrimitive, mUrlString.Length() * sizeof(PRUnichar));
}
}
*outTrans = trans;
NS_IF_ADDREF(*outTrans);
return NS_OK;
}
// static
// note that this can return NS_OK, but a null out param (by design)
nsresult nsTransferableFactory::GetDraggableSelectionData(nsISelection* inSelection,
nsIDOMNode* inRealTargetNode, nsIDOMNode **outImageOrLinkNode, PRBool* outDragSelectedText)
{
NS_ENSURE_ARG(inSelection);
NS_ENSURE_ARG(inRealTargetNode);
NS_ENSURE_ARG_POINTER(outImageOrLinkNode);
*outImageOrLinkNode = nsnull;
*outDragSelectedText = PR_FALSE;
PRBool selectionContainsTarget = PR_FALSE;
PRBool isCollapsed = PR_FALSE;
inSelection->GetIsCollapsed(&isCollapsed);
if (!isCollapsed)
{
inSelection->ContainsNode(inRealTargetNode, PR_FALSE, &selectionContainsTarget);
if (selectionContainsTarget)
{
// track down the anchor node, if any, for the url
nsCOMPtr<nsIDOMNode> selectionStart;
inSelection->GetAnchorNode(getter_AddRefs(selectionStart));
nsCOMPtr<nsIDOMNode> selectionEnd;
inSelection->GetFocusNode(getter_AddRefs(selectionEnd));
// look for a selection around a single node, like an image.
// in this case, drag the image, rather than a serialization of the HTML
// XXX generalize this to other draggable element types?
if (selectionStart == selectionEnd)
{
PRBool hasChildren;
selectionStart->HasChildNodes(&hasChildren);
if (hasChildren)
{
// see if just one node is selected
PRInt32 anchorOffset, focusOffset;
inSelection->GetAnchorOffset(&anchorOffset);
inSelection->GetFocusOffset(&focusOffset);
if (abs(anchorOffset - focusOffset) == 1)
{
nsCOMPtr<nsIContent> selStartContent = do_QueryInterface(selectionStart);
if (selStartContent)
{
PRInt32 childOffset =
(anchorOffset < focusOffset) ? anchorOffset : focusOffset;
nsIContent *childContent =
selStartContent->GetChildAt(childOffset);
// if we find an image, we'll fall into the node-dragging code,
// rather the the selection-dragging code
nsCOMPtr<nsIDOMHTMLImageElement> selectedImage =
do_QueryInterface(childContent);
if (selectedImage)
{
CallQueryInterface(selectedImage, outImageOrLinkNode); // addrefs
return NS_OK;
}
}
}
}
}
// if we didn't find an image, look for an anchor
nsCOMPtr<nsIDOMNode> firstAnchor;
FindFirstAnchor(selectionStart, getter_AddRefs(firstAnchor));
if (firstAnchor)
{
PRBool anchorInSelection = PR_FALSE;
inSelection->ContainsNode(firstAnchor, PR_FALSE, &anchorInSelection);
if (anchorInSelection)
CallQueryInterface(firstAnchor, outImageOrLinkNode); // addrefs
}
*outDragSelectedText = PR_TRUE;
}
}
return NS_OK;
}
// static
nsresult
nsTransferableFactory::SerializeNodeOrSelection(serializationMode inMode, PRUint32 inFlags,
nsIDOMWindow* inWindow, nsIDOMNode* inNode,
nsAString& outResultString, nsAString& outContext, nsAString& outInfo)
{
NS_ENSURE_ARG_POINTER(inWindow);
nsresult rv;
nsCOMPtr<nsIDocumentEncoder> encoder;
static const char *textplain = "text/plain";
if (inMode == serializeAsText)
{
nsCAutoString formatType(NS_DOC_ENCODER_CONTRACTID_BASE);
formatType.Append(textplain);
encoder = do_CreateInstance(formatType.get(), &rv);
}
else
{
encoder = do_CreateInstance(NS_HTMLCOPY_ENCODER_CONTRACTID, &rv);
}
NS_ENSURE_SUCCESS(rv, rv);
nsCOMPtr<nsIDOMDocument> domDoc;
inWindow->GetDocument(getter_AddRefs(domDoc));
nsCOMPtr<nsIDocument> doc = do_QueryInterface(domDoc);
if (!doc) return NS_ERROR_FAILURE;
nsCOMPtr<nsIDOMRange> range;
nsCOMPtr<nsISelection> selection;
if (inNode)
{
// make a range around this node
rv = NS_NewRange(getter_AddRefs(range));
NS_ENSURE_SUCCESS(rv, rv);
rv = range->SelectNode(inNode);
NS_ENSURE_SUCCESS(rv, rv);
}
else
{
inWindow->GetSelection(getter_AddRefs(selection));
inFlags |= nsIDocumentEncoder::OutputSelectionOnly;
}
if (inMode == serializeAsText)
{
rv = encoder->Init(doc, NS_ConvertASCIItoUCS2(textplain), inFlags);
}
else
{
rv = encoder->Init(doc, NS_LITERAL_STRING(kHTMLMime), inFlags);
}
NS_ENSURE_SUCCESS(rv, rv);
if (range)
encoder->SetRange(range);
else if (selection)
encoder->SetSelection(selection);
if (inMode == serializeAsText)
{
outContext.Truncate();
outInfo.Truncate();
return encoder->EncodeToString(outResultString);
}
else
{
return encoder->EncodeToStringWithContext(outResultString, outContext, outInfo);
}
}
//
// GetImage
//
// Given a dom node that's an image, finds the nsIImage associated with it.
//
nsresult
nsTransferableFactory::GetImageFromDOMNode(nsIDOMNode* inNode, nsIImage**outImage)
{
NS_ENSURE_ARG_POINTER(outImage);
*outImage = nsnull;
nsCOMPtr<nsIImageLoadingContent> content(do_QueryInterface(inNode));
if (!content) {
return NS_ERROR_NOT_AVAILABLE;
}
nsCOMPtr<imgIRequest> imgRequest;
content->GetRequest(nsIImageLoadingContent::CURRENT_REQUEST,
getter_AddRefs(imgRequest));
if (!imgRequest) {
return NS_ERROR_NOT_AVAILABLE;
}
nsCOMPtr<imgIContainer> imgContainer;
imgRequest->GetImage(getter_AddRefs(imgContainer));
if (!imgContainer) {
return NS_ERROR_NOT_AVAILABLE;
}
nsCOMPtr<gfxIImageFrame> imgFrame;
imgContainer->GetFrameAt(0, getter_AddRefs(imgFrame));
if (!imgFrame) {
return NS_ERROR_NOT_AVAILABLE;
}
nsCOMPtr<nsIInterfaceRequestor> ir = do_QueryInterface(imgFrame);
if (!ir) {
return NS_ERROR_NOT_AVAILABLE;
}
return CallGetInterface(ir.get(), outImage);
}