MiniJavaCompiler/src/Main.hs

module Main where

import Typecheck
import Parser.Lexer (alexScanTokens)
import Parser.JavaParser
import ByteCode.Builder
import ByteCode.ClassFile
import Data.ByteString (pack, writeFile)
import System.Environment
import System.FilePath.Posix (takeDirectory)

main = do
    args <- getArgs
    let filename = if null args
                   then error "Missing filename, I need to know what to compile"
                   else head args
    let outputDirectory = takeDirectory filename
    print ("Compiling " ++ filename)
    file <- readFile filename

    let untypedAST = parse $ alexScanTokens file
    let typedAST = typeCheckCompilationUnit untypedAST
    let assembledClasses = map (`classBuilder` emptyClassFile) typedAST

    mapM_ (\classFile -> let
            fileContent = pack (serialize classFile)
            fileName = outputDirectory ++ "/" ++ className classFile ++ ".class"
        in Data.ByteString.writeFile fileName fileContent
            ) assembledClasses