diff --git a/Libraries/LibWeb/Frame.h b/Libraries/LibWeb/Frame.h index 0a8ae556e6..cc98cbb629 100644 --- a/Libraries/LibWeb/Frame.h +++ b/Libraries/LibWeb/Frame.h @@ -47,13 +47,15 @@ public: static NonnullRefPtr create(PageView& page_view) { return adopt(*new Frame(page_view)); } ~Frame(); + bool is_main_frame() const { return this == &m_main_frame; } + const Document* document() const { return m_document; } Document* document() { return m_document; } void set_document(Document*); - PageView* page_view() { return m_page_view; } - const PageView* page_view() const { return m_page_view; } + PageView* page_view() { return is_main_frame() ? m_page_view : main_frame().m_page_view; } + const PageView* page_view() const{ return is_main_frame() ? m_page_view : main_frame().m_page_view; } const Gfx::Size& size() const { return m_size; } void set_size(const Gfx::Size&);