83 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Haskell
		
	
	
	
	
	
			
		
		
	
	
			83 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Haskell
		
	
	
	
	
	
| module Hledger.Web.Options
 | |
| where
 | |
| import Prelude
 | |
| import Data.Maybe
 | |
| import System.Console.CmdArgs
 | |
| import System.Console.CmdArgs.Explicit
 | |
| 
 | |
| import Hledger.Cli hiding (progname,version,prognameandversion)
 | |
| import Settings
 | |
| 
 | |
| progname, version :: String
 | |
| progname = "hledger-web"
 | |
| #ifdef VERSION
 | |
| version = VERSION
 | |
| #else
 | |
| version = ""
 | |
| #endif
 | |
| prognameandversion :: String
 | |
| prognameandversion = progname ++ " " ++ version :: String
 | |
| 
 | |
| defbaseurlexample :: String
 | |
| defbaseurlexample = (reverse $ drop 4 $ reverse $ defbaseurl defport) ++ "PORT"
 | |
| 
 | |
| webflags :: [Flag [([Char], [Char])]]
 | |
| webflags = [
 | |
|   flagNone ["server"]  (setboolopt "server") ("log requests, don't auto-exit")
 | |
|  ,flagReq ["base-url"]  (\s opts -> Right $ setopt "base-url" s opts) "URL" ("set the base url (default: "++defbaseurlexample++")")
 | |
|  ,flagReq ["port"]  (\s opts -> Right $ setopt "port" s opts) "PORT" ("listen on this tcp port (default: "++show defport++")")
 | |
|  ]
 | |
|  
 | |
| webmode :: Mode [([Char], [Char])]
 | |
| webmode =  (mode "hledger-web" [("command","web")]
 | |
|             "start serving the hledger web interface"
 | |
|             (argsFlag "[PATTERNS]") []){
 | |
|               modeGroupFlags = Group {
 | |
|                                 groupUnnamed = webflags
 | |
|                                ,groupHidden = [flagNone ["binary-filename"] (setboolopt "binary-filename") "show the download filename for this executable, and exit"]
 | |
|                                ,groupNamed = [generalflagsgroup1]
 | |
|                                }
 | |
|              ,modeHelpSuffix=[
 | |
|                   -- "Reads your ~/.hledger.journal file, or another specified by $LEDGER_FILE or -f, and starts the full-window curses ui."
 | |
|                  ]
 | |
|            }
 | |
| 
 | |
| -- hledger-web options, used in hledger-web and above
 | |
| data WebOpts = WebOpts {
 | |
|      server_   :: Bool
 | |
|     ,base_url_ :: String
 | |
|     ,port_     :: Int
 | |
|     ,cliopts_  :: CliOpts
 | |
|  } deriving (Show)
 | |
| 
 | |
| defwebopts :: WebOpts
 | |
| defwebopts = WebOpts
 | |
|     def
 | |
|     def
 | |
|     def
 | |
|     def
 | |
| 
 | |
| -- instance Default WebOpts where def = defwebopts
 | |
| 
 | |
| toWebOpts :: RawOpts -> IO WebOpts
 | |
| toWebOpts rawopts = do
 | |
|   cliopts <- rawOptsToCliOpts rawopts
 | |
|   let p = fromMaybe defport $ maybeintopt "port" rawopts
 | |
|   return defwebopts {
 | |
|               port_ = p
 | |
|              ,server_ = boolopt "server" rawopts
 | |
|              ,base_url_ = maybe (defbaseurl p) stripTrailingSlash $ maybestringopt "base-url" rawopts
 | |
|              ,cliopts_   = cliopts
 | |
|              }
 | |
|   where
 | |
|     stripTrailingSlash = reverse . dropWhile (=='/') . reverse -- yesod don't like it
 | |
| 
 | |
| checkWebOpts :: WebOpts -> IO WebOpts
 | |
| checkWebOpts opts = do
 | |
|   _ <- checkCliOpts $ cliopts_ opts
 | |
|   return opts
 | |
| 
 | |
| getHledgerWebOpts :: IO WebOpts
 | |
| getHledgerWebOpts = processArgs webmode >>= return . decodeRawOpts >>= toWebOpts >>= checkWebOpts
 | |
| 
 |