@@ -224,7 +224,7 @@ codegenJavaScript target definitions filename outputType = do
224 ("#!/usr/bin/env node\n", "-node")
224 ("#!/usr/bin/env node\n", "-node")
225 JavaScript ->
225 JavaScript ->
226 ("", "-browser")
226 ("", "-browser")
227− path <- getDataDir
227+ path <- getDataDir
228 idrRuntime <- readFile $ path ++ "/js/Runtime-common.js"
228 idrRuntime <- readFile $ path ++ "/js/Runtime-common.js"
229 tgtRuntime <- readFile $ concat [path, "/js/Runtime", runtime, ".js"]
229 tgtRuntime <- readFile $ concat [path, "/js/Runtime", runtime, ".js"]
230 writeFile filename $ intercalate "\n" $ [ header
230 writeFile filename $ intercalate "\n" $ [ header
@@ -246,19 +246,22 @@ codegenJavaScript target definitions filename outputType = do
246 mainLoop :: String
246 mainLoop :: String
247 mainLoop = compileJS $
247 mainLoop = compileJS $
248 JSSeq [ JSAlloc "main" $ Just $ JSFunction [] (
248 JSSeq [ JSAlloc "main" $ Just $ JSFunction [] (
249− jsTailcall $ jsCall "__IDR__runMain0" []
249+ jsTailcall $ jsCall mainFun []
251 , jsCall "main" []
251 , jsCall "main" []
255+ mainFun = idrNamespace ++ translateName (MN 0 "runMain")
254translateIdentifier :: String -> String
257translateIdentifier :: String -> String
255translateIdentifier =
258translateIdentifier =
256 replaceReserved . concatMap replaceBadChars
259 replaceReserved . concatMap replaceBadChars
257 where replaceBadChars :: Char -> String
260 where replaceBadChars :: Char -> String
258 replaceBadChars c
261 replaceBadChars c
259− | ' ' <- c = "_"
262+ | ' ' <- c = "_"
260− | '_' <- c = "__"
263+ | '_' <- c = "__"
261− | isDigit c = "_" ++ [c] ++ "_"
264+ | isDigit c = '_' : show (ord c)
262 | not (isLetter c && isAscii c) = '_' : show (ord c)
265 | not (isLetter c && isAscii c) = '_' : show (ord c)
263 | otherwise = [c]
266 | otherwise = [c]
264 replaceReserved s
267 replaceReserved s
@@ -317,19 +320,24 @@ translateNamespace (SN name) = idrNamespace ++ translateSpecialName name
317translateNamespace NErased = idrNamespace
320translateNamespace NErased = idrNamespace
319translateName :: Name -> String
322translateName :: Name -> String
320−translateName (UN name) = translateIdentifier name
323+translateName (UN name) = 'u' : translateIdentifier name
321−translateName (NS name _) = translateName name
324+translateName (NS name _) = 'n' : translateName name
322−translateName (MN i name) = translateIdentifier name ++ show i
325+translateName (MN i name) = 'm' : translateIdentifier name ++ show i
323−translateName (SN name) = translateSpecialName name
326+translateName (SN name) = 's' : translateSpecialName name
324−translateName NErased = ""
327+translateName NErased = "e"
326translateSpecialName :: SpecialName -> String
329translateSpecialName :: SpecialName -> String
327translateSpecialName name
330translateSpecialName name
331+ | WhereN i m n <- name =
328− | WhereN i m n <- name = translateName m ++ translateName n ++ show i
332+ 'w' : translateName m ++ translateName n ++ show i
333+ | InstanceN n s <- name =
329− | InstanceN n s <- name = translateName n ++ concatMap translateIdentifier s
334+ 'i' : translateName n ++ concatMap translateIdentifier s
335+ | ParentN n s <- name =
330− | ParentN n s <- name = translateName n ++ translateIdentifier s
336+ 'p' : translateName n ++ translateIdentifier s
337+ | MethodN n <- name =
331− | MethodN n <- name = translateName n
338+ 'm' : translateName n
332− | CaseN n <- name = translateName n
340+ 'c' : translateName n
334translateConstant :: Const -> JS
342translateConstant :: Const -> JS
335translateConstant (I i) = JSNum (JSInt i)
343translateConstant (I i) = JSNum (JSInt i)
@@ -436,7 +444,8 @@ translateDeclaration (path, SFun name params stackSize body)
436 assignVar n s = JSAlloc (jsVar n) (Just $ JSRaw s)
444 assignVar n s = JSAlloc (jsVar n) (Just $ JSRaw s)
438 assignAux :: (LVar, String) -> JS
446 assignAux :: (LVar, String) -> JS
447+ assignAux (var, val) =
439− assignAux (var, val) = JSAssign (JSRaw $ translateVariableName var) (JSRaw val)
448+ JSAssign (JSRaw $ translateVariableName var) (JSRaw val)
441 p :: [String]
450 p :: [String]
442 p = map translateName params
451 p = map translateName params
@@ -626,59 +635,9 @@ translateExpression (SError msg) =
626translateExpression (SForeign _ _ "putStr" [(FString, var)]) =
635translateExpression (SForeign _ _ "putStr" [(FString, var)]) =
627 jsCall (idrRTNamespace ++ "print") [JSVar var]
636 jsCall (idrRTNamespace ++ "print") [JSVar var]
629−translateExpression (SForeign _ _ fun args)
638+translateExpression (SForeign _ _ fun args) =
630− | "[]=" `isSuffixOf` fun
639+ ffi fun (map generateWrapper args)
631− , (obj:idx:val:[]) <- args =
632− JSRaw $ concat [object obj, index idx, assign val]
634− | "[]" `isSuffixOf` fun
635− , (obj:idx:[]) <- args =
636− JSRaw $ object obj ++ index idx
638− | "[" `isPrefixOf` fun && "]=" `isSuffixOf` fun
639− , (obj:val:[]) <- args =
640− JSRaw $ concat [object obj, '[' : name ++ "]", assign val]
642− | "[" `isPrefixOf` fun && "]" `isSuffixOf` fun
643− , (obj:[]) <- args =
644− JSRaw $ object obj ++ '[' : name ++ "]"
646− | "." `isPrefixOf` fun, "=" `isSuffixOf` fun
647− , (obj:val:[]) <- args =
648− JSRaw $ concat [object obj, field, assign val]
650− | "." `isPrefixOf` fun
651− , (obj:[]) <- args =
652− JSRaw $ object obj ++ field
654− | "." `isPrefixOf` fun
655− , (obj:[(FUnit, _)]) <- args =
656− JSRaw $ concat [object obj, method, "()"]
658− | "." `isPrefixOf` fun
659− , (obj:as) <- args =
660− JSRaw $ concat [object obj, method, arguments as]
662− | "[]=" `isSuffixOf` fun
663− , (idx:val:[]) <- args =
664− JSRaw $ concat [array, index idx, assign val]
666− | "[]" `isSuffixOf` fun
667− , (idx:[]) <- args =
668− JSRaw $ array ++ index idx
670− | otherwise = JSRaw $ fun ++ arguments args
672− name = filter (`notElem` "[]=") fun
676− object o = translateVariableName (snd o)
677− index i = "[" ++ translateVariableName (snd i) ++ "]"
678− assign v = '=' : generateWrapper v
680− '(' : intercalate "," (map generateWrapper as) ++ ")"
682 generateWrapper (ffunc, name)
641 generateWrapper (ffunc, name)
683 | FFunction <- ffunc =
642 | FFunction <- ffunc =
684 idrRTNamespace ++ "ffiWrap(" ++ translateVariableName name ++ ")"
643 idrRTNamespace ++ "ffiWrap(" ++ translateVariableName name ++ ")"
@@ -727,6 +686,37 @@ translateExpression SNothing = JSNull
727translateExpression e =
686translateExpression e =
728 JSError $ "Not yet implemented: " ++ filter (/= '\'') (show e)
687 JSError $ "Not yet implemented: " ++ filter (/= '\'') (show e)
689+data FFI = FFICode Char | FFIArg Int | FFIError String
691+ffi :: String -> [String] -> JS
692+ffi code args = let parsed = ffiParse code in
693+ case ffiError parsed of
694+ Just err -> JSError err
695+ Nothing -> JSRaw $ renderFFI parsed args
697+ ffiParse :: String -> [FFI]
699+ ffiParse ['%'] = [FFIError "Invalid positional argument"]
700+ ffiParse ('%':'%':ss) = FFICode '%' : ffiParse ss
703+ FFIArg (read $ s : takeWhile isDigit ss) : ffiParse (dropWhile isDigit ss)
705+ [FFIError "Invalid positional argument"]
706+ ffiParse (s:ss) = FFICode s : ffiParse ss
708+ ffiError :: [FFI] -> Maybe String
709+ ffiError [] = Nothing
710+ ffiError ((FFIError s):xs) = Just s
711+ ffiError (x:xs) = ffiError xs
713+ renderFFI :: [FFI] -> [String] -> String
715+ renderFFI ((FFICode c) : fs) args = c : renderFFI fs args
716+ renderFFI ((FFIArg i) : fs) args
717+ | i < length args && i >= 0 = args !! i ++ renderFFI fs args
718+ | otherwise = "Argument index out of bounds"
730data CaseType = ConCase Int
720data CaseType = ConCase Int
731 | TypeCase JSType
721 | TypeCase JSType
732 | DefaultCase
722 | DefaultCase