/*
 * 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 "mainwnd.h"

#include "clientwidget.h"
#include "proxywidget.h"
#include "filewidget.h"

#include "connectdlg.h"
#include "setupproxydlg.h"

#include <QLabel>
#include <QFileDialog>

MainWnd::MainWnd()
{
    setupUi( this );

    pMode = new QLabel( "Idle", this );
    statusBar()->addPermanentWidget( pMode );
}

MainWnd::~MainWnd()
{
}

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

void MainWnd::proxy()
{
    SetupProxyDlg dlg( this );

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

void MainWnd::open()
{
    QString sFile = QFileDialog::getOpenFileName(
        this, "Gats Console - open gats file"
        );
    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, "Gats Console - save gats file"
        );
    if( sFile.isEmpty() )
        return;

    pIo->saveTo( sFile );

    sCurFile = sFile;
}