1 files changed, 59 insertions, 0 deletions

diff --git a/src/gatscon/mainwnd.cpp b/src/gatscon/mainwnd.cpp
index 628988e..bddaf08 100644
--- a/src/gatscon/mainwnd.cpp
+++ b/src/gatscon/mainwnd.cpp
@@ -7,11 +7,15 @@
 #include "connectdlg.h"
 #include "setupproxydlg.h"
 
+#include <QLabel>
 #include <QFileDialog>
 
 MainWnd::MainWnd()
 {
         setupUi( this );
+
+        pMode = new QLabel( "Idle", this );
+        statusBar()->addPermanentWidget( pMode );
 }
 
 MainWnd::~MainWnd()
@@ -23,11 +27,16 @@ void MainWnd::connect()
         ConnectDlg dlg( this );
         if( dlg.exec() == QDialog::Accepted )
         {
+                sCurFile.clear();
                 setCentralWidget(
                         new ClientWidget(
                                 this, dlg.getHostname(), dlg.getPort()
                                 )
                         );
+                pMode->setText(
+                        QString("Client Mode: %1:%2").arg( QString(dlg.getHostname()) ).
+                                arg( dlg.getPort() )
+                        );
         }
 }
 
@@ -37,11 +46,17 @@ void MainWnd::proxy()
 
         if( dlg.exec() == QDialog::Accepted )
         {
+                sCurFile.clear();
                 setCentralWidget(
                         new ProxyWidget(
                                 this, dlg.getPortIn(), dlg.getHostOut(), dlg.getPortOut()
                                 )
                         );
+                pMode->setText(
+                        QString("Proxy Mode: :%1 -> %2:%3").arg( dlg.getPortIn() ).
+                                arg( QString(dlg.getHostOut()) ).
+                                arg( dlg.getPortOut() )
+                        );
         }
 }
 
@@ -53,8 +68,52 @@ void MainWnd::open()
         if( sFile.isEmpty() )
                 return;
 
+        sCurFile = sFile;
         setCentralWidget(
                 new FileWidget( this, sFile )
                 );
+        pMode->setText( QString("File mode: %1").arg( sCurFile ) );
+}
+
+void MainWnd::newFile()
+{
+        sCurFile.clear();
+        setCentralWidget(
+                new FileWidget( this )
+                );
+        pMode->setText( QString("File mode: <untitled>") );
+}
+
+void MainWnd::save()
+{
+        if( sCurFile.isEmpty() )
+        {
+                saveAs();
+        }
+        else
+        {
+                IoBase *pIo = dynamic_cast<IoBase *>(centralWidget());
+                if( !pIo )
+                        return;
+
+                pIo->saveTo( sCurFile );
+        }
+}
+
+void MainWnd::saveAs()
+{
+        IoBase *pIo = dynamic_cast<IoBase *>(centralWidget());
+        if( !pIo )
+                return;
+
+        QString sFile = QFileDialog::getSaveFileName(
+                this, "GatsCon - save gats file"
+                );
+        if( sFile.isEmpty() )
+                return;
+
+        pIo->saveTo( sFile );
+
+        sCurFile = sFile;
 }