Improve getPagePath

psibi · Nov 20, 2018 · c9f8027 · c9f8027
1 parent 4fe8cff
commit c9f8027
Showing 1 changed file with 2 additions and 12 deletions.
diff --git a/app/Main.hs b/app/Main.hs
@@ -1,4 +1,3 @@
-{-#LANGUAGE CPP#-}
 {-#LANGUAGE ScopedTypeVariables#-}
 
 module Main where
@@ -84,17 +83,8 @@ getPagePath :: String -> IO (Maybe FilePath)
 getPagePath page = do
   homeDir <- getHomeDirectory
   let pageDir = homeDir </> tldrDirName </> "tldr" </> "pages"
-      x@(f1:f2:f3:f4:[]) = map (\x -> pageDir </> x </> page <.> "md") checkDirs
-#if MIN_VERSION_base(4,7,0)
-  f1' <- pageExists f1
-  f2' <- pageExists f2
-  f3' <- pageExists f3
-  f4' <- pageExists f4
-  return $ f1' <|> f2' <|> f3' <|> f4'
-#else
-  pageExists f1 <|> pageExists f2 <|> pageExists f3 <|> pageExists f4
-#endif
-
+      paths = map (\x -> pageDir </> x </> page <.> "md") checkDirs
+  foldr (<|>) Nothing <$> mapM pageExists paths
 
 main :: IO ()
 main = do