Renamed library directory

[advent-of-code-22.git] / src / AoC.hs

diff --git a/src/AoC.hs b/src/AoC.hs
new file mode 100644 (file)
index 0000000..764f2cc
--- /dev/null
+++ b/src/AoC.hs
@@ -0,0 +1,13 @@
+module AoC ( getDataFileName ) where
+
+import System.Environment
+
+getDataFileName :: IO String
+getDataFileName =
+  do args <- getArgs
+     progName <- getProgName
+     let baseDataName =  if null args
+                         then progName
+                         else head args 
+     let dataFileName = "data/" ++ baseDataName ++ ".txt"
+     return dataFileName