org.beangle.doc.docx.DocHelper.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of beangle-doc-docx Show documentation
The Beangle Doc Library
The newest version!
/*
 * Copyright (C) 2005, The Beangle Software.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */

package org.beangle.doc.docx

import org.apache.poi.common.usermodel.PictureType
import org.apache.poi.util.Units
import org.apache.poi.xwpf.usermodel.{XWPFDocument, XWPFRun}
import org.beangle.commons.codec.binary.Base64
import org.beangle.commons.collection.Collections
import org.beangle.commons.lang.{Chars, Strings}

import java.io.{ByteArrayInputStream, ByteArrayOutputStream}
import java.net.URL

object DocHelper {

  def toDoc(url: URL, data: collection.Map[String, String]): Array[Byte] = {
    val templateIs = url.openStream()
    val doc = new XWPFDocument(templateIs)
    import scala.jdk.javaapi.CollectionConverters.*

    for (p <- asScala(doc.getParagraphs)) {
      val runs = p.getRuns
      if (runs != null) {
        for (r <- asScala(runs)) fillin(r, data)
      }
    }

    for (tbl <- asScala(doc.getTables)) {
      for (row <- asScala(tbl.getRows)) {
        for (cell <- asScala(row.getTableCells)) {
          for (p <- asScala(cell.getParagraphs)) {
            for (r <- asScala(p.getRuns)) fillin(r, data)
          }
        }
      }
    }
    val bos = new ByteArrayOutputStream()
    doc.write(bos)
    templateIs.close()
    bos.toByteArray
  }

  private def fillin(r: XWPFRun, data: collection.Map[String, String]): Unit = {
    val text = r.getText(0)
    if (text != null) {
      if (text.contains("${")) {
        var processed = replace(text, data)
        if (text.startsWith("[#maxlen")) {
          val max = Integer.parseInt(Strings.substringBetween(text, "[#maxlen", "]").trim())
          processed = processed.substring(processed.indexOf(']') + 1)
          val resultLen = Chars.charLength(processed)
          if (resultLen > max) {
            val scale = java.lang.Double.valueOf(max * 100.0 / resultLen).toInt
            r.setTextScale(scale)
          }
        }
        r.setText(processed, 0)
      } else if (text.startsWith("[#img")) {

        val propertyStr = Strings.substringBetween(text, "[#img ", "/]")
        val p = Strings.split(propertyStr, "=").flatMap(Strings.split)
        val properties = Collections.newMap[String, String]
        var i = 1
        while (i < p.length) {
          if p(i).startsWith("\"") then
            val v = Strings.substringBetween(p(i), "\"", "\"")
            properties.put(p(i - 1), v)
          else if (data.contains(p(i))) {
            properties.put(p(i - 1), data(p(i)))
          }
          i += 2
        }
        if (properties.contains("src")) {
          var src = properties("src")
          if (src.contains("base64,")) {
            src = Strings.substringAfter(src, "base64,")
          }
          r.setText("", 0)
          val width = toEmu(properties("width"))
          val height = toEmu(properties("height"))
          r.addPicture(new ByteArrayInputStream(Base64.decode(src)), PictureType.PNG, "esign.png", width, height)
        } else {
          r.setText("      ", 0)
        }
      }
    }
  }

  private def toEmu(num: String): Int = {
    if (num.endsWith("m")) {
      if (num.endsWith("mm")) {
        Strings.replace(num, "mm", "").toInt * Units.EMU_PER_CENTIMETER / 10
      } else if (num.endsWith("cm")) {
        Strings.replace(num, "cm", "").toInt * Units.EMU_PER_CENTIMETER
      } else {
        throw new RuntimeException(s"Cannot parse ${num} to emu")
      }
    } else {
      num.toInt * Units.EMU_PER_CENTIMETER
    }
  }

  private[docx] def replace(template: String, data: collection.Map[String, String]): String = {
    var text = template
    while (text.contains("${")) {
      val k = Strings.substringBetween(text, "${", "}").trim()
      val v = data.getOrElse(k, "")
      val begin = text.indexOf("${")
      val end = text.indexOf("}") + 1
      text = Strings.replace(text, text.substring(begin, end), v)
    }
    text
  }
}