[chromium] Merge ChromiumDataObject and DataTransferItemListChromium.
[WebKit-https.git] / Source / WebCore / platform / chromium / ChromiumDataObject.cpp
1 /*
2  * Copyright (c) 2008, 2009, Google 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 are
6  * met:
7  * 
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  * 
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32 #include "ChromiumDataObject.h"
33
34 #include "ClipboardMimeTypes.h"
35 #include "ClipboardUtilitiesChromium.h"
36 #include "DataTransferItem.h"
37 #include "ExceptionCode.h"
38 #include "ExceptionCodePlaceholder.h"
39 #include "PlatformSupport.h"
40
41 namespace WebCore {
42
43 PassRefPtr<ChromiumDataObject> ChromiumDataObject::createFromPasteboard()
44 {
45     RefPtr<ChromiumDataObject> dataObject = create();
46     uint64_t sequenceNumber = PlatformSupport::clipboardSequenceNumber(currentPasteboardBuffer());
47     bool ignored;
48     HashSet<String> types = PlatformSupport::clipboardReadAvailableTypes(currentPasteboardBuffer(), &ignored);
49     for (HashSet<String>::const_iterator it = types.begin(); it != types.end(); ++it)
50         dataObject->m_itemList.append(ChromiumDataObjectItem::createFromPasteboard(*it, sequenceNumber));
51     return dataObject.release();
52 }
53
54 PassRefPtr<ChromiumDataObject> ChromiumDataObject::create()
55 {
56     return adoptRef(new ChromiumDataObject());
57 }
58
59 PassRefPtr<ChromiumDataObject> ChromiumDataObject::copy() const
60 {
61     return adoptRef(new ChromiumDataObject(*this));
62 }
63
64 size_t ChromiumDataObject::length() const
65 {
66     return m_itemList.size();
67 }
68
69 PassRefPtr<ChromiumDataObjectItem> ChromiumDataObject::item(unsigned long index)
70 {
71     if (index >= length())
72         return 0;
73     return m_itemList[index];
74 }
75
76 void ChromiumDataObject::deleteItem(unsigned long index)
77 {
78     if (index >= length())
79         return;
80     m_itemList.remove(index);
81 }
82
83 void ChromiumDataObject::clearAll()
84 {
85     m_itemList.clear();
86 }
87
88 void ChromiumDataObject::add(const String& data, const String& type, ExceptionCode& ec)
89 {
90     if (!internalAddStringItem(ChromiumDataObjectItem::createFromString(type, data)))
91         ec = NOT_SUPPORTED_ERR;
92 }
93
94 void ChromiumDataObject::add(PassRefPtr<File> file, ScriptExecutionContext* context)
95 {
96     if (!file)
97         return;
98
99     m_itemList.append(ChromiumDataObjectItem::createFromFile(file));
100 }
101
102 void ChromiumDataObject::clearData(const String& type)
103 {
104     for (size_t i = 0; i < m_itemList.size(); ++i) {
105         if (m_itemList[i]->kind() == DataTransferItem::kindString && m_itemList[i]->type() == type) {
106             // Per the spec, type must be unique among all items of kind 'string'.
107             m_itemList.remove(i);
108             return;
109         }
110     }
111 }
112
113 void ChromiumDataObject::clearAllExceptFiles()
114 {
115     for (size_t i = 0; i < m_itemList.size(); ) {
116         if (m_itemList[i]->kind() != DataTransferItem::kindFile) {
117             m_itemList.remove(i);
118             continue;
119         }
120         ++i;
121     }
122 }
123
124 HashSet<String> ChromiumDataObject::types() const
125 {
126     HashSet<String> results;
127     bool containsFiles = false;
128     for (size_t i = 0; i < m_itemList.size(); ++i) {
129         if (m_itemList[i]->kind() == DataTransferItem::kindString)
130             results.add(m_itemList[i]->type());
131         else if (m_itemList[i]->kind() == DataTransferItem::kindFile)
132             containsFiles = true;
133         else
134             ASSERT_NOT_REACHED();
135     }
136     if (containsFiles)
137         results.add(mimeTypeFiles);
138     return results;
139 }
140
141 String ChromiumDataObject::getData(const String& type) const
142 {
143     for (size_t i = 0; i < m_itemList.size(); ++i)  {
144         if (m_itemList[i]->kind() == DataTransferItem::kindString && m_itemList[i]->type() == type)
145             return m_itemList[i]->internalGetAsString();
146     }
147     return String();
148 }
149
150 bool ChromiumDataObject::setData(const String& type, const String& data)
151 {
152     clearData(type);
153     add(data, type, ASSERT_NO_EXCEPTION);
154     return true;
155 }
156
157 void ChromiumDataObject::urlAndTitle(String& url, String* title) const
158 {
159     RefPtr<ChromiumDataObjectItem> item = findStringItem(mimeTypeTextURIList);
160     if (!item)
161         return;
162     url = convertURIListToURL(item->internalGetAsString());
163     if (title)
164         *title = item->title();
165 }
166
167 void ChromiumDataObject::setURLAndTitle(const String& url, const String& title)
168 {
169     clearData(mimeTypeTextURIList);
170     internalAddStringItem(ChromiumDataObjectItem::createFromURL(url, title));
171 }
172
173 void ChromiumDataObject::htmlAndBaseURL(String& html, KURL& baseURL) const
174 {
175     RefPtr<ChromiumDataObjectItem> item = findStringItem(mimeTypeTextHTML);
176     if (!item)
177         return;
178     html = item->internalGetAsString();
179     baseURL = item->baseURL();
180 }
181
182 void ChromiumDataObject::setHTMLAndBaseURL(const String& html, const KURL& baseURL)
183 {
184     clearData(mimeTypeTextHTML);
185     internalAddStringItem(ChromiumDataObjectItem::createFromHTML(html, baseURL));
186 }
187
188 bool ChromiumDataObject::containsFilenames() const
189 {
190     for (size_t i = 0; i < m_itemList.size(); ++i)
191         if (m_itemList[i]->isFilename())
192             return true;
193     return false;
194 }
195
196 Vector<String> ChromiumDataObject::filenames() const
197 {
198     Vector<String> results;
199     for (size_t i = 0; i < m_itemList.size(); ++i)
200         if (m_itemList[i]->isFilename())
201             results.append(static_cast<File*>(m_itemList[i]->getAsFile().get())->path());
202     return results;
203 }
204
205 void ChromiumDataObject::addFilename(const String& filename)
206 {
207     internalAddFileItem(ChromiumDataObjectItem::createFromFile(File::create(filename)));
208 }
209
210 void ChromiumDataObject::addSharedBuffer(const String& name, PassRefPtr<SharedBuffer> buffer)
211 {
212     internalAddFileItem(ChromiumDataObjectItem::createFromSharedBuffer(name, buffer));
213 }
214
215 ChromiumDataObject::ChromiumDataObject()
216 {
217 }
218
219 ChromiumDataObject::ChromiumDataObject(const ChromiumDataObject& other)
220     : RefCounted<ChromiumDataObject>()
221     , m_itemList(other.m_itemList)
222 {
223 }
224
225 PassRefPtr<ChromiumDataObjectItem> ChromiumDataObject::findStringItem(const String& type) const
226 {
227     for (size_t i = 0; i < m_itemList.size(); ++i) {
228         if (m_itemList[i]->kind() == DataTransferItem::kindString && m_itemList[i]->type() == type)
229             return m_itemList[i];
230     }
231     return 0;
232 }
233
234 bool ChromiumDataObject::internalAddStringItem(PassRefPtr<ChromiumDataObjectItem> item)
235 {
236     ASSERT(item->kind() == DataTransferItem::kindString);
237     for (size_t i = 0; i < m_itemList.size(); ++i)
238         if (m_itemList[i]->kind() == DataTransferItem::kindString && m_itemList[i]->type() == item->type())
239             return false;
240
241     m_itemList.append(item);
242     return true;
243 }
244
245 void ChromiumDataObject::internalAddFileItem(PassRefPtr<ChromiumDataObjectItem> item)
246 {
247     ASSERT(item->kind() == DataTransferItem::kindFile);
248     m_itemList.append(item);
249 }
250
251 } // namespace WebCore