{- FOURMOLU_DISABLE -}
-- The MIT License (MIT)

-- Copyright (c) 2016-2024 Objectionary.com

-- Permission is hereby granted, free of charge, to any person obtaining a copy
-- of this software and associated documentation files (the "Software"), to deal
-- in the Software without restriction, including without limitation the rights
-- to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-- copies of the Software, and to permit persons to whom the Software is
-- furnished to do so, subject to the following conditions:

-- The above copyright notice and this permission notice shall be included
-- in all copies or substantial portions of the Software.

-- THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-- IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-- FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL THE
-- AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-- LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-- OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-- SOFTWARE.
{- FOURMOLU_ENABLE -}
{-# LANGUAGE OverloadedRecordDot #-}
{-# LANGUAGE QuasiQuotes #-}
{-# LANGUAGE RecordWildCards #-}

module Language.EO.Phi.Pipeline.Dataize.PrintConfigs where

import Data.Functor ((<&>))
import Data.List (stripPrefix)
import Data.Maybe
import Language.EO.Phi.Pipeline.Config
import PyF (fmt)

printDataizeConfigs :: PipelineConfig -> [String] -> Bool -> IO ()
printDataizeConfigs :: PipelineConfig -> [String] -> Bool -> IO ()
printDataizeConfigs PipelineConfig
pipelineConfig [String]
phiPrefixes Bool
singleLine = do
  let args :: [String]
args =
        PipelineConfig
pipelineConfig.testSets [TestSet] -> (TestSet -> String) -> [String]
forall (f :: * -> *) a b. Functor f => f a -> (a -> b) -> f b
<&> \TestSet
testSet ->
          let phiInitial :: String
phiInitial = TestSet
testSet.phi.initial
              phi :: String
phi =
                case [Maybe String] -> [String]
forall a. [Maybe a] -> [a]
catMaybes [String -> String -> Maybe String
forall a. Eq a => [a] -> [a] -> Maybe [a]
stripPrefix String
prefix String
phiInitial | String
prefix <- [String]
phiPrefixes] of
                  [] -> String
phiInitial
                  String
x : [String]
_ -> String
x
              atoms :: String
atoms = case TestSet
testSet.atoms of
                Maybe AtomsSet
Nothing -> String
""
                Just (AtomsSet{Maybe [String]
enable :: Maybe [String]
disable :: Maybe [String]
$sel:enable:AtomsSet :: AtomsSet -> Maybe [String]
$sel:disable:AtomsSet :: AtomsSet -> Maybe [String]
..}) ->
                  let enable' :: Maybe String
enable' = [String] -> String
unwords ([String] -> String)
-> ([String] -> [String]) -> [String] -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((\String
atom -> [fmt|--enable-atom {atom}|]) (String -> String) -> [String] -> [String]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$>) ([String] -> String) -> Maybe [String] -> Maybe String
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Maybe [String]
enable
                      disable' :: Maybe String
disable' = [String] -> String
unwords ([String] -> String)
-> ([String] -> [String]) -> [String] -> String
forall b c a. (b -> c) -> (a -> b) -> a -> c
. ((\String
atom -> [fmt|--disable-atom {atom}|]) (String -> String) -> [String] -> [String]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$>) ([String] -> String) -> Maybe [String] -> Maybe String
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Maybe [String]
disable
                   in [String] -> String
unwords ([Maybe String] -> [String]
forall a. [Maybe a] -> [a]
catMaybes [Maybe String
enable', Maybe String
disable'])
              enableTestSet :: String
enableTestSet
                | Bool -> Maybe Bool -> Bool
forall a. a -> Maybe a -> a
fromMaybe Bool
True TestSet
testSet.enable = String
"true"
                | Bool
otherwise = String
"false"
           in [fmt|'{{"atoms": "{atoms}", "phi": "{phi}", "enable": "{enableTestSet}"}}'|]
  String -> IO ()
putStrLn ((if Bool
singleLine then [String] -> String
unwords else [String] -> String
unlines) [String]
args)