source: psi/trunk/src/jltest.cpp

#include"jltest.h"

#include<qpushbutton.h>
#include<qlayout.h>
#include<qcheckbox.h>
#include<qmessagebox.h>
#include<qgroupbox.h>
#include<qprogressbar.h>
#include<qradiobutton.h>
#include<qbuttongroup.h>
#include<qdom.h>
#include<qtimer.h>
#include<stdlib.h>
#include"im.h"
#include"xmpp_jidlink.h"
#include"xmpp_xmlcommon.h"
#include"psiaccount.h"
#include"busywidget.h"

#define TOTAL_SIZE   32768
#define PACKET_SIZE  8192
#define PACKET_DELAY 2000

class JLTestDlg::Private
{
public:
        PsiAccount *pa;
        Jid jid;
        BusyWidget *busy;
        QRadioButton *rb_dtcp, *rb_ibb;
        QPushButton *pb_test;
        QProgressBar *pr_status;
        QButtonGroup *bg_type;
        JidLink *jl;
        QLabel *lb_status;
        int mode;

        int numBytes;
};

JLTestDlg::JLTestDlg(const Jid &j, PsiAccount *pa)
:QDialog(0, 0, false, WDestructiveClose)
{
        d = new Private;
        d->pa = pa;
        d->jid = j;
        d->mode = ModeConnect;
        d->jl = 0;
        setCaption("Link Test");
        init();
        d->pb_test->setText(tr("Connect"));

        d->rb_dtcp->setChecked(true);
}

JLTestDlg::JLTestDlg(const Jid &j, JidLink *jl, PsiAccount *pa)
:QDialog(0, 0, false, WDestructiveClose)
{
        d = new Private;
        d->pa = pa;
        d->jid = j;
        d->mode = ModeAccept;
        d->jl = jl;
        setCaption("Link Test");
        init();
        d->pb_test->setText(tr("Accept"));
        if(jl->type() == JidLink::DTCP)
                d->rb_dtcp->setChecked(true);
        else
                d->rb_ibb->setChecked(true);
        d->bg_type->setEnabled(false);
}

void JLTestDlg::init()
{
        QVBoxLayout *vb = new QVBoxLayout(this, 8);

        // JID
        QLabel *l = new QLabel(QString("Link test with %1").arg(d->jid.full()), this);
        vb->addWidget(l);

        // type selection
        d->bg_type = new QButtonGroup(1, Vertical, "Type", this);
        vb->addWidget(d->bg_type);
        d->rb_dtcp = new QRadioButton("Direct (S5B)", d->bg_type);
        d->rb_ibb = new QRadioButton("Thru-server (IBB)", d->bg_type);

        // status area
        QGroupBox *gr = new QGroupBox(1, Horizontal, "Status", this);
        d->pr_status = new QProgressBar(100, gr);
        d->pr_status->setProgress(0);
        d->lb_status = new QLabel(gr);
        d->lb_status->setText("Idle");

        vb->addWidget(gr);

        vb->addStretch(1);
        QHBoxLayout *hb = new QHBoxLayout(vb);
        d->busy = new BusyWidget(this);
        hb->addWidget(d->busy);
        d->pb_test = new QPushButton(this);
        connect(d->pb_test, SIGNAL(clicked()), SLOT(start()));
        hb->addWidget(d->pb_test);
        QPushButton *pb_close = new QPushButton(tr("Close"), this);
        connect(pb_close, SIGNAL(clicked()), SLOT(close()));
        hb->addWidget(pb_close);

        d->pb_test->setFocus();
}

JLTestDlg::~JLTestDlg()
{
        if(d->jl) {
                d->jl->close();
                delete d->jl;
        }
        delete d;
}

void JLTestDlg::reset()
{
        d->mode = ModeConnect;
        d->pb_test->setEnabled(true);
        d->pb_test->setText(tr("Connect"));
        d->lb_status->setText("Idle");
        d->pr_status->setProgress(0);
        d->bg_type->setEnabled(true);
}

