Initial commit

Lts/Services/FileManagement/pdf_download.cpp

@@ -0,0 +1,82 @@
+/**
+  * @RDE_HEADER@
+  * Download a PDF file from the server
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Hannes Fritsche, NLE, 2019-04-29
+  */
+
+#include "pdf_download.h"
+
+#include "main_handler.h"
+
+#include <tcpserver/tcpserver.h>
+
+namespace FileManagement
+{
+
+CBI_PdfDownload::CBI_PdfDownload(unsigned short port)
+  : CommandInterface(port, "PDF_DOWNLOAD")
+  , m_logger(getLogger("PdfDownload"))
+{}
+
+void CBI_PdfDownload::handleCommandInternal(const QVector<QByteArray> &args)
+{
+  setAutoWriteEOL(false);
+
+  DEBUG_LOG(m_logger) << "Got command for PDF download";
+
+  if (args.count() != 1)
+  {
+    ERROR_LOG(m_logger) << "Invalid arg count";
+    return;
+  }
+  DEBUG_LOG(m_logger) << "Argument count ok";
+
+  bool bOk;
+  int documentId = args.at(0).toInt(&bOk);
+
+  if (!bOk)
+  {
+    const QString err(QStringLiteral("Invalid file id: '%1'").arg(QString::fromUtf8(args.at(0))));
+    writeError(err);
+    ERROR_LOG(m_logger) << err;
+    return;
+  }
+  DEBUG_LOG(m_logger) << "Argument ok";
+
+  MainHandler *mainHandler = static_cast<MainHandler *>(TcpServer::instance(port())->userData());
+
+  Database::FlmFileContent fileContent;
+  mainHandler->db()->selectFlmFileContent(documentId, fileContent);
+
+  if (fileContent.id < 1)
+  {
+    const QString err(QStringLiteral("File with ID '%1' does not exist").arg(QString::fromUtf8(args.at(0))));
+    writeError(err);
+    ERROR_LOG(m_logger) << err;
+    return;
+  }
+  DEBUG_LOG(m_logger) << "File ID valid";
+
+  if (fileContent.fileTypeId != FileManagement::FileType::Pdf)
+  {
+    const QString err(QStringLiteral("File with ID '%1' is not of type Pdf").arg(QString::fromUtf8(args.at(0))));
+    writeError(err);
+    ERROR_LOG(m_logger) << err;
+    return;
+  }
+  DEBUG_LOG(m_logger) << "FileType valid";
+
+  // success
+  DEBUG_LOG(m_logger) << "File ok, sending file \"" << fileContent.name << "\" with ID " << fileContent.id;
+  writeData(fileContent.content);
+}
+
+QString CBI_PdfDownload::help() const
+{
+  return QLatin1String("PDF_DOWNLOAD <fileId> ");
+}
+
+} // namespace FileManagement