aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/moe/nea/firmament/util/SBData.kt
blob: 348f048de9cad3e2486a5ba2a543bde9e46c2834 (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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/*
 * SPDX-FileCopyrightText: 2023 Linnea Gräf <nea@nea.moe>
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 */

package moe.nea.firmament.util

import kotlinx.serialization.SerializationException
import kotlinx.serialization.decodeFromString
import moe.nea.firmament.Firmament
import moe.nea.firmament.events.OutgoingPacketEvent
import moe.nea.firmament.events.ProcessChatEvent
import moe.nea.firmament.events.SkyblockServerUpdateEvent
import moe.nea.firmament.events.WorldReadyEvent
import net.minecraft.network.packet.c2s.play.CommandExecutionC2SPacket
import java.util.*
import kotlin.time.Duration
import kotlin.time.Duration.Companion.seconds

object SBData {
    private val profileRegex = "Profile ID: ([a-z0-9\\-]+)".toRegex()
    var profileId: UUID? = null

    private var lastLocrawSent = Timer()
    private val anyLocrawSent = Timer()
    private val locrawRoundtripTime: Duration = 5.seconds
    private var hasReceivedProfile = false
    private var hasSentLocraw = false
    var locraw: Locraw? = null
    val skyblockLocation: String? get() = locraw?.skyblockLocation
    val hasValidLocraw get() = locraw?.server !in listOf("limbo", null)

    fun init() {
        OutgoingPacketEvent.subscribe { event ->
            if (event.packet is CommandExecutionC2SPacket && event.packet.command == "locraw") {
                anyLocrawSent.markNow()
            }
        }
        ProcessChatEvent.subscribe(receivesCancelled = true) { event ->
            val profileMatch = profileRegex.matchEntire(event.unformattedString)
            if (profileMatch != null) {
                try {
                    profileId = UUID.fromString(profileMatch.groupValues[1])
                    hasReceivedProfile = true
                    if (!hasValidLocraw && !hasSentLocraw && anyLocrawSent.timePassed() > locrawRoundtripTime) {
                        sendLocraw()
                    }
                } catch (e: IllegalArgumentException) {
                    profileId = null
                    e.printStackTrace()
                }
            }
            if (event.unformattedString.startsWith("{")) {
                if (tryReceiveLocraw(event.unformattedString)) {
                    if (lastLocrawSent.timePassed() < locrawRoundtripTime) {
                        lastLocrawSent.markFarPast()
                        event.cancel()
                    }
                    if (!hasValidLocraw && !hasSentLocraw && hasReceivedProfile) {
                        sendLocraw()
                    }
                }
            }
        }

        WorldReadyEvent.subscribe {
            val lastLocraw = locraw
            locraw = null
            SkyblockServerUpdateEvent.publish(SkyblockServerUpdateEvent(lastLocraw, null))
            hasSentLocraw = false
            hasReceivedProfile = false
            profileId = null
        }
    }

    private fun tryReceiveLocraw(unformattedString: String, update: Boolean = true): Boolean = try {
        val lastLocraw = locraw
        val n = Firmament.json.decodeFromString<Locraw>(unformattedString)
        if (update) {
            if (n.gametype != "SKYBLOCK")
                profileId = null
            locraw = n
            SkyblockServerUpdateEvent.publish(SkyblockServerUpdateEvent(lastLocraw, locraw))
        }
        true
    } catch (e: SerializationException) {
        e.printStackTrace()
        false
    } catch (e: IllegalArgumentException) {
        e.printStackTrace()
        false
    }

    fun sendLocraw() {
        hasSentLocraw = true
        lastLocrawSent.markNow()
        val nh = MC.player?.networkHandler ?: return
        nh.sendChatCommand("locraw")
    }


}