[scoper] Add function clauses to scoped InfoTable

src/MiniJuvix/Syntax/Concrete/Scoped/InfoTable.hs

@@ -3,6 +3,7 @@
 module MiniJuvix.Syntax.Concrete.Scoped.InfoTable where
 import MiniJuvix.Prelude
 import MiniJuvix.Syntax.Concrete.Language
+import qualified MiniJuvix.Syntax.Concrete.Scoped.Name as S
 
 newtype FunctionInfo = FunctionInfo {
   _functionInfoType :: Expression
@@ -25,7 +26,8 @@ data InfoTable = InfoTable {
   _infoConstructors :: HashMap ConstructorRef ConstructorInfo,
   _infoAxioms :: HashMap AxiomRef AxiomInfo,
   _infoInductives :: HashMap InductiveRef InductiveInfo,
-  _infoFunctions :: HashMap FunctionRef FunctionInfo
+  _infoFunctions :: HashMap FunctionRef FunctionInfo,
+  _infoFunctionClauses :: HashMap S.Symbol (FunctionClause 'Scoped)
   }
 
 emptyInfoTable :: InfoTable

src/MiniJuvix/Syntax/Concrete/Scoped/Scoper.hs

@@ -720,13 +720,12 @@ checkFunctionClause clause@FunctionClause {..} = do
       clb <- checkParseExpressionAtoms _clauseBody
       put s
       return (clp, clw, clb)
-  return
-    FunctionClause
-      { _clauseOwnerFunction = clauseOwnerFunction',
-        _clausePatterns = clausePatterns',
-        _clauseBody = clauseBody',
-        _clauseWhere = clauseWhere'
-      }
+  registerFunctionClause' FunctionClause
+    { _clauseOwnerFunction = clauseOwnerFunction',
+      _clausePatterns = clausePatterns',
+      _clauseBody = clauseBody',
+      _clauseWhere = clauseWhere'
+    }
   where
     fun = _clauseOwnerFunction
     checkTypeSigInScope :: Sem r S.Symbol

src/MiniJuvix/Syntax/Concrete/Scoped/Scoper/InfoTableBuilder.hs

@@ -11,6 +11,7 @@ data InfoTableBuilder m a where
   RegisterConstructor :: InductiveConstructorDef 'Scoped -> InfoTableBuilder m ()
   RegisterInductive :: InductiveDef 'Scoped -> InfoTableBuilder m ()
   RegisterFunction :: TypeSignature 'Scoped -> InfoTableBuilder m ()
+  RegisterFunctionClause :: FunctionClause 'Scoped -> InfoTableBuilder m ()
 
 makeSem ''InfoTableBuilder
 
@@ -31,6 +32,11 @@ registerAxiom' :: Member InfoTableBuilder r =>
   AxiomDef 'Scoped -> Sem r (AxiomDef 'Scoped)
 registerAxiom' a = registerAxiom a $> a
 
+registerFunctionClause' :: Member InfoTableBuilder r =>
+  FunctionClause 'Scoped -> Sem r (FunctionClause 'Scoped)
+registerFunctionClause' a = registerFunctionClause a $> a
+
+
 toState :: Sem (InfoTableBuilder ': r) a -> Sem (State InfoTable ': r) a
 toState = reinterpret $ \case
   RegisterAxiom d ->
@@ -61,6 +67,10 @@ toState = reinterpret $ \case
       }
       in modify (over infoFunctions  (HashMap.insert ref info)
    )
+  RegisterFunctionClause c -> let
+    key = c ^. clauseOwnerFunction
+    value = c
+    in modify (over infoFunctionClauses (HashMap.insert key value))
 
 runInfoTableBuilder :: Sem (InfoTableBuilder ': r) a -> Sem r (InfoTable, a)
 runInfoTableBuilder = runState emptyInfoTable . toState