[NuBank] Corrigindo issue maxiwell#9 onde o servidor Selenium poderia…

… continuar em execução no caso de exceções.

nubank/Main.hs

@@ -342,6 +342,7 @@ startSeleniumServer = do
 
 stopSeleniumServer :: IO ()
 stopSeleniumServer = do
+        putStrLn "Desligando Selenium Server"
         res <- simpleHTTP (defaultGETRequest_ url) >>= getResponseBody ::IO String
         assert (res == "OKOK") $ return ()
     where
@@ -359,6 +360,42 @@ generateOFX cpf dir fatura =
         putStrLn "Pegando OFX"
         hj <- today
         Tio.writeFile (combine dir fname)  (faturaToOFX fatura hj cpf)
+
+getLoginInfo :: IO (String, String)
+getLoginInfo = do 
+    putStr "CPF: "
+    hFlush stdout
+    cpf <- getLine
+    putStr "Senha: "
+    hFlush stdout
+    senha <- withEcho False getLine
+    putChar '\n'
+    return (cpf, senha)
+
+nuBankScript :: String -> String -> FilePath -> [Flag] -> WD ()
+nuBankScript cpf senha dir flags = do
+    setImplicitWait 200
+    liftIO $ putStrLn "Efetuando login"
+    login cpf senha
+    f0 <- getFatura
+    let fatura = filter isValidTx f0
+
+    when (CSV `elem` flags) $ 
+        liftIO (generateCSV cpf dir fatura)
+    when (null flags || OFX `elem` flags) $ 
+        liftIO (generateOFX cpf dir fatura)
+
+    liftIO $ putStrLn "Deslogando"
+    logout
+
+startSeleniumAndRun :: [Flag] -> IO ()
+startSeleniumAndRun flags = do
+    (cpf, senha) <- getLoginInfo
+    putStrLn "Iniciando Selenium Server"
+    startSeleniumServer
+    dir <- getCurrentDirectory
+    config <- myConfig dir
+    runSession config $ finallyClose $ nuBankScript cpf senha dir flags 
 
 main :: IO()
 main = do
@@ -367,34 +404,8 @@ main = do
 
     if Help `elem`flags || (length args /= length flags) then
         putStrLn usageString
-    else do
-        putStr "CPF: "
-        hFlush stdout
-        cpf <- getLine
-        putStr "Senha: "
-        hFlush stdout
-        senha <- withEcho False getLine
-        putChar '\n'
-
-        putStrLn "Iniciando Selenium Server"
-        startSeleniumServer
-        dir <- getCurrentDirectory
-        config <- myConfig dir
-        runSession config $ do
-            setImplicitWait 200
-            liftIO $ putStrLn "Efetuando login"
-            login cpf senha
-            f0 <- getFatura
-            let fatura = filter isValidTx f0
-
-            when (CSV `elem` flags) $ 
-                liftIO (generateCSV cpf dir fatura)
-            when (null flags || OFX `elem` flags) $ 
-                liftIO (generateOFX cpf dir fatura)
-
-            liftIO $ putStrLn "Deslogando"
-            logout
-            liftIO $ putStrLn "Fechando sessão Selenium"
-            closeSession
-        putStrLn "Desligando Selenium Server"
-        stopSeleniumServer
+    else
+        startSeleniumAndRun flags `catch` exHandler `finally` stopSeleniumServer
+        where 
+            exHandler :: SomeException -> IO ()
+            exHandler e = putStrLn $ "Erro durante a execução" ++ displayException e