X-Git-Url: http://wien.tomnetworks.com/gitweb/?a=blobdiff_plain;f=Mate.hs;h=eb3c8733bab63d9261cc54f6247888e244220fbc;hb=d3f63d65d80aaab4ad8eac43ee1caea7dea09fbd;hp=95daaf8133d890e2445ef353b05072d5aeb9288e;hpb=55dbc4836a164757f92c97f98623cb203ffbed4e;p=mate.git

diff --git a/Mate.hs b/Mate.hs
index 95daaf8..eb3c873 100644
--- a/Mate.hs
+++ b/Mate.hs
@@ -6,8 +6,9 @@ module Main where
 import System.Environment
 import Data.Char
 import Data.List
-import Data.String.Utils
+import Data.List.Split
 import qualified Data.ByteString.Lazy as B
+import Control.Monad
 
 #ifdef DEBUG
 import Text.Printf
@@ -25,34 +26,49 @@ main ::  IO ()
 main = do
   args <- getArgs
   register_signal
-  addClassPath "./"
-  case args of
-    [clspath] -> do
-      let bclspath = B.pack $ map (fromIntegral . ord) clspath
+  parseArgs args False
+
+parseArgs :: [String] -> Bool -> IO ()
+parseArgs ("-jar":jarpath:_) stdcp = do
+  unless stdcp $ addClassPath "./"
+  addClassPathJAR jarpath
+  res <- readMainClass jarpath
+  case res of
+    Nothing -> error "JAR: no MainClass entry found. Try to pass the jar file via -cp instead."
+    Just mc -> do
+      let bclspath = B.pack $ map (fromIntegral . ord) mc
       cls <- getClassFile bclspath
       executeMain bclspath cls
-    ["-jar", jarpath] -> do
-      addClassPathJAR jarpath
-      res <- readMainClass jarpath
-      case res of
-        Nothing -> error "JAR: no MainClass entry found. Try to pass the jar file via -cp instead."
-        Just mc -> do
-          let mc' = replace "." "/" mc
-          let bclspath = B.pack $ map (fromIntegral . ord) mc'
-          cls <- getClassFile bclspath
-          executeMain bclspath cls
-    _ -> error "Usage: mate [<class-file> | -jar <jar-file>]"
+parseArgs ("-cp":cps) cpset = parseArgs ("-classpath":cps) cpset
+parseArgs ("-classpath":cps:xs) False = do
+  let paths = splitOn ":" cps
+  mapM_ addStuff paths
+  parseArgs xs True
+  where
+  addStuff :: String -> IO ()
+  addStuff x
+    | ".jar" `isSuffixOf` x = addClassPathJAR x
+    | otherwise = addClassPath $ x ++ "/"
+parseArgs ("-classpath":xs) _ = parseArgs ("-":xs) True -- usage
+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
+  unless stdcp $ addClassPath "./"
+  let bclspath = B.pack $ map (fromIntegral . ord) clspath
+  cls <- getClassFile bclspath
+  executeMain bclspath cls
+parseArgs _ _ = parseArgs ["-"] False
+
 
 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