Fix record update formatting (#2315)

* Closes #2287

src/Juvix/Compiler/Concrete/Print/Base.hs

@@ -201,7 +201,19 @@ instance (SingI s) => PrettyPrint (PatternAtoms s) where
   ppCode (PatternAtoms ps _) = hsep (ppCode <$> ps)
 
 instance (SingI s) => PrettyPrint (ExpressionAtoms s) where
-  ppCode as = hsep (ppCode <$> as ^. expressionAtoms)
+  ppCode ::
+    forall r.
+    (Members '[ExactPrint, Reader Options] r) =>
+    ExpressionAtoms s ->
+    Sem r ()
+  ppCode as = go (toList $ as ^. expressionAtoms)
+    where
+      go :: [ExpressionAtom s] -> Sem r ()
+      go = \case
+        [] -> return ()
+        [x] -> ppCode x
+        (x : xs@(AtomRecordUpdate {} : _)) -> ppCode x >> go xs
+        (x : xs) -> ppCode x >> space >> go xs
 
 instance (SingI s) => PrettyPrint (Initializer s) where
   ppCode Initializer {..} = do

src/Juvix/Data/Ape/Print.hs

@@ -72,11 +72,16 @@ ppUChain (UChain opFix f links) = do
   pp <- asks (^. apePP)
   let f' = ppLinkExpr fx f
       args = hsep (fmap pp links)
-  f' <+> args
+  f' <> sp <> args
   where
     fx :: Precedence
     fx = opFix ^. fixityPrecedence
 
+    sp :: Sem r ()
+    sp = case fx of
+      PrecUpdate -> return ()
+      _ -> space
+
 ppLinkExpr ::
   (Members '[Reader (ApeParams a), ExactPrint] r) => Precedence -> Cape a -> Sem r ()
 ppLinkExpr opFix e = parensIf cond (ppCape e)

tests/Compilation/positive/test060.juvix

@@ -17,20 +17,19 @@ type Pair (A B : Type) :=
     };
 
 mf : Pair (Pair Bool (List Nat)) (List Nat) → Bool
-  | mkPair@{fst := mkPair@{fst; snd := nil}; snd := zero :: _} := fst
+  | mkPair@{fst := mkPair@{fst; snd := nil};
+           snd := zero :: _} := fst
   | _ := false;
 
 main : Triple Nat Nat Nat :=
   let
     p : Triple Nat Nat Nat := mkTriple 2 2 2;
-    p' :
+    p' : Triple Nat Nat Nat :=
-      Triple Nat Nat Nat :=
       p@Triple{
         fst := fst + 1;
         snd := snd * 3 + thd + fst
       };
-    f :
+    f : Triple Nat Nat Nat -> Triple Nat Nat Nat :=
-      Triple Nat Nat Nat -> Triple Nat Nat Nat :=
       (@Triple{fst := fst * 10});
   in if
     (mf (mkPair (fst := mkPair true nil; snd := 0 :: nil)))