From 5d241e19f8e87ec38e45f58bfa77cc3b33326c69 Mon Sep 17 00:00:00 2001 From: Marty Sluijtman Date: Mon, 31 Jul 2023 14:32:56 +0200 Subject: [PATCH] Link names --- app/Main.hs | 36 ++++++++++++++++++++++++------------ 1 file changed, 24 insertions(+), 12 deletions(-) diff --git a/app/Main.hs b/app/Main.hs index 8944836..9699dd1 100644 --- a/app/Main.hs +++ b/app/Main.hs @@ -12,6 +12,7 @@ data Args = Args , directory :: FilePath , fileHead :: FilePath , verbose :: Bool + , refs :: Bool , cgi :: Bool } @@ -36,6 +37,9 @@ args = Args ( long "verbose" <> short 'v' <> help "Verbose mode") + <*> switch + ( long "references" + <> help "Generate link reference based on first line of document" ) <*> switch ( long "cgi" <> help "Output gemini file header") @@ -49,16 +53,24 @@ main = parseArgs =<< execParser opts <> header "Generate a gemini page index") parseArgs :: Args -> IO() -parseArgs (Args t d h False False) = do - putStrLn $ "# " ++ t - putStrLn h - mkIndex =<< getDirectoryContents d -parseArgs (Args t d h False True) = do - putStr success - putStrLn $ "# " ++ t - putStrLn h - mkIndex =<< getDirectoryContents d -parseArgs _ = putStrLn "fiets" +parseArgs (Args t dir h v c r) = 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" + mkIndex r . filter (isSuffixOf ".gmi") =<< getDirectoryContents dir -mkIndex :: [String] -> IO () -mkIndex = mapM_ (putStrLn . ("=> " ++)) . filter (isSuffixOf ".gmi") +mkIndex :: Bool -> [String] -> IO () +mkIndex True l = mapM_ (putStrLn . ("=> " ++)) l + -- where firstLine f = do + -- print $ head . lines =<< readFile f +mkIndex False l = mapM_ (putStrLn . ("=> " ++)) l + +firstLine :: FilePath -> IO () +firstLine f = do + putStr . head . lines =<< readFile f + +--compose = (++) <$> ("=> " ++) <*> (show . length) <*> (head . lines =<< readFile)