twin/app/Main.hs

module Main where

import System.Directory
import Options.Applicative
import Data.List

success :: String
success = "20 text/gemini; lang=en; charset=utf-8\r\n"

data Args = Args
    { title :: String
    , directory :: FilePath
    , fileHead :: FilePath
    , verbose :: Bool
    , refs :: Bool
    , cgi :: Bool
    }

args :: Parser Args
args = Args
    <$> strOption
          ( long "title"
        <> short 't'
        <> value "Title Left Blank"
        <> help "Document title" )
    <*> strOption
        ( long "directory"
       <> short 'd'
       <> value "./"
       <> help "Directory to parse" )
    <*> strOption
        ( long "header"
       <> short 'H'
       <> value ""
       <> help "Header text" )
    <*> switch
        ( long "verbose"
       <> short 'v'
       <> help "Verbose mode")
    <*> switch
        ( long "references"
       <> short 'R'
       <> help "Generate link reference based on first line of document" )
    <*> switch
        ( long "cgi"
       <> help "Output gemini file header")

main :: IO ()
main = parseArgs =<< execParser opts
  where
    opts = info ( args <**> helper)
        ( fullDesc
       <> progDesc "Generate  gemini page index"
       <> header "Generate a gemini page index")

parseArgs :: Args -> IO()
parseArgs (Args t dir h v r c) = do
    if c then do
            putStr success
            putStrLn $ "# " ++ t
        else putStrLn $ "# " ++ t
    if not (null h)
        then putStr $ "\n\n" ++ h ++ "\n\n"
        else putStr "\n"
    mapM_ (mkIndex r dir) . filter (isSuffixOf ".gmi") =<< getDirectoryContents dir

mkIndex :: Bool -> FilePath -> FilePath -> IO ()
mkIndex True d l = do
    line <- readFile $ d++"/"++l
    putStrLn $ "=> " ++ l ++ " " ++ (head . lines $ line)
mkIndex False _ l = putStrLn $ "=> " ++ l

--compose = (++) <$> ("=> " ++) <*> (show . length) <*> (head . lines =<< readFile)