Skip to content

Commit d8a8ba2

Browse files
authored
Support NamedFieldPuns (#84)
1 parent 698449d commit d8a8ba2

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

examples/34-field-puns.hell

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
data Foo = Foo { bar, mu :: Int }
2+
main = do
3+
let bar = 123
4+
let mu = 666
5+
let r = Main.Foo{bar,mu}
6+
IO.print $ Record.get @"bar" @Int r

src/Hell.hs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -391,6 +391,7 @@ makeConstructRecord qname fields =
391391
$ map
392392
( \case
393393
HSE.FieldUpdate _ (HSE.UnQual _ (HSE.Ident _ i)) expr -> (i, expr)
394+
HSE.FieldPun _ v@(HSE.UnQual _ (HSE.Ident l' i)) -> (i, HSE.Var l' v)
394395
f -> error $ "Invalid field: " ++ show f
395396
)
396397
fields
@@ -2112,7 +2113,7 @@ data File = File {
21122113
parseFile :: String -> IO (Either String File)
21132114
parseFile filePath = do
21142115
string <- ByteString.readFile filePath
2115-
pure $ case HSE.parseModuleWithMode HSE.defaultParseMode {HSE.parseFilename = filePath, HSE.extensions = HSE.extensions HSE.defaultParseMode ++ [HSE.EnableExtension HSE.PatternSignatures, HSE.EnableExtension HSE.DataKinds, HSE.EnableExtension HSE.BlockArguments, HSE.EnableExtension HSE.TypeApplications]} (Text.unpack (dropShebang (Text.decodeUtf8 string))) >>= parseModule of
2116+
pure $ case HSE.parseModuleWithMode HSE.defaultParseMode {HSE.parseFilename = filePath, HSE.extensions = HSE.extensions HSE.defaultParseMode ++ [HSE.EnableExtension HSE.PatternSignatures, HSE.EnableExtension HSE.DataKinds, HSE.EnableExtension HSE.BlockArguments, HSE.EnableExtension HSE.TypeApplications, HSE.EnableExtension HSE.NamedFieldPuns]} (Text.unpack (dropShebang (Text.decodeUtf8 string))) >>= parseModule of
21162117
HSE.ParseFailed l e -> Left $ "Parse error: " <> HSE.prettyPrint l <> ": " <> e
21172118
HSE.ParseOk file -> Right file
21182119

0 commit comments

Comments
 (0)