void JLTestDlg::start()
{
        if(d->mode == ModeConnect) {
                int type;
                if(d->rb_dtcp->isChecked())
                        type = JidLink::DTCP;
                else
                        type = JidLink::IBB;

                d->jl = new JidLink(d->pa->client());
                connect(d->jl, SIGNAL(connected()), SLOT(jl_connected()));
                connect(d->jl, SIGNAL(connectionClosed()), SLOT(jl_connectionClosed()));
                connect(d->jl, SIGNAL(error(int)), SLOT(jl_error(int)));
                connect(d->jl, SIGNAL(bytesWritten(int)), SLOT(jl_bytesWritten(int)));
                connect(d->jl, SIGNAL(readyRead()), SLOT(jl_readyRead()));

                QDomElement e = textTag(d->pa->client()->doc(), "comment", "this is only a test!");
                d->jl->connectToJid(d->jid, type, e);

                d->bg_type->setEnabled(false);
                d->lb_status->setText("Requesting...");
        }
        else {
                connect(d->jl, SIGNAL(connected()), SLOT(jl_connected()));
                connect(d->jl, SIGNAL(connectionClosed()), SLOT(jl_connectionClosed()));
                connect(d->jl, SIGNAL(error(int)), SLOT(jl_error(int)));
                connect(d->jl, SIGNAL(bytesWritten(int)), SLOT(jl_bytesWritten(int)));
                connect(d->jl, SIGNAL(readyRead()), SLOT(jl_readyRead()));

                d->jl->accept();

                d->lb_status->setText("Connecting...");
        }

        d->busy->start();
        d->pb_test->setEnabled(false);
}

void JLTestDlg::jl_connected()
{
        d->lb_status->setText("Transferring data...");
        d->numBytes = 0;

        if(d->mode == ModeConnect)
                doNext();
}

void JLTestDlg::jl_connectionClosed()
{
        d->busy->stop();

        delete d->jl;
        d->jl = 0;

        QMessageBox::information(this, tr("Error"), tr("Connection closed."));

        reset();
}

void JLTestDlg::jl_error(int)
{
        d->busy->stop();

        QString str;
        delete d->jl;
        d->jl = 0;

        /*if(x == DTCPConnection::ErrRequest)
                str = tr("Error during request!");
        else if(x == DTCPConnection::ErrConnect)
                str = tr("Error during connect!");
        else
                str = tr("Socket error!");*/
        str = tr("Stream or connection error.");

        QMessageBox::information(this, tr("Error"), str);

        reset();
}

void JLTestDlg::jl_readyRead()
{
        QByteArray r;
        r = d->jl->read();
        printf("got [%d] bytes\n", r.size());

        d->numBytes += r.size();
        int percent = 100 * d->numBytes / TOTAL_SIZE;
        d->pr_status->setProgress(percent);

        if(d->numBytes >= TOTAL_SIZE) {
                d->busy->stop();

                d->jl->close();
                delete d->jl;
                d->jl = 0;

                d->lb_status->setText("Finished");
        }
}

void JLTestDlg::jl_bytesWritten(int x)
{
        d->numBytes += x;
        int percent = 100 * d->numBytes / TOTAL_SIZE;
        d->pr_status->setProgress(percent);

        if(d->jl->bytesToWrite() > 0)
                return;

        if(d->numBytes >= TOTAL_SIZE) {
                d->busy->stop();

                d->jl->close();
                d->jl->deleteLater();
                d->jl = 0;

                d->lb_status->setText("Finished");
        }
        else {
                QTimer::singleShot(PACKET_DELAY, this, SLOT(doNext()));
        }
}

void JLTestDlg::doNext()
{
        if(!d->jl)
                return;

        writePacket();
}

void JLTestDlg::writePacket()
{
        QByteArray a(PACKET_SIZE);
        for(int n = 0; n < (int)a.size(); ++n)
                a[n] = rand();
        d->jl->write(a);
}