grok reading a field from a struct or ptr.

1 files changed, 13 insertions, 0 deletions

diff --git a/monkeypatch.hs b/monkeypatch.hs
index 5228dce..8e2dcf8 100644
--- a/monkeypatch.hs
+++ b/monkeypatch.hs
@@ -431,9 +431,22 @@ grokExpression fe (CAssign CAssignOp cvar expr _) = do
                     $ Lambda () [hspvar v] (hsvar k)
             }
     return $ (,) (ss ++ [s]) $ mkcomp (hsvar v)
+grokExpression fe (CMember cvar fld isptr _) = do
+    v <- cvarName cvar
+    let fieldlbl = identToString fld
+        hv = v ++ fieldlbl
+        e = App () (App () (hsvar "get")
+                            (TypeApp () (TyPromoted () (PromotedString () fieldlbl fieldlbl))))
+                   (hsvar v)
+        e' = (mkcomp e){ compFree = Map.singleton v () }
+        k = uniqIdentifier "go" (varmap [hv,v,fieldlbl])
+        s = (FormalLambda k <$> fmap (($ Lambda () [hspvar hv] (hsvar k)) . (`infixOp` ">>=")) e')
+                { compIntro = Map.singleton hv () }
+    return $ (,) [s] (mkcomp $ hsvar hv){ compFree = Map.singleton hv () }
 grokExpression fe _ = Nothing
 
 
+
 grokInitialization :: Foldable t1 =>
                       FunctionEnvironment
                       -> t1 (CDeclarationSpecifier t2)