{- |
Module     : Spec.Effect.TreasuryWithdrawalEffect
Maintainer : seungheon.ooh@gmail.com
Description: Sample based testing for Treasury Withdrawal Effect

This module tests the Treasury Withdrawal Effect.
-}
module Spec.Effect.TreasuryWithdrawalEffect (currSymbol, signer, validator, validatorHashTN, scriptContext1, tests) where

import Plutarch.Evaluate
import Plutarch.Builtin
import Plutarch.Api.V1
import Plutus.V1.Ledger.Api
import Plutus.V1.Ledger.Interval qualified as Interval
import Plutus.V1.Ledger.Value qualified as Value

import Data.ByteString.Hash

import Agora.Effect.TreasuryWithdrawal
import Agora.AuthorityToken

import Spec.Util

import Test.Tasty

-- receiverList :: TreasuryWithdrawalDatum
-- receiverList = TreasuryWithdrawalDatum [(mempty, mempty)]

-- | A sample Currency Symbol.
currSymbol :: CurrencySymbol
currSymbol = CurrencySymbol "12312099"

gtSymbol :: CurrencySymbol
gtSymbol = CurrencySymbol "abb"

gtToken :: TokenName
gtToken = TokenName "hey"

-- | A sample 'PubKeyHash'.
signer :: PubKeyHash
signer = "8a30896c4fd5e79843e4ca1bd2cdbaa36f8c0bc3be7401214142019c"

-- | List of users who the effect will pay to.
users :: [Credential]
users =
  PubKeyCredential . PubKeyHash . toBuiltin . sha2
    <$> [ "Orange"
        , "Bottle"
        , "Hello"
        ]

-- | List of users who the effect will pay to.
treasuries :: [Credential]
treasuries =
  ScriptCredential . ValidatorHash . toBuiltin . sha2
    <$> [ "1234"
        , "qwer"
        , "asdf"
        ]

_aa :: [Credential]
_aa = treasuries

-- | Datum for Treasury Withdrawal Effect Validator.
datum :: TreasuryWithdrawalDatum
datum =
  TreasuryWithdrawalDatum
    [ (users !! 0, Value.singleton gtSymbol gtToken 1)
    , (users !! 1, Value.singleton gtSymbol gtToken 1)
    , (users !! 2, Value.singleton gtSymbol gtToken 1)
    ]

-- | Effect validator instance.
validator :: Validator
validator = mkValidator $ treasuryWithdrawalValidator currSymbol

-- | 'TokenName' that represents the hash of the 'Stake' validator.
validatorHashTN :: TokenName
validatorHashTN = let ValidatorHash vh = validatorHash validator in TokenName vh

scriptContext1 :: ScriptContext
scriptContext1 =
  ScriptContext
    { scriptContextTxInfo =
        TxInfo
          { txInfoInputs = 
              [ TxInInfo -- Initiator
                  (TxOutRef "0b2086cbf8b6900f8cb65e012de4516cb66b5cb08a9aaba12a8b88be" 1)
                  TxOut
                    { txOutAddress = Address (ScriptCredential $ validatorHash validator) Nothing
                    , txOutValue = Value.singleton currSymbol validatorHashTN 1 -- Stake ST
                    , txOutDatumHash = Just (DatumHash "")
                    }
              -- , TxInInfo -- Treasury 1
              --     (TxOutRef "0b121212121212121212121212121212121212121212121212121221" 2)
              --     TxOut
              --       { txOutAddress = Address (treasuries !! 0) Nothing
              --       , txOutValue = Value.singleton gtSymbol gtToken 10
              --       , txOutDatumHash = Just (DatumHash "")
              --       }
              -- , TxInInfo -- Treasury 2
              --     (TxOutRef "0b121212121212121212a41212121212121212121212121212121221" 3)
              --     TxOut
              --       { txOutAddress = Address (treasuries !! 1) Nothing
              --       , txOutValue = Value.singleton "1234ab" "LQLQLQL" 10
              --       , txOutDatumHash = Just (DatumHash "")
              --       }
              ]
          , txInfoOutputs = 
              [ TxOut
                  { txOutAddress = Address (users !! 0) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 1
                  , txOutDatumHash = Nothing
                  }
              , TxOut
                  { txOutAddress = Address (users !! 1) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 1
                  , txOutDatumHash = Nothing
                  }
              , TxOut
                  { txOutAddress = Address (users !! 2) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 1
                  , txOutDatumHash = Nothing
                  }
              -- Send left overs to treasuries
              , TxOut
                  { txOutAddress = Address (treasuries !! 0) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 7
                  , txOutDatumHash = Nothing
                  }
              , TxOut
                  { txOutAddress = Address (treasuries !! 1) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 10
                  , txOutDatumHash = Nothing
                  }                  
              ]
          , txInfoFee = Value.singleton "" "" 2
          , txInfoMint = Value.singleton currSymbol validatorHashTN (-1)
          , txInfoDCert = []
          , txInfoWdrl = []
          , txInfoValidRange = Interval.always
          , txInfoSignatories = [signer]
          , txInfoData = []
          , txInfoId = "0b2086cbf8b6900f8cb65e012de4516cb66b5cb08a9aaba12a8b88be"
          }
    , scriptContextPurpose =
        Spending (TxOutRef "0b2086cbf8b6900f8cb65e012de4516cb66b5cb08a9aaba12a8b88be" 1)
    }


