Main.hs

{-# LANGUAGE OverloadedStrings #-}
-- | Main entry point to the application.
module Main where
import CoreDataHs
import QueryBuild
import System.Environment
import Control.Arrow
import Control.Applicative
import Data.Maybe
import Control.Concurrent.Async
import Data.Text(Text, pack, unpack)
import Data.Text.IO
import Data.Monoid
import Prelude hiding(putStrLn, readFile, writeFile)
version :: IO ()
version = putStrLn "coredata-hs version 0.1.0"

progressIO :: Text -> IO ()
progressIO = putStrLn . ("Creating file " <>)

ifEntity :: String -> Maybe Entity -> IO [Entity]
ifEntity s Nothing = error ("No entity named " <> "'" <> s <> "''" <> " exists")
ifEntity _ e = return . maybeToList $ e


parseArgs :: [String] -> IO ()
parseArgs [] = error "no Model name given."
parseArgs ["-V"] = version
parseArgs ["--version"] = version
parseArgs [a] = modelEntities <$> readFile (fullModelPath a) >>= writeFiles
parseArgs [_, "--entity"] = error "No Entity name given"
parseArgs [a, "--entity", c] = modelEntities <$> readFile (fullModelPath a) >>= ifEntity c . findEntity c >>= writeFiles
parseArgs [a, "--entity", b, "--model-version", c] = modelEntities <$> readFile (versionModelPath a c) >>= ifEntity b . findEntity b >>= writeFiles
parseArgs [a, "--model-version", c] = modelEntities <$> readFile (versionModelPath a c) >>= writeFiles
parseArgs _ = putStrLn "Unrecognized command"


writeFiles :: [Entity] -> IO ()
writeFiles e = do		
		a1 <- async $ sequence_ [progressIO p >> writeFile (unpack p) c | (p, c) <- map (intFileName &&& buildDeclaration) e]              		 		  
		a2 <- async $ sequence_ [progressIO p >> writeFile (unpack p) c | (p, c) <- map (impFileName &&& buildImplementation) e]        				  
		wait a1
		wait a2
		putStrLn "Done!"

main :: IO ()
main = getArgs >>= parseArgs