haskdeep-0.1.0.0: src/Options.hs

-- |
-- Module      :  HaskDeep.Command.Options
-- Copyright   :  Mauro Taraborelli 2012
-- License     :  BSD3
--
-- Maintainer  :  maurotaraborelli@gmail.com
-- Stability   :  experimental
-- Portability :  unknown
--
-- Computes hashes traversing recursively through the directory structure.
-- Use a list of known hashes to audit a set of files.

module Options
    (
     Options(..)
    ,OptExecution(..)
    ,OptCompMode(..)
    ,optionsPI
    )
where

import           Prelude hiding (FilePath)

import           Data.Text (Text)
import qualified Data.Text as T
import           Filesystem.Path.CurrentOS (FilePath)
import qualified Filesystem.Path.CurrentOS as FSC
import           Options.Applicative

import           HaskDeep

data Options = Options OptExecution OptCompMode HaskDeepConfiguration

data OptExecution = OptComputation
                  | OptAudit

data OptCompMode = OptMD5
                 | OptSHA1
                 | OptSHA256
                 | OptSkein512

optionsPI :: ParserInfo Options
optionsPI = info (optionsP <**> helper)
            ( fullDesc
              & progDesc "Computes hashes and audit a set of files"
              & header "haskdeep - file hashing and audit" )

configurationP :: Parser HaskDeepConfiguration
configurationP = HaskDeepConfiguration
                 <$> nullOption
                         ( long "root"
                           & short 'r'
                           & metavar "DIRNAME"
                           & help "Root directory - default current directory"
                           & reader fpReader
                           & value (rootDirectory defaultHaskDeepConfiguration))
                 <*> nullOption
                         ( long "known"
                           & short 'k'
                           & metavar "FILENAME"
                           & help "Known hashes file - default known.haskdeep"
                           & reader fpReader
                           & value (knownHashes defaultHaskDeepConfiguration))
                 <*> nullOption
                         ( long "ignore"
                           & short 'i'
                           & metavar "RULE"
                           & help "Regex to ignore files or directories"
                           & reader ignReader
                           & value (ignoreRule defaultHaskDeepConfiguration))

optionsP :: Parser Options
optionsP = Options
           <$> subparser
                   ( command "compute"
                     (info (pure OptComputation)
                               (progDesc "Computes file hashes and saves them to known hashes file"))
                     & command "audit"
                     (info (pure OptAudit)
                               (progDesc "Audit files comparing them to known hashes")))
           <*> nullOption
                   ( long "computation"
                     & short 'c'
                     & metavar "MODE"
                     & help "md5 | sha1 | sha256 | skein512 - default md5"
                     & reader compReader
                     & value OptMD5)
           <*> configurationP

fpReader :: String -> Maybe FilePath
fpReader fp = Just $ FSC.decodeString fp

ignReader :: String -> Maybe (Maybe Text)
ignReader = Just . Just . T.pack

compReader :: String -> Maybe OptCompMode
compReader "md5"      = Just OptMD5
compReader "sha1"     = Just OptSHA1
compReader "sha256"   = Just OptSHA256
compReader "skein512" = Just OptSkein512
compReader _          = Nothing