{-# LANGUAGE CPP #-}
{-# LANGUAGE OverloadedStrings #-}
module Main where

import System.Environment
import Data.Char
import Data.List
import qualified Data.ByteString.Lazy as B

#ifdef DEBUG
import Text.Printf

import JVM.Dump
#endif
import JVM.ClassFile

import Mate.BasicBlocks
import Mate.X86CodeGen
import Mate.MethodPool
import Mate.Types
import Mate.ClassPool

main ::  IO ()
main = do
  args <- getArgs
  register_signal
  initMethodPool
  case args of
    [clspath] -> do
      let bclspath = B.pack $ map (fromIntegral . ord) clspath
      cls <- getClassFile bclspath
#ifdef DEBUG
      dumpClass cls
#endif
      hmap <- parseMethod cls "main"
      case hmap of
        Just hmap' -> do
          let methods = classMethods cls; methods :: [Method Resolved]
          let method = find (\x -> (methodName x) == "main") methods
          case method of
            Just m -> do
              let mi = (MethodInfo "main" bclspath (methodSignature m))
              entry <- compileBB hmap' mi
              addMethodRef entry mi [bclspath]
#ifdef DEBUG
              printf "executing `main' now:\n"
#endif
              executeFuncPtr entry
            Nothing -> error "main not found"
        Nothing -> error "main not found"
    _ -> error "Usage: mate <class-file>"