master..100fa44 · ormolu

.buildkite/pipeline.yml+3−14

since splitsnapshots

unifiedsplit

@@ -1,18 +1,7 @@

1steps:

2− - label: Build and test with GHC 8.6.5

2+ - label: Build and test

3− command: |

4− nix-build --keep-going --no-out-link --argstr ormoluCompiler ghc865

5− timeout: 100

6− - label: Build and test with GHC 8.8.4

7− command: |

8− nix-build --keep-going --no-out-link --argstr ormoluCompiler ghc884

9− timeout: 100

10− - label: Build and test with GHC 8.10.1

11− command: |

12− nix-build --keep-going --no-out-link --argstr ormoluCompiler ghc8101

13− timeout: 100

14− - wait

15− - label: Check formatting

16 command: |

3 command: |

4+ nix-build --keep-going --no-out-link

17 ./format.sh

5 ./format.sh

18 git diff --exit-code --color=always

6 git diff --exit-code --color=always

7+ timeout: 100

.github/workflows/binaries.ymldeleted−39

@@ -1,39 +0,0 @@

1−name: Build binaries

2−on:

3− release:

4− types: [published]

5−jobs:

6− build:

7− strategy:

8− fail-fast: false

9− matrix:

10− os: [ubuntu-latest, macOS-latest, windows-latest]

11− runs-on: ${{ matrix.os }}

12− steps:

13− - uses: actions/checkout@v2

14− - uses: actions/setup-haskell@v1

15− if: runner.os != 'Linux'

16− with:

17− ghc-version: '8.10.1'

18− cabal-version: '3.2'

19− - name: Build Linux binary

20− if: runner.os == 'Linux'

21− run: |

22− docker run \

23− -v $(pwd):/mnt -w /mnt \

24− utdemir/ghc-musl:v14-ghc8101 \

25− bash .github/workflows/build.sh

26− - name: Build ${{ runner.os }} binary

27− if: runner.os != 'Linux'

28− shell: bash

29− run: |

30− cabal build exe:ormolu

31− cp $(find dist-newstyle $ -name ormolu -o -name ormolu.exe $ -type f) ormolu

32− - uses: actions/upload-release-asset@v1

33− env:

34− GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

35− with:

36− upload_url: ${{ github.event.release.upload_url }}

37− asset_path: ormolu

38− asset_name: ormolu-${{ runner.os }}

39− asset_content_type: application/octet-stream

.github/workflows/build.shdeleted−7

@@ -1,7 +0,0 @@

1−set -e

2−

3−cabal update

4−cabal build exe:ormolu --enable-executable-static --ghc-options="-split-sections"

5−

6−cp $(find dist-newstyle -name ormolu -type f) .

7−strip -s ormolu

CHANGELOG.md−137

@@ -1,126 +1,5 @@

1## Unreleased

2

3−* Ormolu no longer overwrites already formatted files. [PR

4− 649](https://github.com/tweag/ormolu/pull/649).

5−

6−## Ormolu 0.1.2.0

7−

8−* Fixed the bug when comments in different styles got glued together after

9− formatting. [Issue 589](https://github.com/tweag/ormolu/issues/589).

10−

11−* Added `-i` as a shortcut for `--mode inplace`. [Issue

12− 467](https://github.com/tweag/ormolu/issues/467).

13−

14−* Improved grouping of top-level declarations. [Issue

15− 466](https://github.com/tweag/ormolu/issues/466).

16−

17−## Ormolu 0.1.1.0

18−

19−* Imports in a import lists are now normalized: duplicate imports are

20− combined/eliminated intelligently.

21−

22−* Import declarations that can be merged are now automatically merged.

23− [Issue 414](https://github.com/tweag/ormolu/issues/414).

24−

25−* The magic comments for disabling and enabling Ormolu now can encompass any

26− fragment of code provided that the remaining code after exclusion of the

27− disabled part is still syntactically correct. [Issue

28− 601](https://github.com/tweag/ormolu/issues/601).

29−

30−* Improved sorting of operators in imports. [Issue

31− 602](https://github.com/tweag/ormolu/issues/602).

32−

33−* Fixed a bug related to trailing space in multiline comments in certain

34− cases. [Issue 603](https://github.com/tweag/ormolu/issues/602).

35−

36−* Added support for formatting linked lists with `(:)` as line terminator.

37− [Issue 478](https://github.com/tweag/ormolu/issues/478).

38−

39−* Fixed rendering of function arguments in multiline layout. [Issue

40− 609](https://github.com/tweag/ormolu/issues/609).

41−

42−* Blank lines between definitions in `let` and `while` bindings are now

43− preserved. [Issue 554](https://github.com/tweag/ormolu/issues/554).

44−

45−* Fixed the bug when type applications stuck to the `$` of TH splices that

46− followed them. [Issue 613](https://github.com/tweag/ormolu/issues/613).

47−

48−* Improved region formatting so that indented fragments—such as definitions

49− inside of `where` clauses—can be formatted. [Issue

50− 572](https://github.com/tweag/ormolu/issues/572).

51−

52−* Fixed the bug related to the de-association of pragma comments. [Issue

53− 619](https://github.com/tweag/ormolu/issues/619).

54−

55−## Ormolu 0.1.0.0

56−

57−* Fixed rendering of type signatures concerning several identifiers. [Issue

58− 566](https://github.com/tweag/ormolu/issues/566).

59−

60−* Fixed an idempotence issue with inline comments in tuples and parentheses.

61− [Issue 450](https://github.com/tweag/ormolu/issues/450).

62−

63−* Fixed an idempotence issue when certain comments were picked up as

64− “continuation” of a series of comments [Issue

65− 449](https://github.com/tweag/ormolu/issues/449).

66−

67−* Fixed an idempotence issue related to different indentation levels in a

68− comment series. [Issue 512](https://github.com/tweag/ormolu/issues/512).

69−

70−* Fixed an idempotence issue related to comments which may happen to be

71− separated from the elements they are attached to by the equality sign.

72− [Issue 340](https://github.com/tweag/ormolu/issues/340).

73−

74−* Fixed an idempotence issue with type synonym and data declarations where

75− the type has a Haddock. [Issue

76− 578](https://github.com/tweag/ormolu/issues/578).

77−

78−* Fix the false positive about AST differences in presence of comments with

79− multiple blank lines in a row. [Issue

80− 518](https://github.com/tweag/ormolu/issues/518).

81−

82−* Fixed rendering of comments around if expressions. [Issue

83− 458](https://github.com/tweag/ormolu/issues/458).

84−

85−* Unnamed fields of data constructors are now documented using the `-- ^`

86− syntax. [Issue 445](https://github.com/tweag/ormolu/issues/445) and [Issue

87− 428](https://github.com/tweag/ormolu/issues/428).

88−

89−* Fixed non-idempotent transformation of partly documented data definition.

90− [Issue 590](https://github.com/tweag/ormolu/issues/590).

91−

92−* Fixed an idempotence issue related to operators. [Issue

93− 522](https://github.com/tweag/ormolu/issues/522).

94−

95−* Renamed the `--check-idempotency` flag to `--check-idempotence`.

96− Apparently only the latter is correct.

97−

98−## Ormolu 0.0.5.0

99−

100−* Grouping of statements in `do`-blocks is now preserved. [Issue

101− 74](https://github.com/tweag/ormolu/issues/74).

102−

103−* Grouping of TH splices is now preserved. [Issue

104− 507](https://github.com/tweag/ormolu/issues/507).

105−

106−* Comments on pragmas are now preserved. [Issue

107− 216](https://github.com/tweag/ormolu/issues/216).

108−

109−* Ormolu can now be enabled and disabled via special comments. [Issue

110− 435](https://github.com/tweag/ormolu/issues/435).

111−

112−* Added experimental support for simple CPP. [Issue

113− 415](https://github.com/tweag/ormolu/issues/415).

114−

115−* Added two new options `--start-line` and `--end-line` that allow us to

116− select a region to format. [Issue

117− 516](https://github.com/tweag/ormolu/issues/516).

118−

119−* Fixed rendering of module headers in the presence of preceding comments or

120− Haddocks. [Issue 561](https://github.com/tweag/ormolu/issues/561).

121−

122−## Ormolu 0.0.4.0

123−

124* When given several files to format, Ormolu does not stop on the first

3* When given several files to format, Ormolu does not stop on the first

125 failure, but tries to format all the files. [Issue

4 failure, but tries to format all the files. [Issue

126 502](https://github.com/tweag/ormolu/issues/502).

5 502](https://github.com/tweag/ormolu/issues/502).

@@ -137,22 +16,6 @@

137 now put on its own line. [Issue

16 now put on its own line. [Issue

138 509](https://github.com/tweag/ormolu/issues/509).

17 509](https://github.com/tweag/ormolu/issues/509).

139

18

140−* Fixed the bug pertaining to rendering of arrow notation with multiline

141− expressions. [Issue 513](https://github.com/tweag/ormolu/issues/513).

142−

143−* Made rendering of data type definitions, value-level applications, and

144− application of types use the same style. Moreover, existential now doesn't

145− cause the data constructor be unconditionally rendered in multiline

146− layout. [Issue 427](https://github.com/tweag/ormolu/issues/427).

147−

148−* Records with a single data constructor are now formatted more compactly.

149− [Issue 425](https://github.com/tweag/ormolu/issues/425).

150−

151−* Switched to `ghc-lib-parser-8.10.1`.

152−

153−* Implemented support for the new language extensions `ImportQualifiedPost`

154− and `StandaloneKindSignatures`.

155−

156## Ormolu 0.0.3.1

19## Ormolu 0.0.3.1

157

20

158* Fixed rendering of record updates with the record dot preprocessor syntax

21* Fixed rendering of record updates with the record dot preprocessor syntax

CONTRIBUTING.md+1−1

@@ -30,7 +30,7 @@ Testing is performed as following:

30* The result of printing is parsed back again and the AST is compared to the

31 AST obtained from the original file. They should match.

32* The output of printer is checked against the expected output.

33−* Idempotence property is verified: formatting already formatted code

33+* Idempotency property is verified: formatting already formatted code

34 results in exactly the same output.

35

36Examples can be organized in sub-directories, see the existing ones for

DESIGN.md+11−11

@@ -24,7 +24,7 @@ It also includes recommendations for future implementers.

24

25We set for the following goals (mostly taken from

26[brittany](https://github.com/lspitzner/brittany)):

27−* Preserve the meaning of the formatted functions when no CPP is used;

27+* Preserve the meaning of the formatted functions;

28* Make reasonable use of screen space;

29* Use linear space and computation time on the size of the input;

30* Preserve comments;

@@ -140,10 +140,8 @@ impossible to guess what they'll be called on.

140

141### CPP

142

143−We allow CPP directives in the input, but we forgo the goal to preserve the

143+Formatting a module which uses CPP directives won't be supported. Instead,

144−meaning of the formatted functions in that case.

145−Instead of supporting CPP better, we hope for a solution to replace CPP to

146−do conditional compilation.

144+we hope for a solution to replace CPP to do conditional compilation.

147

145

148There are the following challenges when formatting a module with CPP:

146There are the following challenges when formatting a module with CPP:

149

147

@@ -182,15 +180,14 @@ True

182```

180```

183

181

184At the time of this writing, formatting this program with Hindent

182At the time of this writing, formatting this program with Hindent

185−or Ormolu produces the same output we would get if the CPP directives

183+produces the same output we would get if the CPP directives were

186−were considered comments:

184+considered comments:

187

185

188```

186```

189−$ ormolu --version

187+$ hindent --version

190−ormolu 0.0.5.0 HEAD fc64eada5c4da7a5b07d2872e253671b48aec115

188+hindent 5.2.7

191−using ghc-lib-parser 8.10.1.20200412

192

189

193−$ ormolu --mode inplace test.hs

190+$ hindent test.hs

194

191

195$ cat test.hs

192$ cat test.hs

196{-# LANGUAGE CPP #-}

193{-# LANGUAGE CPP #-}

@@ -230,6 +227,9 @@ to parse Haskell modules. If CPP is replaced with some language

230extension or mechanism to do conditional compilation, all tools

227extension or mechanism to do conditional compilation, all tools

231will benefit from it.

228will benefit from it.

232

229

230+Therefore, CPP won't be supported. If the CPP extension

231+is enabled, we should signal an error right away.

232+

233### Printing

234

235Just pretty-printing code (following the approach of Hindent) seems sane. It

README.md+13−54

@@ -22,7 +22,7 @@ the following goals in mind:

22* That formatting style aims to result in minimal diffs while still

23 remaining very close to “conventional” Haskell formatting people use.

24* Choose a style compatible with modern dialects of Haskell. As new Haskell

25− extensions enter broad use, we may change the style to accommodate them.

25+ extensions enter broad use, we may change the style to accomodate them.

26* Idempotence: formatting already formatted code doesn't change it.

27* Be well-tested and robust to the point that it can be used in large

28 projects without exposing unfortunate, disappointing bugs here and there.

@@ -45,12 +45,11 @@ Alternatively, `stack` could be used with a `stack.yaml` file as follows.

45

46```console

47$ cat stack.yaml

48−resolver: lts-16.0

48+resolver: lts-14.3

49packages:

50- '.'

51

52−$ stack build # to build

52+$ stack build

53−$ stack install # to install

54```

53```

55

54

56To use Ormolu directly from GitHub with Nix, this snippet may come in handy:

55To use Ormolu directly from GitHub with Nix, this snippet may come in handy:

@@ -92,44 +91,21 @@ formatted output.

92$ ormolu --mode inplace Module.hs

91$ ormolu --mode inplace Module.hs

93```

92```

94

93

95−Use `find` to format a tree recursively:

96−

97−```console

98−$ ormolu --mode inplace $(find . -name '*.hs')

99−```

100−

101−## Magic comments

102−

103−Ormolu understands two magic comments:

104−

105−```haskell

106−{- ORMOLU_DISABLE -}

107−```

108−

109−and

110−

111−```haskell

112−{- ORMOLU_ENABLE -}

113−```

114−

115−This allows us to disable formatting selectively for code between these

116−markers or disable it for the entire file. To achieve the latter, just put

117−`{- ORMOLU_DISABLE -}` at the very top. Note that the source code should

118−still be parseable even without the “excluded” part. Because of that the

119−magic comments cannot be placed arbitrary, but should rather enclose

120−independent top-level definitions.

121−

122## Current limitations

94## Current limitations

123

95

124−* CPP support is experimental. CPP is virtually impossible to handle

96+* Does not handle CPP (wontfix, see [the design document][design]).

125− correctly, so we process them as a sort of unchangeable snippets. This

126− works only in simple cases when CPP conditionals surround top-level

127− declarations. See the [CPP][design-cpp] section in the design notes for

128− a discussion of the dangers.

129* Input modules should be parsable by Haddock, which is a bit stricter

97* Input modules should be parsable by Haddock, which is a bit stricter

130 criterion than just being valid Haskell modules.

98 criterion than just being valid Haskell modules.

131* Various minor idempotence issues, most of them are related to comments.

99* Various minor idempotence issues, most of them are related to comments.

132

100

101+## Editor integration

102+

103+We know of the following editor integrations:

104+

105+* [Emacs][emacs-package]

106+* [VS Code][vs-code-plugin]

107+* vim: [neoformat][neoformat], [vim-ormolu][vim-ormolu]

108+

133## Running on Hackage

109## Running on Hackage

134

110

135It's possible to try Ormolu on arbitrary packages from Hackage. For that

111It's possible to try Ormolu on arbitrary packages from Hackage. For that

@@ -144,22 +120,6 @@ also contain formatted `.hs` files for inspection and original inputs with

144`.hs-original` extension (those are with CPP dropped, exactly what is fed

120`.hs-original` extension (those are with CPP dropped, exactly what is fed

145into Ormolu).

121into Ormolu).

146

122

147−## Editor integration

148−

149−We know of the following editor integrations:

150−

151−* [Emacs][emacs-package]

152−* [VS Code][vs-code-plugin]

153−* vim: [neoformat][neoformat], [vim-ormolu][vim-ormolu]

154−

155−## Arch Linux

156−

157−To install Ormolu on Arch Linux, one can use [the package on AUR][aur]:

158−

159−```console

160−yay -S ormolu

161−```

162−

163## Contributing

123## Contributing

164

124

165See [CONTRIBUTING.md][contributing].

125See [CONTRIBUTING.md][contributing].

@@ -170,7 +130,7 @@ See [LICENSE.md][license].

170

130

172

132

173−[design-cpp]: https://github.com/tweag/ormolu/blob/master/DESIGN.md#cpp

133+[design]: https://github.com/tweag/ormolu/blob/master/DESIGN.md#cpp

174[contributing]: https://github.com/tweag/ormolu/blob/master/CONTRIBUTING.md

134[contributing]: https://github.com/tweag/ormolu/blob/master/CONTRIBUTING.md

175[license]: https://github.com/tweag/ormolu/blob/master/LICENSE.md

135[license]: https://github.com/tweag/ormolu/blob/master/LICENSE.md

176[haskell-src-exts]: https://hackage.haskell.org/package/haskell-src-exts

136[haskell-src-exts]: https://hackage.haskell.org/package/haskell-src-exts

178[vs-code-plugin]: https://marketplace.visualstudio.com/items?itemName=sjurmillidahl.ormolu-vscode

138[vs-code-plugin]: https://marketplace.visualstudio.com/items?itemName=sjurmillidahl.ormolu-vscode

179[vim-ormolu]: https://github.com/sdiehl/vim-ormolu

139[vim-ormolu]: https://github.com/sdiehl/vim-ormolu

180[neoformat]: https://github.com/sbdchd/neoformat

140[neoformat]: https://github.com/sbdchd/neoformat

181−[aur]: https://aur.archlinux.org/packages/ormolu

app/Main.hs+36−48

@@ -4,11 +4,13 @@

4{-# LANGUAGE TemplateHaskell #-}

5{-# LANGUAGE TypeApplications #-}

6

7−module Main (main) where

7+module Main

8+ ( main,

9+ )

10+where

8

11

9import Control.Exception (SomeException, displayException, try)

12import Control.Exception (SomeException, displayException, try)

10import Control.Monad

13import Control.Monad

11−import Data.Bool (bool)

12import Data.Either (lefts)

14import Data.Either (lefts)

13import Data.List (intercalate, sort)

15import Data.List (intercalate, sort)

14import qualified Data.Text.IO as TIO

16import qualified Data.Text.IO as TIO

@@ -35,8 +37,9 @@ main = withPrettyOrmoluExceptions $ do

35 -- It is possible to get IOException, error's and 'OrmoluException's

37 -- It is possible to get IOException, error's and 'OrmoluException's

36 -- from 'formatOne', so we just catch everything.

38 -- from 'formatOne', so we just catch everything.

37 errs <-

39 errs <-

38− lefts <$> mapM (try @SomeException . formatOne' . Just) (sort xs)

40+ lefts <$> mapM (try @SomeException . formatOne' . Just) xs

39 unless (null errs) $ do

41 unless (null errs) $ do

42+ hPutStrLn stderr "Some files failed to format:\n"

40 mapM_ (hPutStrLn stderr . displayException) errs

43 mapM_ (hPutStrLn stderr . displayException) errs

41 exitWith (ExitFailure 102)

44 exitWith (ExitFailure 102)

42

45

@@ -45,7 +48,7 @@ formatOne ::

45 -- | Mode of operation

48 -- | Mode of operation

46 Mode ->

49 Mode ->

47 -- | Configuration

50 -- | Configuration

48− Config RegionIndices ->

51+ Config ->

49 -- | File to format or stdin as 'Nothing'

52 -- | File to format or stdin as 'Nothing'

50 Maybe FilePath ->

53 Maybe FilePath ->

51 IO ()

54 IO ()

@@ -61,19 +64,15 @@ formatOne mode config = \case

61 -- 101 is different from all the other exit codes we already use.

64 -- 101 is different from all the other exit codes we already use.

62 exitWith (ExitFailure 101)

65 exitWith (ExitFailure 101)

63 Just inputFile -> do

66 Just inputFile -> do

64− originalInput <- TIO.readFile inputFile

65− formattedInput <- ormoluFile config inputFile

67+ r <- ormoluFile config inputFile

66 case mode of

68 case mode of

67 Stdout ->

69 Stdout ->

68− TIO.putStr formattedInput

70+ TIO.putStr r

69− InPlace -> do

71+ InPlace ->

70− -- Only write when the contents have changed, in order to avoid

71− -- updating the modified timestamp if the file was already correctly

72− -- formatted.

73− when (formattedInput /= originalInput) $

74− TIO.writeFile inputFile formattedInput

72+ TIO.writeFile inputFile r

75 Check -> do

73 Check -> do

76− when (formattedInput /= originalInput) $

74+ r' <- TIO.readFile inputFile

75+ when (r /= r') $

77 -- 100 is different to all the other exit code that are emitted

76 -- 100 is different to all the other exit code that are emitted

78 -- either from an 'OrmoluException' or from 'error' and

77 -- either from an 'OrmoluException' or from 'error' and

79 -- 'notImplemented'.

78 -- 'notImplemented'.

@@ -82,14 +81,15 @@ formatOne mode config = \case

82----------------------------------------------------------------------------

81----------------------------------------------------------------------------

83-- Command line options parsing.

82-- Command line options parsing.

84

83

85−data Opts = Opts

84+data Opts

85+ = Opts

86− { -- | Mode of operation

86+ { -- | Mode of operation

87− optMode :: !Mode,

87+ optMode :: !Mode,

88− -- | Ormolu 'Config'

88+ -- | Ormolu 'Config'

89− optConfig :: !(Config RegionIndices),

89+ optConfig :: !Config,

90− -- | Haskell source files to format or stdin (when the list is empty)

90+ -- | Haskell source files to format or stdin (when the list is empty)

91− optInputFiles :: ![FilePath]

91+ optInputFiles :: ![FilePath]

92− }

92+ }

93

94-- | Mode of operation.

95data Mode

@@ -139,25 +139,20 @@ optsParserInfo =

139optsParser :: Parser Opts

140optsParser =

141 Opts

142− <$> ( (fmap (bool Stdout InPlace) . switch . mconcat)

143− [ short 'i',

144− help "A shortcut for --mode inplace"

145− ]

146− <|> (option parseMode . mconcat)

142+ <$> (option parseMode . mconcat)

147− [ long "mode",

143+ [ long "mode",

148− short 'm',

144+ short 'm',

149− metavar "MODE",

145+ metavar "MODE",

150− value Stdout,

146+ value Stdout,

151− help "Mode of operation: 'stdout' (default), 'inplace', or 'check'"

147+ help "Mode of operation: 'stdout' (default), 'inplace', or 'check'"

152− ]

148+ ]

153− )

154 <*> configParser

149 <*> configParser

155 <*> (many . strArgument . mconcat)

150 <*> (many . strArgument . mconcat)

156 [ metavar "FILE",

151 [ metavar "FILE",

157 help "Haskell source files to format or stdin (default)"

152 help "Haskell source files to format or stdin (default)"

158 ]

153 ]

159

154

160−configParser :: Parser (Config RegionIndices)

155+configParser :: Parser Config

161configParser =

156configParser =

162 Config

157 Config

163 <$> (fmap (fmap DynOption) . many . strOption . mconcat)

158 <$> (fmap (fmap DynOption) . many . strOption . mconcat)

@@ -177,22 +172,15 @@ configParser =

177 help "Output information useful for debugging"

172 help "Output information useful for debugging"

178 ]

173 ]

179 <*> (switch . mconcat)

174 <*> (switch . mconcat)

180− [ long "check-idempotence",

175+ [ long "tolerate-cpp",

176+ short 'p',

177+ help "Do not fail if CPP pragma is present"

178+ ]

179+ <*> (switch . mconcat)

180+ [ long "check-idempotency",

181 short 'c',

182 help "Fail if formatting is not idempotent"

183 ]

184− <*> ( RegionIndices

185− <$> (optional . option auto . mconcat)

186− [ long "start-line",

187− metavar "START",

188− help "Start line of the region to format (starts from 1)"

189− ]

190− <*> (optional . option auto . mconcat)

191− [ long "end-line",

192− metavar "END",

193− help "End line of the region to format (inclusive)"

194− ]

195− )

196

184

197----------------------------------------------------------------------------

185----------------------------------------------------------------------------

198-- Helpers

186-- Helpers

data/examples/declaration/annotation/annotation-out.hs−1

@@ -18,5 +18,4 @@ foo = 5

18

19data Foo = Foo Int

20{-# ANN type Foo ("HLint: ignore") #-}

21−

22{- Comment -}

21{- Comment -}

data/examples/declaration/data/existential-multiline-out.hs−12

@@ -3,15 +3,3 @@

3data Foo

4 = forall a. MkFoo a (a -> Bool)

5 | forall a. Eq a => MkBar a

6−

7−data Bar

8− = forall x y.

9− Bar x y x y

10−

11−data Baz

12− = forall x y.

13− Baz

14− x

15− y

16− x

17− y

data/examples/declaration/data/existential-multiline.hs−7

@@ -3,10 +3,3 @@

3data Foo

4 = forall a. MkFoo a (a -> Bool)

5 | forall a. Eq a => MkBar a

6−

7−data Bar = forall x y.

8− Bar x y x y

9−

10−data Baz = forall x y.

11− Baz x y x

12− y

data/examples/declaration/data/field-layout/gadt-out.hs+1−2

@@ -6,6 +6,5 @@ data Foo a b where

6 Foo :: Foo Int Int

7 -- | Something else

8 Bar ::

9− Foo

10− Char

9+ Foo Char

11 Char

10 Char

data/examples/declaration/data/field-layout/record-out.hs+9−9

@@ -1,12 +1,12 @@

1module Main where

2

3-- | Foo.

4−data Foo = Foo

4+data Foo

5+ = Foo

5− { -- | Something

6+ { -- | Something

6− foo :: Foo Int Int,

7+ foo :: Foo Int Int,

7− -- | Something else

8+ -- | Something else

8− bar ::

9+ bar ::

9− Bar

10− Char

10+ Bar Char

11− Char

11+ Char

12− }

12+ }

data/examples/declaration/data/gadt/multiline-out.hs+1−1

@@ -8,7 +8,7 @@ data Foo a where

8 Foo ::

9 forall a b.

10 (Show a, Eq b) => -- foo

11− -- bar

11+ -- bar

12 a ->

13 b ->

14 Foo 'Int

data/examples/declaration/data/multiline-arg-parens-out.hs+1−1

@@ -5,6 +5,6 @@ data Foo

5 = Foo

6 Bar

7 (Set Baz) -- and here we go

8− -- and that's it

8+ -- and that's it

9 Text

10 deriving (Eq)

data/examples/declaration/data/partly-documented-out.hsdeleted−5

@@ -1,5 +0,0 @@

1−data Optimisation

2− = PETransform

3− | -- | partial eval and associated transforms

4− GeneralisedNatHack

5− deriving (Show, Eq, Generic)

data/examples/declaration/data/partly-documented.hsdeleted−2

@@ -1,2 +0,0 @@

1−data Optimisation = PETransform | GeneralisedNatHack -- ^ partial eval and associated transforms

2− deriving (Show, Eq, Generic)

data/examples/declaration/data/record-multi-const-out.hsdeleted−17

@@ -1,17 +0,0 @@

1−module Main where

2−

3−-- | Something.

4−data Foo

5− = Foo

6− { -- | X

7− fooX :: Int,

8− -- | Y

9− fooY :: Int

10− }

11− | Bar

12− { -- | X

13− barX :: Int,

14− -- | Y

15− barY :: Int

16− }

17− deriving (Eq, Show)

data/examples/declaration/data/record-multi-const.hsdeleted−13

@@ -1,13 +0,0 @@

1−module Main where

2−

3−-- | Something.

4−

5−data Foo = Foo

6− { fooX :: Int -- ^ X

7− , fooY :: Int -- ^ Y

8− } |

9− Bar

10− { barX :: Int -- ^ X

11− , barY :: Int -- ^ Y

12− }

13− deriving (Eq, Show)

data/examples/declaration/data/record-out.hs+20−19

@@ -1,23 +1,24 @@

1module Main where

2

3-- | Something.

4−data Foo = Foo

4+data Foo

5+ = Foo

5− { -- | X

6+ { -- | X

6− fooX :: Int,

7+ fooX :: Int,

7− -- | Y

8+ -- | Y

8− fooY :: Int,

9+ fooY :: Int,

9− -- | BarBaz

10+ -- | BarBaz

10− fooBar, fooBaz :: NonEmpty (Identity Bool),

11+ fooBar, fooBaz :: NonEmpty (Identity Bool),

11− -- | GagGog

12+ -- | GagGog

12− fooGag,

13+ fooGag,

13− fooGog ::

14+ fooGog ::

14− NonEmpty

15+ NonEmpty

15− ( Indentity

16+ ( Indentity

16− Bool

17+ Bool

17− ),

18+ ),

18− -- | Huh!

19+ -- | Huh!

19− fooFoo,

20+ fooFoo,

20− barBar ::

21+ barBar ::

21− Int

22+ Int

22− }

23+ }

23 deriving (Eq, Show)

24 deriving (Eq, Show)

data/examples/declaration/data/simple-broken-out.hs+2−1

@@ -1,7 +1,8 @@

1module Main where

2

3-- | Here we go.

4+data Foo

4−data Foo = Foo {unFoo :: Int}

5+ = Foo {unFoo :: Int}

5 deriving (Eq)

6 deriving (Eq)

6

7

7-- | And once again.

8-- | And once again.

data/examples/declaration/data/unnamed-field-comment-0-out.hsdeleted−7

@@ -1,7 +0,0 @@

1−data Foo

2− = -- | Bar

3− Bar

4− Field1

5− -- ^ Field 1

6− Field2

7− -- ^ Field 2

data/examples/declaration/data/unnamed-field-comment-0.hsdeleted−5

@@ -1,5 +0,0 @@

1−data Foo

2− = -- | Bar

3− Bar

4− Field1 -- ^ Field 1

5− Field2 -- ^ Field 2

data/examples/declaration/data/unnamed-field-comment-1-out.hsdeleted−5

@@ -1,5 +0,0 @@

1−data X

2− = B

3− !Int

4− -- ^ y

5− C

data/examples/declaration/data/unnamed-field-comment-1.hsdeleted−4

@@ -1,4 +0,0 @@

1−data X

2− = B

3− !Int -- ^ y

4− C

data/examples/declaration/data/with-comment-out.hsdeleted−4

@@ -1,4 +0,0 @@

1−data A

2− = B -- C

3− | -- D

4− E

data/examples/declaration/data/with-comment.hsdeleted−5

@@ -1,5 +0,0 @@

1−data A =

2− B -- C

3−

4− -- D

5− | E

data/examples/declaration/data/with-weird-haddock-out.hsdeleted−4

@@ -1,4 +0,0 @@

1−data PlusLevel' t

2− = -- | @n + ℓ@.

3− Plus Integer (LevelAtom' t)

4− deriving (Show, Data)

data/examples/declaration/data/with-weird-haddock.hsdeleted−2

@@ -1,2 +0,0 @@

1−data PlusLevel' t = Plus Integer (LevelAtom' t) -- ^ @n + ℓ@.

2− deriving (Show, Data)

data/examples/declaration/instance/data-family-instances-gadt-out.hs+2−4

@@ -3,11 +3,9 @@

3

4data instance Bar Int a where

5 SameBar ::

6− Bar

6+ Bar Int

7− Int

8 Int

7 Int

9 CloseBar :: Bar Int Double

8 CloseBar :: Bar Int Double

10 OtherBar ::

9 OtherBar ::

11− Bar

10+ Bar Int

12− Int

13 a

11 a

data/examples/declaration/instance/data-family-instances-newtype-out.hs+4−3

@@ -1,6 +1,7 @@

1{-# LANGUAGE GADTSyntax #-}

2{-# LANGUAGE TypeFamilies #-}

3

4−newtype instance Foo [Double] = DoubleListFoo

4+newtype instance Foo [Double]

5+ = DoubleListFoo

5− { unDoubleListFoo :: Double

6+ { unDoubleListFoo :: Double

6− }

7+ }

data/examples/declaration/signature/type/infix-promoted-type-constructor-out.hs+1

@@ -1,2 +1,3 @@

1fun1 :: Def ('[Ref s (Stored Uint32), IBool] 'T.:-> IBool)

2+

2fun2 :: Def ('[Ref s (Stored Uint32), IBool] ':-> IBool)

3fun2 :: Def ('[Ref s (Stored Uint32), IBool] ':-> IBool)

data/examples/declaration/signature/type/multi-value-out.hs−5

@@ -2,11 +2,6 @@ foo, bar :: Int

2foo = 1

3bar = 2

4

5−a, b, c :: Int

6−a = 1

7−b = 2

8−c = 3

9−

10foo,

5foo,

11 bar,

6 bar,

12 baz ::

7 baz ::

data/examples/declaration/signature/type/multi-value.hs−5

@@ -2,11 +2,6 @@ foo, bar :: Int

2foo = 1

3bar = 2

4

5−a, b, c :: Int

6−a = 1

7−b = 2

8−c = 3

9−

10foo,

5foo,

11 bar,

6 bar,

12 baz :: Int

7 baz :: Int

data/examples/declaration/signature/type/unrelated-out.hsdeleted−4

@@ -1,4 +0,0 @@

1−clientFunc1 :: SomeType1

2−clientFunc2 :: SomeType2

3−clientFunc3 :: SomeType3

4−clientFunc1 :<|> clientFunc2 :<|> clientFunc3 = hoistClient foo bar baz

data/examples/declaration/signature/type/unrelated.hsdeleted−4

@@ -1,4 +0,0 @@

1−clientFunc1 :: SomeType1

2−clientFunc2 :: SomeType2

3−clientFunc3 :: SomeType3

4−clientFunc1 :<|> clientFunc2 :<|> clientFunc3 = hoistClient foo bar baz

data/examples/declaration/splice/grouped-splices-out.hsdeleted−3

@@ -1,3 +0,0 @@

1−$(deriveJSON fieldLabelMod ''A)

2−$(deriveJSON fieldLabelMod ''B)

3−$(deriveJSON fieldLabelMod ''C)

data/examples/declaration/splice/grouped-splices.hsdeleted−3

@@ -1,3 +0,0 @@

1−$(deriveJSON fieldLabelMod ''A)

2−$(deriveJSON fieldLabelMod ''B)

3−$(deriveJSON fieldLabelMod ''C)

data/examples/declaration/splice/splice-decl.hs+1

@@ -15,3 +15,4 @@ foo bar

15-- TemplateHaskell allows Q () at the top level

16do

17 pure []

18+

data/examples/declaration/type-synonyms/with-weird-haddock-out.hsdeleted−3

@@ -1,3 +0,0 @@

1−type Elims =

2− -- | eliminations ordered left-to-right.

3− [Elim]

data/examples/declaration/type-synonyms/with-weird-haddock.hsdeleted−1

@@ -1 +0,0 @@

1−type Elims = [Elim] -- ^ eliminations ordered left-to-right.

data/examples/declaration/value/function/arg-breakpoints-out.hsdeleted−3

@@ -1,3 +0,0 @@

1−foo

2− bar =

3− body

data/examples/declaration/value/function/arg-breakpoints.hsdeleted−3

@@ -1,3 +0,0 @@

1−foo

2− bar =

3− body

data/examples/declaration/value/function/arrow/multiline-case-out.hsdeleted−8

@@ -1,8 +0,0 @@

1−{-# LANGUAGE Arrows #-}

2−

3−f = proc x -> do

4− x <-

5− case x of X -> x

6− -<

7− y

8− a -< b

data/examples/declaration/value/function/arrow/multiline-case.hsdeleted−7

@@ -1,7 +0,0 @@

1−{-# LANGUAGE Arrows #-}

2−

3−f = proc x -> do

4− x

5− <- case x of X -> x

6− -< y

7− a -< b

data/examples/declaration/value/function/arrow/proc-applications-out.hs+5−6

@@ -9,11 +9,10 @@ bar f x =

9 w

10 )

11 ->

12− f -- The value

12+ f -< -- The value

13− -<

14− ( x, -- Foo

13+ ( x, -- Foo

15− w, -- Bar

14+ w, -- Bar

16− z -- Baz

15+ z -- Baz

17− )

16+ )

18

17

19baz x = proc a -> a -<< x

18baz x = proc a -> a -<< x

data/examples/declaration/value/function/arrow/proc-do-complex-out.hs+14−18

@@ -19,11 +19,10 @@ foo

19 )

20 ( b,

21 d

22+ ) -<

23+ ( b + 1, -- Funnel into arrow

24+ d * b

22 )

25 )

23− -<

24− ( b + 1, -- Funnel into arrow

25− d * b

26− )

27 if x `mod` y == 0 -- Basic condition

26 if x `mod` y == 0 -- Basic condition

28 then case e of -- Only left case is relevant

27 then case e of -- Only left case is relevant

29 Left

28 Left

@@ -33,20 +32,18 @@ foo

33 let v =

32 let v =

34 u -- Actually never used

33 u -- Actually never used

35 ^ 2

34 ^ 2

36− in ( returnA

35+ in ( returnA -<

37− -<

38− -- Just do the calculation

36+ -- Just do the calculation

39− (x + y * z)

37+ (x + y * z)

40 )

38 )

41 else do

39 else do

42 let u = x -- Let bindings bind expressions, not commands

40 let u = x -- Let bindings bind expressions, not commands

43− -- Could pattern match directly on x

41+ -- Could pattern match directly on x

44 i <- case u of

42 i <- case u of

45 0 -> (g . h -< u)

43 0 -> (g . h -< u)

46 n ->

44 n ->

47− ( ( h . g

45+ ( ( h . g -<

48− -<

49− y -- First actual use of y

46+ y -- First actual use of y

50 )

47 )

51 )

48 )

52 returnA -< ()

49 returnA -< ()

@@ -54,9 +51,8 @@ foo

54 if i > 0

51 if i > 0

55 then ma -< ()

52 then ma -< ()

56 else returnA -< ()

53 else returnA -< ()

57− returnA

54+ returnA -<

58− -<

59− ( i

55+ ( i

60− + x

56+ + x

61− * y -- Just do the calculation

57+ * y -- Just do the calculation

62− )

58+ )

data/examples/declaration/value/function/arrow/proc-do-simple1-out.hs+2−3

@@ -13,6 +13,5 @@ barbaz f g = proc (a, b) -> do

13

14bazbar f = proc a -> do

15 a <-

16− f

16+ f -<

17− -<

18− a

17+ a

data/examples/declaration/value/function/arrow/proc-do-simple2-out.hs+7−9

@@ -11,12 +11,10 @@ bazbaz f g h = proc (a, b, c) -> do

11 z <-

12 h

13 x

14− y

14+ y -<

15− -<

16− ( a,

15+ ( a,

17− b,

16+ b,

18− c

17+ c

19− )

18+ )

20− returnA

19+ returnA -<

21− -<

22− (x, y, z)

20+ (x, y, z)

data/examples/declaration/value/function/arrow/proc-forms1-out.hs+6−8

@@ -7,15 +7,13 @@ foo1 f g h x =

7 (|

8 test

9 ( h f

10− . h g

10+ . h g -<

11− -<

12− y x

11+ y x

13− . y z

12+ . y z

14 )

13 )

15 ( h g

14 ( h g

16− . h f

15+ . h f -<

17− -<

18− y z

16+ y z

19− . y x

17+ . y x

20 )

18 )

21 |)

19 |)

data/examples/declaration/value/function/arrow/proc-forms2-out.hs+4−6

@@ -12,11 +12,9 @@ bar2 f g h x =

12

13bar3 f g h x =

14 proc (y, z) ->

15− ( (h f . h g)

15+ ( (h f . h g) -<

16− -<

17− (y x) . y z

16+ (y x) . y z

18 ) |||

17 ) |||

19− ( (h g . h f)

18+ ( (h g . h f) -<

20− -<

21− y z . (y x)

19+ y z . (y x)

22 )

20 )

data/examples/declaration/value/function/arrow/proc-ifs-out.hs+4−6

@@ -6,10 +6,8 @@ bar f g = proc a ->

6 if f a

7 then

8 f

9− . g

9+ . g -<

10− -<

11− a

10+ a

12 else

11 else

13− g

12+ g -<

14− -<

15− b

13+ b

data/examples/declaration/value/function/arrow/proc-parentheses-out.hs+4−5

@@ -6,14 +6,13 @@ bar f g = proc a ->

6 ( ( (f)

7 ( g

8 )

9− )

9+ ) -<

10− -<

11− ( ( ( ( ( ( g

10+ ( ( ( ( ( ( g

12− a

11+ a

13− )

14 )

12 )

15 )

13 )

16 )

14 )

17 )

15 )

18 )

16 )

17+ )

19 )

18 )

data/examples/declaration/value/function/arrow/recursive-procs-out.hs+10−14

@@ -3,21 +3,17 @@

3foo f g = proc (x, y) -> do

4 rec a <- f y -< x

5 b <-

6− g x

6+ g x -<

7− -<

8− y

7+ y

9− bar

8+ bar -<

10− -<

11− ( a,

9+ ( a,

12− b

10+ b

13− )

11+ )

14 rec p <-

12 rec p <-

15 f

13 f

16− p

14+ p -<

17− -<

18− a

15+ a

19 rec q <-

16 rec q <-

20 g

17 g

21− q

18+ q -<

22− -<

23− b

19+ b

data/examples/declaration/value/function/awkward-comment-0-out.hsdeleted−6

@@ -1,6 +0,0 @@

1−mergeErrorReply :: ParseError -> Reply s u a -> Reply s u a

2−mergeErrorReply err1 reply -- XXX where to put it?

3− =

4− case reply of

5− Ok x state err2 -> Ok x state (mergeError err1 err2)

6− Error err2 -> Error (mergeError err1 err2)

data/examples/declaration/value/function/awkward-comment-0.hsdeleted−5

@@ -1,5 +0,0 @@

1−mergeErrorReply :: ParseError -> Reply s u a -> Reply s u a

2−mergeErrorReply err1 reply -- XXX where to put it?

3− = case reply of

4− Ok x state err2 -> Ok x state (mergeError err1 err2)

5− Error err2 -> Error (mergeError err1 err2)

data/examples/declaration/value/function/awkward-comment-1-out.hsdeleted−9

@@ -1,9 +0,0 @@

1−doForeign :: Vars -> [Name] -> [Term] -> Idris LExp

2−doForeign x = x

3− where

4− splitArg tm | (_, [_, _, l, r]) <- unApply tm -- pair, two implicits

5− =

6− do

7− let l' = toFDesc l

8− r' <- irTerm (sMN 0 "__foreignCall") vs env r

9− return (l', r')

data/examples/declaration/value/function/awkward-comment-1.hsdeleted−7

@@ -1,7 +0,0 @@

1−doForeign :: Vars -> [Name] -> [Term] -> Idris LExp

2−doForeign x = x

3− where

4− splitArg tm | (_, [_,_,l,r]) <- unApply tm -- pair, two implicits

5− = do let l' = toFDesc l

6− r' <- irTerm (sMN 0 "__foreignCall") vs env r

7− return (l', r')

data/examples/declaration/value/function/blank-lines-let-out.hsdeleted−6

@@ -1,6 +0,0 @@

1−foo =

2− let x = 10

3−

4− y = 11

5− z = 12

6− in x + y + z

data/examples/declaration/value/function/blank-lines-let.hsdeleted−9

@@ -1,9 +0,0 @@

1−foo =

2− let

3−

4− x = 10

5−

6− y = 11

7− z = 12

8−

9− in x + y + z

data/examples/declaration/value/function/blank-lines-where-out.hsdeleted−6

@@ -1,6 +0,0 @@

1−foo = x + y + z

2− where

3− x = 10

4−

5− y = 11

6− z = 12

data/examples/declaration/value/function/blank-lines-where.hsdeleted−7

@@ -1,7 +0,0 @@

1−foo = x + y + z

2− where

3−

4− x = 10

5−

6− y = 11

7− z = 12

data/examples/declaration/value/function/do/blocks-out.hs−1

@@ -15,5 +15,4 @@ readInClause = do

15 <|> do

16 optional g_Semi

17 void allspacing

18−

19 return things

18 return things

data/examples/declaration/value/function/do/comment-alignment-out.hsdeleted−9

@@ -1,9 +0,0 @@

1−main = do

2− case blah of

3− Nothing -> return ()

4− Just xs -> do

5− forM_ xs foo

6− bar

7− -- and here it goes

8− unless bobla $

9− quux

data/examples/declaration/value/function/do/comment-alignment.hsdeleted−9

@@ -1,9 +0,0 @@

1−main = do

2− case blah of

3− Nothing -> return ()

4− Just xs -> do

5− forM_ xs foo

6− bar

7− -- and here it goes

8− unless bobla $

9− quux

data/examples/declaration/value/function/do/comment-spacing-out.hsdeleted−7

@@ -1,7 +0,0 @@

1−foo = do

2− bar

3− -- foo

4− baz

5−

6− -- foo

7− quux

data/examples/declaration/value/function/do/comment-spacing.hsdeleted−7

@@ -1,7 +0,0 @@

1−foo = do

2− bar

3− -- foo

4− baz

5−

6− -- foo

7− quux

data/examples/declaration/value/function/do/recursive-do-rec-out.hs−5

@@ -2,18 +2,13 @@

2

3foo = do

4 rec a <- b + 5

5−

6 let d = c

5 let d = c

7−

8 b <- a * 5

6 b <- a * 5

9−

10 something

7 something

11−

12 c <- a + b

8 c <- a + b

13 print c

9 print c

14 rec something $ do

10 rec something $ do

15 x <- a

11 x <- a

16 print x

12 print x

17−

18 y <- c

13 y <- c

19 print y

14 print y

data/examples/declaration/value/function/if-with-comment-out.hsdeleted−8

@@ -1,8 +0,0 @@

1−foo =

2− if undefined

3− then -- then comment

4− undefined

5− else -- else comment

6−

7− do

8− undefined

data/examples/declaration/value/function/if-with-comment.hsdeleted−8

@@ -1,8 +0,0 @@

1−foo =

2− if undefined

3− -- then comment

4− then undefined

5− -- else comment

6− else

7− do

8− undefined

data/examples/declaration/value/function/implicit-params-out.hs−1

@@ -1,7 +1,6 @@

1{-# LANGUAGE ImplicitParams #-}

2

3sortBy :: (a -> a -> Bool) -> [a] -> [a]

4−sortBy = undefined

5

4

6sort :: (?cmp :: a -> a -> Bool) => [a] -> [a]

5sort :: (?cmp :: a -> a -> Bool) => [a] -> [a]

7sort = sortBy ?cmp

6sort = sortBy ?cmp

data/examples/declaration/value/function/implicit-params.hs−2

@@ -1,8 +1,6 @@

1{-# LANGUAGE ImplicitParams #-}

2sortBy :: (a -> a -> Bool) -> [a] -> [a]

3

4−sortBy = undefined

5−

6sort :: (?cmp :: a -> a -> Bool) => [a] -> [a]

4sort :: (?cmp :: a -> a -> Bool) => [a] -> [a]

7sort = sortBy ?cmp

5sort = sortBy ?cmp

8

6

data/examples/declaration/value/function/infix/dollar-chains-out.hs+3−3

@@ -13,9 +13,9 @@ foo =

13 throwIO (OrmoluCppEnabled path)

14

15foo =

16− bar $

16+ bar

17− baz $

17+ $ baz

18− quux

18+ $ quux

19

20x =

21 case l of { A -> B } $

data/examples/declaration/value/function/infix/hanging-out.hs+3−4

@@ -17,7 +17,6 @@ update =

17 `catch` \case

18 a -> a

19

20−foo =

20+foo = bar

21− bar

22− ++ case foo of

21+ ++ case foo of

23− a -> a

22+ a -> a

data/examples/declaration/value/function/infix/lenses-out.hs+7−8

@@ -1,12 +1,11 @@

1lenses =

2− Just $

3− M.fromList $

2+ Just $ M.fromList $

4− "type" .= ("user.connection" :: Text)

3+ "type" .= ("user.connection" :: Text)

5− # "connection" .= uc

4+ # "connection" .= uc

6− # "user" .= case name of

5+ # "user" .= case name of

7− Just n -> Just $ object ["name" .= n]

6+ Just n -> Just $ object ["name" .= n]

8− Nothing -> Nothing

7+ Nothing -> Nothing

9− # []

8+ # []

10

9

11foo =

10foo =

12 a

11 a

data/examples/declaration/value/function/let-single-line-out.hs+1

@@ -5,6 +5,7 @@ foo x = let x = z where { z = 2 }; a = 3 in x

5foo x = let g :: Int -> Int; g = id in ()

6

7let a = b; c = do { foo; bar }; d = baz in b

8+

8let a = case True of { True -> foo; False -> bar }; b = foo a in b

9let a = case True of { True -> foo; False -> bar }; b = foo a in b

9

10

10foo x = let ?g = id; ?f = g in x

11foo x = let ?g = id; ?f = g in x

data/examples/declaration/value/function/list-notation-0-out.hsdeleted−5

@@ -1,5 +0,0 @@

1−foo =

2− testCase "Foo" testFoo :

3− testCase "Bar" testBar :

4− testCase "Baz" testBaz :

5− []

data/examples/declaration/value/function/list-notation-0.hsdeleted−5

@@ -1,5 +0,0 @@

1−foo =

2− testCase "Foo" testFoo :

3− testCase "Bar" testBar :

4− testCase "Baz" testBaz :

5− []

data/examples/declaration/value/function/list-notation-1-out.hsdeleted−8

@@ -1,8 +0,0 @@

1−instance A.ToJSON UpdateTable where

2− toJSON a =

3− A.object $

4− "TableName" .= updateTableName a :

5− "ProvisionedThroughput" .= updateProvisionedThroughput a :

6− case updateGlobalSecondaryIndexUpdates a of

7− [] -> []

8− l -> ["GlobalSecondaryIndexUpdates" .= l]

data/examples/declaration/value/function/list-notation-1.hsdeleted−7

@@ -1,7 +0,0 @@

1−instance A.ToJSON UpdateTable where

2− toJSON a = A.object

3− $ "TableName" .= updateTableName a

4− : "ProvisionedThroughput" .= updateProvisionedThroughput a

5− : case updateGlobalSecondaryIndexUpdates a of

6− [] -> []

7− l -> [ "GlobalSecondaryIndexUpdates" .= l ]

data/examples/declaration/value/function/list-notation-2-out.hsdeleted−6

@@ -1,6 +0,0 @@

1−-- A list of the element and all its parents up to the root node.

2−getPath tree t =

3− t :

4− case Map.lookup (getId t) tree of

5− Nothing -> []

6− Just parent -> getPath tree parent

data/examples/declaration/value/function/list-notation-2.hsdeleted−5

@@ -1,5 +0,0 @@

1−-- A list of the element and all its parents up to the root node.

2−getPath tree t = t :

3− case Map.lookup (getId t) tree of

4− Nothing -> []

5− Just parent -> getPath tree parent

data/examples/declaration/value/function/list-notation-3-out.hsdeleted−12

@@ -1,12 +0,0 @@

1−foo =

2− reportSDoc "tc.cc" 30 $

3− sep $ do

4− (prettyTCM q <+> " before compilation") : do

5− map (prettyTCM . map unArg . clPats) cls

6−

7−foo =

8− reportSDoc "tc.cc" 30 $

9− sep $ do

10− (prettyTCM q <+> " before compilation") :

11− do

12− map (prettyTCM . map unArg . clPats) cls

data/examples/declaration/value/function/list-notation-3.hsdeleted−10

@@ -1,10 +0,0 @@

1−foo =

2− reportSDoc "tc.cc" 30 $ sep $ do

3− (prettyTCM q <+> " before compilation") : do

4− map (prettyTCM . map unArg . clPats) cls

5−

6−foo =

7− reportSDoc "tc.cc" 30 $ sep $ do

8− (prettyTCM q <+> " before compilation") :

9− do

10− map (prettyTCM . map unArg . clPats) cls

data/examples/declaration/value/function/multi-way-if-out.hs+2−4

@@ -3,14 +3,12 @@

3foo x = if | x == 5 -> 5

4

5bar x y =

6− if

7− | x > y -> x

6+ if | x > y -> x

8 | x < y ->

7 | x < y ->

9 y

8 y

10 | otherwise -> x

9 | otherwise -> x

11

10

12baz =

11baz =

13− if

12+ if | p -> f

14− | p -> f

15 | otherwise -> g

13 | otherwise -> g

16 x

14 x

data/examples/declaration/value/function/operator-comments-2-out.hsdeleted−5

@@ -1,5 +0,0 @@

1−x =

2− y

3− ++ f g -- commentA

4− -- commentB

5− -- commentC

data/examples/declaration/value/function/operator-comments-2.hsdeleted−4

@@ -1,4 +0,0 @@

1−x =

2− y ++ -- commentA

3−-- commentB

4− f g -- commentC

data/examples/declaration/value/function/operators-1-out.hsdeleted−4

@@ -1,4 +0,0 @@

1−foo =

2− f

3− . g

4− =<< h . i

data/examples/declaration/value/function/operators-1.hsdeleted−3

@@ -1,3 +0,0 @@

1−foo = f

2− . g

3− =<< h . i

data/examples/declaration/value/function/operators-2-out.hsdeleted−4

@@ -1,4 +0,0 @@

1−foo n

2− | x || y && z || n ** x

3− || x && n =

4− 42

data/examples/declaration/value/function/operators-2.hsdeleted−4

@@ -1,4 +0,0 @@

1−foo n

2− | x || y && z || n ** x

3− || x && n =

4− 42

data/examples/declaration/value/function/operators-3-out.hsdeleted−3

@@ -1,3 +0,0 @@

1−foo =

2− op <> n <+> colon <+> prettySe <+> text "="

3− <+> prettySe <> text sc

data/examples/declaration/value/function/operators-3.hsdeleted−3

@@ -1,3 +0,0 @@

1−foo =

2− op <> n <+> colon <+> prettySe <+> text "=" <+>

3− prettySe <> text sc

data/examples/declaration/value/function/operators-4-out.hsdeleted−3

@@ -1,3 +0,0 @@

1−foo =

2− line <> bindingOf <+> text "=" <+> tPretty <+> colon

3− <+> align <> prettyPs

data/examples/declaration/value/function/operators-4.hsdeleted−3

@@ -1,3 +0,0 @@

1−foo =

2− line <> bindingOf <+> text "=" <+> tPretty <+> colon <+>

3− align <> prettyPs

data/examples/declaration/value/function/operators-5-out.hsdeleted−5

@@ -1,5 +0,0 @@

1−foo =

2− map bar $

3− [ baz

4− ]

5− ++ quux

data/examples/declaration/value/function/operators-5.hsdeleted−4

@@ -1,4 +0,0 @@

1−foo =

2− map bar $

3− [ baz

4− ] ++ quux

data/examples/declaration/value/function/operators-6-out.hsdeleted−9

@@ -1,9 +0,0 @@

1−type PermuteRef =

2− "a"

3− :> ( "b" :> "c" :> End

4− :<|> "c" :> "b" :> End

5− )

6− :<|> "b"

7− :> ( "a" :> "c" :> End

8− :<|> "c" :> "a" :> End

9− )

data/examples/declaration/value/function/operators-6.hsdeleted−7

@@ -1,7 +0,0 @@

1−type PermuteRef =

2− "a" :> ( "b" :> "c" :> End

3− :<|> "c" :> "b" :> End

4− )

5− :<|> "b" :> ( "a" :> "c" :> End

6− :<|> "c" :> "a" :> End

7− )

data/examples/declaration/value/function/operators-0-out.hs → data/examples/declaration/value/function/operators-out.hsrenamed

The content is the same.

data/examples/declaration/value/function/operators-0.hs → data/examples/declaration/value/function/operators.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/constructors-out.hs → data/examples/declaration/value/function/record-constructors-out.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/constructors.hs → data/examples/declaration/value/function/record-constructors.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/dot-out.hs → data/examples/declaration/value/function/record-dot-out.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/dot.hs → data/examples/declaration/value/function/record-dot.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/inter-comments-out.hs → data/examples/declaration/value/function/record-inter-comments-out.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/inter-comments.hs → data/examples/declaration/value/function/record-inter-comments.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/updaters-out.hs → data/examples/declaration/value/function/record-updaters-out.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/updaters.hs → data/examples/declaration/value/function/record-updaters.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/wildcards-out.hs → data/examples/declaration/value/function/record-wildcards-out.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/wildcards.hs → data/examples/declaration/value/function/record-wildcards.hsrenamed

The content is the same.

data/examples/declaration/value/function/record/hanging-after-equal-con-out.hsdeleted−8

@@ -1,8 +0,0 @@

1−foo =

2− Foo

3− { foo =

4− -- Blah blah

5− case x of

6− Blah -> 1

7− Blim -> 2

8− }

data/examples/declaration/value/function/record/hanging-after-equal-con.hsdeleted−7

@@ -1,7 +0,0 @@

1−foo = Foo

2− { foo =

3− -- Blah blah

4− case x of

5− Blah -> 1

6− Blim -> 2

7− }

data/examples/declaration/value/function/record/hanging-after-equal-update-out.hsdeleted−8

@@ -1,8 +0,0 @@

1−foo =

2− foo

3− { foo =

4− -- Blah blah

5− case x of

6− Blah -> 1

7− Blim -> 2

8− }

data/examples/declaration/value/function/record/hanging-after-equal-update.hsdeleted−7

@@ -1,7 +0,0 @@

1−foo = foo

2− { foo =

3− -- Blah blah

4− case x of

5− Blah -> 1

6− Blim -> 2

7− }

data/examples/declaration/value/function/type-applications-and-splice-out.hsdeleted−4

@@ -1,4 +0,0 @@

1−{-# LANGUAGE TemplateHaskell #-}

2−{-# LANGUAGE TypeApplications #-}

3−

4−staticKey name = [|sing @ $(symFQN name)|]

data/examples/declaration/value/function/type-applications-and-splice.hsdeleted−4

@@ -1,4 +0,0 @@

1−{-# LANGUAGE TemplateHaskell #-}

2−{-# LANGUAGE TypeApplications #-}

3−

4−staticKey name = [| sing @ $(symFQN name) |]

data/examples/import/deduplication-bug-out.hsdeleted−6

@@ -1,6 +0,0 @@

1−import Foo1 (Bar1 (..), Baz1)

2−import Foo2 (Bar2 (..), Baz2)

3−import Foo3 (Bar3 (x1, x2, x3))

4−import Foo4 (Bar4 (x1, x2))

5−import Foo5 (Bar5 (x1))

6−import Foo6 (Bar6 (..))

data/examples/import/deduplication-bug.hsdeleted−6

@@ -1,6 +0,0 @@

1−import Foo1 (Bar1, Baz1, Bar1(..))

2−import Foo2 (Bar2(..), Baz2, Bar2)

3−import Foo3 (Bar3(x1,x3), Bar3(x1, x2))

4−import Foo4 (Bar4(x1), Bar4(x2))

5−import Foo5 (Bar5, Bar5(x1))

6−import Foo6 (Bar6(x1), Bar6(..))

data/examples/import/explicit-imports-out.hs+4−4

@@ -1,12 +1,12 @@

1import qualified MegaModule as M

2+ ( (<<<),

3+ (>>>),

2− ( Either,

4+ Either,

3 Maybe (Just, Nothing),

5 Maybe (Just, Nothing),

4 MaybeT (..),

6 MaybeT (..),

5− Monad (return, (>>), (>>=)),

7+ Monad ((>>), (>>=), return),

6 MonadBaseControl,

8 MonadBaseControl,

7 join,

9 join,

8 liftIO,

10 liftIO,

9 void,

11 void,

10− (<<<),

11− (>>>),

12 )

data/examples/import/explicit-imports-with-comments-out.hs+2−3

@@ -1,7 +1,6 @@

1import qualified MegaModule as M

2 ( -- (1)

3− -- (2)

4− Either, -- (3)

5− (<<<),

3+ (<<<), -- (2)

6 (>>>),

4 (>>>),

5+ Either, -- (3)

7 )

6 )

data/examples/import/merging-0-out.hsdeleted−3

@@ -1,3 +0,0 @@

1−import Foo

2−import Foo (bar, foo)

3−import Foo as F

data/examples/import/merging-0.hsdeleted−4

@@ -1,4 +0,0 @@

1−import Foo

2−import Foo (foo)

3−import Foo (bar)

4−import Foo as F

data/examples/import/merging-1-out.hsdeleted−2

@@ -1,2 +0,0 @@

1−import "bar" Foo (bar)

2−import "foo" Foo (baz, foo)

data/examples/import/merging-1.hsdeleted−3

@@ -1,3 +0,0 @@

1−import "foo" Foo (foo)

2−import "bar" Foo (bar)

3−import "foo" Foo (baz)

data/examples/import/merging-2-out.hsdeleted−2

@@ -1,2 +0,0 @@

1−import Foo hiding (bar4, foo2)

2−import qualified Foo (bar3, foo1)

data/examples/import/merging-2.hsdeleted−4

@@ -1,4 +0,0 @@

1−import qualified Foo (foo1)

2−import Foo hiding (foo2)

3−import qualified Foo (bar3)

4−import Foo hiding (bar4)

data/examples/import/misc-out.hs+6

@@ -1,5 +1,11 @@

1import A hiding

2 ( foobarbazqux,

3+ foobarbazqux,

4+ foobarbazqux,

5+ foobarbazqux,

6+ foobarbazqux,

7+ foobarbazqux,

8+ foobarbazqux,

3 )

9 )

4import {-# SOURCE #-} safe qualified Module as M hiding (a, b, c, d, e, f)

10import {-# SOURCE #-} safe qualified Module as M hiding (a, b, c, d, e, f)

5import Name hiding ()

11import Name hiding ()

data/examples/import/nested-explicit-imports-out.hs+6−6

@@ -1,10 +1,10 @@

1import qualified MegaModule as M

2+ ( (<<<),

3+ (>>>),

2− ( Either,

4+ Either,

3 Monad

5 Monad

4− ( return,

5− (>>),

6+ ( (>>),

6− (>>=)

7+ (>>=),

8+ return

7 ),

9 ),

8− (<<<),

9− (>>>),

10 )

data/examples/import/qualified-post-out.hs+1−1

@@ -1,5 +1,5 @@

1{-# LANGUAGE ImportQualifiedPost #-}

2

3+import Data.Text qualified as T

3import Data.Text qualified (a, b, c)

4import Data.Text qualified (a, b, c)

4import Data.Text qualified hiding (a, b, c)

5import Data.Text qualified hiding (a, b, c)

5−import Data.Text qualified as T

data/examples/import/qualified-prelude-out.hs+1−1

@@ -1,4 +1,4 @@

1module P where

2

3−import Prelude hiding (id, (.))

4import qualified Prelude

3import qualified Prelude

4+import Prelude hiding ((.), id)

data/examples/import/simple-out.hs+3−2

@@ -1,5 +1,6 @@

1import Data.Text

2+import Data.Text

3+import qualified Data.Text as T

4+import qualified Data.Text (a, b, c)

2import Data.Text (a, b, c)

5import Data.Text (a, b, c)

3import Data.Text hiding (a, b, c)

6import Data.Text hiding (a, b, c)

4−import qualified Data.Text (a, b, c)

5−import qualified Data.Text as T

data/examples/import/simple.hs−2

@@ -4,5 +4,3 @@ import qualified Data.Text as T

4import qualified Data.Text (a, c, b)

5import Data.Text (a, b, c)

6import Data.Text hiding (c, b, a)

7−import Data.Text (a, b, c, b, a)

8−import Data.Text hiding (c, b, a, b, c)

data/examples/import/sorted-export-list-out.hsdeleted−1

@@ -1 +0,0 @@

1−import Linear.Vector (Additive (..), (*^), (^*))

data/examples/import/sorted-export-list.hsdeleted−1

@@ -1 +0,0 @@

1−import Linear.Vector (Additive (..), (*^), (^*))

data/examples/module-header/double-dot-with-names-out.hs+6−1

@@ -1,6 +1,11 @@

1{-# LANGUAGE PatternSynonyms #-}

2

3−module ExportSyntax (A (.., NoA), Q (F, ..), G (T, .., U)) where

3+module ExportSyntax

4+ ( A (.., NoA),

5+ Q (F, ..),

6+ G (T, .., U),

7+ )

8+where

4

9

5data A = A | B

10data A = A | B

6

11

data/examples/module-header/multiline2.hs+2−2

@@ -1,2 +1,2 @@

1−module Foo (

1+module Foo

2− foo, bar, baz) where

2+ (foo, bar, baz) where

data/examples/module-header/preceding-comment-with-haddock-out.hs+4−1

@@ -3,6 +3,9 @@

3-}

4

5-- | This is the module's Haddock.

6−module Main (main) where

6+module Main

7+ ( main,

8+ )

9+where

7

10

8main = return ()

11main = return ()

data/examples/module-header/singleline-empty-out.hs+4−1

@@ -1,2 +1,5 @@

1-- | This demonstrates a BUG.

2−module Foo () where

2+module Foo

3+ (

4+ )

5+where

data/examples/module-header/stack-header-0-out.hs−1

@@ -3,5 +3,4 @@

3{-# LANGUAGE OverloadedStrings #-}

4

5main = return ()

6−

7-- stack runhaskell

6-- stack runhaskell

data/examples/module-header/stack-header-1-out.hs−1

@@ -4,5 +4,4 @@

4{-# LANGUAGE OverloadedStrings #-}

5

6main = return ()

7−

8-- stack runhaskell

7-- stack runhaskell

data/examples/module-header/warning-pragma-multiline-out.hs+4−1

@@ -1,6 +1,9 @@

1module Test

2 {-# DEPRECATED "This module is unstable" #-}

3− (foo, bar, baz)

3+ ( foo,

4+ bar,

5+ baz,

6+ )

4where

7where

5

8

6import Blah

9import Blah

data/examples/module-header/warning-pragma-out.hs+1−3

@@ -1,5 +1,3 @@

1module Test

2− {-# WARNING

3− "This module is very internal"

2+ {-# WARNING "This module is very internal" #-}

4− #-}

5where

3where

data/examples/other/argument-comment-out.hs+1

@@ -9,6 +9,7 @@ foo ::

9 -- | Foo

10 Int ->

11 Int

12+

12foo ::

13foo ::

13 Foo a =>

14 Foo a =>

14 -- | Foo

15 -- | Foo

data/examples/other/comment-glued-together-out.hsdeleted−7

@@ -1,7 +0,0 @@

1−module Main (main) where

2−

3−-- | Foo.

4−

5−-- Bar

6−main :: IO ()

7−main = return ()

data/examples/other/comment-glued-together.hsdeleted−6

@@ -1,6 +0,0 @@

1−module Main (main) where

2−

3−{- | Foo. -}

4−-- Bar

5−main :: IO ()

6−main = return ()

data/examples/other/comment-trailing-space-out.hsdeleted−9

@@ -1,9 +0,0 @@

1−data T

2− = {-

3−

4− some multi-line comment

5−

6− with empty lines

7−

8− -}

9− A

data/examples/other/comment-trailing-space.hsdeleted−9

@@ -1,9 +0,0 @@

1−data T

2− = {-

3−

4− some multi-line comment

5−

6− with empty lines

7−

8− -}

9− A

data/examples/other/comment-two-blocks-out.hs+2−2

@@ -2,7 +2,7 @@ newNames :: [(String, String)]

2newNames =

3 let (*) = flip (,)

4 in [ "Control" * "Monad"

5− -- Foo

5+ -- Foo

6

7− -- Bar

7+ -- Bar

8 ]

data/examples/other/cpp/continuation-out.hsdeleted−8

@@ -1,8 +0,0 @@

1−foo :: Int

2−foo = 5

3−

4−#define message_for(a, b) \

5− printf(#a " and " #b ": We love you!\n")

6−

7−bar :: Int

8−bar = 6

data/examples/other/cpp/continuation.hsdeleted−8

@@ -1,8 +0,0 @@

1−foo :: Int

2−foo = 5

3−

4−#define message_for(a, b) \

5− printf(#a " and " #b ": We love you!\n")

6−

7−bar :: Int

8−bar = 6

data/examples/other/cpp/lonely-hash-out.hsdeleted−14

@@ -1,14 +0,0 @@

1−module Main (main) where

2−

3−import Data.Void

4−import Test.Hspec

5−import Test.Hspec.Megaparsec

6−import Text.Megaparsec

7−import Text.Megaparsec.Char

8−

9−# if !MIN_VERSION_base(4,13,0)

10−import Data.Semigroup ((<>))

11−# endif

12−

13−main :: IO ()

14−main = return ()

data/examples/other/cpp/lonely-hash.hsdeleted−14

@@ -1,14 +0,0 @@

1−module Main (main) where

2−

3−import Data.Void

4−import Test.Hspec

5−import Test.Hspec.Megaparsec

6−import Text.Megaparsec

7−import Text.Megaparsec.Char

8−

9−# if !MIN_VERSION_base(4,13,0)

10−import Data.Semigroup ((<>))

11−# endif

12−

13−main :: IO ()

14−main = return ()

data/examples/other/cpp/separation-0a-out.hsdeleted−7

@@ -1,7 +0,0 @@

1−instance Stream s => Monad (ParsecT e s m) where

2− return = pure

3− (>>=) = pBind

4−

5−#if !(MIN_VERSION_base(4,13,0))

6− fail = Fail.fail

7−#endif

data/examples/other/cpp/separation-0a.hsdeleted−6

@@ -1,6 +0,0 @@

1−instance Stream s => Monad (ParsecT e s m) where

2− return = pure

3− (>>=) = pBind

4−#if !(MIN_VERSION_base(4,13,0))

5− fail = Fail.fail

6−#endif

data/examples/other/cpp/separation-0b-out.hsdeleted−10

@@ -1,10 +0,0 @@

1−instance Stream s => Monad (ParsecT e s m) where

2− return = pure

3− (>>=) = pBind

4−

5−#if !(MIN_VERSION_base(4,13,0))

6− fail = Fail.fail

7−#endif

8−

9−foo :: Int

10−foo = undefined

data/examples/other/cpp/separation-0b.hsdeleted−9

@@ -1,9 +0,0 @@

1−instance Stream s => Monad (ParsecT e s m) where

2− return = pure

3− (>>=) = pBind

4−#if !(MIN_VERSION_base(4,13,0))

5− fail = Fail.fail

6−#endif

7−

8−foo :: Int

9−foo = undefined

data/examples/other/cpp/separation-1a-out.hsdeleted−12

@@ -1,12 +0,0 @@

1−decompressingPipe ::

2− (PrimMonad m, MonadThrow m, MonadResource m) =>

3− CompressionMethod ->

4− ConduitT ByteString ByteString m ()

5−decompressingPipe Store = C.awaitForever C.yield

6−decompressingPipe Deflate = Z.decompress $ Z.WindowBits (-15)

7−

8−#ifdef ENABLE_BZIP2

9−decompressingPipe BZip2 = BZ.bunzip2

10−#else

11−decompressingPipe BZip2 = throwM BZip2Unsupported

12−#endif

data/examples/other/cpp/separation-1a.hsdeleted−11

@@ -1,11 +0,0 @@

1−decompressingPipe

2− :: (PrimMonad m, MonadThrow m, MonadResource m)

3− => CompressionMethod

4− -> ConduitT ByteString ByteString m ()

5−decompressingPipe Store = C.awaitForever C.yield

6−decompressingPipe Deflate = Z.decompress $ Z.WindowBits (-15)

7−#ifdef ENABLE_BZIP2

8−decompressingPipe BZip2 = BZ.bunzip2

9−#else

10−decompressingPipe BZip2 = throwM BZip2Unsupported

11−#endif

data/examples/other/cpp/separation-1b-out.hsdeleted−15

@@ -1,15 +0,0 @@

1−decompressingPipe ::

2− (PrimMonad m, MonadThrow m, MonadResource m) =>

3− CompressionMethod ->

4− ConduitT ByteString ByteString m ()

5−decompressingPipe Store = C.awaitForever C.yield

6−decompressingPipe Deflate = Z.decompress $ Z.WindowBits (-15)

7−

8−#ifdef ENABLE_BZIP2

9−decompressingPipe BZip2 = BZ.bunzip2

10−#else

11−decompressingPipe BZip2 = throwM BZip2Unsupported

12−#endif

13−

14−foo :: Int

15−foo = undefined

data/examples/other/cpp/separation-1b.hsdeleted−14

@@ -1,14 +0,0 @@

1−decompressingPipe

2− :: (PrimMonad m, MonadThrow m, MonadResource m)

3− => CompressionMethod

4− -> ConduitT ByteString ByteString m ()

5−decompressingPipe Store = C.awaitForever C.yield

6−decompressingPipe Deflate = Z.decompress $ Z.WindowBits (-15)

7−#ifdef ENABLE_BZIP2

8−decompressingPipe BZip2 = BZ.bunzip2

9−#else

10−decompressingPipe BZip2 = throwM BZip2Unsupported

11−#endif

12−

13−foo :: Int

14−foo = undefined

data/examples/other/cpp/separation-2a-out.hsdeleted−9

@@ -1,9 +0,0 @@

1−ffff, ffffffff :: Natural

2−

3−#ifdef HASKELL_ZIP_DEV_MODE

4−ffff = 200

5−ffffffff = 5000

6−#else

7−ffff = 0xffff

8−ffffffff = 0xffffffff

9−#endif

data/examples/other/cpp/separation-2a.hsdeleted−8

@@ -1,8 +0,0 @@

1−ffff, ffffffff :: Natural

2−#ifdef HASKELL_ZIP_DEV_MODE

3−ffff = 200

4−ffffffff = 5000

5−#else

6−ffff = 0xffff

7−ffffffff = 0xffffffff

8−#endif

data/examples/other/cpp/separation-2b-out.hsdeleted−11

@@ -1,11 +0,0 @@

1−ffff, ffffffff :: Natural

2−#ifdef HASKELL_ZIP_DEV_MODE

3−ffff = 200

4−ffffffff = 5000

5−#else

6−ffff = 0xffff

7−ffffffff = 0xffffffff

8−#endif

9−

10−foo :: Int

11−foo = undefined

data/examples/other/cpp/separation-2b.hsdeleted−11

@@ -1,11 +0,0 @@

1−ffff, ffffffff :: Natural

2−#ifdef HASKELL_ZIP_DEV_MODE

3−ffff = 200

4−ffffffff = 5000

5−#else

6−ffff = 0xffff

7−ffffffff = 0xffffffff

8−#endif

9−

10−foo :: Int

11−foo = undefined

data/examples/other/cpp/shifted-out.hsdeleted−21

@@ -1,21 +0,0 @@

1−sinkData h compression = do

2− (uncompressedSize, crc32, compressedSize) <-

3− case compression of

4− Store ->

5− withCompression

6− dataSink

7− Deflate ->

8− withCompression $

9− Z.compress 9 (Z.WindowBits (-15)) .| dataSink

10−#ifdef ENABLE_BZIP2

11− BZip2 -> withCompression $

12− BZ.bzip2 .| dataSink

13−#else

14− BZip2 -> throwM BZip2Unsupported

15−#endif

16− return

17− DataDescriptor

18− { ddCRC32 = fromIntegral crc32,

19− ddCompressedSize = compressedSize,

20− ddUncompressedSize = uncompressedSize

21− }

data/examples/other/cpp/shifted.hsdeleted−17

@@ -1,17 +0,0 @@

1−sinkData h compression = do

2− (uncompressedSize, crc32, compressedSize) <-

3− case compression of

4− Store -> withCompression

5− dataSink

6− Deflate -> withCompression $

7− Z.compress 9 (Z.WindowBits (-15)) .| dataSink

8−#ifdef ENABLE_BZIP2

9− BZip2 -> withCompression $

10− BZ.bzip2 .| dataSink

11−#else

12− BZip2 -> throwM BZip2Unsupported

13−#endif

14− return DataDescriptor

15− { ddCRC32 = fromIntegral crc32

16− , ddCompressedSize = compressedSize

17− , ddUncompressedSize = uncompressedSize }

data/examples/other/cpp/simple-import-out.hsdeleted−14

@@ -1,14 +0,0 @@

1−module Main (main) where

2−

3−import Data.Void

4−import Test.Hspec

5−import Test.Hspec.Megaparsec

6−import Text.Megaparsec

7−import Text.Megaparsec.Char

8−

9−#if !MIN_VERSION_base(4,13,0)

10−import Data.Semigroup ((<>))

11−#endif

12−

13−main :: IO ()

14−main = return ()

data/examples/other/cpp/simple-import.hsdeleted−14

@@ -1,14 +0,0 @@

1−module Main (main) where

2−

3−import Data.Void

4−import Test.Hspec

5−import Test.Hspec.Megaparsec

6−import Text.Megaparsec

7−import Text.Megaparsec.Char

8−

9−#if !MIN_VERSION_base(4,13,0)

10−import Data.Semigroup ((<>))

11−#endif

12−

13−main :: IO ()

14−main = return ()

data/examples/other/disabling/indented-fragment-out.hsdeleted−7

@@ -1,7 +0,0 @@

1−foo =

2− {- ORMOLU_DISABLE -}

3− testCase "Foo" testFoo :

4− testCase "Bar" testBar :

5− testCase "Baz" testBaz :

6− {- ORMOLU_ENABLE -}

7− []

data/examples/other/disabling/indented-fragment.hsdeleted−7

@@ -1,7 +0,0 @@

1−foo =

2− {- ORMOLU_DISABLE -}

3− testCase "Foo" testFoo :

4− testCase "Bar" testBar :

5− testCase "Baz" testBaz :

6− {- ORMOLU_ENABLE -}

7− []

data/examples/other/disabling/single-definition-out.hsdeleted−10

@@ -1,10 +0,0 @@

1−module Foo (foo, bar) where

2−

3−{- ORMOLU_DISABLE -}

4−foo :: Int -> Int

5−foo = (+5)

6−{- ORMOLU_ENABLE -}

7−

8−bar :: Bool -> Bool

9−bar True = True

10−bar False = True

data/examples/other/disabling/single-definition.hsdeleted−10

@@ -1,10 +0,0 @@

1−module Foo (foo,bar) where

2−

3−{- ORMOLU_DISABLE -}

4−foo :: Int -> Int

5−foo = (+5)

6−{- ORMOLU_ENABLE -}

7−

8−bar :: Bool -> Bool

9−bar True = True

10−bar False = True

data/examples/other/disabling/whole-file-out.hsdeleted−10

@@ -1,10 +0,0 @@

1−{- ORMOLU_DISABLE -}

2−

3−module Foo (foo,bar) where

4−

5−foo :: Int -> Int

6−foo = (+5)

7−

8−bar :: Bool -> Bool

9−bar True = True

10−bar False = True

data/examples/other/disabling/whole-file.hsdeleted−10

@@ -1,10 +0,0 @@

1−{- ORMOLU_DISABLE -}

2−

3−module Foo (foo,bar) where

4−

5−foo :: Int -> Int

6−foo = (+5)

7−

8−bar :: Bool -> Bool

9−bar True = True

10−bar False = True

data/examples/other/inline-comment-0-out.hsdeleted−3

@@ -1,3 +0,0 @@

1−x = ({-a-} b, c)

2−

3−y = ({-a-} b)

data/examples/other/inline-comment-0.hsdeleted−3

@@ -1,3 +0,0 @@

1−x = ({-a-}b, c)

2−

3−y = ({-a-}b)

data/examples/other/inline-comment-1-out.hsdeleted−12

@@ -1,12 +0,0 @@

1−showPs env ((n, _, Let _ t v) : bs) =

2− " " ++ show n ++ " : "

3− ++ showEnv env ({- normalise ctxt env -} t)

4− ++ " = "

5− ++ showEnv env ({- normalise ctxt env -} v)

6− ++ "\n"

7− ++ showPs env bs

8−showPs env ((n, _, b) : bs) =

9− " " ++ show n ++ " : "

10− ++ showEnv env ({- normalise ctxt env -} (binderTy b))

11− ++ "\n"

12− ++ showPs env bs

data/examples/other/inline-comment-1.hsdeleted−9

@@ -1,9 +0,0 @@

1−showPs env ((n, _, Let _ t v):bs)

2− = " " ++ show n ++ " : " ++

3− showEnv env ({- normalise ctxt env -} t) ++ " = " ++

4− showEnv env ({- normalise ctxt env -} v) ++

5− "\n" ++ showPs env bs

6−showPs env ((n, _, b):bs)

7− = " " ++ show n ++ " : " ++

8− showEnv env ({- normalise ctxt env -} (binderTy b)) ++

9− "\n" ++ showPs env bs

data/examples/other/multiline-forall-out.hs+2−1

@@ -12,7 +12,8 @@ data D

12 )

13 (x :: *)

14 (y :: *).

15+ D

15− D (f x y)

16+ (f x y)

16

17

17data G where

18data G where

18 G ::

19 G ::

data/examples/other/multiple-blank-line-comment-out.hsdeleted−9

@@ -1,9 +0,0 @@

1−module A where

2−

3−a =

4− b

5− [ f

6− {-,

7−

8− -}

9− ]

data/examples/other/multiple-blank-line-comment.hsdeleted−10

@@ -1,10 +0,0 @@

1−module A where

2−

3−a =

4− b

5− [ f

6− {-,

7−

8−

9− -}

10− ]

data/examples/other/pragma-comments-after-out.hsdeleted−4

@@ -1,4 +0,0 @@

1−{-# LANGUAGE AllowAmbiguousTypes #-}

2−{-# LANGUAGE ConstraintKinds #-}

3−-- TODO: Fix and delete this pragma

4−{-# OPTIONS_GHC -Wno-incomplete-uni-patterns #-}

data/examples/other/pragma-comments-after.hsdeleted−3

@@ -1,3 +0,0 @@

1−{-# OPTIONS_GHC -Wno-incomplete-uni-patterns #-} -- TODO: Fix and delete this pragma

2−{-# LANGUAGE AllowAmbiguousTypes #-}

3−{-# LANGUAGE ConstraintKinds #-}

data/examples/other/pragma-comments-out.hsdeleted−8

@@ -1,8 +0,0 @@

1−{-# LANGUAGE OverloadedStrings #-}

2−-- TODO This extension is probably too dangerous, remove it.

3−{-# LANGUAGE RecordWildCards #-}

4−-- Avoid warning produced by TH.

5−{-# OPTIONS_GHC -fno-warn-overlapping-patterns #-}

6−

7−-- | Header comment.

8−module Foo () where

data/examples/other/pragma-comments.hsdeleted−11

@@ -1,11 +0,0 @@

1−-- | Header comment.

2−

3−{-# LANGUAGE OverloadedStrings #-}

4−

5−-- Avoid warning produced by TH.

6−{-# OPTIONS_GHC -fno-warn-overlapping-patterns #-}

7−

8−-- TODO This extension is probably too dangerous, remove it.

9−{-# LANGUAGE RecordWildCards #-}

10−

11−module Foo () where

data/examples/other/pragma-out.hs+4−1

@@ -9,4 +9,7 @@

9{-# OPTIONS_HADDOCK prune, show-extensions #-}

10

11-- | Header comment.

12−module Foo () where

12+module Foo

13+ (

14+ )

15+where

data/examples/other/pragma-sorting-out.hs+2−1

@@ -1,9 +1,10 @@

1{-# LANGUAGE NondecreasingIndentation #-}

2{-# LANGUAGE ScopedTypeVariables #-}

3{-# LANGUAGE TypeFamilies #-}

4+{-# LANGUAGE NoMonoLocalBinds #-}

5+

4-- This gap is necessary for stylish Haskell not to re-arrange

6-- This gap is necessary for stylish Haskell not to re-arrange

5-- NoMonoLocalBinds before TypeFamilies

7-- NoMonoLocalBinds before TypeFamilies

6−{-# LANGUAGE NoMonoLocalBinds #-}

7

8

8module Foo

9module Foo

9 ( bar,

10 ( bar,

default.nix+61−108

@@ -1,9 +1,7 @@

1−{ pkgs ? (import ./nix/nixpkgs { inherit system; })

1+{ pkgs ? (import ./nix/nixpkgs) }:

2−, system ? builtins.currentSystem

3−, ormoluCompiler ? "ghc884" # the shell doesn't work with ghc8101 yet

4−}:

5

2

6let

3let

4+ ormoluCompiler = "ghc865";

7 source = pkgs.lib.sourceByRegex ./. [

5 source = pkgs.lib.sourceByRegex ./. [

8 "^.*\.md$"

6 "^.*\.md$"

9 "^app.*$"

7 "^app.*$"

@@ -16,71 +14,67 @@ let

16 overrides = ormoluOverlay;

14 overrides = ormoluOverlay;

17 };

15 };

18 ormoluOverlay = self: super: {

16 ormoluOverlay = self: super: {

17+ "ormolu" = pkgs.haskell.lib.enableCabalFlag

19− "ormolu" = super.callCabal2nixWithOptions "ormolu" source "-fdev" { };

18+ (super.callCabal2nix "ormolu" source { }) "dev";

19+ # Nixpkgs provides ghc-lib-parser-8.8.0.20190424, but we want

20+ # ghc-lib-parser-8.10.1. We disable Haddock generation because it's way

21+ # too slow.

22+ "ghc-lib-parser" = pkgs.haskell.lib.dontHaddock

23+ (super.callHackageDirect {

24+ pkg = "ghc-lib-parser";

25+ ver = "8.10.1.20200324";

26+ sha256 = "0f2c68fzdj2lw6da2zpx7a0cx631im3kylwd85dzx1npsm1vzlbg";

27+ } {});

20 };

28 };

21 ormolize = import ./nix/ormolize {

29 ormolize = import ./nix/ormolize {

22 inherit pkgs;

30 inherit pkgs;

23 inherit haskellPackages;

31 inherit haskellPackages;

24 };

32 };

25− expectedFailures = [

33+ # NOTE We have to exclude some directories for some packages because

26− "Agda"

34+ # Ormolu needs files to be parsable by Haddock which is not always the

35+ # case. For example some tests and examples do not parse.

36+ excludedHackageDirs = {

27− "aws"

37+ "aws" = ["Examples"];

28− "distributed-process"

38+ "distributed-process" = ["benchmarks"];

29− "esqueleto"

39+ "esqueleto" = ["test"];

30− "fay"

40+ "fay" = ["examples"];

31− "hlint"

32− "idris"

33− "intero"

34− "leksah"

35− "pandoc"

36− "pipes"

37− "postgrest"

41+ "postgrest" = ["test"];

38− "purescript"

42+ };

39− "tensorflow"

40− ] ++ (if ormoluCompiler == "ghc8101" then [] else ["lens"]);

41 ormolizedPackages = doCheck:

43 ormolizedPackages = doCheck:

42 pkgs.lib.mapAttrs (name: p: ormolize {

44 pkgs.lib.mapAttrs (name: p: ormolize {

43 package = p;

45 package = p;

44 inherit doCheck;

46 inherit doCheck;

45− expectedFailures =

47+ excludedDirs =

46− if pkgs.lib.lists.any (x: x == name) expectedFailures

48+ if builtins.hasAttr name excludedHackageDirs

47− then ./expected-failures + "/${name}.txt"

49+ then excludedHackageDirs.${name}

48− else null;

50+ else [];

49 }) haskellPackages;

51 }) haskellPackages;

50in {

52in {

51 ormolu = haskellPackages.ormolu;

53 ormolu = haskellPackages.ormolu;

52− # We put the derivations in another attribute set to avoid building them

53− # when nix-build is run.

54− dev = {

55− ormoluShell =

56− haskellPackages.shellFor {

54+ ormoluShell = haskellPackages.shellFor {

57− packages = ps: [

55+ packages = ps: [

58− ps.ormolu

56+ ps.ormolu

59− ];

57+ ];

60− buildInputs = [

58+ buildInputs = [

61− haskellPackages.cabal-install

59+ haskellPackages.cabal-install

62− haskellPackages.ghcid

60+ haskellPackages.ghcid

63− ];

61+ ];

64− };

62+ };

65− withOrmolu = haskellPackages.shellFor {

63+ withOrmolu = haskellPackages.shellFor {

66− packages = ps: [];

64+ packages = ps: [];

67− buildInputs = [

65+ buildInputs = [

68− haskellPackages.cabal-install

69− haskellPackages.ormolu

66+ haskellPackages.ormolu

70− ];

71− };

67+ ];

72 };

68 };

73 inherit ormoluOverlay ormoluCompiler;

69 inherit ormoluOverlay ormoluCompiler;

74 hackage = ormolizedPackages false;

70 hackage = ormolizedPackages false;

75 hackageTests = with pkgs.lib; pkgs.recurseIntoAttrs (

71 hackageTests = with pkgs.lib; pkgs.recurseIntoAttrs (

76 let ps = [

72 let ps = [

77− "Agda"

78 "QuickCheck"

73 "QuickCheck"

79 "ShellCheck"

74 "ShellCheck"

80 "aeson"

75 "aeson"

81 "attoparsec"

76 "attoparsec"

82 "aws"

77 "aws"

83− "brick"

84 "capability"

78 "capability"

85 "cassava"

79 "cassava"

86 "conduit"

80 "conduit"

@@ -92,78 +86,37 @@ in {

92 "hakyll"

86 "hakyll"

93 "haxl"

87 "haxl"

94 "hedgehog"

88 "hedgehog"

95− "hledger"

96 "hlint"

89 "hlint"

97− "http-client"

98− "idris"

99− "intero"

100− "leksah"

101− "lens"

102 "megaparsec"

90 "megaparsec"

103− "optics"

104 "ormolu"

91 "ormolu"

105− "pandoc"

92+ "optics"

106− "parsec3"

107− "pipes"

108 "postgrest"

93 "postgrest"

109− "purescript"

110 "servant"

94 "servant"

111 "servant-server"

95 "servant-server"

112− "stack"

113 "tensorflow"

96 "tensorflow"

114 "text_1_2_4_0"

97 "text_1_2_4_0"

115 "tls"

98 "tls"

116 "yesod-core"

99 "yesod-core"

100+

101+ # Comment idempotence issue

102+

103+ # "Agda"

104+ # "brick"

105+ # "hledger"

106+ # "http-client"

107+ # "idris"

108+ # "intero"

109+ # "leksah"

110+ # "pandoc"

111+ # "pipes"

112+ # "stack"

113+

114+ # Missing language extension

115+

116+ # "lens" #fixed in master

117+ # "purescript"

118+

117 ];

119 ];

118 in listToAttrs (map (p: nameValuePair p (ormolizedPackages true).${p}) ps)

120 in listToAttrs (map (p: nameValuePair p (ormolizedPackages true).${p}) ps)

119 );

121 );

120− regionTests = pkgs.stdenv.mkDerivation {

121− name = "ormolu-region-tests";

122− src = ./region-tests;

123− buildInputs = [

124− haskellPackages.ormolu

125− pkgs.diffutils

126− ];

127− doCheck = true;

128− buildPhase = ''

129− cp src.hs result-all-implicit.hs

130− ormolu --check-idempotence --mode inplace result-all-implicit.hs

131− cp src.hs result-all-explicit.hs

132− ormolu --check-idempotence --mode inplace --start-line 1 --end-line 18 result-all-explicit.hs

133− cp src.hs result-only-start.hs

134− ormolu --check-idempotence --mode inplace --start-line 1 result-only-start.hs

135− cp src.hs result-only-end.hs

136− ormolu --check-idempotence --mode inplace --end-line 18 result-only-end.hs

137− cp src.hs result-6-7.hs

138− ormolu --check-idempotence --mode inplace --start-line 6 --end-line 7 result-6-7.hs

139− cp src.hs result-6-8.hs

140− ormolu --check-idempotence --mode inplace --start-line 6 --end-line 8 result-6-8.hs

141− cp src.hs result-9-12.hs

142− ormolu --check-idempotence --mode inplace --start-line 9 --end-line 12 result-9-12.hs

143− cp src.hs result-17-18.hs

144− ormolu --check-idempotence --mode inplace --start-line 17 --end-line 18 result-17-18.hs

145− '';

146− checkPhase = ''

147− echo result-all-implicit.hs

148− diff --color=always expected-result-all.hs result-all-implicit.hs

149− echo result-all-explicit.hs

150− diff --color=always expected-result-all.hs result-all-explicit.hs

151− echo result-only-start.hs

152− diff --color=always expected-result-all.hs result-only-start.hs

153− echo result-only-end.hs

154− diff --color=always expected-result-all.hs result-only-end.hs

155− echo result-6-7.hs

156− diff --color=always expected-result-6-7.hs result-6-7.hs

157− echo result-6-8.hs

158− diff --color=always expected-result-6-8.hs result-6-8.hs

159− echo result-9-12.hs

160− diff --color=always expected-result-9-12.hs result-9-12.hs

161− echo result-17-18.hs

162− diff --color=always expected-result-17-18.hs result-17-18.hs

163− '';

164− installPhase = ''

165− mkdir "$out"

166− find . -name '*.hs' -exec cp --parents {} $out \;

167− '';

168− };

169}

122}

expected-failures/Agda.txtdeleted−11

@@ -1,11 +0,0 @@

1−Formatting is not idempotent:

2− dist/build/Agda/Syntax/Parser/Lexer.hs<rendered>:1602:17

3− before: " = (check_ac"

4− after: " =\n "

5−Please, consider reporting the bug.

6−

7−Formatting is not idempotent:

8− src/full/Agda/Syntax/Translation/InternalToAbstract.hs<rendered>:769:4

9− before: " nes\n\n "

10− after: " nes\n\n -- Andreas"

11−Please, consider reporting the bug.

expected-failures/aws.txtdeleted−9

@@ -1,9 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− Examples/PutBucketNearLine.hs:5:1-6

3− parse error on input `import'

4−

5−The GHC parser (in Haddock mode) failed:

6− Examples/Sqs.hs:(88,3)-(90,4)

7− parse error on input `{- | Let's make sure the queue was actually deleted and that the same number of queues exist at when

8− | the program ends as when it started.

9− -}'

expected-failures/distributed-process.txtdeleted−7

@@ -1,7 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− benchmarks/Channels.hs:2:1-6

3− parse error on input `import'

4−

5−The GHC parser (in Haddock mode) failed:

6− benchmarks/Spawns.hs:5:1-6

7− parse error on input `import'

expected-failures/esqueleto.txtdeleted−7

@@ -1,7 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− src/Database/Esqueleto/Internal/Internal.hs:379:1

3− lexical error in string/character literal at character 's'

4−

5−The GHC parser (in Haddock mode) failed:

6− test/PostgreSQL/Test.hs:562:9-41

7− parse error on input `-- | Check the result is not null'

expected-failures/fay.txtdeleted−18

@@ -1,18 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− examples/Separated.hs:12:1-6

3− parse error on input `module'

4−

5−The GHC parser (in Haddock mode) failed:

6− examples/calc.hs:(4,1)-(6,2)

7− parse error on input `-- $ fay -p --html-wrapper --html-js-lib jquery.min.js examples/calc.hs

8−-- You also need to download jquery.min.js.

9−--'

10−

11−The GHC parser (in Haddock mode) failed:

12− examples/canvaswater.hs:(6,1)-(7,17)

13− parse error on input `-- | A demonstration of Fay using the canvas element to display a

14−-- simple effect.'

15−

16−The GHC parser (in Haddock mode) failed:

17− examples/data.hs:9:1-6

18− parse error on input `module'

expected-failures/hlint.txtdeleted−11

@@ -1,11 +0,0 @@

1−Formatting is not idempotent:

2− src/Extension.hs<rendered>:19:22

3− before: "DoRec , -} Recursive"

4− after: "DoRec , -}\n "

5−Please, consider reporting the bug.

6−

7−Formatting is not idempotent:

8− src/Hint/Bracket.hs<rendered>:249:50

9− before: "\" x y [r]) {ideaSpan"

10− after: "\" x y [r])\n "

11−Please, consider reporting the bug.

expected-failures/idris.txtdeleted−7

@@ -1,7 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− src/Idris/Parser.hs:1052:1

3− parse error on input `@'

4−

5−The GHC parser (in Haddock mode) failed:

6− src/Idris/Parser/Expr.hs:75:1

7− parse error on input `@'

expected-failures/intero.txtdeleted−5

@@ -1,5 +0,0 @@

1−Formatting is not idempotent:

2− src/InteractiveUI.hs<rendered>:3748:33

3− before: "text \"Try\" <+> doWha"

4− after: "text \"Try\"\n "

5−Please, consider reporting the bug.

expected-failures/leksah.txtdeleted−5

@@ -1,5 +0,0 @@

1−Formatting is not idempotent:

2− src/IDE/Pane/Modules.hs<rendered>:1184:7

3− before: "cr\n -- show"

4− after: "cr\n in -- show"

5−Please, consider reporting the bug.

expected-failures/lens.txtdeleted−3

@@ -1,3 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− src/Control/Exception/Lens.hs:180:13-37

3− parse error on input `AllocationLimitExceeded__'

expected-failures/pandoc.txtdeleted−9

@@ -1,9 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− benchmark/weigh-pandoc.hs:14:1-6

3− parse error on input `import'

4−

5−Formatting is not idempotent:

6− src/Text/Pandoc/Readers/Vimwiki.hs<rendered>:613:19

7− before: ".softbreak <$ endlin"

8− after: ".softbreak\n <$ en"

9−Please, consider reporting the bug.

expected-failures/pipes.txtdeleted−3

@@ -1,3 +0,0 @@

1−AST of input and AST of formatted code differ.

2− at src/Pipes/Core.hs:(128,1)-(151,2)

3−Please, consider reporting the bug.

expected-failures/postgrest.txtdeleted−5

@@ -1,5 +0,0 @@

1−Formatting is not idempotent:

2− src/PostgREST/DbRequestBuilder.hs<rendered>:149:24

3− before: " || -- clients"

4− after: " || ( origin ="

5−Please, consider reporting the bug.

expected-failures/purescript.txtdeleted−18

@@ -1,18 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− src/Language/PureScript/AST/Declarations.hs:479:1-17

3− Invalid type signature: pattern ValueDecl :: ...

4−Perhaps you meant to use PatternSynonyms?

5−

6−The GHC parser (in Haddock mode) failed:

7− src/Language/PureScript/AST/SourcePos.hs:99:1-22

8− Invalid type signature: pattern NullSourceSpan :: ...

9−Perhaps you meant to use PatternSynonyms?

10−

11−The GHC parser (in Haddock mode) failed:

12− src/Language/PureScript/Constants.hs:42:1-15

13− Invalid type signature: pattern Discard :: ...

14−Perhaps you meant to use PatternSynonyms?

15−

16−The GHC parser (in Haddock mode) failed:

17− src/Language/PureScript/Crash.hs:16:1-6

18− parse error on input `import'

expected-failures/tensorflow.txtdeleted−7

@@ -1,7 +0,0 @@

1−The GHC parser (in Haddock mode) failed:

2− tensorflow-mnist/app/Main.hs:50:13-14

3− parse error on input `->'

4−

5−The GHC parser (in Haddock mode) failed:

6− tensorflow-ops/tests/RegressionTest.hs:3:1-6

7− parse error on input `import'

format.sh+4−6

@@ -1,5 +1,5 @@

1#!/usr/bin/env nix-shell

2−#!nix-shell default.nix -A dev.withOrmolu -i bash --pure

2+#!nix-shell default.nix -A withOrmolu -i bash --pure

3#

4# Format Ormolu using current version of Ormolu.

5

@@ -7,8 +7,6 @@ set -e

7

8export LANG="C.UTF-8"

9

10−cabal format

11−

12−ormolu -m inplace $(find app -type f -name "*.hs")

10+ormolu -p -m inplace $(find app -type f -name "*.hs")

13−ormolu -m inplace $(find src -type f $ -name "*.hs" -o -name "*.hs-boot" $)

11+ormolu -p -m inplace $(find src -type f $ -name "*.hs" -o -name "*.hs-boot" $)

14−ormolu -m inplace $(find tests -type f -name "*.hs")

12+ormolu -p -m inplace $(find tests -type f -name "*.hs")

nix/nixpkgs/default.nix+3−7

@@ -1,13 +1,9 @@

1−{ system ? builtins.currentSystem }:

2let

1let

3− rev = "32b46dd897ab2143a609988a04d87452f0bbef59";

2+ rev = "19.09";

4− sha256 = "1gzfrpjnr1bz9zljsyg3a4zrhk8r927sz761mrgcg56dwinkhpjk";

3+ sha256 = "0mhqhq21y5vrr1f30qd2bvydv4bbbslvyzclhw0kdxmkgg3z4c92";

5 nixpkgs = builtins.fetchTarball {

4 nixpkgs = builtins.fetchTarball {

6 url = "https://github.com/NixOS/nixpkgs/archive/${rev}.tar.gz";

5 url = "https://github.com/NixOS/nixpkgs/archive/${rev}.tar.gz";

7 inherit sha256;

6 inherit sha256;

8 };

7 };

9− pkgs = import nixpkgs {

10− inherit system;

11− config.allowUnfree = true;

8+ pkgs = import nixpkgs { config.allowUnfree = true; };

12− };

13in pkgs

9in pkgs

nix/ormolize/default.nix+13−29

@@ -2,48 +2,32 @@

2, haskellPackages

3}:

4{ package

5−, expectedFailures ? null

6, doCheck ? false

5, doCheck ? false

6+, excludedDirs ? []

7}:

8 pkgs.stdenv.mkDerivation rec {

9 name = package.name + "-ormolized";

10 src = package.src;

11 buildInputs = [

12 haskellPackages.cpphs

13+ (if doCheck

13− haskellPackages.ormolu

14+ then haskellPackages.ormolu

14− pkgs.diffutils

15+ else pkgs.haskell.lib.dontCheck haskellPackages.ormolu)

15 pkgs.glibcLocales

16 pkgs.glibcLocales

16 ];

17 ];

17 LANG = "en_US.UTF-8";

18 LANG = "en_US.UTF-8";

19+ excludedDirsRendered =

20+ if excludedDirs == []

21+ then ""

22+ else pkgs.lib.concatMapStringsSep " " (x: "-not -path './" + x + "/*'") excludedDirs;

18 buildPhase = ''

23 buildPhase = ''

19− hs_files=$(find . -name '*.hs')

24+ find . -name '*.hs' ${excludedDirsRendered} -exec bash ${./ormolize.sh} {} \; 2> log.txt

20− for hs_file in $hs_files; do

25+ cat log.txt

21−

22− # drop includes

23− sed -i '/^#include/d' "$hs_file"

24−

25− # deal with CPP

26− cpphs "$hs_file" --noline -DARCH_X86 > "''${hs_file}-nocpp" 2> /dev/null

27−

28− # annoyingly, cpphs cannot modify files in place

29− mv "''${hs_file}-nocpp" "$hs_file"

30−

31− # preserve the original

32− cp "$hs_file" "''${hs_file}-original"

33− done

34−

35− (ormolu --check-idempotence --mode inplace $hs_files || true) 2> log.txt

36 '';

26 '';

37 inherit doCheck;

27 inherit doCheck;

38− checkPhase =

28+ checkPhase = ''

39− if expectedFailures == null

40− then ''

41− echo "No failures expected"

42− if [[ -s log.txt ]]; then exit 1; fi

29+ if [[ -s log.txt ]]; then exit 1; fi

43− ''

44− else ''

45− diff --ignore-blank-lines --color=always ${expectedFailures} log.txt

46− '';

30+ '';

47 installPhase = ''

31 installPhase = ''

48 mkdir "$out"

32 mkdir "$out"

49 find . -name '*.hs-original' -exec cp --parents {} $out \;

33 find . -name '*.hs-original' -exec cp --parents {} $out \;

nix/ormolize/ormolize.shnew file+17

@@ -0,0 +1,17 @@

1+set -euo pipefail

2+

3+# drop includes

4+sed -i '/^#include/d' "$1"

5+

6+# deal with CPP in a fairly straightforward way

7+cpphs "$1" --noline > "$1-nocpp" 2> /dev/null

8+

9+# annoyingly, cpphs cannot modify files in place

10+mv "$1-nocpp" "$1"

11+

12+# preserve the original

13+cp "$1" "$1-original"

14+

15+# run ormolu

16+ormolu --tolerate-cpp --check-idempotency --mode inplace "$1" ||

17+ ormolu --tolerate-cpp --unsafe --mode inplace "$1"

ormolu.cabal+155−174

@@ -1,184 +1,165 @@

1+name: ormolu

2+version: 0.0.3.1

1−cabal-version: 1.18

3+cabal-version: 1.18

2−name: ormolu

4+tested-with: GHC==8.6.5

3−version: 0.1.2.0

4−license: BSD3

5+license: BSD3

5−license-file: LICENSE.md

6+license-file: LICENSE.md

6−maintainer: Mark Karpov <mark.karpov@tweag.io>

7+maintainer: Mark Karpov <mark.karpov@tweag.io>

7−tested-with: ghc ==8.6.5 ghc ==8.8.4 ghc ==8.10.1

8−homepage: https://github.com/tweag/ormolu

8+homepage: https://github.com/tweag/ormolu

9−bug-reports: https://github.com/tweag/ormolu/issues

9+bug-reports: https://github.com/tweag/ormolu/issues

10−synopsis: A formatter for Haskell source code

11−description: A formatter for Haskell source code.

12−category: Development, Formatting

10+category: Development, Formatting

11+synopsis: A formatter for Haskell source code

13−build-type: Simple

12+build-type: Simple

14−data-files:

13+description: A formatter for Haskell source code.

14+extra-doc-files: CONTRIBUTING.md

15+ , CHANGELOG.md

16+ , DESIGN.md

17+ , README.md

15− data/examples/declaration/annotation/*.hs

18+data-files: data/examples/declaration/annotation/*.hs

16− data/examples/declaration/class/*.hs

19+ , data/examples/declaration/class/*.hs

17− data/examples/declaration/data/*.hs

20+ , data/examples/declaration/data/*.hs

18− data/examples/declaration/data/gadt/*.hs

21+ , data/examples/declaration/data/gadt/*.hs

19− data/examples/declaration/default/*.hs

22+ , data/examples/declaration/default/*.hs

20− data/examples/declaration/deriving/*.hs

23+ , data/examples/declaration/deriving/*.hs

21− data/examples/declaration/foreign/*.hs

24+ , data/examples/declaration/foreign/*.hs

22− data/examples/declaration/instance/*.hs

25+ , data/examples/declaration/instance/*.hs

23− data/examples/declaration/rewrite-rule/*.hs

26+ , data/examples/declaration/rewrite-rule/*.hs

24− data/examples/declaration/role-annotation/*.hs

27+ , data/examples/declaration/role-annotation/*.hs

25− data/examples/declaration/signature/complete/*.hs

28+ , data/examples/declaration/signature/complete/*.hs

26− data/examples/declaration/signature/fixity/*.hs

29+ , data/examples/declaration/signature/fixity/*.hs

27− data/examples/declaration/signature/inline/*.hs

30+ , data/examples/declaration/signature/inline/*.hs

28− data/examples/declaration/signature/minimal/*.hs

31+ , data/examples/declaration/signature/minimal/*.hs

29− data/examples/declaration/signature/pattern/*.hs

32+ , data/examples/declaration/signature/pattern/*.hs

30− data/examples/declaration/signature/set-cost-centre/*.hs

33+ , data/examples/declaration/signature/set-cost-centre/*.hs

31− data/examples/declaration/signature/specialize/*.hs

34+ , data/examples/declaration/signature/specialize/*.hs

32− data/examples/declaration/signature/type/*.hs

35+ , data/examples/declaration/signature/type/*.hs

33− data/examples/declaration/splice/*.hs

36+ , data/examples/declaration/splice/*.hs

34− data/examples/declaration/type-families/closed-type-family/*.hs

37+ , data/examples/declaration/type-families/closed-type-family/*.hs

35− data/examples/declaration/type-families/data-family/*.hs

38+ , data/examples/declaration/type-families/data-family/*.hs

36− data/examples/declaration/type-families/type-family/*.hs

39+ , data/examples/declaration/type-families/type-family/*.hs

37− data/examples/declaration/type-synonyms/*.hs

40+ , data/examples/declaration/type-synonyms/*.hs

38− data/examples/declaration/type/*.hs

41+ , data/examples/declaration/type/*.hs

39− data/examples/declaration/value/function/*.hs

42+ , data/examples/declaration/value/function/*.hs

40− data/examples/declaration/value/function/arrow/*.hs

43+ , data/examples/declaration/value/function/arrow/*.hs

41− data/examples/declaration/value/function/comprehension/*.hs

44+ , data/examples/declaration/value/function/comprehension/*.hs

42− data/examples/declaration/value/function/do/*.hs

45+ , data/examples/declaration/value/function/do/*.hs

43− data/examples/declaration/value/function/infix/*.hs

46+ , data/examples/declaration/value/function/infix/*.hs

44− data/examples/declaration/value/function/pattern/*.hs

47+ , data/examples/declaration/value/function/pattern/*.hs

45− data/examples/declaration/value/other/*.hs

48+ , data/examples/declaration/value/other/*.hs

46− data/examples/declaration/value/pattern-synonyms/*.hs

49+ , data/examples/declaration/value/pattern-synonyms/*.hs

47− data/examples/declaration/warning/*.hs

50+ , data/examples/declaration/warning/*.hs

48− data/examples/import/*.hs

51+ , data/examples/import/*.hs

49− data/examples/module-header/*.hs

52+ , data/examples/module-header/*.hs

50− data/examples/other/*.hs

53+ , data/examples/other/*.hs

51−

52−extra-doc-files:

53− CONTRIBUTING.md

54− CHANGELOG.md

55− DESIGN.md

56− README.md

57

54

58source-repository head

55source-repository head

59− type: git

56+ type: git

60− location: https://github.com/tweag/ormolu.git

57+ location: https://github.com/tweag/ormolu.git

61

58

62flag dev

59flag dev

63− description: Turn on development settings.

60+ description: Turn on development settings.

64− default: False

65− manual: True

61+ manual: True

62+ default: False

66

63

67library

64library

65+ hs-source-dirs: src

66+ build-depends: base >= 4.12 && < 5.0

67+ , bytestring >= 0.2 && < 0.11

68+ , containers >= 0.5 && < 0.7

69+ , dlist >= 0.8 && < 0.9

70+ , exceptions >= 0.6 && < 0.11

71+ , ghc-lib-parser >= 8.10 && < 8.11

72+ , mtl >= 2.0 && < 3.0

73+ , syb >= 0.7 && < 0.8

74+ , text >= 0.2 && < 1.3

68− exposed-modules:

75+ exposed-modules: Ormolu

69− Ormolu

70− Ormolu.Config

76+ , Ormolu.Config

71− Ormolu.Diff

77+ , Ormolu.Diff

72− Ormolu.Exception

78+ , Ormolu.Exception

73− Ormolu.Imports

79+ , Ormolu.Imports

74− Ormolu.Parser

80+ , Ormolu.Parser

75− Ormolu.Parser.Anns

81+ , Ormolu.Parser.Anns

76− Ormolu.Parser.CommentStream

82+ , Ormolu.Parser.CommentStream

77− Ormolu.Parser.Pragma

83+ , Ormolu.Parser.Pragma

78− Ormolu.Parser.Result

84+ , Ormolu.Parser.Result

79− Ormolu.Parser.Shebang

80− Ormolu.Printer

85+ , Ormolu.Printer

81− Ormolu.Printer.Combinators

86+ , Ormolu.Printer.Combinators

82− Ormolu.Printer.Comments

87+ , Ormolu.Printer.Comments

83− Ormolu.Printer.Internal

88+ , Ormolu.Printer.Internal

84− Ormolu.Printer.Meat.Common

89+ , Ormolu.Printer.Meat.Common

85− Ormolu.Printer.Meat.Declaration

90+ , Ormolu.Printer.Meat.Declaration

86− Ormolu.Printer.Meat.Declaration.Annotation

91+ , Ormolu.Printer.Meat.Declaration.Annotation

87− Ormolu.Printer.Meat.Declaration.Class

92+ , Ormolu.Printer.Meat.Declaration.Class

88− Ormolu.Printer.Meat.Declaration.Data

93+ , Ormolu.Printer.Meat.Declaration.Data

89− Ormolu.Printer.Meat.Declaration.Default

94+ , Ormolu.Printer.Meat.Declaration.Default

90− Ormolu.Printer.Meat.Declaration.Foreign

95+ , Ormolu.Printer.Meat.Declaration.Foreign

91− Ormolu.Printer.Meat.Declaration.Instance

96+ , Ormolu.Printer.Meat.Declaration.Instance

92− Ormolu.Printer.Meat.Declaration.RoleAnnotation

97+ , Ormolu.Printer.Meat.Declaration.RoleAnnotation

93− Ormolu.Printer.Meat.Declaration.Rule

98+ , Ormolu.Printer.Meat.Declaration.Rule

94− Ormolu.Printer.Meat.Declaration.Signature

99+ , Ormolu.Printer.Meat.Declaration.Signature

95− Ormolu.Printer.Meat.Declaration.Splice

100+ , Ormolu.Printer.Meat.Declaration.Splice

96− Ormolu.Printer.Meat.Declaration.Type

101+ , Ormolu.Printer.Meat.Declaration.Type

97− Ormolu.Printer.Meat.Declaration.TypeFamily

102+ , Ormolu.Printer.Meat.Declaration.TypeFamily

98− Ormolu.Printer.Meat.Declaration.Value

103+ , Ormolu.Printer.Meat.Declaration.Value

99− Ormolu.Printer.Meat.Declaration.Warning

104+ , Ormolu.Printer.Meat.Declaration.Warning

100− Ormolu.Printer.Meat.ImportExport

105+ , Ormolu.Printer.Meat.ImportExport

101− Ormolu.Printer.Meat.Module

106+ , Ormolu.Printer.Meat.Module

102− Ormolu.Printer.Meat.Pragma

107+ , Ormolu.Printer.Meat.Pragma

103− Ormolu.Printer.Meat.Type

108+ , Ormolu.Printer.Meat.Type

104− Ormolu.Printer.Operators

109+ , Ormolu.Printer.Operators

105− Ormolu.Printer.SpanStream

110+ , Ormolu.Printer.SpanStream

106− Ormolu.Processing.Common

107− Ormolu.Processing.Cpp

108− Ormolu.Processing.Postprocess

109− Ormolu.Processing.Preprocess

110− Ormolu.Utils

111+ , Ormolu.Utils

111−

112− hs-source-dirs: src

113− other-modules:

112+ other-modules: GHC

114− GHC

115− GHC.DynFlags

113+ , GHC.DynFlags

116−

117− default-language: Haskell2010

118− build-depends:

119− base >=4.12 && <5.0,

120− bytestring >=0.2 && <0.11,

121− containers >=0.5 && <0.7,

122− dlist >=0.8 && <2.0,

123− exceptions >=0.6 && <0.11,

124− ghc-lib-parser >=8.10 && <8.11,

125− mtl >=2.0 && <3.0,

126− syb >=0.7 && <0.8,

127− text >=0.2 && <1.3

128−

129− if flag(dev)

114+ if flag(dev)

130− ghc-options:

115+ ghc-options: -Wall -Werror -Wcompat

131− -Wall -Werror -Wcompat -Wincomplete-record-updates

116+ -Wincomplete-record-updates

117+ -Wincomplete-uni-patterns

132− -Wincomplete-uni-patterns -Wnoncanonical-monad-instances

118+ -Wnoncanonical-monad-instances

119+ -- https://github.com/haskell/haddock/issues/1116

133− -Wno-missing-home-modules

120+ -Wno-missing-home-modules

134−

135− else

121+ else

136− ghc-options: -O2 -Wall

122+ ghc-options: -O2 -Wall

137−

138−executable ormolu

139− main-is: Main.hs

140− hs-source-dirs: app

141− other-modules: Paths_ormolu

142− default-language: Haskell2010

123+ default-language: Haskell2010

143− build-depends:

144− base >=4.12 && <5.0,

145− ghc-lib-parser >=8.10 && <8.11,

146− gitrev >=1.3 && <1.4,

147− optparse-applicative >=0.14 && <0.17,

148− ormolu -any,

149− text >=0.2 && <1.3

150−

151− if flag(dev)

152− ghc-options:

153− -Wall -Werror -Wcompat -Wincomplete-record-updates

154− -Wincomplete-uni-patterns -Wnoncanonical-monad-instances

155− -optP-Wno-nonportable-include-path

156−

157− else

158− ghc-options: -O2 -Wall -rtsopts

159

124

160test-suite tests

125test-suite tests

161− type: exitcode-stdio-1.0

162− main-is: Spec.hs

126+ main-is: Spec.hs

163− build-tools: hspec-discover >=2.0 && <3.0

164− hs-source-dirs: tests

127+ hs-source-dirs: tests

128+ type: exitcode-stdio-1.0

129+ build-depends: base >= 4.12 && < 5.0

130+ , containers >= 0.5 && < 0.7

131+ , filepath >= 1.2 && < 1.5

132+ , hspec >= 2.0 && < 3.0

133+ , ormolu

134+ , path >= 0.6 && < 0.8

135+ , path-io >= 1.4.2 && < 2.0

136+ , text >= 0.2 && < 1.3

137+ build-tools: hspec-discover >= 2.0 && < 3.0

165− other-modules:

138+ other-modules:

166− Ormolu.Parser.PragmaSpec

139+ Ormolu.Parser.PragmaSpec

167− Ormolu.PrinterSpec

140+ , Ormolu.PrinterSpec

168

141

142+ if flag(dev)

143+ ghc-options: -Wall -Werror

144+ else

145+ ghc-options: -O2 -Wall

169− default-language: Haskell2010

146+ default-language: Haskell2010

170− build-depends:

171− base >=4.12 && <5.0,

172− containers >=0.5 && <0.7,

173− filepath >=1.2 && <1.5,

174− hspec >=2.0 && <3.0,

175− ormolu -any,

176− path >=0.6 && <0.8,

177− path-io >=1.4.2 && <2.0,

178− text >=0.2 && <1.3

179

147

148+executable ormolu

149+ main-is: Main.hs

150+ hs-source-dirs: app

151+ build-depends: base >= 4.12 && < 5.0

152+ , ghc-lib-parser >= 8.10 && < 8.11

153+ , gitrev >= 1.3 && < 1.4

154+ , optparse-applicative >= 0.14 && < 0.16

155+ , ormolu

156+ , text >= 0.2 && < 1.3

157+ other-modules: Paths_ormolu

180− if flag(dev)

158+ if flag(dev)

181− ghc-options: -Wall -Werror

159+ ghc-options: -Wall -Werror -Wcompat

182−

160+ -Wincomplete-record-updates

161+ -Wincomplete-uni-patterns

162+ -Wnoncanonical-monad-instances

183− else

163+ else

184− ghc-options: -O2 -Wall

164+ ghc-options: -O2 -Wall

165+ default-language: Haskell2010

region-tests/expected-result-17-18.hsdeleted−17

@@ -1,17 +0,0 @@

1−{-# LANGUAGE LambdaCase #-}

2−

3−module Foo (

4− foo, bar, baz) where

5−

6−foo :: Int

7−foo = 5

8−

9−bar :: Int -> Int

10−bar = \case

11− 0 -> foo

12− x -> x - foo

13−

14−baz :: Int -> Int

15−baz = gege

16− where

17− gege = 1 + 2

region-tests/expected-result-6-7.hsdeleted−17

@@ -1,17 +0,0 @@

1−{-# LANGUAGE LambdaCase #-}

2−

3−module Foo (

4− foo, bar, baz) where

5−

6−foo :: Int

7−foo = 5

8−

9−bar :: Int -> Int

10−bar = \case

11− 0 -> foo

12− x -> x - foo

13−

14−baz :: Int -> Int

15−baz = gege

16− where

17− gege = 1 + 2

region-tests/expected-result-6-8.hsdeleted−16

@@ -1,16 +0,0 @@

1−{-# LANGUAGE LambdaCase #-}

2−

3−module Foo (

4− foo, bar, baz) where

5−

6−foo :: Int

7−foo = 5

8−bar :: Int -> Int

9−bar = \case

10− 0 -> foo

11− x -> x - foo

12−

13−baz :: Int -> Int

14−baz = gege

15− where

16− gege = 1 + 2

region-tests/expected-result-9-12.hsdeleted−17

@@ -1,17 +0,0 @@

1−{-# LANGUAGE LambdaCase #-}

2−

3−module Foo (

4− foo, bar, baz) where

5−

6−foo :: Int

7−foo = 5

8−

9−bar :: Int -> Int

10−bar = \case

11− 0 -> foo

12− x -> x - foo

13−

14−baz :: Int -> Int

15−baz = gege

16− where

17− gege = 1 + 2

region-tests/expected-result-all.hsdeleted−21

@@ -1,21 +0,0 @@

1−{-# LANGUAGE LambdaCase #-}

2−

3−module Foo

4− ( foo,

5− bar,

6− baz,

7− )

8−where

9−

10−foo :: Int

11−foo = 5

12−

13−bar :: Int -> Int

14−bar = \case

15− 0 -> foo

16− x -> x - foo

17−

18−baz :: Int -> Int

19−baz = gege

20− where

21− gege = 1 + 2

region-tests/src.hsdeleted−17

@@ -1,17 +0,0 @@

1−{-# LANGUAGE LambdaCase #-}

2−

3−module Foo (

4− foo, bar, baz) where

5−

6−foo :: Int

7−foo = 5

8−

9−bar :: Int -> Int

10−bar = \case

11− 0 -> foo

12− x -> x - foo

13−

14−baz :: Int -> Int

15−baz = gege

16− where

17− gege = 1 + 2

shell.nix+2−4

@@ -1,5 +1,3 @@

1−{ pkgs ? (import ./nix/nixpkgs { inherit system; })

1+{ pkgs ? (import ./nix/nixpkgs) }:

2−, system ? builtins.currentSystem

3−}:

4

2

5−(import ./default.nix { inherit pkgs; }).dev.ormoluShell

3+(import ./default.nix { inherit pkgs; }).ormoluShell

src/GHC.hs+3

@@ -1,5 +1,6 @@

1module GHC

2 ( module X,

3+ ParsedSource,

3 )

4 )

4where

5where

5

6

@@ -10,3 +11,5 @@ import GHC.Hs.Instances as X ()

10import Module as X

11import Module as X

11import RdrName as X

12import RdrName as X

12import SrcLoc as X

13import SrcLoc as X

14+

15+type ParsedSource = Located (HsModule GhcPs)

src/Ormolu.hs+7−10

@@ -6,7 +6,6 @@ module Ormolu

6 ormoluFile,

7 ormoluStdin,

8 Config (..),

9− RegionIndices (..),

10 defaultConfig,

9 defaultConfig,

11 DynOption (..),

10 DynOption (..),

12 OrmoluException (..),

11 OrmoluException (..),

@@ -43,15 +42,13 @@ import qualified SrcLoc as GHC

43ormolu ::

42ormolu ::

44 MonadIO m =>

43 MonadIO m =>

45 -- | Ormolu configuration

44 -- | Ormolu configuration

46− Config RegionIndices ->

45+ Config ->

47 -- | Location of source file

46 -- | Location of source file

48 FilePath ->

47 FilePath ->

49 -- | Input to format

48 -- | Input to format

50 String ->

49 String ->

51 m Text

50 m Text

52−ormolu cfgWithIndices path str = do

51+ormolu cfg path str = do

53− let totalLines = length (lines str)

54− cfg = regionIndicesToDeltas totalLines <$> cfgWithIndices

55 (warnings, result0) <-

52 (warnings, result0) <-

56 parseModule' cfg OrmoluParsingFailed path str

53 parseModule' cfg OrmoluParsingFailed path str

57 when (cfgDebug cfg) $ do

54 when (cfgDebug cfg) $ do

@@ -63,7 +60,7 @@ ormolu cfgWithIndices path str = do

63 -- to parse the rendered code back, inside of GHC monad wrapper which will

60 -- to parse the rendered code back, inside of GHC monad wrapper which will

64 -- lead to error messages presenting the exceptions as GHC bugs.

61 -- lead to error messages presenting the exceptions as GHC bugs.

65 let !txt = printModule result0

62 let !txt = printModule result0

66− when (not (cfgUnsafe cfg) || cfgCheckIdempotence cfg) $ do

63+ when (not (cfgUnsafe cfg) || cfgCheckIdempotency cfg) $ do

67 let pathRendered = path ++ "<rendered>"

64 let pathRendered = path ++ "<rendered>"

68 -- Parse the result of pretty-printing again and make sure that AST

65 -- Parse the result of pretty-printing again and make sure that AST

69 -- is the same as AST of original snippet module span positions.

66 -- is the same as AST of original snippet module span positions.

@@ -79,7 +76,7 @@ ormolu cfgWithIndices path str = do

79 Different ss -> liftIO $ throwIO (OrmoluASTDiffers path ss)

76 Different ss -> liftIO $ throwIO (OrmoluASTDiffers path ss)

80 -- Try re-formatting the formatted result to check if we get exactly

77 -- Try re-formatting the formatted result to check if we get exactly

81 -- the same output.

78 -- the same output.

82− when (cfgCheckIdempotence cfg) $

79+ when (cfgCheckIdempotency cfg) $

83 let txt2 = printModule result1

80 let txt2 = printModule result1

84 in case diffText txt txt2 pathRendered of

81 in case diffText txt txt2 pathRendered of

85 Nothing -> return ()

82 Nothing -> return ()

@@ -96,7 +93,7 @@ ormolu cfgWithIndices path str = do

96ormoluFile ::

93ormoluFile ::

97 MonadIO m =>

94 MonadIO m =>

98 -- | Ormolu configuration

95 -- | Ormolu configuration

99− Config RegionIndices ->

96+ Config ->

100 -- | Location of source file

97 -- | Location of source file

101 FilePath ->

98 FilePath ->

102 -- | Resulting rendition

99 -- | Resulting rendition

@@ -111,7 +108,7 @@ ormoluFile cfg path =

111ormoluStdin ::

108ormoluStdin ::

112 MonadIO m =>

109 MonadIO m =>

113 -- | Ormolu configuration

110 -- | Ormolu configuration

114− Config RegionIndices ->

111+ Config ->

115 -- | Resulting rendition

112 -- | Resulting rendition

116 m Text

113 m Text

117ormoluStdin cfg =

114ormoluStdin cfg =

@@ -124,7 +121,7 @@ ormoluStdin cfg =

124parseModule' ::

121parseModule' ::

125 MonadIO m =>

122 MonadIO m =>

126 -- | Ormolu configuration

123 -- | Ormolu configuration

127− Config RegionDeltas ->

124+ Config ->

128 -- | How to obtain 'OrmoluException' to throw when parsing fails

125 -- | How to obtain 'OrmoluException' to throw when parsing fails

129 (GHC.SrcSpan -> String -> OrmoluException) ->

126 (GHC.SrcSpan -> String -> OrmoluException) ->

130 -- | File name to use in errors

127 -- | File name to use in errors

src/Ormolu/Config.hs+23−62

@@ -1,13 +1,7 @@

1−{-# LANGUAGE DeriveFunctor #-}

2−{-# LANGUAGE RecordWildCards #-}

3−

4-- | Configuration options used by the tool.

1-- | Configuration options used by the tool.

5module Ormolu.Config

2module Ormolu.Config

6 ( Config (..),

3 ( Config (..),

7− RegionIndices (..),

8− RegionDeltas (..),

9 defaultConfig,

4 defaultConfig,

10− regionIndicesToDeltas,

11 DynOption (..),

5 DynOption (..),

12 dynOptionToLocatedStr,

6 dynOptionToLocatedStr,

13 )

7 )

@@ -16,72 +10,39 @@ where

16import qualified SrcLoc as GHC

10import qualified SrcLoc as GHC

17

11

18-- | Ormolu configuration.

12-- | Ormolu configuration.

19−data Config region = Config

13+data Config

14+ = Config

20− { -- | Dynamic options to pass to GHC parser

15+ { -- | Dynamic options to pass to GHC parser

21− cfgDynOptions :: ![DynOption],

16+ cfgDynOptions :: ![DynOption],

22− -- | Do formatting faster but without automatic detection of defects

17+ -- | Do formatting faster but without automatic detection of defects

23− cfgUnsafe :: !Bool,

18+ cfgUnsafe :: !Bool,

24− -- | Output information useful for debugging

19+ -- | Output information useful for debugging

25− cfgDebug :: !Bool,

20+ cfgDebug :: !Bool,

21+ -- | Do not fail if CPP pragma is present (still doesn't handle CPP but

22+ -- useful for formatting of files that enable the extension without

23+ -- actually containing CPP macros)

24+ cfgTolerateCpp :: !Bool,

26− -- | Checks if re-formatting the result is idempotent

25+ -- | Checks if re-formatting the result is idempotent.

27− cfgCheckIdempotence :: !Bool,

26+ cfgCheckIdempotency :: !Bool

28− -- | Region selection

29− cfgRegion :: !region

30− }

31− deriving (Eq, Show, Functor)

32−

33−-- | Region selection as the combination of start and end line numbers.

34−data RegionIndices = RegionIndices

35− { -- | Start line of the region to format

36− regionStartLine :: !(Maybe Int),

37− -- | End line of the region to format

38− regionEndLine :: !(Maybe Int)

39− }

40− deriving (Eq, Show)

41−

42−-- | Region selection as the length of the literal prefix and the literal

43−-- suffix.

44−data RegionDeltas = RegionDeltas

45− { -- | Prefix length in number of lines

46− regionPrefixLength :: !Int,

47− -- | Suffix length in number of lines

48− regionSuffixLength :: !Int

49− }

27+ }

50 deriving (Eq, Show)

28 deriving (Eq, Show)

51

29

52−-- | Default @'Config' 'RegionIndices'@.

30+-- | Default 'Config'.

53−defaultConfig :: Config RegionIndices

31+defaultConfig :: Config

54defaultConfig =

32defaultConfig =

55 Config

33 Config

56 { cfgDynOptions = [],

34 { cfgDynOptions = [],

57 cfgUnsafe = False,

35 cfgUnsafe = False,

58 cfgDebug = False,

36 cfgDebug = False,

59− cfgCheckIdempotence = False,

37+ cfgTolerateCpp = False,

60− cfgRegion =

38+ cfgCheckIdempotency = False

61− RegionIndices

62− { regionStartLine = Nothing,

63− regionEndLine = Nothing

64− }

65− }

66−

67−-- | Convert 'RegionIndices' into 'RegionDeltas'.

68−regionIndicesToDeltas ::

69− -- | Total number of lines in the input

70− Int ->

71− -- | Region indices

72− RegionIndices ->

73− -- | Region deltas

74− RegionDeltas

75−regionIndicesToDeltas total RegionIndices {..} =

76− RegionDeltas

77− { regionPrefixLength = maybe 0 (subtract 1) regionStartLine,

78− regionSuffixLength = maybe 0 (total -) regionEndLine

79 }

39 }

80

40

81-- | A wrapper for dynamic options.

41-- | A wrapper for dynamic options.

82−newtype DynOption = DynOption

42+newtype DynOption

43+ = DynOption

83− { unDynOption :: String

44+ { unDynOption :: String

84− }

45+ }

85 deriving (Eq, Ord, Show)

46 deriving (Eq, Ord, Show)

86

47

87-- | Convert 'DynOption' to @'GHC.Located' 'String'@.

48-- | Convert 'DynOption' to @'GHC.Located' 'String'@.

src/Ormolu/Diff.hs+13−15

@@ -8,14 +8,14 @@ module Ormolu.Diff

8 )

9where

10

11+import BasicTypes (SourceText)

11import Data.ByteString (ByteString)

12import Data.ByteString (ByteString)

12import Data.Generics

13import Data.Generics

13import Data.Text (Text)

14import Data.Text (Text)

14import qualified Data.Text as T

15import qualified Data.Text as T

15import qualified FastString as GHC

16import qualified FastString as GHC

16import GHC

17import GHC

17−import Ormolu.Imports (normalizeImports)

18+import Ormolu.Imports (sortImports)

18−import Ormolu.Parser.CommentStream

19import Ormolu.Parser.Result

20import Ormolu.Utils

21

@@ -39,16 +39,14 @@ diffParseResult :: ParseResult -> ParseResult -> Diff

39diffParseResult

40 ParseResult

41 { prCommentStream = cstream0,

42− prParsedSource = hs0

42+ prParsedSource = ps0

43 }

44 ParseResult

45 { prCommentStream = cstream1,

46− prParsedSource = hs1

46+ prParsedSource = ps1

47 } =

48 matchIgnoringSrcSpans cstream0 cstream1

49− <> matchIgnoringSrcSpans

49+ <> matchIgnoringSrcSpans ps0 ps1

50− hs0 {hsmodImports = normalizeImports (hsmodImports hs0)}

51− hs1 {hsmodImports = normalizeImports (hsmodImports hs1)}

52

50

53-- | Compare two values for equality disregarding differences in 'SrcSpan's

51-- | Compare two values for equality disregarding differences in 'SrcSpan's

54-- and the ordering of import lists.

52-- and the ordering of import lists.

@@ -70,7 +68,7 @@ matchIgnoringSrcSpans = genericQuery

70 gzipWithQ

68 gzipWithQ

71 ( genericQuery

69 ( genericQuery

72 `extQ` srcSpanEq

70 `extQ` srcSpanEq

73− `extQ` commentEq

71+ `extQ` hsModuleEq

74 `extQ` sourceTextEq

72 `extQ` sourceTextEq

75 `extQ` hsDocStringEq

73 `extQ` hsDocStringEq

76 `extQ` importDeclQualifiedStyleEq

74 `extQ` importDeclQualifiedStyleEq

@@ -81,14 +79,14 @@ matchIgnoringSrcSpans = genericQuery

81 | otherwise = Different []

79 | otherwise = Different []

82 srcSpanEq :: SrcSpan -> GenericQ Diff

80 srcSpanEq :: SrcSpan -> GenericQ Diff

83 srcSpanEq _ _ = Same

81 srcSpanEq _ _ = Same

84− commentEq :: Comment -> GenericQ Diff

82+ hsModuleEq :: HsModule GhcPs -> GenericQ Diff

85− commentEq (Comment _ x) d =

83+ hsModuleEq hs0 hs1' =

86− case cast d :: Maybe Comment of

84+ case cast hs1' :: Maybe (HsModule GhcPs) of

87 Nothing -> Different []

85 Nothing -> Different []

88− Just (Comment _ y) ->

86+ Just hs1 ->

89− if x == y

87+ matchIgnoringSrcSpans

90− then Same

88+ hs0 {hsmodImports = sortImports (hsmodImports hs0)}

91− else Different []

89+ hs1 {hsmodImports = sortImports (hsmodImports hs1)}

92 sourceTextEq :: SourceText -> GenericQ Diff

90 sourceTextEq :: SourceText -> GenericQ Diff

93 sourceTextEq _ _ = Same

91 sourceTextEq _ _ = Same

94 importDeclQualifiedStyleEq :: ImportDeclQualifiedStyle -> GenericQ Diff

92 importDeclQualifiedStyleEq :: ImportDeclQualifiedStyle -> GenericQ Diff

src/Ormolu/Exception.hs+11−4

@@ -19,7 +19,9 @@ import System.IO

19

20-- | Ormolu exception representing all cases when Ormolu can fail.

21data OrmoluException

22+ = -- | Ormolu does not work with source files that use CPP

23+ OrmoluCppEnabled FilePath

22− = -- | Parsing of original source code failed

24+ | -- | Parsing of original source code failed

23 OrmoluParsingFailed GHC.SrcSpan String

25 OrmoluParsingFailed GHC.SrcSpan String

24 | -- | Parsing of formatted source code failed

26 | -- | Parsing of formatted source code failed

25 OrmoluOutputParsingFailed GHC.SrcSpan String

27 OrmoluOutputParsingFailed GHC.SrcSpan String

@@ -33,8 +35,13 @@ data OrmoluException

33

35

34instance Exception OrmoluException where

36instance Exception OrmoluException where

35 displayException = \case

37 displayException = \case

38+ OrmoluCppEnabled path ->

39+ unlines

40+ [ "CPP is not supported:",

41+ withIndent path

42+ ]

36 OrmoluParsingFailed s e ->

43 OrmoluParsingFailed s e ->

37− showParsingErr "The GHC parser (in Haddock mode) failed:" s [e]

44+ showParsingErr "Parsing of source code failed:" s [e]

38 OrmoluOutputParsingFailed s e ->

45 OrmoluOutputParsingFailed s e ->

39 showParsingErr "Parsing of formatted code failed:" s [e]

46 showParsingErr "Parsing of formatted code failed:" s [e]

40 ++ "Please, consider reporting the bug.\n"

47 ++ "Please, consider reporting the bug.\n"

@@ -74,8 +81,8 @@ withPrettyOrmoluExceptions m = m `catch` h

74 hPutStrLn stderr (displayException e)

81 hPutStrLn stderr (displayException e)

75 exitWith . ExitFailure $

82 exitWith . ExitFailure $

76 case e of

83 case e of

77− -- Error code 1 is for 'error' or 'notImplemented'

84+ -- Error code 1 is for `error` or `notImplemented`

78− -- 2 used to be for erroring out on CPP

85+ OrmoluCppEnabled {} -> 2

79 OrmoluParsingFailed {} -> 3

86 OrmoluParsingFailed {} -> 3

80 OrmoluOutputParsingFailed {} -> 4

87 OrmoluOutputParsingFailed {} -> 4

81 OrmoluASTDiffers {} -> 5

88 OrmoluASTDiffers {} -> 5

src/Ormolu/Imports.hs+44−161

@@ -4,199 +4,82 @@

4

5-- | Manipulations on import lists.

6module Ormolu.Imports

7− ( normalizeImports,

7+ ( sortImports,

8 )

9where

10

11import Data.Bifunctor

12−import Data.Char (isAlphaNum)

13import Data.Function (on)

12import Data.Function (on)

14−import Data.List (foldl', nubBy, sortBy, sortOn)

13+import Data.Generics (gcompare)

15−import Data.Map.Strict (Map)

14+import Data.List (sortBy)

16−import qualified Data.Map.Strict as M

17−import FastString (FastString)

18import GHC hiding (GhcPs, IE)

15import GHC hiding (GhcPs, IE)

19import GHC.Hs.Extension

16import GHC.Hs.Extension

20import GHC.Hs.ImpExp (IE (..))

17import GHC.Hs.ImpExp (IE (..))

21−import Ormolu.Utils (notImplemented, showOutputable)

18+import Ormolu.Utils (notImplemented)

22

19

23−-- | Sort and normalize imports.

20+-- | Sort imports by module name. This also sorts explicit import lists for

21+-- each declaration.

24−normalizeImports :: [LImportDecl GhcPs] -> [LImportDecl GhcPs]

22+sortImports :: [LImportDecl GhcPs] -> [LImportDecl GhcPs]

25−normalizeImports =

23+sortImports = sortBy compareIdecl . fmap (fmap sortImportLists)

26− fmap snd

27− . M.toAscList

28− . M.fromListWith combineImports

29− . fmap (\x -> (importId x, g x))

30 where

24 where

31− g (L l ImportDecl {..}) =

25+ sortImportLists :: ImportDecl GhcPs -> ImportDecl GhcPs

32− L

26+ sortImportLists = \case

33− l

27+ ImportDecl {..} ->

34 ImportDecl

28 ImportDecl

35− { ideclHiding = second (fmap normalizeLies) <$> ideclHiding,

29+ { ideclHiding = second (fmap sortLies) <$> ideclHiding,

36 ..

30 ..

37 }

31 }

38− g _ = notImplemented "XImportDecl"

32+ XImportDecl x -> noExtCon x

39

33

40−-- | Combine two import declarations. It should be assumed that 'ImportId's

41−-- are equal.

42−combineImports ::

43− LImportDecl GhcPs ->

44− LImportDecl GhcPs ->

34+-- | Compare two @'LImportDecl' 'GhcPs'@ things.

45− LImportDecl GhcPs

35+compareIdecl :: LImportDecl GhcPs -> LImportDecl GhcPs -> Ordering

46−combineImports (L lx ImportDecl {..}) (L _ y) =

36+compareIdecl (L _ m0) (L _ m1) =

47− L

37+ case (isPrelude n0, isPrelude n1) of

48− lx

38+ (False, False) -> n0 `compare` n1

49− ImportDecl

39+ (True, False) -> GT

50− { ideclHiding = case (ideclHiding, GHC.ideclHiding y) of

40+ (False, True) -> LT

51− (Just (hiding, L l' xs), Just (_, L _ ys)) ->

41+ (True, True) -> m0 `gcompare` m1

52− Just (hiding, (L l' (normalizeLies (xs ++ ys))))

53− _ -> Nothing,

54− ..

55− }

56−combineImports _ _ = notImplemented "XImportDecl"

57−

58−-- | Import id, a collection of all things that justify having a separate

59−-- import entry. This is used for merging of imports. If two imports have

60−-- the same 'ImportId' they can be merged.

61−data ImportId = ImportId

62− { importIsPrelude :: Bool,

63− importIdName :: ModuleName,

64− importPkgQual :: Maybe FastString,

65− importSource :: Bool,

66− importSafe :: Bool,

67− importQualified :: Bool,

68− importImplicit :: Bool,

69− importAs :: Maybe ModuleName,

70− importHiding :: Maybe Bool

71− }

72− deriving (Eq, Ord)

73−

74−-- | Obtain an 'ImportId' for a given import.

75−importId :: LImportDecl GhcPs -> ImportId

76−importId (L _ ImportDecl {..}) =

77− ImportId

78− { importIsPrelude = isPrelude,

79− importIdName = moduleName,

80− importPkgQual = sl_fs <$> ideclPkgQual,

81− importSource = ideclSource,

82− importSafe = ideclSafe,

83− importQualified = case ideclQualified of

84− QualifiedPre -> True

85− QualifiedPost -> True

86− NotQualified -> False,

87− importImplicit = ideclImplicit,

88− importAs = unLoc <$> ideclAs,

89− importHiding = fst <$> ideclHiding

90− }

91 where

42 where

43+ n0 = unLoc (ideclName m0)

44+ n1 = unLoc (ideclName m1)

92− isPrelude = moduleNameString moduleName == "Prelude"

45+ isPrelude = (== "Prelude") . moduleNameString

93− moduleName = unLoc ideclName

94−importId _ = notImplemented "XImportDecl"

95

46

96−-- | Normalize a collection of import\/export items.

47+-- | Sort located import or export.

97−normalizeLies :: [LIE GhcPs] -> [LIE GhcPs]

48+sortLies :: [LIE GhcPs] -> [LIE GhcPs]

98−normalizeLies = sortOn (getIewn . unLoc) . M.elems . foldl' combine M.empty

49+sortLies = sortBy (compareIE `on` unLoc) . fmap (fmap sortThings)

99− where

100− combine ::

101− Map IEWrappedNameOrd (LIE GhcPs) ->

102− LIE GhcPs ->

103− Map IEWrappedNameOrd (LIE GhcPs)

104− combine m (L new_l new) =

105− let wname = getIewn new

106− normalizeWNames =

107− nubBy (\x y -> compareLIewn x y == EQ) . sortBy compareLIewn

108− alter = \case

109− Nothing -> Just . L new_l $

110− case new of

111− IEThingWith NoExtField n wildcard g flbl ->

112− IEThingWith NoExtField n wildcard (normalizeWNames g) flbl

113− other -> other

114− Just old ->

115− let f = \case

116− IEVar NoExtField n -> IEVar NoExtField n

117− IEThingAbs NoExtField _ -> new

118− IEThingAll NoExtField n -> IEThingAll NoExtField n

119− IEThingWith NoExtField n wildcard g flbl ->

120− case new of

121− IEVar NoExtField _ ->

122− error "Ormolu.Imports broken presupposition"

123− IEThingAbs NoExtField _ ->

124− IEThingWith NoExtField n wildcard g flbl

125− IEThingAll NoExtField n' ->

126− IEThingAll NoExtField n'

127− IEThingWith NoExtField n' wildcard' g' flbl' ->

128− let combinedWildcard =

129− case (wildcard, wildcard') of

130− (IEWildcard _, _) -> IEWildcard 0

131− (_, IEWildcard _) -> IEWildcard 0

132− _ -> NoIEWildcard

133− in IEThingWith

134− NoExtField

135− n'

136− combinedWildcard

137− (normalizeWNames (g <> g'))

138− flbl'

139− IEModuleContents NoExtField _ -> notImplemented "IEModuleContents"

140− IEGroup NoExtField _ _ -> notImplemented "IEGroup"

141− IEDoc NoExtField _ -> notImplemented "IEDoc"

142− IEDocNamed NoExtField _ -> notImplemented "IEDocNamed"

143− XIE x -> noExtCon x

144− IEModuleContents NoExtField _ -> notImplemented "IEModuleContents"

145− IEGroup NoExtField _ _ -> notImplemented "IEGroup"

146− IEDoc NoExtField _ -> notImplemented "IEDoc"

147− IEDocNamed NoExtField _ -> notImplemented "IEDocNamed"

148− XIE x -> noExtCon x

149− in Just (f <$> old)

150− in M.alter alter wname m

151

50

152−-- | A wrapper for @'IEWrappedName' 'RdrName'@ that allows us to define an

51+-- | Sort imports\/exports inside of 'IEThingWith'.

153−-- 'Ord' instance for it.

52+sortThings :: IE GhcPs -> IE GhcPs

154−newtype IEWrappedNameOrd = IEWrappedNameOrd (IEWrappedName RdrName)

53+sortThings = \case

155− deriving (Eq)

54+ IEThingWith NoExtField x w xs fl ->

55+ IEThingWith NoExtField x w (sortBy (compareIewn `on` unLoc) xs) fl

56+ other -> other

156

57

157−instance Ord IEWrappedNameOrd where

58+-- | Compare two located imports or exports.

158− compare (IEWrappedNameOrd x) (IEWrappedNameOrd y) = compareIewn x y

59+compareIE :: IE GhcPs -> IE GhcPs -> Ordering

60+compareIE = compareIewn `on` getIewn

159

61

160-- | Project @'IEWrappedName' 'RdrName'@ from @'IE' 'GhcPs'@.

62-- | Project @'IEWrappedName' 'RdrName'@ from @'IE' 'GhcPs'@.

161−getIewn :: IE GhcPs -> IEWrappedNameOrd

63+getIewn :: IE GhcPs -> IEWrappedName RdrName

162getIewn = \case

64getIewn = \case

163− IEVar NoExtField x -> IEWrappedNameOrd (unLoc x)

65+ IEVar NoExtField x -> unLoc x

164− IEThingAbs NoExtField x -> IEWrappedNameOrd (unLoc x)

66+ IEThingAbs NoExtField x -> unLoc x

165− IEThingAll NoExtField x -> IEWrappedNameOrd (unLoc x)

67+ IEThingAll NoExtField x -> unLoc x

166− IEThingWith NoExtField x _ _ _ -> IEWrappedNameOrd (unLoc x)

68+ IEThingWith NoExtField x _ _ _ -> unLoc x

167 IEModuleContents NoExtField _ -> notImplemented "IEModuleContents"

69 IEModuleContents NoExtField _ -> notImplemented "IEModuleContents"

168 IEGroup NoExtField _ _ -> notImplemented "IEGroup"

70 IEGroup NoExtField _ _ -> notImplemented "IEGroup"

169 IEDoc NoExtField _ -> notImplemented "IEDoc"

71 IEDoc NoExtField _ -> notImplemented "IEDoc"

170 IEDocNamed NoExtField _ -> notImplemented "IEDocNamed"

72 IEDocNamed NoExtField _ -> notImplemented "IEDocNamed"

171 XIE x -> noExtCon x

73 XIE x -> noExtCon x

172

74

173−-- | Like 'compareIewn' for located wrapped names.

174−compareLIewn :: LIEWrappedName RdrName -> LIEWrappedName RdrName -> Ordering

175−compareLIewn = compareIewn `on` unLoc

176−

177-- | Compare two @'IEWrapppedName' 'RdrName'@ things.

75-- | Compare two @'IEWrapppedName' 'RdrName'@ things.

178compareIewn :: IEWrappedName RdrName -> IEWrappedName RdrName -> Ordering

76compareIewn :: IEWrappedName RdrName -> IEWrappedName RdrName -> Ordering

179−compareIewn (IEName x) (IEName y) = unLoc x `compareRdrName` unLoc y

77+compareIewn (IEName x) (IEName y) = unLoc x `compare` unLoc y

180compareIewn (IEName _) (IEPattern _) = LT

78compareIewn (IEName _) (IEPattern _) = LT

181compareIewn (IEName _) (IEType _) = LT

79compareIewn (IEName _) (IEType _) = LT

182compareIewn (IEPattern _) (IEName _) = GT

80compareIewn (IEPattern _) (IEName _) = GT

183−compareIewn (IEPattern x) (IEPattern y) = unLoc x `compareRdrName` unLoc y

81+compareIewn (IEPattern x) (IEPattern y) = unLoc x `compare` unLoc y

184compareIewn (IEPattern _) (IEType _) = LT

82compareIewn (IEPattern _) (IEType _) = LT

185compareIewn (IEType _) (IEName _) = GT

83compareIewn (IEType _) (IEName _) = GT

186compareIewn (IEType _) (IEPattern _) = GT

84compareIewn (IEType _) (IEPattern _) = GT

187−compareIewn (IEType x) (IEType y) = unLoc x `compareRdrName` unLoc y

85+compareIewn (IEType x) (IEType y) = unLoc x `compare` unLoc y

188−

189−compareRdrName :: RdrName -> RdrName -> Ordering

190−compareRdrName x y =

191− case (getNameStr x, getNameStr y) of

192− ([], []) -> EQ

193− ((_ : _), []) -> GT

194− ([], (_ : _)) -> LT

195− ((x' : _), (y' : _)) ->

196− case (isAlphaNum x', isAlphaNum y') of

197− (False, False) -> x `compare` y

198− (True, False) -> LT

199− (False, True) -> GT

200− (True, True) -> x `compare` y

201− where

202− getNameStr = showOutputable . rdrNameOcc

src/Ormolu/Parser.hs+100−52

@@ -12,11 +12,12 @@ where

12import Bag (bagToList)

13import qualified CmdLineParser as GHC

14import Control.Exception

15+import Control.Monad

15import Control.Monad.IO.Class

16import Control.Monad.IO.Class

16−import qualified Data.List as L

17+import Data.List ((\\), foldl', isPrefixOf, sortOn)

17import qualified Data.List.NonEmpty as NE

18import qualified Data.List.NonEmpty as NE

19+import Data.Maybe (catMaybes)

18import Data.Ord (Down (Down))

20import Data.Ord (Down (Down))

19−import qualified Data.Text as T

20import DynFlags as GHC

21import DynFlags as GHC

21import ErrUtils (Severity (..), errMsgSeverity, errMsgSpan)

22import ErrUtils (Severity (..), errMsgSeverity, errMsgSpan)

22import qualified FastString as GHC

23import qualified FastString as GHC

@@ -31,8 +32,6 @@ import Ormolu.Exception

31import Ormolu.Parser.Anns

32import Ormolu.Parser.Anns

32import Ormolu.Parser.CommentStream

33import Ormolu.Parser.CommentStream

33import Ormolu.Parser.Result

34import Ormolu.Parser.Result

34−import Ormolu.Processing.Preprocess (preprocess)

35−import Ormolu.Utils (incSpanLine, removeIndentation)

36import qualified Panic as GHC

35import qualified Panic as GHC

37import qualified Parser as GHC

36import qualified Parser as GHC

38import qualified StringBuffer as GHC

37import qualified StringBuffer as GHC

@@ -41,7 +40,7 @@ import qualified StringBuffer as GHC

41parseModule ::

40parseModule ::

42 MonadIO m =>

41 MonadIO m =>

43 -- | Ormolu configuration

42 -- | Ormolu configuration

44− Config RegionDeltas ->

43+ Config ->

45 -- | File name (only for source location annotations)

44 -- | File name (only for source location annotations)

46 FilePath ->

45 FilePath ->

47 -- | Input for parser

46 -- | Input for parser

@@ -50,10 +49,8 @@ parseModule ::

50 ( [GHC.Warn],

49 ( [GHC.Warn],

51 Either (SrcSpan, String) ParseResult

50 Either (SrcSpan, String) ParseResult

52 )

51 )

53−parseModule Config {..} path rawInput = liftIO $ do

52+parseModule Config {..} path input' = liftIO $ do

54− let (literalPrefix, indentedInput, literalSuffix, extraComments) =

53+ let (input, extraComments) = extractCommentsFromLines path input'

55− preprocess path rawInput cfgRegion

56− (input, indent) = removeIndentation indentedInput

57 -- It's important that 'setDefaultExts' is done before

54 -- It's important that 'setDefaultExts' is done before

58 -- 'parsePragmasIntoDynFlags', because otherwise we might enable an

55 -- 'parsePragmasIntoDynFlags', because otherwise we might enable an

59 -- extension that was explicitly disabled in the file.

56 -- extension that was explicitly disabled in the file.

@@ -63,7 +60,7 @@ parseModule Config {..} path rawInput = liftIO $ do

63 (setDefaultExts baseDynFlags)

60 (setDefaultExts baseDynFlags)

64 extraOpts = dynOptionToLocatedStr <$> cfgDynOptions

61 extraOpts = dynOptionToLocatedStr <$> cfgDynOptions

65 (warnings, dynFlags) <-

62 (warnings, dynFlags) <-

66− parsePragmasIntoDynFlags baseFlags extraOpts path rawInput >>= \case

63+ parsePragmasIntoDynFlags baseFlags extraOpts path input' >>= \case

67 Right res -> pure res

64 Right res -> pure res

68 Left err ->

65 Left err ->

69 let loc =

66 let loc =

@@ -71,6 +68,8 @@ parseModule Config {..} path rawInput = liftIO $ do

71 (mkSrcLoc (GHC.mkFastString path) 1 1)

68 (mkSrcLoc (GHC.mkFastString path) 1 1)

72 (mkSrcLoc (GHC.mkFastString path) 1 1)

69 (mkSrcLoc (GHC.mkFastString path) 1 1)

73 in throwIO (OrmoluParsingFailed loc err)

70 in throwIO (OrmoluParsingFailed loc err)

71+ when (GHC.xopt Cpp dynFlags && not cfgTolerateCpp) $

72+ throwIO (OrmoluCppEnabled path)

74 let useRecordDot =

73 let useRecordDot =

75 "record-dot-preprocessor" == pgm_F dynFlags

74 "record-dot-preprocessor" == pgm_F dynFlags

76 || any

75 || any

@@ -78,18 +77,15 @@ parseModule Config {..} path rawInput = liftIO $ do

78 (pluginModNames dynFlags)

77 (pluginModNames dynFlags)

79 pStateErrors = \pstate ->

78 pStateErrors = \pstate ->

80 let errs = bagToList $ GHC.getErrorMessages pstate dynFlags

79 let errs = bagToList $ GHC.getErrorMessages pstate dynFlags

81− fixupErrSpan = incSpanLine (regionPrefixLength cfgRegion)

82− in case L.sortOn (Down . SeverityOrd . errMsgSeverity) errs of

80+ in case sortOn (Down . SeverityOrd . errMsgSeverity) errs of

83 [] -> Nothing

81 [] -> Nothing

84− err : _ ->

85− -- Show instance returns a short error message

82+ err : _ -> Just (errMsgSpan err, show err) -- Show instance returns a short error message

86− Just (fixupErrSpan (errMsgSpan err), show err)

87 r = case runParser GHC.parseModule dynFlags path input of

83 r = case runParser GHC.parseModule dynFlags path input of

88 GHC.PFailed pstate ->

84 GHC.PFailed pstate ->

89 case pStateErrors pstate of

85 case pStateErrors pstate of

90 Just err -> Left err

86 Just err -> Left err

91− Nothing -> error "PFailed does not have an error"

87+ Nothing -> error "invariant violation: PFailed does not have an error"

92− GHC.POk pstate (L _ hsModule) ->

88+ GHC.POk pstate pmod ->

93 case pStateErrors pstate of

89 case pStateErrors pstate of

94 -- Some parse errors (pattern/arrow syntax in expr context)

90 -- Some parse errors (pattern/arrow syntax in expr context)

95 -- do not cause a parse error, but they are replaced with "_"

91 -- do not cause a parse error, but they are replaced with "_"

@@ -97,33 +93,20 @@ parseModule Config {..} path rawInput = liftIO $ do

97 -- later stages; but we fail in those cases.

93 -- later stages; but we fail in those cases.

98 Just err -> Left err

94 Just err -> Left err

99 Nothing ->

95 Nothing ->

100− let (stackHeader, shebangs, pragmas, comments) =

101− mkCommentStream input extraComments pstate

96+ let (comments, exts, shebangs) = mkCommentStream extraComments pstate

102 in Right

97 in Right

103 ParseResult

98 ParseResult

104− { prParsedSource = hsModule,

99+ { prParsedSource = pmod,

105 prAnns = mkAnns pstate,

100 prAnns = mkAnns pstate,

106− prStackHeader = stackHeader,

107− prShebangs = shebangs,

108− prPragmas = pragmas,

109 prCommentStream = comments,

101 prCommentStream = comments,

102+ prExtensions = exts,

103+ prShebangs = shebangs,

110 prUseRecordDot = useRecordDot,

104 prUseRecordDot = useRecordDot,

111 prImportQualifiedPost =

105 prImportQualifiedPost =

112− GHC.xopt ImportQualifiedPost dynFlags,

106+ GHC.xopt ImportQualifiedPost dynFlags

113− prLiteralPrefix = T.pack literalPrefix,

114− prLiteralSuffix = T.pack literalSuffix,

115− prIndent = indent

116 }

107 }

117 return (warnings, r)

108 return (warnings, r)

118

109

119−-- | Enable all language extensions that we think should be enabled by

120−-- default for ease of use.

121−setDefaultExts :: DynFlags -> DynFlags

122−setDefaultExts flags = L.foldl' xopt_set flags autoExts

123− where

124− autoExts = allExts L.\\ manualExts

125− allExts = [minBound .. maxBound]

126−

127-- | Extensions that are not enabled automatically and should be activated

110-- | Extensions that are not enabled automatically and should be activated

128-- by user.

111-- by user.

129manualExts :: [Extension]

112manualExts :: [Extension]

@@ -150,6 +133,9 @@ manualExts =

150 -- decision of enabling this style is left to the user

133 -- decision of enabling this style is left to the user

151 ]

134 ]

152

135

136+----------------------------------------------------------------------------

137+-- Helpers (taken from ghc-exactprint)

138+

153-- | Run a 'GHC.P' computation.

139-- | Run a 'GHC.P' computation.

154runParser ::

140runParser ::

155 -- | Computation to run

141 -- | Computation to run

@@ -168,27 +154,67 @@ runParser parser flags filename input = GHC.unP parser parseState

168 buffer = GHC.stringToStringBuffer input

154 buffer = GHC.stringToStringBuffer input

169 parseState = GHC.mkPState flags buffer location

155 parseState = GHC.mkPState flags buffer location

170

156

171−-- | Wrap GHC's 'Severity' to add 'Ord' instance.

157+-- | Transform given lines possibly returning comments extracted from them.

172−newtype SeverityOrd = SeverityOrd Severity

158+-- This handles LINE pragmas and shebangs.

159+extractCommentsFromLines ::

160+ -- | File name, just to use in the spans

161+ FilePath ->

162+ -- | List of lines from that file

163+ String ->

164+ -- | Adjusted lines together with comments extracted from them

165+ (String, [Located String])

166+extractCommentsFromLines path =

167+ unlines' . unzip . zipWith (extractCommentFromLine path) [1 ..] . lines

168+ where

169+ unlines' (a, b) = (unlines a, catMaybes b)

173

170

174−instance Eq SeverityOrd where

171+-- | Transform a given line possibly returning a comment extracted from it.

175− s1 == s2 = compare s1 s2 == EQ

172+extractCommentFromLine ::

173+ -- | File name, just to use in the spans

174+ FilePath ->

175+ -- | Line number of this line

176+ Int ->

177+ -- | The actual line

178+ String ->

179+ -- | Adjusted line and possibly a comment extracted from it

180+ (String, Maybe (Located String))

181+extractCommentFromLine path line s

182+ | "{-# LINE" `isPrefixOf` s =

183+ let (pragma, res) = getPragma s

184+ size = length pragma

185+ ss = mkSrcSpan (mkSrcLoc' 1) (mkSrcLoc' (size + 1))

186+ in (res, Just $ L ss pragma)

187+ | isShebang s =

188+ let ss = mkSrcSpan (mkSrcLoc' 1) (mkSrcLoc' (length s))

189+ in ("", Just $ L ss s)

190+ | otherwise = (s, Nothing)

191+ where

192+ mkSrcLoc' = mkSrcLoc (GHC.mkFastString path) line

176

193

177−instance Ord SeverityOrd where

194+-- | Take a line pragma and output its replacement (where line pragma is

178− compare (SeverityOrd s1) (SeverityOrd s2) =

195+-- replaced with spaces) and the contents of the pragma itself.

179− compare (f s1) (f s2)

196+getPragma ::

180− where

197+ -- | Pragma line to analyze

181− f :: Severity -> Int

198+ String ->

182− f SevOutput = 1

199+ -- | Contents of the pragma and its replacement line

183− f SevFatal = 2

200+ (String, String)

184− f SevInteractive = 3

201+getPragma [] = error "Ormolu.Parser.getPragma: input must not be empty"

185− f SevDump = 4

202+getPragma s@(x : xs)

186− f SevInfo = 5

203+ | "#-}" `isPrefixOf` s = ("#-}", " " ++ drop 3 s)

187− f SevWarning = 6

204+ | otherwise =

188− f SevError = 7

205+ let (prag, remline) = getPragma xs

206+ in (x : prag, ' ' : remline)

207+

208+-- | Enable all language extensions that we think should be enabled by

209+-- default for ease of use.

210+setDefaultExts :: DynFlags -> DynFlags

211+setDefaultExts flags = foldl' GHC.xopt_set flags autoExts

212+ where

213+ autoExts = allExts \\ manualExts

214+ allExts = [minBound .. maxBound]

189

215

190----------------------------------------------------------------------------

216----------------------------------------------------------------------------

191−-- Helpers taken from HLint

217+-- More helpers (taken from HLint)

192

218

193parsePragmasIntoDynFlags ::

219parsePragmasIntoDynFlags ::

194 -- | Pre-set 'DynFlags'

220 -- | Pre-set 'DynFlags'

@@ -217,3 +243,25 @@ parsePragmasIntoDynFlags flags extraOpts filepath str =

217 reportErr

243 reportErr

218 (GHC.handleSourceError reportErr act)

244 (GHC.handleSourceError reportErr act)

219 reportErr e = return $ Left (show e)

245 reportErr e = return $ Left (show e)

246+

247+----------------------------------------------------------------------------

248+-- Even more helpers

249+

250+-- Wrap GHC's ErrUtils.Severity to add Ord instance

251+newtype SeverityOrd = SeverityOrd Severity

252+

253+instance Eq SeverityOrd where

254+ s1 == s2 = compare s1 s2 == EQ

255+

256+instance Ord SeverityOrd where

257+ compare (SeverityOrd s1) (SeverityOrd s2) =

258+ compare (f s1) (f s2)

259+ where

260+ f :: Severity -> Int

261+ f SevOutput = 1

262+ f SevFatal = 2

263+ f SevInteractive = 3

264+ f SevDump = 4

265+ f SevInfo = 5

266+ f SevWarning = 6

267+ f SevError = 7

src/Ormolu/Parser/CommentStream.hs+61−147

@@ -4,74 +4,78 @@

4

5-- | Functions for working with comment stream.

6module Ormolu.Parser.CommentStream

7− ( -- * Comment stream

8− CommentStream (..),

7+ ( CommentStream (..),

9− mkCommentStream,

10− showCommentStream,

11−

12− -- * Comment

13 Comment (..),

8 Comment (..),

14− unComment,

9+ mkCommentStream,

15− hasAtomsBefore,

10+ isShebang,

11+ isPrevHaddock,

16 isMultilineComment,

12 isMultilineComment,

13+ showCommentStream,

17 )

14 )

18where

15where

19

16

20import Data.Char (isSpace)

17import Data.Char (isSpace)

21import Data.Data (Data)

18import Data.Data (Data)

22−import qualified Data.List as L

19+import Data.Either (partitionEithers)

20+import Data.List (dropWhileEnd, isPrefixOf, sortOn)

23import Data.List.NonEmpty (NonEmpty (..))

21import Data.List.NonEmpty (NonEmpty (..))

24import qualified Data.List.NonEmpty as NE

22import qualified Data.List.NonEmpty as NE

25import Data.Maybe (mapMaybe)

23import Data.Maybe (mapMaybe)

26import qualified GHC

24import qualified GHC

27import qualified Lexer as GHC

25import qualified Lexer as GHC

28import Ormolu.Parser.Pragma

26import Ormolu.Parser.Pragma

29−import Ormolu.Parser.Shebang

30−import Ormolu.Processing.Common

31−import Ormolu.Utils (onTheSameLine, showOutputable)

27+import Ormolu.Utils (showOutputable)

32import SrcLoc

28import SrcLoc

33

29

34−----------------------------------------------------------------------------

35−-- Comment stream

36−

37-- | A stream of 'RealLocated' 'Comment's in ascending order with respect to

30-- | A stream of 'RealLocated' 'Comment's in ascending order with respect to

38-- beginning of corresponding spans.

31-- beginning of corresponding spans.

39newtype CommentStream = CommentStream [RealLocated Comment]

32newtype CommentStream = CommentStream [RealLocated Comment]

40 deriving (Eq, Data, Semigroup, Monoid)

33 deriving (Eq, Data, Semigroup, Monoid)

41

34

35+-- | A wrapper for a single comment. The 'NonEmpty' list inside contains

36+-- lines of multiline comment @{- … -}@ or just single item\/line otherwise.

37+newtype Comment = Comment (NonEmpty String)

38+ deriving (Eq, Show, Data)

39+

42-- | Create 'CommentStream' from 'GHC.PState'. The pragmas and shebangs are

40-- | Create 'CommentStream' from 'GHC.PState'. The pragmas and shebangs are

43-- removed from the 'CommentStream'. Shebangs are only extracted from the

41-- removed from the 'CommentStream'. Shebangs are only extracted from the

44-- comments that come from the first argument.

42-- comments that come from the first argument.

45mkCommentStream ::

43mkCommentStream ::

46− -- | Original input

47− String ->

48 -- | Extra comments to include

44 -- | Extra comments to include

49 [Located String] ->

45 [Located String] ->

50 -- | Parser state to use for comment extraction

46 -- | Parser state to use for comment extraction

51 GHC.PState ->

47 GHC.PState ->

52− -- | Stack header, shebangs, pragmas, and comment stream

48+ -- | Comment stream, a set of extracted pragmas, and extracted shebangs

53− ( Maybe (RealLocated Comment),

49+ (CommentStream, [Pragma], [Located String])

54− [Shebang],

55− [([RealLocated Comment], Pragma)],

56− CommentStream

57− )

58−mkCommentStream input extraComments pstate =

50+mkCommentStream extraComments pstate =

59− ( mstackHeader,

51+ ( CommentStream $

60− shebangs,

52+ mkComment <$> sortOn (realSrcSpanStart . getRealSrcSpan) comments,

61 pragmas,

53 pragmas,

62− CommentStream comments

54+ shebangs

63 )

55 )

64 where

56 where

65− (comments, pragmas) = extractPragmas input rawComments1

57+ (comments, pragmas) = partitionEithers (partitionComments <$> rawComments)

66− (rawComments1, mstackHeader) = extractStackHeader rawComments0

58+ rawComments =

67− rawComments0 =

68− L.sortOn (realSrcSpanStart . getRealSrcSpan) . mapMaybe toRealSpan $

59+ mapMaybe toRealSpan $

69 otherExtraComments

60 otherExtraComments

70 ++ mapMaybe (liftMaybe . fmap unAnnotationComment) (GHC.comment_q pstate)

61 ++ mapMaybe (liftMaybe . fmap unAnnotationComment) (GHC.comment_q pstate)

71 ++ concatMap

62 ++ concatMap

72 (mapMaybe (liftMaybe . fmap unAnnotationComment) . snd)

63 (mapMaybe (liftMaybe . fmap unAnnotationComment) . snd)

73 (GHC.annotations_comments pstate)

64 (GHC.annotations_comments pstate)

74− (shebangs, otherExtraComments) = extractShebangs extraComments

65+ (shebangs, otherExtraComments) = span (isShebang . unLoc) extraComments

66+

67+-- | Return 'True' if given 'String' is a shebang.

68+isShebang :: String -> Bool

69+isShebang str = "#!" `isPrefixOf` str

70+

71+-- | Test whether a 'Comment' looks like a Haddock following a definition,

72+-- i.e. something starting with @-- ^@.

73+isPrevHaddock :: Comment -> Bool

74+isPrevHaddock (Comment (x :| _)) = "-- ^" `isPrefixOf` x

75+

76+-- | Is this comment multiline-style?

77+isMultilineComment :: Comment -> Bool

78+isMultilineComment (Comment (x :| _)) = "{-" `isPrefixOf` x

75

79

76-- | Pretty-print a 'CommentStream'.

80-- | Pretty-print a 'CommentStream'.

77showCommentStream :: CommentStream -> String

81showCommentStream :: CommentStream -> String

@@ -82,117 +86,28 @@ showCommentStream (CommentStream xs) =

82 showComment (GHC.L l str) = showOutputable l ++ " " ++ show str

86 showComment (GHC.L l str) = showOutputable l ++ " " ++ show str

83

87

84----------------------------------------------------------------------------

88----------------------------------------------------------------------------

85−-- Comment

89+-- Helpers

86−

87−-- | A wrapper for a single comment. The 'Bool' indicates whether there were

88−-- atoms before beginning of the comment in the original input. The

89−-- 'NonEmpty' list inside contains lines of multiline comment @{- … -}@ or

90−-- just single item\/line otherwise.

91−data Comment = Comment Bool (NonEmpty String)

92− deriving (Eq, Show, Data)

93

90

94-- | Normalize comment string. Sometimes one multi-line comment is turned

91-- | Normalize comment string. Sometimes one multi-line comment is turned

95-- into several lines for subsequent outputting with correct indentation for

92-- into several lines for subsequent outputting with correct indentation for

96-- each line.

93-- each line.

97−mkComment ::

94+mkComment :: RealLocated String -> RealLocated Comment

98− -- | Lines of original input with their indices

95+mkComment (L l s) =

99− [(Int, String)] ->

96+ L l . Comment . fmap dropTrailing $

100− -- | Raw comment string

97+ if "{-" `isPrefixOf` s

101− RealLocated String ->

98+ then case NE.nonEmpty (lines s) of

102− -- | Remaining lines of original input and the constructed 'Comment'

99+ Nothing -> s :| []

103− ([(Int, String)], RealLocated Comment)

100+ Just (x :| xs) ->

104−mkComment ls (L l s) = (ls', comment)

101+ let getIndent y =

102+ if all isSpace y

103+ then startIndent

104+ else length (takeWhile isSpace y)

105+ n = minimum (startIndent : fmap getIndent xs)

106+ in x :| (drop n <$> xs)

107+ else s :| []

105 where

108 where

106− comment =

107− L l . Comment atomsBefore . removeConseqBlanks . fmap dropTrailing $

108− if "{-" `L.isPrefixOf` s

109− then case NE.nonEmpty (lines s) of

110− Nothing -> s :| []

111− Just (x :| xs) ->

112− let getIndent y =

113− if all isSpace y || y == endDisabling

114− then startIndent

115− else length (takeWhile isSpace y)

116− n = minimum (startIndent : fmap getIndent xs)

117− removeIndent y =

118− if y == endDisabling

119− then y

120− else drop n y

121− in x :| (removeIndent <$> xs)

122− else s :| []

123− (atomsBefore, ls') =

124− case dropWhile ((< commentLine) . fst) ls of

125− [] -> (False, [])

126− ((_, i) : ls'') ->

127− case take 2 (dropWhile isSpace i) of

128− "--" -> (False, ls'')

129− "{-" -> (False, ls'')

130− _ -> (True, ls'')

131− dropTrailing = L.dropWhileEnd isSpace

109+ dropTrailing = dropWhileEnd isSpace

132 startIndent = srcSpanStartCol l - 1

110 startIndent = srcSpanStartCol l - 1

133− commentLine = srcSpanStartLine l

134−

135−-- | Get a collection of lines from a 'Comment'.

136−unComment :: Comment -> NonEmpty String

137−unComment (Comment _ xs) = xs

138−

139−-- | Check whether the 'Comment' had some non-whitespace atoms in front of

140−-- it in the original input.

141−hasAtomsBefore :: Comment -> Bool

142−hasAtomsBefore (Comment atomsBefore _) = atomsBefore

143−

144−-- | Is this comment multiline-style?

145−isMultilineComment :: Comment -> Bool

146−isMultilineComment (Comment _ (x :| _)) = "{-" `L.isPrefixOf` x

147−

148−----------------------------------------------------------------------------

149−-- Helpers

150−

151−-- | Detect and extract stack header if it is present.

152−extractStackHeader ::

153− -- | Comment stream to analyze

154− [RealLocated String] ->

155− ([RealLocated String], Maybe (RealLocated Comment))

156−extractStackHeader = \case

157− [] -> ([], Nothing)

158− (x : xs) ->

159− let comment = snd (mkComment [] x)

160− in if isStackHeader (unRealSrcSpan comment)

161− then (xs, Just comment)

162− else (x : xs, Nothing)

163− where

164− isStackHeader (Comment _ (x :| _)) =

165− "stack" `L.isPrefixOf` dropWhile isSpace (drop 2 x)

166−

167−-- | Extract pragmas and their associated comments.

168−extractPragmas ::

169− -- | Input

170− String ->

171− -- | Comment stream to analyze

172− [RealLocated String] ->

173− ([RealLocated Comment], [([RealLocated Comment], Pragma)])

174−extractPragmas input = go initialLs id id

175− where

176− initialLs = zip [1 ..] (lines input)

177− go ls csSoFar pragmasSoFar = \case

178− [] -> (csSoFar [], pragmasSoFar [])

179− (x : xs) ->

180− case parsePragma (unRealSrcSpan x) of

181− Nothing ->

182− let (ls', x') = mkComment ls x

183− in go ls' (csSoFar . (x' :)) pragmasSoFar xs

184− Just pragma ->

185− let combined ys = (csSoFar ys, pragma)

186− go' ls' ys rest = go ls' id (pragmasSoFar . (combined ys :)) rest

187− in case xs of

188− [] -> go' ls [] xs

189− (y : ys) ->

190− let (ls', y') = mkComment ls y

191− in if onTheSameLine

192− (RealSrcSpan (getRealSrcSpan x))

193− (RealSrcSpan (getRealSrcSpan y))

194− then go' ls' [y'] ys

195− else go' ls [] xs

196

111

197-- | Get a 'String' from 'GHC.AnnotationComment'.

112-- | Get a 'String' from 'GHC.AnnotationComment'.

198unAnnotationComment :: GHC.AnnotationComment -> Maybe String

113unAnnotationComment :: GHC.AnnotationComment -> Maybe String

@@ -214,13 +129,12 @@ toRealSpan :: Located a -> Maybe (RealLocated a)

214toRealSpan (L (RealSrcSpan l) a) = Just (L l a)

129toRealSpan (L (RealSrcSpan l) a) = Just (L l a)

215toRealSpan _ = Nothing

130toRealSpan _ = Nothing

216

131

217−-- | Remove consecutive blank lines.

132+-- | If a given comment is a pragma, return it in parsed form in 'Right'.

218−removeConseqBlanks :: NonEmpty String -> NonEmpty String

133+-- Otherwise return the original comment unchanged.

219−removeConseqBlanks (x :| xs) = x :| go (null x) id xs

134+partitionComments ::

220− where

135+ RealLocated String ->

221− go seenBlank acc = \case

136+ Either (RealLocated String) Pragma

222− [] -> acc []

137+partitionComments input =

223− (y : ys) ->

138+ case parsePragma (unRealSrcSpan input) of

224− if seenBlank && null y

139+ Nothing -> Left input

225− then go True acc ys

140+ Just pragma -> Right pragma

226− else go (null y) (acc . (y :)) ys

src/Ormolu/Parser/Pragma.hs+4−4

@@ -10,7 +10,7 @@ where

10

11import Control.Monad

12import Data.Char (isSpace, toLower)

13−import qualified Data.List as L

13+import Data.List

14import qualified EnumSet as ES

15import FastString (mkFastString, unpackFS)

16import qualified Lexer as L

@@ -35,8 +35,8 @@ parsePragma ::

35 String ->

36 Maybe Pragma

37parsePragma input = do

38− inputNoPrefix <- L.stripPrefix "{-#" input

38+ inputNoPrefix <- stripPrefix "{-#" input

39− guard ("#-}" `L.isSuffixOf` input)

39+ guard ("#-}" `isSuffixOf` input)

40 let contents = take (length inputNoPrefix - 3) inputNoPrefix

41 (pragmaName, cs) = (break isSpace . dropWhile isSpace) contents

42 case toLower <$> pragmaName of

@@ -46,7 +46,7 @@ parsePragma input = do

46 _ -> Nothing

47 where

48 trimSpaces :: String -> String

49− trimSpaces = L.dropWhileEnd isSpace . dropWhile isSpace

49+ trimSpaces = dropWhileEnd isSpace . dropWhile isSpace

50

51-- | Assuming the input consists of a series of tokens from a language

52-- pragma, return the set of enabled extensions.

src/Ormolu/Parser/Result.hs+17−26

@@ -7,38 +7,29 @@ module Ormolu.Parser.Result

7 )

8where

9

10−import Data.Text (Text)

11import GHC

10import GHC

12import Ormolu.Parser.Anns

11import Ormolu.Parser.Anns

13import Ormolu.Parser.CommentStream

12import Ormolu.Parser.CommentStream

14import Ormolu.Parser.Pragma (Pragma)

13import Ormolu.Parser.Pragma (Pragma)

15−import Ormolu.Parser.Shebang (Shebang)

16

14

17-- | A collection of data that represents a parsed module in Ormolu.

15-- | A collection of data that represents a parsed module in Ormolu.

18−data ParseResult = ParseResult

16+data ParseResult

17+ = ParseResult

19− { -- | 'ParsedSource' from GHC

18+ { -- | 'ParsedSource' from GHC

20− prParsedSource :: HsModule GhcPs,

19+ prParsedSource :: ParsedSource,

21− -- | Ormolu-specfic representation of annotations

20+ -- | Ormolu-specfic representation of annotations

22− prAnns :: Anns,

21+ prAnns :: Anns,

23− -- | Stack header

22+ -- | Comment stream

24− prStackHeader :: Maybe (RealLocated Comment),

23+ prCommentStream :: CommentStream,

24+ -- | Extensions enabled in that module

25+ prExtensions :: [Pragma],

25− -- | Shebangs found in the input

26+ -- | Shebangs found in the input

26− prShebangs :: [Shebang],

27+ prShebangs :: [Located String],

27− -- | Pragmas and the associated comments

28− prPragmas :: [([RealLocated Comment], Pragma)],

29− -- | Comment stream

30− prCommentStream :: CommentStream,

31− -- | Whether or not record dot syntax is enabled

28+ -- | Whether or not record dot syntax is enabled

32− prUseRecordDot :: Bool,

29+ prUseRecordDot :: Bool,

33− -- | Whether or not ImportQualifiedPost is enabled

30+ -- | Whether or not ImportQualifiedPost is enabled

34− prImportQualifiedPost :: Bool,

31+ prImportQualifiedPost :: Bool

35− -- | Literal prefix

36− prLiteralPrefix :: Text,

37− -- | Literal suffix

38− prLiteralSuffix :: Text,

39− -- | Indentation level, can be non-zero in case of region formatting

40− prIndent :: Int

41− }

32+ }

42

33

43-- | Pretty-print a 'ParseResult'.

34-- | Pretty-print a 'ParseResult'.

44prettyPrintParseResult :: ParseResult -> String

35prettyPrintParseResult :: ParseResult -> String

src/Ormolu/Parser/Shebang.hsdeleted−27

@@ -1,27 +0,0 @@

1−{-# LANGUAGE DeriveDataTypeable #-}

2−

3−-- | A module for dealing with shebangs.

4−module Ormolu.Parser.Shebang

5− ( Shebang (..),

6− extractShebangs,

7− isShebang,

8− )

9−where

10−

11−import Data.Data (Data)

12−import qualified Data.List as L

13−import SrcLoc

14−

15−-- | A wrapper for a shebang.

16−newtype Shebang = Shebang (Located String)

17− deriving (Eq, Data)

18−

19−-- | Extract shebangs from the beginning of a comment stream.

20−extractShebangs :: [Located String] -> ([Shebang], [Located String])

21−extractShebangs comments = (Shebang <$> shebangs, rest)

22− where

23− (shebangs, rest) = span (isShebang . unLoc) comments

24−

25−-- | Return 'True' if given 'String' is a shebang.

26−isShebang :: String -> Bool

27−isShebang str = "#!" `L.isPrefixOf` str

src/Ormolu/Printer.hs+11−17

@@ -11,7 +11,6 @@ import Ormolu.Parser.Result

11import Ormolu.Printer.Combinators

12import Ormolu.Printer.Meat.Module

13import Ormolu.Printer.SpanStream

14−import Ormolu.Processing.Postprocess (postprocess)

15

14

16-- | Render a module.

15-- | Render a module.

17printModule ::

16printModule ::

@@ -20,19 +19,14 @@ printModule ::

20 -- | Resulting rendition

19 -- | Resulting rendition

21 Text

20 Text

22printModule ParseResult {..} =

21printModule ParseResult {..} =

23− prLiteralPrefix <> region <> prLiteralSuffix

24− where

25− region =

26− postprocess prIndent $

27− runR

22+ runR

28− ( p_hsModule

23+ ( p_hsModule

29− prStackHeader

30− prShebangs

24+ prShebangs

31− prPragmas

25+ prExtensions

32− prImportQualifiedPost

26+ prImportQualifiedPost

33− prParsedSource

27+ prParsedSource

34− )

28+ )

35− (mkSpanStream prParsedSource)

29+ (mkSpanStream prParsedSource)

36− prCommentStream

30+ prCommentStream

37− prAnns

31+ prAnns

38− prUseRecordDot

32+ prUseRecordDot

src/Ormolu/Printer/Combinators.hs+29−39

@@ -21,7 +21,6 @@ module Ormolu.Printer.Combinators

21 space,

22 newline,

23 inci,

24− inciIf,

25 located,

24 located,

26 located',

25 located',

27 switchLayout,

26 switchLayout,

@@ -52,57 +51,57 @@ module Ormolu.Printer.Combinators

52

51

53 -- ** Literals

52 -- ** Literals

54 comma,

53 comma,

55− commaDel,

56− equals,

57

54

58− -- ** Stateful markers

55+ -- ** Comments

59− SpanMark (..),

60− spanMarkSpan,

61 HaddockStyle (..),

56 HaddockStyle (..),

62− setSpanMark,

57+ setLastCommentSpan,

63− getSpanMark,

58+ getLastCommentSpan,

64 )

59 )

65where

60where

66

61

67import Control.Monad

62import Control.Monad

63+import Data.Data (Data)

68import Data.List (intersperse)

64import Data.List (intersperse)

69import Data.Text (Text)

65import Data.Text (Text)

70import Ormolu.Printer.Comments

66import Ormolu.Printer.Comments

71import Ormolu.Printer.Internal

67import Ormolu.Printer.Internal

68+import Ormolu.Utils (isModule)

72import SrcLoc

69import SrcLoc

73

70

74----------------------------------------------------------------------------

71----------------------------------------------------------------------------

75-- Basic

72-- Basic

76

73

77−-- | Indent the inner expression if the first argument is 'True'.

78−inciIf ::

79− -- | Whether to indent

80− Bool ->

81− -- | The expression to indent

82− R () ->

83− R ()

84−inciIf b m = if b then inci m else m

85−

86-- | Enter a 'Located' entity. This combinator handles outputting comments

74-- | Enter a 'Located' entity. This combinator handles outputting comments

87-- and sets layout (single-line vs multi-line) for the inner computation.

75-- and sets layout (single-line vs multi-line) for the inner computation.

88-- Roughly, the rule for using 'located' is that every time there is a

76-- Roughly, the rule for using 'located' is that every time there is a

89-- 'Located' wrapper, it should be “discharged” with a corresponding

77-- 'Located' wrapper, it should be “discharged” with a corresponding

90-- 'located' invocation.

78-- 'located' invocation.

91located ::

79located ::

80+ Data a =>

92 -- | Thing to enter

81 -- | Thing to enter

93 Located a ->

82 Located a ->

94 -- | How to render inner value

83 -- | How to render inner value

95 (a -> R ()) ->

84 (a -> R ()) ->

96 R ()

85 R ()

97−located (L (UnhelpfulSpan _) a) f = f a

86+located loc f = do

98−located (L (RealSrcSpan l) a) f = do

87+ let withRealLocated (L l a) g =

88+ case l of

89+ UnhelpfulSpan _ -> return ()

90+ RealSrcSpan l' -> g (L l' a)

99− spitPrecedingComments l

91+ withRealLocated loc spitPrecedingComments

92+ let setEnclosingSpan =

93+ case getLoc loc of

94+ UnhelpfulSpan _ -> id

95+ RealSrcSpan orf ->

96+ if isModule (unLoc loc)

97+ then id

100− withEnclosingSpan l $

98+ else withEnclosingSpan orf

101− switchLayout [RealSrcSpan l] (f a)

99+ setEnclosingSpan $ switchLayout [getLoc loc] (f (unLoc loc))

102− spitFollowingComments l

100+ withRealLocated loc spitFollowingComments

103

101

104-- | A version of 'located' with arguments flipped.

102-- | A version of 'located' with arguments flipped.

105located' ::

103located' ::

104+ Data a =>

106 -- | How to render inner value

105 -- | How to render inner value

107 (a -> R ()) ->

106 (a -> R ()) ->

108 -- | Thing to enter

107 -- | Thing to enter

@@ -185,12 +184,10 @@ sepSemi f xs = vlayout singleLine multiLine

185 xs' ->

184 xs' ->

186 if ub

185 if ub

187 then do

186 then do

188− txt "{"

187+ txt "{ "

189− space

190− sep (txt ";" >> space) (dontUseBraces . f) xs'

188+ sep (txt "; ") (dontUseBraces . f) xs'

191− space

192− txt "}"

189+ txt " }"

193− else sep (txt ";" >> space) f xs'

190+ else sep (txt "; ") f xs'

194 multiLine =

191 multiLine =

195 sep newline (dontUseBraces . f) xs

192 sep newline (dontUseBraces . f) xs

196

193

@@ -203,7 +200,6 @@ data BracketStyle

203 N

200 N

204 | -- | Shifted one level

201 | -- | Shifted one level

205 S

202 S

206− deriving (Eq, Show)

207

203

208-- | Surround given entity by backticks.

204-- | Surround given entity by backticks.

209backticks :: R () -> R ()

205backticks :: R () -> R ()

@@ -280,7 +276,9 @@ brackets_ needBreaks open close style m = sitcc (vlayout singleLine multiLine)

280 then newline >> inci m

276 then newline >> inci m

281 else space >> sitcc m

277 else space >> sitcc m

282 newline

278 newline

279+ case style of

280+ N -> txt close

283− inciIf (style == S) (txt close)

281+ S -> inci (txt close)

284

282

285----------------------------------------------------------------------------

283----------------------------------------------------------------------------

286-- Literals

284-- Literals

@@ -288,11 +286,3 @@ brackets_ needBreaks open close style m = sitcc (vlayout singleLine multiLine)

288-- | Print @,@.

286-- | Print @,@.

289comma :: R ()

287comma :: R ()

290comma = txt ","

288comma = txt ","

291−

292−-- | Delimiting combination with 'comma'. To be used with 'sep'.

293−commaDel :: R ()

294−commaDel = comma >> breakpoint

295−

296−-- | Print @=@. Do not use @'txt' "="@.

297−equals :: R ()

298−equals = interferingTxt "="

src/Ormolu/Printer/Comments.hs+76−62

@@ -1,4 +1,3 @@

1−{-# LANGUAGE LambdaCase #-}

2{-# LANGUAGE OverloadedStrings #-}

1{-# LANGUAGE OverloadedStrings #-}

3

2

4-- | Helpers for formatting of comments. This is low-level code, use

3-- | Helpers for formatting of comments. This is low-level code, use

@@ -7,17 +6,21 @@ module Ormolu.Printer.Comments

7 ( spitPrecedingComments,

6 ( spitPrecedingComments,

8 spitFollowingComments,

7 spitFollowingComments,

9 spitRemainingComments,

8 spitRemainingComments,

10− spitCommentNow,

9+ spitStackHeader,

11− spitCommentPending,

12 )

10 )

13where

11where

14

12

15import Control.Monad

13import Control.Monad

14+import Data.Char (isSpace)

15+import Data.Coerce (coerce)

16+import Data.Data (Data)

17+import Data.List (isPrefixOf)

16import qualified Data.List.NonEmpty as NE

18import qualified Data.List.NonEmpty as NE

17−import Data.Maybe (listToMaybe)

19+import Data.List.NonEmpty (NonEmpty (..))

18import qualified Data.Text as T

20import qualified Data.Text as T

19import Ormolu.Parser.CommentStream

21import Ormolu.Parser.CommentStream

20import Ormolu.Printer.Internal

22import Ormolu.Printer.Internal

23+import Ormolu.Utils (isModule)

21import SrcLoc

24import SrcLoc

22

25

23----------------------------------------------------------------------------

26----------------------------------------------------------------------------

@@ -25,91 +28,108 @@ import SrcLoc

25

28

26-- | Output all preceding comments for an element at given location.

29-- | Output all preceding comments for an element at given location.

27spitPrecedingComments ::

30spitPrecedingComments ::

31+ Data a =>

28− -- | Span of the element to attach comments to

32+ -- | AST element to attach comments to

29− RealSrcSpan ->

33+ RealLocated a ->

30 R ()

34 R ()

31spitPrecedingComments ref = do

35spitPrecedingComments ref = do

32 gotSome <- handleCommentSeries (spitPrecedingComment ref)

36 gotSome <- handleCommentSeries (spitPrecedingComment ref)

33 when gotSome $ do

37 when gotSome $ do

34− lastMark <- getSpanMark

35 -- Insert a blank line between the preceding comments and the thing

38 -- Insert a blank line between the preceding comments and the thing

36 -- after them if there was a blank line in the input.

39 -- after them if there was a blank line in the input.

40+ lastSpn <- fmap snd <$> getLastCommentSpan

37− when (needsNewlineBefore ref lastMark) newline

41+ when (needsNewlineBefore (getRealSrcSpan ref) lastSpn) newline

38

42

39-- | Output all comments following an element at given location.

43-- | Output all comments following an element at given location.

40spitFollowingComments ::

44spitFollowingComments ::

45+ Data a =>

41− -- | Span of the element to attach comments to

46+ -- | AST element of attach comments to

42− RealSrcSpan ->

47+ RealLocated a ->

43 R ()

48 R ()

44spitFollowingComments ref = do

49spitFollowingComments ref = do

45− trimSpanStream ref

50+ trimSpanStream (getRealSrcSpan ref)

46 void $ handleCommentSeries (spitFollowingComment ref)

51 void $ handleCommentSeries (spitFollowingComment ref)

47

52

48-- | Output all remaining comments in the comment stream.

53-- | Output all remaining comments in the comment stream.

49spitRemainingComments :: R ()

54spitRemainingComments :: R ()

50−spitRemainingComments = do

51− -- Make sure we have a blank a line between the last definition and the

52− -- trailing comments.

53− newline

54− void $ handleCommentSeries spitRemainingComment

55+spitRemainingComments = void $ handleCommentSeries spitRemainingComment

56+

57+-- | If there is a stack header in the comment stream, print it.

58+spitStackHeader :: R ()

59+spitStackHeader = do

60+ let isStackHeader (Comment (x :| _)) =

61+ "stack" `isPrefixOf` dropWhile isSpace (drop 2 x)

62+ mstackHeader <- popComment (isStackHeader . unRealSrcSpan)

63+ forM_ mstackHeader $ \(L spn x) -> do

64+ spitCommentNow spn x

65+ newline

55

66

56----------------------------------------------------------------------------

67----------------------------------------------------------------------------

57-- Single-comment functions

68-- Single-comment functions

58

69

59-- | Output a single preceding comment for an element at given location.

70-- | Output a single preceding comment for an element at given location.

60spitPrecedingComment ::

71spitPrecedingComment ::

72+ Data a =>

61− -- | Span of the element to attach comments to

73+ -- | AST element to attach comments to

74+ RealLocated a ->

75+ -- | Location of last comment in the series

62− RealSrcSpan ->

76+ Maybe RealSrcSpan ->

63 -- | Are we done?

77 -- | Are we done?

64 R Bool

78 R Bool

65−spitPrecedingComment ref = do

79+spitPrecedingComment (L ref a) mlastSpn = do

66− mlastMark <- getSpanMark

67 let p (L l _) = realSrcSpanEnd l <= realSrcSpanStart ref

80 let p (L l _) = realSrcSpanEnd l <= realSrcSpanStart ref

68 withPoppedComment p $ \l comment -> do

81 withPoppedComment p $ \l comment -> do

69− lineSpans <- thisLineSpans

82+ dirtyLine <-

70− let thisCommentLine = srcLocLine (realSrcSpanStart l)

83+ case mlastSpn of

71− needsNewline =

84+ -- When the current line is dirty it means that something that can

72− case listToMaybe lineSpans of

85+ -- have comments attached to it is already on the line. To avoid

73− Nothing -> False

86+ -- problems with idempotence we cannot output the first comment

74− Just spn -> srcLocLine (realSrcSpanEnd spn) /= thisCommentLine

87+ -- immediately because it'll be attached to the previous element (on

88+ -- the same line) on the next run, so we play safe here and output

89+ -- an extra 'newline' in this case.

90+ Nothing -> isLineDirty -- only for very first preceding comment

91+ Just _ -> return False

75− when (needsNewline || needsNewlineBefore l mlastMark) newline

92+ when (dirtyLine || needsNewlineBefore l mlastSpn) newline

76 spitCommentNow l comment

93 spitCommentNow l comment

77− if theSameLinePre l ref

94+ if theSameLinePre l ref && not (isModule a)

78 then space

95 then space

79 else newline

96 else newline

80

97

81-- | Output a comment that follows element at given location immediately on

98-- | Output a comment that follows element at given location immediately on

82-- the same line, if there is any.

99-- the same line, if there is any.

83spitFollowingComment ::

100spitFollowingComment ::

101+ Data a =>

84 -- | AST element to attach comments to

102 -- | AST element to attach comments to

103+ RealLocated a ->

104+ -- | Location of last comment in the series

85− RealSrcSpan ->

105+ Maybe RealSrcSpan ->

86 -- | Are we done?

106 -- | Are we done?

87 R Bool

107 R Bool

88−spitFollowingComment ref = do

108+spitFollowingComment (L ref a) mlastSpn = do

89− mlastMark <- getSpanMark

90 mnSpn <- nextEltSpan

109 mnSpn <- nextEltSpan

91 -- Get first enclosing span that is not equal to reference span, i.e. it's

110 -- Get first enclosing span that is not equal to reference span, i.e. it's

92 -- truly something enclosing the AST element.

111 -- truly something enclosing the AST element.

93 meSpn <- getEnclosingSpan (/= ref)

112 meSpn <- getEnclosingSpan (/= ref)

94− withPoppedComment (commentFollowsElt ref mnSpn meSpn mlastMark) $ \l comment ->

113+ withPoppedComment (commentFollowsElt ref mnSpn meSpn mlastSpn) $ \l comment ->

95− if theSameLinePost l ref

114+ if theSameLinePost l ref && not (isModule a)

96 then

115 then

97 if isMultilineComment comment

116 if isMultilineComment comment

98 then space >> spitCommentNow l comment

117 then space >> spitCommentNow l comment

99 else spitCommentPending OnTheSameLine l comment

118 else spitCommentPending OnTheSameLine l comment

100 else do

119 else do

101− when (needsNewlineBefore l mlastMark) $

120+ when (needsNewlineBefore l mlastSpn) $

102 registerPendingCommentLine OnNextLine ""

121 registerPendingCommentLine OnNextLine ""

103 spitCommentPending OnNextLine l comment

122 spitCommentPending OnNextLine l comment

104

123

105-- | Output a single remaining comment from the comment stream.

124-- | Output a single remaining comment from the comment stream.

106spitRemainingComment ::

125spitRemainingComment ::

126+ -- | Location of last comment in the series

127+ Maybe RealSrcSpan ->

107 -- | Are we done?

128 -- | Are we done?

108 R Bool

129 R Bool

109−spitRemainingComment = do

130+spitRemainingComment mlastSpn =

110− mlastMark <- getSpanMark

111 withPoppedComment (const True) $ \l comment -> do

131 withPoppedComment (const True) $ \l comment -> do

112− when (needsNewlineBefore l mlastMark) newline

132+ when (needsNewlineBefore l mlastSpn) newline

113 spitCommentNow l comment

133 spitCommentNow l comment

114 newline

134 newline

115

135

@@ -120,13 +140,13 @@ spitRemainingComment = do

120handleCommentSeries ::

140handleCommentSeries ::

121 -- | Given location of previous comment, output the next comment

141 -- | Given location of previous comment, output the next comment

122 -- returning 'True' if we're done

142 -- returning 'True' if we're done

123− R Bool ->

143+ (Maybe RealSrcSpan -> R Bool) ->

124 -- | Whether we printed any comments

144 -- | Whether we printed any comments

125 R Bool

145 R Bool

126handleCommentSeries f = go False

146handleCommentSeries f = go False

127 where

147 where

128 go gotSome = do

148 go gotSome = do

129− done <- f

149+ done <- getLastCommentSpan >>= f . fmap snd

130 if done

150 if done

131 then return gotSome

151 then return gotSome

132 else go True

152 else go True

@@ -152,14 +172,13 @@ needsNewlineBefore ::

152 -- | Current comment span

172 -- | Current comment span

153 RealSrcSpan ->

173 RealSrcSpan ->

154 -- | Last printed comment span

174 -- | Last printed comment span

155− Maybe SpanMark ->

175+ Maybe RealSrcSpan ->

156 Bool

176 Bool

157−needsNewlineBefore _ (Just (HaddockSpan _ _)) = True

177+needsNewlineBefore l mlastSpn =

158−needsNewlineBefore l mlastMark =

178+ case mlastSpn of

159− case spanMarkSpan <$> mlastMark of

160 Nothing -> False

179 Nothing -> False

161− Just lastMark ->

180+ Just lastSpn ->

162− srcSpanStartLine l > srcSpanEndLine lastMark + 1

181+ srcSpanStartLine l > srcSpanEndLine lastSpn + 1

163

182

164-- | Is the preceding comment and AST element are on the same line?

183-- | Is the preceding comment and AST element are on the same line?

165theSameLinePre ::

184theSameLinePre ::

@@ -190,11 +209,11 @@ commentFollowsElt ::

190 -- | Location of enclosing AST element

209 -- | Location of enclosing AST element

191 Maybe RealSrcSpan ->

210 Maybe RealSrcSpan ->

192 -- | Location of last comment in the series

211 -- | Location of last comment in the series

193− Maybe SpanMark ->

212+ Maybe RealSrcSpan ->

194 -- | Comment to test

213 -- | Comment to test

195 RealLocated Comment ->

214 RealLocated Comment ->

196 Bool

215 Bool

197−commentFollowsElt ref mnSpn meSpn mlastMark (L l comment) =

216+commentFollowsElt ref mnSpn meSpn mlastSpn (L l comment) =

198 -- A comment follows a AST element if all 4 conditions are satisfied:

217 -- A comment follows a AST element if all 4 conditions are satisfied:

199 goesAfter

218 goesAfter

200 && logicallyFollows

219 && logicallyFollows

@@ -207,8 +226,9 @@ commentFollowsElt ref mnSpn meSpn mlastMark (L l comment) =

207 -- 2) The comment logically belongs to the element, four cases:

226 -- 2) The comment logically belongs to the element, four cases:

208 logicallyFollows =

227 logicallyFollows =

209 theSameLinePost l ref -- a) it's on the same line

228 theSameLinePost l ref -- a) it's on the same line

229+ || isPrevHaddock comment -- b) it's a Haddock string starting with -- ^

210− || continuation -- b) it's a continuation of a comment block

230+ || continuation -- c) it's a continuation of a comment block

211− || lastInEnclosing -- c) it's the last element in the enclosing construct

231+ || lastInEnclosing -- d) it's the last element in the enclosing construct

212

232

213 -- 3) There is no other AST element between this element and the comment:

233 -- 3) There is no other AST element between this element and the comment:

214 noEltBetween =

234 noEltBetween =

@@ -235,15 +255,9 @@ commentFollowsElt ref mnSpn meSpn mlastMark (L l comment) =

235 >= abs (startColumn ref - startColumn l)

255 >= abs (startColumn ref - startColumn l)

236 )

256 )

237 continuation =

257 continuation =

238− -- A comment is a continuation when it doesn't have non-whitespace

239− -- lexemes in front of it and goes right after the previous comment.

240− not (hasAtomsBefore comment)

241− && ( case mlastMark of

258+ case mlastSpn of

242− Just (HaddockSpan _ _) -> False

259+ Nothing -> False

243− Just (CommentSpan spn) ->

244− srcSpanEndLine spn + 1 == srcSpanStartLine l

260+ Just spn -> srcSpanEndLine spn + 1 == srcSpanStartLine l

245− _ -> False

246− )

247 lastInEnclosing =

261 lastInEnclosing =

248 case meSpn of

262 case meSpn of

249 -- When there is no enclosing element, return false

263 -- When there is no enclosing element, return false

@@ -265,9 +279,9 @@ spitCommentNow spn comment = do

265 . sequence_

279 . sequence_

266 . NE.intersperse newline

280 . NE.intersperse newline

267 . fmap (txt . T.pack)

281 . fmap (txt . T.pack)

268− . unComment

282+ . coerce

269 $ comment

283 $ comment

270− setSpanMark (CommentSpan spn)

284+ setLastCommentSpan Nothing spn

271

285

272-- | Output a 'Comment' at the end of correct line or after it depending on

286-- | Output a 'Comment' at the end of correct line or after it depending on

273-- 'CommentPosition'. Used for comments that may potentially follow on the

287-- 'CommentPosition'. Used for comments that may potentially follow on the

@@ -281,6 +295,6 @@ spitCommentPending position spn comment = do

281 . sequence_

295 . sequence_

282 . NE.toList

296 . NE.toList

283 . fmap (registerPendingCommentLine position . T.pack)

297 . fmap (registerPendingCommentLine position . T.pack)

284− . unComment

298+ . coerce

285 $ comment

299 $ comment

286− setSpanMark (CommentSpan spn)

300+ setLastCommentSpan Nothing spn

src/Ormolu/Printer/Internal.hs+96−149

@@ -1,5 +1,4 @@

1{-# LANGUAGE GeneralizedNewtypeDeriving #-}

2−{-# LANGUAGE LambdaCase #-}

3{-# LANGUAGE OverloadedStrings #-}

2{-# LANGUAGE OverloadedStrings #-}

4{-# LANGUAGE RankNTypes #-}

3{-# LANGUAGE RankNTypes #-}

5

4

@@ -14,10 +13,10 @@ module Ormolu.Printer.Internal

14

13

15 -- * Internal functions

14 -- * Internal functions

16 txt,

15 txt,

17− interferingTxt,

18 atom,

16 atom,

19 space,

17 space,

20 newline,

18 newline,

19+ isLineDirty,

21 useRecordDot,

20 useRecordDot,

22 inci,

21 inci,

23 sitcc,

22 sitcc,

@@ -39,14 +38,9 @@ module Ormolu.Printer.Internal

39 popComment,

38 popComment,

40 getEnclosingSpan,

39 getEnclosingSpan,

41 withEnclosingSpan,

40 withEnclosingSpan,

42− thisLineSpans,

43−

44− -- * Stateful markers

45− SpanMark (..),

46− spanMarkSpan,

47 HaddockStyle (..),

41 HaddockStyle (..),

48− setSpanMark,

42+ setLastCommentSpan,

49− getSpanMark,

43+ getLastCommentSpan,

50

44

51 -- * Annotations

45 -- * Annotations

52 getAnns,

46 getAnns,

@@ -79,44 +73,45 @@ newtype R a = R (ReaderT RC (State SC) a)

79

73

80-- | Reader context of 'R'. This should be used when we control rendering by

74-- | Reader context of 'R'. This should be used when we control rendering by

81-- enclosing certain expressions with wrappers.

75-- enclosing certain expressions with wrappers.

82−data RC = RC

76+data RC

77+ = RC

83− { -- | Indentation level, as the column index we need to start from after

78+ { -- | Indentation level, as the column index we need to start from after

84− -- a newline if we break lines

79+ -- a newline if we break lines

85− rcIndent :: !Int,

80+ rcIndent :: !Int,

86− -- | Current layout

81+ -- | Current layout

87− rcLayout :: Layout,

82+ rcLayout :: Layout,

88− -- | Spans of enclosing elements of AST

83+ -- | Spans of enclosing elements of AST

89− rcEnclosingSpans :: [RealSrcSpan],

84+ rcEnclosingSpans :: [RealSrcSpan],

90− -- | Collection of annotations

85+ -- | Collection of annotations

91− rcAnns :: Anns,

86+ rcAnns :: Anns,

92− -- | Whether the last expression in the layout can use braces

87+ -- | Whether the last expression in the layout can use braces

93− rcCanUseBraces :: Bool,

88+ rcCanUseBraces :: Bool,

94− -- | Whether the source could have used the record dot preprocessor

89+ -- | Whether the source could have used the record dot preprocessor

95− rcUseRecDot :: Bool

90+ rcUseRecDot :: Bool

96− }

91+ }

97

92

98-- | State context of 'R'.

93-- | State context of 'R'.

99−data SC = SC

94+data SC

95+ = SC

100− { -- | Index of the next column to render

96+ { -- | Index of the next column to render

101− scColumn :: !Int,

97+ scColumn :: !Int,

102− -- | Indentation level that was used for the current line

103− scIndent :: !Int,

104− -- | Rendered source code so far

98+ -- | Rendered source code so far

105− scBuilder :: Builder,

99+ scBuilder :: Builder,

106− -- | Span stream

100+ -- | Span stream

107− scSpanStream :: SpanStream,

101+ scSpanStream :: SpanStream,

108− -- | Spans of atoms that have been printed on the current line so far

109− scThisLineSpans :: [RealSrcSpan],

110− -- | Comment stream

102+ -- | Comment stream

111− scCommentStream :: CommentStream,

103+ scCommentStream :: CommentStream,

112− -- | Pending comment lines (in reverse order) to be inserted before next

104+ -- | Pending comment lines (in reverse order) to be inserted before next

113− -- newline, 'Int' is the indentation level

105+ -- newline, 'Int' is the indentation level

114− scPendingComments :: ![(CommentPosition, Text)],

106+ scPendingComments :: ![(CommentPosition, Int, Text)],

107+ -- | Whether the current line is “dirty”, that is, already contains

108+ -- atoms that can have comments attached to them

109+ scDirtyLine :: !Bool,

115− -- | Whether to output a space before the next output

110+ -- | Whether to output a space before the next output

116− scRequestedDelimiter :: !RequestedDelimiter,

111+ scRequestedDelimiter :: !RequestedDelimiter,

117− -- | An auxiliary marker for keeping track of last output element

112+ -- | Span of last output comment

118− scSpanMark :: !(Maybe SpanMark)

113+ scLastCommentSpan :: !(Maybe (Maybe HaddockStyle, RealSrcSpan))

119− }

114+ }

120

115

121-- | Make sure next output is delimited by one of the following.

116-- | Make sure next output is delimited by one of the following.

122data RequestedDelimiter

117data RequestedDelimiter

@@ -177,36 +172,18 @@ runR (R m) sstream cstream anns recDot =

177 sc =

172 sc =

178 SC

173 SC

179 { scColumn = 0,

174 { scColumn = 0,

180− scIndent = 0,

181 scBuilder = mempty,

175 scBuilder = mempty,

182 scSpanStream = sstream,

176 scSpanStream = sstream,

183− scThisLineSpans = [],

184 scCommentStream = cstream,

177 scCommentStream = cstream,

185 scPendingComments = [],

178 scPendingComments = [],

179+ scDirtyLine = False,

186 scRequestedDelimiter = VeryBeginning,

180 scRequestedDelimiter = VeryBeginning,

187− scSpanMark = Nothing

181+ scLastCommentSpan = Nothing

188 }

182 }

189

183

190----------------------------------------------------------------------------

184----------------------------------------------------------------------------

191-- Internal functions

185-- Internal functions

192

186

193−-- | Type of the thing to output. Influences the primary low-level rendering

194−-- function 'spit'.

195−data SpitType

196− = -- | Simple opaque text that breaks comment series.

197− SimpleText

198− | -- | Like 'SimpleText', but assume that when this text is inserted it

199− -- will separate an 'Atom' and its pending comments, so insert an extra

200− -- 'newline' in that case to force the pending comments and continue on

201− -- a fresh line.

202− InterferingText

203− | -- | An atom that typically have span information in the AST and can

204− -- have comments attached to it.

205− Atom

206− | -- | Used for rendering comment lines.

207− CommentPart

208− deriving (Show, Eq)

209−

210-- | Output a fixed 'Text' fragment. The argument may not contain any line

187-- | Output a fixed 'Text' fragment. The argument may not contain any line

211-- breaks. 'txt' is used to output all sorts of “fixed” bits of syntax like

188-- breaks. 'txt' is used to output all sorts of “fixed” bits of syntax like

212-- keywords and pipes @|@ in functional dependencies.

189-- keywords and pipes @|@ in functional dependencies.

@@ -218,15 +195,7 @@ txt ::

218 -- | 'Text' to output

195 -- | 'Text' to output

219 Text ->

196 Text ->

220 R ()

197 R ()

221−txt = spit SimpleText

198+txt = spit False False

222−

223−-- | Similar to 'txt' but the text inserted this way is assumed to break the

224−-- “link” between the preceding atom and its pending comments.

225−interferingTxt ::

226− -- | 'Text' to output

227− Text ->

228− R ()

229−interferingTxt = spit InterferingText

230

199

231-- | Output 'Outputable' fragment of AST. This can be used to output numeric

200-- | Output 'Outputable' fragment of AST. This can be used to output numeric

232-- literals and similar. Everything that doesn't have inner structure but

201-- literals and similar. Everything that doesn't have inner structure but

@@ -235,61 +204,48 @@ atom ::

235 Outputable a =>

204 Outputable a =>

236 a ->

205 a ->

237 R ()

206 R ()

238−atom = spit Atom . T.pack . showOutputable

207+atom = spit True False . T.pack . showOutputable

239

208

240-- | Low-level non-public helper to define 'txt' and 'atom'.

209-- | Low-level non-public helper to define 'txt' and 'atom'.

241spit ::

210spit ::

242− -- | Type of the thing to spit

211+ -- | Should we mark the line as dirty?

243− SpitType ->

212+ Bool ->

213+ -- | Used during outputting of pending comments?

214+ Bool ->

244 -- | 'Text' to output

215 -- | 'Text' to output

245 Text ->

216 Text ->

246 R ()

217 R ()

247−spit _ "" = return ()

218+spit dirty printingComments txt' = do

248−spit stype text = do

249 requestedDel <- R (gets scRequestedDelimiter)

219 requestedDel <- R (gets scRequestedDelimiter)

250− pendingComments <- R (gets scPendingComments)

251− when (stype == InterferingText && not (null pendingComments)) newline

252 case requestedDel of

220 case requestedDel of

253 RequestedNewline -> do

221 RequestedNewline -> do

254 R . modify $ \sc ->

222 R . modify $ \sc ->

255 sc

223 sc

256 { scRequestedDelimiter = RequestedNothing

224 { scRequestedDelimiter = RequestedNothing

257 }

225 }

258− case stype of

226+ if printingComments

259− CommentPart -> newlineRaw

227+ then newlineRaw

260− _ -> newline

228+ else newline

261 _ -> return ()

229 _ -> return ()

262 R $ do

230 R $ do

263 i <- asks rcIndent

231 i <- asks rcIndent

264 c <- gets scColumn

232 c <- gets scColumn

265− closestEnclosing <- listToMaybe <$> asks rcEnclosingSpans

233+ let spaces =

266− let indentedTxt = spaces <> text

234+ if c < i

267− spaces = T.replicate spacesN " "

235+ then T.replicate (i - c) " "

268− spacesN =

269− if c == 0

270− then i

271− else bool 0 1 (requestedDel == RequestedSpace)

236+ else bool mempty " " (requestedDel == RequestedSpace)

237+ indentedTxt = spaces <> txt'

272 modify $ \sc ->

238 modify $ \sc ->

273 sc

239 sc

274 { scBuilder = scBuilder sc <> fromText indentedTxt,

240 { scBuilder = scBuilder sc <> fromText indentedTxt,

275 scColumn = scColumn sc + T.length indentedTxt,

241 scColumn = scColumn sc + T.length indentedTxt,

276− scIndent =

242+ scDirtyLine = scDirtyLine sc || dirty,

277− if c == 0

278− then i

279− else scIndent sc,

280− scThisLineSpans =

281− let xs = scThisLineSpans sc

282− in case stype of

283− Atom -> case closestEnclosing of

284− Nothing -> xs

285− Just x -> x : xs

286− _ -> xs,

287 scRequestedDelimiter = RequestedNothing,

243 scRequestedDelimiter = RequestedNothing,

288− scSpanMark =

244+ scLastCommentSpan =

289 -- If there are pending comments, do not reset last comment

245 -- If there are pending comments, do not reset last comment

290 -- location.

246 -- location.

291− if (stype == CommentPart) || (not . null . scPendingComments) sc

247+ if printingComments || (not . null . scPendingComments) sc

292− then scSpanMark sc

248+ then scLastCommentSpan sc

293 else Nothing

249 else Nothing

294 }

250 }

295

251

@@ -320,22 +276,21 @@ space = R . modify $ \sc ->

320-- hard to output more than one blank newline in a row.

276-- hard to output more than one blank newline in a row.

321newline :: R ()

277newline :: R ()

322newline = do

278newline = do

323− indent <- R (gets scIndent)

324 cs <- reverse <$> R (gets scPendingComments)

279 cs <- reverse <$> R (gets scPendingComments)

325 case cs of

280 case cs of

326 [] -> newlineRaw

281 [] -> newlineRaw

327− ((position, _) : _) -> do

282+ ((position, _, _) : _) -> do

328 case position of

283 case position of

329 OnTheSameLine -> space

284 OnTheSameLine -> space

330 OnNextLine -> newlineRaw

285 OnNextLine -> newlineRaw

331− R . forM_ cs $ \(_, text) ->

286+ R . forM_ cs $ \(_, indent, txt') ->

332 let modRC rc =

287 let modRC rc =

333 rc

288 rc

334 { rcIndent = indent

289 { rcIndent = indent

335 }

290 }

336 R m = do

291 R m = do

337− unless (T.null text) $

292+ unless (T.null txt') $

338− spit CommentPart text

293+ spit False True txt'

339 newlineRaw

294 newlineRaw

340 in local modRC m

295 in local modRC m

341 R . modify $ \sc ->

296 R . modify $ \sc ->

@@ -356,8 +311,7 @@ newlineRaw = R . modify $ \sc ->

356 VeryBeginning -> builderSoFar

311 VeryBeginning -> builderSoFar

357 _ -> builderSoFar <> "\n",

312 _ -> builderSoFar <> "\n",

358 scColumn = 0,

313 scColumn = 0,

359− scIndent = 0,

314+ scDirtyLine = False,

360− scThisLineSpans = [],

361 scRequestedDelimiter = case scRequestedDelimiter sc of

315 scRequestedDelimiter = case scRequestedDelimiter sc of

362 AfterNewline -> RequestedNewline

316 AfterNewline -> RequestedNewline

363 RequestedNewline -> RequestedNewline

317 RequestedNewline -> RequestedNewline

@@ -365,9 +319,14 @@ newlineRaw = R . modify $ \sc ->

365 _ -> AfterNewline

319 _ -> AfterNewline

366 }

320 }

367

321

322+-- | Check if the current line is “dirty”, that is, there is something on it

323+-- that can have comments attached to it.

324+isLineDirty :: R Bool

325+isLineDirty = R (gets scDirtyLine)

326+

368-- | Return 'True' if we should print record dot syntax.

327-- | Return 'True' if we should print record dot syntax.

369useRecordDot :: R Bool

328useRecordDot :: R Bool

370−useRecordDot = R (asks rcUseRecDot)

329+useRecordDot = R $ asks rcUseRecDot

371

330

372-- | Increase indentation level by one indentation step for the inner

331-- | Increase indentation level by one indentation step for the inner

373-- computation. 'inci' should be used when a part of code must be more

332-- computation. 'inci' should be used when a part of code must be more

@@ -384,7 +343,8 @@ inci (R m) = R (local modRC m)

384

343

385-- | Set indentation level for the inner computation equal to current

344-- | Set indentation level for the inner computation equal to current

386-- column. This makes sure that the entire inner block is uniformly

345-- column. This makes sure that the entire inner block is uniformly

387−-- \"shifted\" to the right.

346+-- \"shifted\" to the right. Only works (and makes sense) when enclosing

347+-- layout is multi-line.

388sitcc :: R () -> R ()

348sitcc :: R () -> R ()

389sitcc (R m) = do

349sitcc (R m) = do

390 requestedDel <- R (gets scRequestedDelimiter)

350 requestedDel <- R (gets scRequestedDelimiter)

@@ -392,9 +352,16 @@ sitcc (R m) = do

392 c <- R (gets scColumn)

352 c <- R (gets scColumn)

393 let modRC rc =

353 let modRC rc =

394 rc

354 rc

395− { rcIndent = max i (c + bool 0 1 (requestedDel == RequestedSpace))

355+ { rcIndent = max i c + bool 0 1 (requestedDel == RequestedSpace)

396 }

356 }

357+ vlayout (R m) . R $ do

358+ modify $ \sc ->

359+ sc

360+ { scRequestedDelimiter = case requestedDel of

361+ RequestedSpace -> RequestedNothing

362+ other -> other

363+ }

397− R (local modRC m)

364+ local modRC m

398

365

399-- | Set 'Layout' for internal computation.

366-- | Set 'Layout' for internal computation.

400enterLayout :: Layout -> R () -> R ()

367enterLayout :: Layout -> R () -> R ()

@@ -435,10 +402,11 @@ registerPendingCommentLine ::

435 -- | 'Text' to output

402 -- | 'Text' to output

436 Text ->

403 Text ->

437 R ()

404 R ()

438−registerPendingCommentLine position text = R $ do

405+registerPendingCommentLine position txt' = R $ do

406+ i <- asks rcIndent

439 modify $ \sc ->

407 modify $ \sc ->

440 sc

408 sc

441− { scPendingComments = (position, text) : scPendingComments sc

409+ { scPendingComments = (position, i, txt') : scPendingComments sc

442 }

410 }

443

411

444-- | Drop elements that begin before or at the same place as given

412-- | Drop elements that begin before or at the same place as given

@@ -497,29 +465,6 @@ withEnclosingSpan spn (R m) = R (local modRC m)

497 { rcEnclosingSpans = spn : rcEnclosingSpans rc

465 { rcEnclosingSpans = spn : rcEnclosingSpans rc

498 }

466 }

499

467

500−-- | Get spans on this line so far.

501−thisLineSpans :: R [RealSrcSpan]

502−thisLineSpans = R (gets scThisLineSpans)

503−

504−----------------------------------------------------------------------------

505−-- Stateful markers

506−

507−-- | An auxiliary marker for keeping track of last output element.

508−data SpanMark

509− = -- | Haddock comment

510− HaddockSpan HaddockStyle RealSrcSpan

511− | -- | Non-haddock comment

512− CommentSpan RealSrcSpan

513− | -- | A statement in a do-block and such span

514− StatementSpan RealSrcSpan

515−

516−-- | Project 'RealSrcSpan' from 'SpanMark'.

517−spanMarkSpan :: SpanMark -> RealSrcSpan

518−spanMarkSpan = \case

519− HaddockSpan _ s -> s

520− CommentSpan s -> s

521− StatementSpan s -> s

522−

523-- | Haddock string style.

468-- | Haddock string style.

524data HaddockStyle

469data HaddockStyle

525 = -- | @-- |@

470 = -- | @-- |@

@@ -532,18 +477,20 @@ data HaddockStyle

532 Named String

477 Named String

533

478

534-- | Set span of last output comment.

479-- | Set span of last output comment.

535−setSpanMark ::

480+setLastCommentSpan ::

536− -- | Span mark to set

481+ -- | 'HaddockStyle' or 'Nothing' if it's a non-Haddock comment

537− SpanMark ->

482+ Maybe HaddockStyle ->

483+ -- | Location of last printed comment

484+ RealSrcSpan ->

538 R ()

485 R ()

539−setSpanMark spnMark = R . modify $ \sc ->

486+setLastCommentSpan mhStyle spn = R . modify $ \sc ->

540 sc

487 sc

541− { scSpanMark = Just spnMark

488+ { scLastCommentSpan = Just (mhStyle, spn)

542 }

489 }

543

490

544-- | Get span of last output comment.

491-- | Get span of last output comment.

545−getSpanMark :: R (Maybe SpanMark)

492+getLastCommentSpan :: R (Maybe (Maybe HaddockStyle, RealSrcSpan))

546−getSpanMark = R (gets scSpanMark)

493+getLastCommentSpan = R (gets scLastCommentSpan)

547

494

548----------------------------------------------------------------------------

495----------------------------------------------------------------------------

549-- Annotations

496-- Annotations

@@ -567,7 +514,7 @@ dontUseBraces (R r) = R (local (\i -> i {rcCanUseBraces = False}) r)

567

514

568-- | Return 'True' if we can use braces in this context.

515-- | Return 'True' if we can use braces in this context.

569canUseBraces :: R Bool

516canUseBraces :: R Bool

570−canUseBraces = R (asks rcCanUseBraces)

517+canUseBraces = R $ asks rcCanUseBraces

571

518

572----------------------------------------------------------------------------

519----------------------------------------------------------------------------

573-- Constants

520-- Constants

src/Ormolu/Printer/Meat/Common.hs+11−13

@@ -17,12 +17,14 @@ where

17

18import Control.Monad

19import Data.List (isPrefixOf)

20+import Data.Maybe (isJust)

20import qualified Data.Text as T

21import qualified Data.Text as T

21import GHC hiding (GhcPs, IE)

22import GHC hiding (GhcPs, IE)

22import Name (nameStableString)

23import Name (nameStableString)

23import OccName (OccName (..))

24import OccName (OccName (..))

24import Ormolu.Printer.Combinators

25import Ormolu.Printer.Combinators

25import Ormolu.Utils

26import Ormolu.Utils

27+import RdrName (RdrName (..))

26

28

27-- | Data and type family style.

29-- | Data and type family style.

28data FamilyStyle

30data FamilyStyle

@@ -114,14 +116,14 @@ p_qualName mname occName = do

114p_infixDefHelper ::

116p_infixDefHelper ::

115 -- | Whether to format in infix style

117 -- | Whether to format in infix style

116 Bool ->

118 Bool ->

117− -- | Whether to bump indentation for arguments

119+ -- | Indentation-bumping wrapper

118− Bool ->

120+ (R () -> R ()) ->

119 -- | How to print the operator\/name

121 -- | How to print the operator\/name

120 R () ->

122 R () ->

121 -- | How to print the arguments

123 -- | How to print the arguments

122 [R ()] ->

124 [R ()] ->

123 R ()

125 R ()

124−p_infixDefHelper isInfix indentArgs name args =

126+p_infixDefHelper isInfix inci' name args =

125 case (isInfix, args) of

127 case (isInfix, args) of

126 (True, p0 : p1 : ps) -> do

128 (True, p0 : p1 : ps) -> do

127 let parens' =

129 let parens' =

@@ -131,18 +133,18 @@ p_infixDefHelper isInfix indentArgs name args =

131 parens' $ do

133 parens' $ do

132 p0

134 p0

133 breakpoint

135 breakpoint

134− inci . sitcc $ do

136+ inci $ sitcc $ do

135 name

137 name

136 space

138 space

137 p1

139 p1

138− unless (null ps) . inciIf indentArgs $ do

140+ unless (null ps) . inci' $ do

139 breakpoint

141 breakpoint

140 sitcc (sep breakpoint sitcc ps)

142 sitcc (sep breakpoint sitcc ps)

141 (_, ps) -> do

143 (_, ps) -> do

142 name

144 name

143 unless (null ps) $ do

145 unless (null ps) $ do

144 breakpoint

146 breakpoint

145− inciIf indentArgs $ sitcc (sep breakpoint sitcc args)

147+ inci' $ sitcc (sep breakpoint sitcc args)

146

148

147-- | Print a Haddock.

149-- | Print a Haddock.

148p_hsDocString ::

150p_hsDocString ::

@@ -154,11 +156,7 @@ p_hsDocString ::

154 LHsDocString ->

156 LHsDocString ->

155 R ()

157 R ()

156p_hsDocString hstyle needsNewline (L l str) = do

158p_hsDocString hstyle needsNewline (L l str) = do

157− let isCommentSpan = \case

159+ goesAfterComment <- isJust <$> getLastCommentSpan

158− HaddockSpan _ _ -> True

159− CommentSpan _ -> True

160− _ -> False

161− goesAfterComment <- maybe False isCommentSpan <$> getSpanMark

162 -- Make sure the Haddock is separated by a newline from other comments.

160 -- Make sure the Haddock is separated by a newline from other comments.

163 when goesAfterComment newline

161 when goesAfterComment newline

164 forM_ (zip (splitDocString str) (True : repeat False)) $ \(x, isFirst) -> do

162 forM_ (zip (splitDocString str) (True : repeat False)) $ \(x, isFirst) -> do

@@ -177,8 +175,8 @@ p_hsDocString hstyle needsNewline (L l str) = do

177 -- It's often the case that the comment itself doesn't have a span

175 -- It's often the case that the comment itself doesn't have a span

178 -- attached to it and instead its location can be obtained from

176 -- attached to it and instead its location can be obtained from

179 -- nearest enclosing span.

177 -- nearest enclosing span.

180− getEnclosingSpan (const True) >>= mapM_ (setSpanMark . HaddockSpan hstyle)

178+ getEnclosingSpan (const True) >>= mapM_ (setLastCommentSpan (Just hstyle))

181− RealSrcSpan spn -> setSpanMark (HaddockSpan hstyle spn)

179+ RealSrcSpan spn -> setLastCommentSpan (Just hstyle) spn

182

180

183-- | Print anchor of named doc section.

181-- | Print anchor of named doc section.

184p_hsDocName :: String -> R ()

182p_hsDocName :: String -> R ()

src/Ormolu/Printer/Meat/Declaration.hs+58−76

@@ -12,7 +12,7 @@ module Ormolu.Printer.Meat.Declaration

12where

13

14import Data.List (sort)

15−import Data.List.NonEmpty (NonEmpty (..), (<|))

15+import Data.List.NonEmpty ((<|), NonEmpty (..))

16import qualified Data.List.NonEmpty as NE

17import GHC hiding (InlinePragma)

18import OccName (occNameFS)

@@ -34,6 +34,7 @@ import Ormolu.Printer.Meat.Declaration.Value

34import Ormolu.Printer.Meat.Declaration.Warning

35import Ormolu.Printer.Meat.Type

36import Ormolu.Utils

37+import RdrName (rdrNameOcc)

37

38

38data UserGrouping

39data UserGrouping

39 = -- | Always put newlines where we think they should be

40 = -- | Always put newlines where we think they should be

@@ -54,27 +55,25 @@ p_hsDeclsRespectGrouping :: FamilyStyle -> [LHsDecl GhcPs] -> R ()

54p_hsDeclsRespectGrouping = p_hsDecls' Respect

55p_hsDeclsRespectGrouping = p_hsDecls' Respect

55

56

56p_hsDecls' :: UserGrouping -> FamilyStyle -> [LHsDecl GhcPs] -> R ()

57p_hsDecls' :: UserGrouping -> FamilyStyle -> [LHsDecl GhcPs] -> R ()

57−p_hsDecls' grouping style decls =

58+p_hsDecls' grouping style decls = sepSemi id $

58− sepSemi id $

59− -- Return a list of rendered declarations, adding a newline to separate

59+ -- Return a list of rendered declarations, adding a newline to separate

60− -- groups.

60+ -- groups.

61− case groupDecls decls of

61+ case groupDecls decls of

62− [] -> []

62+ [] -> []

63− (x : xs) -> renderGroup x ++ concat (zipWith renderGroupWithPrev (x : xs) xs)

63+ (x : xs) -> renderGroup x ++ concat (zipWith renderGroupWithPrev (x : xs) xs)

64 where

65 renderGroup = NE.toList . fmap (located' $ dontUseBraces . p_hsDecl style)

66 renderGroupWithPrev prev curr =

67 -- We can omit a blank line when the user didn't add one, but we must

68 -- ensure we always add blank lines around documented declarations

69− case grouping of

69+ if or

70− Disregard ->

70+ [ grouping == Disregard,

71− breakpoint : renderGroup curr

72− Respect ->

73− if separatedByBlankNE getLoc prev curr

71+ separatedByBlank getLoc prev curr,

74− || isDocumented prev

72+ isDocumented prev,

75− || isDocumented curr

73+ isDocumented curr

74+ ]

76− then breakpoint : renderGroup curr

75+ then breakpoint : renderGroup curr

77− else renderGroup curr

76+ else renderGroup curr

78

77

79-- | Is a declaration group documented?

78-- | Is a declaration group documented?

80isDocumented :: NonEmpty (LHsDecl GhcPs) -> Bool

79isDocumented :: NonEmpty (LHsDecl GhcPs) -> Bool

@@ -85,6 +84,9 @@ isDocumented = any (isHaddock . unLoc)

85 isHaddock _ = False

84 isHaddock _ = False

86

85

87-- | Group relevant declarations together.

86-- | Group relevant declarations together.

87+--

88+-- Add a declaration to a group iff it is relevant to either the first or

89+-- the last declaration of the group.

88groupDecls :: [LHsDecl GhcPs] -> [NonEmpty (LHsDecl GhcPs)]

90groupDecls :: [LHsDecl GhcPs] -> [NonEmpty (LHsDecl GhcPs)]

89groupDecls [] = []

91groupDecls [] = []

90groupDecls (l@(L _ DocNext) : xs) =

92groupDecls (l@(L _ DocNext) : xs) =

@@ -93,13 +95,18 @@ groupDecls (l@(L _ DocNext) : xs) =

93 case groupDecls xs of

95 case groupDecls xs of

94 [] -> [l :| []]

96 [] -> [l :| []]

95 (x : xs') -> (l <| x) : xs'

97 (x : xs') -> (l <| x) : xs'

96−groupDecls (header : xs) =

98+groupDecls (lhdr : xs) =

97− let (grp, rest) = flip span (zip (header : xs) xs) $ \(previous, current) ->

99+ let -- Pick the first decl as the group header

100+ hdr = unLoc lhdr

101+ -- Zip rest of the decls with their previous decl

102+ zipped = zip (lhdr : xs) xs

103+ -- Pick decls from the tail if they are relevant to the group header

104+ -- or the previous decl.

105+ (grp, rest) = flip span zipped $ \(L _ prev, L _ cur) ->

98− let relevantToHdr = groupedDecls header current

106+ let relevantToHdr = groupedDecls hdr cur

99− relevantToPrev = groupedDecls previous current

107+ relevantToPrev = groupedDecls prev cur

100− isDeclSeries = declSeries previous current

101− in isDeclSeries || relevantToHdr || relevantToPrev

108+ in relevantToHdr || relevantToPrev

102− in (header :| map snd grp) : groupDecls (map snd rest)

109+ in (lhdr :| map snd grp) : groupDecls (map snd rest)

103

110

104p_hsDecl :: FamilyStyle -> HsDecl GhcPs -> R ()

111p_hsDecl :: FamilyStyle -> HsDecl GhcPs -> R ()

105p_hsDecl style = \case

112p_hsDecl style = \case

@@ -158,54 +165,35 @@ p_instDecl style = \case

158

165

159p_derivDecl :: DerivDecl GhcPs -> R ()

166p_derivDecl :: DerivDecl GhcPs -> R ()

160p_derivDecl = \case

167p_derivDecl = \case

161− d@DerivDecl {} -> p_standaloneDerivDecl d

168+ d@DerivDecl {..} -> p_standaloneDerivDecl d

162 XDerivDecl x -> noExtCon x

169 XDerivDecl x -> noExtCon x

163

170

164-- | Determine if these declarations should be grouped together.

171-- | Determine if these declarations should be grouped together.

165groupedDecls ::

172groupedDecls ::

166− LHsDecl GhcPs ->

173+ HsDecl GhcPs ->

167− LHsDecl GhcPs ->

168− Bool

169−groupedDecls (L l_x x') (L l_y y') =

170− case (x', y') of

171− (TypeSignature ns, FunctionBody ns') -> ns `intersects` ns'

172− (TypeSignature ns, DefaultSignature ns') -> ns `intersects` ns'

173− (DefaultSignature ns, TypeSignature ns') -> ns `intersects` ns'

174− (DefaultSignature ns, FunctionBody ns') -> ns `intersects` ns'

175− (x, FunctionBody ns) | Just ns' <- isPragma x -> ns `intersects` ns'

176− (FunctionBody ns, x) | Just ns' <- isPragma x -> ns `intersects` ns'

177− (x, DataDeclaration n) | Just ns <- isPragma x -> n `elem` ns

178− (DataDeclaration n, x)

179− | Just ns <- isPragma x ->

180− let f = occNameFS . rdrNameOcc in f n `elem` map f ns

181− (x, y)

182− | Just ns <- isPragma x,

183− Just ns' <- isPragma y ->

184− ns `intersects` ns'

185− (x, TypeSignature ns) | Just ns' <- isPragma x -> ns `intersects` ns'

186− (TypeSignature ns, x) | Just ns' <- isPragma x -> ns `intersects` ns'

187− (PatternSignature ns, Pattern n) -> n `elem` ns

188− (KindSignature n, DataDeclaration n') -> n == n'

189− (KindSignature n, ClassDeclaration n') -> n == n'

190− (KindSignature n, FamilyDeclaration n') -> n == n'

191− -- Special case for TH splices, we look at locations

192− (Splice, Splice) -> not (separatedByBlank id l_x l_y)

193− -- This looks only at Haddocks, normal comments are handled elsewhere

194− (DocNext, _) -> True

195− (_, DocPrev) -> True

196− _ -> False

197−

198−-- | Detect declaration series that should not have blanks between them.

199−declSeries ::

200− LHsDecl GhcPs ->

201− LHsDecl GhcPs ->

174+ HsDecl GhcPs ->

202 Bool

175 Bool

203−declSeries (L _ x) (L _ y) =

176+groupedDecls (TypeSignature ns) (FunctionBody ns') = ns `intersects` ns'

204− case (x, y) of

177+groupedDecls (TypeSignature ns) (DefaultSignature ns') = ns `intersects` ns'

205− ( SigD NoExtField (TypeSig NoExtField _ _),

178+groupedDecls (DefaultSignature ns) (TypeSignature ns') = ns `intersects` ns'

206− SigD NoExtField (TypeSig NoExtField _ _)

179+groupedDecls (DefaultSignature ns) (FunctionBody ns') = ns `intersects` ns'

207− ) -> True

180+groupedDecls x (FunctionBody ns) | Just ns' <- isPragma x = ns `intersects` ns'

181+groupedDecls (FunctionBody ns) x | Just ns' <- isPragma x = ns `intersects` ns'

182+groupedDecls x (DataDeclaration n) | Just ns <- isPragma x = n `elem` ns

183+groupedDecls (DataDeclaration n) x

184+ | Just ns <- isPragma x =

185+ let f = occNameFS . rdrNameOcc in f n `elem` map f ns

186+groupedDecls x y | Just ns <- isPragma x, Just ns' <- isPragma y = ns `intersects` ns'

187+groupedDecls x (TypeSignature ns) | Just ns' <- isPragma x = ns `intersects` ns'

188+groupedDecls (TypeSignature ns) x | Just ns' <- isPragma x = ns `intersects` ns'

189+groupedDecls (PatternSignature ns) (Pattern n) = n `elem` ns

190+groupedDecls (KindSignature n) (DataDeclaration n') = n == n'

191+groupedDecls (KindSignature n) (ClassDeclaration n') = n == n'

192+groupedDecls (KindSignature n) (FamilyDeclaration n') = n == n'

193+-- This looks only at Haddocks, normal comments are handled elsewhere

194+groupedDecls DocNext _ = True

195+groupedDecls _ DocPrev = True

208− _ -> False

196+groupedDecls _ _ = False

209

197

210intersects :: Ord a => [a] -> [a] -> Bool

198intersects :: Ord a => [a] -> [a] -> Bool

211intersects a b = go (sort a) (sort b)

199intersects a b = go (sort a) (sort b)

@@ -230,11 +218,6 @@ isPragma = \case

230 WarningPragma n -> Just n

218 WarningPragma n -> Just n

231 _ -> Nothing

219 _ -> Nothing

232

220

233−-- Declarations that do not refer to names

234−

235−pattern Splice :: HsDecl GhcPs

236−pattern Splice <- SpliceD NoExtField (SpliceDecl NoExtField _ _)

237−

238-- Declarations referring to a single name

221-- Declarations referring to a single name

239

222

240pattern

223pattern

@@ -299,10 +282,9 @@ patSigRdrNames (SigD NoExtField (PatSynSig NoExtField ns _)) = Just $ map unLoc

299patSigRdrNames _ = Nothing

282patSigRdrNames _ = Nothing

300

283

301warnSigRdrNames :: HsDecl GhcPs -> Maybe [RdrName]

284warnSigRdrNames :: HsDecl GhcPs -> Maybe [RdrName]

302−warnSigRdrNames (WarningD NoExtField (Warnings NoExtField _ ws)) = Just $

285+warnSigRdrNames (WarningD NoExtField (Warnings NoExtField _ ws)) = Just $ flip concatMap ws $ \case

303− flip concatMap ws $ \case

304− L _ (Warning NoExtField ns _) -> map unLoc ns

286+ L _ (Warning NoExtField ns _) -> map unLoc ns

305− L _ (XWarnDecl x) -> noExtCon x

287+ L _ (XWarnDecl x) -> noExtCon x

306warnSigRdrNames _ = Nothing

288warnSigRdrNames _ = Nothing

307

289

308patBindNames :: Pat GhcPs -> [RdrName]

290patBindNames :: Pat GhcPs -> [RdrName]

src/Ormolu/Printer/Meat/Declaration.hs-boot+1

@@ -9,4 +9,5 @@ import Ormolu.Printer.Combinators

9import Ormolu.Printer.Meat.Common

10

11p_hsDecls :: FamilyStyle -> [LHsDecl GhcPs] -> R ()

12+

12p_hsDeclsRespectGrouping :: FamilyStyle -> [LHsDecl GhcPs] -> R ()

13p_hsDeclsRespectGrouping :: FamilyStyle -> [LHsDecl GhcPs] -> R ()

src/Ormolu/Printer/Meat/Declaration/Class.hs+3−2

@@ -18,6 +18,7 @@ import Ormolu.Printer.Combinators

18import Ormolu.Printer.Meat.Common

19import {-# SOURCE #-} Ormolu.Printer.Meat.Declaration

20import Ormolu.Printer.Meat.Type

21+import RdrName (RdrName (..))

21

22

22p_classDecl ::

23p_classDecl ::

23 LHsContext GhcPs ->

24 LHsContext GhcPs ->

@@ -58,7 +59,7 @@ p_classDecl ctx name HsQTvs {..} fixity fdeps csigs cdefs cats catdefs cdocs = d

58 switchLayout signatureSpans $

59 switchLayout signatureSpans $

59 p_infixDefHelper

60 p_infixDefHelper

60 (isInfix fixity)

61 (isInfix fixity)

61− True

62+ inci

62 (p_rdrName name)

63 (p_rdrName name)

63 (located' p_hsTyVarBndr <$> hsq_explicit)

64 (located' p_hsTyVarBndr <$> hsq_explicit)

64 inci (p_classFundeps fdeps)

65 inci (p_classFundeps fdeps)

@@ -82,7 +83,7 @@ p_classFundeps fdeps = unless (null fdeps) $ do

82 breakpoint

83 breakpoint

83 txt "|"

84 txt "|"

84 space

85 space

85− inci $ sep commaDel (sitcc . located' p_funDep) fdeps

86+ sitcc $ sep (comma >> breakpoint) (sitcc . located' p_funDep) fdeps

86

87

87p_funDep :: FunDep (Located RdrName) -> R ()

88p_funDep :: FunDep (Located RdrName) -> R ()

88p_funDep (before, after) = do

89p_funDep (before, after) = do

src/Ormolu/Printer/Meat/Declaration/Data.hs+31−54

@@ -15,6 +15,8 @@ import Ormolu.Printer.Combinators

15import Ormolu.Printer.Meat.Common

16import Ormolu.Printer.Meat.Type

17import Ormolu.Utils

18+import RdrName (RdrName (..))

19+import SrcLoc (Located)

18

20

19p_dataDecl ::

21p_dataDecl ::

20 -- | Whether to format as data family

22 -- | Whether to format as data family

@@ -41,7 +43,7 @@ p_dataDecl style name tpats fixity HsDataDefn {..} = do

41 inci $

43 inci $

42 p_infixDefHelper

44 p_infixDefHelper

43 (isInfix fixity)

45 (isInfix fixity)

44− True

46+ inci

45 (p_rdrName name)

47 (p_rdrName name)

46 (located' p_hsType <$> tpats)

48 (located' p_hsType <$> tpats)

47 case dd_kindSig of

49 case dd_kindSig of

@@ -59,37 +61,25 @@ p_dataDecl style name tpats fixity HsDataDefn {..} = do

59 breakpoint

61 breakpoint

60 txt "where"

62 txt "where"

61 breakpoint

63 breakpoint

62− sepSemi (located' (p_conDecl False)) dd_cons

64+ sepSemi (located' p_conDecl) dd_cons

63− else switchLayout (getLoc name : (getLoc <$> dd_cons)) . inci $ do

65+ else switchLayout (getLoc name : (getLoc <$> dd_cons))

64− let singleConstRec = isSingleConstRec dd_cons

66+ $ inci

65− if singleConstRec

67+ $ do

66− then space

67− else

68− if hasHaddocks dd_cons

69− then newline

70− else breakpoint

68+ breakpoint

71− equals

69+ txt "="

72− space

70+ space

73− layout <- getLayout

74− let s =

71+ let s =

75− if layout == MultiLine || hasHaddocks dd_cons

72+ vlayout

76− then newline >> txt "|" >> space

73+ (space >> txt "|" >> space)

77− else space >> txt "|" >> space

74+ (newline >> txt "|" >> space)

78− sitcc' =

79− if singleConstRec

80− then id

81− else sitcc

82− sep s (sitcc' . located' (p_conDecl singleConstRec)) dd_cons

75+ sep s (sitcc . located' p_conDecl) dd_cons

83 unless (null $ unLoc dd_derivs) breakpoint

76 unless (null $ unLoc dd_derivs) breakpoint

84 inci . located dd_derivs $ \xs ->

77 inci . located dd_derivs $ \xs ->

85 sep newline (located' p_hsDerivingClause) xs

78 sep newline (located' p_hsDerivingClause) xs

86p_dataDecl _ _ _ _ (XHsDataDefn x) = noExtCon x

79p_dataDecl _ _ _ _ (XHsDataDefn x) = noExtCon x

87

80

88−p_conDecl ::

89− Bool ->

90− ConDecl GhcPs ->

81+p_conDecl :: ConDecl GhcPs -> R ()

91− R ()

92−p_conDecl singleConstRec = \case

82+p_conDecl = \case

93 ConDeclGADT {..} -> do

83 ConDeclGADT {..} -> do

94 mapM_ (p_hsDocString Pipe True) con_doc

84 mapM_ (p_hsDocString Pipe True) con_doc

95 let conDeclSpn =

85 let conDeclSpn =

@@ -104,10 +94,11 @@ p_conDecl singleConstRec = \case

104 (c : cs) -> do

94 (c : cs) -> do

105 p_rdrName c

95 p_rdrName c

106 unless (null cs) . inci $ do

96 unless (null cs) . inci $ do

107− commaDel

97+ comma

98+ breakpoint

108− sep commaDel p_rdrName cs

99+ sitcc $ sep (comma >> breakpoint) p_rdrName cs

100+ space

109 inci $ do

101 inci $ do

110− space

111 txt "::"

102 txt "::"

112 let interArgBreak =

103 let interArgBreak =

113 if hasDocStrings (unLoc con_res_ty)

104 if hasDocStrings (unLoc con_res_ty)

@@ -135,27 +126,26 @@ p_conDecl singleConstRec = \case

135 p_hsType (unLoc con_res_ty)

126 p_hsType (unLoc con_res_ty)

136 ConDeclH98 {..} -> do

127 ConDeclH98 {..} -> do

137 mapM_ (p_hsDocString Pipe True) con_doc

128 mapM_ (p_hsDocString Pipe True) con_doc

138− let conDeclWithContextSpn =

129+ let conDeclSpn =

130+ [getLoc con_name]

139− [getLoc con_forall]

131+ <> [getLoc con_forall]

140 <> fmap getLoc con_ex_tvs

132 <> fmap getLoc con_ex_tvs

141 <> maybeToList (fmap getLoc con_mb_cxt)

133 <> maybeToList (fmap getLoc con_mb_cxt)

142− <> conDeclSpn

143− conDeclSpn =

144− getLoc con_name : conArgsSpans con_args

134+ <> conArgsSpans con_args

145− switchLayout conDeclWithContextSpn $ do

135+ switchLayout conDeclSpn $ do

146 when (unLoc con_forall) $ do

136 when (unLoc con_forall) $ do

147 p_forallBndrs ForallInvis p_hsTyVarBndr con_ex_tvs

137 p_forallBndrs ForallInvis p_hsTyVarBndr con_ex_tvs

148 breakpoint

138 breakpoint

149 forM_ con_mb_cxt p_lhsContext

139 forM_ con_mb_cxt p_lhsContext

150− switchLayout conDeclSpn $ case con_args of

140+ case con_args of

151 PrefixCon xs -> do

141 PrefixCon xs -> do

152 p_rdrName con_name

142 p_rdrName con_name

153 unless (null xs) breakpoint

143 unless (null xs) breakpoint

154− inci . sitcc $ sep breakpoint (sitcc . located' p_hsTypePostDoc) xs

144+ inci . sitcc $ sep breakpoint (sitcc . located' p_hsType) xs

155 RecCon l -> do

145 RecCon l -> do

156 p_rdrName con_name

146 p_rdrName con_name

157 breakpoint

147 breakpoint

158− inciIf (not singleConstRec) (located l p_conDeclFields)

148+ inci $ located l p_conDeclFields

159 InfixCon x y -> do

149 InfixCon x y -> do

160 located x p_hsType

150 located x p_hsType

161 breakpoint

151 breakpoint

@@ -204,9 +194,9 @@ p_hsDerivingClause HsDerivingClause {..} = do

204 let derivingWhat = located deriv_clause_tys $ \case

194 let derivingWhat = located deriv_clause_tys $ \case

205 [] -> txt "()"

195 [] -> txt "()"

206 xs ->

196 xs ->

207− parens N $

197+ parens N . sitcc $

208 sep

198 sep

209− commaDel

199+ (comma >> breakpoint)

210 (sitcc . located' p_hsType . hsib_body)

200 (sitcc . located' p_hsType . hsib_body)

211 xs

201 xs

212 space

202 space

@@ -246,16 +236,3 @@ isInfix :: LexicalFixity -> Bool

246isInfix = \case

236isInfix = \case

247 Infix -> True

237 Infix -> True

248 Prefix -> False

238 Prefix -> False

249−

250−isSingleConstRec :: [LConDecl GhcPs] -> Bool

251−isSingleConstRec [(L _ ConDeclH98 {..})] =

252− case con_args of

253− RecCon _ -> True

254− _ -> False

255−isSingleConstRec _ = False

256−

257−hasHaddocks :: [LConDecl GhcPs] -> Bool

258−hasHaddocks = any (f . unLoc)

259− where

260− f ConDeclH98 {..} = isJust con_doc

261− f _ = False

src/Ormolu/Printer/Meat/Declaration/Default.hs+2−2

@@ -15,6 +15,6 @@ p_defaultDecl = \case

15 DefaultDecl NoExtField ts -> do

16 txt "default"

17 breakpoint

18− inci . parens N $

18+ inci . parens N . sitcc $

19− sep commaDel (sitcc . located' p_hsType) ts

19+ sep (comma >> breakpoint) (sitcc . located' p_hsType) ts

20 XDefaultDecl x -> noExtCon x

src/Ormolu/Printer/Meat/Declaration/Instance.hs+9−5

@@ -32,7 +32,9 @@ p_standaloneDerivDecl DerivDecl {..} = do

32 txt "instance"

33 breakpoint

34 match_overlap_mode deriv_overlap_mode breakpoint

35+ if toIndent

36+ then inci typesAfterInstance

35− inciIf toIndent typesAfterInstance

37+ else typesAfterInstance

36 txt "deriving"

38 txt "deriving"

37 space

39 space

38 case deriv_strategy of

40 case deriv_strategy of

@@ -88,10 +90,12 @@ p_clsInstDecl = \case

88 unless (null allDecls) $ do

90 unless (null allDecls) $ do

89 breakpoint

91 breakpoint

90 txt "where"

92 txt "where"

91− unless (null allDecls) . inci $ do

93+ unless (null allDecls)

94+ $ inci

95+ $ do

92− -- Ensure whitespace is added after where clause.

96+ -- Ensure whitespace is added after where clause.

93− breakpoint

97+ breakpoint

94− dontUseBraces $ p_hsDeclsRespectGrouping Associated allDecls

98+ dontUseBraces $ p_hsDeclsRespectGrouping Associated allDecls

95 XHsImplicitBndrs x -> noExtCon x

99 XHsImplicitBndrs x -> noExtCon x

96 XClsInstDecl x -> noExtCon x

100 XClsInstDecl x -> noExtCon x

97

101

src/Ormolu/Printer/Meat/Declaration/RoleAnnotation.hs+2

@@ -12,6 +12,8 @@ import CoAxiom

12import GHC

13import Ormolu.Printer.Combinators

14import Ormolu.Printer.Meat.Common

15+import RdrName (RdrName (..))

16+import SrcLoc (Located)

15

17

16p_roleAnnot :: RoleAnnotDecl GhcPs -> R ()

18p_roleAnnot :: RoleAnnotDecl GhcPs -> R ()

17p_roleAnnot = \case

19p_roleAnnot = \case

src/Ormolu/Printer/Meat/Declaration/Rule.hs+2−2

@@ -18,7 +18,7 @@ import Ormolu.Printer.Meat.Type

18p_ruleDecls :: RuleDecls GhcPs -> R ()

19p_ruleDecls = \case

20 HsRules NoExtField _ xs ->

21− pragma "RULES" $

21+ pragma "RULES" . sitcc $

22 sep breakpoint (sitcc . located' p_ruleDecl) xs

23 XRuleDecls x -> noExtCon x

24

@@ -43,7 +43,7 @@ p_ruleDecl = \case

43 inci $ do

44 located lhs p_hsExpr

45 space

46− equals

46+ txt "="

47 inci $ do

48 breakpoint

49 located rhs p_hsExpr

src/Ormolu/Printer/Meat/Declaration/Signature.hs+27−29

@@ -14,6 +14,7 @@ where

14import BasicTypes

15import BooleanFormula

16import Control.Monad

17+import Data.Bool (bool)

17import GHC

18import GHC

18import Ormolu.Printer.Combinators

19import Ormolu.Printer.Combinators

19import Ormolu.Printer.Meat.Common

20import Ormolu.Printer.Meat.Common

@@ -47,22 +48,24 @@ p_typeSig indentTail (n : ns) hswc = do

47 p_rdrName n

48 p_rdrName n

48 if null ns

49 if null ns

49 then p_typeAscription hswc

50 then p_typeAscription hswc

50− else inciIf indentTail $ do

51+ else bool id inci indentTail $ do

51− commaDel

52+ comma

53+ breakpoint

52− sep commaDel p_rdrName ns

54+ sep (comma >> breakpoint) p_rdrName ns

53 p_typeAscription hswc

55 p_typeAscription hswc

54

56

55p_typeAscription ::

57p_typeAscription ::

56 LHsSigWcType GhcPs ->

58 LHsSigWcType GhcPs ->

57 R ()

59 R ()

58−p_typeAscription HsWC {..} = inci $ do

60+p_typeAscription HsWC {..} = do

59 space

61 space

62+ inci $ do

60− txt "::"

63+ txt "::"

61− let t = hsib_body hswc_body

64+ let t = hsib_body hswc_body

62− if hasDocStrings (unLoc t)

65+ if hasDocStrings (unLoc t)

63− then newline

66+ then newline

64− else breakpoint

67+ else breakpoint

65− located t p_hsType

68+ located t p_hsType

66p_typeAscription (XHsWildCardBndrs x) = noExtCon x

69p_typeAscription (XHsWildCardBndrs x) = noExtCon x

67

70

68p_patSynSig ::

71p_patSynSig ::

@@ -71,11 +74,7 @@ p_patSynSig ::

71 R ()

74 R ()

72p_patSynSig names hsib = do

75p_patSynSig names hsib = do

73 txt "pattern"

76 txt "pattern"

74− let body =

75− p_typeSig

76− False

77− names

78− HsWC {hswc_ext = NoExtField, hswc_body = hsib}

77+ let body = p_typeSig False names HsWC {hswc_ext = NoExtField, hswc_body = hsib}

79 if length names > 1

78 if length names > 1

80 then breakpoint >> inci body

79 then breakpoint >> inci body

81 else space >> body

80 else space >> body

@@ -104,7 +103,7 @@ p_fixSig = \case

104 space

103 space

105 atom n

104 atom n

106 space

105 space

107− sitcc $ sep commaDel p_rdrName names

106+ sitcc $ sep (comma >> breakpoint) p_rdrName names

108 XFixitySig x -> noExtCon x

107 XFixitySig x -> noExtCon x

109

108

110p_inlineSig ::

109p_inlineSig ::

@@ -143,7 +142,7 @@ p_specSig name ts InlinePragma {..} = pragmaBraces $ do

143 space

142 space

144 txt "::"

143 txt "::"

145 breakpoint

144 breakpoint

146− inci $ sep commaDel (located' p_hsType . hsib_body) ts

145+ inci $ sep (comma >> breakpoint) (located' p_hsType . hsib_body) ts

147

146

148p_inlineSpec :: InlineSpec -> R ()

147p_inlineSpec :: InlineSpec -> R ()

149p_inlineSpec = \case

148p_inlineSpec = \case

@@ -187,13 +186,13 @@ p_booleanFormula = \case

187 And xs ->

186 And xs ->

188 sitcc $

187 sitcc $

189 sep

188 sep

190− commaDel

189+ (comma >> breakpoint)

191 (located' p_booleanFormula)

190 (located' p_booleanFormula)

192 xs

191 xs

193 Or xs ->

192 Or xs ->

194 sitcc $

193 sitcc $

195 sep

194 sep

196− (breakpoint >> txt "|" >> space)

195+ (breakpoint >> txt "| ")

197 (located' p_booleanFormula)

196 (located' p_booleanFormula)

198 xs

197 xs

199 Parens l -> located l (parens N . p_booleanFormula)

198 Parens l -> located l (parens N . p_booleanFormula)

@@ -207,7 +206,7 @@ p_completeSig ::

207p_completeSig cs' mty =

206p_completeSig cs' mty =

208 located cs' $ \cs ->

207 located cs' $ \cs ->

209 pragma "COMPLETE" . inci $ do

208 pragma "COMPLETE" . inci $ do

210− sep commaDel p_rdrName cs

209+ sitcc $ sep (comma >> breakpoint) p_rdrName cs

211 forM_ mty $ \ty -> do

210 forM_ mty $ \ty -> do

212 space

211 space

213 txt "::"

212 txt "::"

@@ -224,13 +223,12 @@ p_sccSig loc literal = pragma "SCC" . inci $ do

224p_standaloneKindSig :: StandaloneKindSig GhcPs -> R ()

223p_standaloneKindSig :: StandaloneKindSig GhcPs -> R ()

225p_standaloneKindSig (StandaloneKindSig NoExtField name bndrs) = do

224p_standaloneKindSig (StandaloneKindSig NoExtField name bndrs) = do

226 txt "type"

225 txt "type"

227− inci $ do

228− space

226+ space

229− p_rdrName name

227+ p_rdrName name

230− space

228+ space

231− txt "::"

229+ txt "::"

232− breakpoint

230+ breakpoint

233− case bndrs of

231+ inci $ case bndrs of

234− HsIB NoExtField sig -> located sig p_hsType

232+ HsIB NoExtField sig -> located sig p_hsType

235− XHsImplicitBndrs x -> noExtCon x

233+ XHsImplicitBndrs x -> noExtCon x

236p_standaloneKindSig (XStandaloneKindSig c) = noExtCon c

234p_standaloneKindSig (XStandaloneKindSig c) = noExtCon c

src/Ormolu/Printer/Meat/Declaration/Type.hs+5−5

@@ -11,6 +11,8 @@ import GHC

11import Ormolu.Printer.Combinators

12import Ormolu.Printer.Meat.Common

13import Ormolu.Printer.Meat.Type

14+import RdrName (RdrName (..))

15+import SrcLoc (Located)

14

16

15p_synDecl ::

17p_synDecl ::

16 -- | Type constructor

18 -- | Type constructor

@@ -28,13 +30,11 @@ p_synDecl name fixity HsQTvs {..} t = do

28 switchLayout (getLoc name : map getLoc hsq_explicit) $

30 switchLayout (getLoc name : map getLoc hsq_explicit) $

29 p_infixDefHelper

31 p_infixDefHelper

30 (case fixity of Infix -> True; _ -> False)

32 (case fixity of Infix -> True; _ -> False)

31− True

33+ inci

32 (p_rdrName name)

34 (p_rdrName name)

33 (map (located' p_hsTyVarBndr) hsq_explicit)

35 (map (located' p_hsTyVarBndr) hsq_explicit)

34 space

36 space

35− equals

37+ txt "="

36− if hasDocStrings (unLoc t)

37− then newline

38− else breakpoint

38+ breakpoint

39 inci (located t p_hsType)

40p_synDecl _ _ (XLHsQTyVars x) _ = noExtCon x

src/Ormolu/Printer/Meat/Declaration/TypeFamily.hs+7−5

@@ -9,6 +9,7 @@ module Ormolu.Printer.Meat.Declaration.TypeFamily

9 )

10where

11

12+import BasicTypes (LexicalFixity (..))

12import Control.Monad

13import Control.Monad

13import Data.Maybe (isNothing)

14import Data.Maybe (isNothing)

14import GHC

15import GHC

@@ -16,6 +17,7 @@ import Ormolu.Printer.Combinators

16import Ormolu.Printer.Meat.Common

17import Ormolu.Printer.Meat.Common

17import Ormolu.Printer.Meat.Type

18import Ormolu.Printer.Meat.Type

18import Ormolu.Utils

19import Ormolu.Utils

20+import SrcLoc (GenLocated (..), Located)

19

21

20p_famDecl :: FamilyStyle -> FamilyDecl GhcPs -> R ()

22p_famDecl :: FamilyStyle -> FamilyDecl GhcPs -> R ()

21p_famDecl style FamilyDecl {fdTyVars = HsQTvs {..}, ..} = do

23p_famDecl style FamilyDecl {fdTyVars = HsQTvs {..}, ..} = do

@@ -31,7 +33,7 @@ p_famDecl style FamilyDecl {fdTyVars = HsQTvs {..}, ..} = do

31 switchLayout (getLoc fdLName : (getLoc <$> hsq_explicit)) $

33 switchLayout (getLoc fdLName : (getLoc <$> hsq_explicit)) $

32 p_infixDefHelper

34 p_infixDefHelper

33 (isInfix fdFixity)

35 (isInfix fdFixity)

34− True

36+ inci

35 (p_rdrName fdLName)

37 (p_rdrName fdLName)

36 (located' p_hsTyVarBndr <$> hsq_explicit)

38 (located' p_hsTyVarBndr <$> hsq_explicit)

37 let resultSig = p_familyResultSigL fdResultSig

39 let resultSig = p_familyResultSigL fdResultSig

@@ -69,7 +71,7 @@ p_familyResultSigL l =

69 breakpoint

71 breakpoint

70 located k p_hsType

72 located k p_hsType

71 TyVarSig NoExtField bndr -> Just $ do

73 TyVarSig NoExtField bndr -> Just $ do

72− equals

74+ txt "="

73 breakpoint

75 breakpoint

74 located bndr p_hsTyVarBndr

76 located bndr p_hsTyVarBndr

75 XFamilyResultSig x ->

77 XFamilyResultSig x ->

@@ -92,16 +94,16 @@ p_tyFamInstEqn HsIB {hsib_body = FamEqn {..}} = do

92 Just bndrs -> do

94 Just bndrs -> do

93 p_forallBndrs ForallInvis p_hsTyVarBndr bndrs

95 p_forallBndrs ForallInvis p_hsTyVarBndr bndrs

94 breakpoint

96 breakpoint

95− inciIf (not $ null feqn_bndrs) $ do

97+ (if null feqn_bndrs then id else inci) $ do

96 let famLhsSpn = getLoc feqn_tycon : fmap (getLoc . typeArgToType) feqn_pats

98 let famLhsSpn = getLoc feqn_tycon : fmap (getLoc . typeArgToType) feqn_pats

97 switchLayout famLhsSpn $

99 switchLayout famLhsSpn $

98 p_infixDefHelper

100 p_infixDefHelper

99 (isInfix feqn_fixity)

101 (isInfix feqn_fixity)

100− True

102+ inci

101 (p_rdrName feqn_tycon)

103 (p_rdrName feqn_tycon)

102 (located' p_hsType . typeArgToType <$> feqn_pats)

104 (located' p_hsType . typeArgToType <$> feqn_pats)

103 space

105 space

104− equals

106+ txt "="

105 breakpoint

107 breakpoint

106 inci (located feqn_rhs p_hsType)

108 inci (located feqn_rhs p_hsType)

107p_tyFamInstEqn HsIB {hsib_body = XFamEqn x} = noExtCon x

109p_tyFamInstEqn HsIB {hsib_body = XFamEqn x} = noExtCon x

src/Ormolu/Printer/Meat/Declaration/Value.hs+174−210

@@ -1,5 +1,4 @@

1{-# LANGUAGE LambdaCase #-}

2−{-# LANGUAGE MultiWayIf #-}

3{-# LANGUAGE OverloadedStrings #-}

2{-# LANGUAGE OverloadedStrings #-}

4{-# LANGUAGE RecordWildCards #-}

3{-# LANGUAGE RecordWildCards #-}

5{-# LANGUAGE TypeApplications #-}

4{-# LANGUAGE TypeApplications #-}

@@ -22,12 +21,12 @@ import Data.Char (isPunctuation, isSymbol)

22import Data.Data hiding (Infix, Prefix)

21import Data.Data hiding (Infix, Prefix)

23import Data.Functor ((<&>))

22import Data.Functor ((<&>))

24import Data.List (intersperse, sortOn)

23import Data.List (intersperse, sortOn)

25−import Data.List.NonEmpty (NonEmpty (..), (<|))

24+import Data.List.NonEmpty ((<|), NonEmpty ((:|)))

26import qualified Data.List.NonEmpty as NE

25import qualified Data.List.NonEmpty as NE

27import Data.Text (Text)

26import Data.Text (Text)

28import qualified Data.Text as Text

27import qualified Data.Text as Text

29import GHC

28import GHC

30−import OccName (occNameString)

29+import OccName (mkVarOcc)

31import Ormolu.Printer.Combinators

30import Ormolu.Printer.Combinators

32import Ormolu.Printer.Internal

31import Ormolu.Printer.Internal

33import Ormolu.Printer.Meat.Common

32import Ormolu.Printer.Meat.Common

@@ -36,6 +35,8 @@ import Ormolu.Printer.Meat.Declaration.Signature

36import Ormolu.Printer.Meat.Type

35import Ormolu.Printer.Meat.Type

37import Ormolu.Printer.Operators

36import Ormolu.Printer.Operators

38import Ormolu.Utils

37import Ormolu.Utils

38+import RdrName (RdrName (..), rdrNameOcc)

39+import SrcLoc (combineSrcSpans, isOneLineSpan)

39

40

40-- | Style of a group of equations.

41-- | Style of a group of equations.

41data MatchGroupStyle

42data MatchGroupStyle

@@ -61,7 +62,7 @@ data Placement

61 -- should use it and avoid bumping one level

62 -- should use it and avoid bumping one level

62 -- of indentation

63 -- of indentation

63 Hanging

64 Hanging

64− deriving (Eq, Show)

65+ deriving (Eq)

65

66

66p_valDecl :: HsBindLR GhcPs GhcPs -> R ()

67p_valDecl :: HsBindLR GhcPs GhcPs -> R ()

67p_valDecl = \case

68p_valDecl = \case

@@ -180,24 +181,25 @@ p_match' placer render style isInfix strictness m_pats GRHSs {..} = do

180 NoSrcStrict -> return ()

181 NoSrcStrict -> return ()

181 SrcStrict -> txt "!"

182 SrcStrict -> txt "!"

182 SrcLazy -> txt "~"

183 SrcLazy -> txt "~"

183− indentBody <- case NE.nonEmpty m_pats of

184+ inci' <- case NE.nonEmpty m_pats of

184− Nothing ->

185− False <$ case style of

185+ Nothing -> id <$ case style of

186− Function name -> p_rdrName name

186+ Function name -> p_rdrName name

187− _ -> return ()

187+ _ -> return ()

188 Just ne_pats -> do

189− let combinedSpans = case style of

189+ let combinedSpans =

190− Function name -> combineSrcSpans (getLoc name) patSpans

190+ combineSrcSpans' $

191− _ -> patSpans

192− patSpans = combineSrcSpans' (getLoc <$> ne_pats)

191+ getLoc <$> ne_pats

192+ inci' =

193− indentBody = not (isOneLineSpan combinedSpans)

193+ if isOneLineSpan combinedSpans

194+ then id

195+ else inci

194 switchLayout [combinedSpans] $ do

196 switchLayout [combinedSpans] $ do

195 let stdCase = sep breakpoint (located' p_pat) m_pats

197 let stdCase = sep breakpoint (located' p_pat) m_pats

196 case style of

198 case style of

197 Function name ->

199 Function name ->

198 p_infixDefHelper

200 p_infixDefHelper

199 isInfix

201 isInfix

200− indentBody

202+ inci'

201 (p_rdrName name)

203 (p_rdrName name)

202 (located' p_pat <$> m_pats)

204 (located' p_pat <$> m_pats)

203 PatternBind -> stdCase

205 PatternBind -> stdCase

@@ -212,32 +214,40 @@ p_match' placer render style isInfix strictness m_pats GRHSs {..} = do

212 when needsSpace space

214 when needsSpace space

213 sitcc stdCase

215 sitcc stdCase

214 LambdaCase -> stdCase

216 LambdaCase -> stdCase

215− return indentBody

217+ return inci'

216 let -- Calculate position of end of patterns. This is useful when we decide

218 let -- Calculate position of end of patterns. This is useful when we decide

217 -- about putting certain constructions in hanging positions.

219 -- about putting certain constructions in hanging positions.

218 endOfPats = case NE.nonEmpty m_pats of

220 endOfPats = case NE.nonEmpty m_pats of

219 Nothing -> case style of

221 Nothing -> case style of

220− Function name -> Just (getLoc name)

222+ Function name -> (Just . srcSpanEnd . getLoc) name

221 _ -> Nothing

223 _ -> Nothing

222− Just pats -> (Just . getLoc . NE.last) pats

224+ Just pats -> (Just . srcSpanEnd . getLoc . NE.last) pats

223 isCase = \case

225 isCase = \case

224 Case -> True

226 Case -> True

225 LambdaCase -> True

227 LambdaCase -> True

226 _ -> False

228 _ -> False

227− hasGuards = withGuards grhssGRHSs

229+ let hasGuards = withGuards grhssGRHSs

230+ unless (length grhssGRHSs > 1) $

231+ case style of

232+ Function _ | hasGuards -> return ()

233+ Function _ -> space >> txt "="

234+ PatternBind -> space >> txt "="

235+ s | isCase s && hasGuards -> return ()

236+ _ -> space >> txt "->"

228− grhssSpan =

237+ let grhssSpan =

229 combineSrcSpans' $

238 combineSrcSpans' $

230 getGRHSSpan . unLoc <$> NE.fromList grhssGRHSs

239 getGRHSSpan . unLoc <$> NE.fromList grhssGRHSs

231 patGrhssSpan =

240 patGrhssSpan =

232 maybe

241 maybe

233 grhssSpan

242 grhssSpan

234− (combineSrcSpans grhssSpan . srcLocSpan . srcSpanEnd)

243+ (combineSrcSpans grhssSpan . srcLocSpan)

235 endOfPats

244 endOfPats

236 placement =

245 placement =

237 case endOfPats of

246 case endOfPats of

238 Nothing -> blockPlacement placer grhssGRHSs

247 Nothing -> blockPlacement placer grhssGRHSs

239 Just spn ->

248 Just spn ->

249+ if isOneLineSpan

240− if onTheSameLine spn grhssSpan

250+ (mkSrcSpan spn (srcSpanStart grhssSpan))

241 then blockPlacement placer grhssGRHSs

251 then blockPlacement placer grhssGRHSs

242 else Normal

252 else Normal

243 p_body = do

253 p_body = do

@@ -253,14 +263,7 @@ p_match' placer render style isInfix strictness m_pats GRHSs {..} = do

253 txt "where"

263 txt "where"

254 unless whereIsEmpty breakpoint

264 unless whereIsEmpty breakpoint

255 inci $ located grhssLocalBinds p_hsLocalBinds

265 inci $ located grhssLocalBinds p_hsLocalBinds

256− inciIf indentBody $ do

266+ inci' $ do

257− unless (length grhssGRHSs > 1) $

258− case style of

259− Function _ | hasGuards -> return ()

260− Function _ -> space >> inci equals

261− PatternBind -> space >> inci equals

262− s | isCase s && hasGuards -> return ()

263− _ -> space >> txt "->"

264 switchLayout [patGrhssSpan] $

267 switchLayout [patGrhssSpan] $

265 placeHanging placement p_body

268 placeHanging placement p_body

266 inci p_where

269 inci p_where

@@ -284,24 +287,24 @@ p_grhs' placer render style (GRHS NoExtField guards body) =

284 xs -> do

287 xs -> do

285 txt "|"

288 txt "|"

286 space

289 space

287− sitcc (sep commaDel (sitcc . located' p_stmt) xs)

290+ sitcc (sep (comma >> breakpoint) (sitcc . located' p_stmt) xs)

288 space

291 space

289− inci $ case style of

292+ txt $ case style of

290− EqualSign -> equals

293+ EqualSign -> "="

291− RightArrow -> txt "->"

294+ RightArrow -> "->"

292 placeHanging placement p_body

295 placeHanging placement p_body

293 where

296 where

294 placement =

297 placement =

295 case endOfGuards of

298 case endOfGuards of

296 Nothing -> placer (unLoc body)

299 Nothing -> placer (unLoc body)

297 Just spn ->

300 Just spn ->

298− if onTheSameLine spn (getLoc body)

301+ if isOneLineSpan (mkSrcSpan spn (srcSpanStart (getLoc body)))

299 then placer (unLoc body)

302 then placer (unLoc body)

300 else Normal

303 else Normal

301 endOfGuards =

304 endOfGuards =

302 case NE.nonEmpty guards of

305 case NE.nonEmpty guards of

303 Nothing -> Nothing

306 Nothing -> Nothing

304− Just gs -> (Just . getLoc . NE.last) gs

307+ Just gs -> (Just . srcSpanEnd . getLoc . NE.last) gs

305 p_body = located body render

308 p_body = located body render

306p_grhs' _ _ _ (XGRHS x) = noExtCon x

309p_grhs' _ _ _ (XGRHS x) = noExtCon x

307

310

@@ -309,14 +312,13 @@ p_hsCmd :: HsCmd GhcPs -> R ()

309p_hsCmd = \case

312p_hsCmd = \case

310 HsCmdArrApp NoExtField body input arrType _ -> do

313 HsCmdArrApp NoExtField body input arrType _ -> do

311 located body p_hsExpr

314 located body p_hsExpr

312− breakpoint

315+ space

313− inci $ do

314− case arrType of

316+ case arrType of

315− HsFirstOrderApp -> txt "-<"

317+ HsFirstOrderApp -> txt "-<"

316− HsHigherOrderApp -> txt "-<<"

318+ HsHigherOrderApp -> txt "-<<"

317− placeHanging (exprPlacement (unLoc input)) $

319+ placeHanging (exprPlacement (unLoc input)) $

318− located input p_hsExpr

320+ located input p_hsExpr

319− HsCmdArrForm NoExtField form Prefix _ cmds -> banana $ do

321+ HsCmdArrForm NoExtField form Prefix _ cmds -> banana $ sitcc $ do

320 located form p_hsExpr

322 located form p_hsExpr

321 unless (null cmds) $ do

323 unless (null cmds) $ do

322 breakpoint

324 breakpoint

@@ -344,7 +346,7 @@ p_hsCmd = \case

344 txt "do"

346 txt "do"

345 newline

347 newline

346 inci . located es $

348 inci . located es $

347− sitcc . sep newline (sitcc . withSpacing (p_stmt' cmdPlacement p_hsCmd))

349+ sitcc . sep newline (located' (sitcc . p_stmt' cmdPlacement p_hsCmd))

348 HsCmdWrap {} -> notImplemented "HsCmdWrap"

350 HsCmdWrap {} -> notImplemented "HsCmdWrap"

349 XCmd x -> noExtCon x

351 XCmd x -> noExtCon x

350

352

@@ -353,35 +355,6 @@ p_hsCmdTop = \case

353 HsCmdTop NoExtField cmd -> located cmd p_hsCmd

355 HsCmdTop NoExtField cmd -> located cmd p_hsCmd

354 XCmdTop x -> noExtCon x

356 XCmdTop x -> noExtCon x

355

357

356−-- | Render an expression preserving blank lines between such consecutive

357−-- expressions found in the original source code.

358−withSpacing ::

359− -- | Rendering function

360− (a -> R ()) ->

361− -- | Entity to render

362− Located a ->

363− R ()

364−withSpacing f l = located l $ \x -> do

365− case getLoc l of

366− UnhelpfulSpan _ -> f x

367− RealSrcSpan currentSpn -> do

368− getSpanMark >>= \case

369− -- Spacing before comments will be handled by the code

370− -- that prints comments, so we just have to deal with

371− -- blank lines between statements here.

372− Just (StatementSpan lastSpn) ->

373− if srcSpanStartLine currentSpn > srcSpanEndLine lastSpn + 1

374− then newline

375− else return ()

376− _ -> return ()

377− f x

378− -- In some cases the (f x) expression may insert a new mark. We want

379− -- to be careful not to override comment marks.

380− getSpanMark >>= \case

381− Just (HaddockSpan _ _) -> return ()

382− Just (CommentSpan _) -> return ()

383− _ -> setSpanMark (StatementSpan currentSpn)

384−

385p_stmt :: Stmt GhcPs (LHsExpr GhcPs) -> R ()

358p_stmt :: Stmt GhcPs (LHsExpr GhcPs) -> R ()

386p_stmt = p_stmt' exprPlacement p_hsExpr

359p_stmt = p_stmt' exprPlacement p_hsExpr

387

360

@@ -453,7 +426,7 @@ p_stmt' placer render = \case

453 RecStmt {..} -> do

426 RecStmt {..} -> do

454 txt "rec"

427 txt "rec"

455 space

428 space

456− sitcc $ sepSemi (withSpacing (p_stmt' placer render)) recS_stmts

429+ sitcc $ sepSemi (located' (p_stmt' placer render)) recS_stmts

457 XStmtLR c -> noExtCon c

430 XStmtLR c -> noExtCon c

458

431

459gatherStmt :: ExprLStmt GhcPs -> [[ExprLStmt GhcPs]]

432gatherStmt :: ExprLStmt GhcPs -> [[ExprLStmt GhcPs]]

@@ -471,31 +444,34 @@ gatherStmtBlock (XParStmtBlock x) = noExtCon x

471p_hsLocalBinds :: HsLocalBindsLR GhcPs GhcPs -> R ()

444p_hsLocalBinds :: HsLocalBindsLR GhcPs GhcPs -> R ()

472p_hsLocalBinds = \case

445p_hsLocalBinds = \case

473 HsValBinds NoExtField (ValBinds NoExtField bag lsigs) -> do

446 HsValBinds NoExtField (ValBinds NoExtField bag lsigs) -> do

447+ let ssStart =

448+ either

449+ (srcSpanStart . getLoc)

450+ (srcSpanStart . getLoc)

451+ items =

452+ (Left <$> bagToList bag) ++ (Right <$> lsigs)

453+ p_item (Left x) = located x p_valDecl

454+ p_item (Right x) = located x p_sigDecl

455+ -- Assigns 'False' to the last element, 'True' to the rest.

456+ markInit :: [a] -> [(Bool, a)]

457+ markInit [] = []

458+ markInit [x] = [(False, x)]

459+ markInit (x : xs) = (True, x) : markInit xs

474 -- When in a single-line layout, there is a chance that the inner

460 -- When in a single-line layout, there is a chance that the inner

475 -- elements will also contain semicolons and they will confuse the

461 -- elements will also contain semicolons and they will confuse the

476 -- parser. so we request braces around every element except the last.

462 -- parser. so we request braces around every element except the last.

477 br <- layoutToBraces <$> getLayout

463 br <- layoutToBraces <$> getLayout

478− let items =

464+ sitcc $

479− let injectLeft (L l x) = L l (Left x)

465+ sepSemi

480− injectRight (L l x) = L l (Right x)

466+ (\(m, i) -> (if m then br else id) $ p_item i)

481− in (injectLeft <$> bagToList bag) ++ (injectRight <$> lsigs)

467+ (markInit $ sortOn ssStart items)

482− positionToBracing = \case

483− SinglePos -> id

484− FirstPos -> br

485− MiddlePos -> br

486− LastPos -> id

487− p_item' (p, item) =

488− positionToBracing p $

489− withSpacing (either p_valDecl p_sigDecl) item

490− binds = sortOn (srcSpanStart . getLoc) items

491− sitcc $ sepSemi p_item' (attachRelativePos binds)

492 HsValBinds NoExtField _ -> notImplemented "HsValBinds"

468 HsValBinds NoExtField _ -> notImplemented "HsValBinds"

493 HsIPBinds NoExtField (IPBinds NoExtField xs) ->

469 HsIPBinds NoExtField (IPBinds NoExtField xs) ->

494 -- Second argument of IPBind is always Left before type-checking.

470 -- Second argument of IPBind is always Left before type-checking.

495 let p_ipBind (IPBind NoExtField (Left name) expr) = do

471 let p_ipBind (IPBind NoExtField (Left name) expr) = do

496 atom name

472 atom name

497 space

473 space

498− equals

474+ txt "="

499 breakpoint

475 breakpoint

500 useBraces $ inci $ located expr p_hsExpr

476 useBraces $ inci $ located expr p_hsExpr

501 p_ipBind (IPBind NoExtField (Right _) _) =

477 p_ipBind (IPBind NoExtField (Right _) _) =

@@ -514,12 +490,15 @@ p_hsRecField HsRecField {..} = do

514 p_rdrName hsRecFieldLbl

490 p_rdrName hsRecFieldLbl

515 unless hsRecPun $ do

491 unless hsRecPun $ do

516 space

492 space

517− equals

493+ txt "="

518− let placement =

519− if onTheSameLine (getLoc hsRecFieldLbl) (getLoc hsRecFieldArg)

520− then exprPlacement (unLoc hsRecFieldArg)

494+ let placement = exprPlacement (unLoc hsRecFieldArg)

521− else Normal

522− placeHanging placement (located hsRecFieldArg p_hsExpr)

495+ placeHanging placement $ located hsRecFieldArg p_hsExpr

496+

497+p_hsTupArg :: HsTupArg GhcPs -> R ()

498+p_hsTupArg = \case

499+ Present NoExtField x -> located x p_hsExpr

500+ Missing NoExtField -> pure ()

501+ XTupArg x -> noExtCon x

523

502

524p_hsExpr :: HsExpr GhcPs -> R ()

503p_hsExpr :: HsExpr GhcPs -> R ()

525p_hsExpr = p_hsExpr' N

504p_hsExpr = p_hsExpr' N

@@ -584,21 +563,23 @@ p_hsExpr' s = \case

584 -- do-block or case expression it is not a good idea. It seems

563 -- do-block or case expression it is not a good idea. It seems

585 -- to be safe to always bump indentation when the function

564 -- to be safe to always bump indentation when the function

586 -- expression is parenthesised.

565 -- expression is parenthesised.

587− doIndent =

566+ indent =

588 case func of

567 case func of

589− L _ (HsPar NoExtField _) -> True

568+ L _ (HsPar NoExtField _) -> inci

590− L _ (HsAppType NoExtField _ _) -> True

569+ L _ (HsAppType NoExtField _ _) -> inci

591− L _ (HsMultiIf NoExtField _) -> True

570+ L _ (HsMultiIf NoExtField _) -> inci

571+ L spn _ ->

592− L spn _ -> isOneLineSpan spn

572+ if isOneLineSpan spn

593− ub <-

573+ then inci

574+ else id

594− getLayout <&> \case

575+ ub <- getLayout <&> \case

595− SingleLine -> useBraces

576+ SingleLine -> useBraces

596− MultiLine -> id

577+ MultiLine -> id

597 ub $ do

578 ub $ do

598 located func (p_hsExpr' s)

579 located func (p_hsExpr' s)

599 breakpoint

580 breakpoint

600− inciIf doIndent $ sep breakpoint (located' p_hsExpr) initp

581+ indent $ sep breakpoint (located' p_hsExpr) initp

601− inciIf doIndent $ do

582+ indent $ do

602 unless (null initp) breakpoint

583 unless (null initp) breakpoint

603 located lastp p_hsExpr

584 located lastp p_hsExpr

604 Hanging -> do

585 Hanging -> do

@@ -613,15 +594,10 @@ p_hsExpr' s = \case

613 breakpoint

594 breakpoint

614 inci $ do

595 inci $ do

615 txt "@"

596 txt "@"

616− -- Insert a space when the type is represented as a TH splice to avoid

617− -- gluing @ and $ together.

618− case unLoc (hswc_body a) of

619− HsSpliceTy {} -> space

620− _ -> return ()

621 located (hswc_body a) p_hsType

597 located (hswc_body a) p_hsType

622 OpApp NoExtField x op y -> do

598 OpApp NoExtField x op y -> do

623 let opTree = OpBranch (exprOpTree x) op (exprOpTree y)

599 let opTree = OpBranch (exprOpTree x) op (exprOpTree y)

624− p_exprOpTree s (reassociateOpTree getOpName opTree)

600+ p_exprOpTree True s (reassociateOpTree getOpName opTree)

625 NegApp NoExtField e _ -> do

601 NegApp NoExtField e _ -> do

626 txt "-"

602 txt "-"

627 space

603 space

@@ -638,27 +614,22 @@ p_hsExpr' s = \case

638 let isRecordDot' = isRecordDot (unLoc op) (getLoc x)

614 let isRecordDot' = isRecordDot (unLoc op) (getLoc x)

639 unless (useRecordDot' && isRecordDot') breakpoint

615 unless (useRecordDot' && isRecordDot') breakpoint

640 inci (located x p_hsExpr)

616 inci (located x p_hsExpr)

641− ExplicitTuple NoExtField args boxity ->

617+ ExplicitTuple NoExtField args boxity -> do

642 let isSection = any (isMissing . unLoc) args

618 let isSection = any (isMissing . unLoc) args

643 isMissing = \case

619 isMissing = \case

644 Missing NoExtField -> True

620 Missing NoExtField -> True

645 _ -> False

621 _ -> False

646− p_arg = \case

647− Present NoExtField x -> located x p_hsExpr

648− Missing NoExtField -> pure ()

649− XTupArg x -> noExtCon x

650− p_larg = sitcc . located' p_arg

651− parens' =

622+ let parens' =

652 case boxity of

623 case boxity of

653 Boxed -> parens

624 Boxed -> parens

654 Unboxed -> parensHash

625 Unboxed -> parensHash

655− in if isSection

626+ if isSection

656− then

627+ then

657− switchLayout [] . parens' s $

628+ switchLayout [] . parens' s $

658− sep comma p_larg args

629+ sep comma (located' p_hsTupArg) args

659− else

630+ else

660− switchLayout (getLoc <$> args) . parens' s $

631+ switchLayout (getLoc <$> args) . parens' s . sitcc $

661− sep commaDel p_larg args

632+ sep (comma >> breakpoint) (sitcc . located' p_hsTupArg) args

662 ExplicitSum NoExtField tag arity e ->

633 ExplicitSum NoExtField tag arity e ->

663 p_unboxedSum N tag arity (located e p_hsExpr)

634 p_unboxedSum N tag arity (located e p_hsExpr)

664 HsCase NoExtField e mgroup ->

635 HsCase NoExtField e mgroup ->

@@ -666,9 +637,8 @@ p_hsExpr' s = \case

666 HsIf NoExtField _ if' then' else' ->

637 HsIf NoExtField _ if' then' else' ->

667 p_if exprPlacement p_hsExpr if' then' else'

638 p_if exprPlacement p_hsExpr if' then' else'

668 HsMultiIf NoExtField guards -> do

639 HsMultiIf NoExtField guards -> do

669− txt "if"

640+ txt "if "

670− breakpoint

671− inci . inci $ sep newline (located' (p_grhs RightArrow)) guards

641+ inci . inci . sitcc $ sep newline (located' (p_grhs RightArrow)) guards

672 HsLet NoExtField localBinds e ->

642 HsLet NoExtField localBinds e ->

673 p_let p_hsExpr localBinds e

643 p_let p_hsExpr localBinds e

674 HsDo NoExtField ctx es -> do

644 HsDo NoExtField ctx es -> do

@@ -678,17 +648,17 @@ p_hsExpr' s = \case

678 ub <- layoutToBraces <$> getLayout

648 ub <- layoutToBraces <$> getLayout

679 inci $

649 inci $

680 sepSemi

650 sepSemi

681− (ub . withSpacing (p_stmt' exprPlacement (p_hsExpr' S)))

651+ (located' (ub . p_stmt' exprPlacement (p_hsExpr' S)))

682 (unLoc es)

652 (unLoc es)

683− compBody = brackets N . located es $ \xs -> do

653+ compBody = brackets N $ located es $ \xs -> do

684 let p_parBody =

654 let p_parBody =

685 sep

655 sep

686− (breakpoint >> txt "|" >> space)

656+ (breakpoint >> txt "| ")

687 p_seqBody

657 p_seqBody

688 p_seqBody =

658 p_seqBody =

689 sitcc

659 sitcc

690 . sep

660 . sep

691− commaDel

661+ (comma >> breakpoint)

692 (located' (sitcc . p_stmt))

662 (located' (sitcc . p_stmt))

693 stmts = init xs

663 stmts = init xs

694 yield = last xs

664 yield = last xs

@@ -709,8 +679,8 @@ p_hsExpr' s = \case

709 ParStmtCtxt _ -> notImplemented "ParStmtCtxt"

679 ParStmtCtxt _ -> notImplemented "ParStmtCtxt"

710 TransStmtCtxt _ -> notImplemented "TransStmtCtxt"

680 TransStmtCtxt _ -> notImplemented "TransStmtCtxt"

711 ExplicitList _ _ xs ->

681 ExplicitList _ _ xs ->

712− brackets s $

682+ brackets s . sitcc $

713− sep commaDel (sitcc . located' p_hsExpr) xs

683+ sep (comma >> breakpoint) (sitcc . located' p_hsExpr) xs

714 RecordCon {..} -> do

684 RecordCon {..} -> do

715 located rcon_con_name atom

685 located rcon_con_name atom

716 breakpoint

686 breakpoint

@@ -726,8 +696,8 @@ p_hsExpr' s = \case

726 case rec_dotdot of

696 case rec_dotdot of

727 Just {} -> [txt ".."]

697 Just {} -> [txt ".."]

728 Nothing -> []

698 Nothing -> []

729− inci . braces N $

699+ inci . braces N . sitcc $

730− sep commaDel sitcc (fields <> dotdot)

700+ sep (comma >> breakpoint) sitcc (fields <> dotdot)

731 RecordUpd {..} -> do

701 RecordUpd {..} -> do

732 located rupd_expr p_hsExpr

702 located rupd_expr p_hsExpr

733 useRecordDot' <- useRecordDot

703 useRecordDot' <- useRecordDot

@@ -745,9 +715,9 @@ p_hsExpr' s = \case

745 Unambiguous _ n -> n

715 Unambiguous _ n -> n

746 XAmbiguousFieldOcc x -> noExtCon x

716 XAmbiguousFieldOcc x -> noExtCon x

747 }

717 }

748− inci . braces N $

718+ inci . braces N . sitcc $

749 sep

719 sep

750− commaDel

720+ (comma >> breakpoint)

751 (sitcc . located' (p_hsRecField . updName))

721 (sitcc . located' (p_hsRecField . updName))

752 rupd_flds

722 rupd_flds

753 ExprWithTySig NoExtField x HsWC {hswc_body = HsIB {..}} -> sitcc $ do

723 ExprWithTySig NoExtField x HsWC {hswc_body = HsIB {..}} -> sitcc $ do

@@ -760,22 +730,22 @@ p_hsExpr' s = \case

760 ExprWithTySig NoExtField _ (XHsWildCardBndrs x) -> noExtCon x

730 ExprWithTySig NoExtField _ (XHsWildCardBndrs x) -> noExtCon x

761 ArithSeq NoExtField _ x ->

731 ArithSeq NoExtField _ x ->

762 case x of

732 case x of

763− From from -> brackets s $ do

733+ From from -> brackets s . sitcc $ do

764 located from p_hsExpr

734 located from p_hsExpr

765 breakpoint

735 breakpoint

766 txt ".."

736 txt ".."

767− FromThen from next -> brackets s $ do

737+ FromThen from next -> brackets s . sitcc $ do

768− sep commaDel (located' p_hsExpr) [from, next]

738+ sitcc $ sep (comma >> breakpoint) (located' p_hsExpr) [from, next]

769 breakpoint

739 breakpoint

770 txt ".."

740 txt ".."

771− FromTo from to -> brackets s $ do

741+ FromTo from to -> brackets s . sitcc $ do

772 located from p_hsExpr

742 located from p_hsExpr

773 breakpoint

743 breakpoint

774 txt ".."

744 txt ".."

775 space

745 space

776 located to p_hsExpr

746 located to p_hsExpr

777− FromThenTo from next to -> brackets s $ do

747+ FromThenTo from next to -> brackets s . sitcc $ do

778− sep commaDel (located' p_hsExpr) [from, next]

748+ sitcc $ sep (comma >> breakpoint) (located' p_hsExpr) [from, next]

779 breakpoint

749 breakpoint

780 txt ".."

750 txt ".."

781 space

751 space

@@ -825,7 +795,7 @@ p_patSynBind PSB {..} = do

825 breakpoint

795 breakpoint

826 located psb_def p_pat

796 located psb_def p_pat

827 ImplicitBidirectional -> do

797 ImplicitBidirectional -> do

828− equals

798+ txt "="

829 breakpoint

799 breakpoint

830 located psb_def p_pat

800 located psb_def p_pat

831 ExplicitBidirectional mgroup -> do

801 ExplicitBidirectional mgroup -> do

@@ -852,8 +822,8 @@ p_patSynBind PSB {..} = do

852 inci $ do

822 inci $ do

853 switchLayout (getLoc . recordPatSynPatVar <$> xs) $ do

823 switchLayout (getLoc . recordPatSynPatVar <$> xs) $ do

854 unless (null xs) breakpoint

824 unless (null xs) breakpoint

855− braces N $

825+ braces N . sitcc $

856− sep commaDel (p_rdrName . recordPatSynPatVar) xs

826+ sep (comma >> breakpoint) (p_rdrName . recordPatSynPatVar) xs

857 rhs

827 rhs

858 InfixCon l r -> do

828 InfixCon l r -> do

859 switchLayout [getLoc l, getLoc r] $ do

829 switchLayout [getLoc l, getLoc r] $ do

@@ -907,12 +877,10 @@ p_if placer render if' then' else' = do

907 breakpoint

877 breakpoint

908 inci $ do

878 inci $ do

909 txt "then"

879 txt "then"

910− space

911 located then' $ \x ->

880 located then' $ \x ->

912 placeHanging (placer x) (render x)

881 placeHanging (placer x) (render x)

913 breakpoint

882 breakpoint

914 txt "else"

883 txt "else"

915− space

916 located else' $ \x ->

884 located else' $ \x ->

917 placeHanging (placer x) (render x)

885 placeHanging (placer x) (render x)

918

886

@@ -949,13 +917,13 @@ p_pat = \case

949 txt "!"

917 txt "!"

950 located pat p_pat

918 located pat p_pat

951 ListPat NoExtField pats ->

919 ListPat NoExtField pats ->

952− brackets S $ sep commaDel (located' p_pat) pats

920+ brackets S . sitcc $ sep (comma >> breakpoint) (located' p_pat) pats

953 TuplePat NoExtField pats boxing -> do

921 TuplePat NoExtField pats boxing -> do

954− let parens' =

922+ let f =

955 case boxing of

923 case boxing of

956 Boxed -> parens S

924 Boxed -> parens S

957 Unboxed -> parensHash S

925 Unboxed -> parensHash S

958− parens' $ sep commaDel (sitcc . located' p_pat) pats

926+ f . sitcc $ sep (comma >> breakpoint) (sitcc . located' p_pat) pats

959 SumPat NoExtField pat tag arity ->

927 SumPat NoExtField pat tag arity ->

960 p_unboxedSum S tag arity (located pat p_pat)

928 p_unboxedSum S tag arity (located pat p_pat)

961 ConPatIn pat details ->

929 ConPatIn pat details ->

@@ -971,7 +939,7 @@ p_pat = \case

971 let f = \case

939 let f = \case

972 Nothing -> txt ".."

940 Nothing -> txt ".."

973 Just x -> located x p_pat_hsRecField

941 Just x -> located x p_pat_hsRecField

974− inci . braces N . sep commaDel f $

942+ inci . braces N . sitcc . sep (comma >> breakpoint) f $

975 case dotdot of

943 case dotdot of

976 Nothing -> Just <$> fields

944 Nothing -> Just <$> fields

977 Just (L _ n) -> (Just <$> take n fields) ++ [Nothing]

945 Just (L _ n) -> (Just <$> take n fields) ++ [Nothing]

@@ -1012,7 +980,7 @@ p_pat_hsRecField HsRecField {..} = do

1012 p_rdrName (rdrNameFieldOcc x)

980 p_rdrName (rdrNameFieldOcc x)

1013 unless hsRecPun $ do

981 unless hsRecPun $ do

1014 space

982 space

1015− equals

983+ txt "="

1016 breakpoint

984 breakpoint

1017 inci (located hsRecFieldArg p_pat)

985 inci (located hsRecFieldArg p_pat)

1018

986

@@ -1021,15 +989,17 @@ p_unboxedSum s tag arity m = do

1021 let before = tag - 1

989 let before = tag - 1

1022 after = arity - before - 1

990 after = arity - before - 1

1023 args = replicate before Nothing <> [Just m] <> replicate after Nothing

991 args = replicate before Nothing <> [Just m] <> replicate after Nothing

1024− f x =

992+ f (x, i) = do

993+ let isFirst = i == 0

994+ isLast = i == arity - 1

1025 case x :: Maybe (R ()) of

995 case x :: Maybe (R ()) of

1026 Nothing ->

996 Nothing ->

1027− space

997+ unless (isFirst || isLast) space

1028 Just m' -> do

998 Just m' -> do

1029− space

999+ unless isFirst space

1030 m'

1000 m'

1031− space

1001+ unless isLast space

1032− parensHash s $ sep (txt "|") f args

1002+ parensHash s $ sep (txt "|") f (zip args [0 ..])

1033

1003

1034p_hsSplice :: HsSplice GhcPs -> R ()

1004p_hsSplice :: HsSplice GhcPs -> R ()

1035p_hsSplice = \case

1005p_hsSplice = \case

@@ -1231,10 +1201,11 @@ exprPlacement = \case

1231 HsCase NoExtField _ _ -> Hanging

1201 HsCase NoExtField _ _ -> Hanging

1232 HsDo NoExtField DoExpr _ -> Hanging

1202 HsDo NoExtField DoExpr _ -> Hanging

1233 HsDo NoExtField MDoExpr _ -> Hanging

1203 HsDo NoExtField MDoExpr _ -> Hanging

1234− OpApp NoExtField _ op y ->

1204+ -- If the rightmost expression in an operator chain is hanging, make the

1235− case (fmap getOpNameStr . getOpName . unLoc) op of

1205+ -- whole block hanging; so that we can use the common @f = foo $ do@

1206+ -- style.

1236− Just "$" -> exprPlacement (unLoc y)

1207+ OpApp NoExtField _ _ y -> exprPlacement (unLoc y)

1237− _ -> Normal

1208+ -- Same thing for function applications (usually with -XBlockArguments)

1238 HsApp NoExtField _ y -> exprPlacement (unLoc y)

1209 HsApp NoExtField _ y -> exprPlacement (unLoc y)

1239 HsProc NoExtField p _ ->

1210 HsProc NoExtField p _ ->

1240 -- Indentation breaks if pattern is longer than one line and left

1211 -- Indentation breaks if pattern is longer than one line and left

@@ -1260,16 +1231,15 @@ getOpName = \case

1260 HsVar NoExtField (L _ a) -> Just a

1231 HsVar NoExtField (L _ a) -> Just a

1261 _ -> Nothing

1232 _ -> Nothing

1262

1233

1263−getOpNameStr :: RdrName -> String

1264−getOpNameStr = occNameString . rdrNameOcc

1265−

1266p_exprOpTree ::

1234p_exprOpTree ::

1235+ -- | Can use special handling of dollar?

1236+ Bool ->

1267 -- | Bracket style to use

1237 -- | Bracket style to use

1268 BracketStyle ->

1238 BracketStyle ->

1269 OpTree (LHsExpr GhcPs) (LHsExpr GhcPs) ->

1239 OpTree (LHsExpr GhcPs) (LHsExpr GhcPs) ->

1270 R ()

1240 R ()

1271−p_exprOpTree s (OpNode x) = located x (p_hsExpr' s)

1241+p_exprOpTree _ s (OpNode x) = located x (p_hsExpr' s)

1272−p_exprOpTree s (OpBranch x op y) = do

1242+p_exprOpTree isDollarSpecial s (OpBranch x op y) = do

1273 -- If the beginning of the first argument and the second argument are on

1243 -- If the beginning of the first argument and the second argument are on

1274 -- the same line, and the second argument has a hanging form, use hanging

1244 -- the same line, and the second argument has a hanging form, use hanging

1275 -- placement.

1245 -- placement.

@@ -1291,54 +1261,44 @@ p_exprOpTree s (OpBranch x op y) = do

1291 MultiLine -> case placement of

1261 MultiLine -> case placement of

1292 Hanging -> useBraces

1262 Hanging -> useBraces

1293 Normal -> dontUseBraces

1263 Normal -> dontUseBraces

1294− opNameStr = (fmap getOpNameStr . getOpName . unLoc) op

1264+ gotDollar = case getOpName (unLoc op) of

1295− gotDollar = opNameStr == Just "$"

1265+ Just rname -> mkVarOcc "$" == rdrNameOcc rname

1296− gotColon = opNameStr == Just ":"

1266+ _ -> False

1297− gotRecordDot = isRecordDot (unLoc op) (opTreeLoc y)

1298 lhs =

1267 lhs =

1299 switchLayout [opTreeLoc x] $

1268 switchLayout [opTreeLoc x] $

1300− p_exprOpTree s x

1269+ p_exprOpTree (not gotDollar) s x

1301− p_op = located op (opWrapper . p_hsExpr)

1270+ let p_op = located op (opWrapper . p_hsExpr)

1302− p_y = switchLayout [opTreeLoc y] (p_exprOpTree N y)

1271+ p_y = switchLayout [opTreeLoc y] (p_exprOpTree True N y)

1303 isSection = case (opTreeLoc x, getLoc op) of

1272 isSection = case (opTreeLoc x, getLoc op) of

1304 (RealSrcSpan treeSpan, RealSrcSpan opSpan) ->

1273 (RealSrcSpan treeSpan, RealSrcSpan opSpan) ->

1305 srcSpanEndCol treeSpan /= srcSpanStartCol opSpan

1274 srcSpanEndCol treeSpan /= srcSpanStartCol opSpan

1306 _ -> False

1275 _ -> False

1307− isDoBlock = \case

1308− OpNode (L _ HsDo {}) -> True

1309− _ -> False

1310 useRecordDot' <- useRecordDot

1276 useRecordDot' <- useRecordDot

1311− if

1277+ let isRecordDot' = isRecordDot (unLoc op) (opTreeLoc y)

1312− | gotColon -> do

1278+ if useRecordDot' && isRecordDot'

1313− lhs

1279+ then do

1280+ lhs

1281+ when isSection space

1282+ p_op

1283+ p_y

1284+ else

1285+ if isDollarSpecial

1286+ && gotDollar

1287+ && placement

1288+ == Normal

1289+ && isOneLineSpan (opTreeLoc x)

1290+ then do

1291+ useBraces lhs

1314− space

1292+ space

1315− p_op

1293+ p_op

1316− case placement of

1294+ breakpoint

1317− Hanging -> do

1295+ inci p_y

1296+ else do

1297+ ub lhs

1298+ placeHanging placement $ do

1299+ p_op

1318 space

1300 space

1319 p_y

1301 p_y

1320− Normal -> do

1321− breakpoint

1322− inciIf (isDoBlock y) p_y

1323− | gotDollar

1324− && isOneLineSpan (opTreeLoc x)

1325− && placement == Normal -> do

1326− useBraces lhs

1327− space

1328− p_op

1329− breakpoint

1330− inci p_y

1331− | useRecordDot' && gotRecordDot -> do

1332− lhs

1333− when isSection space

1334− p_op

1335− p_y

1336− | otherwise -> do

1337− ub lhs

1338− placeHanging placement $ do

1339− p_op

1340− space

1341− p_y

1342

1302

1343-- | Return 'True' if given expression is a record-dot operator expression.

1303-- | Return 'True' if given expression is a record-dot operator expression.

1344isRecordDot ::

1304isRecordDot ::

@@ -1349,10 +1309,14 @@ isRecordDot ::

1349 Bool

1309 Bool

1350isRecordDot op (RealSrcSpan ySpan) = case op of

1310isRecordDot op (RealSrcSpan ySpan) = case op of

1351 HsVar NoExtField (L (RealSrcSpan opSpan) opName) ->

1311 HsVar NoExtField (L (RealSrcSpan opSpan) opName) ->

1352− (getOpNameStr opName == ".") && (srcSpanEndCol opSpan == srcSpanStartCol ySpan)

1312+ isDot opName && (srcSpanEndCol opSpan == srcSpanStartCol ySpan)

1353 _ -> False

1313 _ -> False

1354isRecordDot _ _ = False

1314isRecordDot _ _ = False

1355

1315

1316+-- | Check whether a given 'RdrName' is the dot operator.

1317+isDot :: RdrName -> Bool

1318+isDot name = rdrNameOcc name == mkVarOcc "."

1319+

1356-- | Get annotations for the enclosing element.

1320-- | Get annotations for the enclosing element.

1357getEnclosingAnns :: R [AnnKeywordId]

1321getEnclosingAnns :: R [AnnKeywordId]

1358getEnclosingAnns = do

1322getEnclosingAnns = do

src/Ormolu/Printer/Meat/Declaration/Value.hs-boot+4

@@ -11,7 +11,11 @@ import GHC

11import Ormolu.Printer.Combinators

12

13p_valDecl :: HsBindLR GhcPs GhcPs -> R ()

14+

14p_pat :: Pat GhcPs -> R ()

15p_pat :: Pat GhcPs -> R ()

16+

15p_hsExpr :: HsExpr GhcPs -> R ()

17p_hsExpr :: HsExpr GhcPs -> R ()

18+

16p_hsSplice :: HsSplice GhcPs -> R ()

19p_hsSplice :: HsSplice GhcPs -> R ()

20+

17p_stringLit :: String -> R ()

21p_stringLit :: String -> R ()

src/Ormolu/Printer/Meat/Declaration/Warning.hs+8−5

@@ -27,14 +27,17 @@ p_warnDecl (XWarnDecl x) = noExtCon x

27p_moduleWarning :: WarningTxt -> R ()

28p_moduleWarning wtxt = do

29 let (pragmaText, lits) = warningText wtxt

30+ switchLayout (getLoc <$> lits)

31+ $ inci

30− inci $ pragma pragmaText $ inci $ p_lits lits

32+ $ pragma pragmaText (inci $ p_lits lits)

31

33

32p_topLevelWarning :: [Located RdrName] -> WarningTxt -> R ()

34p_topLevelWarning :: [Located RdrName] -> WarningTxt -> R ()

33p_topLevelWarning fnames wtxt = do

35p_topLevelWarning fnames wtxt = do

34 let (pragmaText, lits) = warningText wtxt

36 let (pragmaText, lits) = warningText wtxt

35− switchLayout (fmap getLoc fnames ++ fmap getLoc lits) $

37+ switchLayout (fmap getLoc fnames ++ fmap getLoc lits)

36− pragma pragmaText . inci $ do

38+ $ pragma pragmaText . inci

39+ $ do

37− sep commaDel p_rdrName fnames

40+ sitcc $ sep (comma >> breakpoint) p_rdrName fnames

38 breakpoint

41 breakpoint

39 p_lits lits

42 p_lits lits

40

43

@@ -46,4 +49,4 @@ warningText = \case

46p_lits :: [Located StringLiteral] -> R ()

49p_lits :: [Located StringLiteral] -> R ()

47p_lits = \case

50p_lits = \case

48 [l] -> atom l

51 [l] -> atom l

49− ls -> brackets N $ sep commaDel atom ls

52+ ls -> brackets N . sitcc $ sep (comma >> breakpoint) atom ls

src/Ormolu/Printer/Meat/ImportExport.hs+21−26

@@ -11,9 +11,9 @@ where

11

12import Control.Monad

13import GHC

14+import GHC.Hs.ImpExp (IE (..))

14import Ormolu.Printer.Combinators

15import Ormolu.Printer.Combinators

15import Ormolu.Printer.Meat.Common

16import Ormolu.Printer.Meat.Common

16−import Ormolu.Utils (RelativePos (..), attachRelativePos)

17

18p_hsmodExports :: [LIE GhcPs] -> R ()

19p_hsmodExports [] = do

@@ -21,12 +21,9 @@ p_hsmodExports [] = do

21 breakpoint'

22 txt ")"

23p_hsmodExports xs =

24− parens N $ do

24+ parens N . sitcc $ do

25 layout <- getLayout

26− sep

27− breakpoint

28− (\(p, l) -> sitcc (located l (p_lie layout p)))

26+ sep breakpoint (sitcc . located' (uncurry (p_lie layout))) (attachPositions xs)

29− (attachRelativePos xs)

30

27

31p_hsmodImport :: Bool -> ImportDecl GhcPs -> R ()

28p_hsmodImport :: Bool -> ImportDecl GhcPs -> R ()

32p_hsmodImport useQualifiedPost ImportDecl {..} = do

29p_hsmodImport useQualifiedPost ImportDecl {..} = do

@@ -65,17 +62,14 @@ p_hsmodImport useQualifiedPost ImportDecl {..} = do

65 Nothing -> return ()

62 Nothing -> return ()

66 Just (_, L _ xs) -> do

63 Just (_, L _ xs) -> do

67 breakpoint

64 breakpoint

68− parens N $ do

65+ parens N . sitcc $ do

69 layout <- getLayout

66 layout <- getLayout

70− sep

71− breakpoint

72− (\(p, l) -> sitcc (located l (p_lie layout p)))

67+ sep breakpoint (sitcc . located' (uncurry (p_lie layout))) (attachPositions xs)

73− (attachRelativePos xs)

74 newline

68 newline

75p_hsmodImport _ (XImportDecl x) = noExtCon x

69p_hsmodImport _ (XImportDecl x) = noExtCon x

76

70

77−p_lie :: Layout -> RelativePos -> IE GhcPs -> R ()

71+p_lie :: Layout -> (Int, Int) -> IE GhcPs -> R ()

78−p_lie encLayout relativePos = \case

72+p_lie encLayout (i, totalItems) = \case

79 IEVar NoExtField l1 -> do

73 IEVar NoExtField l1 -> do

80 located l1 p_ieWrappedName

74 located l1 p_ieWrappedName

81 p_comma

75 p_comma

@@ -93,8 +87,9 @@ p_lie encLayout relativePos = \case

93 inci $ do

87 inci $ do

94 let names :: [R ()]

88 let names :: [R ()]

95 names = located' p_ieWrappedName <$> xs

89 names = located' p_ieWrappedName <$> xs

96− parens N . sep commaDel sitcc $

90+ parens N . sitcc

91+ $ sep (comma >> breakpoint) sitcc

97− case w of

92+ $ case w of

98 NoIEWildcard -> names

93 NoIEWildcard -> names

99 IEWildcard n ->

94 IEWildcard n ->

100 let (before, after) = splitAt n names

95 let (before, after) = splitAt n names

@@ -104,11 +99,7 @@ p_lie encLayout relativePos = \case

104 located l1 p_hsmodName

99 located l1 p_hsmodName

105 p_comma

100 p_comma

106 IEGroup NoExtField n str -> do

101 IEGroup NoExtField n str -> do

107− case relativePos of

108− SinglePos -> return ()

109− FirstPos -> return ()

110− MiddlePos -> newline

111− LastPos -> newline

102+ unless (i == 0) newline

112 p_hsDocString (Asterisk n) False (noLoc str)

103 p_hsDocString (Asterisk n) False (noLoc str)

113 IEDoc NoExtField str ->

104 IEDoc NoExtField str ->

114 p_hsDocString Pipe False (noLoc str)

105 p_hsDocString Pipe False (noLoc str)

@@ -117,10 +108,14 @@ p_lie encLayout relativePos = \case

117 where

108 where

118 p_comma =

109 p_comma =

119 case encLayout of

110 case encLayout of

120− SingleLine ->

111+ SingleLine -> unless (i + 1 == totalItems) comma

121− case relativePos of

122− SinglePos -> return ()

123− FirstPos -> comma

124− MiddlePos -> comma

125− LastPos -> return ()

126 MultiLine -> comma

112 MultiLine -> comma

113+

114+-- | Attach positions to 'Located' things in a list.

115+attachPositions ::

116+ [Located a] ->

117+ [Located ((Int, Int), a)]

118+attachPositions xs =

119+ let f i (L l x) = L l ((i, n), x)

120+ n = length xs

121+ in zipWith f [0 ..] xs

src/Ormolu/Printer/Meat/Module.hs+20−23

@@ -11,10 +11,8 @@ where

11import Control.Monad

12import qualified Data.Text as T

13import GHC

14−import Ormolu.Imports (normalizeImports)

14+import Ormolu.Imports

15−import Ormolu.Parser.CommentStream

16import Ormolu.Parser.Pragma

15import Ormolu.Parser.Pragma

17−import Ormolu.Parser.Shebang

18import Ormolu.Printer.Combinators

16import Ormolu.Printer.Combinators

19import Ormolu.Printer.Comments

17import Ormolu.Printer.Comments

20import Ormolu.Printer.Meat.Common

18import Ormolu.Printer.Meat.Common

@@ -25,28 +23,28 @@ import Ormolu.Printer.Meat.Pragma

25

23

26-- | Render a module.

24-- | Render a module.

27p_hsModule ::

25p_hsModule ::

28− -- | Stack header

29− Maybe (RealLocated Comment) ->

30 -- | Shebangs

26 -- | Shebangs

31− [Shebang] ->

27+ [Located String] ->

32− -- | Pragmas and the associated comments

28+ -- | Pragmas

33− [([RealLocated Comment], Pragma)] ->

29+ [Pragma] ->

34 -- | Whether to use postfix qualified imports

30 -- | Whether to use postfix qualified imports

35 Bool ->

31 Bool ->

36 -- | AST to print

32 -- | AST to print

37− HsModule GhcPs ->

33+ ParsedSource ->

38 R ()

34 R ()

39−p_hsModule mstackHeader shebangs pragmas qualifiedPost HsModule {..} = do

35+p_hsModule shebangs pragmas qualifiedPost (L moduleSpan HsModule {..}) = do

40− let deprecSpan = maybe [] (\(L s _) -> [s]) hsmodDeprecMessage

36+ -- If span of exports in multiline, the whole thing is multiline. This is

37+ -- especially important because span of module itself always seems to have

38+ -- length zero, so it's not reliable for layout selection.

41− exportSpans = maybe [] (\(L s _) -> [s]) hsmodExports

39+ let exportSpans = maybe [] (\(L s _) -> [s]) hsmodExports

40+ deprecSpan = maybe [] (\(L s _) -> [s]) hsmodDeprecMessage

41+ spans' = exportSpans ++ deprecSpan ++ [moduleSpan]

42− switchLayout (deprecSpan <> exportSpans) $ do

42+ switchLayout spans' $ do

43− forM_ shebangs $ \(Shebang x) ->

43+ forM_ shebangs $ \x ->

44 located x $ \shebang -> do

45 txt (T.pack shebang)

46 newline

47− forM_ mstackHeader $ \(L spn comment) -> do

47+ spitStackHeader

48− spitCommentNow spn comment

49− newline

50 newline

48 newline

51 p_pragmas pragmas

49 p_pragmas pragmas

52 newline

50 newline

@@ -56,20 +54,19 @@ p_hsModule mstackHeader shebangs pragmas qualifiedPost HsModule {..} = do

56 located hsmodName' $ \name -> do

54 located hsmodName' $ \name -> do

57 forM_ hsmodHaddockModHeader (p_hsDocString Pipe True)

55 forM_ hsmodHaddockModHeader (p_hsDocString Pipe True)

58 p_hsmodName name

56 p_hsmodName name

59− breakpoint

60 forM_ hsmodDeprecMessage $ \w -> do

57 forM_ hsmodDeprecMessage $ \w -> do

61− located' p_moduleWarning w

62 breakpoint

58 breakpoint

59+ located' p_moduleWarning w

63 case hsmodExports of

60 case hsmodExports of

64 Nothing -> return ()

61 Nothing -> return ()

65− Just l -> do

62+ Just hsmodExports' -> do

66− located l $ \exports -> do

67− inci (p_hsmodExports exports)

68 breakpoint

63 breakpoint

64+ inci (p_hsmodExports (unLoc hsmodExports'))

65+ breakpoint

69 txt "where"

66 txt "where"

70 newline

67 newline

71 newline

68 newline

72− forM_ (normalizeImports hsmodImports) (located' (p_hsmodImport qualifiedPost))

69+ forM_ (sortImports hsmodImports) (located' (p_hsmodImport qualifiedPost))

73 newline

70 newline

74 switchLayout (getLoc <$> hsmodDecls) $ do

71 switchLayout (getLoc <$> hsmodDecls) $ do

75 p_hsDecls Free hsmodDecls

72 p_hsDecls Free hsmodDecls

src/Ormolu/Printer/Meat/Pragma.hs+12−22

@@ -8,16 +8,12 @@ module Ormolu.Printer.Meat.Pragma

8 )

9where

10

11−import Control.Monad

12import Data.Char (isUpper)

11import Data.Char (isUpper)

13−import qualified Data.List as L

14import Data.Maybe (listToMaybe)

12import Data.Maybe (listToMaybe)

13+import qualified Data.Set as S

15import qualified Data.Text as T

14import qualified Data.Text as T

16−import Ormolu.Parser.CommentStream

17import Ormolu.Parser.Pragma (Pragma (..))

15import Ormolu.Parser.Pragma (Pragma (..))

18import Ormolu.Printer.Combinators

16import Ormolu.Printer.Combinators

19−import Ormolu.Printer.Comments

20−import SrcLoc

21

17

22-- | Pragma classification.

18-- | Pragma classification.

23data PragmaTy

19data PragmaTy

@@ -46,31 +42,25 @@ data LanguagePragmaClass

46 Final

42 Final

47 deriving (Eq, Ord)

43 deriving (Eq, Ord)

48

44

49−-- | Print a collection of 'Pragma's with their associated comments.

50−p_pragmas :: [([RealLocated Comment], Pragma)] -> R ()

45+p_pragmas :: [Pragma] -> R ()

51−p_pragmas ps = do

46+p_pragmas ps =

52− let prepare = L.sortOn snd . L.nub . concatMap analyze

47+ let prepare = concatMap $ \case

53− analyze = \case

54− (cs, PragmaLanguage xs) ->

48+ PragmaLanguage xs ->

55− let f x = (cs, (Language (classifyLanguagePragma x), x))

49+ let f x = (Language (classifyLanguagePragma x), x)

56 in f <$> xs

50 in f <$> xs

57− (cs, PragmaOptionsGHC x) -> [(cs, (OptionsGHC, x))]

51+ PragmaOptionsGHC x -> [(OptionsGHC, x)]

58− (cs, PragmaOptionsHaddock x) -> [(cs, (OptionsHaddock, x))]

52+ PragmaOptionsHaddock x -> [(OptionsHaddock, x)]

59− forM_ (prepare ps) $ \(cs, (pragmaTy, x)) ->

53+ in mapM_ (uncurry p_pragma) (S.toAscList . S.fromList . prepare $ ps)

60− p_pragma cs pragmaTy x

61

54

62−p_pragma :: [RealLocated Comment] -> PragmaTy -> String -> R ()

55+p_pragma :: PragmaTy -> String -> R ()

63−p_pragma comments ty x = do

56+p_pragma ty c = do

64− forM_ comments $ \(L l comment) -> do

65− spitCommentNow l comment

66− newline

67 txt "{-# "

57 txt "{-# "

68 txt $ case ty of

58 txt $ case ty of

69 Language _ -> "LANGUAGE"

59 Language _ -> "LANGUAGE"

70 OptionsGHC -> "OPTIONS_GHC"

60 OptionsGHC -> "OPTIONS_GHC"

71 OptionsHaddock -> "OPTIONS_HADDOCK"

61 OptionsHaddock -> "OPTIONS_HADDOCK"

72 space

62 space

73− txt (T.pack x)

63+ txt (T.pack c)

74 txt " #-}"

64 txt " #-}"

75 newline

65 newline

76

66

src/Ormolu/Printer/Meat/Type.hs+27−48

@@ -5,7 +5,6 @@

5-- | Rendering of types.

6module Ormolu.Printer.Meat.Type

7 ( p_hsType,

8− p_hsTypePostDoc,

9 hasDocStrings,

8 hasDocStrings,

10 p_hsContext,

9 p_hsContext,

11 p_hsTyVarBndr,

10 p_hsTyVarBndr,

@@ -24,22 +23,14 @@ import Ormolu.Printer.Operators

24import Ormolu.Utils

23import Ormolu.Utils

25

24

26p_hsType :: HsType GhcPs -> R ()

25p_hsType :: HsType GhcPs -> R ()

27−p_hsType t = p_hsType' (hasDocStrings t) PipeStyle t

26+p_hsType t = p_hsType' (hasDocStrings t) t

28

27

29−p_hsTypePostDoc :: HsType GhcPs -> R ()

30−p_hsTypePostDoc t = p_hsType' (hasDocStrings t) CaretStyle t

31−

32−-- | How to render Haddocks associated with a type.

33−data TypeDocStyle

34− = PipeStyle

35− | CaretStyle

36−

37−p_hsType' :: Bool -> TypeDocStyle -> HsType GhcPs -> R ()

28+p_hsType' :: Bool -> HsType GhcPs -> R ()

38−p_hsType' multilineArgs docStyle = \case

29+p_hsType' multilineArgs = \case

39 HsForAllTy NoExtField visibility bndrs t -> do

30 HsForAllTy NoExtField visibility bndrs t -> do

40 p_forallBndrs visibility p_hsTyVarBndr bndrs

31 p_forallBndrs visibility p_hsTyVarBndr bndrs

41 interArgBreak

32 interArgBreak

42− p_hsTypeR (unLoc t)

33+ p_hsType' multilineArgs (unLoc t)

43 HsQualTy NoExtField qs t -> do

34 HsQualTy NoExtField qs t -> do

44 located qs p_hsContext

35 located qs p_hsContext

45 space

36 space

@@ -58,20 +49,10 @@ p_hsType' multilineArgs docStyle = \case

58 _ -> return ()

49 _ -> return ()

59 NotPromoted -> return ()

50 NotPromoted -> return ()

60 p_rdrName n

51 p_rdrName n

61− HsAppTy NoExtField f x -> do

52+ HsAppTy NoExtField f x -> sitcc $ do

62− let -- In order to format type applications with multiple parameters

63− -- nicer, traverse the AST to gather the function and all the

64− -- parameters together.

65− gatherArgs f' knownArgs =

66− case f' of

67− L _ (HsAppTy _ l r) -> gatherArgs l (r : knownArgs)

68− _ -> (f', knownArgs)

69− (func, args) = gatherArgs f [x]

70− switchLayout (getLoc f : fmap getLoc args) . sitcc $ do

71− located func p_hsType

53+ located f p_hsType

72− breakpoint

54+ breakpoint

73− inci $

74− sep breakpoint (located' p_hsType) args

55+ inci (located x p_hsType)

75 HsAppKindTy _ ty kd -> sitcc $ do

56 HsAppKindTy _ ty kd -> sitcc $ do

76 -- The first argument is the location of the "@..." part. Not 100% sure,

57 -- The first argument is the location of the "@..." part. Not 100% sure,

77 -- but I think we can ignore it as long as we use 'located' on both the

58 -- but I think we can ignore it as long as we use 'located' on both the

@@ -98,9 +79,10 @@ p_hsType' multilineArgs docStyle = \case

98 HsBoxedTuple -> parens N

79 HsBoxedTuple -> parens N

99 HsConstraintTuple -> parens N

80 HsConstraintTuple -> parens N

100 HsBoxedOrConstraintTuple -> parens N

81 HsBoxedOrConstraintTuple -> parens N

82+ in parens' . sitcc $

101− in parens' $ sep commaDel (sitcc . located' p_hsType) xs

83+ sep (comma >> breakpoint) (sitcc . located' p_hsType) xs

102 HsSumTy NoExtField xs ->

84 HsSumTy NoExtField xs ->

103− parensHash N $

85+ parensHash N . sitcc $

104 sep (txt "|" >> breakpoint) (sitcc . located' p_hsType) xs

86 sep (txt "|" >> breakpoint) (sitcc . located' p_hsType) xs

105 HsOpTy NoExtField x op y ->

87 HsOpTy NoExtField x op y ->

106 sitcc $

88 sitcc $

@@ -117,20 +99,14 @@ p_hsType' multilineArgs docStyle = \case

117 HsStarTy NoExtField _ -> txt "*"

99 HsStarTy NoExtField _ -> txt "*"

118 HsKindSig NoExtField t k -> sitcc $ do

100 HsKindSig NoExtField t k -> sitcc $ do

119 located t p_hsType

101 located t p_hsType

120− space

102+ space -- FIXME

121 txt "::"

103 txt "::"

122− breakpoint

104+ space

123 inci (located k p_hsType)

105 inci (located k p_hsType)

124 HsSpliceTy NoExtField splice -> p_hsSplice splice

106 HsSpliceTy NoExtField splice -> p_hsSplice splice

125− HsDocTy NoExtField t str ->

107+ HsDocTy NoExtField t str -> do

126− case docStyle of

127− PipeStyle -> do

128− p_hsDocString Pipe True str

108+ p_hsDocString Pipe True str

129− located t p_hsType

130− CaretStyle -> do

131− located t p_hsType

109+ located t p_hsType

132− newline

133− p_hsDocString Caret False str

134 HsBangTy NoExtField (HsSrcBang _ u s) t -> do

110 HsBangTy NoExtField (HsSrcBang _ u s) t -> do

135 case u of

111 case u of

136 SrcUnpack -> txt "{-# UNPACK #-}" >> space

112 SrcUnpack -> txt "{-# UNPACK #-}" >> space

@@ -153,14 +129,14 @@ p_hsType' multilineArgs docStyle = \case

153 case (p, xs) of

129 case (p, xs) of

154 (IsPromoted, L _ t : _) | isPromoted t -> space

130 (IsPromoted, L _ t : _) | isPromoted t -> space

155 _ -> return ()

131 _ -> return ()

156− sep commaDel (sitcc . located' p_hsType) xs

132+ sitcc $ sep (comma >> breakpoint) (sitcc . located' p_hsType) xs

157 HsExplicitTupleTy NoExtField xs -> do

133 HsExplicitTupleTy NoExtField xs -> do

158 txt "'"

134 txt "'"

159 parens N $ do

135 parens N $ do

160 case xs of

136 case xs of

161 L _ t : _ | isPromoted t -> space

137 L _ t : _ | isPromoted t -> space

162 _ -> return ()

138 _ -> return ()

163− sep commaDel (located' p_hsType) xs

139+ sep (comma >> breakpoint) (located' p_hsType) xs

164 HsTyLit NoExtField t ->

140 HsTyLit NoExtField t ->

165 case t of

141 case t of

166 HsStrTy (SourceText s) _ -> p_stringLit s

142 HsStrTy (SourceText s) _ -> p_stringLit s

@@ -170,14 +146,14 @@ p_hsType' multilineArgs docStyle = \case

170 where

146 where

171 isPromoted = \case

147 isPromoted = \case

172 HsTyVar _ IsPromoted _ -> True

148 HsTyVar _ IsPromoted _ -> True

173− HsExplicitTupleTy {} -> True

174 HsExplicitListTy {} -> True

149 HsExplicitListTy {} -> True

150+ HsExplicitTupleTy {} -> True

175 _ -> False

151 _ -> False

176 interArgBreak =

152 interArgBreak =

177 if multilineArgs

153 if multilineArgs

178 then newline

154 then newline

179 else breakpoint

155 else breakpoint

180− p_hsTypeR = p_hsType' multilineArgs docStyle

156+ p_hsTypeR = p_hsType' multilineArgs

181

157

182-- | Return 'True' if at least one argument in 'HsType' has a doc string

158-- | Return 'True' if at least one argument in 'HsType' has a doc string

183-- attached to it.

159-- attached to it.

@@ -191,7 +167,9 @@ p_hsContext :: HsContext GhcPs -> R ()

191p_hsContext = \case

167p_hsContext = \case

192 [] -> txt "()"

168 [] -> txt "()"

193 [x] -> located x p_hsType

169 [x] -> located x p_hsType

170+ xs ->

171+ parens N . sitcc $

194− xs -> parens N $ sep commaDel (sitcc . located' p_hsType) xs

172+ sep (comma >> breakpoint) (sitcc . located' p_hsType) xs

195

173

196p_hsTyVarBndr :: HsTyVarBndr GhcPs -> R ()

174p_hsTyVarBndr :: HsTyVarBndr GhcPs -> R ()

197p_hsTyVarBndr = \case

175p_hsTyVarBndr = \case

@@ -221,14 +199,15 @@ p_forallBndrs vis p tyvars =

221

199

222p_conDeclFields :: [LConDeclField GhcPs] -> R ()

200p_conDeclFields :: [LConDeclField GhcPs] -> R ()

223p_conDeclFields xs =

201p_conDeclFields xs =

202+ braces N . sitcc $

224− braces N $ sep commaDel (sitcc . located' p_conDeclField) xs

203+ sep (comma >> breakpoint) (sitcc . located' p_conDeclField) xs

225

204

226p_conDeclField :: ConDeclField GhcPs -> R ()

205p_conDeclField :: ConDeclField GhcPs -> R ()

227p_conDeclField ConDeclField {..} = do

206p_conDeclField ConDeclField {..} = do

228 mapM_ (p_hsDocString Pipe True) cd_fld_doc

207 mapM_ (p_hsDocString Pipe True) cd_fld_doc

229 sitcc $

208 sitcc $

230 sep

209 sep

231− commaDel

210+ (comma >> breakpoint)

232 (located' (p_rdrName . rdrNameFieldOcc))

211 (located' (p_rdrName . rdrNameFieldOcc))

233 cd_fld_names

212 cd_fld_names

234 space

213 space

@@ -270,6 +249,6 @@ tyVarToType = \case

270 -- <https://gitlab.haskell.org/ghc/ghc/issues/17404>. This is fine as

249 -- <https://gitlab.haskell.org/ghc/ghc/issues/17404>. This is fine as

271 -- long as 'tyVarToType' does not get applied to right-hand sides of

250 -- long as 'tyVarToType' does not get applied to right-hand sides of

272 -- declarations.

251 -- declarations.

273− HsParTy NoExtField . noLoc $

252+ HsParTy NoExtField $ noLoc $

274 HsKindSig NoExtField (noLoc (HsTyVar NoExtField NotPromoted tvar)) kind

253 HsKindSig NoExtField (noLoc (HsTyVar NoExtField NotPromoted tvar)) kind

275 XTyVarBndr x -> noExtCon x

254 XTyVarBndr x -> noExtCon x

src/Ormolu/Printer/Operators.hs+66−50

@@ -10,14 +10,16 @@ module Ormolu.Printer.Operators

10 )

11where

12

13+import BasicTypes (Fixity (..), SourceText (NoSourceText), compareFixity, defaultFixity)

13import Data.Function (on)

14import Data.Function (on)

14−import qualified Data.List as L

15+import Data.List

15−import Data.Map.Strict (Map)

16−import qualified Data.Map.Strict as M

17import Data.Maybe (fromMaybe, mapMaybe)

16import Data.Maybe (fromMaybe, mapMaybe)

17+import Data.Ord (Down (Down), comparing)

18import GHC

19−import OccName (occNameString)

19+import OccName (mkVarOcc)

20import Ormolu.Utils (unSrcSpan)

21+import RdrName (mkRdrUnqual)

22+import SrcLoc (combineSrcSpans)

21

23

22-- | Intermediate representation of operator trees. It has two type

24-- | Intermediate representation of operator trees. It has two type

23-- parameters: @ty@ is the type of sub-expressions, while @op@ is the type

25-- parameters: @ty@ is the type of sub-expressions, while @op@ is the type

@@ -57,7 +59,7 @@ reassociateOpTree getOpName opTree =

57reassociateOpTreeWith ::

59reassociateOpTreeWith ::

58 forall ty op.

60 forall ty op.

59 -- | Fixity map for operators

61 -- | Fixity map for operators

60− Map String Fixity ->

62+ [(RdrName, Fixity)] ->

61 -- | How to get the name of an operator

63 -- | How to get the name of an operator

62 (op -> Maybe RdrName) ->

64 (op -> Maybe RdrName) ->

63 -- | Original 'OpTree'

65 -- | Original 'OpTree'

@@ -68,13 +70,13 @@ reassociateOpTreeWith fixityMap getOpName = go

68 where

70 where

69 fixityOf :: op -> Fixity

71 fixityOf :: op -> Fixity

70 fixityOf op = fromMaybe defaultFixity $ do

72 fixityOf op = fromMaybe defaultFixity $ do

71− s <- occNameString . rdrNameOcc <$> getOpName op

73+ opName <- getOpName op

72− M.lookup s fixityMap

74+ lookup opName fixityMap

73 -- Here, left branch is already associated and the root alongside with

75 -- Here, left branch is already associated and the root alongside with

74 -- the right branch is right-associated. This function picks up one item

76 -- the right branch is right-associated. This function picks up one item

75 -- from the right and inserts it correctly to the left.

77 -- from the right and inserts it correctly to the left.

76 --

78 --

77− -- Also, we are using the 'compareFixity' function which tells if the

79+ -- Also, we are using the 'compareFixity' function which returns if the

78 -- expression should associate to right.

80 -- expression should associate to right.

79 go :: OpTree ty op -> OpTree ty op

81 go :: OpTree ty op -> OpTree ty op

80 -- base cases

82 -- base cases

@@ -94,16 +96,6 @@ reassociateOpTreeWith fixityMap getOpName = go

94 then go $ OpBranch (OpBranch l op (go $ OpBranch r op' l')) op'' r'

96 then go $ OpBranch (OpBranch l op (go $ OpBranch r op' l')) op'' r'

95 else go $ OpBranch (OpBranch (OpBranch l op r) op' l') op'' r'

97 else go $ OpBranch (OpBranch (OpBranch l op r) op' l') op'' r'

96

98

97−-- | A score assigned to an operator.

98−data Score

99− = -- | The operator was placed at the beginning of a line

100− AtBeginning Int

101− | -- | The operator was placed at the end of a line

102− AtEnd

103− | -- | The operator was placed in between arguments on a single line

104− InBetween

105− deriving (Eq, Ord)

106−

107-- | Build a map of inferred 'Fixity's from an 'OpTree'.

99-- | Build a map of inferred 'Fixity's from an 'OpTree'.

108buildFixityMap ::

100buildFixityMap ::

109 forall ty op.

101 forall ty op.

@@ -112,26 +104,23 @@ buildFixityMap ::

112 -- | Operator tree

104 -- | Operator tree

113 OpTree (Located ty) (Located op) ->

105 OpTree (Located ty) (Located op) ->

114 -- | Fixity map

106 -- | Fixity map

115− Map String Fixity

107+ [(RdrName, Fixity)]

116buildFixityMap getOpName opTree =

108buildFixityMap getOpName opTree =

117− addOverrides

118− . M.fromList

119− . concatMap (\(i, ns) -> map (\(n, _) -> (n, fixity i InfixL)) ns)

109+ concatMap (\(i, ns) -> map (\(n, _) -> (n, fixity i InfixL)) ns)

120− . zip [2 ..]

110+ . zip [0 ..]

121− . L.groupBy ((==) `on` snd)

111+ . groupBy (doubleWithinEps 0.00001 `on` snd)

122− . selectScores

112+ . (overrides ++)

113+ . modeScores

123 $ score opTree

114 $ score opTree

124 where

115 where

125− addOverrides :: Map String Fixity -> Map String Fixity

116+ -- Add a special case for ($), since it is pretty unlikely for someone

126− addOverrides m =

117+ -- to override it.

127− M.fromList

118+ overrides :: [(RdrName, Double)]

128− [ ("$", fixity 0 InfixR),

119+ overrides =

129− (":", fixity 1 InfixR),

120+ [ (mkRdrUnqual $ mkVarOcc "$", -1)

130− (".", fixity 100 InfixL)

121+ ]

131− ]

122+ -- Assign scores to operators based on their location in the source.

132− `M.union` m

133− fixity = Fixity NoSourceText

134− score :: OpTree (Located ty) (Located op) -> [(String, Score)]

123+ score :: OpTree (Located ty) (Located op) -> [(RdrName, Double)]

135 score (OpNode _) = []

124 score (OpNode _) = []

136 score (OpBranch l o r) = fromMaybe (score r) $ do

125 score (OpBranch l o r) = fromMaybe (score r) $ do

137 -- If we fail to get any of these, 'defaultFixity' will be used by

126 -- If we fail to get any of these, 'defaultFixity' will be used by

@@ -141,27 +130,48 @@ buildFixityMap getOpName opTree =

141 oe <- srcSpanEndLine <$> unSrcSpan (getLoc o) -- operator end

130 oe <- srcSpanEndLine <$> unSrcSpan (getLoc o) -- operator end

142 rb <- srcSpanStartLine <$> unSrcSpan (opTreeLoc r) -- right begin

131 rb <- srcSpanStartLine <$> unSrcSpan (opTreeLoc r) -- right begin

143 oc <- srcSpanStartCol <$> unSrcSpan (getLoc o) -- operator column

132 oc <- srcSpanStartCol <$> unSrcSpan (getLoc o) -- operator column

144− opName <- occNameString . rdrNameOcc <$> getOpName (unLoc o)

133+ opName <- getOpName (unLoc o)

145 let s

134 let s

146− | le < ob = AtBeginning oc

135+ | le < ob =

136+ -- if the operator is in the beginning of a line, assign

137+ -- a score relative to its column within range [0, 1).

138+ fromIntegral oc / fromIntegral (maxCol + 1)

147− | oe < rb = AtEnd

139+ | oe < rb =

140+ -- if the operator is in the end of the line, assign the

141+ -- score 1.

142+ 1

148− | otherwise = InBetween

143+ | otherwise =

144+ 2 -- otherwise, assign a high score.

149 return $ (opName, s) : score r

145 return $ (opName, s) : score r

150− selectScores :: [(String, Score)] -> [(String, Score)]

146+ -- Pick the most common score per 'RdrName'.

151− selectScores =

147+ modeScores :: [(RdrName, Double)] -> [(RdrName, Double)]

148+ modeScores =

152− L.sortOn snd

149+ sortOn snd

153 . mapMaybe

150 . mapMaybe

154 ( \case

151 ( \case

155 [] -> Nothing

152 [] -> Nothing

156− xs@((n, _) : _) -> Just (n, selectScore $ map snd xs)

153+ xs@((n, _) : _) -> Just (n, mode $ map snd xs)

157 )

154 )

158− . L.groupBy ((==) `on` fst)

155+ . groupBy ((==) `on` fst)

159− . L.sort

156+ . sort

160− selectScore :: [Score] -> Score

157+ -- Return the most common number, leaning to the smaller

161− selectScore xs =

158+ -- one in case of a tie.

162− case filter (/= InBetween) xs of

159+ mode :: [Double] -> Double

163− [] -> InBetween

160+ mode =

161+ head

162+ . minimumBy (comparing (Down . length))

163+ . groupBy (doubleWithinEps 0.0001)

164+ . sort

165+ -- The start column of the rightmost operator.

166+ maxCol = go opTree

167+ where

168+ go (OpNode (L _ _)) = 0

169+ go (OpBranch l (L o _) r) =

164− xs' -> maximum xs'

170+ maximum

171+ [ go l,

172+ maybe 0 srcSpanStartCol (unSrcSpan o),

173+ go r

174+ ]

165

175

166----------------------------------------------------------------------------

176----------------------------------------------------------------------------

167-- Helpers

177-- Helpers

@@ -175,3 +185,9 @@ normalizeOpTree (OpBranch (OpNode l) lop r) =

175 OpBranch (OpNode l) lop (normalizeOpTree r)

185 OpBranch (OpNode l) lop (normalizeOpTree r)

176normalizeOpTree (OpBranch (OpBranch l' lop' r') lop r) =

186normalizeOpTree (OpBranch (OpBranch l' lop' r') lop r) =

177 normalizeOpTree (OpBranch l' lop' (OpBranch r' lop r))

187 normalizeOpTree (OpBranch l' lop' (OpBranch r' lop r))

188+

189+fixity :: Int -> FixityDirection -> Fixity

190+fixity = Fixity NoSourceText

191+

192+doubleWithinEps :: Double -> Double -> Double -> Bool

193+doubleWithinEps eps a b = abs (a - b) < eps

src/Ormolu/Processing/Common.hsdeleted−27

@@ -1,27 +0,0 @@

1−{-# LANGUAGE OverloadedStrings #-}

2−

3−-- | Common definitions for pre- and post- processing.

4−module Ormolu.Processing.Common

5− ( OrmoluState (..),

6− startDisabling,

7− endDisabling,

8− )

9−where

10−

11−import Data.String (IsString (..))

12−

13−-- | Ormolu state.

14−data OrmoluState

15− = -- | Enabled

16− OrmoluEnabled

17− | -- | Disabled

18− OrmoluDisabled

19− deriving (Eq, Show)

20−

21−-- | Marker for the beginning of the region where Ormolu should be disabled.

22−startDisabling :: IsString s => s

23−startDisabling = "{- ORMOLU_DISABLE_START"

24−

25−-- | Marker for the end of the region where Ormolu should be disabled.

26−endDisabling :: IsString s => s

27−endDisabling = "ORMOLU_DISABLE_END -}"

src/Ormolu/Processing/Cpp.hsdeleted−70

@@ -1,70 +0,0 @@

1−{-# LANGUAGE OverloadedStrings #-}

2−

3−-- | Support for CPP.

4−module Ormolu.Processing.Cpp

5− ( State (..),

6− processLine,

7− unmaskLine,

8− )

9−where

10−

11−import Control.Monad

12−import Data.Char (isSpace)

13−import qualified Data.List as L

14−import Data.Maybe (isJust)

15−import Data.String

16−import Data.Text (Text)

17−import qualified Data.Text as T

18−

19−-- | State of the CPP processor.

20−data State

21− = -- | Outside of CPP directives

22− Outside

23− | -- | In a conditional expression

24− InConditional

25− | -- | In a continuation (after @\\@)

26− InContinuation

27− deriving (Eq, Show)

28−

29−-- | Automatically mask the line when needed and update the 'State'.

30−processLine :: String -> State -> (String, State)

31−processLine line state

32− | for "define " = (masked, state')

33− | for "include " = (masked, state')

34− | for "undef " = (masked, state')

35− | for "ifdef " = (masked, InConditional)

36− | for "ifndef " = (masked, InConditional)

37− | for "if " = (masked, InConditional)

38− | for "else" = (masked, InConditional)

39− | for "elif" = (masked, InConditional)

40− | for "endif" = (masked, state')

41− | otherwise =

42− case state of

43− Outside -> (line, Outside)

44− InConditional -> (masked, InConditional)

45− InContinuation -> (masked, state')

46− where

47− for directive = isJust $ do

48− s <- dropWhile isSpace <$> L.stripPrefix "#" line

49− void (L.stripPrefix directive s)

50− masked = maskLine line

51− state' =

52− if "\\" `L.isSuffixOf` line

53− then InContinuation

54− else Outside

55−

56−-- | Mask the given line.

57−maskLine :: String -> String

58−maskLine x = maskPrefix ++ x

59−

60−-- | If the given line is masked, unmask it. Otherwise return the line

61−-- unchanged.

62−unmaskLine :: Text -> Text

63−unmaskLine x =

64− case T.stripPrefix maskPrefix (T.stripStart x) of

65− Nothing -> x

66− Just x' -> x'

67−

68−-- | Mask prefix for CPP.

69−maskPrefix :: IsString s => s

70−maskPrefix = "-- ORMOLU_CPP_MASK"

src/Ormolu/Processing/Postprocess.hsdeleted−31

@@ -1,31 +0,0 @@

1−{-# LANGUAGE OverloadedStrings #-}

2−{-# LANGUAGE ViewPatterns #-}

3−

4−-- | Postprocessing for the results of printing.

5−module Ormolu.Processing.Postprocess

6− ( postprocess,

7− )

8−where

9−

10−import Data.Text (Text)

11−import qualified Data.Text as T

12−import Ormolu.Processing.Common

13−import qualified Ormolu.Processing.Cpp as Cpp

14−

15−-- | Postprocess output of the formatter.

16−postprocess ::

17− -- | Desired indentation level

18− Int ->

19− -- | Input to process

20− Text ->

21− Text

22−postprocess indent =

23− T.unlines

24− . fmap indentLine

25− . fmap Cpp.unmaskLine

26− . filter (not . magicComment)

27− . T.lines

28− where

29− magicComment (T.stripStart -> x) =

30− x == startDisabling || x == endDisabling

31− indentLine x = T.replicate indent " " <> x

src/Ormolu/Processing/Preprocess.hsdeleted−149

@@ -1,149 +0,0 @@

1−{-# LANGUAGE BangPatterns #-}

2−{-# LANGUAGE LambdaCase #-}

3−{-# LANGUAGE RecordWildCards #-}

4−

5−-- | Preprocessing for input source code.

6−module Ormolu.Processing.Preprocess

7− ( preprocess,

8− )

9−where

10−

11−import Control.Monad

12−import Data.Char (isSpace)

13−import qualified Data.List as L

14−import Data.Maybe (isJust, maybeToList)

15−import FastString

16−import Ormolu.Config (RegionDeltas (..))

17−import Ormolu.Parser.Shebang (isShebang)

18−import Ormolu.Processing.Common

19−import qualified Ormolu.Processing.Cpp as Cpp

20−import SrcLoc

21−

22−-- | Transform given input possibly returning comments extracted from it.

23−-- This handles LINE pragmas, CPP, shebangs, and the magic comments for

24−-- enabling\/disabling of Ormolu.

25−preprocess ::

26− -- | File name, just to use in the spans

27− FilePath ->

28− -- | Input to process

29− String ->

30− -- | Region deltas

31− RegionDeltas ->

32− -- | Literal prefix, pre-processed input, literal suffix, extra comments

33− (String, String, String, [Located String])

34−preprocess path input RegionDeltas {..} =

35− go 1 OrmoluEnabled Cpp.Outside id id regionLines

36− where

37− (prefixLines, otherLines) = splitAt regionPrefixLength (lines input)

38− (regionLines, suffixLines) =

39− let regionLength = length otherLines - regionSuffixLength

40− in splitAt regionLength otherLines

41− go !n ormoluState cppState inputSoFar csSoFar = \case

42− [] ->

43− let input' = unlines (inputSoFar [])

44− in ( unlines prefixLines,

45− case ormoluState of

46− OrmoluEnabled -> input'

47− OrmoluDisabled -> input' ++ endDisabling,

48− unlines suffixLines,

49− csSoFar []

50− )

51− (x : xs) ->

52− let (x', ormoluState', cppState', cs) =

53− processLine path n ormoluState cppState x

54− in go

55− (n + 1)

56− ormoluState'

57− cppState'

58− (inputSoFar . (x' :))

59− (csSoFar . (maybeToList cs ++))

60− xs

61−

62−-- | Transform a given line possibly returning a comment extracted from it.

63−processLine ::

64− -- | File name, just to use in the spans

65− FilePath ->

66− -- | Line number of this line

67− Int ->

68− -- | Whether Ormolu is currently enabled

69− OrmoluState ->

70− -- | CPP state

71− Cpp.State ->

72− -- | The actual line

73− String ->

74− -- | Adjusted line and possibly a comment extracted from it

75− (String, OrmoluState, Cpp.State, Maybe (Located String))

76−processLine path n ormoluState Cpp.Outside line

77− | "{-# LINE" `L.isPrefixOf` line =

78− let (pragma, res) = getPragma line

79− size = length pragma

80− ss = mkSrcSpan (mkSrcLoc' 1) (mkSrcLoc' (size + 1))

81− in (res, ormoluState, Cpp.Outside, Just (L ss pragma))

82− | isOrmoluEnable line =

83− case ormoluState of

84− OrmoluEnabled ->

85− (enableMarker, OrmoluEnabled, Cpp.Outside, Nothing)

86− OrmoluDisabled ->

87− (endDisabling ++ enableMarker, OrmoluEnabled, Cpp.Outside, Nothing)

88− | isOrmoluDisable line =

89− case ormoluState of

90− OrmoluEnabled ->

91− (disableMarker ++ startDisabling, OrmoluDisabled, Cpp.Outside, Nothing)

92− OrmoluDisabled ->

93− (disableMarker, OrmoluDisabled, Cpp.Outside, Nothing)

94− | isShebang line =

95− let ss = mkSrcSpan (mkSrcLoc' 1) (mkSrcLoc' (length line))

96− in ("", ormoluState, Cpp.Outside, Just (L ss line))

97− | otherwise =

98− let (line', cppState') = Cpp.processLine line Cpp.Outside

99− in (line', ormoluState, cppState', Nothing)

100− where

101− mkSrcLoc' = mkSrcLoc (mkFastString path) n

102−processLine _ _ ormoluState cppState line =

103− let (line', cppState') = Cpp.processLine line cppState

104− in (line', ormoluState, cppState', Nothing)

105−

106−-- | Take a line pragma and output its replacement (where line pragma is

107−-- replaced with spaces) and the contents of the pragma itself.

108−getPragma ::

109− -- | Pragma line to analyze

110− String ->

111− -- | Contents of the pragma and its replacement line

112− (String, String)

113−getPragma [] = error "Ormolu.Preprocess.getPragma: input must not be empty"

114−getPragma s@(x : xs)

115− | "#-}" `L.isPrefixOf` s = ("#-}", " " ++ drop 3 s)

116− | otherwise =

117− let (prag, remline) = getPragma xs

118− in (x : prag, ' ' : remline)

119−

120−-- | Canonical enable marker.

121−enableMarker :: String

122−enableMarker = "{- ORMOLU_ENABLE -}"

123−

124−-- | Canonical disable marker.

125−disableMarker :: String

126−disableMarker = "{- ORMOLU_DISABLE -}"

127−

128−-- | Return 'True' if the given string is an enabling marker.

129−isOrmoluEnable :: String -> Bool

130−isOrmoluEnable = magicComment "ORMOLU_ENABLE"

131−

132−-- | Return 'True' if the given string is a disabling marker.

133−isOrmoluDisable :: String -> Bool

134−isOrmoluDisable = magicComment "ORMOLU_DISABLE"

135−

136−-- | Construct a function for whitespace-insensitive matching of string.

137−magicComment ::

138− -- | What to expect

139− String ->

140− -- | String to test

141− String ->

142− -- | Whether or not the two strings watch

143− Bool

144−magicComment expected s0 = isJust $ do

145− let trim = dropWhile isSpace

146− s1 <- trim <$> L.stripPrefix "{-" (trim s0)

147− s2 <- trim <$> L.stripPrefix expected s1

148− s3 <- L.stripPrefix "-}" s2

149− guard (all isSpace s3)

src/Ormolu/Utils.hs+16−74

@@ -1,66 +1,45 @@

1{-# LANGUAGE LambdaCase #-}

2{-# LANGUAGE OverloadedStrings #-}

3−{-# LANGUAGE ViewPatterns #-}

4

3

5-- | Random utilities used by the code.

4-- | Random utilities used by the code.

6module Ormolu.Utils

5module Ormolu.Utils

7− ( RelativePos (..),

8− attachRelativePos,

9− combineSrcSpans',

6+ ( combineSrcSpans',

7+ isModule,

10 notImplemented,

8 notImplemented,

11 showOutputable,

9 showOutputable,

12 splitDocString,

10 splitDocString,

13 typeArgToType,

11 typeArgToType,

14 unSrcSpan,

12 unSrcSpan,

15− incSpanLine,

16 separatedByBlank,

13 separatedByBlank,

17− separatedByBlankNE,

18− onTheSameLine,

19− removeIndentation,

20 )

14 )

21where

15where

22

16

23−import Data.Char (isSpace)

17+import Data.Data (Data, showConstr, toConstr)

24import Data.List (dropWhileEnd)

18import Data.List (dropWhileEnd)

25−import Data.List.NonEmpty (NonEmpty (..))

26import qualified Data.List.NonEmpty as NE

19import qualified Data.List.NonEmpty as NE

20+import Data.List.NonEmpty (NonEmpty (..))

27import Data.Maybe (fromMaybe)

21import Data.Maybe (fromMaybe)

28import Data.Text (Text)

22import Data.Text (Text)

29import qualified Data.Text as T

23import qualified Data.Text as T

30import GHC

24import GHC

31−import GHC.DynFlags (baseDynFlags)

25+import GHC.Hs.Doc (HsDocString, unpackHDS)

32import qualified Outputable as GHC

26import qualified Outputable as GHC

33

27

34−-- | Relative positions in a list.

35−data RelativePos

36− = SinglePos

37− | FirstPos

38− | MiddlePos

39− | LastPos

40− deriving (Eq, Show)

41−

42−-- | Attach 'RelativePos'es to elements of a given list.

43−attachRelativePos :: [a] -> [(RelativePos, a)]

44−attachRelativePos = \case

45− [] -> []

46− [x] -> [(SinglePos, x)]

47− (x : xs) -> (FirstPos, x) : markLast xs

48− where

49− markLast [] = []

50− markLast [x] = [(LastPos, x)]

51− markLast (x : xs) = (MiddlePos, x) : markLast xs

52−

53-- | Combine all source spans from the given list.

28-- | Combine all source spans from the given list.

54combineSrcSpans' :: NonEmpty SrcSpan -> SrcSpan

29combineSrcSpans' :: NonEmpty SrcSpan -> SrcSpan

55combineSrcSpans' (x :| xs) = foldr combineSrcSpans x xs

30combineSrcSpans' (x :| xs) = foldr combineSrcSpans x xs

56

31

32+-- | Return 'True' if given element of AST is module.

33+isModule :: Data a => a -> Bool

34+isModule x = showConstr (toConstr x) == "HsModule"

35+

57-- | Placeholder for things that are not yet implemented.

36-- | Placeholder for things that are not yet implemented.

58notImplemented :: String -> a

37notImplemented :: String -> a

59notImplemented msg = error $ "not implemented yet: " ++ msg

38notImplemented msg = error $ "not implemented yet: " ++ msg

60

39

61-- | Pretty-print an 'GHC.Outputable' thing.

40-- | Pretty-print an 'GHC.Outputable' thing.

62showOutputable :: GHC.Outputable o => o -> String

41showOutputable :: GHC.Outputable o => o -> String

63−showOutputable = GHC.showSDoc baseDynFlags . GHC.ppr

42+showOutputable = GHC.showSDocUnsafe . GHC.ppr

64

43

65-- | Split and normalize a doc string. The result is a list of lines that

44-- | Split and normalize a doc string. The result is a list of lines that

66-- make up the comment.

45-- make up the comment.

@@ -99,57 +78,20 @@ splitDocString docStr =

99 then dropSpace <$> xs

78 then dropSpace <$> xs

100 else xs

79 else xs

101

80

102−-- | Get 'LHsType' out of 'LHsTypeArg'.

103typeArgToType :: LHsTypeArg p -> LHsType p

81typeArgToType :: LHsTypeArg p -> LHsType p

104typeArgToType = \case

82typeArgToType = \case

105 HsValArg tm -> tm

83 HsValArg tm -> tm

106 HsTypeArg _ ty -> ty

84 HsTypeArg _ ty -> ty

107 HsArgPar _ -> notImplemented "HsArgPar"

85 HsArgPar _ -> notImplemented "HsArgPar"

108

86

109−-- | Get 'RealSrcSpan' out of 'SrcSpan' if the span is “helpful”.

110unSrcSpan :: SrcSpan -> Maybe RealSrcSpan

87unSrcSpan :: SrcSpan -> Maybe RealSrcSpan

111−unSrcSpan = \case

112− RealSrcSpan r -> Just r

88+unSrcSpan (RealSrcSpan r) = Just r

113− UnhelpfulSpan _ -> Nothing

89+unSrcSpan (UnhelpfulSpan _) = Nothing

114−

115−-- | Increment line number in a 'SrcSpan'.

116−incSpanLine :: Int -> SrcSpan -> SrcSpan

117−incSpanLine i = \case

118− RealSrcSpan s ->

119− let start = realSrcSpanStart s

120− end = realSrcSpanEnd s

121− incLine x =

122− let file = srcLocFile x

123− line = srcLocLine x

124− col = srcLocCol x

125− in mkRealSrcLoc file (line + i) col

126− in RealSrcSpan (mkRealSrcSpan (incLine start) (incLine end))

127− UnhelpfulSpan x -> UnhelpfulSpan x

128

90

129−-- | Do two declarations have a blank between them?

91+-- | Do two declaration groups have a blank between them?

130−separatedByBlank :: (a -> SrcSpan) -> a -> a -> Bool

92+separatedByBlank :: (a -> SrcSpan) -> NonEmpty a -> NonEmpty a -> Bool

131separatedByBlank loc a b =

93separatedByBlank loc a b =

132 fromMaybe False $ do

94 fromMaybe False $ do

133− endA <- srcSpanEndLine <$> unSrcSpan (loc a)

95+ endA <- srcSpanEndLine <$> unSrcSpan (loc $ NE.last a)

134− startB <- srcSpanStartLine <$> unSrcSpan (loc b)

96+ startB <- srcSpanStartLine <$> unSrcSpan (loc $ NE.head b)

135 pure (startB - endA >= 2)

97 pure (startB - endA >= 2)

136−

137−-- | Do two declaration groups have a blank between them?

138−separatedByBlankNE :: (a -> SrcSpan) -> NonEmpty a -> NonEmpty a -> Bool

139−separatedByBlankNE loc a b = separatedByBlank loc (NE.last a) (NE.head b)

140−

141−-- | Return 'True' if one span ends on the same line the second one starts.

142−onTheSameLine :: SrcSpan -> SrcSpan -> Bool

143−onTheSameLine a b =

144− isOneLineSpan (mkSrcSpan (srcSpanEnd a) (srcSpanStart b))

145−

146−-- | Remove indentation from a given 'String'. Return the input with

147−-- indentation removed and the detected indentation level.

148−removeIndentation :: String -> (String, Int)

149−removeIndentation (lines -> xs) = (unlines (drop n <$> xs), n)

150− where

151− n = minimum (getIndent <$> xs)

152− getIndent y =

153− if all isSpace y

154− then 0

155− else length (takeWhile isSpace y)

tests/Ormolu/PrinterSpec.hs+7−4

@@ -1,6 +1,9 @@

1{-# LANGUAGE TemplateHaskell #-}

2

3−module Ormolu.PrinterSpec (spec) where

3+module Ormolu.PrinterSpec

4+ ( spec,

5+ )

6+where

4

7

5import Control.Exception

8import Control.Exception

6import Control.Monad

9import Control.Monad

@@ -62,7 +65,7 @@ deriveOutput path =

62-- | A version of 'shouldBe' that is specialized to comparing 'Text' values.

65-- | A version of 'shouldBe' that is specialized to comparing 'Text' values.

63-- It also prints multi-line snippets in a more readable form.

66-- It also prints multi-line snippets in a more readable form.

64shouldMatch :: Bool -> Text -> Text -> Expectation

67shouldMatch :: Bool -> Text -> Text -> Expectation

65−shouldMatch idempotenceTest actual expected =

68+shouldMatch idempotencyTest actual expected =

66 when (actual /= expected) . expectationFailure $

69 when (actual /= expected) . expectationFailure $

67 unlines

70 unlines

68 [ ">>>>>>>>>>>>>>>>>>>>>> expected (" ++ pass ++ "):",

71 [ ">>>>>>>>>>>>>>>>>>>>>> expected (" ++ pass ++ "):",

@@ -72,8 +75,8 @@ shouldMatch idempotenceTest actual expected =

72 ]

75 ]

73 where

76 where

74 pass =

77 pass =

75− if idempotenceTest

78+ if idempotencyTest

76− then "idempotence pass"

79+ then "idempotency pass"

77 else "first pass"

80 else "first pass"

78

81

79examplesDir :: Path Rel Dir

82examplesDir :: Path Rel Dir