{-# LANGUAGE LambdaCase #-}

{- |
Module      : Haskoin.Test.Script
Copyright   : No rights reserved
License     : MIT
Maintainer  : jprupp@protonmail.ch
Stability   : experimental
Portability : POSIX
-}
module Haskoin.Util.Arbitrary.Script where

import Crypto.Secp256k1
import qualified Data.ByteString as B
import Data.Maybe
import Data.Word
import Haskoin.Address
import Haskoin.Constants
import Haskoin.Data
import Haskoin.Keys.Common
import Haskoin.Script
import Haskoin.Transaction.Common
import Haskoin.Util
import Haskoin.Util.Arbitrary.Address
import Haskoin.Util.Arbitrary.Crypto
import Haskoin.Util.Arbitrary.Keys
import Haskoin.Util.Arbitrary.Util
import Test.QuickCheck

-- | Arbitrary 'Script' with random script ops.
arbitraryScript :: Gen Script
arbitraryScript :: Gen Script
arbitraryScript = [ScriptOp] -> Script
Script ([ScriptOp] -> Script) -> Gen [ScriptOp] -> Gen Script
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen ScriptOp -> Gen [ScriptOp]
forall a. Gen a -> Gen [a]
listOf Gen ScriptOp
arbitraryScriptOp

-- | Arbitrary 'ScriptOp' (push operations have random data).
arbitraryScriptOp :: Gen ScriptOp
arbitraryScriptOp :: Gen ScriptOp
arbitraryScriptOp =
    [Gen ScriptOp] -> Gen ScriptOp
forall a. [Gen a] -> Gen a
oneof
        -- Pushing Data
        [ ByteString -> ScriptOp
opPushData (ByteString -> ScriptOp) -> Gen ByteString -> Gen ScriptOp
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen ByteString
arbitraryBS1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_0
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_1NEGATE
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RESERVED
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_3
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_4
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_5
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_6
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_7
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_8
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_9
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_10
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_11
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_12
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_13
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_14
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_15
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_16
        , -- Flow control
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_VER
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_IF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOTIF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_VERIF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_VERNOTIF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ELSE
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ENDIF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_VERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RETURN
        , -- Stack operations
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_TOALTSTACK
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_FROMALTSTACK
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_IFDUP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_DEPTH
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_DROP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_DUP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NIP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_OVER
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_PICK
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ROLL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ROT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SWAP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_TUCK
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2DROP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2DUP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_3DUP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2OVER
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2ROT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2SWAP
        , -- Splice
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CAT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SUBSTR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_LEFT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RIGHT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SIZE
        , -- Bitwise logic
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_INVERT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_AND
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_OR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_XOR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_EQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_EQUALVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RESERVED1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RESERVED2
        , -- Arithmetic
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_1ADD
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_1SUB
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2MUL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2DIV
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NEGATE
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ABS
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_0NOTEQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ADD
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SUB
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_MUL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_DIV
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_MOD
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_LSHIFT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RSHIFT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_BOOLAND
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_BOOLOR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NUMEQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NUMEQUALVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NUMNOTEQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_LESSTHAN
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_GREATERTHAN
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_LESSTHANOREQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_GREATERTHANOREQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_MIN
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_MAX
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_WITHIN
        , -- Crypto
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RIPEMD160
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SHA1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SHA256
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_HASH160
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_HASH256
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CODESEPARATOR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKSIG
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKSIGVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKMULTISIG
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKMULTISIGVERIFY
        , -- Expansion
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKLOCKTIMEVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKSEQUENCEVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP4
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP5
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP6
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP7
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP8
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP9
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP10
        , -- Bitcoin Cash Nov 2018 hard fork
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKDATASIG
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKDATASIGVERIFY
        , -- Bitcoin Cash May 2020 hard fork
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_REVERSEBYTES
        , -- Other
          ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_PUBKEYHASH
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_PUBKEY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptOp -> Gen ScriptOp) -> ScriptOp -> Gen ScriptOp
forall a b. (a -> b) -> a -> b
$ Word8 -> ScriptOp
OP_INVALIDOPCODE Word8
0xff
        ]

-- | Arbtirary 'ScriptOp' with a value in @[OP_1 .. OP_16]@.
arbitraryIntScriptOp :: Gen ScriptOp
arbitraryIntScriptOp :: Gen ScriptOp
arbitraryIntScriptOp =
    [ScriptOp] -> Gen ScriptOp
