Create reference files from unpacked archives with helper program

1 files changed, 26 insertions, 0 deletions

diff --git a/data/make-reference-files.hs b/data/make-reference-files.hs
new file mode 100644
index 000000000..2e64dc51f
--- /dev/null
+++ b/data/make-reference-files.hs
@@ -0,0 +1,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)