479e0caaf739019789840b0f876f6d4f7964b3ee
[WebKit-https.git] / Source / WebCore / platform / win / DragDataWin.cpp
1 /*
2  * Copyright (C) 2007, 2008 Apple Inc. All rights reserved.
3  * Copyright (C) 2012 Baidu Inc. All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
18  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
22  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
24  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
25  */
26
27 #include "config.h"
28 #include "DragData.h"
29
30 #include "COMPtr.h"
31 #include "ClipboardUtilitiesWin.h"
32 #include "TextEncoding.h"
33 #include <objidl.h>
34 #include <shlwapi.h>
35 #include <wininet.h>
36 #include <wtf/Forward.h>
37 #include <wtf/HashMap.h>
38 #include <wtf/RefPtr.h>
39 #include <wtf/text/WTFString.h>
40 #include <wtf/text/win/WCharStringExtras.h>
41
42 namespace WebCore {
43
44 DragData::DragData(const DragDataMap& data, const IntPoint& clientPosition, const IntPoint& globalPosition,
45     DragOperation sourceOperationMask, DragApplicationFlags flags)
46     : m_clientPosition(clientPosition)
47     , m_globalPosition(globalPosition)
48     , m_platformDragData(0)
49     , m_draggingSourceOperationMask(sourceOperationMask)
50     , m_applicationFlags(flags)
51     , m_dragDataMap(data)
52 {
53 }
54
55 bool DragData::containsURL(FilenameConversionPolicy filenamePolicy) const
56 {
57     if (m_platformDragData)
58         return SUCCEEDED(m_platformDragData->QueryGetData(urlWFormat())) 
59             || SUCCEEDED(m_platformDragData->QueryGetData(urlFormat()))
60             || (filenamePolicy == ConvertFilenames
61                 && (SUCCEEDED(m_platformDragData->QueryGetData(filenameWFormat()))
62                     || SUCCEEDED(m_platformDragData->QueryGetData(filenameFormat()))));
63     return m_dragDataMap.contains(urlWFormat()->cfFormat) || m_dragDataMap.contains(urlFormat()->cfFormat)
64         || (filenamePolicy == ConvertFilenames && (m_dragDataMap.contains(filenameWFormat()->cfFormat) || m_dragDataMap.contains(filenameFormat()->cfFormat)));
65 }
66
67 const DragDataMap& DragData::dragDataMap()
68 {
69     if (!m_dragDataMap.isEmpty() || !m_platformDragData)
70         return m_dragDataMap;
71     // Enumerate clipboard content and load it in the map.
72     COMPtr<IEnumFORMATETC> itr;
73
74     if (FAILED(m_platformDragData->EnumFormatEtc(DATADIR_GET, &itr)) || !itr)
75         return m_dragDataMap;
76
77     FORMATETC dataFormat;
78     while (itr->Next(1, &dataFormat, 0) == S_OK) {
79         Vector<String> dataStrings;
80         getClipboardData(m_platformDragData, &dataFormat, dataStrings);
81         if (!dataStrings.isEmpty())
82             m_dragDataMap.set(dataFormat.cfFormat, dataStrings); 
83     }
84     return m_dragDataMap;
85 }
86
87 void DragData::getDragFileDescriptorData(int& size, String& pathname)
88 {
89     size = 0;
90     if (m_platformDragData)
91         getFileDescriptorData(m_platformDragData, size, pathname);
92 }
93
94 void DragData::getDragFileContentData(int size, void* dataBlob)
95 {
96     if (m_platformDragData)
97         getFileContentData(m_platformDragData, size, dataBlob);
98 }
99
100 String DragData::asURL(FilenameConversionPolicy filenamePolicy, String* title) const
101 {
102     return (m_platformDragData) ? getURL(m_platformDragData, filenamePolicy, title) : getURL(&m_dragDataMap, filenamePolicy, title);
103 }
104
105 bool DragData::containsFiles() const
106 {
107 #if USE(CF)
108     return (m_platformDragData) ? SUCCEEDED(m_platformDragData->QueryGetData(cfHDropFormat())) : m_dragDataMap.contains(cfHDropFormat()->cfFormat);
109 #else
110     return false;
111 #endif
112 }
113
114 unsigned DragData::numberOfFiles() const
115 {
116 #if USE(CF)
117     if (!m_platformDragData)
118         return 0;
119
120     STGMEDIUM medium;
121     if (FAILED(m_platformDragData->GetData(cfHDropFormat(), &medium)))
122         return 0;
123
124     HDROP hdrop = static_cast<HDROP>(GlobalLock(medium.hGlobal));
125
126     if (!hdrop)
127         return 0;
128
129     unsigned numFiles = DragQueryFileW(hdrop, 0xFFFFFFFF, 0, 0);
130
131     DragFinish(hdrop);
132     GlobalUnlock(medium.hGlobal);
133
134     return numFiles;
135 #else
136     return 0;
137 #endif
138 }
139
140 Vector<String> DragData::asFilenames() const
141 {
142     Vector<String> result;
143
144 #if USE(CF)
145     if (m_platformDragData) {
146         WCHAR filename[MAX_PATH];
147
148         STGMEDIUM medium;
149         if (FAILED(m_platformDragData->GetData(cfHDropFormat(), &medium)))
150             return result;
151
152         HDROP hdrop = reinterpret_cast<HDROP>(GlobalLock(medium.hGlobal)); 
153
154         if (!hdrop)
155             return result;
156
157         const unsigned numFiles = DragQueryFileW(hdrop, 0xFFFFFFFF, 0, 0);
158         for (unsigned i = 0; i < numFiles; i++) {
159             if (!DragQueryFileW(hdrop, i, filename, WTF_ARRAY_LENGTH(filename)))
160                 continue;
161             result.append(nullTerminatedWCharToString(filename));
162         }
163
164         // Free up memory from drag
165         DragFinish(hdrop);
166
167         GlobalUnlock(medium.hGlobal);
168         return result;
169     }
170     result = m_dragDataMap.get(cfHDropFormat()->cfFormat);
171 #endif
172
173     return result;
174 }
175
176 bool DragData::containsPlainText() const
177 {
178     if (m_platformDragData)
179         return SUCCEEDED(m_platformDragData->QueryGetData(plainTextWFormat()))
180             || SUCCEEDED(m_platformDragData->QueryGetData(plainTextFormat()));
181     return m_dragDataMap.contains(plainTextWFormat()->cfFormat) || m_dragDataMap.contains(plainTextFormat()->cfFormat);
182 }
183
184 String DragData::asPlainText() const
185 {
186     return (m_platformDragData) ? getPlainText(m_platformDragData) : getPlainText(&m_dragDataMap);
187 }
188
189 bool DragData::containsColor() const
190 {
191     return false;
192 }
193
194 bool DragData::canSmartReplace() const
195 {
196     if (m_platformDragData)
197         return SUCCEEDED(m_platformDragData->QueryGetData(smartPasteFormat()));
198     return m_dragDataMap.contains(smartPasteFormat()->cfFormat);
199 }
200
201 bool DragData::containsCompatibleContent(DraggingPurpose) const
202 {
203     return containsPlainText() || containsURL()
204         || ((m_platformDragData) ? (containsHTML(m_platformDragData) || containsFilenames(m_platformDragData))
205             : (containsHTML(&m_dragDataMap) || containsFilenames(&m_dragDataMap)))
206         || containsColor();
207 }
208
209 Color DragData::asColor() const
210 {
211     return Color();
212 }
213
214 }