forall a. [a] -> Gen a
elements
        [ ScriptOp
OP_1
        , ScriptOp
OP_2
        , ScriptOp
OP_3
        , ScriptOp
OP_4
        , ScriptOp
OP_5
        , ScriptOp
OP_6
        , ScriptOp
OP_7
        , ScriptOp
OP_8
        , ScriptOp
OP_9
        , ScriptOp
OP_10
        , ScriptOp
OP_11
        , ScriptOp
OP_12
        , ScriptOp
OP_13
        , ScriptOp
OP_14
        , ScriptOp
OP_15
        , ScriptOp
OP_16
        ]

-- | Arbitrary 'PushDataType'.
arbitraryPushDataType :: Gen PushDataType
arbitraryPushDataType :: Gen PushDataType
arbitraryPushDataType = [PushDataType] -> Gen PushDataType
forall a. [a] -> Gen a
elements [PushDataType
OPCODE, PushDataType
OPDATA1, PushDataType
OPDATA2, PushDataType
OPDATA4]

-- | Arbitrary 'SigHash' (including invalid/unknown sighash codes).
arbitrarySigHash :: Gen SigHash
arbitrarySigHash :: Gen SigHash
arbitrarySigHash = Word32 -> SigHash
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word32 -> SigHash) -> Gen Word32 -> Gen SigHash
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (Gen Word32
forall a. Arbitrary a => Gen a
arbitrary :: Gen Word32)

-- | Arbitrary valid 'SigHash'.
arbitraryValidSigHash :: Network -> Gen SigHash
arbitraryValidSigHash :: Network -> Gen SigHash
arbitraryValidSigHash Network
net = do
    SigHash
sh <- [SigHash] -> Gen SigHash
forall a. [a] -> Gen a
elements [SigHash
sigHashAll, SigHash
sigHashNone, SigHash
sigHashSingle]
    SigHash -> SigHash
f1 <-
        [SigHash -> SigHash] -> Gen (SigHash -> SigHash)
forall a. [a] -> Gen a
elements ([SigHash -> SigHash] -> Gen (SigHash -> SigHash))
-> [SigHash -> SigHash] -> Gen (SigHash -> SigHash)
forall a b. (a -> b) -> a -> b
$
            if Maybe Word32 -> Bool
forall a. Maybe a -> Bool
isJust (Network -> Maybe Word32
getSigHashForkId Network
net)
                then [SigHash -> SigHash
forall a. a -> a
id, SigHash -> SigHash
setForkIdFlag]
                else [SigHash -> SigHash
forall a. a -> a
id]
    SigHash -> SigHash
f2 <- [SigHash -> SigHash] -> Gen (SigHash -> SigHash)
forall a. [a] -> Gen a
elements [SigHash -> SigHash
forall a. a -> a
id, SigHash -> SigHash
setAnyoneCanPayFlag]
    SigHash -> Gen SigHash
forall (m :: * -> *) a. Monad m => a -> m a
return (SigHash -> Gen SigHash) -> SigHash -> Gen SigHash
forall a b. (a -> b) -> a -> b
$ SigHash -> SigHash
f1 (SigHash -> SigHash) -> SigHash -> SigHash
forall a b. (a -> b) -> a -> b
$ SigHash -> SigHash
f2 SigHash
sh

arbitrarySigHashFlag :: Gen SigHashFlag
arbitrarySigHashFlag :: Gen SigHashFlag
arbitrarySigHashFlag =
    [SigHashFlag] -> Gen SigHashFlag
forall a. [a] -> Gen a
elements
        [ SigHashFlag
SIGHASH_ALL
        , SigHashFlag
SIGHASH_NONE
        , SigHashFlag
SIGHASH_SINGLE
        , SigHashFlag
SIGHASH_FORKID
        , SigHashFlag
SIGHASH_ANYONECANPAY
        ]

{- | Arbitrary message hash, private key and corresponding 'TxSignature'. The
 signature is generated deterministically using a random message and a random
 private key.
-}
arbitraryTxSignature :: Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature :: Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net = do
    (Hash256
m, SecKey
key, Sig
sig) <- Gen (Hash256, SecKey, Sig)
arbitrarySignature
    SigHash
