[Win] Remove -DUCHAR_TYPE=wchar_t stopgap and learn to live with char16_t.
[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
41 namespace WebCore {
42
43 DragData::DragData(const DragDataMap& data, const IntPoint& clientPosition, const IntPoint& globalPosition,
44     DragOperation sourceOperationMask, DragApplicationFlags flags)
45     : m_clientPosition(clientPosition)
46     , m_globalPosition(globalPosition)
47     , m_platformDragData(0)
48     , m_draggingSourceOperationMask(sourceOperationMask)
49     , m_applicationFlags(flags)
50     , m_dragDataMap(data)
51 {
52 }
53
54 bool DragData::containsURL(FilenameConversionPolicy filenamePolicy) const
55 {
56     if (m_platformDragData)
57         return SUCCEEDED(m_platformDragData->QueryGetData(urlWFormat())) 
58             || SUCCEEDED(m_platformDragData->QueryGetData(urlFormat()))
59             || (filenamePolicy == ConvertFilenames
60                 && (SUCCEEDED(m_platformDragData->QueryGetData(filenameWFormat()))
61                     || SUCCEEDED(m_platformDragData->QueryGetData(filenameFormat()))));
62     return m_dragDataMap.contains(urlWFormat()->cfFormat) || m_dragDataMap.contains(urlFormat()->cfFormat)
63         || (filenamePolicy == ConvertFilenames && (m_dragDataMap.contains(filenameWFormat()->cfFormat) || m_dragDataMap.contains(filenameFormat()->cfFormat)));
64 }
65
66 const DragDataMap& DragData::dragDataMap()
67 {
68     if (!m_dragDataMap.isEmpty() || !m_platformDragData)
69         return m_dragDataMap;
70     // Enumerate clipboard content and load it in the map.
71     COMPtr<IEnumFORMATETC> itr;
72
73     if (FAILED(m_platformDragData->EnumFormatEtc(DATADIR_GET, &itr)) || !itr)
74         return m_dragDataMap;
75
76     FORMATETC dataFormat;
77     while (itr->Next(1, &dataFormat, 0) == S_OK) {
78         Vector<String> dataStrings;
79         getClipboardData(m_platformDragData, &dataFormat, dataStrings);
80         if (!dataStrings.isEmpty())
81             m_dragDataMap.set(dataFormat.cfFormat, dataStrings); 
82     }
83     return m_dragDataMap;
84 }
85
86 void DragData::getDragFileDescriptorData(int& size, String& pathname)
87 {
88     size = 0;
89     if (m_platformDragData)
90         getFileDescriptorData(m_platformDragData, size, pathname);
91 }
92
93 void DragData::getDragFileContentData(int size, void* dataBlob)
94 {
95     if (m_platformDragData)
96         getFileContentData(m_platformDragData, size, dataBlob);
97 }
98
99 String DragData::asURL(FilenameConversionPolicy filenamePolicy, String* title) const
100 {
101     return (m_platformDragData) ? getURL(m_platformDragData, filenamePolicy, title) : getURL(&m_dragDataMap, filenamePolicy, title);
102 }
103
104 bool DragData::containsFiles() const
105 {
106 #if USE(CF)
107     return (m_platformDragData) ? SUCCEEDED(m_platformDragData->QueryGetData(cfHDropFormat())) : m_dragDataMap.contains(cfHDropFormat()->cfFormat);
108 #else
109     return false;
110 #endif
111 }
112
113 unsigned DragData::numberOfFiles() const
114 {
115 #if USE(CF)
116     if (!m_platformDragData)
117         return 0;
118
119     STGMEDIUM medium;
120     if (FAILED(m_platformDragData->GetData(cfHDropFormat(), &medium)))
121         return 0;
122
123     HDROP hdrop = static_cast<HDROP>(GlobalLock(medium.hGlobal));
124
125     if (!hdrop)
126         return 0;
127
128     unsigned numFiles = DragQueryFileW(hdrop, 0xFFFFFFFF, 0, 0);
129
130     DragFinish(hdrop);
131     GlobalUnlock(medium.hGlobal);
132
133     return numFiles;
134 #else
135     return 0;
136 #endif
137 }
138
139 Vector<String> DragData::asFilenames() const
140 {
141     Vector<String> result;
142
143 #if USE(CF)
144     if (m_platformDragData) {
145         WCHAR filename[MAX_PATH];
146
147         STGMEDIUM medium;
148         if (FAILED(m_platformDragData->GetData(cfHDropFormat(), &medium)))
149             return result;
150
151         HDROP hdrop = reinterpret_cast<HDROP>(GlobalLock(medium.hGlobal)); 
152
153         if (!hdrop)
154             return result;
155
156         const unsigned numFiles = DragQueryFileW(hdrop, 0xFFFFFFFF, 0, 0);
157         for (unsigned i = 0; i < numFiles; i++) {
158             if (!DragQueryFileW(hdrop, i, filename, WTF_ARRAY_LENGTH(filename)))
159                 continue;
160             result.append(filename);
161         }
162
163         // Free up memory from drag
164         DragFinish(hdrop);
165
166         GlobalUnlock(medium.hGlobal);
167         return result;
168     }
169     result = m_dragDataMap.get(cfHDropFormat()->cfFormat);
170 #endif
171
172     return result;
173 }
174
175 bool DragData::containsPlainText() const
176 {
177     if (m_platformDragData)
178         return SUCCEEDED(m_platformDragData->QueryGetData(plainTextWFormat()))
179             || SUCCEEDED(m_platformDragData->QueryGetData(plainTextFormat()));
180     return m_dragDataMap.contains(plainTextWFormat()->cfFormat) || m_dragDataMap.contains(plainTextFormat()->cfFormat);
181 }
182
183 String DragData::asPlainText() const
184 {
185     return (m_platformDragData) ? getPlainText(m_platformDragData) : getPlainText(&m_dragDataMap);
186 }
187
188 bool DragData::containsColor() const
189 {
190     return false;
191 }
192
193 bool DragData::canSmartReplace() const
194 {
195     if (m_platformDragData)
196         return SUCCEEDED(m_platformDragData->QueryGetData(smartPasteFormat()));
197     return m_dragDataMap.contains(smartPasteFormat()->cfFormat);
198 }
199
200 bool DragData::containsCompatibleContent(DraggingPurpose) const
201 {
202     return containsPlainText() || containsURL()
203         || ((m_platformDragData) ? (containsHTML(m_platformDragData) || containsFilenames(m_platformDragData))
204             : (containsHTML(&m_dragDataMap) || containsFilenames(&m_dragDataMap)))
205         || containsColor();
206 }
207
208 Color DragData::asColor() const
209 {
210     return Color();
211 }
212
213 }