From 5475a3868ceee55d733a47ce63a30b16b59ae22d Mon Sep 17 00:00:00 2001
From: Simon Michael <simon@joyful.com>
Date: Fri, 16 Feb 2007 12:24:13 +0000
Subject: [PATCH] fix -f option

---
 Options.hs | 41 ++++++++++++++++++++---------------------
 hledger.hs |  4 ++--
 2 files changed, 22 insertions(+), 23 deletions(-)

diff --git a/Options.hs b/Options.hs
index 3f6b58082..0c3e629ad 100644
--- a/Options.hs
+++ b/Options.hs
@@ -8,36 +8,35 @@ import Data.Maybe (fromMaybe)
 import Utils
 
 
-data Flag = Version | File String | ShowSubs
-            deriving (Show,Eq)
-    
-options :: [OptDescr Flag]
-options = [
-            Option ['v'] ["version"] (NoArg Version)     "show version number"
-          , Option ['f'] ["file"]    (OptArg inp "FILE") "ledger file, or - to read stdin"
-          , Option ['s'] ["subtotal"] (NoArg ShowSubs)     "balance: show sub-accounts" --; register: show subtotals"
-          ]
+usageHeader = "Usage: hledger [OPTIONS] register|balance [MATCHARGS]"
 
-inp :: Maybe String -> Flag
-inp  = File . fromMaybe "stdin"
-    
 getOptions :: [String] -> IO ([Flag], [String])
 getOptions argv =
     case getOpt RequireOrder options argv of
       (o,n,[]  ) -> return (o,n)
       (_,_,errs) -> ioError (userError (concat errs ++ usageInfo usageHeader options))
 
-usageHeader = "Usage: hledger [OPTIONS] register|balance [MATCHARGS]"
+options :: [OptDescr Flag]
+options = [
+            Option ['v'] ["version"] (NoArg Version)     "show version number"
+          , Option ['f'] ["file"]    (OptArg readFileOpt "FILE") "ledger file, or - to read stdin"
+          , Option ['s'] ["subtotal"] (NoArg ShowSubs)     "balance: show sub-accounts" --; register: show subtotals"
+          ]
 
-get_content :: Flag -> Maybe String
-get_content (File s) = Just s
+data Flag = Version | File String | ShowSubs deriving (Show,Eq)
+    
+readFileOpt :: Maybe String -> Flag
+readFileOpt  = File . fromMaybe "stdin"
+    
+getFile :: Flag -> String
+getFile (File s) = s
+getFile _ = []
 
-defaultLedgerFile = "~/ledger.dat"
-
-getLedgerFilePath :: IO String
-getLedgerFilePath = do
-  defaultpath <- tildeExpand defaultLedgerFile
-  getEnv "LEDGER" `catch` \_ -> return defaultpath >>= return
+getLedgerFilePath :: [Flag] -> IO String
+getLedgerFilePath opts = do
+  defaultpath <- tildeExpand "~/ledger.dat"
+  envordefault <- getEnv "LEDGER" `catch` \_ -> return defaultpath
+  return $ last $ [envordefault] ++ (filter (/= "") (map getFile opts))
 
 -- ledger pattern args are a list of account patterns optionally followed
 -- by -- and a list of description patterns
diff --git a/hledger.hs b/hledger.hs
index 18431fcee..e2118d2ee 100644
--- a/hledger.hs
+++ b/hledger.hs
@@ -42,11 +42,11 @@ test = do
 
 register :: [Flag] -> [String] -> IO ()
 register opts args = do 
-  getLedgerFilePath >>= parseLedgerFile >>= doWithParsed (printRegister opts args)
+  getLedgerFilePath opts >>= parseLedgerFile >>= doWithParsed (printRegister opts args)
 
 balance :: [Flag] -> [String] -> IO ()
 balance opts args = do
-  getLedgerFilePath >>= parseLedgerFile >>= doWithParsed (printBalance opts args)
+  getLedgerFilePath opts >>= parseLedgerFile >>= doWithParsed (printBalance opts args)
 
 -- utils