sh <- (Word8 -> SigHash
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word8 -> SigHash) -> Gen Word8 -> Gen SigHash
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (Gen Word8
forall a. Arbitrary a => Gen a
arbitrary :: Gen Word8)) Gen SigHash -> (SigHash -> Bool) -> Gen SigHash
forall a. Gen a -> (a -> Bool) -> Gen a
`suchThat` SigHash -> Bool
filterBad
    let txsig :: TxSignature
txsig = Sig -> SigHash -> TxSignature
TxSignature Sig
sig SigHash
sh
    (TxHash, SecKey, TxSignature) -> Gen (TxHash, SecKey, TxSignature)
forall (m :: * -> *) a. Monad m => a -> m a
return (Hash256 -> TxHash
TxHash Hash256
m, SecKey
key, TxSignature
txsig)
  where
    filterBad :: SigHash -> Bool
filterBad SigHash
sh =
        Bool -> Bool
not (Bool -> Bool) -> Bool -> Bool
forall a b. (a -> b) -> a -> b
$
            SigHash -> Bool
isSigHashUnknown SigHash
sh
                Bool -> Bool -> Bool
|| Maybe Word32 -> Bool
forall a. Maybe a -> Bool
isNothing (Network -> Maybe Word32
getSigHashForkId Network
net) Bool -> Bool -> Bool
&& SigHash -> Bool
hasForkIdFlag SigHash
sh

-- | Arbitrary transaction signature that could also be empty.
arbitraryTxSignatureEmpty :: Network -> Gen TxSignature
arbitraryTxSignatureEmpty :: Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net =
    [(Int, Gen TxSignature)] -> Gen TxSignature
forall a. [(Int, Gen a)] -> Gen a
frequency
        [ (Int
1, TxSignature -> Gen TxSignature
forall (m :: * -> *) a. Monad m => a -> m a
return TxSignature
TxSignatureEmpty)
        , (Int
10, (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ((TxHash, SecKey, TxSignature) -> TxSignature)
-> Gen (TxHash, SecKey, TxSignature) -> Gen TxSignature
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net)
        ]

-- | Arbitrary m of n parameters.
arbitraryMSParam :: Gen (Int, Int)
arbitraryMSParam :: Gen (Int, Int)
arbitraryMSParam = do
    Int
m <- (Int, Int) -> Gen Int
forall a. Random a => (a, a) -> Gen a
choose (Int
1, Int
16)
    Int
n <- (Int, Int) -> Gen Int
forall a. Random a => (a, a) -> Gen a
choose (Int
m, Int
16)
    (Int, Int) -> Gen (Int, Int)
forall (m :: * -> *) a. Monad m => a -> m a
return (Int
m, Int
n)

-- | Arbitrary 'ScriptOutput' (Can by any valid type).
arbitraryScriptOutput :: Network -> Gen ScriptOutput
arbitraryScriptOutput :: Network -> Gen ScriptOutput
arbitraryScriptOutput Network
net =
    [Gen ScriptOutput] -> Gen ScriptOutput
forall a. [Gen a] -> Gen a
oneof ([Gen ScriptOutput] -> Gen ScriptOutput)
-> [Gen ScriptOutput] -> Gen ScriptOutput
forall a b. (a -> b) -> a -> b
$
        [ Gen ScriptOutput
arbitraryPKOutput
        , Gen ScriptOutput
arbitraryPKHashOutput
        , Gen ScriptOutput
arbitraryMSOutput
        , Gen ScriptOutput
arbitrarySHOutput
        , Gen ScriptOutput
arbitraryDCOutput
        ]
            [Gen ScriptOutput] -> [Gen ScriptOutput] -> [Gen ScriptOutput]
forall a. [a] -> [a] -> [a]
++ if Network -> Bool
getSegWit Network
net
                then
                    [ Gen ScriptOutput
arbitraryWPKHashOutput
                    , Gen ScriptOutput
arbitraryWSHOutput
                    , Gen ScriptOutput
arbitraryWitOutput
                    ]
                else []

{- | Arbitrary 'ScriptOutput' of type 'PayPK', 'PayPKHash' or 'PayMS'
 (Not 'PayScriptHash', 'DataCarrier', or SegWit)
-}
arbitrarySimpleOutput :: Gen ScriptOutput
arbitrarySimpleOutput :: Gen ScriptOutput
arbitrarySimpleOutput =
    [Gen ScriptOutput] -> Gen ScriptOutput
forall a. [Gen a] -> Gen a
oneof
        [ Gen ScriptOutput
arbitraryPKOutput
        , Gen ScriptOutput
arbitraryPKHashOutput
        , Gen ScriptOutput
arbitraryMSOutput
        ]

-- | Arbitrary 'ScriptOutput' of type 'PayPK'
arbitraryPKOutput :: Gen ScriptOutput
arbitraryPKOutput :: Gen ScriptOutput
arbitraryPKOutput = PubKeyI -> ScriptOutput
PayPK (PubKeyI -> ScriptOutput)
-> ((SecKeyI, PubKeyI) -> PubKeyI)
-> (SecKeyI, PubKeyI)
-> ScriptOutput
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ((SecKeyI, PubKeyI) -> ScriptOutput)
-> Gen (SecKeyI, PubKeyI) -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen (SecKeyI, PubKeyI)
arbitraryKeyPair

-- | Arbitrary 'ScriptOutput' of type 'PayPKHash'
arbitraryPKHashOutput :: Gen ScriptOutput
arbitraryPKHashOutput :: Gen ScriptOutput
arbitraryPKHashOutput = Hash160 -> ScriptOutput
PayPKHash (Hash160 -> ScriptOutput) -> Gen Hash160 -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen Hash160
arbitraryHash160

-- | Arbitrary 'PayWitnessPKHash' output.
arbitraryWPKHashOutput :: Gen ScriptOutput
arbitraryWPKHashOutput :: Gen ScriptOutput
arbitraryWPKHashOutput = Hash160 -> ScriptOutput
PayWitnessPKHash (Hash160 -> ScriptOutput) -> Gen Hash160 -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen Hash160
arbitraryHash160

-- | Arbitrary 'PayWitnessScriptHash' output.
arbitraryWSHOutput :: Gen ScriptOutput
arbitraryWSHOutput :: Gen ScriptOutput
arbitraryWSHOutput = Hash256 -> ScriptOutput
PayWitnessScriptHash (Hash256 -> ScriptOutput) -> Gen Hash256 -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen Hash256
arbitraryHash256

arbitraryWitOutput :: Gen ScriptOutput
arbitraryWitOutput :: Gen ScriptOutput
arbitraryWitOutput = do
    Word8
ver <- (Word8, Word8) -> Gen Word8
forall a. Random a => (a, a) -> Gen a
choose (Word8
1, Word8
16)
    Int
len <- (Int, Int) -> Gen Int
forall a. Random a => (a, a) -> Gen a
choose (Int
2, Int
40)
    [Word8]
ws <- Int -> Gen Word8 -> Gen [Word8]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
len Gen Word8
forall a. Arbitrary a => Gen a
arbitrary
    let bs :: ByteString
bs = [Word8] -> ByteString
B.pack [Word8]
ws
    ScriptOutput -> Gen ScriptOutput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptOutput -> Gen ScriptOutput)
