src/AoC.hs

   1 module AoC ( getDataFileName ) where
   2
   3 import System.Environment
   4
   5 getDataFileName :: IO String
   6 getDataFileName =
   7   do args <- getArgs
   8      progName <- getProgName
   9      let baseDataName =  if null args
  10                          then progName
  11                          else head args
  12      let dataFileName = "data/" ++ baseDataName ++ ".txt"
  13      return dataFileName