Source/WebCore/platform/chromium/ChromiumDataObject.cpp

   1 /*
   2  * Copyright (c) 2008, 2009, 2012 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, const String& displayName)
 206 {
 207     internalAddFileItem(ChromiumDataObjectItem::createFromFile(File::createWithName(filename, displayName)));
 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     : m_modifierKeyState(0)
 217 {
 218 }
 219
 220 ChromiumDataObject::ChromiumDataObject(const ChromiumDataObject& other)
 221     : RefCounted<ChromiumDataObject>()
 222     , m_itemList(other.m_itemList)
 223     , m_modifierKeyState(0)
 224 {
 225 }
 226
 227 PassRefPtr<ChromiumDataObjectItem> ChromiumDataObject::findStringItem(const String& type) const
 228 {
 229     for (size_t i = 0; i < m_itemList.size(); ++i) {
 230         if (m_itemList[i]->kind() == DataTransferItem::kindString && m_itemList[i]->type() == type)
 231             return m_itemList[i];
 232     }
 233     return 0;
 234 }
 235
 236 bool ChromiumDataObject::internalAddStringItem(PassRefPtr<ChromiumDataObjectItem> item)
 237 {
 238     ASSERT(item->kind() == DataTransferItem::kindString);
 239     for (size_t i = 0; i < m_itemList.size(); ++i)
 240         if (m_itemList[i]->kind() == DataTransferItem::kindString && m_itemList[i]->type() == item->type())
 241             return false;
 242
 243     m_itemList.append(item);
 244     return true;
 245 }
 246
 247 void ChromiumDataObject::internalAddFileItem(PassRefPtr<ChromiumDataObjectItem> item)
 248 {
 249     ASSERT(item->kind() == DataTransferItem::kindFile);
 250     m_itemList.append(item);
 251 }
 252
 253 } // namespace WebCore