1 files changed, 42 insertions, 0 deletions
diff --git a/src/main/kotlin/moe/nea/firmament/util/regex.kt b/src/main/kotlin/moe/nea/firmament/util/regex.kt
index 4cc3f03..9de2b36 100644
--- a/src/main/kotlin/moe/nea/firmament/util/regex.kt
+++ b/src/main/kotlin/moe/nea/firmament/util/regex.kt
@@ -1,5 +1,6 @@
 /*
  * SPDX-FileCopyrightText: 2023 Linnea Gräf <nea@nea.moe>
+ * SPDX-FileCopyrightText: 2024 Linnea Gräf <nea@nea.moe>
  *
  * SPDX-License-Identifier: GPL-3.0-or-later
  */
@@ -8,6 +9,10 @@ package moe.nea.firmament.util
 
 import java.util.regex.Matcher
 import java.util.regex.Pattern
+import org.intellij.lang.annotations.Language
+import kotlin.time.Duration
+import kotlin.time.Duration.Companion.minutes
+import kotlin.time.Duration.Companion.seconds
 
 inline fun <T> String.ifMatches(regex: Regex, block: (MatchResult) -> T): T? =
     regex.matchEntire(this)?.let(block)
@@ -16,3 +21,40 @@ inline fun <T> Pattern.useMatch(string: String, block: Matcher.() -> T): T? =
     matcher(string)
         .takeIf(Matcher::matches)
         ?.let(block)
+
+@Language("RegExp")
+val TIME_PATTERN = "[0-9]+[ms]"
+
+@Language("RegExp")
+val SHORT_NUMBER_FORMAT = "[0-9]+(?:,[0-9]+)*(?:\\.[0-9]+)?[kKmMbB]?"
+
+
+val siScalars = mapOf(
+    'k' to 1_000.0,
+    'K' to 1_000.0,
+    'm' to 1_000_000.0,
+    'M' to 1_000_000.0,
+    'b' to 1_000_000_000.0,
+    'B' to 1_000_000_000.0,
+)
+
+fun parseTimePattern(text: String): Duration {
+    val length = text.dropLast(1).toInt()
+    return when (text.last()) {
+        'm' -> length.minutes
+        's' -> length.seconds
+        else -> error("Invalid pattern for time $text")
+    }
+}
+
+fun parseShortNumber(string: String): Double {
+    var k = string.replace(",", "")
+    val scalar = k.last()
+    var scalarMultiplier = siScalars[scalar]
+    if (scalarMultiplier == null) {
+        scalarMultiplier = 1.0
+    } else {
+        k = k.dropLast(1)
+    }
+    return k.toDouble() * scalarMultiplier
+}