net.adamcin.vltpack.ChecksumCalculator.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vltpack-maven-plugin Show documentation

Maven plugin defining the vltpack packaging lifecycle and appropriate goals for use in creating application and content packages targeting the Adobe CRX / AEM / Granite platforms

There is a newer version: 1.3.0

Show newest version

package net.adamcin.vltpack

import java.io.File
import java.security.MessageDigest
import scalax.io.Resource

class ChecksumCalculator {
  final val md = MessageDigest.getInstance("SHA-1")
  final val nullBytes = "null".getBytes("UTF-8")

  private def update(item: Array[Byte]): ChecksumCalculator = {
    md.update(item)
    this
  }

  def addNull(): ChecksumCalculator = {
    md.update(nullBytes)
    this
  }

  def add(item: AnyVal): ChecksumCalculator = {
    Option(item) match {
      case Some(v) => add(v.toString)
      case None => addNull()
    }
  }

  def add(item: String): ChecksumCalculator = {
    Option(item) match {
      case Some(s) => update(s.getBytes("UTF-8"))
      case None => addNull()
    }
  }

  def add(item: File): ChecksumCalculator = {
    Option(item) match {
      case Some(file) => add(file.getAbsolutePath)
      case None => addNull()
    }
  }

  def addContents(item: File): ChecksumCalculator = {
    Option(item) match {
      case Some(file) => {
        if (file.exists()) {
          Resource.fromFile(file).bytes.sliding(1024).foreach((bytes) => update(bytes.toArray))
          this
        } else {
          addNull()
        }
      }
      case None => addNull()
    }
  }

  def add(item: Map[String, String]): ChecksumCalculator = {
    Option(item) match {
      case Some(map) => add(map.toString)
      case None => addNull()
    }
  }

  def calculate(): String = md.digest().map(0xFF & _).map { "%02x".format(_) }.mkString("")
}