/*
 * Copyright (C) 2007-2012 Xagasoft, All rights reserved.
 *
 * This file is part of the libgats library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#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 );
}