summaryrefslogtreecommitdiffhomepage
path: root/lib/Language/Haskell/Stylish/Align.hs
diff options
context:
space:
mode:
authorSean Whitton <spwhitton@spwhitton.name>2021-01-17 11:15:37 -0700
committerSean Whitton <spwhitton@spwhitton.name>2021-01-17 11:15:37 -0700
commit3130faccf7c9a9a7697e246884e2b60fd4b1f9de (patch)
treeab171724845fe928ef05692c27351be933228ec2 /lib/Language/Haskell/Stylish/Align.hs
parentfd8bfa2853825504c2dbc7678154ac8d56d47035 (diff)
parent84770e33bb6286c163c3b2b10fa98d264f6672b8 (diff)
downloadstylish-haskell-3130faccf7c9a9a7697e246884e2b60fd4b1f9de.tar.gz
Merge tag 'v0.12.2.0'
v0.12.2.0 - 0.12.2.0 (2020-10-08) * align: Add a new option for aligning only adjacent items (by 1Computer1) * align: Add support for aligning MultiWayIf syntax (by 1Computer1) * data: Fix some issues with record field padding * module_header: Add separate_lists option * imports: Respect separate_lists for (..) imports * data: Make sorting deriving list optional (by Maxim Koltsov)
Diffstat (limited to 'lib/Language/Haskell/Stylish/Align.hs')
-rw-r--r--lib/Language/Haskell/Stylish/Align.hs53
1 files changed, 26 insertions, 27 deletions
diff --git a/lib/Language/Haskell/Stylish/Align.hs b/lib/Language/Haskell/Stylish/Align.hs
index 1f28d7a..c8a092f 100644
--- a/lib/Language/Haskell/Stylish/Align.hs
+++ b/lib/Language/Haskell/Stylish/Align.hs
@@ -8,7 +8,7 @@ module Language.Haskell.Stylish.Align
--------------------------------------------------------------------------------
import Data.List (nub)
-import qualified Language.Haskell.Exts as H
+import qualified SrcLoc as S
--------------------------------------------------------------------------------
@@ -51,49 +51,48 @@ data Alignable a = Alignable
, aRightLead :: !Int
} deriving (Show)
-
--------------------------------------------------------------------------------
-- | Create changes that perform the alignment.
+
align
- :: Maybe Int -- ^ Max columns
- -> [Alignable H.SrcSpan] -- ^ Alignables
- -> [Change String] -- ^ Changes performing the alignment.
+ :: Maybe Int -- ^ Max columns
+ -> [Alignable S.RealSrcSpan] -- ^ Alignables
+ -> [Change String] -- ^ Changes performing the alignment
align _ [] = []
align maxColumns alignment
- -- Do not make any change if we would go past the maximum number of columns.
- | exceedsColumns (longestLeft + longestRight) = []
- | not (fixable alignment) = []
- | otherwise = map align' alignment
+ -- Do not make an changes if we would go past the maximum number of columns
+ | exceedsColumns (longestLeft + longestRight) = []
+ | not (fixable alignment) = []
+ | otherwise = map align' alignment
where
exceedsColumns i = case maxColumns of
- Nothing -> False -- No number exceeds a maximum column count of
- -- Nothing, because there is no limit to exceed.
- Just c -> i > c
+ Nothing -> False
+ Just c -> i > c
- -- The longest thing in the left column.
- longestLeft = maximum $ map (H.srcSpanEndColumn . aLeft) alignment
+ -- The longest thing in the left column
+ longestLeft = maximum $ map (S.srcSpanEndCol . aLeft) alignment
- -- The longest thing in the right column.
+ -- The longest thing in the right column
longestRight = maximum
- [ H.srcSpanEndColumn (aRight a) - H.srcSpanStartColumn (aRight a)
- + aRightLead a
- | a <- alignment
- ]
-
- align' a = changeLine (H.srcSpanStartLine $ aContainer a) $ \str ->
- let column = H.srcSpanEndColumn $ aLeft a
- (pre, post) = splitAt column str
- in [padRight longestLeft (trimRight pre) ++ trimLeft post]
+ [ S.srcSpanEndCol (aRight a) - S.srcSpanStartCol (aRight a)
+ + aRightLead a
+ | a <- alignment
+ ]
+ align' a = changeLine (S.srcSpanStartLine $ aContainer a) $ \str ->
+ let column = S.srcSpanEndCol $ aLeft a
+ (pre, post) = splitAt column str
+ in [padRight longestLeft (trimRight pre) ++ trimLeft post]
--------------------------------------------------------------------------------
-- | Checks that all the alignables appear on a single line, and that they do
-- not overlap.
-fixable :: [Alignable H.SrcSpan] -> Bool
+
+fixable :: [Alignable S.RealSrcSpan] -> Bool
fixable [] = False
fixable [_] = False
fixable fields = all singleLine containers && nonOverlapping containers
where
containers = map aContainer fields
- singleLine s = H.srcSpanStartLine s == H.srcSpanEndLine s
- nonOverlapping ss = length ss == length (nub $ map H.srcSpanStartLine ss)
+ singleLine s = S.srcSpanStartLine s == S.srcSpanEndLine s
+ nonOverlapping ss = length ss == length (nub $ map S.srcSpanStartLine ss)