copilot-prettyprinter: Pretty-print struct update expressions. Refs C…

…opilot-Language#526. This commit adds pretty-printing of expressions that update fields of a struct.
GaloisInc · Aug 21, 2024 · 905cfe5 · 905cfe5
1 parent 6034e6d
commit 905cfe5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/copilot-prettyprinter/src/Copilot/PrettyPrint.hs b/copilot-prettyprinter/src/Copilot/PrettyPrint.hs
@@ -100,6 +100,11 @@ ppOp2 op = case op of
   BwShiftL _ _ -> ppInfix "<<"
   BwShiftR _ _ -> ppInfix ">>"
   Index    _   -> ppInfix ".!!"
+  UpdateField (Struct _) _ f -> \ doc1 doc2 ->
+    text "(UpdateStruct" <+> doc1 <+>
+    text "at" <+> text (accessorName f) <+>
+    text "to" <+> doc2 <> text ")"
+  UpdateField _ _ _ -> impossible "ppOp2" "Copilot.PrettyPrint"
 
 -- | Pretty-print a ternary operation.
 ppOp3 :: Op3 a b c d -> Doc -> Doc -> Doc -> Doc