ladybird/Userland/Libraries/LibWeb/HTML/DataTransferItem.cpp
Timothy Flynn b978dba8bd LibWeb: Implement DataTransferItem.prototype.getAsFile()
This is used on imgur after drag-and-dropping an image onto its page.
2024-08-23 10:10:34 +01:00

117 lines
3.9 KiB
C++

/*
* Copyright (c) 2024, Tim Flynn <trflynn89@serenityos.org>
*
* SPDX-License-Identifier: BSD-2-Clause
*/
#include <LibJS/Runtime/Realm.h>
#include <LibWeb/Bindings/DataTransferItemPrototype.h>
#include <LibWeb/Bindings/Intrinsics.h>
#include <LibWeb/FileAPI/File.h>
#include <LibWeb/HTML/DataTransfer.h>
#include <LibWeb/HTML/DataTransferItem.h>
namespace Web::HTML {
JS_DEFINE_ALLOCATOR(DataTransferItem);
JS::NonnullGCPtr<DataTransferItem> DataTransferItem::create(JS::Realm& realm, JS::NonnullGCPtr<DataTransfer> data_transfer, size_t item_index)
{
return realm.heap().allocate<DataTransferItem>(realm, realm, data_transfer, item_index);
}
DataTransferItem::DataTransferItem(JS::Realm& realm, JS::NonnullGCPtr<DataTransfer> data_transfer, size_t item_index)
: PlatformObject(realm)
, m_data_transfer(data_transfer)
, m_item_index(item_index)
{
}
DataTransferItem::~DataTransferItem() = default;
void DataTransferItem::initialize(JS::Realm& realm)
{
Base::initialize(realm);
WEB_SET_PROTOTYPE_FOR_INTERFACE(DataTransferItem);
}
void DataTransferItem::visit_edges(JS::Cell::Visitor& visitor)
{
Base::visit_edges(visitor);
visitor.visit(m_data_transfer);
}
// https://html.spec.whatwg.org/multipage/dnd.html#dom-datatransferitem-kind
String DataTransferItem::kind() const
{
// The kind attribute must return the empty string if the DataTransferItem object is in the disabled mode; otherwise
// it must return the string given in the cell from the second column of the following table from the row whose cell
// in the first column contains the drag data item kind of the item represented by the DataTransferItem object:
//
// Kind | String
// ---------------
// Text | "string"
// File | "file"
if (!mode().has_value())
return {};
auto const& item = m_data_transfer->drag_data(*m_item_index);
switch (item.kind) {
case DragDataStoreItem::Kind::Text:
return "string"_string;
case DragDataStoreItem::Kind::File:
return "file"_string;
}
VERIFY_NOT_REACHED();
}
// https://html.spec.whatwg.org/multipage/dnd.html#dom-datatransferitem-type
String DataTransferItem::type() const
{
// The type attribute must return the empty string if the DataTransferItem object is in the disabled mode; otherwise
// it must return the drag data item type string of the item represented by the DataTransferItem object.
if (!mode().has_value())
return {};
auto const& item = m_data_transfer->drag_data(*m_item_index);
return item.type_string;
}
Optional<DragDataStore::Mode> DataTransferItem::mode() const
{
if (!m_item_index.has_value())
return {};
return m_data_transfer->mode();
}
// https://html.spec.whatwg.org/multipage/dnd.html#dom-datatransferitem-getasfile
JS::GCPtr<FileAPI::File> DataTransferItem::get_as_file() const
{
auto& realm = this->realm();
// 1. If the DataTransferItem object is not in the read/write mode or the read-only mode, then return null
if (mode() != DragDataStore::Mode::ReadWrite && mode() != DragDataStore::Mode::ReadOnly)
return nullptr;
auto const& item = m_data_transfer->drag_data(*m_item_index);
// 2. If the drag data item kind is not File, then return null.
if (item.kind != DragDataStoreItem::Kind::File)
return nullptr;
// 3. Return a new File object representing the actual data of the item represented by the DataTransferItem object.
auto blob = FileAPI::Blob::create(realm, item.data, item.type_string);
// FIXME: The FileAPI should use ByteString for file names.
auto file_name = MUST(String::from_byte_string(item.file_name));
// FIXME: Fill in other fields (e.g. last_modified).
FileAPI::FilePropertyBag options {};
options.type = item.type_string;
return MUST(FileAPI::File::create(realm, { JS::make_handle(blob) }, file_name, move(options)));
}
}