summaryrefslogtreecommitdiff
path: root/data/make-reference-files.hs
blob: 2e64dc51fb146717010a5594a86f8a822f609d76 (plain)
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
import System.Environment
import System.Directory
import Codec.Archive.Zip
import qualified Data.ByteString.Lazy as BS
import qualified Control.Exception as E
import System.IO.Error (isDoesNotExistError)

mkzip :: String -> IO ()
mkzip fmt = do
  let dir    = "data/"++fmt
      output = "data/reference."++fmt
  cd <- getCurrentDirectory
  setCurrentDirectory dir
  archive <- addFilesToArchive [OptRecursive] emptyArchive ["."]
  setCurrentDirectory cd
  removeIfExists output
  BS.writeFile output $ fromArchive archive

removeIfExists :: FilePath -> IO ()
removeIfExists fileName = removeFile fileName `E.catch` handleExists
  where handleExists e
          | isDoesNotExistError e = return ()
          | otherwise = E.throwIO e

main :: IO ()
main = getArgs >>= mkzip . (!!0)