Smarter filename handling for AoC library

diff --git a/src/AoC.hs b/src/AoC.hs
index 90b5c62b18f58c8e9fde85044090d99d0ac4e6d8..b3773a4ed4ebfe914caa445e0a9ee094721aa798 100644 (file)
--- a/src/AoC.hs
+++ b/src/AoC.hs
@@ -1,16 +1,18 @@
 module AoC ( getDataFileName ) where
 
 import System.Environment
+import Data.Char
 
 getDataFileName :: IO String
 getDataFileName =
   do args <- getArgs
      progName <- getProgName
+     let progName' = reverse $ dropWhile isLetter $ reverse progName
      let baseDataName =  if null args
-                         then progName
+                         then progName'
                          else head args 
      let baseDataName' =  if length baseDataName < 5
-                          then progName ++ baseDataName
+                          then progName' ++ baseDataName
                           else baseDataName
      let dataFileName = "../data/" ++ baseDataName' ++ ".txt"
      return dataFileName