Replace static_casts with to* functions for document types.
[WebKit-https.git] / Source / WebCore / page / DragController.cpp
1 /*
2  * Copyright (C) 2007, 2009, 2010 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #include "config.h"
27 #include "DragController.h"
28
29 #if ENABLE(DRAG_SUPPORT)
30
31 #include "CachedImage.h"
32 #include "Clipboard.h"
33 #include "ClipboardAccessPolicy.h"
34 #include "CachedResourceLoader.h"
35 #include "Document.h"
36 #include "DocumentFragment.h"
37 #include "DragActions.h"
38 #include "DragClient.h"
39 #include "DragData.h"
40 #include "DragSession.h"
41 #include "DragState.h"
42 #include "Editor.h"
43 #include "EditorClient.h"
44 #include "Element.h"
45 #include "EventHandler.h"
46 #include "ExceptionCodePlaceholder.h"
47 #include "FloatRect.h"
48 #include "Frame.h"
49 #include "FrameLoadRequest.h"
50 #include "FrameLoader.h"
51 #include "FrameSelection.h"
52 #include "FrameView.h"
53 #include "HTMLAnchorElement.h"
54 #include "HTMLInputElement.h"
55 #include "HTMLNames.h"
56 #include "HTMLPlugInElement.h"
57 #include "HitTestRequest.h"
58 #include "HitTestResult.h"
59 #include "Image.h"
60 #include "ImageOrientation.h"
61 #include "MoveSelectionCommand.h"
62 #include "Node.h"
63 #include "Page.h"
64 #include "PlatformKeyboardEvent.h"
65 #include "PluginDocument.h"
66 #include "PluginViewBase.h"
67 #include "RenderFileUploadControl.h"
68 #include "RenderImage.h"
69 #include "RenderView.h"
70 #include "ReplaceSelectionCommand.h"
71 #include "ResourceRequest.h"
72 #include "SecurityOrigin.h"
73 #include "Settings.h"
74 #include "ShadowRoot.h"
75 #include "StylePropertySet.h"
76 #include "Text.h"
77 #include "TextEvent.h"
78 #include "htmlediting.h"
79 #include "markup.h"
80 #include <wtf/CurrentTime.h>
81 #include <wtf/RefPtr.h>
82
83 namespace WebCore {
84
85 static PlatformMouseEvent createMouseEvent(DragData* dragData)
86 {
87     bool shiftKey, ctrlKey, altKey, metaKey;
88     shiftKey = ctrlKey = altKey = metaKey = false;
89     int keyState = dragData->modifierKeyState();
90     shiftKey = static_cast<bool>(keyState & PlatformEvent::ShiftKey);
91     ctrlKey = static_cast<bool>(keyState & PlatformEvent::CtrlKey);
92     altKey = static_cast<bool>(keyState & PlatformEvent::AltKey);
93     metaKey = static_cast<bool>(keyState & PlatformEvent::MetaKey);
94
95     return PlatformMouseEvent(dragData->clientPosition(), dragData->globalPosition(),
96                               LeftButton, PlatformEvent::MouseMoved, 0, shiftKey, ctrlKey, altKey,
97                               metaKey, currentTime());
98 }
99
100 DragController::DragController(Page* page, DragClient* client)
101     : m_page(page)
102     , m_client(client)
103     , m_documentUnderMouse(0)
104     , m_dragInitiator(0)
105     , m_fileInputElementUnderMouse(0)
106     , m_documentIsHandlingDrag(false)
107     , m_dragDestinationAction(DragDestinationActionNone)
108     , m_dragSourceAction(DragSourceActionNone)
109     , m_didInitiateDrag(false)
110     , m_sourceDragOperation(DragOperationNone)
111 {
112     ASSERT(m_client);
113 }
114
115 DragController::~DragController()
116 {
117     m_client->dragControllerDestroyed();
118 }
119
120 PassOwnPtr<DragController> DragController::create(Page* page, DragClient* client)
121 {
122     return adoptPtr(new DragController(page, client));
123 }
124
125 static PassRefPtr<DocumentFragment> documentFragmentFromDragData(DragData* dragData, Frame* frame, RefPtr<Range> context,
126                                           bool allowPlainText, bool& chosePlainText)
127 {
128     ASSERT(dragData);
129     chosePlainText = false;
130
131     Document* document = context->ownerDocument();
132     ASSERT(document);
133     if (document && dragData->containsCompatibleContent()) {
134         if (PassRefPtr<DocumentFragment> fragment = dragData->asFragment(frame, context, allowPlainText, chosePlainText))
135             return fragment;
136
137         if (dragData->containsURL(frame, DragData::DoNotConvertFilenames)) {
138             String title;
139             String url = dragData->asURL(frame, DragData::DoNotConvertFilenames, &title);
140             if (!url.isEmpty()) {
141                 RefPtr<HTMLAnchorElement> anchor = HTMLAnchorElement::create(document);
142                 anchor->setHref(url);
143                 if (title.isEmpty()) {
144                     // Try the plain text first because the url might be normalized or escaped.
145                     if (dragData->containsPlainText())
146                         title = dragData->asPlainText(frame);
147                     if (title.isEmpty())
148                         title = url;
149                 }
150                 RefPtr<Node> anchorText = document->createTextNode(title);
151                 anchor->appendChild(anchorText, IGNORE_EXCEPTION);
152                 RefPtr<DocumentFragment> fragment = document->createDocumentFragment();
153                 fragment->appendChild(anchor, IGNORE_EXCEPTION);
154                 return fragment.get();
155             }
156         }
157     }
158     if (allowPlainText && dragData->containsPlainText()) {
159         chosePlainText = true;
160         return createFragmentFromText(context.get(), dragData->asPlainText(frame)).get();
161     }
162
163     return 0;
164 }
165
166 bool DragController::dragIsMove(FrameSelection* selection, DragData* dragData)
167 {
168     return m_documentUnderMouse == m_dragInitiator && selection->isContentEditable() && selection->isRange() && !isCopyKeyDown(dragData);
169 }
170
171 // FIXME: This method is poorly named.  We're just clearing the selection from the document this drag is exiting.
172 void DragController::cancelDrag()
173 {
174     m_page->dragCaretController()->clear();
175 }
176
177 void DragController::dragEnded()
178 {
179     m_dragInitiator = 0;
180     m_didInitiateDrag = false;
181     m_page->dragCaretController()->clear();
182     
183     m_client->dragEnded();
184 }
185
186 DragSession DragController::dragEntered(DragData* dragData)
187 {
188     return dragEnteredOrUpdated(dragData);
189 }
190
191 void DragController::dragExited(DragData* dragData)
192 {
193     ASSERT(dragData);
194     Frame* mainFrame = m_page->mainFrame();
195
196     if (RefPtr<FrameView> v = mainFrame->view()) {
197         ClipboardAccessPolicy policy = (!m_documentUnderMouse || m_documentUnderMouse->securityOrigin()->isLocal()) ? ClipboardReadable : ClipboardTypesReadable;
198         RefPtr<Clipboard> clipboard = Clipboard::create(policy, dragData, mainFrame);
199         clipboard->setSourceOperation(dragData->draggingSourceOperationMask());
200         mainFrame->eventHandler()->cancelDragAndDrop(createMouseEvent(dragData), clipboard.get());
201         clipboard->setAccessPolicy(ClipboardNumb);    // invalidate clipboard here for security
202     }
203     mouseMovedIntoDocument(0);
204     if (m_fileInputElementUnderMouse)
205         m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(false);
206     m_fileInputElementUnderMouse = 0;
207 }
208
209 DragSession DragController::dragUpdated(DragData* dragData)
210 {
211     return dragEnteredOrUpdated(dragData);
212 }
213
214 bool DragController::performDrag(DragData* dragData)
215 {
216     ASSERT(dragData);
217     m_documentUnderMouse = m_page->mainFrame()->documentAtPoint(dragData->clientPosition());
218     if ((m_dragDestinationAction & DragDestinationActionDHTML) && m_documentIsHandlingDrag) {
219         m_client->willPerformDragDestinationAction(DragDestinationActionDHTML, dragData);
220         RefPtr<Frame> mainFrame = m_page->mainFrame();
221         bool preventedDefault = false;
222         if (mainFrame->view()) {
223             // Sending an event can result in the destruction of the view and part.
224             RefPtr<Clipboard> clipboard = Clipboard::create(ClipboardReadable, dragData, mainFrame.get());
225             clipboard->setSourceOperation(dragData->draggingSourceOperationMask());
226             preventedDefault = mainFrame->eventHandler()->performDragAndDrop(createMouseEvent(dragData), clipboard.get());
227             clipboard->setAccessPolicy(ClipboardNumb); // Invalidate clipboard here for security
228         }
229         if (preventedDefault) {
230             m_documentUnderMouse = 0;
231             return true;
232         }
233     }
234
235     if ((m_dragDestinationAction & DragDestinationActionEdit) && concludeEditDrag(dragData)) {
236         m_documentUnderMouse = 0;
237         return true;
238     }
239
240     m_documentUnderMouse = 0;
241
242     if (operationForLoad(dragData) == DragOperationNone)
243         return false;
244
245     m_client->willPerformDragDestinationAction(DragDestinationActionLoad, dragData);
246     m_page->mainFrame()->loader()->load(FrameLoadRequest(m_page->mainFrame(), ResourceRequest(dragData->asURL(m_page->mainFrame()))));
247     return true;
248 }
249
250 void DragController::mouseMovedIntoDocument(Document* newDocument)
251 {
252     if (m_documentUnderMouse == newDocument)
253         return;
254
255     // If we were over another document clear the selection
256     if (m_documentUnderMouse)
257         cancelDrag();
258     m_documentUnderMouse = newDocument;
259 }
260
261 DragSession DragController::dragEnteredOrUpdated(DragData* dragData)
262 {
263     ASSERT(dragData);
264     ASSERT(m_page->mainFrame());
265     mouseMovedIntoDocument(m_page->mainFrame()->documentAtPoint(dragData->clientPosition()));
266
267     m_dragDestinationAction = m_client->actionMaskForDrag(dragData);
268     if (m_dragDestinationAction == DragDestinationActionNone) {
269         cancelDrag(); // FIXME: Why not call mouseMovedIntoDocument(0)?
270         return DragSession();
271     }
272
273     DragSession dragSession;
274     m_documentIsHandlingDrag = tryDocumentDrag(dragData, m_dragDestinationAction, dragSession);
275     if (!m_documentIsHandlingDrag && (m_dragDestinationAction & DragDestinationActionLoad))
276         dragSession.operation = operationForLoad(dragData);
277     return dragSession;
278 }
279
280 static HTMLInputElement* asFileInput(Node* node)
281 {
282     ASSERT(node);
283
284     HTMLInputElement* inputElement = node->toInputElement();
285
286     // If this is a button inside of the a file input, move up to the file input.
287     if (inputElement && inputElement->isTextButton() && inputElement->treeScope()->rootNode()->isShadowRoot())
288         inputElement = toShadowRoot(inputElement->treeScope()->rootNode())->host()->toInputElement();
289
290     return inputElement && inputElement->isFileUpload() ? inputElement : 0;
291 }
292
293 // This can return null if an empty document is loaded.
294 static Element* elementUnderMouse(Document* documentUnderMouse, const IntPoint& p)
295 {
296     Frame* frame = documentUnderMouse->frame();
297     float zoomFactor = frame ? frame->pageZoomFactor() : 1;
298     LayoutPoint point = roundedLayoutPoint(FloatPoint(p.x() * zoomFactor, p.y() * zoomFactor));
299
300     HitTestRequest request(HitTestRequest::ReadOnly | HitTestRequest::Active);
301     HitTestResult result(point);
302     documentUnderMouse->renderView()->hitTest(request, result);
303
304     Node* n = result.innerNode();
305     while (n && !n->isElementNode())
306         n = n->parentNode();
307     if (n)
308         n = n->deprecatedShadowAncestorNode();
309
310     return static_cast<Element*>(n);
311 }
312
313 bool DragController::tryDocumentDrag(DragData* dragData, DragDestinationAction actionMask, DragSession& dragSession)
314 {
315     ASSERT(dragData);
316
317     if (!m_documentUnderMouse)
318         return false;
319
320     if (m_dragInitiator && !m_documentUnderMouse->securityOrigin()->canReceiveDragData(m_dragInitiator->securityOrigin()))
321         return false;
322
323     bool isHandlingDrag = false;
324     if (actionMask & DragDestinationActionDHTML) {
325         isHandlingDrag = tryDHTMLDrag(dragData, dragSession.operation);
326         // Do not continue if m_documentUnderMouse has been reset by tryDHTMLDrag.
327         // tryDHTMLDrag fires dragenter event. The event listener that listens
328         // to this event may create a nested message loop (open a modal dialog),
329         // which could process dragleave event and reset m_documentUnderMouse in
330         // dragExited.
331         if (!m_documentUnderMouse)
332             return false;
333     }
334
335     // It's unclear why this check is after tryDHTMLDrag.
336     // We send drag events in tryDHTMLDrag and that may be the reason.
337     RefPtr<FrameView> frameView = m_documentUnderMouse->view();
338     if (!frameView)
339         return false;
340
341     if (isHandlingDrag) {
342         m_page->dragCaretController()->clear();
343         return true;
344     }
345
346     if ((actionMask & DragDestinationActionEdit) && canProcessDrag(dragData)) {
347         if (dragData->containsColor()) {
348             dragSession.operation = DragOperationGeneric;
349             return true;
350         }
351
352         IntPoint point = frameView->windowToContents(dragData->clientPosition());
353         Element* element = elementUnderMouse(m_documentUnderMouse.get(), point);
354         if (!element)
355             return false;
356         
357         HTMLInputElement* elementAsFileInput = asFileInput(element);
358         if (m_fileInputElementUnderMouse != elementAsFileInput) {
359             if (m_fileInputElementUnderMouse)
360                 m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(false);
361             m_fileInputElementUnderMouse = elementAsFileInput;
362         }
363         
364         if (!m_fileInputElementUnderMouse)
365             m_page->dragCaretController()->setCaretPosition(m_documentUnderMouse->frame()->visiblePositionForPoint(point));
366
367         Frame* innerFrame = element->document()->frame();
368         dragSession.operation = dragIsMove(innerFrame->selection(), dragData) ? DragOperationMove : DragOperationCopy;
369         dragSession.mouseIsOverFileInput = m_fileInputElementUnderMouse;
370         dragSession.numberOfItemsToBeAccepted = 0;
371
372         unsigned numberOfFiles = dragData->numberOfFiles();
373         if (m_fileInputElementUnderMouse) {
374             if (m_fileInputElementUnderMouse->disabled())
375                 dragSession.numberOfItemsToBeAccepted = 0;
376             else if (m_fileInputElementUnderMouse->multiple())
377                 dragSession.numberOfItemsToBeAccepted = numberOfFiles;
378             else if (numberOfFiles > 1)
379                 dragSession.numberOfItemsToBeAccepted = 0;
380             else
381                 dragSession.numberOfItemsToBeAccepted = 1;
382             
383             if (!dragSession.numberOfItemsToBeAccepted)
384                 dragSession.operation = DragOperationNone;
385             m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(dragSession.numberOfItemsToBeAccepted);
386         } else {
387             // We are not over a file input element. The dragged item(s) will only
388             // be loaded into the view the number of dragged items is 1.
389             dragSession.numberOfItemsToBeAccepted = numberOfFiles != 1 ? 0 : 1;
390         }
391         
392         return true;
393     }
394     
395     // We are not over an editable region. Make sure we're clearing any prior drag cursor.
396     m_page->dragCaretController()->clear();
397     if (m_fileInputElementUnderMouse)
398         m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(false);
399     m_fileInputElementUnderMouse = 0;
400     return false;
401 }
402
403 DragSourceAction DragController::delegateDragSourceAction(const IntPoint& rootViewPoint)
404 {
405     m_dragSourceAction = m_client->dragSourceActionMaskForPoint(rootViewPoint);
406     return m_dragSourceAction;
407 }
408
409 DragOperation DragController::operationForLoad(DragData* dragData)
410 {
411     ASSERT(dragData);
412     Document* doc = m_page->mainFrame()->documentAtPoint(dragData->clientPosition());
413
414     bool pluginDocumentAcceptsDrags = false;
415
416     if (doc && doc->isPluginDocument()) {
417         const Widget* widget = toPluginDocument(doc)->pluginWidget();
418         const PluginViewBase* pluginView = (widget && widget->isPluginViewBase()) ? static_cast<const PluginViewBase*>(widget) : 0;
419
420         if (pluginView)
421             pluginDocumentAcceptsDrags = pluginView->shouldAllowNavigationFromDrags();
422     }
423
424     if (doc && (m_didInitiateDrag || (doc->isPluginDocument() && !pluginDocumentAcceptsDrags) || doc->rendererIsEditable()))
425         return DragOperationNone;
426     return dragOperation(dragData);
427 }
428
429 static bool setSelectionToDragCaret(Frame* frame, VisibleSelection& dragCaret, RefPtr<Range>& range, const IntPoint& point)
430 {
431     frame->selection()->setSelection(dragCaret);
432     if (frame->selection()->isNone()) {
433         dragCaret = frame->visiblePositionForPoint(point);
434         frame->selection()->setSelection(dragCaret);
435         range = dragCaret.toNormalizedRange();
436     }
437     return !frame->selection()->isNone() && frame->selection()->isContentEditable();
438 }
439
440 bool DragController::dispatchTextInputEventFor(Frame* innerFrame, DragData* dragData)
441 {
442     ASSERT(m_page->dragCaretController()->hasCaret());
443     String text = m_page->dragCaretController()->isContentRichlyEditable() ? "" : dragData->asPlainText(innerFrame);
444     Node* target = innerFrame->editor()->findEventTargetFrom(m_page->dragCaretController()->caretPosition());
445     return target->dispatchEvent(TextEvent::createForDrop(innerFrame->document()->domWindow(), text), IGNORE_EXCEPTION);
446 }
447
448 bool DragController::concludeEditDrag(DragData* dragData)
449 {
450     ASSERT(dragData);
451
452     RefPtr<HTMLInputElement> fileInput = m_fileInputElementUnderMouse;
453     if (m_fileInputElementUnderMouse) {
454         m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(false);
455         m_fileInputElementUnderMouse = 0;
456     }
457
458     if (!m_documentUnderMouse)
459         return false;
460
461     IntPoint point = m_documentUnderMouse->view()->windowToContents(dragData->clientPosition());
462     Element* element = elementUnderMouse(m_documentUnderMouse.get(), point);
463     if (!element)
464         return false;
465     RefPtr<Frame> innerFrame = element->ownerDocument()->frame();
466     ASSERT(innerFrame);
467
468     if (m_page->dragCaretController()->hasCaret() && !dispatchTextInputEventFor(innerFrame.get(), dragData))
469         return true;
470
471     if (dragData->containsColor()) {
472         Color color = dragData->asColor();
473         if (!color.isValid())
474             return false;
475         RefPtr<Range> innerRange = innerFrame->selection()->toNormalizedRange();
476         RefPtr<StylePropertySet> style = StylePropertySet::create();
477         style->setProperty(CSSPropertyColor, color.serialized(), false);
478         if (!innerFrame->editor()->shouldApplyStyle(style.get(), innerRange.get()))
479             return false;
480         m_client->willPerformDragDestinationAction(DragDestinationActionEdit, dragData);
481         innerFrame->editor()->applyStyle(style.get(), EditActionSetColor);
482         return true;
483     }
484
485     if (dragData->containsFiles() && fileInput) {
486         // fileInput should be the element we hit tested for, unless it was made
487         // display:none in a drop event handler.
488         ASSERT(fileInput == element || !fileInput->renderer());
489         if (fileInput->disabled())
490             return false;
491
492         return fileInput->receiveDroppedFiles(dragData);
493     }
494
495     if (!m_page->dragController()->canProcessDrag(dragData)) {
496         m_page->dragCaretController()->clear();
497         return false;
498     }
499
500     VisibleSelection dragCaret = m_page->dragCaretController()->caretPosition();
501     m_page->dragCaretController()->clear();
502     RefPtr<Range> range = dragCaret.toNormalizedRange();
503     RefPtr<Element> rootEditableElement = innerFrame->selection()->rootEditableElement();
504
505     // For range to be null a WebKit client must have done something bad while
506     // manually controlling drag behaviour
507     if (!range)
508         return false;
509     CachedResourceLoader* cachedResourceLoader = range->ownerDocument()->cachedResourceLoader();
510     ResourceCacheValidationSuppressor validationSuppressor(cachedResourceLoader);
511     if (dragIsMove(innerFrame->selection(), dragData) || dragCaret.isContentRichlyEditable()) {
512         bool chosePlainText = false;
513         RefPtr<DocumentFragment> fragment = documentFragmentFromDragData(dragData, innerFrame.get(), range, true, chosePlainText);
514         if (!fragment || !innerFrame->editor()->shouldInsertFragment(fragment, range, EditorInsertActionDropped)) {
515             return false;
516         }
517
518         m_client->willPerformDragDestinationAction(DragDestinationActionEdit, dragData);
519         if (dragIsMove(innerFrame->selection(), dragData)) {
520             // NSTextView behavior is to always smart delete on moving a selection,
521             // but only to smart insert if the selection granularity is word granularity.
522             bool smartDelete = innerFrame->editor()->smartInsertDeleteEnabled();
523             bool smartInsert = smartDelete && innerFrame->selection()->granularity() == WordGranularity && dragData->canSmartReplace();
524             applyCommand(MoveSelectionCommand::create(fragment, dragCaret.base(), smartInsert, smartDelete));
525         } else {
526             if (setSelectionToDragCaret(innerFrame.get(), dragCaret, range, point)) {
527                 ReplaceSelectionCommand::CommandOptions options = ReplaceSelectionCommand::SelectReplacement | ReplaceSelectionCommand::PreventNesting;
528                 if (dragData->canSmartReplace())
529                     options |= ReplaceSelectionCommand::SmartReplace;
530                 if (chosePlainText)
531                     options |= ReplaceSelectionCommand::MatchStyle;
532                 applyCommand(ReplaceSelectionCommand::create(m_documentUnderMouse.get(), fragment, options));
533             }
534         }
535     } else {
536         String text = dragData->asPlainText(innerFrame.get());
537         if (text.isEmpty() || !innerFrame->editor()->shouldInsertText(text, range.get(), EditorInsertActionDropped)) {
538             return false;
539         }
540
541         m_client->willPerformDragDestinationAction(DragDestinationActionEdit, dragData);
542         if (setSelectionToDragCaret(innerFrame.get(), dragCaret, range, point))
543             applyCommand(ReplaceSelectionCommand::create(m_documentUnderMouse.get(), createFragmentFromText(range.get(), text),  ReplaceSelectionCommand::SelectReplacement | ReplaceSelectionCommand::MatchStyle | ReplaceSelectionCommand::PreventNesting));
544     }
545
546     if (rootEditableElement) {
547         if (Frame* frame = rootEditableElement->document()->frame())
548             frame->eventHandler()->updateDragStateAfterEditDragIfNeeded(rootEditableElement.get());
549     }
550
551     return true;
552 }
553
554 bool DragController::canProcessDrag(DragData* dragData)
555 {
556     ASSERT(dragData);
557
558     if (!dragData->containsCompatibleContent())
559         return false;
560
561     IntPoint point = m_page->mainFrame()->view()->windowToContents(dragData->clientPosition());
562     HitTestResult result = HitTestResult(point);
563     if (!m_page->mainFrame()->contentRenderer())
564         return false;
565
566     result = m_page->mainFrame()->eventHandler()->hitTestResultAtPoint(point, HitTestRequest::ReadOnly | HitTestRequest::Active | HitTestRequest::AllowShadowContent);
567
568     if (!result.innerNonSharedNode())
569         return false;
570
571     if (dragData->containsFiles() && asFileInput(result.innerNonSharedNode()))
572         return true;
573
574     if (result.innerNonSharedNode()->isPluginElement()) {
575         HTMLPlugInElement* plugin = static_cast<HTMLPlugInElement*>(result.innerNonSharedNode());
576         if (!plugin->canProcessDrag() && !result.innerNonSharedNode()->rendererIsEditable())
577             return false;
578     } else if (!result.innerNonSharedNode()->rendererIsEditable())
579         return false;
580
581     if (m_didInitiateDrag && m_documentUnderMouse == m_dragInitiator && result.isSelected())
582         return false;
583
584     return true;
585 }
586
587 static DragOperation defaultOperationForDrag(DragOperation srcOpMask)
588 {
589     // This is designed to match IE's operation fallback for the case where
590     // the page calls preventDefault() in a drag event but doesn't set dropEffect.
591     if (srcOpMask == DragOperationEvery)
592         return DragOperationCopy;
593     if (srcOpMask == DragOperationNone)
594         return DragOperationNone;
595     if (srcOpMask & DragOperationMove || srcOpMask & DragOperationGeneric)
596         return DragOperationMove;
597     if (srcOpMask & DragOperationCopy)
598         return DragOperationCopy;
599     if (srcOpMask & DragOperationLink)
600         return DragOperationLink;
601     
602     // FIXME: Does IE really return "generic" even if no operations were allowed by the source?
603     return DragOperationGeneric;
604 }
605
606 bool DragController::tryDHTMLDrag(DragData* dragData, DragOperation& operation)
607 {
608     ASSERT(dragData);
609     ASSERT(m_documentUnderMouse);
610     RefPtr<Frame> mainFrame = m_page->mainFrame();
611     RefPtr<FrameView> viewProtector = mainFrame->view();
612     if (!viewProtector)
613         return false;
614
615     ClipboardAccessPolicy policy = m_documentUnderMouse->securityOrigin()->isLocal() ? ClipboardReadable : ClipboardTypesReadable;
616     RefPtr<Clipboard> clipboard = Clipboard::create(policy, dragData, mainFrame.get());
617     DragOperation srcOpMask = dragData->draggingSourceOperationMask();
618     clipboard->setSourceOperation(srcOpMask);
619
620     PlatformMouseEvent event = createMouseEvent(dragData);
621     if (!mainFrame->eventHandler()->updateDragAndDrop(event, clipboard.get())) {
622         clipboard->setAccessPolicy(ClipboardNumb);    // invalidate clipboard here for security
623         return false;
624     }
625
626     operation = clipboard->destinationOperation();
627     if (clipboard->dropEffectIsUninitialized())
628         operation = defaultOperationForDrag(srcOpMask);
629     else if (!(srcOpMask & operation)) {
630         // The element picked an operation which is not supported by the source
631         operation = DragOperationNone;
632     }
633
634     clipboard->setAccessPolicy(ClipboardNumb);    // invalidate clipboard here for security
635     return true;
636 }
637
638 Node* DragController::draggableNode(const Frame* src, Node* startNode, const IntPoint& dragOrigin, DragState& state) const
639 {
640     state.m_dragType = (src->selection()->contains(dragOrigin)) ? DragSourceActionSelection : DragSourceActionNone;
641
642     for (const RenderObject* renderer = startNode->renderer(); renderer; renderer = renderer->parent()) {
643         Node* node = renderer->nonPseudoNode();
644         if (!node)
645             // Anonymous render blocks don't correspond to actual DOM nodes, so we skip over them
646             // for the purposes of finding a draggable node.
647             continue;
648         if (!(state.m_dragType & DragSourceActionSelection) && node->isTextNode() && node->canStartSelection())
649             // In this case we have a click in the unselected portion of text. If this text is
650             // selectable, we want to start the selection process instead of looking for a parent
651             // to try to drag.
652             return 0;
653         if (node->isElementNode()) {
654             EUserDrag dragMode = renderer->style()->userDrag();
655             if ((m_dragSourceAction & DragSourceActionDHTML) && dragMode == DRAG_ELEMENT) {
656                 state.m_dragType = static_cast<DragSourceAction>(state.m_dragType | DragSourceActionDHTML);
657                 return node;
658             }
659             if (dragMode == DRAG_AUTO) {
660                 if ((m_dragSourceAction & DragSourceActionImage)
661                     && node->hasTagName(HTMLNames::imgTag)
662                     && src->settings()
663                     && src->settings()->loadsImagesAutomatically()) {
664                     state.m_dragType = static_cast<DragSourceAction>(state.m_dragType | DragSourceActionImage);
665                     return node;
666                 }
667                 if ((m_dragSourceAction & DragSourceActionLink)
668                     && node->hasTagName(HTMLNames::aTag)
669                     && static_cast<HTMLAnchorElement*>(node)->isLiveLink()) {
670                     state.m_dragType = static_cast<DragSourceAction>(state.m_dragType | DragSourceActionLink);
671                     return node;
672                 }
673             }
674         }
675     }
676
677     // We either have nothing to drag or we have a selection and we're not over a draggable element.
678     return (state.m_dragType & DragSourceActionSelection) ? startNode : 0;
679 }
680
681 static CachedImage* getCachedImage(Element* element)
682 {
683     ASSERT(element);
684     RenderObject* renderer = element->renderer();
685     if (!renderer || !renderer->isImage())
686         return 0;
687     RenderImage* image = toRenderImage(renderer);
688     return image->cachedImage();
689 }
690
691 static Image* getImage(Element* element)
692 {
693     ASSERT(element);
694     CachedImage* cachedImage = getCachedImage(element);
695     // Don't use cachedImage->imageForRenderer() here as that may return BitmapImages for cached SVG Images.
696     // Users of getImage() want access to the SVGImage, in order to figure out the filename extensions,
697     // which would be empty when asking the cached BitmapImages.
698     return (cachedImage && !cachedImage->errorOccurred()) ?
699         cachedImage->image() : 0;
700 }
701
702 static void prepareClipboardForImageDrag(Frame* source, Clipboard* clipboard, Element* node, const KURL& linkURL, const KURL& imageURL, const String& label)
703 {
704     if (node->isContentRichlyEditable()) {
705         RefPtr<Range> range = source->document()->createRange();
706         range->selectNode(node, ASSERT_NO_EXCEPTION);
707         source->selection()->setSelection(VisibleSelection(range.get(), DOWNSTREAM));
708     }
709     clipboard->declareAndWriteDragImage(node, !linkURL.isEmpty() ? linkURL : imageURL, label, source);
710 }
711
712 static IntPoint dragLocForDHTMLDrag(const IntPoint& mouseDraggedPoint, const IntPoint& dragOrigin, const IntPoint& dragImageOffset, bool isLinkImage)
713 {
714     // dragImageOffset is the cursor position relative to the lower-left corner of the image.
715 #if PLATFORM(MAC)
716     // We add in the Y dimension because we are a flipped view, so adding moves the image down.
717     const int yOffset = dragImageOffset.y();
718 #else
719     const int yOffset = -dragImageOffset.y();
720 #endif
721
722     if (isLinkImage)
723         return IntPoint(mouseDraggedPoint.x() - dragImageOffset.x(), mouseDraggedPoint.y() + yOffset);
724
725     return IntPoint(dragOrigin.x() - dragImageOffset.x(), dragOrigin.y() + yOffset);
726 }
727
728 static IntPoint dragLocForSelectionDrag(Frame* src)
729 {
730     IntRect draggingRect = enclosingIntRect(src->selection()->bounds());
731     int xpos = draggingRect.maxX();
732     xpos = draggingRect.x() < xpos ? draggingRect.x() : xpos;
733     int ypos = draggingRect.maxY();
734 #if PLATFORM(MAC)
735     // Deal with flipped coordinates on Mac
736     ypos = draggingRect.y() > ypos ? draggingRect.y() : ypos;
737 #else
738     ypos = draggingRect.y() < ypos ? draggingRect.y() : ypos;
739 #endif
740     return IntPoint(xpos, ypos);
741 }
742
743 bool DragController::startDrag(Frame* src, const DragState& state, DragOperation srcOp, const PlatformMouseEvent& dragEvent, const IntPoint& dragOrigin)
744 {
745     ASSERT(src);
746
747     if (!src->view() || !src->contentRenderer())
748         return false;
749
750     HitTestResult hitTestResult = src->eventHandler()->hitTestResultAtPoint(dragOrigin, HitTestRequest::ReadOnly | HitTestRequest::Active | HitTestRequest::AllowShadowContent);
751     if (!state.m_dragSrc->contains(hitTestResult.innerNode()))
752         // The original node being dragged isn't under the drag origin anymore... maybe it was
753         // hidden or moved out from under the cursor. Regardless, we don't want to start a drag on
754         // something that's not actually under the drag origin.
755         return false;
756     KURL linkURL = hitTestResult.absoluteLinkURL();
757     KURL imageURL = hitTestResult.absoluteImageURL();
758
759     IntPoint mouseDraggedPoint = src->view()->windowToContents(dragEvent.position());
760
761     m_draggingImageURL = KURL();
762     m_sourceDragOperation = srcOp;
763
764     DragImageRef dragImage = 0;
765     IntPoint dragLoc(0, 0);
766     IntPoint dragImageOffset(0, 0);
767
768     Clipboard* clipboard = state.m_dragClipboard.get();
769     if (state.m_dragType == DragSourceActionDHTML)
770         dragImage = clipboard->createDragImage(dragImageOffset);
771     if (state.m_dragType == DragSourceActionSelection || !imageURL.isEmpty() || !linkURL.isEmpty())
772         // Selection, image, and link drags receive a default set of allowed drag operations that
773         // follows from:
774         // http://trac.webkit.org/browser/trunk/WebKit/mac/WebView/WebHTMLView.mm?rev=48526#L3430
775         m_sourceDragOperation = static_cast<DragOperation>(m_sourceDragOperation | DragOperationGeneric | DragOperationCopy);
776
777     // We allow DHTML/JS to set the drag image, even if its a link, image or text we're dragging.
778     // This is in the spirit of the IE API, which allows overriding of pasteboard data and DragOp.
779     if (dragImage) {
780         dragLoc = dragLocForDHTMLDrag(mouseDraggedPoint, dragOrigin, dragImageOffset, !linkURL.isEmpty());
781         m_dragOffset = dragImageOffset;
782     }
783
784     bool startedDrag = true; // optimism - we almost always manage to start the drag
785
786     Node* node = state.m_dragSrc.get();
787
788     Image* image = getImage(static_cast<Element*>(node));
789     if (state.m_dragType == DragSourceActionSelection) {
790         if (!clipboard->hasData()) {
791             if (enclosingTextFormControl(src->selection()->start()))
792                 clipboard->writePlainText(src->editor()->selectedText());
793             else {
794                 RefPtr<Range> selectionRange = src->selection()->toNormalizedRange();
795                 ASSERT(selectionRange);
796
797                 clipboard->writeRange(selectionRange.get(), src);
798             }
799         }
800         m_client->willPerformDragSourceAction(DragSourceActionSelection, dragOrigin, clipboard);
801         if (!dragImage) {
802             dragImage = createDragImageForSelection(src);
803             dragLoc = dragLocForSelectionDrag(src);
804             m_dragOffset = IntPoint(dragOrigin.x() - dragLoc.x(), dragOrigin.y() - dragLoc.y());
805         }
806         doSystemDrag(dragImage, dragLoc, dragOrigin, clipboard, src, false);
807     } else if (!imageURL.isEmpty() && node && node->isElementNode() && image && !image->isNull()
808                && (m_dragSourceAction & DragSourceActionImage)) {
809         // We shouldn't be starting a drag for an image that can't provide an extension.
810         // This is an early detection for problems encountered later upon drop.
811         ASSERT(!image->filenameExtension().isEmpty());
812         Element* element = static_cast<Element*>(node);
813         if (!clipboard->hasData()) {
814             m_draggingImageURL = imageURL;
815             prepareClipboardForImageDrag(src, clipboard, element, linkURL, imageURL, hitTestResult.altDisplayString());
816         }
817
818         m_client->willPerformDragSourceAction(DragSourceActionImage, dragOrigin, clipboard);
819
820         if (!dragImage) {
821             IntRect imageRect = hitTestResult.imageRect();
822             imageRect.setLocation(m_page->mainFrame()->view()->rootViewToContents(src->view()->contentsToRootView(imageRect.location())));
823             doImageDrag(element, dragOrigin, hitTestResult.imageRect(), clipboard, src, m_dragOffset);
824         } else
825             // DHTML defined drag image
826             doSystemDrag(dragImage, dragLoc, dragOrigin, clipboard, src, false);
827
828     } else if (!linkURL.isEmpty() && (m_dragSourceAction & DragSourceActionLink)) {
829         if (!clipboard->hasData())
830             // Simplify whitespace so the title put on the clipboard resembles what the user sees
831             // on the web page. This includes replacing newlines with spaces.
832             clipboard->writeURL(linkURL, hitTestResult.textContent().simplifyWhiteSpace(), src);
833
834         if (src->selection()->isCaret() && src->selection()->isContentEditable()) {
835             // a user can initiate a drag on a link without having any text
836             // selected.  In this case, we should expand the selection to
837             // the enclosing anchor element
838             Position pos = src->selection()->base();
839             Node* node = enclosingAnchorElement(pos);
840             if (node)
841                 src->selection()->setSelection(VisibleSelection::selectionFromContentsOfNode(node));
842         }
843
844         m_client->willPerformDragSourceAction(DragSourceActionLink, dragOrigin, clipboard);
845         if (!dragImage) {
846             dragImage = createDragImageForLink(linkURL, hitTestResult.textContent(), src);
847             IntSize size = dragImageSize(dragImage);
848             m_dragOffset = IntPoint(-size.width() / 2, -LinkDragBorderInset);
849             dragLoc = IntPoint(mouseDraggedPoint.x() + m_dragOffset.x(), mouseDraggedPoint.y() + m_dragOffset.y());
850         }
851         doSystemDrag(dragImage, dragLoc, mouseDraggedPoint, clipboard, src, true);
852     } else if (state.m_dragType == DragSourceActionDHTML) {
853         ASSERT(m_dragSourceAction & DragSourceActionDHTML);
854         m_client->willPerformDragSourceAction(DragSourceActionDHTML, dragOrigin, clipboard);
855         doSystemDrag(dragImage, dragLoc, dragOrigin, clipboard, src, false);
856     } else {
857         // draggableNode() determined an image or link node was draggable, but it turns out the
858         // image or link had no URL, so there is nothing to drag.
859         startedDrag = false;
860     }
861
862     if (dragImage)
863         deleteDragImage(dragImage);
864     return startedDrag;
865 }
866
867 void DragController::doImageDrag(Element* element, const IntPoint& dragOrigin, const IntRect& rect, Clipboard* clipboard, Frame* frame, IntPoint& dragImageOffset)
868 {
869     IntPoint mouseDownPoint = dragOrigin;
870     DragImageRef dragImage;
871     IntPoint origin;
872
873     Image* image = getImage(element);
874     if (image && image->size().height() * image->size().width() <= MaxOriginalImageArea
875         && (dragImage = createDragImageFromImage(image, element->renderer() ? element->renderer()->shouldRespectImageOrientation() : DoNotRespectImageOrientation))) {
876         IntSize originalSize = rect.size();
877         origin = rect.location();
878
879         dragImage = fitDragImageToMaxSize(dragImage, rect.size(), maxDragImageSize());
880         dragImage = dissolveDragImageToFraction(dragImage, DragImageAlpha);
881         IntSize newSize = dragImageSize(dragImage);
882
883         // Properly orient the drag image and orient it differently if it's smaller than the original
884         float scale = newSize.width() / (float)originalSize.width();
885         float dx = origin.x() - mouseDownPoint.x();
886         dx *= scale;
887         origin.setX((int)(dx + 0.5));
888 #if PLATFORM(MAC)
889         //Compensate for accursed flipped coordinates in cocoa
890         origin.setY(origin.y() + originalSize.height());
891 #endif
892         float dy = origin.y() - mouseDownPoint.y();
893         dy *= scale;
894         origin.setY((int)(dy + 0.5));
895     } else {
896         dragImage = createDragImageIconForCachedImage(getCachedImage(element));
897         if (dragImage)
898             origin = IntPoint(DragIconRightInset - dragImageSize(dragImage).width(), DragIconBottomInset);
899     }
900
901     dragImageOffset = mouseDownPoint + origin;
902     doSystemDrag(dragImage, dragImageOffset, dragOrigin, clipboard, frame, false);
903
904     deleteDragImage(dragImage);
905 }
906
907 void DragController::doSystemDrag(DragImageRef image, const IntPoint& dragLoc, const IntPoint& eventPos, Clipboard* clipboard, Frame* frame, bool forLink)
908 {
909     m_didInitiateDrag = true;
910     m_dragInitiator = frame->document();
911     // Protect this frame and view, as a load may occur mid drag and attempt to unload this frame
912     RefPtr<Frame> frameProtector = m_page->mainFrame();
913     RefPtr<FrameView> viewProtector = frameProtector->view();
914     m_client->startDrag(image, viewProtector->rootViewToContents(frame->view()->contentsToRootView(dragLoc)),
915         viewProtector->rootViewToContents(frame->view()->contentsToRootView(eventPos)), clipboard, frameProtector.get(), forLink);
916     // DragClient::startDrag can cause our Page to dispear, deallocating |this|.
917     if (!frameProtector->page())
918         return;
919
920     cleanupAfterSystemDrag();
921 }
922
923 // Manual drag caret manipulation
924 void DragController::placeDragCaret(const IntPoint& windowPoint)
925 {
926     mouseMovedIntoDocument(m_page->mainFrame()->documentAtPoint(windowPoint));
927     if (!m_documentUnderMouse)
928         return;
929     Frame* frame = m_documentUnderMouse->frame();
930     FrameView* frameView = frame->view();
931     if (!frameView)
932         return;
933     IntPoint framePoint = frameView->windowToContents(windowPoint);
934
935     m_page->dragCaretController()->setCaretPosition(frame->visiblePositionForPoint(framePoint));
936 }
937
938 } // namespace WebCore
939
940 #endif // ENABLE(DRAG_SUPPORT)