import Data.List
import Data.List.Split
import qualified Data.ByteString.Lazy as B
+import Control.Monad
#ifdef DEBUG
import Text.Printf
parseArgs :: [String] -> Bool -> IO ()
parseArgs ("-jar":jarpath:_) stdcp = do
- if not stdcp then addClassPath "./" else return ()
+ unless stdcp $ addClassPath "./"
addClassPathJAR jarpath
res <- readMainClass jarpath
case res of
parseArgs (('-':_):_) _ = error "Usage: mate [-cp|-classpath <cp1:cp2:..>] [<class-file> | -jar <jar-file>]"
-- first argument which isn't prefixed by '-' should be a class file
parseArgs (clspath:_) stdcp = do
- if not stdcp then addClassPath "./" else return ()
+ unless stdcp $ addClassPath "./"
let bclspath = B.pack $ map (fromIntegral . ord) clspath
cls <- getClassFile bclspath
executeMain bclspath cls
executeMain :: B.ByteString -> Class Direct -> IO ()
executeMain bclspath cls = do
- hmap <- parseMethod cls "main"
- case hmap of
- Just hmap' -> do
- let methods = classMethods cls; methods :: [Method Direct]
- let method = find (\x -> methodName x == "main") methods
- case method of
- Just m -> do
- let mi = MethodInfo "main" bclspath $ methodSignature m
+ let methods = classMethods cls; methods :: [Method Direct]
+ case find (\x -> methodName x == "main") methods of
+ Just m -> do
+ let mi = MethodInfo "main" bclspath $ methodSignature m
+ hmap <- parseMethod cls "main" $ methodSignature m
+ case hmap of
+ Just hmap' -> do
entry <- compileBB hmap' mi
addMethodRef entry mi [bclspath]
#ifdef DEBUG