before removing label tweaks

Author: mmontin

src/Cooked/Attack/DatumHijacking.hs

@@ -140,5 +140,5 @@ datumHijackingAttack (DatumHijackingParams branching mChange select) = do
         simple
         (\out -> (\owner -> set txSkelOutOwnerL (toPKHOrVScript owner) out) <$> mChange out)
         select
-  addLabelTweak $ DatumHijackingLabel modified
+  insertInTweak txSkelLabelsL $ TxSkelLabel $ DatumHijackingLabel modified
   return modified

src/Cooked/Attack/DatumTampering.hs

@@ -97,5 +97,5 @@ datumTamperingAttack DatumTamperingParams {..} = do
   modified <-
     modifyTweakFromParams $
       ModifyTweakParams tdpBranching tdpOptic txSkelOutDatumTypedAT tdpModification tdpIndexPred
-  addLabelTweak $ DatumTamperingLabel modified
+  insertInTweak txSkelLabelsL $ TxSkelLabel $ DatumTamperingLabel modified
   return modified

src/Cooked/Attack/OutputsReordering.hs

@@ -14,7 +14,7 @@ import Control.Monad
 import Cooked.Pretty.Class
 import Cooked.Skeleton
 import Cooked.Tweak.Common
-import Cooked.Tweak.Labels
+import Cooked.Tweak.Insertion
 import Data.List (permutations)
 import Polysemy (Members, Sem)
 import Polysemy.NonDet
@@ -65,7 +65,7 @@ outputsReorderingAttack params = do
           Shuffle -> permutations outputs
           ManualReordering f -> f outputs
           _ -> []
-  addLabelTweak OutputsReorderingLabel
+  insertInTweak txSkelLabelsL $ TxSkelLabel OutputsReorderingLabel
   where
     valid :: Int -> Int -> Bool
     valid i iMax = i >= 0 && i < iMax

src/Cooked/Attack/PeerTampering.hs

@@ -20,7 +20,7 @@ import Control.Monad
 import Cooked.Pretty.Class
 import Cooked.Skeleton
 import Cooked.Tweak.Common
-import Cooked.Tweak.Labels
+import Cooked.Tweak.Insertion
 import Cooked.Tweak.Modification
 import Optics.Core
 import Plutus.Script.Utils.Address qualified as Script
@@ -98,5 +98,5 @@ peerTamperingAttack (PeerTamperingParams branching mChanges) = do
         branching
         ((txSkelAllocatedPeersT % userPubKeyHashI) `adjoin` (txSkelRedeemedPeersT % userPubKeyHashI))
       $ \pkh -> if pkh == existing then targets else []
-  addLabelTweak $ PeerTamperingLabel modified
+  insertInTweak txSkelLabelsL $ TxSkelLabel $ PeerTamperingLabel modified
   return modified

src/Cooked/Attack/RedeemerTampering.hs

@@ -127,5 +127,5 @@ redeemerTamperingAttack RedeemerTamperingParams {..} = do
   modified <-
     modifyTweakFromParams $
       ModifyTweakParams trpBranching trpOptic txSkelRedeemerTypedAT trpModification trpIndexPred
-  addLabelTweak $ RedeemerTamperingLabel modified
+  insertInTweak txSkelLabelsL $ TxSkelLabel $ RedeemerTamperingLabel modified
   return modified

src/Cooked/Attack/TokenDuplication.hs

@@ -21,7 +21,7 @@ import Control.Monad
 import Cooked.Pretty.Class
 import Cooked.Skeleton
 import Cooked.Tweak.Common
-import Cooked.Tweak.Labels
+import Cooked.Tweak.Insertion
 import Cooked.Tweak.Mint
 import Cooked.Tweak.Outputs
 import Optics.Core
@@ -121,6 +121,6 @@ tokenDuplicationAttack TokenDuplicationParams {..} = do
   -- We redirect the extra value to an attacker
   addOutputTweak $ atpThief `receives` Value totalIncrement
   -- We label the transaction by the added tokens
-  addLabelTweak $ TokenDuplicationLabel totalIncrement
+  insertInTweak txSkelLabelsL $ TxSkelLabel $ TokenDuplicationLabel totalIncrement
   -- We return the added tokens
   return totalIncrement

src/Cooked/Attack/ValidityTampering.hs

@@ -128,5 +128,5 @@ validityTamperingAttack (ValidityTamperingParams optics change) = do
         OneBranchForAllFoci
         (txSkelValidityRangeL % castOptic @A_Traversal optics)
         change
-  addLabelTweak $ ValidityTamperingLabel currentValidityRange
+  insertInTweak txSkelLabelsL $ TxSkelLabel $ ValidityTamperingLabel currentValidityRange
   return currentValidityRange

src/Cooked/Tweak/Insertion.hs

@@ -83,7 +83,7 @@ insertInTweak ::
   a ->
   Sem effs ()
 insertInTweak (castOptic @A_Traversal -> optic) a = do
-  guardPredTweak optic $ view $ contains a
+  guardTweak $ optic % contains a % filtered not
   setTweak (optic % contains a) True
 
 -- * Inserting elements in maps
@@ -100,5 +100,5 @@ insertAtTweak ::
   v ->
   Sem effs ()
 insertAtTweak (castOptic @A_Traversal -> optic) k v = do
-  guardTweak $ optic % at k
+  guardTweak $ optic % at k % _Nothing
   setTweak (optic % at k) (Just v)