1 files changed, 131 insertions, 0 deletions

diff --git a/c++-libbu++/src/gatscon/proxywidget.cpp b/c++-libbu++/src/gatscon/proxywidget.cpp
new file mode 100644
index 0000000..215f95f
--- /dev/null
+++ b/c++-libbu++/src/gatscon/proxywidget.cpp
@@ -0,0 +1,131 @@
+#include "proxywidget.h"
+#include "proxythread.h"
+
+#include "gatstotree.h"
+#include "treetogats.h"
+
+#include <QMessageBox>
+
+#include <gats/gatsstream.h>
+#include <gats/types.h>
+#include <bu/sio.h>
+#include <bu/file.h>
+
+using namespace Bu;
+
+ProxyWidget::ProxyWidget( QWidget *pParent, int iPortIn,
+                const QByteArray baHost, int iPortOut ) :
+        QWidget( pParent ),
+        pPrx( NULL )
+{
+        setupUi( this );
+
+        pPrx = new ProxyThread( this, iPortIn, baHost, iPortOut );
+
+        connect( pPrx, SIGNAL(gotConnection()),
+                this, SLOT(gotConnection()), Qt::QueuedConnection );
+        connect( pPrx, SIGNAL(recv( Gats::Object *)),
+                this, SLOT(clientRecv(Gats::Object *)), Qt::QueuedConnection );
+        connect( pPrx->pHost, SIGNAL(recv( Gats::Object *)),
+                this, SLOT(hostRecv(Gats::Object *)), Qt::QueuedConnection );
+
+        pPrx->start();
+}
+
+ProxyWidget::~ProxyWidget()
+{
+}
+
+void ProxyWidget::saveTo( const QString &sFile )
+{
+        File fOut( sFile.toAscii().constData(), File::WriteNew );
+        Gats::GatsStream gsOut( fOut );
+        QTreeWidgetItem *pRoot = twHistory->invisibleRootItem();
+        for( int j = 0; j < pRoot->childCount(); j++ )
+        {
+                Gats::Object *pObj = treeToGats( pRoot->child( j ) );
+                gsOut.writeObject( pObj );
+                delete pObj;
+        }
+}
+
+void ProxyWidget::sendToClient()
+{
+        try
+        {
+                Gats::Object *pObj = Gats::Object::strToGats(
+                        leGats->text().toAscii().constData()
+                        );
+                sio << "Send: " << *pObj << sio.nl;
+                QTreeWidgetItem *pIt = new QTreeWidgetItem(
+                        twHistory->invisibleRootItem()
+                        );
+                pIt->setText( 0, "proxy -> client" );
+                gatsToTree( pIt, pObj );
+                pPrx->send( pObj );
+                delete pObj;
+
+                leGats->setText("");
+                leGats->setFocus();
+        }
+        catch( Bu::ExceptionBase &e )
+        {
+                QMessageBox::critical( this, "Gats Console - Error", e.what() );
+        }
+}
+
+void ProxyWidget::sendToServer()
+{
+        try
+        {
+                Gats::Object *pObj = Gats::Object::strToGats(
+                        leGats->text().toAscii().constData()
+                        );
+                sio << "Send: " << *pObj << sio.nl;
+                QTreeWidgetItem *pIt = new QTreeWidgetItem(
+                        twHistory->invisibleRootItem()
+                        );
+                pIt->setText( 0, "proxy -> host" );
+                gatsToTree( pIt, pObj );
+                pPrx->pHost->send( pObj );
+                delete pObj;
+
+                leGats->setText("");
+                leGats->setFocus();
+        }
+        catch( Bu::ExceptionBase &e )
+        {
+                QMessageBox::critical( this, "Gats Console - Error", e.what() );
+        }
+}
+
+void ProxyWidget::clientRecv( Gats::Object *pObj )
+{
+        sio << "Recv: " << *pObj << sio.nl;
+
+        QTreeWidgetItem *pIt = new QTreeWidgetItem(
+                twHistory->invisibleRootItem()
+                );
+        pIt->setText( 0, "client -> host" );
+        gatsToTree( pIt, pObj );
+        delete pObj;
+}
+
+void ProxyWidget::hostRecv( Gats::Object *pObj )
+{
+        sio << "Recv: " << *pObj << sio.nl;
+
+        QTreeWidgetItem *pIt = new QTreeWidgetItem(
+                twHistory->invisibleRootItem()
+                );
+        pIt->setText( 0, "host -> client" );
+        gatsToTree( pIt, pObj );
+        delete pObj;
+}
+
+void ProxyWidget::gotConnection()
+{
+        lwConnect->stop();
+        swRoot->setCurrentIndex( 1 );
+}
+