rooturlschemehandler.cpp

Go to the documentation of this file.

// Author: Sergey Linev <S.Linev@gsi.de>
// Date: 2017-06-29
// Warning: This is part of the ROOT 7 prototype! It will change without notice. It might trigger earthquakes. Feedback is welcome!
 
 
/*************************************************************************
 * Copyright (C) 1995-2019, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
 
#include "rooturlschemehandler.h"
 
#include "rootwebpage.h" // only because of logger channel
#include <cstring>
 
#include <QBuffer>
#include <QByteArray>
#include <QFile>
#include <QWebEngineUrlRequestJob>
 
#include <ROOT/RLogger.hxx>
 
#include "THttpServer.h"
#include "THttpCallArg.h"
#include "TBase64.h"
 
/** \class UrlRequestJobHolder
\ingroup qt5webdisplay
 
Class UrlRequestJobHolder
Required to monitor state of QWebEngineUrlRequestJob
Qt can delete object at any time, therefore one connects destroy signal
from the request to clear pointer
 
*/
 
/////////////////////////////////////////////////////////////////
/// Constructor
 
UrlRequestJobHolder::UrlRequestJobHolder(QWebEngineUrlRequestJob *req) : QObject(), fRequest(req)
{
   if (fRequest)
      connect(fRequest, &QObject::destroyed, this, &UrlRequestJobHolder::onRequestDeleted);
}
 
/////////////////////////////////////////////////////////////////
/// destroyed signal handler
 
void UrlRequestJobHolder::onRequestDeleted(QObject *obj)
{
   if (fRequest == obj)
      fRequest = nullptr;
}
 
/////////////////////////////////////////////////////////////////
/// Reset holder
 
void UrlRequestJobHolder::reset()
{
   if (fRequest)
      disconnect(fRequest, &QObject::destroyed, this, &UrlRequestJobHolder::onRequestDeleted);
   fRequest = nullptr;
}
 
// ===================================================================
 
/////////////////////////////////////////////////////////////////////////////////////
/// Class TWebGuiCallArg
/// Specialized handler of requests in THttpServer with QWebEngine
////////////////////////////////////////////////////////////////////////////////////
 
class TWebGuiCallArg : public THttpCallArg {
 
protected:
   UrlRequestJobHolder fRequest;
 
public:
   explicit TWebGuiCallArg(QWebEngineUrlRequestJob *req = nullptr) : THttpCallArg(), fRequest(req) {}
 
   virtual ~TWebGuiCallArg() {}
 
   /** provide WS kind  */
   const char *GetWSKind() const override { return "rawlongpoll"; }
 
   /** provide WS platform, intentionally keep qt5 here while it only used on client side */
   const char *GetWSPlatform() const override { return "qt5"; }
 
   void SendFile(const char *fname)
   {
      const char *mime = THttpServer::GetMimeType(fname);
 
      QBuffer *buffer = new QBuffer;
 
      QFile file(fname);
      buffer->open(QIODevice::WriteOnly);
      if (file.open(QIODevice::ReadOnly)) {
         QByteArray arr = file.readAll();
         if (strstr(fname, ".mjs") && !strcmp(mime, "text/javascript")) {
            const char *mark1 = "///_begin_exclude_in_qt5web_", *mark2 = "///_end_exclude_in_qt5web_";
            auto p1 = arr.indexOf(mark1);
            auto p2 = arr.indexOf(mark2);
            if ((p1 > 0) && (p2 > p1)) arr.remove(p1, p2 - p1 + strlen(mark2));
         }
 
         buffer->write(arr);
      }
      file.close();
      buffer->close();
 
      QWebEngineUrlRequestJob *req = fRequest.req();
 
      if (req) {
         buffer->connect(req, &QObject::destroyed, buffer, &QObject::deleteLater);
         req->reply(mime, buffer);
         fRequest.reset();
      } else {
         delete buffer;
      }
   }
 
   void HttpReplied() override
   {
      QWebEngineUrlRequestJob *req = fRequest.req();
 
      if (!req) {
         R__LOG_ERROR(QtWebDisplayLog()) << "Qt5 request already processed path " << GetPathName() << " file " << GetFileName();
         return;
      }
 
      if (Is404()) {
         R__LOG_ERROR(QtWebDisplayLog()) << "Qt5 request FAIL path " << GetPathName() << " file " << GetFileName();
 
         req->fail(QWebEngineUrlRequestJob::UrlNotFound);
         // abort request
      } else if (IsFile()) {
         // send file
         SendFile((const char *)GetContent());
      } else {
 
         QBuffer *buffer = new QBuffer;
 
         buffer->open(QIODevice::WriteOnly);
 
         buffer->write((const char *)GetContent(), GetContentLength());
 
         buffer->close();
 
         buffer->connect(req, &QObject::destroyed, buffer, &QObject::deleteLater);
 
         req->reply(GetContentType(), buffer);
      }
 
      fRequest.reset();
   }
};
 
 
/////////////////////////////////////////////////////////////////
/// Returns fully qualified URL, required to open in QWindow
 
QString RootUrlSchemeHandler::MakeFullUrl(THttpServer *serv, const QString &url)
{
   // TODO: provide support for many servers
   fServer = serv;
 
   QString res = "rootscheme://root.server1";
   res.append(url);
   return res;
}
 
 
///////////////////////////////////////////////////////////////////////////////////////////////////
/// Start processing of emulated HTTP request in WebEngine scheme handler
/// Either one reads file or redirect request to THttpServer
 
void RootUrlSchemeHandler::requestStarted(QWebEngineUrlRequestJob *request)
{
   QUrl url = request->requestUrl();
 
   if (!fServer) {
      R__LOG_ERROR(QtWebDisplayLog()) << "Server not specified when request is started";
      request->fail(QWebEngineUrlRequestJob::UrlNotFound);
      return;
   }
 
   QString inp_path = url.path();
   QString inp_query = url.query();
   QString inp_method = request->requestMethod();
 
   // printf("REQUEST PATH:%s QUERY:%s\n", inp_path.toLatin1().data(), inp_query.toLatin1().data());
 
   auto arg = std::make_shared<TWebGuiCallArg>(request);
 
   TString fname;
 
   // process file
   if (fServer->IsFileRequested(inp_path.toLatin1().data(), fname)) {
      arg->SendFile(fname.Data());
      return;
   }
 
   // Analyze and cut post data as soon as possible
   TString query = inp_query.toLatin1().data();
   Int_t pos = query.Index("&post=");
   if (pos != kNPOS) {
      TString buf = TBase64::Decode(query.Data() + pos + 6);
      arg->SetPostData(std::string(buf.Data()));
      query.Resize(pos);
   }
 
   arg->SetPathAndFileName(inp_path.toLatin1().data());
   arg->SetQuery(query.Data());
   arg->SetMethod(inp_method.toLatin1().data());
   arg->SetTopName("webgui");
 
   // can process immediately - function called in main thread
   fServer->SubmitHttp(arg, kTRUE);
}