File indexing completed on 2024-05-05 04:48:36

 /***************************************************************************
  *   Copyright (C) 2009, 2010 by Richard J. Moore <rich@kde.org>           *
  *                                                                         *
  *   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 option) 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 "NetworkAccessViewer.h"
 
 #include "core/support/Debug.h"
 
 #include <QNetworkReply>
 #include <QNetworkRequest>
 #include <QSignalMapper>
 
 NetworkAccessViewer::NetworkAccessViewer( QWidget *parent )
     : QObject( parent )
 {
     dialog = new QDialog( parent );
     networkRequestsDialog = new Ui::NetworkRequestsDialog;
     networkRequestsDialog->setupUi(dialog);
 
     mapper = new QSignalMapper(this);
     connect( mapper, QOverload<QObject*>::of(&QSignalMapper::mapped),
              this, &NetworkAccessViewer::requestFinished );
 
     connect( networkRequestsDialog->requestList, &QTreeWidget::currentItemChanged, this, &NetworkAccessViewer::showItemDetails );
     connect( networkRequestsDialog->clearButton, &QPushButton::clicked, this, &NetworkAccessViewer::clear );
 }
 
 NetworkAccessViewer::~NetworkAccessViewer()
 {
     delete networkRequestsDialog;
 }
 
 void NetworkAccessViewer::addRequest( QNetworkAccessManager::Operation op, const QNetworkRequest&req, QIODevice *outgoingData, QNetworkReply *reply )
 {
     Q_UNUSED( outgoingData )
     // Add to list of requests
     QStringList cols;
     switch( op ) {
     case   QNetworkAccessManager::HeadOperation:
         cols << QString::fromLatin1("HEAD");
         break;
     case   QNetworkAccessManager::GetOperation:
         cols << QString::fromLatin1("GET");
         break;
     case   QNetworkAccessManager::PutOperation:
         cols << QString::fromLatin1("PUT");
         break;
     case   QNetworkAccessManager::PostOperation:
         cols << QString::fromLatin1("POST");
         break;
     default:
         qWarning() << "Unknown network operation";
     }
     cols << req.url().toString();
     cols << tr("Pending");
 
     QTreeWidgetItem *item = new QTreeWidgetItem( cols );
     networkRequestsDialog->requestList->addTopLevelItem( item );
     networkRequestsDialog->requestList->scrollToItem( item );
 
     // Add to maps
     requestMap.insert( reply, req );
     itemMap.insert( reply, item );
     itemRequestMap.insert( item, req );
 
     mapper->setMapping( reply, reply );
     connect( reply, &QNetworkReply::finished,
              mapper, QOverload<>::of(&QSignalMapper::map) );
 }
 
 void NetworkAccessViewer::show()
 {
     dialog->show();
 }
 
 void NetworkAccessViewer::hide()
 {
     dialog->hide();
 }
 
 void NetworkAccessViewer::clear()
 {
     requestMap.clear();
     itemMap.clear();
     itemReplyMap.clear();
     itemRequestMap.clear();
     networkRequestsDialog->requestList->clear();
     networkRequestsDialog->requestDetails->clear();
     networkRequestsDialog->responseDetails->clear();
 }
 
 void NetworkAccessViewer::requestFinished( QObject *replyObject )
 {
     QNetworkReply *reply = qobject_cast<QNetworkReply *>( replyObject );
     if( !reply )
     {
         warning() << __PRETTY_FUNCTION__ << "Failed to downcast reply";
         return;
     }
     QTreeWidgetItem *item = itemMap.value( reply );
     if( !item )
     {
         warning() << __PRETTY_FUNCTION__ << "null item";
         return;
     }
 
     // Record the reply headers
     QList<QByteArray> headerValues;
     foreach( const QByteArray &header, reply->rawHeaderList() ) {
         headerValues += reply->rawHeader( header );
     }
     QPair< QList<QByteArray>, QList<QByteArray> > replyHeaders;
     replyHeaders.first = reply->rawHeaderList();
     replyHeaders.second = headerValues;
     itemReplyMap[item] = replyHeaders;
 
     // Display the request
     int status = reply->attribute( QNetworkRequest::HttpStatusCodeAttribute ).toInt();
     QString reason = reply->attribute( QNetworkRequest::HttpReasonPhraseAttribute ).toString();
     item->setText( 2, tr("%1 %2").arg(status).arg(reason) );
 
     QString length = reply->header( QNetworkRequest::ContentLengthHeader ).toString();
     item->setText( 3, length );
 
     QString contentType = reply->header( QNetworkRequest::ContentTypeHeader ).toString();
     item->setText( 4, contentType );
 
     if ( status == 302 ) {
         QUrl target = reply->attribute( QNetworkRequest::RedirectionTargetAttribute ).toUrl();
         item->setText( 5, tr("Redirect: %1").arg( target.toString() ) );
     }
 }
 
 void NetworkAccessViewer::showItemDetails( QTreeWidgetItem *item )
 {
     // Show request details
     QTreeWidget *reqTree = networkRequestsDialog->requestDetails;
     reqTree->clear();
 
     QNetworkRequest req = itemRequestMap[item];
     QByteArray header;
     foreach( header, req.rawHeaderList() ) {
         QTreeWidgetItem *item = new QTreeWidgetItem();
         item->setText( 0, QString::fromLatin1( header ) );
         item->setText( 1, QString::fromLatin1( req.rawHeader( header ) ) );
         reqTree->addTopLevelItem( item );
     }
 
     // Show reply headers
     QTreeWidget *respTree = networkRequestsDialog->responseDetails;
     respTree->clear();
 
     QPair< QList<QByteArray>, QList<QByteArray> > replyHeaders = itemReplyMap[item];
     for ( int i = 0; i < replyHeaders.first.count(); i++ ) {
         QTreeWidgetItem *item = new QTreeWidgetItem();
         item->setText( 0, QString::fromLatin1( replyHeaders.first[i] ) );
         item->setText( 1, QString::fromLatin1( replyHeaders.second[i] ) );
         respTree->addTopLevelItem( item );
     }
 }