Use ModState for action buttons

Author: DeathsGun

src/main/java/xyz/deathsgun/modmanager/ModManager.java

@@ -75,7 +75,13 @@ public static String getVersion() {
 
     public static ModState getState(SummarizedMod mod) {
         Optional<ModContainer> installedMod = FabricMods.getModContainerByMod(mod);
-        return installedMod.map(modContainer -> updateCheckService.isUpdateAvailable(mod, modContainer.getMetadata()) ? ModState.OUTDATED : ModState.INSTALLED).orElse(ModState.DOWNLOADABLE);
+        if (installedMod.isEmpty()) {
+            return getModManipulationManager().isInstalled(mod) ? ModState.INSTALLED : ModState.DOWNLOADABLE;
+        }
+        if (getModManipulationManager().isMarkedUninstalled(mod)) {
+            return ModState.DOWNLOADABLE;
+        }
+        return updateCheckService.isUpdateAvailable(mod, installedMod.get().getMetadata()) ? ModState.OUTDATED : ModState.INSTALLED;
     }
 
     @Override

src/main/java/xyz/deathsgun/modmanager/gui/ModDetailScreen.java

@@ -80,22 +80,25 @@ protected void init() {
     }
 
     private void handleActionClick(ButtonWidget buttonWidget) {
+        if (exception != null) {
+            buttonWidget.active = true;
+            MinecraftClient.getInstance().openScreen(new ModManagerErrorScreen(this, exception));
+            return;
+        }
         buttonWidget.active = false;
-        String key = ((TranslatableText) buttonWidget.getMessage()).getKey();
-        switch (key) {
-            case "modmanager.message.install" -> {
+        switch (ModManager.getState(summarizedMod)) {
+            case OUTDATED -> {
+                buttonWidget.setMessage(new TranslatableText("modmanager.message.updating"));
+                ModManager.getModManipulationManager().updateMod(summarizedMod, this::handleTaskResult);
+            }
+            case DOWNLOADABLE -> {
                 buttonWidget.setMessage(new TranslatableText("modmanager.message.installing"));
                 ModManager.getModManipulationManager().installMod(summarizedMod, this::handleTaskResult);
             }
-            case "modmanager.message.remove" -> {
+            case INSTALLED -> {
                 buttonWidget.setMessage(new TranslatableText("modmanager.message.removing"));
                 ModManager.getModManipulationManager().removeMod(summarizedMod, this::handleTaskResult);
             }
-            case "modmanager.message.showError" -> {
-                buttonWidget.active = true;
-                MinecraftClient.getInstance().openScreen(new ModManagerErrorScreen(this, exception));
-            }
-            case "modmanager.message.update" -> buttonWidget.setMessage(new TranslatableText("modmanager.message.updating"));
         }
     }
 
@@ -110,14 +113,11 @@ void updateActionButton() {
             actionButton.setMessage(new TranslatableText("modmanager.message.showError"));
             return;
         }
-        if (ModManager.getModManipulationManager().isInstalled(summarizedMod)) {
-            if (ModManager.getModManipulationManager().isMarkedUninstalled(summarizedMod)) {
-                actionButton.setMessage(new TranslatableText("modmanager.message.install"));
-            }
-            actionButton.setMessage(new TranslatableText("modmanager.message.remove"));
-            return;
+        switch (ModManager.getState(summarizedMod)) {
+            case DOWNLOADABLE -> actionButton.setMessage(new TranslatableText("modmanager.message.install"));
+            case INSTALLED -> actionButton.setMessage(new TranslatableText("modmanager.message.remove"));
+            case OUTDATED -> actionButton.setMessage(new TranslatableText("modmanager.message.update"));
         }
-        actionButton.setMessage(new TranslatableText("modmanager.message.install"));
     }
 
     @Override

src/main/java/xyz/deathsgun/modmanager/manager/ModManipulationManager.java

@@ -53,6 +53,10 @@ public boolean isInstalled(SummarizedMod mod) {
     }
 
     public boolean isMarkedUninstalled(SummarizedMod summarizedMod) {
-        return this.uninstalledMods.add(summarizedMod.id());
+        return this.uninstalledMods.contains(summarizedMod.id());
+    }
+
+    public void updateMod(SummarizedMod summarizedMod, TaskCallback taskCallback) {
+
     }
 }