From 51e79a2bbcd1652d1980df301a72da72ee82f175 Mon Sep 17 00:00:00 2001
From: FalseHonesty <thefalsehonesty@gmail.com>
Date: Sat, 23 May 2020 13:03:13 -0400
Subject: [PATCH] LibWeb: Add hook to HtmlView when a new document is set

---
 Libraries/LibWeb/HtmlView.cpp | 3 +++
 Libraries/LibWeb/HtmlView.h   | 1 +
 2 files changed, 4 insertions(+)

diff --git a/Libraries/LibWeb/HtmlView.cpp b/Libraries/LibWeb/HtmlView.cpp
index 57403b9a61..49bd81d61d 100644
--- a/Libraries/LibWeb/HtmlView.cpp
+++ b/Libraries/LibWeb/HtmlView.cpp
@@ -89,6 +89,9 @@ void HtmlView::set_document(Document* new_document)
 
     main_frame().set_document(new_document);
 
+    if (on_set_document)
+        on_set_document(new_document);
+
     if (new_document) {
         new_document->on_layout_updated = [this] {
             layout_and_sync_size();
diff --git a/Libraries/LibWeb/HtmlView.h b/Libraries/LibWeb/HtmlView.h
index 4804a12087..649535227e 100644
--- a/Libraries/LibWeb/HtmlView.h
+++ b/Libraries/LibWeb/HtmlView.h
@@ -65,6 +65,7 @@ public:
     Function<void(const URL&)> on_load_start;
     Function<void(const Gfx::Bitmap&)> on_favicon_change;
     Function<void(const URL&)> on_url_drop;
+    Function<void(Document*)> on_set_document;
 
     virtual bool accepts_focus() const override { return true; }