fix auto updater not working with pre-releases

Jordyrat · Oct 3, 2024 · 5b175b0 · 5b175b0
1 parent 5f93621
commit 5b175b0
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/src/main/kotlin/com/ratons/config/features/About.java b/src/main/kotlin/com/ratons/config/features/About.java
@@ -1,8 +1,10 @@
 package com.ratons.config.features;
 
 import at.hannibal2.skyhanni.deps.moulconfig.annotations.ConfigEditorBoolean;
+import at.hannibal2.skyhanni.deps.moulconfig.annotations.ConfigEditorDropdown;
 import at.hannibal2.skyhanni.deps.moulconfig.annotations.ConfigOption;
 import com.google.gson.annotations.Expose;
+import com.ratons.features.misc.update.UpdateStream;
 
 public class About {
 
@@ -11,6 +13,11 @@ public class About {
     @ConfigEditorBoolean
     public boolean autoUpdates = true;
 
+    @Expose
+    @ConfigOption(name = "Update Stream", desc = "Select the update stream.")
+    @ConfigEditorDropdown
+    public UpdateStream updateStream = UpdateStream.FULL;
+
     @Expose
     @ConfigOption(name = "Debug", desc = "Developer setting for checking debug messages.")
     @ConfigEditorBoolean

diff --git a/src/main/kotlin/com/ratons/features/misc/update/UpdateManager.kt b/src/main/kotlin/com/ratons/features/misc/update/UpdateManager.kt
@@ -54,8 +54,8 @@ object UpdateManager {
         checkUpdate()
     }
 
-    fun checkUpdate() {
-        activePromise = context.checkUpdate("full")
+    fun checkUpdate(stream: String = config.updateStream.streamName) {
+        activePromise = context.checkUpdate(stream)
             .thenAcceptAsync({
                 potentialUpdate = it
                 if (it.isUpdateAvailable) {

diff --git a/src/main/kotlin/com/ratons/features/misc/update/UpdateStream.kt b/src/main/kotlin/com/ratons/features/misc/update/UpdateStream.kt
@@ -0,0 +1,11 @@
+package com.ratons.features.misc.update
+
+@Suppress("unused")
+enum class UpdateStream(private val displayName: String, val streamName: String = "") {
+    NONE("None"),
+    PRERELEASE("Pre-release", "pre"),
+    FULL("Full", "full"),
+    ;
+
+    override fun toString() = displayName
+}