aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/moe/nea/firmament/util/regex.kt
blob: 9de2b36c30de29e46144590457a123233c6dc1ba (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/*
 * 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
 */

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
}