Implement ui::ClipboardMus.

ui/views/mus/clipboard_mus.cc

+// Copyright 2016 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 "ui/views/mus/clipboard_mus.h"
+
+#include "base/logging.h"
+#include "base/stl_util.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/threading/thread_restrictions.h"
+#include "mojo/common/common_type_converters.h"
+#include "services/shell/public/cpp/connector.h"
+#include "third_party/skia/include/core/SkBitmap.h"
+#include "ui/base/clipboard/custom_data_helper.h"
+#include "ui/gfx/codec/png_codec.h"
+
+namespace views {
+namespace {
+
+mus::mojom::Clipboard::Type GetType(ui::ClipboardType type) {
+  switch (type) {
+    case ui::CLIPBOARD_TYPE_COPY_PASTE:
+      return mus::mojom::Clipboard::Type::COPY_PASTE;
+    case ui::CLIPBOARD_TYPE_SELECTION:
+      return mus::mojom::Clipboard::Type::SELECTION;
+    case ui::CLIPBOARD_TYPE_DRAG:
+      return mus::mojom::Clipboard::Type::DRAG;
+  }
+
+  NOTREACHED();
+  return mus::mojom::Clipboard::Type::COPY_PASTE;
+}
+
+// The source URL of copied HTML.
+const char kInternalSourceURL[] = "chromium/internal-url";
+
+}  // namespace
+
+ClipboardMus::ClipboardMus() {}
+
+ClipboardMus::~ClipboardMus() {}
+
+void ClipboardMus::Init(shell::Connector* connector) {
+  connector->ConnectToInterface("mojo:mus", &clipboard_);
+}
+
+// TODO(erg): This isn't optimal. It would be better to move the entire
+// FormatType system to mime types throughout chrome, but that's a very large
+// change.
+mojo::String ClipboardMus::GetMimeTypeFor(const FormatType& format) {
+  if (format.Equals(GetUrlFormatType()) || format.Equals(GetUrlWFormatType()))
+    return mus::mojom::kMimeTypeURIList;
+  if (format.Equals(GetMozUrlFormatType()))
+    return mus::mojom::kMimeTypeMozillaURL;
+  if (format.Equals(GetPlainTextFormatType()) ||
+      format.Equals(GetPlainTextWFormatType())) {
+    return mus::mojom::kMimeTypeText;
+  }
+  if (format.Equals(GetHtmlFormatType()))
+    return mus::mojom::kMimeTypeHTML;
+  if (format.Equals(GetRtfFormatType()))
+    return mus::mojom::kMimeTypeRTF;
+  if (format.Equals(GetBitmapFormatType()))
+    return mus::mojom::kMimeTypePNG;
+  if (format.Equals(GetWebKitSmartPasteFormatType()))
+    return kMimeTypeWebkitSmartPaste;
+  if (format.Equals(GetWebCustomDataFormatType()))
+    return kMimeTypeWebCustomData;
+  if (format.Equals(GetPepperCustomDataFormatType()))
+    return kMimeTypePepperCustomData;
+
+  // TODO(erg): This isn't optimal, but it's the best we can do. On windows,
+  // this will return strings that aren't MIME types, though they'll be
+  // unique and should be serializable on the other side of the mojo
+  // connection.
+  return format.Serialize();
+}
+
+bool ClipboardMus::HasMimeType(const mojo::Array<mojo::String>& available_types,
+                               const std::string& type) const {
+  return ContainsValue(available_types, type);
+}
+
+uint64_t ClipboardMus::GetSequenceNumber(ui::ClipboardType type) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  uint64_t sequence_number = 0;
+  clipboard_->GetSequenceNumber(GetType(type), &sequence_number);
+  return sequence_number;
+}
+
+bool ClipboardMus::IsFormatAvailable(const FormatType& format,
+                                     ui::ClipboardType type) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+
+  uint64_t sequence_number = 0;
+  mojo::Array<mojo::String> available_types;
+  clipboard_->GetAvailableMimeTypes(GetType(type), &sequence_number,
+                                    &available_types);
+
+  mojo::String format_in_mime = GetMimeTypeFor(format);
+  return ContainsValue(available_types, format_in_mime);
+}
+
+void ClipboardMus::Clear(ui::ClipboardType type) {
+  // Sends the data to mus server.
+  uint64_t sequence_number = 0;
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  clipboard_->WriteClipboardData(GetType(type), nullptr,
+                                 &sequence_number);
+}
+
+void ClipboardMus::ReadAvailableTypes(ui::ClipboardType type,
+                                      std::vector<base::string16>* types,
+                                      bool* contains_filenames) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+
+  uint64_t sequence_number = 0;
+  mojo::Array<mojo::String> available_types;
+  clipboard_->GetAvailableMimeTypes(GetType(type), &sequence_number,
+                                    &available_types);
+
+  types->clear();
+  if (HasMimeType(available_types, mus::mojom::kMimeTypeText))
+    types->push_back(base::UTF8ToUTF16(mus::mojom::kMimeTypeText));
+  if (HasMimeType(available_types, mus::mojom::kMimeTypeHTML))
+    types->push_back(base::UTF8ToUTF16(mus::mojom::kMimeTypeHTML));
+  if (HasMimeType(available_types, mus::mojom::kMimeTypeRTF))
+    types->push_back(base::UTF8ToUTF16(mus::mojom::kMimeTypeRTF));
+  if (HasMimeType(available_types, mus::mojom::kMimeTypePNG))
+    types->push_back(base::UTF8ToUTF16(mus::mojom::kMimeTypePNG));
+
+  if (HasMimeType(available_types, kMimeTypeWebCustomData)) {
+    mojo::Array<uint8_t> custom_data;
+    uint64_t sequence_number = 0;
+    if (clipboard_->ReadClipboardData(GetType(type), kMimeTypeWebCustomData,
+                                      &sequence_number, &custom_data)) {
+      ui::ReadCustomDataTypes(&custom_data.front(), custom_data.size(), types);
+    }
+  }
+
+  *contains_filenames = false;
+}
+
+void ClipboardMus::ReadText(ui::ClipboardType type,
+                            base::string16* result) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  mojo::Array<uint8_t> text_data;
+  uint64_t sequence_number = 0;
+  if (clipboard_->ReadClipboardData(GetType(type),
+                                    mojo::String(mus::mojom::kMimeTypeText),
+                                    &sequence_number, &text_data)) {
+    std::string text = text_data.To<std::string>();
+    *result = base::UTF8ToUTF16(text);
+  }
+}
+
+void ClipboardMus::ReadAsciiText(ui::ClipboardType type,
+                                 std::string* result) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  mojo::Array<uint8_t> text_data;
+  uint64_t sequence_number = 0;
+  if (clipboard_->ReadClipboardData(GetType(type),
+                                    mojo::String(mus::mojom::kMimeTypeText),
+                                    &sequence_number, &text_data)) {
+    *result = text_data.To<std::string>();
+  }
+}
+
+void ClipboardMus::ReadHTML(ui::ClipboardType type,
+                            base::string16* markup,
+                            std::string* src_url,
+                            uint32_t* fragment_start,
+                            uint32_t* fragment_end) const {
+  markup->clear();
+  if (src_url)
+    src_url->clear();
+  *fragment_start = 0;
+  *fragment_end = 0;
+
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  mojo::Array<uint8_t> html_data;
+  uint64_t sequence_number = 0;
+  if (clipboard_->ReadClipboardData(GetType(type),
+                                    mojo::String(mus::mojom::kMimeTypeHTML),
+                                    &sequence_number, &html_data)) {
+    *markup = base::UTF8ToUTF16(html_data.To<std::string>());
+    *fragment_end = static_cast<uint32_t>(markup->length());
+
+    // We only bother fetching the source url if we were the ones who wrote
+    // this html data to the clipboard.
+    mojo::Array<uint8_t> url_data;
+    if (clipboard_->ReadClipboardData(GetType(type), kInternalSourceURL,
+                                      &sequence_number, &url_data)) {
+      *src_url = url_data.To<std::string>();
+    }
+  }
+}
+
+void ClipboardMus::ReadRTF(ui::ClipboardType type, std::string* result) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  mojo::Array<uint8_t> rtf_data;
+  uint64_t sequence_number = 0;
+  if (clipboard_->ReadClipboardData(
+          GetType(type), mojo::String(mus::mojom::kMimeTypeRTF),
+          &sequence_number, &rtf_data)) {
+    *result = rtf_data.To<std::string>();
+  }
+}
+
+SkBitmap ClipboardMus::ReadImage(ui::ClipboardType type) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  mojo::Array<uint8_t> data;
+  uint64_t sequence_number = 0;
+  if (clipboard_->ReadClipboardData(
+          GetType(type), mojo::String(mus::mojom::kMimeTypePNG),
+          &sequence_number, &data)) {
+    SkBitmap bitmap;
+    if (gfx::PNGCodec::Decode(&data.front(), data.size(), &bitmap))
+      return SkBitmap(bitmap);
+  }
+
+  return SkBitmap();
+}
+
+void ClipboardMus::ReadCustomData(ui::ClipboardType clipboard_type,
+                                  const base::string16& type,
+                                  base::string16* result) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  mojo::Array<uint8_t> custom_data;
+  uint64_t sequence_number = 0;
+  if (clipboard_->ReadClipboardData(GetType(clipboard_type),
+                                    mojo::String(kMimeTypeWebCustomData),
+                                    &sequence_number, &custom_data)) {
+    ui::ReadCustomDataForType(&custom_data.front(), custom_data.size(), type,
+                              result);
+  }
+}
+
+void ClipboardMus::ReadBookmark(base::string16* title, std::string* url) const {
+  // TODO(erg): This is NOTIMPLEMENTED() on all linux platforms?
+  NOTIMPLEMENTED();
+}
+
+void ClipboardMus::ReadData(const FormatType& format,
+                            std::string* result) const {
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  mojo::Array<uint8_t> data;
+  uint64_t sequence_number = 0;
+  if (clipboard_->ReadClipboardData(mus::mojom::Clipboard::Type::COPY_PASTE,
+                                    GetMimeTypeFor(format),
+                                    &sequence_number, &data)) {
+    *result = data.To<std::string>();
+  }
+}
+
+void ClipboardMus::WriteObjects(ui::ClipboardType type,
+                                const ObjectMap& objects) {
+  current_clipboard_.reset(new mojo::Map<mojo::String, mojo::Array<uint8_t>>);
+  for (const auto& p : objects)
+    DispatchObject(static_cast<ObjectType>(p.first), p.second);
+
+  // Sends the data to mus server.
+  uint64_t sequence_number = 0;
+  base::ThreadRestrictions::ScopedAllowWait allow_wait;
+  clipboard_->WriteClipboardData(GetType(type), std::move(*current_clipboard_),
+                                 &sequence_number);
+  current_clipboard_.reset();
+}
+
+void ClipboardMus::WriteText(const char* text_data, size_t text_len) {
+  DCHECK(current_clipboard_);
+  current_clipboard_->insert(
+      mus::mojom::kMimeTypeText,
+      mojo::Array<uint8_t>::From(base::StringPiece(text_data, text_len)));
+}
+
+void ClipboardMus::WriteHTML(const char* markup_data,
+                             size_t markup_len,
+                             const char* url_data,
+                             size_t url_len) {
+  DCHECK(current_clipboard_);
+  current_clipboard_->insert(
+      mus::mojom::kMimeTypeHTML,
+      mojo::Array<uint8_t>::From(base::StringPiece(markup_data, markup_len)));
+  if (url_len > 0) {
+    current_clipboard_->insert(
+        kInternalSourceURL,
+        mojo::Array<uint8_t>::From(base::StringPiece(url_data, url_len)));
+  }
+}
+
+void ClipboardMus::WriteRTF(const char* rtf_data, size_t data_len) {
+  DCHECK(current_clipboard_);
+  current_clipboard_->insert(
+      mus::mojom::kMimeTypeRTF,
+      mojo::Array<uint8_t>::From(base::StringPiece(rtf_data, data_len)));
+}
+
+void ClipboardMus::WriteBookmark(const char* title_data,
+                                 size_t title_len,
+                                 const char* url_data,
+                                 size_t url_len) {
+  // Writes a Mozilla url (UTF16: URL, newline, title)
+  base::string16 bookmark =
+      base::UTF8ToUTF16(base::StringPiece(url_data, url_len)) +
+      base::ASCIIToUTF16("\n") +
+      base::UTF8ToUTF16(base::StringPiece(title_data, title_len));
+
+  DCHECK(current_clipboard_);
+  current_clipboard_->insert(mus::mojom::kMimeTypeMozillaURL,
+                             mojo::Array<uint8_t>::From(bookmark));
+}
+
+void ClipboardMus::WriteWebSmartPaste() {
+  DCHECK(current_clipboard_);
+  current_clipboard_->insert(kMimeTypeWebkitSmartPaste, mojo::Array<uint8_t>());
+}
+
+void ClipboardMus::WriteBitmap(const SkBitmap& bitmap) {
+  DCHECK(current_clipboard_);
+  // Encode the bitmap as a PNG for transport.
+  std::vector<unsigned char> output;
+  if (gfx::PNGCodec::FastEncodeBGRASkBitmap(bitmap, false, &output)) {
+    current_clipboard_->insert(mus::mojom::kMimeTypePNG,
+                               mojo::Array<uint8_t>::From(output));
+  }
+}
+
+void ClipboardMus::WriteData(const FormatType& format,
+                             const char* data_data,
+                             size_t data_len) {
+  DCHECK(current_clipboard_);
+  current_clipboard_->insert(
+      GetMimeTypeFor(format),
+      mojo::Array<uint8_t>::From(base::StringPiece(data_data, data_len)));
+}
+
+}  // namespace views