{-# LANGUAGE OverloadedStrings #-}

module Main (main) where

import Data.Text         (Text)
import Data.Void         (Void)
import Dhall.Core        (Expr)
import Dhall.CsvToDhall
    ( defaultConversion
    , dhallFromCsv
    , resolveSchemaExpr
    , typeCheckSchemaExpr
    )
import Dhall.Src         (Src)
import System.FilePath   (dropExtension, replaceExtension, takeBaseName)
import Test.Tasty        (TestTree)
import Test.Tasty.Silver (findByExtension)

import qualified Data.Csv
import qualified Data.Text         as Text
import qualified Data.Text.IO
import qualified Dhall.Core        as D
import qualified Dhall.Csv
import qualified Dhall.Csv.Util
import qualified Dhall.CsvToDhall  as CsvToDhall
import qualified GHC.IO.Encoding
import qualified Test.Tasty
import qualified Test.Tasty.Silver as Silver


main :: IO ()
main = do
    GHC.IO.Encoding.setLocaleEncoding GHC.IO.Encoding.utf8

    testTree <- goldenTests
    Test.Tasty.defaultMain testTree

goldenTests :: IO TestTree
goldenTests = do
    dhallToCsvTree <- dhallToCsvGolden
    csvToDhallTree <- csvToDhallGolden
    noHeaderCsvToDhallTree <- noHeaderCsvToDhallGolden
    return $ Test.Tasty.testGroup "dhall-csv"
        [ dhallToCsvTree
        , csvToDhallTree
        , noHeaderCsvToDhallTree
        ]

dhallToCsvGolden :: IO TestTree
dhallToCsvGolden = do
    dhallFiles <- findByExtension [".dhall"] "./tasty/data/dhall-to-csv"
    return $ Test.Tasty.testGroup "dhall-to-csv"
        [ Silver.goldenVsAction
            (takeBaseName dhallFile)
            csvFile
            (Dhall.Csv.codeToValue Nothing =<< Data.Text.IO.readFile dhallFile)
            Dhall.Csv.Util.encodeCsvDefault
        | dhallFile <- dhallFiles
        , let csvFile = replaceExtension dhallFile ".csv"
        ]

csvToDhallGolden :: IO TestTree
csvToDhallGolden = do
    csvFiles <- findByExtension [".csv"] "./tasty/data/csv-to-dhall"
    return $ Test.Tasty.testGroup "csv-to-dhall"
        [ Silver.goldenVsAction
            (takeBaseName csvFile)
            dhallFile
            (getSchemaAndCsv csvFile True schema)
            (showExpressionOrError . (uncurry (dhallFromCsv defaultConversion)))
        | csvFile <- csvFiles
        , let dhallFile = replaceExtension csvFile ".dhall"
        , let schema = Text.pack $ (dropExtension csvFile) ++ "_schema.dhall"
        ]

noHeaderCsvToDhallGolden :: IO TestTree
noHeaderCsvToDhallGolden = do
    csvFiles <- findByExtension [".csv"] "./tasty/data/no-header-csv-to-dhall"
    return $ Test.Tasty.testGroup "csv-to-dhall"
        [ Silver.goldenVsAction
            (takeBaseName csvFile)
            dhallFile
            (getSchemaAndCsv csvFile False schema)
            (showExpressionOrError . (uncurry (dhallFromCsv defaultConversion)))
        | csvFile <- csvFiles
        , let dhallFile = replaceExtension csvFile ".dhall"
        , let schema = Text.pack $ (dropExtension csvFile) ++ "_schema.dhall"
        ]

textToCsv :: Bool -> Text -> IO [Data.Csv.NamedRecord]
textToCsv hasHeader txt =
    case Dhall.Csv.Util.decodeCsvDefault hasHeader txt of
        Left err -> fail err
        Right csv -> return csv

getSchemaAndCsv :: FilePath -> Bool -> Text -> IO (Expr Src Void, [Data.Csv.NamedRecord])
getSchemaAndCsv csvFile hasHeader schema = do
    finalSchema <- typeCheckSchemaExpr id =<< resolveSchemaExpr schema
    csv <- textToCsv hasHeader =<< Data.Text.IO.readFile csvFile
    return (finalSchema, csv)

showExpressionOrError :: Either CsvToDhall.CompileError (Expr Src Void) -> Text
showExpressionOrError (Left err) = Text.pack $ show err
showExpressionOrError (Right expr) = (D.pretty expr) <> "\n"