1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
module Language.Haskell.Stylish.Step.Data where
import Data.List (find, intercalate)
import Data.Maybe (maybeToList)
import qualified Language.Haskell.Exts as H
import Language.Haskell.Exts.Comments
import Language.Haskell.Stylish.Block
import Language.Haskell.Stylish.Editor
import Language.Haskell.Stylish.Step
import Language.Haskell.Stylish.Util
import Prelude hiding (init)
datas :: H.Module l -> [H.Decl l]
datas (H.Module _ _ _ _ decls) = decls
datas _ = []
type ChangeLine = Change String
step :: Int -> Step
step indentSize = makeStep "Data" (step' indentSize)
step' :: Int -> Lines -> Module -> Lines
step' indentSize ls (module', allComments) = applyChanges changes ls
where
datas' = datas $ fmap linesFromSrcSpan module'
changes = datas' >>= maybeToList . changeDecl allComments indentSize
findComment :: LineBlock -> [Comment] -> Maybe Comment
findComment lb = find commentOnLine
where
commentOnLine (Comment _ (H.SrcSpan _ start _ end _) _) =
blockStart lb == start && blockEnd lb == end
commentsWithin :: LineBlock -> [Comment] -> [Comment]
commentsWithin lb = filter within
where
within (Comment _ (H.SrcSpan _ start _ end _) _) =
start >= blockStart lb && end <= blockEnd lb
changeDecl :: [Comment] -> Int -> H.Decl LineBlock -> Maybe ChangeLine
changeDecl _ _ (H.DataDecl _ (H.DataType _) Nothing _ [] _) = Nothing
changeDecl allComments indentSize (H.DataDecl block (H.DataType _) Nothing dhead decls derivings)
| null $ commentsWithin block allComments = Just $ change block (const $ concat newLines)
| otherwise = Nothing
where
newLines = fmap constructors zipped ++ [fmap (indented . H.prettyPrint) derivings]
zipped = zip decls ([1..] ::[Int])
constructors (decl, 1) = processConstructor allComments typeConstructor indentSize decl
constructors (decl, _) = processConstructor allComments (indented "| ") indentSize decl
typeConstructor = "data " <> H.prettyPrint dhead <> " = "
indented = indent indentSize
changeDecl _ _ _ = Nothing
processConstructor :: [Comment] -> String -> Int -> H.QualConDecl LineBlock -> [String]
processConstructor allComments init indentSize (H.QualConDecl _ _ _ (H.RecDecl _ dname fields)) = do
init <> H.prettyPrint dname : n1 : ns ++ [indented "}"]
where
n1 = processName "{ " ( extractField $ head fields)
ns = fmap (processName ", " . extractField) (tail fields)
processName prefix (fnames, _type, Nothing) =
indented prefix <> intercalate ", " (fmap H.prettyPrint fnames) <> " :: " <> H.prettyPrint _type
processName prefix (fnames, _type, (Just (Comment _ _ c))) =
indented prefix <> intercalate ", " (fmap H.prettyPrint fnames) <> " :: " <> H.prettyPrint _type <> " --" <> c
extractField (H.FieldDecl lb names _type) = (names, _type, findComment lb allComments)
indented = indent indentSize
processConstructor _ init _ decl = [init <> trimLeft (H.prettyPrint decl)]
|