tests :: [TestTree]
tests =
  [ testGroup
    "effect"
    [ effectFailsWith "test1" (treasuryWithdrawalValidator currSymbol) datum scriptContext1]
  ]

_asdfa :: IO ()
_asdfa = do
    let (res, _budget, traces) = evalScript $ compile ((treasuryWithdrawalValidator currSymbol) # pforgetData (pconstantData datum) # pforgetData (pconstantData ()) # pconstant scriptContext1)
    case res of
      Left e -> do
        putStrLn $ show e <> " Traces: " <> show traces
      Right _v ->
        pure ()

_test :: IO ()
_test = do
    let (res, _budget, traces) = evalScript $ compile (authorityTokensValidIn # pconstant currSymbol # (pconstant $
                                            TxOut { txOutAddress = Address (ScriptCredential $ validatorHash validator) Nothing ,
                                                    txOutValue = Value.singleton currSymbol validatorHashTN 1,
                                                    txOutDatumHash = Just (DatumHash "")}))
    case res of
      Left e -> do
        putStrLn $ show e <> " Traces: " <> show traces
      Right _v ->
        pure ()

_test2 :: IO()
_test2 = do
    let (res, _budget, traces) = evalScript $ compile (singleAuthorityTokenBurned (pconstant currSymbol) (pconstantData tinfo) (pconstant mv))
    case res of
      Left e -> do
        putStrLn $ show e <> " Traces: " <> show traces
      Right _v ->
        putStrLn $ show res
    where
      mv = mempty -- Value.singleton currSymbol validatorHashTN (1)
      tinfo =         TxInfo
          { txInfoInputs = 
              [ TxInInfo -- Initiator
                  (TxOutRef "0b2086cbf8b6900f8cb65e012de4516cb66b5cb08a9aaba12a8b88be" 1)
                  TxOut
                    { txOutAddress = Address (ScriptCredential $ validatorHash validator) Nothing
                    , txOutValue = mempty
                    , txOutDatumHash = Just (DatumHash "")
                    }
              , TxInInfo -- Treasury 1
                  (TxOutRef "0b121212121212121212121212121212121212121212121212121221" 2)
                  TxOut
                    { txOutAddress = Address (treasuries !! 0) Nothing
                    , txOutValue = Value.singleton gtSymbol gtToken 10
                    , txOutDatumHash = Just (DatumHash "")
                    }
              , TxInInfo -- Treasury 2
                  (TxOutRef "0b121212121212121212a41212121212121212121212121212121221" 3)
                  TxOut
                    { txOutAddress = Address (treasuries !! 1) Nothing
                    , txOutValue = Value.singleton "1234ab" "LQLQLQL" 10
                    , txOutDatumHash = Just (DatumHash "")
                    }
              ]
          , txInfoOutputs = 
              [ TxOut
                  { txOutAddress = Address (users !! 0) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 1
                  , txOutDatumHash = Nothing
                  }
              , TxOut
                  { txOutAddress = Address (users !! 1) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 1
                  , txOutDatumHash = Nothing
                  }
              , TxOut
                  { txOutAddress = Address (users !! 2) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 1
                  , txOutDatumHash = Nothing
                  }
              -- Send left overs to treasuries
              , TxOut
                  { txOutAddress = Address (treasuries !! 0) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 7
                  , txOutDatumHash = Nothing
                  }
              , TxOut
                  { txOutAddress = Address (treasuries !! 1) Nothing
                  , txOutValue = Value.singleton gtSymbol gtToken 10
                  , txOutDatumHash = Nothing
                  }                  
              ]
          , txInfoFee = Value.singleton "" "" 2
          , txInfoMint = mempty -- Value.singleton currSymbol validatorHashTN (-1)
          , txInfoDCert = []
          , txInfoWdrl = []
          , txInfoValidRange = Interval.always
          , txInfoSignatories = [signer]
          , txInfoData = []
          , txInfoId = "0b2086cbf8b6900f8cb65e012de4516cb66b5cb08a9aaba12a8b88be"
          }