X-Git-Url: https://git.lukelau.me/?p=kaleidoscope-hs.git;a=blobdiff_plain;f=Main.hs;h=816692b758ceb7dce309611947d84935df97e6e8;hp=76a9bdb5d48831da2e59c811ecbcbb5a379bdfb5;hb=38600a47841e2c5837b5693b32c35d2093864458;hpb=30a26b7d2b0e17ea523ee34cb5d37242a38882df diff --git a/Main.hs b/Main.hs index 76a9bdb..816692b 100644 --- a/Main.hs +++ b/Main.hs @@ -1 +1,35 @@ -main = pure () +{-# LANGUAGE OverloadedStrings #-} + +import AST +import Utils +import Control.Monad.IO.Class +import qualified Data.Text.Lazy.IO as Text +import LLVM.AST.Constant +import LLVM.AST.Float +import LLVM.AST.Operand +import LLVM.AST.Type as Type +import LLVM.IRBuilder +import LLVM.Pretty +import System.IO +import Text.Read (readMaybe) + +main = buildModuleT "main" repl + +repl :: ModuleBuilderT IO () +repl = do + liftIO $ hPutStr stderr "ready> " + ast <- liftIO $ readMaybe <$> getLine + case ast of + Nothing -> liftIO $ hPutStrLn stderr "Couldn't parse" + Just x -> do + hoist $ buildAST x + mostRecentDef >>= liftIO . Text.hPutStrLn stderr . ppll + repl + where + +buildAST :: AST -> ModuleBuilder Operand +buildAST (TopLevelExpr x) = function "__anon_expr" [] Type.double $ + const $ buildExpr x >>= ret + +buildExpr :: Expr -> IRBuilderT ModuleBuilder Operand +buildExpr (Num x) = pure $ ConstantOperand (Float (Double x))