File inputs only accept UTI types that can be inserted into contenteditable areas...
[WebKit.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 void DragData::asFilenames(Vector<String>& result) const
140 {
141 #if USE(CF)
142     if (m_platformDragData) {
143         WCHAR filename[MAX_PATH];
144
145         STGMEDIUM medium;
146         if (FAILED(m_platformDragData->GetData(cfHDropFormat(), &medium)))
147             return;
148
149         HDROP hdrop = reinterpret_cast<HDROP>(GlobalLock(medium.hGlobal)); 
150
151         if (!hdrop)
152             return;
153
154         const unsigned numFiles = DragQueryFileW(hdrop, 0xFFFFFFFF, 0, 0);
155         for (unsigned i = 0; i < numFiles; i++) {
156             if (!DragQueryFileW(hdrop, i, filename, WTF_ARRAY_LENGTH(filename)))
157                 continue;
158             result.append(static_cast<UChar*>(filename)); 
159         }
160
161         // Free up memory from drag
162         DragFinish(hdrop);
163
164         GlobalUnlock(medium.hGlobal);
165         return;
166     }
167     result = m_dragDataMap.get(cfHDropFormat()->cfFormat);
168 #endif
169 }
170
171 bool DragData::containsPlainText() const
172 {
173     if (m_platformDragData)
174         return SUCCEEDED(m_platformDragData->QueryGetData(plainTextWFormat()))
175             || SUCCEEDED(m_platformDragData->QueryGetData(plainTextFormat()));
176     return m_dragDataMap.contains(plainTextWFormat()->cfFormat) || m_dragDataMap.contains(plainTextFormat()->cfFormat);
177 }
178
179 String DragData::asPlainText() const
180 {
181     return (m_platformDragData) ? getPlainText(m_platformDragData) : getPlainText(&m_dragDataMap);
182 }
183
184 bool DragData::containsColor() const
185 {
186     return false;
187 }
188
189 bool DragData::canSmartReplace() const
190 {
191     if (m_platformDragData)
192         return SUCCEEDED(m_platformDragData->QueryGetData(smartPasteFormat()));
193     return m_dragDataMap.contains(smartPasteFormat()->cfFormat);
194 }
195
196 bool DragData::containsCompatibleContent(DraggingPurpose) const
197 {
198     return containsPlainText() || containsURL()
199         || ((m_platformDragData) ? (containsHTML(m_platformDragData) || containsFilenames(m_platformDragData))
200             : (containsHTML(&m_dragDataMap) || containsFilenames(&m_dragDataMap)))
201         || containsColor();
202 }
203
204 Color DragData::asColor() const
205 {
206     return Color();
207 }
208
209 }