Merge pull request #14 from 414owen/os/pretty-records

feat: Use vsep to make records legible
phile314 · Jul 17, 2024 · f984c5c · f984c5c
2 parents b12ae79 + a81b06e
commit f984c5c
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/Pinch/Generate/Pretty.hs b/src/Pinch/Generate/Pretty.hs
@@ -154,8 +154,15 @@ instance Pretty Deriving where
 
 instance Pretty ConDecl where
   pretty (ConDecl n args) = hsep $ [ pretty n ] ++ map pretty args
-  pretty (RecConDecl n args) = hsep $ [ pretty n, "{", fields, "}" ]
-    where fields = cList $ map (\(f, v) -> pretty f <+> "::" <+> pretty v) args
+  pretty (RecConDecl n fields) = pretty n
+    <> case fields of
+      [] -> "{}"
+      ((f, t) : xs) -> line
+        <> "{" <+> pretty f <+> "::" <+> pretty t
+          <> line
+          <> vsep (map (\(f', v) -> "," <+> pretty f' <+> "::" <+> pretty v) xs)
+          <> line
+          <> "}"
 
 instance Pretty InstHead where
   pretty (InstHead cs n ty) = "instance" <> context <+> pretty n <+> pretty ty <+> "where"