1 files changed, 55 insertions, 0 deletions
diff --git a/src/main/kotlin/util/regex.kt b/src/main/kotlin/util/regex.kt
new file mode 100644
index 0000000..3ce5bd8
--- /dev/null
+++ b/src/main/kotlin/util/regex.kt
@@ -0,0 +1,55 @@
+
+
+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)
+
+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
+}