@@ -1,9 +1,5 @@
1−{-# LANGUAGE OverloadedStrings #-}
2−{-# LANGUAGE NoImplicitPrelude #-}
3−{-# LANGUAGE TypeApplications #-}
4−{-# LANGUAGE DeriveGeneric #-}
5−{-# LANGUAGE DeriveAnyClass #-}
6−{-# LANGUAGE RecordWildCards #-}
1+{-# LANGUAGE OverloadedStrings, DeriveGeneric, TypeApplications, NoImplicitPrelude #-}
2+{-# LANGUAGE DeriveAnyClass, LambdaCase, ScopedTypeVariables, RecordWildCards #-}
73
84module Control.Hext.Processor
95 ( processFiles
@@ -11,9 +7,11 @@ module Control.Hext.Processor
117
128import Relude
139
14−import Data.List hiding (unlines)
15−import System.Directory.Tree
10+import Control.Hext.Config
1611import Data.Haskell.Parser
12+import Data.List hiding (head, unlines)
13+import qualified Data.Text as T
14+import System.Directory.Tree
1715
1816getFiles :: FilePath -> IO [FilePath]
1917getFiles dir = do
@@ -29,18 +27,40 @@ getFiles dir = do
2927 getHs (Dir name paths) | not $ "." `isPrefixOf` name = paths >>= getHs
3028 getHs _ = []
3129
32−processFile :: [Text] -> Text -> Text
33−processFile extensions file =
34− (unlines . fmap wrapExtension) extensions
30+processFile :: Maybe LineLimit -> Text -> Text -> Text
31+processFile limit extensions file =
32+ extensions
3533 <> "\n"
3634 <> withoutExtensions file
37− where
38− wrapExtension :: Text -> Text
39− wrapExtension e = "{-# LANGUAGE " <> e <> " #-}"
4035
41−processFiles :: [Text] -> FilePath -> IO ()
42−processFiles extensions dir = do
36+wrapExtension :: Text -> Text
37+wrapExtension ee = "{-# LANGUAGE " <> ee <> " #-}"
38+
39+baseLength :: Int
40+baseLength = T.length $ wrapExtension ""
41+
42+processFiles :: Maybe LineLimit -> [Text] -> FilePath -> IO ()
43+processFiles limit extensions dir = do
4344 files <- getFiles dir
4445 forM_ files $ \file -> do
4546 contents <- readFileText file
46− writeFileText file . processFile extensions $ contents
47+ writeFileText file . processFile limit (wrapExtensions extensions) $ contents
48+ where
49+ wrapExtensions :: [Text] -> Text
50+ wrapExtensions [] = ""
51+ wrapExtensions extensions@(e:ee) = case limit of
52+ Nothing -> unlines . fmap wrapExtension $ extensions
53+ Just Unlimited -> unlines . pure . wrapExtension . mconcat . intersperse ", " $ extensions
54+ Just (Limited n) ->
55+ let emptyExtra = ([], ee, baseLength + T.length e)
56+ ee' = unfoldr (\case
57+ Nothing -> Nothing
58+ (Just current@(_, [], _)) -> Just (current, Nothing)
59+ (Just current@(soFar, ne:next, l)) ->
60+ let thing = (ne:soFar, next, l + T.length ne + 2)
61+ in Just (current, Just thing))
62+ (Just emptyExtra)
63+ (extraExtensions, eLeft, _) =
64+ fromMaybe emptyExtra . viaNonEmpty head . reverse . filter (\(_, _, n') -> n' <= n) $ ee'
65+ in (wrapExtension . mconcat . intersperse ", " $ e:extraExtensions)
66+ <> "\n" <> wrapExtensions eLeft