-> ScriptOutput -> Gen ScriptOutput
forall a b. (a -> b) -> a -> b
$ Word8 -> ByteString -> ScriptOutput
PayWitness Word8
ver ByteString
bs

-- | Arbitrary 'ScriptOutput' of type 'PayMS'.
arbitraryMSOutput :: Gen ScriptOutput
arbitraryMSOutput :: Gen ScriptOutput
arbitraryMSOutput = do
    (Int
m, Int
n) <- Gen (Int, Int)
arbitraryMSParam
    [PubKeyI]
keys <- ((SecKeyI, PubKeyI) -> PubKeyI)
-> [(SecKeyI, PubKeyI)] -> [PubKeyI]
forall a b. (a -> b) -> [a] -> [b]
map (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ([(SecKeyI, PubKeyI)] -> [PubKeyI])
-> Gen [(SecKeyI, PubKeyI)] -> Gen [PubKeyI]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int -> Gen (SecKeyI, PubKeyI) -> Gen [(SecKeyI, PubKeyI)]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
n Gen (SecKeyI, PubKeyI)
arbitraryKeyPair
    ScriptOutput -> Gen ScriptOutput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptOutput -> Gen ScriptOutput)
-> ScriptOutput -> Gen ScriptOutput
forall a b. (a -> b) -> a -> b
$ [PubKeyI] -> Int -> ScriptOutput
PayMulSig [PubKeyI]
keys Int
m

-- | Arbitrary 'ScriptOutput' of type 'PayMS', only using compressed keys.
arbitraryMSOutputC :: Gen ScriptOutput
arbitraryMSOutputC :: Gen ScriptOutput
arbitraryMSOutputC = do
    (Int
m, Int
n) <- Gen (Int, Int)
arbitraryMSParam
    [PubKeyI]
