1 files changed, 24 insertions, 1 deletions

diff --git a/src/gatscon/mainwnd.cpp b/src/gatscon/mainwnd.cpp
index c86e1c2..f20a531 100644
--- a/src/gatscon/mainwnd.cpp
+++ b/src/gatscon/mainwnd.cpp
@@ -1,5 +1,10 @@
 #include "mainwnd.h"
+
 #include "clientwidget.h"
+#include "proxywidget.h"
+
+#include "connectdlg.h"
+#include "setupproxydlg.h"
 
 MainWnd::MainWnd()
 {
@@ -12,11 +17,29 @@ MainWnd::~MainWnd()
 
 void MainWnd::connect()
 {
-        setCentralWidget( new ClientWidget( this ) );
+        ConnectDlg dlg( this );
+        if( dlg.exec() == QDialog::Accepted )
+        {
+                setCentralWidget(
+                        new ClientWidget(
+                                this, dlg.getHostname(), dlg.getPort()
+                                )
+                        );
+        }
 }
 
 void MainWnd::proxy()
 {
+        SetupProxyDlg dlg( this );
+
+        if( dlg.exec() == QDialog::Accepted )
+        {
+                setCentralWidget(
+                        new ProxyWidget(
+                                this, dlg.getPortIn(), dlg.getHostOut(), dlg.getPortOut()
+                                )
+                        );
+        }
 }
 
 void MainWnd::open()