1 files changed, 8 insertions, 5 deletions
diff --git a/src/lib/navigation/navigationbutton.h b/src/lib/navigation/navigationbutton.h
index 44e3ceb..678d48d 100644
--- a/src/lib/navigation/navigationbutton.h
+++ b/src/lib/navigation/navigationbutton.h
@@ -33,26 +33,29 @@ public:
 
     enum Type {
         BackButton,
-        ForwardButton
+        ForwardButton,
+        ReloadButton,
+        StopButton
     };
 
     explicit NavigationButton(Type type, QWidget *parent = nullptr);
 
-    void setView(const WebView *view);
+    void setView(WebView *view);
 
 signals:
 
 private slots:
-    void updateEnableState();
+    void updateOnLoadStarted();
+    void updateOnLoadFinished();
     void doAction();
     void prepareMenu();
 
 private:
     Type m_type;
     QMenu *menu;
-    const WebView *m_view;
+    WebView *m_view;
 
-    QMetaObject::Connection loadFinishedConnection;
+    QMetaObject::Connection loadStartedConnection, loadFinishedConnection;
 };
 
 #endif // NAVIGATIONBUTTON_H