keys <-
        ((SecKeyI, PubKeyI) -> PubKeyI)
-> [(SecKeyI, PubKeyI)] -> [PubKeyI]
forall a b. (a -> b) -> [a] -> [b]
map (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd
            ([(SecKeyI, PubKeyI)] -> [PubKeyI])
-> Gen [(SecKeyI, PubKeyI)] -> Gen [PubKeyI]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int -> Gen (SecKeyI, PubKeyI) -> Gen [(SecKeyI, PubKeyI)]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
n (Gen (SecKeyI, PubKeyI)
arbitraryKeyPair Gen (SecKeyI, PubKeyI)
-> ((SecKeyI, PubKeyI) -> Bool) -> Gen (SecKeyI, PubKeyI)
forall a. Gen a -> (a -> Bool) -> Gen a
`suchThat` (PubKeyI -> Bool
pubKeyCompressed (PubKeyI -> Bool)
-> ((SecKeyI, PubKeyI) -> PubKeyI) -> (SecKeyI, PubKeyI) -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd))
    ScriptOutput -> Gen ScriptOutput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptOutput -> Gen ScriptOutput)
-> ScriptOutput -> Gen ScriptOutput
forall a b. (a -> b) -> a -> b
$ [PubKeyI] -> Int -> ScriptOutput
PayMulSig [PubKeyI]
keys Int
m

-- | Arbitrary 'ScriptOutput' of type 'PayScriptHash'.
arbitrarySHOutput :: Gen ScriptOutput
arbitrarySHOutput :: Gen ScriptOutput
arbitrarySHOutput = Hash160 -> ScriptOutput
PayScriptHash (Hash160 -> ScriptOutput)
-> (Address -> Hash160) -> Address -> ScriptOutput
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Address -> Hash160
getAddrHash160 (Address -> ScriptOutput) -> Gen Address -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen Address
arbitraryScriptAddress

-- | Arbitrary 'ScriptOutput' of type 'DataCarrier'.
arbitraryDCOutput :: Gen ScriptOutput
arbitraryDCOutput :: Gen ScriptOutput
arbitraryDCOutput = ByteString -> ScriptOutput
DataCarrier (ByteString -> ScriptOutput) -> Gen ByteString -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen ByteString
arbitraryBS1

-- | Arbitrary 'ScriptInput'.
arbitraryScriptInput :: Network -> Gen ScriptInput
arbitraryScriptInput :: Network -> Gen ScriptInput
arbitraryScriptInput Network
net =
    [Gen ScriptInput] -> Gen ScriptInput
forall a. [Gen a] -> Gen a
oneof
        [ Network -> Gen ScriptInput
arbitraryPKInput Network
net
        , Network -> Gen ScriptInput
arbitraryPKHashInput Network
net
        , Network -> Gen ScriptInput
arbitraryMSInput Network
net
        , Network -> Gen ScriptInput
arbitrarySHInput Network
net
        ]

{- | Arbitrary 'ScriptInput' of type 'SpendPK', 'SpendPKHash' or 'SpendMulSig'
 (not 'ScriptHashInput')
-}
arbitrarySimpleInput :: Network -> Gen ScriptInput
arbitrarySimpleInput :: Network -> Gen ScriptInput
arbitrarySimpleInput Network
net =
    [Gen ScriptInput] -> Gen ScriptInput
forall a. [Gen a] -> Gen a
oneof
        [ Network -> Gen ScriptInput
arbitraryPKInput Network
net
        , Network -> Gen ScriptInput
arbitraryPKHashInput Network
net
        , Network -> Gen ScriptInput
arbitraryMSInput Network
net
        ]

-- | Arbitrary 'ScriptInput' of type 'SpendPK'.
arbitraryPKInput :: Network -> Gen ScriptInput
arbitraryPKInput :: Network -> Gen ScriptInput
arbitraryPKInput Network
net = SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput)
-> (TxSignature -> SimpleInput) -> TxSignature -> ScriptInput
forall b c a. (b -> c) -> (a -> b) -> a -> c
. TxSignature -> SimpleInput
SpendPK (TxSignature -> ScriptInput) -> Gen TxSignature -> Gen ScriptInput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net

