autotypes/app/Main.hs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

module Main where

import AutoTypes.Unify as U
import Data.Aeson (Value, decodeFileStrict', encode)
import qualified Data.ByteString.Lazy as B
import Data.Maybe
import System.Environment (getArgs)
import System.FilePath (takeFileName)

main :: IO ()
main = do
  filePaths <- getArgs
  types <-
    mapMaybeM
      ( \filePath -> do
          Just value <- decodeFileStrict' filePath
          pure (U.fromJson value)
      )
      filePaths
  B.putStr
    ( encode
        ( head
            ( foldr1
                (\ls rs -> (concat [unify1 l r | l <- ls, r <- rs]))
                (map (: []) types)
            )
        )
    )

mapMaybeM = (fmap catMaybes .) . mapM