arbtt-0.5: src/recover-main.hs

module Main where
import System.Directory
import System.FilePath
import System.Console.GetOpt
import System.Environment
import System.Exit
import System.IO
import qualified Data.Map as M
import Data.Version (showVersion)
import Data.Maybe
import Control.Monad

import TimeLog
import Data
import CommonStartup

import Paths_arbtt (version)

data Flag = Help | Version | InFile String | OutFile String
        deriving Eq

versionStr = "arbtt-recover " ++ showVersion version
header = "Usage: arbtt-recover [OPTIONS...]"

options :: [OptDescr Flag]
options =
     [ Option "h?"     ["help"]
              (NoArg Help)
              "show this help"
     , Option "V"      ["version"]
              (NoArg Version)
              "show the version number"
     , Option "i"      ["infile"]
               (ReqArg InFile "FILE")
               "read from this file instead of ~/.arbtt/capture.log"
     , Option "o"      ["outfile"]
               (ReqArg OutFile "FILE")
               "write to this file instead of ~/.arbtt/capture.log.recovered"
     ]


main = do
  commonStartup
  args <- getArgs
  flags <- case getOpt Permute options args of
          (o,[],[]) | Help `notElem` o  && Version `notElem` o -> return o
          (o,_,_) | Version `elem` o -> do
                hPutStrLn stderr versionStr
                exitSuccess
          (o,_,_) | Help `elem` o -> do
                hPutStr stderr (usageInfo header options)
                exitSuccess
          (_,_,errs) -> do
                hPutStr stderr (concat errs ++ usageInfo header options)
                exitFailure

  dir <- getAppUserDataDirectory "arbtt"

  let captureFilename =
        fromMaybe (dir </> "capture.log") $ listToMaybe $
        mapMaybe (\f -> case f of { InFile f -> Just f; _ -> Nothing}) $
        flags
  let saveFilename =
        fromMaybe (dir </> "capture.log.recovered") $ listToMaybe $
        mapMaybe (\f -> case f of { OutFile f -> Just f; _ -> Nothing}) $
        flags

  captures <- recoverTimeLog captureFilename :: IO (TimeLog CaptureData)
  writeTimeLog saveFilename captures