1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api
21 import akka.actor.typed.ActorSystem
22 import akka.serialization.{Serialization, SerializationExtension}
23 
24 import java.nio.ByteOrder
25 import stamina.ByteString
26 
27 object StaminaTestUtils {
28 
29 //  copied from org.apache.logging.log4j.core.config.plugins.convert.HexConverter
30   @SuppressWarnings(Array("org.wartremover.warts.While"))
31   private def parseHexBinary(s: String): Array[Byte] = {
32     val len = s.length
33     val data = new Array[Byte](len / 2)
34     var i = 0
35     while ({ i < len }) {
36       data(i / 2) = ((Character.digit(s.charAt(i), 16) << 4) + Character.digit(s.charAt(i + 1), 16)).toByte
37       i += 2
38     }
39     data
40   }
41 
42   private val initialCharactersToSkip: Int = 2
43 
44   @SuppressWarnings(Array("org.wartremover.warts.TryPartial"))
45   private def makeEventSerializer(implicit system: ActorSystem[_]) =
46     SerializationExtension(system)
47       .serializerOf(new Serialization.Settings(system.settings.config).Serializers("make-serializer"))
48       .get
49 
50   @SuppressWarnings(Array("org.wartremover.warts.AsInstanceOf"))
51   def deserializeEventFromJson[A](eventKey: String, eventAsJsonString: String, version: Int = 1)(
52     implicit system: ActorSystem[_]
53   ): A = {
54     implicit val byteOrder: ByteOrder = java.nio.ByteOrder.LITTLE_ENDIAN
55     val bytes: Array[Byte] = ByteString.newBuilder
56       .putInt(eventKey.length)
57       .putBytes(eventKey.getBytes("UTF-8"))
58       .putInt(version)
59       .append(ByteString(eventAsJsonString))
60       .result()
61       .toArray
62 
63     makeEventSerializer.fromBinary(bytes).asInstanceOf[A]
64   }
65 
66   @SuppressWarnings(Array("org.wartremover.warts.AsInstanceOf"))
67   def deserializeEventFromHexa[A](serialized: String)(implicit system: ActorSystem[_]): A = {
68     val bytes: Array[Byte] = parseHexBinary(serialized.substring(initialCharactersToSkip))
69 
70     makeEventSerializer.fromBinary(bytes).asInstanceOf[A]
71   }
72 
73   def getVersionFromHexa(serialized: String): Int = {
74     val bytes: Array[Byte] = parseHexBinary(serialized.substring(initialCharactersToSkip))
75     val start = bytes.take(4)
76     val skip = start.map(_.toInt).sum
77 
78     bytes.slice(skip + 4, skip + 8).map(_.toInt).sum
79   }
80 
81   def getEventNameFromHexa(serialized: String): String = {
82     val bytes: Array[Byte] = parseHexBinary(serialized.substring(initialCharactersToSkip))
83     val start = bytes.take(4)
84     val skip = start.map(_.toInt).sum
85     new String(bytes.slice(4, skip + 4), "UTF-8")
86   }
87 
88 }