diff options
Diffstat (limited to 'src/modules/http/httpfiletransfer.cpp')
| -rw-r--r-- | src/modules/http/httpfiletransfer.cpp | 489 |
1 files changed, 489 insertions, 0 deletions
diff --git a/src/modules/http/httpfiletransfer.cpp b/src/modules/http/httpfiletransfer.cpp new file mode 100644 index 00000000..09c3b1f9 --- /dev/null +++ b/src/modules/http/httpfiletransfer.cpp @@ -0,0 +1,489 @@ +//================================================================================== +// +// File : httpfiletransfer.h +// Creation date : Tue Apr 22 2003 02:00:12 GMT by Szymon Stefanek +// +// This config is part of the KVirc irc client distribution +// Copyright (C) 2003 Szymon Stefanek (pragma at kvirc dot net) +// +// This program is FREE software. You can redistribute it and/or +// modify it under the terms of the GNU General Public License +// as published by the Free Software Foundation; either version 2 +// of the License, or (at your opinion) any later version. +// +// This program is distributed in the HOPE that it will be USEFUL, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. +// See the GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program. If not, write to the Free Software Foundation, +// Inc. ,51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +// +//================================================================================== +#include "httpfiletransfer.h" + +#include "kvi_app.h" +#include "kvi_out.h" +#include "kvi_locale.h" +#include "kvi_window.h" +#include "kvi_iconmanager.h" +#include "kvi_netutils.h" +#include "kvi_kvs_eventtriggers.h" +#include "kvi_console.h" + +#include "kvi_kvs_script.h" + +#include "kvi_tal_popupmenu.h" +#include <qpainter.h> + +static KviPointerList<KviHttpFileTransfer> * g_pHttpFileTransfers = 0; +static QPixmap * g_pHttpIcon = 0; + + +KviHttpFileTransfer::KviHttpFileTransfer() +: KviFileTransfer() +{ + init(); // ensure we're initialized + g_pHttpFileTransfers->append(this); + + m_tStartTime = kvi_unixTime(); + m_tTransferStartTime = 0; + m_tTransferEndTime = 0; + + m_bNotifyCompletion = true; + m_bAutoClean = false; + m_pAutoCleanTimer = 0; + m_bNoOutput = false; + + m_pHttpRequest = new KviHttpRequest(); + + connect(m_pHttpRequest,SIGNAL(status(const QString &)),this,SLOT(statusMessage(const QString &))); + connect(m_pHttpRequest,SIGNAL(terminated(bool)),this,SLOT(transferTerminated(bool))); + connect(m_pHttpRequest,SIGNAL(header(KviPointerHashTable<const char *,KviStr> *)),this,SLOT(headersReceived(KviPointerHashTable<const char *,KviStr> *))); + connect(m_pHttpRequest,SIGNAL(resolvingHost(const QString &)),this,SLOT(resolvingHost(const QString &))); + connect(m_pHttpRequest,SIGNAL(requestSent(const QStringList &)),this,SLOT(requestSent(const QStringList &))); + connect(m_pHttpRequest,SIGNAL(contactingHost(const QString &)),this,SLOT(contactingHost(const QString &))); + connect(m_pHttpRequest,SIGNAL(receivedResponse(const QString &)),this,SLOT(receivedResponse(const QString &))); + connect(m_pHttpRequest,SIGNAL(connectionEstabilished()),this,SLOT(connectionEstabilished())); + + m_eGeneralStatus = Initializing; + m_szStatusString = __tr2qs_ctx("Initializing","http"); +} + +KviHttpFileTransfer::~KviHttpFileTransfer() +{ + g_pHttpFileTransfers->removeRef(this); + delete m_pHttpRequest; + if(m_pAutoCleanTimer) + { + m_pAutoCleanTimer->stop(); + delete m_pAutoCleanTimer; + } +} + +void KviHttpFileTransfer::autoClean() +{ + die(); +} + +void KviHttpFileTransfer::die() +{ + delete this; +} + +QString KviHttpFileTransfer::localFileName() +{ + return m_pHttpRequest->fileName(); +} + +void KviHttpFileTransfer::abort() +{ + m_pHttpRequest->abort(); +} + +void KviHttpFileTransfer::fillContextPopup(KviTalPopupMenu * m,int column) +{ + int id = m->insertItem(__tr2qs_ctx("Abort","http"),this,SLOT(abort())); + if(!active())m->setItemEnabled(id,false); +} + +bool KviHttpFileTransfer::active() +{ + return ((m_eGeneralStatus == Connecting) || (m_eGeneralStatus == Downloading)); +} + +void KviHttpFileTransfer::displayPaint(QPainter * p,int column,int width,int height) +{ + QString txt; + bool bIsTerminated = ((m_eGeneralStatus == Success) || (m_eGeneralStatus == Failure)); + + switch(column) + { + case COLUMN_TRANSFERTYPE: + { + int offset = 0; + switch(m_eGeneralStatus) + { + case Initializing: offset = 0; break; + case Connecting: offset = 0; break; + case Downloading: offset = 48; break; + case Success: offset = 96; break; + case Failure: offset = 144; break; + } + p->drawPixmap(3,3,*g_pHttpIcon,offset,0,48,64); + } + break; + case COLUMN_FILEINFO: + { + QFontMetrics fm(p->font()); + + QString szFrom = __tr2qs_ctx("From: ","http"); + QString szTo = __tr2qs_ctx("To: ","http"); + + int daW1 = fm.width(szFrom); + int daW2 = fm.width(szTo); + if(daW1 < daW2)daW1 = daW2; + int iLineSpacing = fm.lineSpacing(); + + p->setPen(Qt::black); + + int iY = 4; + + p->drawText(4 + daW1,iY,width - (8 + daW1),height - 8,Qt::AlignTop | Qt::AlignLeft,m_pHttpRequest->url().url().ptr()); + iY += iLineSpacing; + if(!(m_pHttpRequest->fileName().isEmpty())) + { + p->drawText(4 + daW1,iY,width - (8 + daW1),height - 8,Qt::AlignTop | Qt::AlignLeft,m_pHttpRequest->fileName()); + } + iY += iLineSpacing; + + + p->setPen(Qt::darkGray); + + p->drawText(4,4,width - 8,height - 8,Qt::AlignTop | Qt::AlignLeft,szFrom); + p->drawText(4,4 + iLineSpacing,width - 8,height - 8,Qt::AlignTop | Qt::AlignLeft,szTo); + + p->setPen(QColor(180,180,200)); + + iLineSpacing += 2; + + p->drawRect(4,height - (iLineSpacing + 4),width - 8,iLineSpacing); + p->fillRect(5,height - (iLineSpacing + 3),width - 10,iLineSpacing - 2,bIsTerminated ? QColor(210,210,210) : QColor(190,190,240)); + p->setPen(Qt::black); + p->drawText(7,height - (iLineSpacing + 4),width - 14,iLineSpacing,Qt::AlignVCenter | Qt::AlignLeft,m_szStatusString); + } + break; + case COLUMN_PROGRESS: + { + QFontMetrics fm(p->font()); + + unsigned int uTotal = m_pHttpRequest->totalSize(); + unsigned int uRecvd = m_pHttpRequest->receivedSize(); + int iW = width - 8; + + p->setPen(bIsTerminated ? Qt::lightGray : QColor(210,210,240)); + p->drawRect(4,4,iW,12); + + int iAvgSpeed = -1; + int iEta = -1; + + if(m_tTransferStartTime > 0) + { + int tSpan = kvi_timeSpan(m_tTransferEndTime > 0 ? m_tTransferEndTime : kvi_unixTime(),m_tTransferStartTime); + if(tSpan > 0) + { + //debug("SPAN: %d (%d - %d)",tSpan,m_tTransferEndTime > 0 ? m_tTransferEndTime : kvi_unixTime(),m_tTransferStartTime); + iAvgSpeed = uRecvd / tSpan; + if(!bIsTerminated && (uTotal >= uRecvd)) + { + unsigned int uRemaining = uTotal - uRecvd; + iEta = uRemaining / iAvgSpeed; + } + } + } + + if(uTotal > 0) + { + double dPerc = (double)(((double)uRecvd) * 100.0) / (double)uTotal; + iW -= 2; + int iL = (int) ((((double)iW) * dPerc) / 100.0); + //iR = iW - iL; + p->fillRect(5,5,iL,10,bIsTerminated ? QColor(140,110,110) : QColor(200,100,100)); + + txt = QString(__tr2qs_ctx("%1 of %2 (%3 %)","http")).arg(KviQString::makeSizeReadable(uRecvd)) + .arg(KviQString::makeSizeReadable(uTotal)).arg(dPerc,0,'f',2); + } else { + txt = KviQString::makeSizeReadable(m_pHttpRequest->receivedSize()); + } + + p->setPen(Qt::black); + + p->drawText(4,19,width - 8,height - 8,Qt::AlignTop | Qt::AlignLeft,txt); + + int iLeftHalf = (iW - 2) / 2; + int iRightHalf = iW - (iLeftHalf + 1); + int iLineSpacing = fm.lineSpacing() + 2; + + /* + txt = __tr2qs_ctx("Spd:","dcc"); + txt += " "; + if(iInstantSpeed >= 0) + { + QString tmpisp; + KviNetUtils::formatNetworkBandwidthString(tmpisp,iAvgSpeed); + txt += tmpisp; + } else { + txt += "? B/s"; + } + */ + + txt = __tr2qs_ctx("Avg:","dcc"); + txt += " "; + if(iAvgSpeed >= 0) + { + QString tmpspd; + KviNetUtils::formatNetworkBandwidthString(tmpspd,iAvgSpeed); + txt += tmpspd; + } else { + txt += "? B/s"; + } + + + + int iDaH = height - (iLineSpacing + 4); + + p->setPen(QColor(180,180,200)); + p->drawRect(4,iDaH,iLeftHalf,iLineSpacing); + p->fillRect(5,iDaH + 1,iLeftHalf - 2,iLineSpacing - 2,bIsTerminated ? QColor(210,210,210) : QColor(190,190,240)); + p->setPen(bIsTerminated ? Qt::darkGray : Qt::black); + p->drawText(6,iDaH,iLeftHalf - 4,iLineSpacing,Qt::AlignLeft | Qt::AlignVCenter,txt); + + unsigned int uD,uH,uM,uS; + + if(bIsTerminated) + { + KviTimeUtils::secondsToDaysHoursMinsSecs(kvi_timeSpan(m_tTransferEndTime,m_tTransferStartTime),&uD,&uH,&uM,&uS); + txt = "TOT: "; + if(uD > 0)txt += QString(__tr2qs_ctx("%1d %2h %3m %4s","http")).arg(uD).arg(uH).arg(uM).arg(uS); + else if(uH > 0)txt += QString(__tr2qs_ctx("%2h %3m %4s","http")).arg(uH).arg(uM).arg(uS); + else txt += QString(__tr2qs_ctx("%3m %4s","http")).arg(uM).arg(uS); + } else { + if(iEta >= 0) + { + KviTimeUtils::secondsToDaysHoursMinsSecs(iEta,&uD,&uH,&uM,&uS); + txt = "ETA: "; + if(uD > 0)txt += QString(__tr2qs_ctx("%1d %2h %3m %4s","http")).arg(uD).arg(uH).arg(uM).arg(uS); + else if(uH > 0)txt += QString(__tr2qs_ctx("%2h %3m %4s","http")).arg(uH).arg(uM).arg(uS); + else txt += QString(__tr2qs_ctx("%3m %4s","http")).arg(uM).arg(uS); + } else { + txt = "ETA: Unknown"; + } + } + + p->setPen(QColor(180,180,200)); + p->drawRect(width - (4 + iRightHalf),iDaH,iRightHalf,iLineSpacing); + p->fillRect(width - (3 + iRightHalf),iDaH + 1,iRightHalf - 2,iLineSpacing - 2,bIsTerminated ? QColor(210,210,210) : QColor(190,190,240)); + p->setPen(bIsTerminated ? Qt::darkGray : Qt::black); + p->drawText(width - (2 + iRightHalf),iDaH,iRightHalf - 4,iLineSpacing,Qt::AlignLeft | Qt::AlignVCenter,txt); + + } + break; + } +} + +int KviHttpFileTransfer::displayHeight(int iLineSpacing) +{ + int iH = (iLineSpacing * 3) + 10; + return iH >= 70 ? iH : 70; +} + +QString KviHttpFileTransfer::tipText() +{ + QString s; + s = QString("<table><tr><td bgcolor=\"#000000\"><font color=\"#FFFFFF\"><b>HTTP Transfer (ID %1)</b></font></td></tr>").arg(id()); + + if(m_lRequest.count() > 0) + { + s += "<tr><td bgcolor=\"#404040\"><font color=\"#FFFFFF\">Request Headers</font></td></tr>"; + s += "<tr><td bgcolor=\"#C0C0C0\">"; + for(QStringList::ConstIterator it = m_lRequest.begin();it != m_lRequest.end();++it) + { + s += " "; + s += *it; + s += "<br>"; + } + s += "</td></tr>"; + } + + if(m_lHeaders.count() > 0) + { + s += "<tr><td bgcolor=\"#404040\"><font color=\"#FFFFFF\">Response Headers</font></td></tr>"; + s += "<tr><td bgcolor=\"#C0C0C0\">"; + for(QStringList::ConstIterator it = m_lHeaders.begin();it != m_lHeaders.end();++it) + { + s += " "; + s += *it; + s += "<br>"; + } + s += "</td></tr>"; + } + + s += "<table>"; + + return s; +} + +void KviHttpFileTransfer::init() +{ + if(g_pHttpFileTransfers)return; + g_pHttpFileTransfers = new KviPointerList<KviHttpFileTransfer>; + g_pHttpFileTransfers->setAutoDelete(false); + + QPixmap * pix = g_pIconManager->getImage("kvi_httpicons.png"); + if(pix)g_pHttpIcon = new QPixmap(*pix); + else g_pHttpIcon = new QPixmap(192,48); +} + +void KviHttpFileTransfer::done() +{ + if(!g_pHttpFileTransfers)return; + while(KviHttpFileTransfer * t = g_pHttpFileTransfers->first()) + delete t; + delete g_pHttpFileTransfers; + g_pHttpFileTransfers = 0; + delete g_pHttpIcon; + g_pHttpIcon = 0; +} + +unsigned int KviHttpFileTransfer::runningTransfers() +{ + if(!g_pHttpFileTransfers)return 0; + return g_pHttpFileTransfers->count(); +} + +void KviHttpFileTransfer::requestSent(const QStringList &requestHeaders) +{ + m_szStatusString = __tr2qs_ctx("Request sent, waiting for reply...","http"); + displayUpdate(); + + KviWindow * out = transferWindow(); + if(!out)return; + + if(!m_bNoOutput) + out->output(KVI_OUT_GENERICSTATUS,__tr2qs_ctx("[HTTP %d]: Request data sent:","http"),id()); + + for(QStringList::ConstIterator it = requestHeaders.begin();it != requestHeaders.end();++it) + { + if(!m_bNoOutput) + out->output(KVI_OUT_GENERICSTATUS,"[HTTP %d]: %s",id(),(*it).utf8().data()); + } + + m_lRequest = requestHeaders; +} + +void KviHttpFileTransfer::connectionEstabilished() +{ + m_szStatusString = __tr2qs_ctx("Connection established, sending request","http"); + displayUpdate(); +} + +void KviHttpFileTransfer::resolvingHost(const QString &hostname) +{ + m_szStatusString = __tr2qs_ctx("Resolving host %1","http").arg(hostname); + displayUpdate(); +} + +void KviHttpFileTransfer::contactingHost(const QString &ipandport) +{ + m_szStatusString = __tr2qs_ctx("Contacting host %1","http").arg(ipandport); + displayUpdate(); +} + +void KviHttpFileTransfer::receivedResponse(const QString &response) +{ + m_lHeaders.clear(); + m_lHeaders.append(response); + m_szStatusString = __tr2qs_ctx("Transferring data (%1)","http").arg(response); + m_tTransferStartTime = kvi_unixTime(); + m_eGeneralStatus = Downloading; + displayUpdate(); +} + +void KviHttpFileTransfer::statusMessage(const QString &txt) +{ + KviWindow * out = transferWindow(); + if(out && (!m_bNoOutput)) + out->output(KVI_OUT_GENERICSTATUS,"[HTTP %d]: %Q",id(),&txt); +} + +void KviHttpFileTransfer::transferTerminated(bool bSuccess) +{ + KviWindow * out = transferWindow(); + + m_tTransferEndTime = kvi_unixTime(); + + KviKvsVariantList vParams; + vParams.append(new KviKvsVariant(bSuccess)); + vParams.append(new KviKvsVariant(m_pHttpRequest->url().url())); + vParams.append(new KviKvsVariant(m_pHttpRequest->fileName())); + vParams.append(new KviKvsVariant(m_vMagicIdentifier)); + + if(m_szCompletionCallback.isNull()) + { + KVS_TRIGGER_EVENT(KviEvent_OnHTTPGetTerminated,out ? out : (KviWindow *)(g_pApp->activeConsole()),&vParams) + } else { + KviKvsScript::run(m_szCompletionCallback,out ? out : (KviWindow *)(g_pApp->activeConsole()),&vParams); + } + + if(bSuccess) + { + m_szStatusString = __tr2qs_ctx("Transfer completed","http"); + m_eGeneralStatus = Success; + displayUpdate(); + if(out && (!m_bNoOutput))out->output(KVI_OUT_GENERICSUCCESS,__tr2qs_ctx("[HTTP %d]: Transfer completed","http"),id()); + g_pApp->fileDownloadTerminated(true,m_pHttpRequest->url().url().ptr(),m_pHttpRequest->fileName(),QString::null,QString::null,!m_bNotifyCompletion); + } else { + m_szStatusString = __tr2qs_ctx("Transfer failed","http"); + m_szStatusString += ": "; + m_szStatusString += m_pHttpRequest->lastError(); + m_eGeneralStatus = Failure; + displayUpdate(); + if(out && (!m_bNoOutput))out->output(KVI_OUT_GENERICERROR,__tr2qs_ctx("[HTTP %d]: Transfer failed: %Q","http"),id(),&(m_pHttpRequest->lastError())); + g_pApp->fileDownloadTerminated(false,m_pHttpRequest->url().url().ptr(),m_pHttpRequest->fileName(),QString::null,m_pHttpRequest->lastError(),!m_bNotifyCompletion); + } + + if(m_bAutoClean) + { + if(m_pAutoCleanTimer)delete m_pAutoCleanTimer; + m_pAutoCleanTimer = new QTimer(); + connect(m_pAutoCleanTimer,SIGNAL(timeout()),this,SLOT(autoClean())); + m_pAutoCleanTimer->start(100,true); + } +} + +void KviHttpFileTransfer::headersReceived(KviPointerHashTable<const char *,KviStr> *h) +{ + if(!h)return; + KviWindow * out = transferWindow(); + + if(out && (!m_bNoOutput))out->output(KVI_OUT_GENERICSTATUS,__tr2qs_ctx("[HTTP %d]: Response headers:","http"),id()); + KviPointerHashTableIterator<const char *,KviStr> it(*h); + while(KviStr * s = it.current()) + { + QString szHeader = it.currentKey(); + szHeader += ": "; + szHeader += s->ptr(); + m_lHeaders.append(szHeader); + if(out && (!m_bNoOutput))out->output(KVI_OUT_GENERICSTATUS,"[HTTP %d]: %s: %s",id(),it.currentKey(),s->ptr()); + ++it; + } +} + +bool KviHttpFileTransfer::startDownload() +{ + m_eGeneralStatus = Connecting; + return m_pHttpRequest->start(); +} + +#include "m_httpfiletransfer.moc" |