Multiple file support for generating assembly

1 files changed, 15 insertions, 1 deletions

diff --git a/src/Fun/Tree.hs b/src/Fun/Tree.hs
index 26b415f..817a077 100644
--- a/src/Fun/Tree.hs
+++ b/src/Fun/Tree.hs
@@ -26,10 +26,12 @@ data Block = FunctionBlock
   } -- | DataBlock .. TODO
   deriving Show
 
+type FunctionSignature = [Type]
+
 data FunctionDeclaration = FunctionDeclaration
   { dName       :: String
   , dVisibility :: Visibility
-  , dTypes      :: [Type]
+  , dTypes      :: FunctionSignature
   , dFlags      :: [FunctionFlag]
   }
   deriving Show
@@ -55,3 +57,15 @@ data FunctionBody = FunctionBody
 
 data FunctionBodyElement = Statement String | FunctionBodyIdentifier String | FunctionBodyParameter String | FunctionBodyString String | FunctionBodyNumber Integer
   deriving Show
+
+----
+
+-- TODO: This can be optimized
+getFunction :: Tree -> String -> FunctionSignature -> Maybe Block
+getFunction (Tree ps) f sig =
+  let fromProgram (Program bs) =
+        filter (\b -> (dName . bDecl) b == f && (dTypes . bDecl) b == sig) bs
+      fromTree = filter (\b -> length b == 1) (map fromProgram ps)
+  in  case fromTree of
+        [[f]] -> Just f
+        _     -> Nothing