Enable start button only when source and target is selected

1 files changed, 25 insertions, 1 deletions

diff --git a/dialog.cpp b/dialog.cpp
index 7b5145b..73a0ff2 100644
--- a/dialog.cpp
+++ b/dialog.cpp
@@ -33,11 +33,14 @@ void Dialog::RefreshDrives(void)
     m_target->Clear();
     m_target->Append(wxT("~/tmp/alpine"));
 #endif
+    if (!m_target->IsEmpty())
+        m_target->SetSelection(0);
 }
 
 void Dialog::OnRefreshDrives(wxCommandEvent &event)
 {
     RefreshDrives();
+    UpdateStartButton();
 }
 
 void Dialog::OnStartButton(wxCommandEvent &event)
@@ -47,5 +50,26 @@ void Dialog::OnStartButton(wxCommandEvent &event)
 
 void Dialog::OnCancelButton(wxCommandEvent &event)
 {
-	Close();	
+    Close();	
 }
+
+void Dialog::UpdateStartButton(void)
+{
+    if ((m_target->GetSelection() != wxNOT_FOUND)
+            && wxFileExists(m_iso_picker->GetPath())) {
+        m_start_button->Enable();
+    } else {
+        m_start_button->Disable();
+    }
+}
+
+void Dialog::OnIsoChange(wxFileDirPickerEvent &event)
+{
+    UpdateStartButton();
+}
+
+void Dialog::OnTargetChange(wxCommandEvent &event)
+{
+    UpdateStartButton();
+}
+