-- | Arbitrary 'ScriptInput' of type 'SpendPK'.
arbitraryPKHashInput :: Network -> Gen ScriptInput
arbitraryPKHashInput :: Network -> Gen ScriptInput
arbitraryPKHashInput Network
net = do
    TxSignature
sig <- Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net
    PubKeyI
key <- (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ((SecKeyI, PubKeyI) -> PubKeyI)
-> Gen (SecKeyI, PubKeyI) -> Gen PubKeyI
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen (SecKeyI, PubKeyI)
arbitraryKeyPair
    ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput) -> SimpleInput -> ScriptInput
forall a b. (a -> b) -> a -> b
$ TxSignature -> PubKeyI -> SimpleInput
SpendPKHash TxSignature
sig PubKeyI
key

-- | Like 'arbitraryPKHashInput' without empty signatures.
arbitraryPKHashInputFull :: Network -> Gen ScriptInput
arbitraryPKHashInputFull :: Network -> Gen ScriptInput
arbitraryPKHashInputFull Network
net = do
    TxSignature
sig <- (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ((TxHash, SecKey, TxSignature) -> TxSignature)
-> Gen (TxHash, SecKey, TxSignature) -> Gen TxSignature
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net
    PubKeyI
key <- (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ((SecKeyI, PubKeyI) -> PubKeyI)
-> Gen (SecKeyI, PubKeyI) -> Gen PubKeyI
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen (SecKeyI, PubKeyI)
arbitraryKeyPair
    ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput) -> SimpleInput -> ScriptInput
forall a b. (a -> b) -> a -> b
$ TxSignature -> PubKeyI -> SimpleInput
SpendPKHash TxSignature
sig PubKeyI
key

-- | Like above but only compressed.
arbitraryPKHashInputFullC :: Network -> Gen ScriptInput
arbitraryPKHashInputFullC :: Network -> Gen ScriptInput
arbitraryPKHashInputFullC Network
net = do
    TxSignature
sig <- (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ((TxHash, SecKey, TxSignature) -> TxSignature)
-> Gen (TxHash, SecKey, TxSignature) -> Gen TxSignature
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net
    PubKeyI
key <- ((SecKeyI, PubKeyI) -> PubKeyI)
-> Gen (SecKeyI, PubKeyI) -> Gen PubKeyI
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd (Gen (SecKeyI, PubKeyI) -> Gen PubKeyI)
-> Gen (SecKeyI, PubKeyI) -> Gen PubKeyI
forall a b. (a -> b) -> a -> b
$ Gen (SecKeyI, PubKeyI)
arbitraryKeyPair Gen (SecKeyI, PubKeyI)
-> ((SecKeyI, PubKeyI) -> Bool) -> Gen (SecKeyI, PubKeyI)
forall a. Gen a -> (a -> Bool) -> Gen a
`suchThat` (PubKeyI -> Bool
pubKeyCompressed (PubKeyI -> Bool)
-> ((SecKeyI, PubKeyI) -> PubKeyI) -> (SecKeyI, PubKeyI) -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd)
    ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput) -> SimpleInput -> ScriptInput
forall a b. (a -> b) -> a -> b
$ TxSignature -> PubKeyI -> SimpleInput
SpendPKHash TxSignature
sig PubKeyI
key

-- | Arbitrary 'ScriptInput' of type 'SpendMulSig'.
arbitraryMSInput :: Network -> Gen ScriptInput
arbitraryMSInput :: Network -> Gen ScriptInput
arbitraryMSInput Network
net = do
    Int
m <- (Int, Int) -> Int
forall a b. (a, b) -> a
fst ((Int, Int) -> Int) -> Gen (Int, Int) -> Gen Int
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen (Int, Int)
arbitraryMSParam
    [TxSignature]
sigs <- Int -> Gen TxSignature -> Gen [TxSignature]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net)
    ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput) -> SimpleInput -> ScriptInput
