worked on the UpdatePromptDialog

src/singletons/Updates.cpp

@@ -141,12 +141,40 @@ void Updates::checkForUpdates()
 #endif
 }
 
-Updates::UpdateStatus Updates::getStatus() const
+Updates::Status Updates::getStatus() const
 {
     return this->status_;
 }
 
-void Updates::setStatus_(UpdateStatus status)
+bool Updates::shouldShowUpdateButton() const
+{
+    switch (this->getStatus()) {
+        case UpdateAvailable:
+        case SearchFailed:
+        case Downloading:
+        case DownloadFailed:
+        case WriteFileFailed:
+            return true;
+
+        default:
+            return false;
+    }
+}
+
+bool Updates::isError() const
+{
+    switch (this->getStatus()) {
+        case SearchFailed:
+        case DownloadFailed:
+        case WriteFileFailed:
+            return true;
+
+        default:
+            return false;
+    }
+}
+
+void Updates::setStatus_(Status status)
 {
     if (this->status_ != status) {
         this->status_ = status;

src/singletons/Updates.hpp

@@ -10,7 +10,7 @@ class Updates
     Updates();
 
 public:
-    enum UpdateStatus {
+    enum Status {
         None,
         Searching,
         UpdateAvailable,
@@ -28,18 +28,21 @@ public:
     const QString &getCurrentVersion() const;
     const QString &getOnlineVersion() const;
     void installUpdates();
-    UpdateStatus getStatus() const;
+    Status getStatus() const;
 
-    pajlada::Signals::Signal<UpdateStatus> statusUpdated;
+    bool shouldShowUpdateButton() const;
+    bool isError() const;
+
+    pajlada::Signals::Signal<Status> statusUpdated;
 
 private:
     QString currentVersion_;
     QString onlineVersion_;
-    UpdateStatus status_ = None;
+    Status status_ = None;
 
     QString updateUrl_;
 
-    void setStatus_(UpdateStatus status);
+    void setStatus_(Status status);
 };
 
 }  // namespace chatterino