-{-# LANGUAGE ScopedTypeVariables, RecordWildCards #-}
+{-# LANGUAGE DoAndIfThenElse #-}
module Main
where
-import Data.Maybe (fromJust)
-import System.Console.CmdArgs (cmdArgsRun)
+import System.IO (
+ Handle,
+ IOMode (WriteMode),
+ openBinaryFile,
+ stdout)
-import Text.XML.HXT.Core
+import CommandLine (show_help)
+import Configuration (Cfg(..), get_cfg)
+import LWN.Page (epublish, page_from_url)
-import CommandLine (Args(..), apply_args)
-import Epublishable
-import LWN.ArticlePage
-import LWN.FullPage
+-- | 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
-my_read :: String -> IOStateArrow s b XmlTree
-my_read =
- readDocument [ withValidate no,
- withParseHTML yes,
- withInputEncoding utf8,
- withWarnings no ]
main :: IO ()
main = do
- Args{..} <- apply_args
- print article
-
- -- let article_html = my_read "test/fixtures/501317-article.html"
- -- ioap <- parse article_html
- -- let article_page :: ArticlePage = fromJust $ ioap
- -- epublish article_page "single_article.epub"
-
- -- let page_html = my_read "test/fixtures/500848-page.html"
- -- ioap_f <- parse page_html
- -- let full_page :: FullPage = fromJust $ ioap_f
- -- epublish full_page "full_page.epub"
-
- -- let bigpage_html = my_read "test/fixtures/50844-bigpage.html"
- -- ioap_bp <- parse bigpage_html
- -- let bigpage :: FullPage = fromJust $ ioap_bp
- -- epublish bigpage "bigpage.epub"
-
- -- putStrLn "Done."
+ 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 ()