forall a b. (a -> b) -> a -> b
$ [TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs

-- | Arbitrary 'ScriptInput' of type 'ScriptHashInput'.
arbitrarySHInput :: Network -> Gen ScriptInput
arbitrarySHInput :: Network -> Gen ScriptInput
arbitrarySHInput Network
net = do
    ScriptInput
i <- Network -> Gen ScriptInput
arbitrarySimpleInput Network
net
    SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput (ScriptInput -> SimpleInput
getRegularInput ScriptInput
i) (ScriptOutput -> ScriptInput)
-> Gen ScriptOutput -> Gen ScriptInput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen ScriptOutput
arbitrarySimpleOutput

{- | Arbitrary 'ScriptInput' of type 'ScriptHashInput' containing a
 'RedeemScript' of type 'PayMulSig' and an input of type 'SpendMulSig'.
-}
arbitraryMulSigSHInput :: Network -> Gen ScriptInput
arbitraryMulSigSHInput :: Network -> Gen ScriptInput
arbitraryMulSigSHInput Network
net =
    Gen ScriptOutput
arbitraryMSOutput Gen ScriptOutput
-> (ScriptOutput -> Gen ScriptInput) -> Gen ScriptInput
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
        rdm :: ScriptOutput
rdm@(PayMulSig [PubKeyI]
_ Int
m) -> do
            [TxSignature]
sigs <- Int -> Gen TxSignature -> Gen [TxSignature]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net)
            ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput ([TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs) ScriptOutput
rdm
        ScriptOutput
_ -> Gen ScriptInput
forall a. HasCallStack => a
undefined

{- | Arbitrary 'ScriptInput' of type 'ScriptHashInput' containing a
 'RedeemScript' of type 'PayMulSig' and an input of type 'SpendMulSig'.
-}
arbitraryMulSigSHInputC :: Network -> Gen ScriptInput
arbitraryMulSigSHInputC :: Network -> Gen ScriptInput
arbitraryMulSigSHInputC Network
net =
    Gen ScriptOutput
arbitraryMSOutputC Gen ScriptOutput
-> (ScriptOutput -> Gen ScriptInput) -> Gen ScriptInput
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
        rdm :: ScriptOutput
rdm@(PayMulSig [PubKeyI]
_ Int
m) -> do
            [TxSignature]
sigs <- Int -> Gen TxSignature -> Gen [TxSignature]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net)
            ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput ([TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs) ScriptOutput
rdm
        ScriptOutput
_ -> Gen ScriptInput
forall a. HasCallStack => a
undefined

-- | Like 'arbitraryMulSigSHCInput' with no empty signatures.
arbitraryMulSigSHInputFull :: Network -> Gen ScriptInput
arbitraryMulSigSHInputFull :: Network -> Gen ScriptInput
arbitraryMulSigSHInputFull Network
net =
    Gen ScriptOutput
arbitraryMSOutput Gen ScriptOutput
-> (ScriptOutput -> Gen ScriptInput) -> Gen ScriptInput
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
        rdm :: ScriptOutput
rdm@(PayMulSig [PubKeyI]
_ Int
m) -> do
            [TxSignature]
sigs <- ((TxHash, SecKey, TxSignature) -> TxSignature)
-> [(TxHash, SecKey, TxSignature)] -> [TxSignature]
forall a b. (a -> b) -> [a] -> [b]
map (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ([(TxHash, SecKey, TxSignature)] -> [TxSignature])
-> Gen [(TxHash, SecKey, TxSignature)] -> Gen [TxSignature]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int
-> Gen (TxHash, SecKey, TxSignature)
-> Gen [(TxHash, SecKey, TxSignature)]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net)
            ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput ([TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs) ScriptOutput
rdm
        ScriptOutput
_ -> Gen ScriptInput
forall a. HasCallStack => a
undefined

-- | Like 'arbitraryMulSigSHCInput' with no empty signatures.
arbitraryMulSigSHInputFullC :: Network -> Gen ScriptInput
arbitraryMulSigSHInputFullC :: Network -> Gen ScriptInput
arbitraryMulSigSHInputFullC Network
net =
    Gen ScriptOutput
arbitraryMSOutputC Gen ScriptOutput
-> (ScriptOutput -> Gen ScriptInput) -> Gen ScriptInput
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
        rdm :: ScriptOutput
rdm@(PayMulSig [PubKeyI]
_ Int
m) -> do
            [TxSignature]
sigs <- ((TxHash, SecKey, TxSignature) -> TxSignature)
-> [(TxHash, SecKey, TxSignature)] -> [TxSignature]
forall a b. (a -> b) -> [a] -> [b]
map (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ([(TxHash, SecKey, TxSignature)] -> [TxSignature])
-> Gen [(TxHash, SecKey, TxSignature)] -> Gen [TxSignature]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int
-> Gen (TxHash, SecKey, TxSignature)
-> Gen [(TxHash, SecKey, TxSignature)]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net)
            ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput ([TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs) ScriptOutput
rdm
        ScriptOutput
_ -> Gen ScriptInput
forall a. HasCallStack => a
undefined