From 8b52e2ef3d69dcfdce4435e6ca04103a07bbb677 Mon Sep 17 00:00:00 2001
From: parax <incubusnb@gmail.com>
Date: Fri, 18 Mar 2022 20:30:54 -0600
Subject: [PATCH] Fixed: purchaseSleeveAug checks shock value

---
 src/NetscriptFunctions/Sleeve.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/NetscriptFunctions/Sleeve.ts b/src/NetscriptFunctions/Sleeve.ts
index 3b37dcb39..f3aaafa8d 100644
--- a/src/NetscriptFunctions/Sleeve.ts
+++ b/src/NetscriptFunctions/Sleeve.ts
@@ -288,7 +288,11 @@ export function NetscriptSleeve(player: IPlayer, workerScript: WorkerScript, hel
       helper.updateDynamicRam("purchaseSleeveAug", getRamCost(player, "sleeve", "purchaseSleeveAug"));
       checkSleeveAPIAccess("purchaseSleeveAug");
       checkSleeveNumber("purchaseSleeveAug", sleeveNumber);
-
+      
+      if (player.sleeves[sleeveNumber].shock > 0){
+        throw helper.makeRuntimeErrorMsg("sleeve.purchaseSleeveAug", `Sleeve shock too high: Sleeve ${sleeveNumber}`);
+      }
+      
       const aug = Augmentations[augName];
       if (!aug) {
         throw helper.makeRuntimeErrorMsg("sleeve.purchaseSleeveAug", `Invalid aug: ${augName}`);