X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Flwn-epub.git;a=blobdiff_plain;f=src%2FMain.hs;h=a1230d60e836c08e849f9c40e78b34e6c4925151;hp=e91654da85a6446d32a02afed636a106dd514a7e;hb=fc0052e451aa03675ebd9a128dfa46573b9357d7;hpb=8e5616a377196f0a200947173d4c78a3dca8a55f diff --git a/src/Main.hs b/src/Main.hs index e91654d..a1230d6 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -1,14 +1,49 @@ +{-# LANGUAGE DoAndIfThenElse #-} module Main where -data Article = Article { headline :: String, - byline :: String, - body_html :: String } - deriving (Eq, Show) +import System.Directory (doesFileExist) +import System.IO ( + Handle, + IOMode (WriteMode), + openBinaryFile, + stdout) -parse_article :: String -> String -parse_article _ = "" +import CommandLine (show_help) +import Configuration (Cfg(..), get_cfg) +import LWN.Article (real_article_path) +import LWN.HTTP (get_login_cookie) +import LWN.Page (epublish, page_from_url) + + +-- | If we're given an empty path, return a handle to +-- 'stdout'. Otherwise, open the given file and return a read/write +-- handle to that. +get_output_handle :: FilePath -> IO Handle +get_output_handle path = + if (null path) then + return stdout + else + openBinaryFile path WriteMode + + +argument_is_file :: Cfg -> IO Bool +argument_is_file cfg = do + path <- real_article_path (article cfg) + doesFileExist path main :: IO () main = do - putStrLn "Hello, world." + cfg' <- get_cfg + aif <- argument_is_file cfg' + cfg <- case aif of + False -> get_login_cookie cfg' + True -> return cfg' + page <- page_from_url cfg (article cfg) + case page of + Just p -> do + output_handle <- get_output_handle (output cfg) + epublish p output_handle + Nothing -> do + _ <- show_help + return ()