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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
|
package at.hannibal2.skyhanni.features.misc.update
import at.hannibal2.skyhanni.SkyHanniMod
import at.hannibal2.skyhanni.config.features.About.UpdateStream
import at.hannibal2.skyhanni.events.ConfigLoadEvent
import at.hannibal2.skyhanni.events.LorenzTickEvent
import at.hannibal2.skyhanni.skyhannimodule.SkyHanniModule
import at.hannibal2.skyhanni.utils.ChatUtils
import at.hannibal2.skyhanni.utils.ConditionalUtils.onToggle
import at.hannibal2.skyhanni.utils.DelayedRun
import at.hannibal2.skyhanni.utils.LorenzLogger
import com.google.gson.JsonElement
import io.github.moulberry.notenoughupdates.util.ApiUtil
import io.github.notenoughupdates.moulconfig.observer.Property
import io.github.notenoughupdates.moulconfig.processor.MoulConfigProcessor
import moe.nea.libautoupdate.CurrentVersion
import moe.nea.libautoupdate.PotentialUpdate
import moe.nea.libautoupdate.UpdateContext
import moe.nea.libautoupdate.UpdateSource
import moe.nea.libautoupdate.UpdateTarget
import moe.nea.libautoupdate.UpdateUtils
import net.minecraft.client.Minecraft
import net.minecraftforge.common.MinecraftForge
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent
import java.util.concurrent.CompletableFuture
import javax.net.ssl.HttpsURLConnection
@SkyHanniModule
object UpdateManager {
private val logger = LorenzLogger("update_manager")
private var _activePromise: CompletableFuture<*>? = null
private var activePromise: CompletableFuture<*>?
get() = _activePromise
set(value) {
_activePromise?.cancel(true)
_activePromise = value
}
var updateState: UpdateState = UpdateState.NONE
private set
fun getNextVersion(): String? {
return potentialUpdate?.update?.versionNumber?.asString
}
@SubscribeEvent
fun onConfigLoad(event: ConfigLoadEvent) {
SkyHanniMod.feature.about.updateStream.onToggle {
reset()
}
}
@SubscribeEvent
fun onTick(event: LorenzTickEvent) {
Minecraft.getMinecraft().thePlayer ?: return
MinecraftForge.EVENT_BUS.unregister(this)
if (config.autoUpdates || config.fullAutoUpdates)
checkUpdate()
}
fun injectConfigProcessor(processor: MoulConfigProcessor<*>) {
processor.registerConfigEditor(ConfigVersionDisplay::class.java) { option, _ ->
GuiOptionEditorUpdateCheck(option)
}
}
fun isCurrentlyBeta(): Boolean {
return SkyHanniMod.version.contains("beta", ignoreCase = true)
}
private val config get() = SkyHanniMod.feature.about
fun reset() {
updateState = UpdateState.NONE
_activePromise = null
potentialUpdate = null
logger.log("Reset update state")
}
fun checkUpdate(forceDownload: Boolean = false, forcedUpdateStream: UpdateStream = config.updateStream.get()) {
var updateStream = forcedUpdateStream
if (updateState != UpdateState.NONE) {
logger.log("Trying to perform update check while another update is already in progress")
return
}
logger.log("Starting update check")
val currentStream = config.updateStream.get()
if (currentStream != UpdateStream.BETA && (updateStream == UpdateStream.BETA || isCurrentlyBeta())) {
config.updateStream = Property.of(UpdateStream.BETA)
updateStream = UpdateStream.BETA
}
activePromise = context.checkUpdate(updateStream.stream)
.thenAcceptAsync({
logger.log("Update check completed")
if (updateState != UpdateState.NONE) {
logger.log("This appears to be the second update check. Ignoring this one")
return@thenAcceptAsync
}
potentialUpdate = it
if (it.isUpdateAvailable) {
updateState = UpdateState.AVAILABLE
if (config.fullAutoUpdates || forceDownload) {
ChatUtils.chat("§aSkyHanni found a new update: ${it.update.versionName}, starting to download now.")
queueUpdate()
} else if (config.autoUpdates) {
ChatUtils.chatAndOpenConfig(
"§aSkyHanni found a new update: ${it.update.versionName}. " +
"Check §b/sh download update §afor more info.",
config::autoUpdates
)
}
} else if (forceDownload) {
ChatUtils.chat("§aSkyHanni didn't find a new update.")
}
}, DelayedRun.onThread)
}
fun queueUpdate() {
if (updateState != UpdateState.AVAILABLE) {
logger.log("Trying to enqueue an update while another one is already downloaded or none is present")
}
updateState = UpdateState.QUEUED
activePromise = CompletableFuture.supplyAsync {
logger.log("Update download started")
potentialUpdate!!.prepareUpdate()
}.thenAcceptAsync({
logger.log("Update download completed, setting exit hook")
updateState = UpdateState.DOWNLOADED
potentialUpdate!!.executePreparedUpdate()
ChatUtils.chat("Download of update complete. ")
ChatUtils.chat("§aThe update will be installed after your next restart.")
}, DelayedRun.onThread)
}
private val context = UpdateContext(
UpdateSource.githubUpdateSource("hannibal002", "SkyHanni"),
UpdateTarget.deleteAndSaveInTheSameFolder(UpdateManager::class.java),
object : CurrentVersion {
val normalDelegate = CurrentVersion.ofTag(SkyHanniMod.version)
override fun display(): String {
if (SkyHanniMod.feature.dev.debug.alwaysOutdated)
return "Force Outdated"
return normalDelegate.display()
}
override fun isOlderThan(element: JsonElement): Boolean {
if (SkyHanniMod.feature.dev.debug.alwaysOutdated)
return true
return normalDelegate.isOlderThan(element)
}
override fun toString(): String {
return "ForceOutdateDelegate($normalDelegate)"
}
},
SkyHanniMod.MODID,
)
init {
context.cleanup()
UpdateUtils.patchConnection {
if (it is HttpsURLConnection) {
ApiUtil.patchHttpsRequest(it)
}
}
}
enum class UpdateState {
AVAILABLE,
QUEUED,
DOWNLOADED,
NONE
}
private var potentialUpdate: PotentialUpdate? = null
}
|
