1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2021 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.technical
21 
22 import scala.util.chaining.scalaUtilChainingOps
23 
24 final case class IpAndHash(ip: String, hash: String)
25 
26 object IpAndHash {
27   private val IPv6SafeSegments = 3
28   private val IPv6SegmentsCount = 8
29 
30   private def obfuscateIpV4(ip: String): String =
31     s"${ip.take(ip.lastIndexOf("."))}.x"
32 
33   @SuppressWarnings(Array("org.wartremover.warts.Throw"))
34   private val obfuscateIpV6: String => String =
35     _.split("::")
36       .map(_.split(':').filterNot(_.isEmpty))
37       .pipe({
38         case Array()     => Array[String]()
39         case Array(elem) => elem
40         case Array(left, right) =>
41           val numberOfSegments = left.length + right.length
42           val numberOfCompressedSegments = IPv6SegmentsCount - numberOfSegments
43           left ++ Array.fill(numberOfCompressedSegments)("0") ++ right
44         case _ => throw new Error("Invalid IPv6 format")
45       })
46       .take(IPv6SafeSegments)
47       .mkString(":")
48       .concat("::")
49 
50   def obfuscateIp(ip: String): String = ip match {
51     case ip if ip.contains(":") => obfuscateIpV6(ip)
52     case _                      => obfuscateIpV4(ip)
53   }
54 }