From be2416af8ad65a92d0565370aaf1c424f39e71fc Mon Sep 17 00:00:00 2001
From: Darkborderman <reastw1234@gmail.com>
Date: Fri, 13 Sep 2024 18:37:19 +0800
Subject: [PATCH] Update

---
 .../debris/StarSystemDesignDebrisScript.java  |  4 +--
 .../entity/StarSystemDesignEntityScript.java  | 34 +++++++++++++++++++
 2 files changed, 36 insertions(+), 2 deletions(-)

diff --git a/src/mod/starsystemdesign/rulecmd/debris/StarSystemDesignDebrisScript.java b/src/mod/starsystemdesign/rulecmd/debris/StarSystemDesignDebrisScript.java
index 71e733c..50c4b27 100644
--- a/src/mod/starsystemdesign/rulecmd/debris/StarSystemDesignDebrisScript.java
+++ b/src/mod/starsystemdesign/rulecmd/debris/StarSystemDesignDebrisScript.java
@@ -44,6 +44,7 @@ public boolean execute(String ruleId, InteractionDialogAPI dialog, List<Misc.Tok
             } else if(terrain.getPlugin().getClass() == NebulaTerrainPlugin.class){
                 opts.setEnabled("StarSystemDesignDebrisRemoveOption", true);
             } // AsteroidBelt and other ring will not removed because of its sprite will still exist
+
             opts.addOption("Back", "StarSystemDesignDebrisBackOption");
             opts.setShortcut("StarSystemDesignDebrisBackOption", Keyboard.KEY_ESCAPE, false, false, false, false);
         }else{
@@ -53,12 +54,11 @@ public boolean execute(String ruleId, InteractionDialogAPI dialog, List<Misc.Tok
                     if(terrain.getPlugin().getClass() == DebrisFieldTerrainPlugin.class){
                         DebrisFieldTerrainPlugin debris = (DebrisFieldTerrainPlugin)terrain.getPlugin();
                         debris.getParams().lastsDays = 1f;
-                        dialog.dismiss();
                     }
                     else {
                         Global.getSector().getPlayerFleet().getStarSystem().removeEntity(target);
-                        dialog.dismiss();
                     }
+                    dialog.dismiss();
                 }
             }
         }
diff --git a/src/mod/starsystemdesign/rulecmd/entity/StarSystemDesignEntityScript.java b/src/mod/starsystemdesign/rulecmd/entity/StarSystemDesignEntityScript.java
index 15e5d6d..f251fd2 100644
--- a/src/mod/starsystemdesign/rulecmd/entity/StarSystemDesignEntityScript.java
+++ b/src/mod/starsystemdesign/rulecmd/entity/StarSystemDesignEntityScript.java
@@ -4,12 +4,17 @@
 import java.util.Map;
 
 import org.apache.log4j.Logger;
+import org.lwjgl.input.Keyboard;
 
 import com.fs.starfarer.api.Global;
 import com.fs.starfarer.api.campaign.InteractionDialogAPI;
+import com.fs.starfarer.api.campaign.OptionPanelAPI;
+import com.fs.starfarer.api.campaign.SectorEntityToken;
 import com.fs.starfarer.api.campaign.rules.MemoryAPI;
 import com.fs.starfarer.api.impl.campaign.rulecmd.BaseCommandPlugin;
+import com.fs.starfarer.api.impl.campaign.terrain.DebrisFieldTerrainPlugin;
 import com.fs.starfarer.api.util.Misc;
+import com.fs.starfarer.campaign.CampaignTerrain;
 
 import mod.starsystemdesign.rulecmd.debris.StarSystemDesignDebrisScript;
 
@@ -18,6 +23,35 @@ public class StarSystemDesignEntityScript extends BaseCommandPlugin{
 
     @Override
     public boolean execute(String ruleId, InteractionDialogAPI dialog, List<Misc.Token> params, Map<String, MemoryAPI> memoryMap) {
+        OptionPanelAPI opts = dialog.getOptionPanel();
+        opts.clearOptions();
+
+        String arg = null;
+        SectorEntityToken target = dialog.getInteractionTarget().getOrbitFocus();
+        try{
+            arg = params.get(0).getString(memoryMap);
+        }catch(IndexOutOfBoundsException e){}
+
+        if(arg == null){
+            opts.addOption("Remove object", "StarSystemDesignDebrisRemoveOption");
+            opts.setEnabled("StarSystemDesignDebrisRemoveOption", false);
+            opts.addOption("Back", "StarSystemDesignDebrisRemoveOption");
+            opts.setShortcut("StarSystemDesignDebrisRemoveOption", Keyboard.KEY_ESCAPE, false, false, false, false);
+        }else{
+            switch(arg){
+                case "remove":{
+                    CampaignTerrain terrain = (CampaignTerrain) target;
+                    if(terrain.getPlugin().getClass() == DebrisFieldTerrainPlugin.class){
+                        DebrisFieldTerrainPlugin debris = (DebrisFieldTerrainPlugin)terrain.getPlugin();
+                        debris.getParams().lastsDays = 1f;
+                    }
+                    else {
+                        Global.getSector().getPlayerFleet().getStarSystem().removeEntity(target);
+                    }
+                    dialog.dismiss();
+                }
+            }
+        }
         return true;
     }
 }