X-Git-Url: http://gitweb.michael.orlitzky.com/?p=dead%2Flwn-epub.git;a=blobdiff_plain;f=src%2FMain.hs;h=88ba39c5ff493cd9b3b0acafc9992738d080138f;hp=da760e45af7a58676b8b6cfdd2c0198de6c41d78;hb=10f322ce20600de109c4643967b6ce3f61f69bf6;hpb=44549423b80c9bf8edf5e942bc8ac457d9817195 diff --git a/src/Main.hs b/src/Main.hs index da760e4..88ba39c 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -1,7 +1,37 @@ +{-# LANGUAGE DoAndIfThenElse #-} module Main where +import System.IO ( + Handle, + IOMode (WriteMode), + openBinaryFile, + stdout) + +import CommandLine (show_help) +import Configuration (Cfg(..), get_cfg) +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 + main :: IO () main = do - putStrLn "Hello, world." + cfg <- get_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 ()