Move URL from WebCore to WTF
[WebKit-https.git] / Source / WebCore / page / DragController.cpp
1 /*
2  * Copyright (C) 2007-2016 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 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 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 #include "HTMLAnchorElement.h"
30 #include "SVGAElement.h"
31
32 #if ENABLE(DRAG_SUPPORT)
33 #include "CachedImage.h"
34 #include "CachedResourceLoader.h"
35 #include "DataTransfer.h"
36 #include "Document.h"
37 #include "DocumentFragment.h"
38 #include "DragActions.h"
39 #include "DragClient.h"
40 #include "DragData.h"
41 #include "DragImage.h"
42 #include "DragState.h"
43 #include "Editing.h"
44 #include "Editor.h"
45 #include "EditorClient.h"
46 #include "ElementAncestorIterator.h"
47 #include "EventHandler.h"
48 #include "File.h"
49 #include "FloatRect.h"
50 #include "FocusController.h"
51 #include "Frame.h"
52 #include "FrameLoadRequest.h"
53 #include "FrameLoader.h"
54 #include "FrameSelection.h"
55 #include "FrameView.h"
56 #include "HTMLAttachmentElement.h"
57 #include "HTMLImageElement.h"
58 #include "HTMLInputElement.h"
59 #include "HTMLParserIdioms.h"
60 #include "HTMLPlugInElement.h"
61 #include "HitTestRequest.h"
62 #include "HitTestResult.h"
63 #include "Image.h"
64 #include "ImageOrientation.h"
65 #include "MoveSelectionCommand.h"
66 #include "Page.h"
67 #include "Pasteboard.h"
68 #include "PlatformKeyboardEvent.h"
69 #include "PluginDocument.h"
70 #include "PluginViewBase.h"
71 #include "Position.h"
72 #include "PromisedAttachmentInfo.h"
73 #include "RenderAttachment.h"
74 #include "RenderFileUploadControl.h"
75 #include "RenderImage.h"
76 #include "RenderView.h"
77 #include "ReplaceSelectionCommand.h"
78 #include "ResourceRequest.h"
79 #include "SecurityOrigin.h"
80 #include "Settings.h"
81 #include "ShadowRoot.h"
82 #include "StyleProperties.h"
83 #include "Text.h"
84 #include "TextEvent.h"
85 #include "VisiblePosition.h"
86 #include "markup.h"
87
88 #if ENABLE(DATA_INTERACTION)
89 #include "SelectionRect.h"
90 #endif
91
92 #include <wtf/RefPtr.h>
93 #include <wtf/SetForScope.h>
94 #endif
95
96 #if ENABLE(DATA_DETECTION)
97 #include "DataDetection.h"
98 #endif
99
100 namespace WebCore {
101
102 bool isDraggableLink(const Element& element)
103 {
104     if (is<HTMLAnchorElement>(element)) {
105         auto& anchorElement = downcast<HTMLAnchorElement>(element);
106         if (!anchorElement.isLiveLink())
107             return false;
108 #if ENABLE(DATA_DETECTION)
109         return !DataDetection::isDataDetectorURL(anchorElement.href());
110 #else
111         return true;
112 #endif
113     }
114     if (is<SVGAElement>(element))
115         return element.isLink();
116     return false;
117 }
118
119 #if ENABLE(DRAG_SUPPORT)
120     
121 static PlatformMouseEvent createMouseEvent(const DragData& dragData)
122 {
123     bool shiftKey = false;
124     bool ctrlKey = false;
125     bool altKey = false;
126     bool metaKey = false;
127
128     PlatformKeyboardEvent::getCurrentModifierState(shiftKey, ctrlKey, altKey, metaKey);
129
130     return PlatformMouseEvent(dragData.clientPosition(), dragData.globalPosition(),
131                               LeftButton, PlatformEvent::MouseMoved, 0, shiftKey, ctrlKey, altKey,
132                               metaKey, WallTime::now(), ForceAtClick, NoTap);
133 }
134
135 DragController::DragController(Page& page, DragClient& client)
136     : m_page(page)
137     , m_client(client)
138     , m_numberOfItemsToBeAccepted(0)
139     , m_dragHandlingMethod(DragHandlingMethod::None)
140     , m_dragDestinationAction(DragDestinationActionNone)
141     , m_dragSourceAction(DragSourceActionNone)
142     , m_didInitiateDrag(false)
143     , m_sourceDragOperation(DragOperationNone)
144 {
145 }
146
147 DragController::~DragController()
148 {
149     m_client.dragControllerDestroyed();
150 }
151
152 static RefPtr<DocumentFragment> documentFragmentFromDragData(const DragData& dragData, Frame& frame, Range& context, bool allowPlainText, bool& chosePlainText)
153 {
154     chosePlainText = false;
155
156     Document& document = context.ownerDocument();
157     if (dragData.containsCompatibleContent()) {
158         if (auto fragment = frame.editor().webContentFromPasteboard(*Pasteboard::createForDragAndDrop(dragData), context, allowPlainText, chosePlainText))
159             return fragment;
160
161         if (dragData.containsURL(DragData::DoNotConvertFilenames)) {
162             String title;
163             String url = dragData.asURL(DragData::DoNotConvertFilenames, &title);
164             if (!url.isEmpty()) {
165                 auto anchor = HTMLAnchorElement::create(document);
166                 anchor->setHref(url);
167                 if (title.isEmpty()) {
168                     // Try the plain text first because the url might be normalized or escaped.
169                     if (dragData.containsPlainText())
170                         title = dragData.asPlainText();
171                     if (title.isEmpty())
172                         title = url;
173                 }
174                 anchor->appendChild(document.createTextNode(title));
175                 auto fragment = document.createDocumentFragment();
176                 fragment->appendChild(anchor);
177                 return WTFMove(fragment);
178             }
179         }
180     }
181     if (allowPlainText && dragData.containsPlainText()) {
182         chosePlainText = true;
183         return createFragmentFromText(context, dragData.asPlainText()).ptr();
184     }
185
186     return nullptr;
187 }
188
189 #if !PLATFORM(IOS_FAMILY)
190
191 DragOperation DragController::platformGenericDragOperation()
192 {
193     return DragOperationMove;
194 }
195
196 #endif
197
198 bool DragController::dragIsMove(FrameSelection& selection, const DragData& dragData)
199 {
200     const VisibleSelection& visibleSelection = selection.selection();
201     return m_documentUnderMouse == m_dragInitiator && visibleSelection.isContentEditable() && visibleSelection.isRange() && !isCopyKeyDown(dragData);
202 }
203
204 void DragController::clearDragCaret()
205 {
206     m_page.dragCaretController().clear();
207 }
208
209 void DragController::dragEnded()
210 {
211     m_dragInitiator = nullptr;
212     m_didInitiateDrag = false;
213     m_documentUnderMouse = nullptr;
214     clearDragCaret();
215     
216     m_client.dragEnded();
217 }
218
219 DragOperation DragController::dragEntered(const DragData& dragData)
220 {
221     return dragEnteredOrUpdated(dragData);
222 }
223
224 void DragController::dragExited(const DragData& dragData)
225 {
226     auto& mainFrame = m_page.mainFrame();
227     if (mainFrame.view())
228         mainFrame.eventHandler().cancelDragAndDrop(createMouseEvent(dragData), Pasteboard::createForDragAndDrop(dragData), dragData.draggingSourceOperationMask(), dragData.containsFiles());
229     mouseMovedIntoDocument(nullptr);
230     if (m_fileInputElementUnderMouse)
231         m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(false);
232     m_fileInputElementUnderMouse = nullptr;
233 }
234
235 DragOperation DragController::dragUpdated(const DragData& dragData)
236 {
237     return dragEnteredOrUpdated(dragData);
238 }
239
240 inline static bool dragIsHandledByDocument(DragController::DragHandlingMethod dragHandlingMethod)
241 {
242     if (dragHandlingMethod == DragController::DragHandlingMethod::None)
243         return false;
244
245     if (dragHandlingMethod == DragController::DragHandlingMethod::PageLoad)
246         return false;
247
248     return true;
249 }
250
251 bool DragController::performDragOperation(const DragData& dragData)
252 {
253     SetForScope<bool> isPerformingDrop(m_isPerformingDrop, true);
254     TemporarySelectionChange ignoreSelectionChanges(m_page.focusController().focusedOrMainFrame(), std::nullopt, TemporarySelectionOption::IgnoreSelectionChanges);
255
256     m_documentUnderMouse = m_page.mainFrame().documentAtPoint(dragData.clientPosition());
257
258     ShouldOpenExternalURLsPolicy shouldOpenExternalURLsPolicy = ShouldOpenExternalURLsPolicy::ShouldNotAllow;
259     if (m_documentUnderMouse)
260         shouldOpenExternalURLsPolicy = m_documentUnderMouse->shouldOpenExternalURLsPolicyToPropagate();
261
262     if ((m_dragDestinationAction & DragDestinationActionDHTML) && dragIsHandledByDocument(m_dragHandlingMethod)) {
263         m_client.willPerformDragDestinationAction(DragDestinationActionDHTML, dragData);
264         Ref<Frame> mainFrame(m_page.mainFrame());
265         bool preventedDefault = false;
266         if (mainFrame->view())
267             preventedDefault = mainFrame->eventHandler().performDragAndDrop(createMouseEvent(dragData), Pasteboard::createForDragAndDrop(dragData), dragData.draggingSourceOperationMask(), dragData.containsFiles());
268         if (preventedDefault) {
269             clearDragCaret();
270             m_documentUnderMouse = nullptr;
271             return true;
272         }
273     }
274
275     if ((m_dragDestinationAction & DragDestinationActionEdit) && concludeEditDrag(dragData)) {
276         m_client.didConcludeEditDrag();
277         m_documentUnderMouse = nullptr;
278         clearDragCaret();
279         return true;
280     }
281
282     m_documentUnderMouse = nullptr;
283     clearDragCaret();
284
285     if (operationForLoad(dragData) == DragOperationNone)
286         return false;
287
288     auto urlString = dragData.asURL();
289     if (urlString.isEmpty())
290         return false;
291
292     m_client.willPerformDragDestinationAction(DragDestinationActionLoad, dragData);
293     m_page.mainFrame().loader().load(FrameLoadRequest(m_page.mainFrame(), { urlString }, shouldOpenExternalURLsPolicy));
294     return true;
295 }
296
297 void DragController::mouseMovedIntoDocument(Document* newDocument)
298 {
299     if (m_documentUnderMouse == newDocument)
300         return;
301
302     // If we were over another document clear the selection
303     if (m_documentUnderMouse)
304         clearDragCaret();
305     m_documentUnderMouse = newDocument;
306 }
307
308 DragOperation DragController::dragEnteredOrUpdated(const DragData& dragData)
309 {
310     mouseMovedIntoDocument(m_page.mainFrame().documentAtPoint(dragData.clientPosition()));
311
312     m_dragDestinationAction = dragData.dragDestinationAction();
313     if (m_dragDestinationAction == DragDestinationActionNone) {
314         clearDragCaret(); // FIXME: Why not call mouseMovedIntoDocument(nullptr)?
315         return DragOperationNone;
316     }
317
318     DragOperation dragOperation = DragOperationNone;
319     m_dragHandlingMethod = tryDocumentDrag(dragData, m_dragDestinationAction, dragOperation);
320     if (m_dragHandlingMethod == DragHandlingMethod::None && (m_dragDestinationAction & DragDestinationActionLoad)) {
321         dragOperation = operationForLoad(dragData);
322         if (dragOperation != DragOperationNone)
323             m_dragHandlingMethod = DragHandlingMethod::PageLoad;
324     } else if (m_dragHandlingMethod == DragHandlingMethod::SetColor)
325         dragOperation = DragOperationCopy;
326
327     updateSupportedTypeIdentifiersForDragHandlingMethod(m_dragHandlingMethod, dragData);
328     return dragOperation;
329 }
330
331 static HTMLInputElement* asFileInput(Node& node)
332 {
333     if (!is<HTMLInputElement>(node))
334         return nullptr;
335
336     auto* inputElement = &downcast<HTMLInputElement>(node);
337
338     // If this is a button inside of the a file input, move up to the file input.
339     if (inputElement->isTextButton() && is<ShadowRoot>(inputElement->treeScope().rootNode())) {
340         auto& host = *downcast<ShadowRoot>(inputElement->treeScope().rootNode()).host();
341         inputElement = is<HTMLInputElement>(host) ? &downcast<HTMLInputElement>(host) : nullptr;
342     }
343
344     return inputElement && inputElement->isFileUpload() ? inputElement : nullptr;
345 }
346
347 #if ENABLE(INPUT_TYPE_COLOR)
348 static bool isEnabledColorInput(Node& node, bool setToShadowAncestor)
349 {
350     Node* candidate = setToShadowAncestor ? node.deprecatedShadowAncestorNode() : &node;
351     if (is<HTMLInputElement>(*candidate)) {
352         auto& input = downcast<HTMLInputElement>(*candidate);
353         if (input.isColorControl() && !input.isDisabledFormControl())
354             return true;
355     }
356
357     return false;
358 }
359 #endif
360
361 // This can return null if an empty document is loaded.
362 static Element* elementUnderMouse(Document* documentUnderMouse, const IntPoint& p)
363 {
364     Frame* frame = documentUnderMouse->frame();
365     float zoomFactor = frame ? frame->pageZoomFactor() : 1;
366     LayoutPoint point(p.x() * zoomFactor, p.y() * zoomFactor);
367
368     HitTestResult result(point);
369     documentUnderMouse->renderView()->hitTest(HitTestRequest(), result);
370
371     Node* node = result.innerNode();
372     while (node && !is<Element>(*node))
373         node = node->parentNode();
374     if (node)
375         node = node->deprecatedShadowAncestorNode();
376
377     return downcast<Element>(node);
378 }
379
380 #if !ENABLE(DATA_INTERACTION)
381
382 void DragController::updateSupportedTypeIdentifiersForDragHandlingMethod(DragHandlingMethod, const DragData&) const
383 {
384 }
385
386 #endif
387
388 DragController::DragHandlingMethod DragController::tryDocumentDrag(const DragData& dragData, DragDestinationAction actionMask, DragOperation& dragOperation)
389 {
390     if (!m_documentUnderMouse)
391         return DragHandlingMethod::None;
392
393     if (m_dragInitiator && !m_documentUnderMouse->securityOrigin().canReceiveDragData(m_dragInitiator->securityOrigin()))
394         return DragHandlingMethod::None;
395
396     bool isHandlingDrag = false;
397     if (actionMask & DragDestinationActionDHTML) {
398         isHandlingDrag = tryDHTMLDrag(dragData, dragOperation);
399         // Do not continue if m_documentUnderMouse has been reset by tryDHTMLDrag.
400         // tryDHTMLDrag fires dragenter event. The event listener that listens
401         // to this event may create a nested message loop (open a modal dialog),
402         // which could process dragleave event and reset m_documentUnderMouse in
403         // dragExited.
404         if (!m_documentUnderMouse)
405             return DragHandlingMethod::None;
406     }
407
408     // It's unclear why this check is after tryDHTMLDrag.
409     // We send drag events in tryDHTMLDrag and that may be the reason.
410     RefPtr<FrameView> frameView = m_documentUnderMouse->view();
411     if (!frameView)
412         return DragHandlingMethod::None;
413
414     if (isHandlingDrag) {
415         clearDragCaret();
416         m_numberOfItemsToBeAccepted = dragData.numberOfFiles();
417         return DragHandlingMethod::NonDefault;
418     }
419
420     if ((actionMask & DragDestinationActionEdit) && canProcessDrag(dragData)) {
421         if (dragData.containsColor()) {
422             dragOperation = DragOperationGeneric;
423             return DragHandlingMethod::SetColor;
424         }
425
426         IntPoint point = frameView->windowToContents(dragData.clientPosition());
427         Element* element = elementUnderMouse(m_documentUnderMouse.get(), point);
428         if (!element)
429             return DragHandlingMethod::None;
430         
431         HTMLInputElement* elementAsFileInput = asFileInput(*element);
432         if (m_fileInputElementUnderMouse != elementAsFileInput) {
433             if (m_fileInputElementUnderMouse)
434                 m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(false);
435             m_fileInputElementUnderMouse = elementAsFileInput;
436         }
437         
438         if (!m_fileInputElementUnderMouse)
439             m_page.dragCaretController().setCaretPosition(m_documentUnderMouse->frame()->visiblePositionForPoint(point));
440         else
441             clearDragCaret();
442
443         Frame* innerFrame = element->document().frame();
444         dragOperation = dragIsMove(innerFrame->selection(), dragData) ? DragOperationMove : DragOperationCopy;
445         m_numberOfItemsToBeAccepted = 0;
446
447         unsigned numberOfFiles = dragData.numberOfFiles();
448         if (m_fileInputElementUnderMouse) {
449             if (m_fileInputElementUnderMouse->isDisabledFormControl())
450                 m_numberOfItemsToBeAccepted = 0;
451             else if (m_fileInputElementUnderMouse->multiple())
452                 m_numberOfItemsToBeAccepted = numberOfFiles;
453             else if (numberOfFiles > 1)
454                 m_numberOfItemsToBeAccepted = 0;
455             else
456                 m_numberOfItemsToBeAccepted = 1;
457             
458             if (!m_numberOfItemsToBeAccepted)
459                 dragOperation = DragOperationNone;
460             m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(m_numberOfItemsToBeAccepted);
461         } else {
462             // We are not over a file input element. The dragged item(s) will only
463             // be loaded into the view the number of dragged items is 1.
464             m_numberOfItemsToBeAccepted = numberOfFiles != 1 ? 0 : 1;
465         }
466
467         if (m_fileInputElementUnderMouse)
468             return DragHandlingMethod::UploadFile;
469
470         if (m_page.dragCaretController().isContentRichlyEditable())
471             return DragHandlingMethod::EditRichText;
472
473         return DragHandlingMethod::EditPlainText;
474     }
475     
476     // We are not over an editable region. Make sure we're clearing any prior drag cursor.
477     clearDragCaret();
478     if (m_fileInputElementUnderMouse)
479         m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(false);
480     m_fileInputElementUnderMouse = nullptr;
481     return DragHandlingMethod::None;
482 }
483
484 DragSourceAction DragController::delegateDragSourceAction(const IntPoint& rootViewPoint)
485 {
486     m_dragSourceAction = m_client.dragSourceActionMaskForPoint(rootViewPoint);
487     return m_dragSourceAction;
488 }
489
490 DragOperation DragController::operationForLoad(const DragData& dragData)
491 {
492     Document* document = m_page.mainFrame().documentAtPoint(dragData.clientPosition());
493
494     bool pluginDocumentAcceptsDrags = false;
495
496     if (is<PluginDocument>(document)) {
497         const Widget* widget = downcast<PluginDocument>(*document).pluginWidget();
498         const PluginViewBase* pluginView = is<PluginViewBase>(widget) ? downcast<PluginViewBase>(widget) : nullptr;
499
500         if (pluginView)
501             pluginDocumentAcceptsDrags = pluginView->shouldAllowNavigationFromDrags();
502     }
503
504     if (document && (m_didInitiateDrag || (is<PluginDocument>(*document) && !pluginDocumentAcceptsDrags) || document->hasEditableStyle()))
505         return DragOperationNone;
506     return dragOperation(dragData);
507 }
508
509 static bool setSelectionToDragCaret(Frame* frame, VisibleSelection& dragCaret, RefPtr<Range>& range, const IntPoint& point)
510 {
511     Ref<Frame> protector(*frame);
512     frame->selection().setSelection(dragCaret);
513     if (frame->selection().selection().isNone()) {
514         dragCaret = frame->visiblePositionForPoint(point);
515         frame->selection().setSelection(dragCaret);
516         range = dragCaret.toNormalizedRange();
517     }
518     return !frame->selection().isNone() && frame->selection().selection().isContentEditable();
519 }
520
521 bool DragController::dispatchTextInputEventFor(Frame* innerFrame, const DragData& dragData)
522 {
523     ASSERT(m_page.dragCaretController().hasCaret());
524     String text = m_page.dragCaretController().isContentRichlyEditable() ? emptyString() : dragData.asPlainText();
525     Element* target = innerFrame->editor().findEventTargetFrom(m_page.dragCaretController().caretPosition());
526     // FIXME: What guarantees target is not null?
527     auto event = TextEvent::createForDrop(&innerFrame->windowProxy(), text);
528     target->dispatchEvent(event);
529     return !event->defaultPrevented();
530 }
531
532 bool DragController::concludeEditDrag(const DragData& dragData)
533 {
534     RefPtr<HTMLInputElement> fileInput = m_fileInputElementUnderMouse;
535     if (m_fileInputElementUnderMouse) {
536         m_fileInputElementUnderMouse->setCanReceiveDroppedFiles(false);
537         m_fileInputElementUnderMouse = nullptr;
538     }
539
540     if (!m_documentUnderMouse)
541         return false;
542
543     IntPoint point = m_documentUnderMouse->view()->windowToContents(dragData.clientPosition());
544     Element* element = elementUnderMouse(m_documentUnderMouse.get(), point);
545     if (!element)
546         return false;
547     RefPtr<Frame> innerFrame = element->document().frame();
548     ASSERT(innerFrame);
549
550     if (m_page.dragCaretController().hasCaret() && !dispatchTextInputEventFor(innerFrame.get(), dragData))
551         return true;
552
553     if (dragData.containsColor()) {
554         Color color = dragData.asColor();
555         if (!color.isValid())
556             return false;
557 #if ENABLE(INPUT_TYPE_COLOR)
558         if (isEnabledColorInput(*element, false)) {
559             auto& input = downcast<HTMLInputElement>(*element);
560             input.setValue(color.serialized(), DispatchInputAndChangeEvent);
561             return true;
562         }
563 #endif
564         auto innerRange = innerFrame->selection().toNormalizedRange();
565         if (!innerRange)
566             return false;
567         auto style = MutableStyleProperties::create();
568         style->setProperty(CSSPropertyColor, color.serialized(), false);
569         if (!innerFrame->editor().shouldApplyStyle(style.ptr(), innerRange.get()))
570             return false;
571         m_client.willPerformDragDestinationAction(DragDestinationActionEdit, dragData);
572         innerFrame->editor().applyStyle(style.ptr(), EditAction::SetColor);
573         return true;
574     }
575
576     if (dragData.containsFiles() && fileInput) {
577         // fileInput should be the element we hit tested for, unless it was made
578         // display:none in a drop event handler.
579         ASSERT(fileInput == element || !fileInput->renderer());
580         if (fileInput->isDisabledFormControl())
581             return false;
582
583         return fileInput->receiveDroppedFiles(dragData);
584     }
585
586     if (!m_page.dragController().canProcessDrag(dragData))
587         return false;
588
589     VisibleSelection dragCaret = m_page.dragCaretController().caretPosition();
590     RefPtr<Range> range = dragCaret.toNormalizedRange();
591     RefPtr<Element> rootEditableElement = innerFrame->selection().selection().rootEditableElement();
592
593     // For range to be null a WebKit client must have done something bad while
594     // manually controlling drag behaviour
595     if (!range)
596         return false;
597
598     ResourceCacheValidationSuppressor validationSuppressor(range->ownerDocument().cachedResourceLoader());
599     auto& editor = innerFrame->editor();
600     bool isMove = dragIsMove(innerFrame->selection(), dragData);
601     if (isMove || dragCaret.isContentRichlyEditable()) {
602         bool chosePlainText = false;
603         RefPtr<DocumentFragment> fragment = documentFragmentFromDragData(dragData, *innerFrame, *range, true, chosePlainText);
604         if (!fragment || !editor.shouldInsertFragment(*fragment, range.get(), EditorInsertAction::Dropped))
605             return false;
606
607         m_client.willPerformDragDestinationAction(DragDestinationActionEdit, dragData);
608
609         if (editor.client() && editor.client()->performTwoStepDrop(*fragment, *range, isMove))
610             return true;
611
612         if (isMove) {
613             // NSTextView behavior is to always smart delete on moving a selection,
614             // but only to smart insert if the selection granularity is word granularity.
615             bool smartDelete = editor.smartInsertDeleteEnabled();
616             bool smartInsert = smartDelete && innerFrame->selection().granularity() == WordGranularity && dragData.canSmartReplace();
617             MoveSelectionCommand::create(fragment.releaseNonNull(), dragCaret.base(), smartInsert, smartDelete)->apply();
618         } else {
619             if (setSelectionToDragCaret(innerFrame.get(), dragCaret, range, point)) {
620                 OptionSet<ReplaceSelectionCommand::CommandOption> options { ReplaceSelectionCommand::SelectReplacement, ReplaceSelectionCommand::PreventNesting };
621                 if (dragData.canSmartReplace())
622                     options.add(ReplaceSelectionCommand::SmartReplace);
623                 if (chosePlainText)
624                     options.add(ReplaceSelectionCommand::MatchStyle);
625                 ReplaceSelectionCommand::create(*m_documentUnderMouse, fragment.releaseNonNull(), options, EditAction::InsertFromDrop)->apply();
626             }
627         }
628     } else {
629         String text = dragData.asPlainText();
630         if (text.isEmpty() || !editor.shouldInsertText(text, range.get(), EditorInsertAction::Dropped))
631             return false;
632
633         m_client.willPerformDragDestinationAction(DragDestinationActionEdit, dragData);
634         RefPtr<DocumentFragment> fragment = createFragmentFromText(*range, text);
635         if (!fragment)
636             return false;
637
638         if (editor.client() && editor.client()->performTwoStepDrop(*fragment, *range, isMove))
639             return true;
640
641         if (setSelectionToDragCaret(innerFrame.get(), dragCaret, range, point))
642             ReplaceSelectionCommand::create(*m_documentUnderMouse, fragment.get(), { ReplaceSelectionCommand::SelectReplacement, ReplaceSelectionCommand::MatchStyle, ReplaceSelectionCommand::PreventNesting }, EditAction::InsertFromDrop)->apply();
643     }
644
645     if (rootEditableElement) {
646         if (Frame* frame = rootEditableElement->document().frame())
647             frame->eventHandler().updateDragStateAfterEditDragIfNeeded(*rootEditableElement);
648     }
649
650     return true;
651 }
652
653 bool DragController::canProcessDrag(const DragData& dragData)
654 {
655     IntPoint point = m_page.mainFrame().view()->windowToContents(dragData.clientPosition());
656     HitTestResult result = HitTestResult(point);
657     if (!m_page.mainFrame().contentRenderer())
658         return false;
659
660     result = m_page.mainFrame().eventHandler().hitTestResultAtPoint(point, HitTestRequest::ReadOnly | HitTestRequest::Active);
661
662     if (!result.innerNonSharedNode())
663         return false;
664
665     DragData::DraggingPurpose dragPurpose = DragData::DraggingPurpose::ForEditing;
666     if (asFileInput(*result.innerNonSharedNode()))
667         dragPurpose = DragData::DraggingPurpose::ForFileUpload;
668 #if ENABLE(INPUT_TYPE_COLOR)
669     else if (isEnabledColorInput(*result.innerNonSharedNode(), true))
670         dragPurpose = DragData::DraggingPurpose::ForColorControl;
671 #endif
672
673     if (!dragData.containsCompatibleContent(dragPurpose))
674         return false;
675
676     if (dragPurpose == DragData::DraggingPurpose::ForFileUpload)
677         return true;
678
679 #if ENABLE(INPUT_TYPE_COLOR)
680     if (dragPurpose == DragData::DraggingPurpose::ForColorControl)
681         return true;
682 #endif
683
684     if (is<HTMLPlugInElement>(*result.innerNonSharedNode())) {
685         if (!downcast<HTMLPlugInElement>(result.innerNonSharedNode())->canProcessDrag() && !result.innerNonSharedNode()->hasEditableStyle())
686             return false;
687     } else if (!result.innerNonSharedNode()->hasEditableStyle())
688         return false;
689
690     if (m_didInitiateDrag && m_documentUnderMouse == m_dragInitiator && result.isSelected())
691         return false;
692
693     return true;
694 }
695
696 static DragOperation defaultOperationForDrag(DragOperation srcOpMask)
697 {
698     // This is designed to match IE's operation fallback for the case where
699     // the page calls preventDefault() in a drag event but doesn't set dropEffect.
700     if (srcOpMask == DragOperationEvery)
701         return DragOperationCopy;
702     if (srcOpMask == DragOperationNone)
703         return DragOperationNone;
704     if (srcOpMask & DragOperationMove)
705         return DragOperationMove;
706     if (srcOpMask & DragOperationGeneric)
707         return DragController::platformGenericDragOperation();
708     if (srcOpMask & DragOperationCopy)
709         return DragOperationCopy;
710     if (srcOpMask & DragOperationLink)
711         return DragOperationLink;
712     
713     // FIXME: Does IE really return "generic" even if no operations were allowed by the source?
714     return DragOperationGeneric;
715 }
716
717 bool DragController::tryDHTMLDrag(const DragData& dragData, DragOperation& operation)
718 {
719     ASSERT(m_documentUnderMouse);
720     Ref<Frame> mainFrame(m_page.mainFrame());
721     RefPtr<FrameView> viewProtector = mainFrame->view();
722     if (!viewProtector)
723         return false;
724
725     DragOperation sourceOperation = dragData.draggingSourceOperationMask();
726     auto targetResponse = mainFrame->eventHandler().updateDragAndDrop(createMouseEvent(dragData), [&dragData]() { return Pasteboard::createForDragAndDrop(dragData); }, sourceOperation, dragData.containsFiles());
727     if (!targetResponse.accept)
728         return false;
729
730     if (!targetResponse.operation)
731         operation = defaultOperationForDrag(sourceOperation);
732     else if (!(sourceOperation & targetResponse.operation.value())) // The element picked an operation which is not supported by the source
733         operation = DragOperationNone;
734     else
735         operation = targetResponse.operation.value();
736
737     return true;
738 }
739
740 static bool imageElementIsDraggable(const HTMLImageElement& image, const Frame& sourceFrame)
741 {
742     if (sourceFrame.settings().loadsImagesAutomatically())
743         return true;
744
745     auto* renderer = image.renderer();
746     if (!is<RenderImage>(renderer))
747         return false;
748
749     auto* cachedImage = downcast<RenderImage>(*renderer).cachedImage();
750     return cachedImage && !cachedImage->errorOccurred() && cachedImage->imageForRenderer(renderer);
751 }
752
753 #if ENABLE(ATTACHMENT_ELEMENT)
754
755 static RefPtr<HTMLAttachmentElement> enclosingAttachmentElement(Element& element)
756 {
757     if (is<HTMLAttachmentElement>(element))
758         return downcast<HTMLAttachmentElement>(&element);
759
760     if (is<HTMLAttachmentElement>(element.parentOrShadowHostElement()))
761         return downcast<HTMLAttachmentElement>(element.parentOrShadowHostElement());
762
763     return { };
764 }
765
766 #endif
767
768 Element* DragController::draggableElement(const Frame* sourceFrame, Element* startElement, const IntPoint& dragOrigin, DragState& state) const
769 {
770     state.type = (sourceFrame->selection().contains(dragOrigin)) ? DragSourceActionSelection : DragSourceActionNone;
771     if (!startElement)
772         return nullptr;
773 #if ENABLE(ATTACHMENT_ELEMENT)
774     if (auto attachment = enclosingAttachmentElement(*startElement)) {
775         auto selection = sourceFrame->selection().selection();
776         bool isSingleAttachmentSelection = selection.start() == Position(attachment.get(), Position::PositionIsBeforeAnchor) && selection.end() == Position(attachment.get(), Position::PositionIsAfterAnchor);
777         bool isAttachmentElementInCurrentSelection = false;
778         if (auto selectedRange = selection.toNormalizedRange()) {
779             auto compareResult = selectedRange->compareNode(*attachment);
780             isAttachmentElementInCurrentSelection = !compareResult.hasException() && compareResult.releaseReturnValue() == Range::NODE_INSIDE;
781         }
782
783         if (!isAttachmentElementInCurrentSelection || isSingleAttachmentSelection) {
784             state.type = DragSourceActionAttachment;
785             return attachment.get();
786         }
787     }
788 #endif
789
790     for (auto* element = startElement; element; element = element->parentOrShadowHostElement()) {
791         auto* renderer = element->renderer();
792         if (!renderer)
793             continue;
794
795         UserDrag dragMode = renderer->style().userDrag();
796         if ((m_dragSourceAction & DragSourceActionDHTML) && dragMode == UserDrag::Element) {
797             state.type = static_cast<DragSourceAction>(state.type | DragSourceActionDHTML);
798             return element;
799         }
800         if (dragMode == UserDrag::Auto) {
801             if ((m_dragSourceAction & DragSourceActionImage)
802                 && is<HTMLImageElement>(*element)
803                 && imageElementIsDraggable(downcast<HTMLImageElement>(*element), *sourceFrame)) {
804                 state.type = static_cast<DragSourceAction>(state.type | DragSourceActionImage);
805                 return element;
806             }
807             if ((m_dragSourceAction & DragSourceActionLink) && isDraggableLink(*element)) {
808                 state.type = static_cast<DragSourceAction>(state.type | DragSourceActionLink);
809                 return element;
810             }
811 #if ENABLE(ATTACHMENT_ELEMENT)
812             if ((m_dragSourceAction & DragSourceActionAttachment)
813                 && is<HTMLAttachmentElement>(*element)
814                 && downcast<HTMLAttachmentElement>(*element).file()) {
815                 state.type = static_cast<DragSourceAction>(state.type | DragSourceActionAttachment);
816                 return element;
817             }
818 #endif
819 #if ENABLE(INPUT_TYPE_COLOR)
820             if ((m_dragSourceAction & DragSourceActionColor)
821                 && isEnabledColorInput(*element, false)) {
822                 state.type = static_cast<DragSourceAction>(state.type | DragSourceActionColor);
823                 return element;
824             }
825 #endif
826         }
827     }
828
829     // We either have nothing to drag or we have a selection and we're not over a draggable element.
830     return (state.type & DragSourceActionSelection) ? startElement : nullptr;
831 }
832
833 static CachedImage* getCachedImage(Element& element)
834 {
835     RenderObject* renderer = element.renderer();
836     if (!is<RenderImage>(renderer))
837         return nullptr;
838     auto& image = downcast<RenderImage>(*renderer);
839     return image.cachedImage();
840 }
841
842 static Image* getImage(Element& element)
843 {
844     CachedImage* cachedImage = getCachedImage(element);
845     // Don't use cachedImage->imageForRenderer() here as that may return BitmapImages for cached SVG Images.
846     // Users of getImage() want access to the SVGImage, in order to figure out the filename extensions,
847     // which would be empty when asking the cached BitmapImages.
848     return (cachedImage && !cachedImage->errorOccurred()) ?
849         cachedImage->image() : nullptr;
850 }
851
852 static void selectElement(Element& element)
853 {
854     RefPtr<Range> range = element.document().createRange();
855     range->selectNode(element);
856     element.document().frame()->selection().setSelection(VisibleSelection(*range, DOWNSTREAM));
857 }
858
859 static IntPoint dragLocForDHTMLDrag(const IntPoint& mouseDraggedPoint, const IntPoint& dragOrigin, const IntPoint& dragImageOffset, bool isLinkImage)
860 {
861     // dragImageOffset is the cursor position relative to the lower-left corner of the image.
862 #if PLATFORM(MAC)
863     // We add in the Y dimension because we are a flipped view, so adding moves the image down.
864     const int yOffset = dragImageOffset.y();
865 #else
866     const int yOffset = -dragImageOffset.y();
867 #endif
868
869     if (isLinkImage)
870         return IntPoint(mouseDraggedPoint.x() - dragImageOffset.x(), mouseDraggedPoint.y() + yOffset);
871
872     return IntPoint(dragOrigin.x() - dragImageOffset.x(), dragOrigin.y() + yOffset);
873 }
874
875 static FloatPoint dragImageAnchorPointForSelectionDrag(Frame& frame, const IntPoint& mouseDraggedPoint)
876 {
877     IntRect draggingRect = enclosingIntRect(frame.selection().selectionBounds());
878
879     float x = (mouseDraggedPoint.x() - draggingRect.x()) / (float)draggingRect.width();
880     float y = (mouseDraggedPoint.y() - draggingRect.y()) / (float)draggingRect.height();
881
882     return FloatPoint { x, y };
883 }
884
885 static IntPoint dragLocForSelectionDrag(Frame& src)
886 {
887     IntRect draggingRect = enclosingIntRect(src.selection().selectionBounds());
888     int xpos = draggingRect.maxX();
889     xpos = draggingRect.x() < xpos ? draggingRect.x() : xpos;
890     int ypos = draggingRect.maxY();
891 #if PLATFORM(COCOA)
892     // Deal with flipped coordinates on Mac
893     ypos = draggingRect.y() > ypos ? draggingRect.y() : ypos;
894 #else
895     ypos = draggingRect.y() < ypos ? draggingRect.y() : ypos;
896 #endif
897     return IntPoint(xpos, ypos);
898 }
899
900 bool DragController::startDrag(Frame& src, const DragState& state, DragOperation srcOp, const PlatformMouseEvent& dragEvent, const IntPoint& dragOrigin, HasNonDefaultPasteboardData hasData)
901 {
902     if (!src.view() || !src.contentRenderer() || !state.source)
903         return false;
904
905     Ref<Frame> protector(src);
906     HitTestResult hitTestResult = src.eventHandler().hitTestResultAtPoint(dragOrigin, HitTestRequest::ReadOnly | HitTestRequest::Active);
907
908     bool sourceContainsHitNode = state.source->containsIncludingShadowDOM(hitTestResult.innerNode());
909     if (!sourceContainsHitNode) {
910         // The original node being dragged isn't under the drag origin anymore... maybe it was
911         // hidden or moved out from under the cursor. Regardless, we don't want to start a drag on
912         // something that's not actually under the drag origin.
913         return false;
914     }
915
916     URL linkURL = hitTestResult.absoluteLinkURL();
917     URL imageURL = hitTestResult.absoluteImageURL();
918
919     IntPoint mouseDraggedPoint = src.view()->windowToContents(dragEvent.position());
920
921     m_draggingImageURL = URL();
922     m_sourceDragOperation = srcOp;
923
924     DragImage dragImage;
925     IntPoint dragLoc(0, 0);
926     IntPoint dragImageOffset(0, 0);
927
928     ASSERT(state.dataTransfer);
929
930     DataTransfer& dataTransfer = *state.dataTransfer;
931     if (state.type == DragSourceActionDHTML) {
932         dragImage = DragImage { dataTransfer.createDragImage(dragImageOffset) };
933         // We allow DHTML/JS to set the drag image, even if its a link, image or text we're dragging.
934         // This is in the spirit of the IE API, which allows overriding of pasteboard data and DragOp.
935         if (dragImage) {
936             dragLoc = dragLocForDHTMLDrag(mouseDraggedPoint, dragOrigin, dragImageOffset, !linkURL.isEmpty());
937             m_dragOffset = dragImageOffset;
938         }
939     }
940
941     if (state.type == DragSourceActionSelection || !imageURL.isEmpty() || !linkURL.isEmpty()) {
942         // Selection, image, and link drags receive a default set of allowed drag operations that
943         // follows from:
944         // http://trac.webkit.org/browser/trunk/WebKit/mac/WebView/WebHTMLView.mm?rev=48526#L3430
945         m_sourceDragOperation = static_cast<DragOperation>(m_sourceDragOperation | DragOperationGeneric | DragOperationCopy);
946     }
947
948     ASSERT(state.source);
949     Element& element = *state.source;
950
951     bool mustUseLegacyDragClient = hasData == HasNonDefaultPasteboardData::Yes || m_client.useLegacyDragClient();
952
953     IntRect dragImageBounds;
954     Image* image = getImage(element);
955     if (state.type == DragSourceActionSelection) {
956         PasteboardWriterData pasteboardWriterData;
957
958         if (hasData == HasNonDefaultPasteboardData::No) {
959             if (src.selection().selection().isNone()) {
960                 // The page may have cleared out the selection in the dragstart handler, in which case we should bail
961                 // out of the drag, since there is no content to write to the pasteboard.
962                 return false;
963             }
964
965             // FIXME: This entire block is almost identical to the code in Editor::copy, and the code should be shared.
966             RefPtr<Range> selectionRange = src.selection().toNormalizedRange();
967             ASSERT(selectionRange);
968
969             src.editor().willWriteSelectionToPasteboard(selectionRange.get());
970
971             if (enclosingTextFormControl(src.selection().selection().start())) {
972                 if (mustUseLegacyDragClient)
973                     dataTransfer.pasteboard().writePlainText(src.editor().selectedTextForDataTransfer(), Pasteboard::CannotSmartReplace);
974                 else {
975                     PasteboardWriterData::PlainText plainText;
976                     plainText.canSmartCopyOrDelete = false;
977                     plainText.text = src.editor().selectedTextForDataTransfer();
978                     pasteboardWriterData.setPlainText(WTFMove(plainText));
979                 }
980             } else {
981                 if (mustUseLegacyDragClient) {
982 #if PLATFORM(COCOA) || PLATFORM(GTK)
983                     src.editor().writeSelectionToPasteboard(dataTransfer.pasteboard());
984 #else
985                     // FIXME: Convert all other platforms to match Mac and delete this.
986                     dataTransfer.pasteboard().writeSelection(*selectionRange, src.editor().canSmartCopyOrDelete(), src, IncludeImageAltTextForDataTransfer);
987 #endif
988                 } else {
989 #if PLATFORM(COCOA)
990                     src.editor().writeSelection(pasteboardWriterData);
991 #endif
992                 }
993             }
994
995             src.editor().didWriteSelectionToPasteboard();
996         }
997         m_client.willPerformDragSourceAction(DragSourceActionSelection, dragOrigin, dataTransfer);
998         if (!dragImage) {
999             TextIndicatorData textIndicator;
1000             dragImage = DragImage { dissolveDragImageToFraction(createDragImageForSelection(src, textIndicator), DragImageAlpha) };
1001             if (textIndicator.contentImage)
1002                 dragImage.setIndicatorData(textIndicator);
1003             dragLoc = dragLocForSelectionDrag(src);
1004             m_dragOffset = IntPoint(dragOrigin.x() - dragLoc.x(), dragOrigin.y() - dragLoc.y());
1005         }
1006
1007         if (!dragImage)
1008             return false;
1009
1010         if (mustUseLegacyDragClient) {
1011             doSystemDrag(WTFMove(dragImage), dragLoc, dragOrigin, src, state, { });
1012             return true;
1013         }
1014
1015         DragItem dragItem;
1016         dragItem.imageAnchorPoint = dragImageAnchorPointForSelectionDrag(src, mouseDraggedPoint);
1017         dragItem.image = WTFMove(dragImage);
1018         dragItem.data = WTFMove(pasteboardWriterData);
1019
1020         beginDrag(WTFMove(dragItem), src, dragOrigin, mouseDraggedPoint, dataTransfer, DragSourceActionSelection);
1021
1022         return true;
1023     }
1024
1025     if (!src.document()->securityOrigin().canDisplay(linkURL)) {
1026         src.document()->addConsoleMessage(MessageSource::Security, MessageLevel::Error, "Not allowed to drag local resource: " + linkURL.stringCenterEllipsizedToLength());
1027         return false;
1028     }
1029
1030     if (!imageURL.isEmpty() && image && !image->isNull() && (m_dragSourceAction & DragSourceActionImage)) {
1031         // We shouldn't be starting a drag for an image that can't provide an extension.
1032         // This is an early detection for problems encountered later upon drop.
1033         ASSERT(!image->filenameExtension().isEmpty());
1034
1035 #if ENABLE(ATTACHMENT_ELEMENT)
1036         auto attachmentInfo = promisedAttachmentInfo(src, element);
1037 #else
1038         PromisedAttachmentInfo attachmentInfo;
1039 #endif
1040
1041         if (hasData == HasNonDefaultPasteboardData::No) {
1042             m_draggingImageURL = imageURL;
1043             if (element.isContentRichlyEditable())
1044                 selectElement(element);
1045             if (!attachmentInfo)
1046                 declareAndWriteDragImage(dataTransfer, element, !linkURL.isEmpty() ? linkURL : imageURL, hitTestResult.altDisplayString());
1047         }
1048
1049         m_client.willPerformDragSourceAction(DragSourceActionImage, dragOrigin, dataTransfer);
1050
1051         if (!dragImage)
1052             doImageDrag(element, dragOrigin, hitTestResult.imageRect(), src, m_dragOffset, state, WTFMove(attachmentInfo));
1053         else {
1054             // DHTML defined drag image
1055             doSystemDrag(WTFMove(dragImage), dragLoc, dragOrigin, src, state, WTFMove(attachmentInfo));
1056         }
1057
1058         return true;
1059     }
1060
1061     if (!linkURL.isEmpty() && (m_dragSourceAction & DragSourceActionLink)) {
1062         PasteboardWriterData pasteboardWriterData;
1063
1064         String textContentWithSimplifiedWhiteSpace = hitTestResult.textContent().simplifyWhiteSpace();
1065
1066         if (hasData == HasNonDefaultPasteboardData::No) {
1067             // Simplify whitespace so the title put on the dataTransfer resembles what the user sees
1068             // on the web page. This includes replacing newlines with spaces.
1069             if (mustUseLegacyDragClient)
1070                 src.editor().copyURL(linkURL, textContentWithSimplifiedWhiteSpace, dataTransfer.pasteboard());
1071             else
1072                 pasteboardWriterData.setURLData(src.editor().pasteboardWriterURL(linkURL, textContentWithSimplifiedWhiteSpace));
1073         } else {
1074             // Make sure the pasteboard also contains trustworthy link data
1075             // but don't overwrite more general pasteboard types.
1076             PasteboardURL pasteboardURL;
1077             pasteboardURL.url = linkURL;
1078             pasteboardURL.title = hitTestResult.textContent();
1079             dataTransfer.pasteboard().writeTrustworthyWebURLsPboardType(pasteboardURL);
1080         }
1081
1082         const VisibleSelection& sourceSelection = src.selection().selection();
1083         if (sourceSelection.isCaret() && sourceSelection.isContentEditable()) {
1084             // a user can initiate a drag on a link without having any text
1085             // selected.  In this case, we should expand the selection to
1086             // the enclosing anchor element
1087             Position pos = sourceSelection.base();
1088             Node* node = enclosingAnchorElement(pos);
1089             if (node)
1090                 src.selection().setSelection(VisibleSelection::selectionFromContentsOfNode(node));
1091         }
1092
1093         m_client.willPerformDragSourceAction(DragSourceActionLink, dragOrigin, dataTransfer);
1094         if (!dragImage) {
1095             TextIndicatorData textIndicator;
1096             dragImage = DragImage { createDragImageForLink(element, linkURL, textContentWithSimplifiedWhiteSpace, textIndicator, src.settings().fontRenderingMode(), m_page.deviceScaleFactor()) };
1097             if (dragImage) {
1098                 m_dragOffset = dragOffsetForLinkDragImage(dragImage.get());
1099                 dragLoc = IntPoint(dragOrigin.x() + m_dragOffset.x(), dragOrigin.y() + m_dragOffset.y());
1100                 dragImage = DragImage { platformAdjustDragImageForDeviceScaleFactor(dragImage.get(), m_page.deviceScaleFactor()) };
1101                 if (textIndicator.contentImage)
1102                     dragImage.setIndicatorData(textIndicator);
1103             }
1104         }
1105
1106         if (mustUseLegacyDragClient) {
1107             doSystemDrag(WTFMove(dragImage), dragLoc, dragOrigin, src, state, { });
1108             return true;
1109         }
1110
1111         DragItem dragItem;
1112         dragItem.imageAnchorPoint = dragImage ? anchorPointForLinkDragImage(dragImage.get()) : FloatPoint();
1113         dragItem.image = WTFMove(dragImage);
1114         dragItem.data = WTFMove(pasteboardWriterData);
1115
1116         beginDrag(WTFMove(dragItem), src, dragOrigin, mouseDraggedPoint, dataTransfer, DragSourceActionSelection);
1117
1118         return true;
1119     }
1120
1121 #if ENABLE(ATTACHMENT_ELEMENT)
1122     if (is<HTMLAttachmentElement>(element) && m_dragSourceAction & DragSourceActionAttachment) {
1123         auto& attachment = downcast<HTMLAttachmentElement>(element);
1124         auto* attachmentRenderer = attachment.renderer();
1125
1126         src.editor().setIgnoreSelectionChanges(true);
1127         auto previousSelection = src.selection().selection();
1128         selectElement(element);
1129
1130         PromisedAttachmentInfo promisedAttachment;
1131         if (hasData == HasNonDefaultPasteboardData::No) {
1132             promisedAttachment = promisedAttachmentInfo(src, attachment);
1133             auto& editor = src.editor();
1134             if (!promisedAttachment && editor.client()) {
1135 #if PLATFORM(COCOA)
1136                 // Otherwise, if no file URL is specified, call out to the injected bundle to populate the pasteboard with data.
1137                 editor.willWriteSelectionToPasteboard(src.selection().toNormalizedRange().get());
1138                 editor.writeSelectionToPasteboard(dataTransfer.pasteboard());
1139                 editor.didWriteSelectionToPasteboard();
1140 #endif
1141             }
1142         }
1143         
1144         m_client.willPerformDragSourceAction(DragSourceActionAttachment, dragOrigin, dataTransfer);
1145         
1146         if (!dragImage) {
1147             TextIndicatorData textIndicator;
1148             if (attachmentRenderer)
1149                 attachmentRenderer->setShouldDrawBorder(false);
1150             dragImage = DragImage { dissolveDragImageToFraction(createDragImageForSelection(src, textIndicator), DragImageAlpha) };
1151             if (attachmentRenderer)
1152                 attachmentRenderer->setShouldDrawBorder(true);
1153             if (textIndicator.contentImage)
1154                 dragImage.setIndicatorData(textIndicator);
1155             dragLoc = dragLocForSelectionDrag(src);
1156             m_dragOffset = IntPoint(dragOrigin.x() - dragLoc.x(), dragOrigin.y() - dragLoc.y());
1157         }
1158         doSystemDrag(WTFMove(dragImage), dragLoc, dragOrigin, src, state, WTFMove(promisedAttachment));
1159         if (!element.isContentRichlyEditable())
1160             src.selection().setSelection(previousSelection);
1161         src.editor().setIgnoreSelectionChanges(false);
1162         return true;
1163     }
1164 #endif
1165
1166 #if ENABLE(INPUT_TYPE_COLOR)
1167     bool isColorControl = is<HTMLInputElement>(state.source) && downcast<HTMLInputElement>(*state.source).isColorControl();
1168     if (isColorControl && m_dragSourceAction & DragSourceActionColor) {
1169         auto& input = downcast<HTMLInputElement>(*state.source);
1170         auto color = input.valueAsColor();
1171
1172         Path visiblePath;
1173         dragImage = DragImage { createDragImageForColor(color, input.boundsInRootViewSpace(), input.document().page()->pageScaleFactor(), visiblePath) };
1174         dragImage.setVisiblePath(visiblePath);
1175         dataTransfer.pasteboard().write(color);
1176         dragImageOffset = IntPoint { dragImageSize(dragImage.get()) };
1177         dragLoc = dragLocForDHTMLDrag(mouseDraggedPoint, dragOrigin, dragImageOffset, false);
1178
1179         m_client.willPerformDragSourceAction(DragSourceActionColor, dragOrigin, dataTransfer);
1180         doSystemDrag(WTFMove(dragImage), dragLoc, dragOrigin, src, state, { });
1181         return true;
1182     }
1183 #endif
1184
1185     if (state.type == DragSourceActionDHTML && dragImage) {
1186         ASSERT(m_dragSourceAction & DragSourceActionDHTML);
1187         m_client.willPerformDragSourceAction(DragSourceActionDHTML, dragOrigin, dataTransfer);
1188         doSystemDrag(WTFMove(dragImage), dragLoc, dragOrigin, src, state, { });
1189         return true;
1190     }
1191
1192     return false;
1193 }
1194
1195 void DragController::doImageDrag(Element& element, const IntPoint& dragOrigin, const IntRect& layoutRect, Frame& frame, IntPoint& dragImageOffset, const DragState& state, PromisedAttachmentInfo&& attachmentInfo)
1196 {
1197     IntPoint mouseDownPoint = dragOrigin;
1198     DragImage dragImage;
1199     IntPoint scaledOrigin;
1200
1201     if (!element.renderer())
1202         return;
1203
1204     ImageOrientationDescription orientationDescription(element.renderer()->shouldRespectImageOrientation(), element.renderer()->style().imageOrientation());
1205
1206     Image* image = getImage(element);
1207     if (image && !layoutRect.isEmpty() && shouldUseCachedImageForDragImage(*image) && (dragImage = DragImage { createDragImageFromImage(image, element.renderer() ? orientationDescription : ImageOrientationDescription()) })) {
1208         dragImage = DragImage { fitDragImageToMaxSize(dragImage.get(), layoutRect.size(), maxDragImageSize()) };
1209         IntSize fittedSize = dragImageSize(dragImage.get());
1210
1211         dragImage = DragImage { platformAdjustDragImageForDeviceScaleFactor(dragImage.get(), m_page.deviceScaleFactor()) };
1212         dragImage = DragImage { dissolveDragImageToFraction(dragImage.get(), DragImageAlpha) };
1213
1214         // Properly orient the drag image and orient it differently if it's smaller than the original.
1215         float scale = fittedSize.width() / (float)layoutRect.width();
1216         float dx = scale * (layoutRect.x() - mouseDownPoint.x());
1217         float originY = layoutRect.y();
1218 #if PLATFORM(COCOA)
1219         // Compensate for accursed flipped coordinates in Cocoa.
1220         originY += layoutRect.height();
1221 #endif
1222         float dy = scale * (originY - mouseDownPoint.y());
1223         scaledOrigin = IntPoint((int)(dx + 0.5), (int)(dy + 0.5));
1224     } else {
1225         if (CachedImage* cachedImage = getCachedImage(element)) {
1226             dragImage = DragImage { createDragImageIconForCachedImageFilename(cachedImage->response().suggestedFilename()) };
1227             if (dragImage) {
1228                 dragImage = DragImage { platformAdjustDragImageForDeviceScaleFactor(dragImage.get(), m_page.deviceScaleFactor()) };
1229                 scaledOrigin = IntPoint(DragIconRightInset - dragImageSize(dragImage.get()).width(), DragIconBottomInset);
1230             }
1231         }
1232     }
1233
1234     if (!dragImage)
1235         return;
1236
1237     dragImageOffset = mouseDownPoint + scaledOrigin;
1238     doSystemDrag(WTFMove(dragImage), dragImageOffset, dragOrigin, frame, state, WTFMove(attachmentInfo));
1239 }
1240
1241 void DragController::beginDrag(DragItem dragItem, Frame& frame, const IntPoint& mouseDownPoint, const IntPoint& mouseDraggedPoint, DataTransfer& dataTransfer, DragSourceAction dragSourceAction)
1242 {
1243     ASSERT(!m_client.useLegacyDragClient());
1244
1245     m_didInitiateDrag = true;
1246     m_dragInitiator = frame.document();
1247
1248     // Protect this frame and view, as a load may occur mid drag and attempt to unload this frame
1249     Ref<Frame> mainFrameProtector(m_page.mainFrame());
1250     RefPtr<FrameView> viewProtector = mainFrameProtector->view();
1251
1252     auto mouseDownPointInRootViewCoordinates = viewProtector->rootViewToContents(frame.view()->contentsToRootView(mouseDownPoint));
1253     auto mouseDraggedPointInRootViewCoordinates = viewProtector->rootViewToContents(frame.view()->contentsToRootView(mouseDraggedPoint));
1254
1255     m_client.beginDrag(WTFMove(dragItem), frame, mouseDownPointInRootViewCoordinates, mouseDraggedPointInRootViewCoordinates, dataTransfer, dragSourceAction);
1256 }
1257
1258 void DragController::doSystemDrag(DragImage image, const IntPoint& dragLoc, const IntPoint& eventPos, Frame& frame, const DragState& state, PromisedAttachmentInfo&& promisedAttachmentInfo)
1259 {
1260     m_didInitiateDrag = true;
1261     m_dragInitiator = frame.document();
1262     // Protect this frame and view, as a load may occur mid drag and attempt to unload this frame
1263     Ref<Frame> frameProtector(m_page.mainFrame());
1264     RefPtr<FrameView> viewProtector = frameProtector->view();
1265
1266     DragItem item;
1267     item.image = WTFMove(image);
1268     item.sourceAction = state.type;
1269     item.promisedAttachmentInfo = WTFMove(promisedAttachmentInfo);
1270
1271     auto eventPositionInRootViewCoordinates = frame.view()->contentsToRootView(eventPos);
1272     auto dragLocationInRootViewCoordinates = frame.view()->contentsToRootView(dragLoc);
1273     item.eventPositionInContentCoordinates = viewProtector->rootViewToContents(eventPositionInRootViewCoordinates);
1274     item.dragLocationInContentCoordinates = viewProtector->rootViewToContents(dragLocationInRootViewCoordinates);
1275     item.dragLocationInWindowCoordinates = viewProtector->contentsToWindow(item.dragLocationInContentCoordinates);
1276     if (auto element = state.source) {
1277         auto dataTransferImageElement = state.dataTransfer->dragImageElement();
1278         if (state.type == DragSourceActionDHTML) {
1279             // If the drag image has been customized, fall back to positioning the preview relative to the drag event location.
1280             IntSize dragPreviewSize;
1281             if (dataTransferImageElement)
1282                 dragPreviewSize = dataTransferImageElement->boundsInRootViewSpace().size();
1283             else {
1284                 dragPreviewSize = dragImageSize(item.image.get());
1285                 if (auto* page = frame.page())
1286                     dragPreviewSize.scale(1 / page->deviceScaleFactor());
1287             }
1288             item.dragPreviewFrameInRootViewCoordinates = { dragLocationInRootViewCoordinates, WTFMove(dragPreviewSize) };
1289         } else {
1290             // We can position the preview using the bounds of the drag source element.
1291             item.dragPreviewFrameInRootViewCoordinates = element->boundsInRootViewSpace();
1292         }
1293
1294         RefPtr<Element> link;
1295         if (element->isLink())
1296             link = element;
1297         else {
1298             for (auto& currentElement : elementLineage(element.get())) {
1299                 if (currentElement.isLink()) {
1300                     link = &currentElement;
1301                     break;
1302                 }
1303             }
1304         }
1305         if (link) {
1306             auto titleAttribute = link->attributeWithoutSynchronization(HTMLNames::titleAttr);
1307             item.title = titleAttribute.isEmpty() ? link->innerText() : titleAttribute.string();
1308             item.url = frame.document()->completeURL(stripLeadingAndTrailingHTMLSpaces(link->getAttribute(HTMLNames::hrefAttr)));
1309         }
1310     }
1311     m_client.startDrag(WTFMove(item), *state.dataTransfer, frameProtector.get());
1312     // DragClient::startDrag can cause our Page to dispear, deallocating |this|.
1313     if (!frameProtector->page())
1314         return;
1315
1316     cleanupAfterSystemDrag();
1317 }
1318
1319 // Manual drag caret manipulation
1320 void DragController::placeDragCaret(const IntPoint& windowPoint)
1321 {
1322     mouseMovedIntoDocument(m_page.mainFrame().documentAtPoint(windowPoint));
1323     if (!m_documentUnderMouse)
1324         return;
1325     Frame* frame = m_documentUnderMouse->frame();
1326     FrameView* frameView = frame->view();
1327     if (!frameView)
1328         return;
1329     IntPoint framePoint = frameView->windowToContents(windowPoint);
1330
1331     m_page.dragCaretController().setCaretPosition(frame->visiblePositionForPoint(framePoint));
1332 }
1333
1334 bool DragController::shouldUseCachedImageForDragImage(const Image& image) const
1335 {
1336 #if ENABLE(DATA_INTERACTION)
1337     UNUSED_PARAM(image);
1338     return true;
1339 #else
1340     return image.size().height() * image.size().width() <= MaxOriginalImageArea;
1341 #endif
1342 }
1343
1344 #if !PLATFORM(COCOA)
1345
1346 String DragController::platformContentTypeForBlobType(const String& type) const
1347 {
1348     return type;
1349 }
1350
1351 #endif
1352
1353 #if ENABLE(ATTACHMENT_ELEMENT)
1354
1355 PromisedAttachmentInfo DragController::promisedAttachmentInfo(Frame& frame, Element& element)
1356 {
1357     auto* client = frame.editor().client();
1358     if (!client || !client->supportsClientSideAttachmentData())
1359         return { };
1360
1361     RefPtr<HTMLAttachmentElement> attachment;
1362     if (is<HTMLAttachmentElement>(element))
1363         attachment = &downcast<HTMLAttachmentElement>(element);
1364     else if (is<HTMLImageElement>(element))
1365         attachment = downcast<HTMLImageElement>(element).attachmentElement();
1366
1367     if (!attachment)
1368         return { };
1369
1370     Vector<String> additionalTypes;
1371     Vector<RefPtr<SharedBuffer>> additionalData;
1372 #if PLATFORM(COCOA)
1373     frame.editor().getPasteboardTypesAndDataForAttachment(element, additionalTypes, additionalData);
1374 #endif
1375
1376     if (auto* file = attachment->file())
1377         return { file->url(), platformContentTypeForBlobType(file->type()), file->name(), { }, WTFMove(additionalTypes), WTFMove(additionalData) };
1378
1379     return { { }, { }, { }, attachment->uniqueIdentifier(), WTFMove(additionalTypes), WTFMove(additionalData) };
1380 }
1381
1382 #endif // ENABLE(ATTACHMENT_ELEMENT)
1383
1384 #endif // ENABLE(DRAG_SUPPORT)
1385
1386 } // namespace WebCore