Grok assignment to field.

1 files changed, 16 insertions, 0 deletions

diff --git a/monkeypatch.hs b/monkeypatch.hs
index e4ccbc1..461d3c1 100644
--- a/monkeypatch.hs
+++ b/monkeypatch.hs
@@ -383,6 +383,22 @@ grokStatement (CBlockStmt (CExpr (Just (C.CCall (CVar (C.Ident "__assert_fail" _
         x' = fmap (\y -> App () (hsvar "error") y) x
     return $ fmap (\y -> Lambda () [hspvar k] y) x'
 grokStatement (CBlockStmt (CExpr (Just
+                (CAssign CAssignOp
+                   (CMember cvar fld isptr _) expr _)) _)) = do
+    (xs,x) <- grokExpression expr
+    v <- cvarName cvar
+    let fieldlbl = identToString fld
+        k1 = uniqIdentifier "go" (compFree x)
+        fieldinit = comp x
+        x' = x
+            { comp = infixOp
+                            (App () (App () (App () (hsvar "set")
+                                               (TypeApp () (TyPromoted () (PromotedString () fieldlbl fieldlbl))))
+                                            (hsvar v))
+                                    fieldinit) ">>" (hsvar k1)
+            }
+    return $ fmap (\y -> Lambda () [hspvar k1] y) $ foldr applyComputation x' xs
+grokStatement (CBlockStmt (CExpr (Just
                 (CAssign CAssignOp cvarnew
                     (C.CCall cvarfun [] _) _)) _)) = do
     v <- cvarName cvarnew