feat(BuyCommand): Implement IPredictable

Computerdores · Apr 19, 2024 · ab7917c · ab7917c
1 parent e34ba6f
commit ab7917c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/AdvancedTerminalAPI/AdvancedTerminalAPI/Vanillin/Commands/BuyCommand.cs b/AdvancedTerminalAPI/AdvancedTerminalAPI/Vanillin/Commands/BuyCommand.cs
@@ -4,13 +4,19 @@
 
 namespace Computerdores.AdvancedTerminalAPI.Vanillin.Commands; 
 
-public class BuyCommand : ICommand, IAliasable {
+public class BuyCommand : ICommand, IAliasable, IPredictable {
     private bool _awaitingConfirmation;
 
     private ICommand _command;
-    
+
     public string GetName() => "buy";
 
+    public string PredictInput(string partialInput, ITerminal terminal) {
+        return _awaitingConfirmation
+            ? Util.PredictConfirmation(partialInput)
+            : FromPlayerInput(terminal.GetDriver().VanillaTerminal, partialInput).GetName();
+    }
+
     public CommandResult Execute(string input, ITerminal terminal) {
         if (_awaitingConfirmation) return _command.Execute(input, terminal);