Introduced doneShort(), doneInt() and doneLong()

Author: catap

CHANGELOG.md

@@ -3,6 +3,7 @@
 All notable changes to this project will be documented in this file.
 
 ## [unreleased]
+- Introduced `update(Short)`, `update(Int)` and `update(Long)`.
 
 ## [2.2.0] - 2021-01-05
 - Introduced `doneShort()`, `doneInt()` and `doneLong()`.

shared/src/main/scala/ky/korins/blake3/Hasher.scala

@@ -29,6 +29,21 @@ trait Hasher {
    */
   def update(input: Byte): Hasher
 
+  /**
+   * Updates a hasher by specified short, returns the same hasher
+   */
+  def update(input: Short): Hasher
+
+  /**
+   * Updates a hasher by specified short, returns the same hasher
+   */
+  def update(input: Int): Hasher
+
+  /**
+   * Updates a hasher by specified short, returns the same hasher
+   */
+  def update(input: Long): Hasher
+
   /**
    * Updates a hasher by specified string, returns the same hasher
    */

shared/src/main/scala/ky/korins/blake3/HasherImpl.scala

@@ -95,6 +95,45 @@ private[blake3] class HasherImpl(
     this
   }
 
+  // Simplified version of update(Array[Byte])
+  def update(input: Short): Hasher = synchronized {
+    var v = input
+    var i = 0
+    while (i < 2) {
+      finalizeWhenCompleted()
+      chunkState.update(v.toByte)
+      v = (v >> 8).toShort
+      i += 1
+    }
+    this
+  }
+
+  // Simplified version of update(Array[Byte])
+  def update(input: Int): Hasher = synchronized {
+    var v = input
+    var i = 0
+    while (i < 4) {
+      finalizeWhenCompleted()
+      chunkState.update(v.toByte)
+      v >>= 8
+      i += 1
+    }
+    this
+  }
+
+  // Simplified version of update(Array[Byte])
+  def update(input: Long): Hasher = synchronized {
+    var v = input
+    var i = 0
+    while (i < 8) {
+      finalizeWhenCompleted()
+      chunkState.update(v.toByte)
+      v >>= 8
+      i += 1
+    }
+    this
+  }
+
   def update(input: InputStream): Hasher = synchronized {
     val bytes = new Array[Byte](CHUNK_LEN)
 

shared/src/test/scala/ky/korins/blake3/Blake3Test.scala

@@ -102,6 +102,59 @@ class Blake3Test extends AnyWordSpec with should.Matchers {
     Blake3.hex(key, 42) shouldBe Blake3.hex(key.getBytes, 42)
   }
 
+  "Short works as Byte" in {
+    val shortHash = Blake3
+      .newHasher()
+      .update(0x6af6.toShort)
+      .doneLong()
+
+    val byteHash = Blake3
+      .newHasher()
+      .update(0xf6.toByte)
+      .update(0x6a.toByte)
+      .doneLong()
+
+    shortHash shouldBe byteHash
+  }
+
+  "Int works as Byte" in {
+    val shortHash = Blake3
+      .newHasher()
+      .update(0x7eaa6af6)
+      .doneLong()
+
+    val byteHash = Blake3
+      .newHasher()
+      .update(0xf6.toByte)
+      .update(0x6a.toByte)
+      .update(0xaa.toByte)
+      .update(0x7e.toByte)
+      .doneLong()
+
+    shortHash shouldBe byteHash
+  }
+
+  "Long works as Byte" in {
+    val shortHash = Blake3
+      .newHasher()
+      .update(0x462d22e57eaa6af6L)
+      .doneLong()
+
+    val byteHash = Blake3
+      .newHasher()
+      .update(0xf6.toByte)
+      .update(0x6a.toByte)
+      .update(0xaa.toByte)
+      .update(0x7e.toByte)
+      .update(0xe5.toByte)
+      .update(0x22.toByte)
+      .update(0x2d.toByte)
+      .update(0x46.toByte)
+      .doneLong()
+
+    shortHash shouldBe byteHash
+  }
+
   "Output encoding works" in {
     val input = "some text"