Add glue for supporting custom MIME types in DataTransfer.

webkit/tools/test_shell/mock_webclipboard_impl.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "webkit/tools/test_shell/mock_webclipboard_impl.h"
 
+#include <algorithm>
+
 #include "base/logging.h"
 #include "base/stl_util.h"
 #include "base/string_util.h"
 #include "net/base/escape.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebCommon.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebDragData.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebImage.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebURL.h"
 #include "webkit/glue/webclipboard_impl.h"
 #include "webkit/glue/webkit_glue.h"
 #include "webkit/support/webkit_support_gfx.h"
 
 #if WEBKIT_USING_CG
 #include <ApplicationServices/ApplicationServices.h>
 #include <CoreFoundation/CoreFoundation.h>
 #endif
 
+using WebKit::WebDragData;
 using WebKit::WebString;
 using WebKit::WebURL;
 using WebKit::WebVector;
 
 MockWebClipboardImpl::MockWebClipboardImpl() {
 }
 
 MockWebClipboardImpl::~MockWebClipboardImpl() {
 }
 
@@ skipping 34 matching lines @@
   std::vector<WebString> results;
   if (!m_plainText.isEmpty()) {
     results.push_back(WebString("text/plain"));
   }
   if (!m_htmlText.isEmpty()) {
     results.push_back(WebString("text/html"));
   }
   if (!m_image.isNull()) {
     results.push_back(WebString("image/png"));
   }
+  for (size_t i = 0; i < m_customData.size(); ++i) {
+    CHECK(std::find(results.begin(), results.end(), m_customData[i].m_type) ==
+          results.end());
+    results.push_back(m_customData[i].m_type);
+  }
   return results;
 }
 
 WebKit::WebString MockWebClipboardImpl::readPlainText(
     WebKit::WebClipboard::Buffer buffer) {
   return m_plainText;
 }
 
 // TODO(wtc): set output argument *url.
 WebKit::WebString MockWebClipboardImpl::readHTML(
@@ skipping 29 matching lines @@
                                 CGImageGetBytesPerRow(image),
                                 false,
                                 &encoded_image);
   CFRelease(image_data_ref);
 #endif
   data.assign(reinterpret_cast<char*>(vector_as_array(&encoded_image)),
               encoded_image.size());
   return data;
 }
 
+WebKit::WebString MockWebClipboardImpl::readCustomData(
+    WebKit::WebClipboard::Buffer buffer,
+    const WebKit::WebString& type) {
+  for (size_t i = 0; i < m_customData.size(); ++i) {
+    if (m_customData[i].m_type == type) {
+      return m_customData[i].m_data;
+    }
+  }
+  return WebKit::WebString();
+}
+
 void MockWebClipboardImpl::writeHTML(
     const WebKit::WebString& htmlText, const WebKit::WebURL& url,
     const WebKit::WebString& plainText, bool writeSmartPaste) {
   m_htmlText = htmlText;
   m_plainText = plainText;
   m_image.reset();
+  m_customData = WebVector<WebDragData::CustomData>();
   m_writeSmartPaste = writeSmartPaste;
 }
 
 void MockWebClipboardImpl::writePlainText(const WebKit::WebString& plain_text) {
   m_htmlText = WebKit::WebString();
   m_plainText = plain_text;
   m_image.reset();
+  m_customData = WebVector<WebDragData::CustomData>();
   m_writeSmartPaste = false;
 }
 
 void MockWebClipboardImpl::writeURL(
     const WebKit::WebURL& url, const WebKit::WebString& title) {
   m_htmlText = WebString::fromUTF8(
       webkit_glue::WebClipboardImpl::URLToMarkup(url, title));
   m_plainText = url.spec().utf16();
   m_image.reset();
+  m_customData = WebVector<WebDragData::CustomData>();
   m_writeSmartPaste = false;
 }
 
 void MockWebClipboardImpl::writeImage(const WebKit::WebImage& image,
     const WebKit::WebURL& url, const WebKit::WebString& title) {
   if (!image.isNull()) {
     m_htmlText = WebString::fromUTF8(
         webkit_glue::WebClipboardImpl::URLToImageMarkup(url, title));
     m_plainText = m_htmlText;
     m_image = image;
+    m_customData = WebVector<WebDragData::CustomData>();
     m_writeSmartPaste = false;
   }
 }
 
 void MockWebClipboardImpl::writeDataObject(const WebKit::WebDragData& data) {
   m_htmlText = data.htmlText();
   m_plainText = data.plainText();
   m_image.reset();
+  m_customData = data.customData();
   m_writeSmartPaste = false;
 }