Recognise "FOURMOLU_ENABLE" etc.

georgefst · georgefst · commit a15b22977f4b · 2020-09-24T23:27:39.000+01:00
diff --git a/README.md b/README.md
@@ -76,22 +76,24 @@ Fourmolu can be integrated with your editor via the [Haskell Language Server](ht
 Fourmolu understands two magic comments:
 
 ```haskell
-{- ORMOLU_DISABLE -}
+{- FOURMOLU_DISABLE -}
 ```
 
 and
 
 ```haskell
-{- ORMOLU_ENABLE -}
+{- FOURMOLU_ENABLE -}
 ```
 
 This allows us to disable formatting selectively for code between these
 markers or disable it for the entire file. To achieve the latter, just put
-`{- ORMOLU_DISABLE -}` at the very top. Note that the source code should
+`{- FOURMOLU_DISABLE -}` at the very top. Note that the source code should
 still be parseable even without the “excluded” part. Because of that the
 magic comments cannot be placed arbitrary, but should rather enclose
 independent top-level definitions.
 
+`{- ORMOLU_DISABLE -}` and `{- ORMOLU_ENABLE -}`, respectively, can be used to the same effect.
+
 ## Current limitations
 
 * CPP support is experimental. CPP is virtually impossible to handle
diff --git a/src/Ormolu/Processing/Preprocess.hs b/src/Ormolu/Processing/Preprocess.hs
@@ -127,11 +127,11 @@ disableMarker = "{- ORMOLU_DISABLE -}"
 
 -- | Return 'True' if the given string is an enabling marker.
 isOrmoluEnable :: String -> Bool
-isOrmoluEnable = magicComment "ORMOLU_ENABLE"
+isOrmoluEnable s = magicComment "ORMOLU_ENABLE" s || magicComment "FOURMOLU_ENABLE" s
 
 -- | Return 'True' if the given string is a disabling marker.
 isOrmoluDisable :: String -> Bool
-isOrmoluDisable = magicComment "ORMOLU_DISABLE"
+isOrmoluDisable s = magicComment "ORMOLU_DISABLE" s || magicComment "FOURMOLU_DISABLE" s
 
 -- | Construct a function for whitespace-insensitive matching of string.
 magicComment ::
