From 07750774cff03251523edc1b1f585196363b1b65 Mon Sep 17 00:00:00 2001
From: Nico Weber <thakis@chromium.org>
Date: Fri, 22 Mar 2024 13:09:52 -0400
Subject: [PATCH] AK: Allow creating a MaybeOwned<Superclass> from a
 MaybeOwned<Subclass>

---
 AK/MaybeOwned.h | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/AK/MaybeOwned.h b/AK/MaybeOwned.h
index 38e3c20d119..dd87626fb4d 100644
--- a/AK/MaybeOwned.h
+++ b/AK/MaybeOwned.h
@@ -32,6 +32,12 @@ public:
     MaybeOwned(MaybeOwned&&) = default;
     MaybeOwned& operator=(MaybeOwned&&) = default;
 
+    template<DerivedFrom<T> U>
+    MaybeOwned(MaybeOwned<U>&& other)
+        : m_handle(downcast<U, T>(move(other.m_handle)))
+    {
+    }
+
     T* ptr()
     {
         if (m_handle.template has<T*>())
@@ -57,7 +63,22 @@ public:
     bool is_owned() const { return m_handle.template has<NonnullOwnPtr<T>>(); }
 
 private:
-    Variant<NonnullOwnPtr<T>, T*> m_handle;
+    template<typename F>
+    friend class MaybeOwned;
+
+    template<typename HT>
+    using Handle = Variant<NonnullOwnPtr<HT>, HT*>;
+
+    template<typename U, typename D>
+    Handle<D> downcast(Handle<U>&& variant)
+    {
+        if (variant.template has<U*>())
+            return variant.template get<U*>();
+        else
+            return static_cast<NonnullOwnPtr<T>&&>(move(variant.template get<NonnullOwnPtr<U>>()));
+    }
+
+    Handle<T> m_handle;
 };
 
 }