1 files changed, 125 insertions, 0 deletions
diff --git a/Userland/Libraries/LibGUI/DragOperation.cpp b/Userland/Libraries/LibGUI/DragOperation.cpp
new file mode 100644
index 0000000000..060f25fc62
--- /dev/null
+++ b/Userland/Libraries/LibGUI/DragOperation.cpp
@@ -0,0 +1,125 @@
+/*
+ * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ *    list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <AK/Badge.h>
+#include <AK/SharedBuffer.h>
+#include <LibCore/EventLoop.h>
+#include <LibCore/MimeData.h>
+#include <LibGUI/DragOperation.h>
+#include <LibGUI/WindowServerConnection.h>
+#include <LibGfx/Bitmap.h>
+
+namespace GUI {
+
+static DragOperation* s_current_drag_operation;
+
+DragOperation::DragOperation(Core::Object* parent)
+    : Core::Object(parent)
+{
+}
+
+DragOperation::~DragOperation()
+{
+}
+
+DragOperation::Outcome DragOperation::exec()
+{
+    ASSERT(!s_current_drag_operation);
+    ASSERT(!m_event_loop);
+    ASSERT(m_mime_data);
+
+    int bitmap_id = -1;
+    Gfx::IntSize bitmap_size;
+    RefPtr<Gfx::Bitmap> shared_bitmap;
+    if (m_mime_data->has_format("image/x-raw-bitmap")) {
+        auto data = m_mime_data->data("image/x-raw-bitmap");
+        auto bitmap = Gfx::Bitmap::create_from_serialized_byte_buffer(move(data));
+        shared_bitmap = bitmap->to_bitmap_backed_by_shared_buffer();
+        shared_bitmap->shared_buffer()->share_with(WindowServerConnection::the().server_pid());
+        bitmap_id = shared_bitmap->shbuf_id();
+        bitmap_size = shared_bitmap->size();
+    }
+
+    auto response = WindowServerConnection::the().send_sync<Messages::WindowServer::StartDrag>(
+        m_mime_data->text(),
+        m_mime_data->all_data(),
+        bitmap_id, bitmap_size);
+
+    if (!response->started()) {
+        m_outcome = Outcome::Cancelled;
+        return m_outcome;
+    }
+
+    s_current_drag_operation = this;
+    m_event_loop = make<Core::EventLoop>();
+    auto result = m_event_loop->exec();
+    m_event_loop = nullptr;
+    dbgln("{}: event loop returned with result {}", class_name(), result);
+    remove_from_parent();
+    s_current_drag_operation = nullptr;
+    return m_outcome;
+}
+
+void DragOperation::done(Outcome outcome)
+{
+    ASSERT(m_outcome == Outcome::None);
+    m_outcome = outcome;
+    m_event_loop->quit(0);
+}
+
+void DragOperation::notify_accepted(Badge<WindowServerConnection>)
+{
+    ASSERT(s_current_drag_operation);
+    s_current_drag_operation->done(Outcome::Accepted);
+}
+
+void DragOperation::notify_cancelled(Badge<WindowServerConnection>)
+{
+    if (s_current_drag_operation)
+        s_current_drag_operation->done(Outcome::Cancelled);
+}
+
+void DragOperation::set_text(const String& text)
+{
+    if (!m_mime_data)
+        m_mime_data = Core::MimeData::construct();
+    m_mime_data->set_text(text);
+}
+void DragOperation::set_bitmap(const Gfx::Bitmap* bitmap)
+{
+    if (!m_mime_data)
+        m_mime_data = Core::MimeData::construct();
+    if (bitmap)
+        m_mime_data->set_data("image/x-raw-bitmap", bitmap->serialize_to_byte_buffer());
+}
+void DragOperation::set_data(const String& data_type, const String& data)
+{
+    if (!m_mime_data)
+        m_mime_data = Core::MimeData::construct();
+    m_mime_data->set_data(data_type, data.to_byte_buffer());
+}
+
+}