1 files changed, 8 insertions, 5 deletions
diff --git a/src/timeoutdialog.h b/src/timeoutdialog.h
index eb853f3..b70f753 100644
--- a/src/timeoutdialog.h
+++ b/src/timeoutdialog.h
@@ -7,24 +7,27 @@
 #include <QProgressDialog>
 #include <QTimer>
 
-class TimeOutDialog : public QProgressDialog
+namespace Ui {
+class TimeOutDialog;
+}
+
+class TimeOutDialog : public QDialog
 {
 	Q_OBJECT
 
 public:
-	TimeOutDialog(int time, QWidget *parent = 0);
+	TimeOutDialog(int time, QWidget *parent = nullptr);
 	bool isActive() const { return _timer.isActive(); }
+	bool wasCanceled() const { return _time > 0; }
 
 private:
+	Ui::TimeOutDialog *_ui;
 	int             _time;
 	QTimer          _timer;
 
 protected:
 	void hideEvent(QHideEvent *event) override;
 
-public slots:
-	void cancel();
-
 private slots:
 	void update();
 };