#!/usr/bin/env runghc
import Distribution.Simple
import System.Directory
import System.FilePath
import Control.Monad
import System.Info
import System.Exit
import List
 
inf  = "infinity"
-- files to remove
bins = [inf] ++ 
       ["util" </> "unlambda"]    
-- directories to clean
dirs = [ "."
       , "util" ] ++
       map ("src" </>) [ "Infinity"
                       , "Infinity" </> "Plugins"]
 
main = defaultMainWithHooks $ simpleUserHooks {
      postBuild = copyInfinity,
      postClean = cleanInfinity
}
 
copyInfinity _ _ _ _ = do
  copyFile (joinPath ["dist","build","infinity",inf]) inf
  ignore (createDirectory "Log" >> createDirectory "State")
 
cleanInfinity _ _ _ _ = do
  mapM_ (ignore . removeFile) bins
  mapM_ clean dirs
 where
   clean d   = getDirectoryContents d >>= mapM_ (check d)
   check d f = when (any (`isSuffixOf` f) exts)
                (removeFile (d </> f))
   exts      = ["~",".o",".hi",".o-boot",".hi-boot"]
 
 
-- convenience
ignore a = Prelude.catch a (\_ -> return ())