summaryrefslogtreecommitdiff
path: root/data/make-reference-files.hs
blob: 77e518496561fddb5dac4549108a987fd762f93e (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
27
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)
import System.FilePath

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)