// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef THIRD_PARTY_BLINK_RENDERER_PLATFORM_MOJO_DRAG_MOJOM_TRAITS_H_
#define THIRD_PARTY_BLINK_RENDERER_PLATFORM_MOJO_DRAG_MOJOM_TRAITS_H_

#include <stdint.h>

#include <optional>
#include <variant>
#include <vector>

#include "mojo/public/cpp/base/big_buffer.h"
#include "mojo/public/cpp/base/file_path_mojom_traits.h"
#include "mojo/public/cpp/bindings/string_traits_wtf.h"
#include "mojo/public/cpp/bindings/struct_traits.h"
#include "mojo/public/cpp/bindings/union_traits.h"
#include "services/network/public/mojom/referrer_policy.mojom-forward.h"
#include "third_party/blink/public/common/page/source_effect_allowed_mojom_util.h"
#include "third_party/blink/public/mojom/data_transfer/data_transfer.mojom-shared.h"
#include "third_party/blink/public/mojom/drag/drag.mojom-shared.h"
#include "third_party/blink/public/mojom/file_system_access/file_system_access_data_transfer_token.mojom-blink.h"
#include "third_party/blink/public/platform/web_drag_data.h"
#include "third_party/blink/renderer/platform/blob/serialized_blob_mojom_traits.h"
#include "third_party/blink/renderer/platform/mojo/kurl_mojom_traits.h"
#include "third_party/blink/renderer/platform/mojo/string16_mojom_traits.h"
#include "third_party/blink/renderer/platform/platform_export.h"
#include "third_party/blink/renderer/platform/weborigin/kurl.h"
#include "third_party/blink/renderer/platform/wtf/forward.h"

namespace mojo {

template <>
struct PLATFORM_EXPORT StructTraits<blink::mojom::DragItemStringDataView,
                                    blink::WebDragData::StringItem> {
  static blink::String string_type(const blink::WebDragData::StringItem& item);
  static blink::String string_data(const blink::WebDragData::StringItem& item);
  static blink::String title(const blink::WebDragData::StringItem& item);
  static std::optional<blink::KURL> base_url(
      const blink::WebDragData::StringItem& item);
  static bool Read(blink::mojom::DragItemStringDataView data,
                   blink::WebDragData::StringItem* out);
};

template <>
struct PLATFORM_EXPORT StructTraits<blink::mojom::DataTransferFileDataView,
                                    blink::WebDragData::FilenameItem> {
  static base::FilePath path(const blink::WebDragData::FilenameItem& item);
  static base::FilePath display_name(
      const blink::WebDragData::FilenameItem& item);
  static mojo::PendingRemote<
      blink::mojom::blink::FileSystemAccessDataTransferToken>
  file_system_access_token(const blink::WebDragData::FilenameItem& item);
  static bool Read(blink::mojom::DataTransferFileDataView data,
                   blink::WebDragData::FilenameItem* out);
};

template <>
struct PLATFORM_EXPORT StructTraits<blink::mojom::DragItemBinaryDataView,
                                    blink::WebDragData::BinaryDataItem> {
  static mojo_base::BigBuffer data(
      const blink::WebDragData::BinaryDataItem& item);
  static bool is_image_accessible(
      const blink::WebDragData::BinaryDataItem& item);
  static blink::KURL source_url(const blink::WebDragData::BinaryDataItem& item);
  static base::FilePath filename_extension(
      const blink::WebDragData::BinaryDataItem& item);
  static blink::String content_disposition(
      const blink::WebDragData::BinaryDataItem& item);
  static bool Read(blink::mojom::DragItemBinaryDataView data,
                   blink::WebDragData::BinaryDataItem* out);
};

template <>
struct PLATFORM_EXPORT
    StructTraits<blink::mojom::DragItemFileSystemFileDataView,
                 blink::WebDragData::FileSystemFileItem> {
  static blink::KURL url(const blink::WebDragData::FileSystemFileItem& item);
  static int64_t size(const blink::WebDragData::FileSystemFileItem& item);
  static blink::String file_system_id(
      const blink::WebDragData::FileSystemFileItem& item);
  static scoped_refptr<blink::BlobDataHandle> serialized_blob(
      const blink::WebDragData::FileSystemFileItem& item);
  static bool Read(blink::mojom::DragItemFileSystemFileDataView data,
                   blink::WebDragData::FileSystemFileItem* out);
};

template <>
struct PLATFORM_EXPORT
    UnionTraits<blink::mojom::DragItemDataView, blink::WebDragData::Item> {
  static const blink::WebDragData::StringItem& string(
      const blink::WebDragData::Item& item) {
    return std::get<blink::WebDragData::StringItem>(item);
  }
  static const blink::WebDragData::FilenameItem& file(
      const blink::WebDragData::Item& item) {
    return std::get<blink::WebDragData::FilenameItem>(item);
  }
  static const blink::WebDragData::BinaryDataItem& binary(
      const blink::WebDragData::Item& item) {
    return std::get<blink::WebDragData::BinaryDataItem>(item);
  }
  static const blink::WebDragData::FileSystemFileItem& file_system_file(
      const blink::WebDragData::Item& item) {
    return std::get<blink::WebDragData::FileSystemFileItem>(item);
  }
  static bool Read(blink::mojom::DragItemDataView data,
                   blink::WebDragData::Item* out);
  static blink::mojom::DragItemDataView::Tag GetTag(
      const blink::WebDragData::Item& item);
};

template <>
struct PLATFORM_EXPORT
    StructTraits<blink::mojom::DragDataDataView, blink::WebDragData> {
  static const std::vector<blink::WebDragData::Item>& items(
      const blink::WebDragData& drag_data);
  static blink::String file_system_id(const blink::WebDragData& drag_data);
  static bool force_default_action(const blink::WebDragData& drag_data);
  static std::optional<blink::mojom::SourceEffectAllowed> source_effect_allowed(
      const blink::WebDragData& drag_data) {
    const blink::WebString source_effect_allowed =
        drag_data.SourceEffectAllowed();
    if (source_effect_allowed.IsNull()) {
      return std::optional<blink::mojom::SourceEffectAllowed>();
    }
    return blink::SourceEffectAllowedFromString(source_effect_allowed.Utf8());
  }
  static network::mojom::ReferrerPolicy referrer_policy(
      const blink::WebDragData& drag_data);
  static bool Read(blink::mojom::DragDataDataView data,
                   blink::WebDragData* out);
};

}  // namespace mojo

#endif  // THIRD_PARTY_BLINK_RENDERER_PLATFORM_MOJO_DRAG_MOJOM_TRAITS_H_