88 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Haskell
		
	
	
	
	
	
			
		
		
	
	
			88 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Haskell
		
	
	
	
	
	
{-# LANGUAGE TemplateHaskell, CPP #-}
 | 
						|
{-|
 | 
						|
 | 
						|
-}
 | 
						|
 | 
						|
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"
 | 
						|
            mainargsflag []){
 | 
						|
              modeGroupFlags = Group {
 | 
						|
                                groupUnnamed = webflags
 | 
						|
                               ,groupHidden = [flagNone ["binary-filename"] (setboolopt "binary-filename") "show the download filename for this executable, and exit"]
 | 
						|
                               ,groupNamed = [(generalflagstitle, generalflags1)]
 | 
						|
                               }
 | 
						|
             ,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 <- toCliOpts 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
 | 
						|
 |