120 lines
5.0 KiB
Haskell
120 lines
5.0 KiB
Haskell
{-# LANGUAGE OverloadedStrings #-}
|
|
{-# LANGUAGE RecordWildCards #-}
|
|
{-# LANGUAGE TemplateHaskell #-}
|
|
module Ringo.ArgParser (ProgArgs(..), parseArgs) where
|
|
|
|
import qualified Data.Text as Text
|
|
import qualified Distribution.Package as P
|
|
import qualified Distribution.PackageDescription as P
|
|
import qualified Distribution.CurrentPackageDescription as P
|
|
import qualified Distribution.Text as DText
|
|
|
|
import Data.List (intercalate)
|
|
import Options.Applicative
|
|
import System.Environment (getProgName)
|
|
|
|
import Ringo.Types
|
|
|
|
data ProgArgs = ProgArgs
|
|
{ progSettings :: Settings
|
|
, progInputFile :: FilePath
|
|
, progOutputDir :: FilePath
|
|
} deriving (Eq, Show)
|
|
|
|
settingsParser :: Parser Settings
|
|
settingsParser = let Settings {..} = defSettings
|
|
in Settings
|
|
<$> (Text.pack <$> strOption (long "dim-prefix"
|
|
<> short 'd'
|
|
<> value (Text.unpack settingDimPrefix)
|
|
<> showDefault
|
|
<> help "Prefix for dimension tables"))
|
|
<*> (Text.pack <$> strOption (long "fact-prefix"
|
|
<> short 'f'
|
|
<> value (Text.unpack settingFactPrefix)
|
|
<> showDefault
|
|
<> help "Prefix for fact tables"))
|
|
<*> option auto (let timeunits = map show [Second ..]
|
|
in long "timeunit"
|
|
<> short 't'
|
|
<> value settingTimeUnit
|
|
<> showDefault
|
|
<> completeWith timeunits
|
|
<> help ("Time unit granularity for fact tables. Possible values: "
|
|
++ intercalate ", " timeunits))
|
|
<*> minorOption "avg-count-col-suffix"
|
|
settingAvgCountColumSuffix
|
|
"Suffix for average count columns"
|
|
<*> minorOption "avg-sum-col-suffix"
|
|
settingAvgSumColumnSuffix
|
|
"Suffix for average sum columns"
|
|
<*> minorOption "dim-id-col-name"
|
|
settingDimTableIdColumnName
|
|
"Name of dimension table id columns"
|
|
<*> minorOption "dim-id-col-type"
|
|
settingDimTableIdColumnType
|
|
"Type of dimension table id columns"
|
|
<*> minorOption "fact-count-col-type"
|
|
settingFactCountColumnType
|
|
"Type of fact table count columns"
|
|
<*> option auto (long "fact-count-distinct-error-rate"
|
|
<> hidden
|
|
<> value settingFactCountDistinctErrorRate
|
|
<> showDefault
|
|
<> help "Error rate for count distinct calulations")
|
|
<*> minorOption "fact-infix"
|
|
settingFactInfix
|
|
"Infix for fact tables"
|
|
<*> minorOption "dependencies-json-file"
|
|
settingDependenciesJSONFileName
|
|
"Name of the output dependencies json file"
|
|
<*> minorOption "facts-json-file"
|
|
settingFactsJSONFileName
|
|
"Name of the output facts json file"
|
|
<*> minorOption "dimensions-json-file"
|
|
settingDimensionJSONFileName
|
|
"Name of the output dimensions json file"
|
|
<*> option auto (long "foreign-key-id-coalesce-val"
|
|
<> hidden
|
|
<> value settingForeignKeyIdCoalesceValue
|
|
<> showDefault
|
|
<> help "Value to coalease missing foriegn key ids to, in fact tables")
|
|
<*> minorOption "tablename-suffix-template"
|
|
settingTableNameSuffixTemplate
|
|
"Suffix template for table names in SQL"
|
|
where
|
|
minorOption longDesc defValue helpTxt =
|
|
Text.pack <$> strOption (long longDesc
|
|
<> hidden
|
|
<> value (Text.unpack defValue)
|
|
<> showDefault
|
|
<> help helpTxt)
|
|
|
|
progArgsParser :: Parser ProgArgs
|
|
progArgsParser =
|
|
ProgArgs
|
|
<$> settingsParser
|
|
<*> argument str (metavar "INPUT"
|
|
<> action "file"
|
|
<> help "Input file")
|
|
<*> argument str (metavar "OUTPUT"
|
|
<> action "directory"
|
|
<> help "Output directory")
|
|
|
|
versionParser :: String -> Parser (a -> a)
|
|
versionParser progName = infoOption (progName ++ " " ++ version)
|
|
(long "version"
|
|
<> help "Print version information")
|
|
where
|
|
version = $(P.getField (DText.display . P.pkgVersion . P.package))
|
|
|
|
parseArgs :: IO ProgArgs
|
|
parseArgs = do
|
|
progName <- getProgName
|
|
execParser $
|
|
info (helper <*> versionParser progName <*> progArgsParser)
|
|
(fullDesc
|
|
<> progDesc $(P.getField P.description)
|
|
<> header (progName ++ " - " ++ $(P.getField P.synopsis))
|
|
<> footer ("© " ++ $(P.getField P.copyright) ++ ". " ++ $(P.getField P.homepage)))
|