From cd1b635b2ae90d9ac2d8b1779183a1fbd8c5fd5c Mon Sep 17 00:00:00 2001
From: vlad <vlad@driver.xyz>
Date: Tue, 13 Jun 2017 16:12:20 -0700
Subject: Adding domain entities

---
 .../xyz/driver/pdsuicommon/pdf/PdfRenderer.scala   |  12 +++
 .../pdsuicommon/pdf/WkHtmlToPdfRenderer.scala      | 104 +++++++++++++++++++++
 2 files changed, 116 insertions(+)
 create mode 100644 src/main/scala/xyz/driver/pdsuicommon/pdf/PdfRenderer.scala
 create mode 100644 src/main/scala/xyz/driver/pdsuicommon/pdf/WkHtmlToPdfRenderer.scala

(limited to 'src/main/scala/xyz/driver/pdsuicommon/pdf')

diff --git a/src/main/scala/xyz/driver/pdsuicommon/pdf/PdfRenderer.scala b/src/main/scala/xyz/driver/pdsuicommon/pdf/PdfRenderer.scala
new file mode 100644
index 0000000..93be2cc
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuicommon/pdf/PdfRenderer.scala
@@ -0,0 +1,12 @@
+package xyz.driver.pdsuicommon.pdf
+
+import java.nio.file.Path
+
+trait PdfRenderer {
+
+  def render(html: String, documentName: String, force: Boolean = false): Path
+
+  def delete(documentName: String): Unit
+
+  def getPath(fileName: String): Path
+}
diff --git a/src/main/scala/xyz/driver/pdsuicommon/pdf/WkHtmlToPdfRenderer.scala b/src/main/scala/xyz/driver/pdsuicommon/pdf/WkHtmlToPdfRenderer.scala
new file mode 100644
index 0000000..9c14646
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuicommon/pdf/WkHtmlToPdfRenderer.scala
@@ -0,0 +1,104 @@
+package xyz.driver.pdsuicommon.pdf
+
+import java.io.IOException
+import java.nio.file._
+
+import io.github.cloudify.scala.spdf._
+import xyz.driver.pdsuicommon.logging._
+import xyz.driver.pdsuicommon.pdf.WkHtmlToPdfRenderer.Settings
+
+object WkHtmlToPdfRenderer {
+
+  final case class Settings(downloadsDir: String) {
+
+    lazy val downloadsPath: Path = getPathFrom(downloadsDir)
+
+    private def getPathFrom(x: String): Path = {
+      val dirPath = if (x.startsWith("/")) Paths.get(x)
+      else {
+        val workingDir = Paths.get(".")
+        workingDir.resolve(x)
+      }
+
+      dirPath.toAbsolutePath.normalize()
+    }
+  }
+}
+
+class WkHtmlToPdfRenderer(settings: Settings) extends PdfRenderer with PhiLogging {
+
+  private val pdf = Pdf(new PdfConfig {
+    disableJavascript := true
+    disableExternalLinks := true
+    disableInternalLinks := true
+    printMediaType := Some(true)
+    orientation := Portrait
+    pageSize := "A4"
+    lowQuality := true
+  })
+
+  override def render(html: String, documentName: String, force: Boolean = false): Path = {
+    checkedCreate(html, documentName, force)
+  }
+
+  override def delete(documentName: String): Unit = {
+    logger.trace(phi"delete(${Unsafe(documentName)})")
+
+    val file = getPath(documentName)
+    logger.debug(phi"File: $file")
+    if (Files.deleteIfExists(file)) {
+      logger.info(phi"Deleted")
+    } else {
+      logger.warn(phi"Doesn't exist")
+    }
+  }
+
+  override def getPath(documentName: String): Path = {
+    settings.downloadsPath.resolve(s"$documentName.pdf").toAbsolutePath
+  }
+
+  protected def checkedCreate[SourceT: SourceDocumentLike](src: SourceT, fileName: String, force: Boolean): Path = {
+    logger.trace(phi"checkedCreate(fileName=${Unsafe(fileName)}, force=$force)")
+
+    val dest = getPath(fileName)
+    logger.debug(phi"Destination file: $dest")
+
+    if (force || !dest.toFile.exists()) {
+      logger.trace(phi"Force refresh the file")
+      val newDocPath = forceCreate(src, dest)
+      logger.info(phi"Updated")
+      newDocPath
+    } else if (dest.toFile.exists()) {
+      logger.trace(phi"Already exists")
+      dest
+    } else {
+      logger.trace(phi"The file does not exist")
+      val newDocPath = forceCreate(src, dest)
+      logger.info(phi"Created")
+      newDocPath
+    }
+  }
+
+  protected def forceCreate[SourceT: SourceDocumentLike](src: SourceT, dest: Path): Path = {
+    logger.trace(phi"forceCreate[${Unsafe(src.getClass.getName)}](dest=$dest)")
+
+    val destTemp = Files.createTempFile("driver", ".pdf")
+    val destTempFile = destTemp.toFile
+
+    Files.createDirectories(dest.getParent)
+
+    val retCode = pdf.run(src, destTempFile)
+    lazy val pdfSize = destTempFile.length()
+    if (retCode != 0) {
+      // Try to google "wkhtmltopdf returns {retCode}"
+      throw new IOException(s"Can create the document, the return code is $retCode")
+    } else if (pdfSize == 0) {
+      // Anything could happen, e.g. https://github.com/wkhtmltopdf/wkhtmltopdf/issues/2540
+      throw new IOException("The pdf is empty")
+    } else {
+      logger.debug(phi"Size: ${Unsafe(pdfSize)}B")
+      Files.move(destTemp, dest, StandardCopyOption.REPLACE_EXISTING)
+      dest
+    }
+  }
+}
-- 
cgit v1.2.3