bced32bbe846d87949540f3c58ae19683c5eb014
[WebKit-https.git] / Source / WebCore / platform / chromium / DataTransferItemChromium.cpp
1 /*
2  * Copyright (C) 2011 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 "DataTransferItemChromium.h"
33
34 #if ENABLE(DATA_TRANSFER_ITEMS)
35
36 #include "Blob.h"
37 #include "Clipboard.h"
38 #include "ClipboardChromium.h"
39 #include "ClipboardMimeTypes.h"
40 #include "ClipboardUtilitiesChromium.h"
41 #include "DataTransferItem.h"
42 #include "File.h"
43 #include "PlatformSupport.h"
44 #include "SharedBuffer.h"
45 #include "StringCallback.h"
46
47 namespace WebCore {
48
49 PassRefPtr<DataTransferItemChromium> DataTransferItemChromium::createFromString(const String& type, const String& data)
50 {
51     RefPtr<DataTransferItemChromium> item = adoptRef(new DataTransferItemChromium(DataTransferItem::kindString, type));
52     item->m_data = data;
53     return item.release();
54 }
55
56 PassRefPtr<DataTransferItemChromium> DataTransferItemChromium::createFromFile(PassRefPtr<File> file)
57 {
58     RefPtr<DataTransferItemChromium> item = adoptRef(new DataTransferItemChromium(DataTransferItem::kindFile, file->type()));
59     item->m_file = file;
60     return item.release();
61 }
62
63 PassRefPtr<DataTransferItemChromium> DataTransferItemChromium::createFromURL(const String& url, const String& title)
64 {
65     RefPtr<DataTransferItemChromium> item = adoptRef(new DataTransferItemChromium(DataTransferItem::kindString, mimeTypeTextURIList));
66     item->m_data = url;
67     item->m_title = title;
68     return item.release();
69 }
70
71 PassRefPtr<DataTransferItemChromium> DataTransferItemChromium::createFromHTML(const String& html, const KURL& baseURL)
72 {
73     RefPtr<DataTransferItemChromium> item = adoptRef(new DataTransferItemChromium(DataTransferItem::kindString, mimeTypeTextHTML));
74     item->m_data = html;
75     item->m_baseURL = baseURL;
76     return item.release();
77 }
78
79 PassRefPtr<DataTransferItemChromium> DataTransferItemChromium::createFromSharedBuffer(const String& name, PassRefPtr<SharedBuffer> buffer)
80 {
81     RefPtr<DataTransferItemChromium> item = adoptRef(new DataTransferItemChromium(DataTransferItem::kindFile, String()));
82     item->m_sharedBuffer = buffer;
83     item->m_title = name;
84     return item.release();
85 }
86
87 PassRefPtr<DataTransferItemChromium> DataTransferItemChromium::createFromPasteboard(const String& type, uint64_t sequenceNumber)
88 {
89     if (type == mimeTypeImagePng)
90         return adoptRef(new DataTransferItemChromium(DataTransferItem::kindFile, type, sequenceNumber));
91     return adoptRef(new DataTransferItemChromium(DataTransferItem::kindString, type, sequenceNumber));
92 }
93
94 DataTransferItemChromium::DataTransferItemChromium(const String& kind, const String& type)
95     : m_source(InternalSource)
96     , m_kind(kind)
97     , m_type(type)
98     , m_sequenceNumber(0)
99 {
100 }
101
102 DataTransferItemChromium::DataTransferItemChromium(const String& kind, const String& type, uint64_t sequenceNumber)
103     : m_source(PasteboardSource)
104     , m_kind(kind)
105     , m_type(type)
106     , m_sequenceNumber(sequenceNumber)
107 {
108 }
109
110 void DataTransferItemChromium::getAsString(PassRefPtr<StringCallback> callback, ScriptExecutionContext* context) const
111 {
112     if (!callback || kind() != DataTransferItem::kindString)
113         return;
114
115     callback->scheduleCallback(context, internalGetAsString());
116 }
117
118 PassRefPtr<Blob> DataTransferItemChromium::getAsFile() const
119 {
120     if (kind() != DataTransferItem::kindFile)
121         return 0;
122
123     if (m_source == InternalSource) {
124         if (m_file)
125             return m_file;
126         ASSERT(m_sharedBuffer);
127         // FIXME: This code is currently impossible--we never populate m_sharedBuffer when dragging
128         // in. At some point though, we may need to support correctly converting a shared buffer
129         // into a file.
130         return 0;
131     }
132
133     ASSERT(m_source == PasteboardSource);
134     if (type() == mimeTypeImagePng) {
135         // FIXME: This is pretty inefficient. We copy the data from the browser
136         // to the renderer. We then place it in a blob in WebKit, which
137         // registers it and copies it *back* to the browser. When a consumer
138         // wants to read the data, we then copy the data back into the renderer.
139         // https://bugs.webkit.org/show_bug.cgi?id=58107 has been filed to track
140         // improvements to this code (in particular, add a registerClipboardBlob
141         // method to the blob registry; that way the data is only copied over
142         // into the renderer when it's actually read, not when the blob is
143         // initially constructed).
144         RefPtr<SharedBuffer> data = PlatformSupport::clipboardReadImage(PasteboardPrivate::StandardBuffer);
145         RefPtr<RawData> rawData = RawData::create();
146         rawData->mutableData()->append(data->data(), data->size());
147         OwnPtr<BlobData> blobData = BlobData::create();
148         blobData->appendData(rawData, 0, -1);
149         blobData->setContentType(mimeTypeImagePng);
150         return Blob::create(blobData.release(), data->size());
151     }
152
153     return 0;
154 }
155
156 String DataTransferItemChromium::internalGetAsString() const
157 {
158     ASSERT(m_kind == DataTransferItem::kindString);
159
160     if (m_source == InternalSource)
161         return m_data;
162
163     ASSERT(m_source == PasteboardSource);
164
165     String data;
166     // This is ugly but there's no real alternative.
167     if (m_type == mimeTypeTextPlain)
168         data = PlatformSupport::clipboardReadPlainText(currentPasteboardBuffer());
169     else if (m_type == mimeTypeTextHTML) {
170         KURL ignoredSourceURL;
171         unsigned ignored;
172         PlatformSupport::clipboardReadHTML(currentPasteboardBuffer(), &data, &ignoredSourceURL, &ignored, &ignored);
173     } else
174         data = PlatformSupport::clipboardReadCustomData(currentPasteboardBuffer(), m_type);
175
176     return PlatformSupport::clipboardSequenceNumber(currentPasteboardBuffer()) == m_sequenceNumber ? data : String();
177 }
178
179 bool DataTransferItemChromium::isFilename() const
180 {
181     // FIXME: https://bugs.webkit.org/show_bug.cgi?id=81261: When we properly support File dragout,
182     // we'll need to make sure this works as expected for DragDataChromium.
183     return m_kind == DataTransferItem::kindFile && m_file;
184 }
185
186 } // namespace WebCore
187
188 #endif // ENABLE(DATA_TRANSFER_ITEMS)