-- | Utility functions to read sample contracts (for testing).

module Test.Util.Contracts
       ( getIllTypedContracts
       , getWellTypedContracts
       , getWellTypedMichelsonContracts
       , getWellTypedMorleyContracts
       ) where

import Data.List (isSuffixOf)
import System.Directory (listDirectory)
import System.FilePath ((</>))

getIllTypedContracts :: IO [FilePath]
getIllTypedContracts =
  concatMapM (flip getContractsWithExtension "contracts/ill-typed")
  [".tz", ".mtz"]

getWellTypedContracts :: IO [FilePath]
getWellTypedContracts = getWellTypedMichelsonContracts <> getWellTypedMorleyContracts

getWellTypedMichelsonContracts :: IO [FilePath]
getWellTypedMichelsonContracts = concatMapM (getContractsWithExtension ".tz") wellTypedContractDirs

getWellTypedMorleyContracts :: IO [FilePath]
getWellTypedMorleyContracts = concatMapM (getContractsWithExtension ".mtz") wellTypedContractDirs

getContractsWithExtension :: String -> FilePath -> IO [FilePath]
getContractsWithExtension ext dir = mapMaybe convertPath <$> listDirectory dir
  where
    convertPath :: FilePath -> Maybe FilePath
    convertPath fileName
      | (ext `isSuffixOf` fileName) =
        Just (dir </> fileName)
      | otherwise = Nothing

wellTypedContractDirs :: [FilePath]
wellTypedContractDirs = ["contracts", "contracts/A1", "contracts/tezos_examples"]