@@ -51,39 +51,61 @@ codegenJavaScript definitions filename outputType = do
52translateIdentifier :: String -> String
52translateIdentifier :: String -> String
53translateIdentifier =
53translateIdentifier =
54− concatMap replaceBadChars
54+ replaceReserved . concatMap replaceBadChars
55 where replaceBadChars :: Char -> String
55 where replaceBadChars :: Char -> String
56− replaceBadChars ' ' = "_"
57− replaceBadChars '_' = "__"
58− replaceBadChars '@' = "_at"
59− replaceBadChars '[' = "_OSB"
60− replaceBadChars ']' = "_CSB"
61− replaceBadChars '(' = "_OP"
62− replaceBadChars ')' = "_CP"
63− replaceBadChars '{' = "_OB"
64− replaceBadChars '}' = "_CB"
65− replaceBadChars '!' = "_bang"
66− replaceBadChars '#' = "_hash"
67− replaceBadChars '.' = "_dot"
68− replaceBadChars ',' = "_comma"
69− replaceBadChars ':' = "_colon"
70− replaceBadChars '+' = "_plus"
71− replaceBadChars '-' = "_minus"
72− replaceBadChars '*' = "_times"
73− replaceBadChars '<' = "_lt"
74− replaceBadChars '>' = "_gt"
75− replaceBadChars '=' = "_eq"
76− replaceBadChars '|' = "_pipe"
77− replaceBadChars '&' = "_amp"
78− replaceBadChars '/' = "_SL"
79− replaceBadChars '\\' = "_BSL"
80− replaceBadChars '%' = "_per"
81− replaceBadChars '?' = "_que"
82− replaceBadChars '~' = "_til"
83− replaceBadChars '\'' = "_apo"
84 replaceBadChars c
56 replaceBadChars c
85 | isDigit c = "_" ++ [c] ++ "_"
59 | isDigit c = "_" ++ [c] ++ "_"
60+ | not (isLetter c && isAscii c) = '_' : show (ord c)
86 | otherwise = [c]
61 | otherwise = [c]
63+ | s `elem` reserved = '_' : s
88translateNamespace :: Name -> [String]
110translateNamespace :: Name -> [String]
89translateNamespace (UN _) = [idrNamespace]
111translateNamespace (UN _) = [idrNamespace]
@@ -112,8 +134,8 @@ translateConstant c =
113translateParameterlist =
135translateParameterlist =
114 map translateParameter
136 map translateParameter
115− where translateParameter (MN i name) = name ++ show i
137+ where translateParameter (MN i name) = translateIdentifier name ++ show i
116− translateParameter (UN name) = name
138+ translateParameter (UN name) = translateIdentifier name
118translateDeclaration :: ([String], SDecl) -> String
140translateDeclaration :: ([String], SDecl) -> String
119translateDeclaration (path, SFun name params stackSize body) =
141translateDeclaration (path, SFun name params stackSize body) =
@@ -327,11 +349,50 @@ translateExpression (SError msg) =
327translateExpression (SForeign _ _ "putStr" [(FString, var)]) =
349translateExpression (SForeign _ _ "putStr" [(FString, var)]) =
328 "__IDRRT__.print(" ++ translateVariableName var ++ ");"
350 "__IDRRT__.print(" ++ translateVariableName var ++ ");"
330−translateExpression (SForeign _ _ fun args) =
352+translateExpression (SForeign _ _ fun args)
331− fun
353+ | "." `isPrefixOf` fun, "[]=" `isSuffixOf` fun
332− ++ "("
354+ , (obj:idx:val:[]) <- args =
355+ concat [object obj, field, index idx, assign val]
357+ | "." `isPrefixOf` fun, "[]" `isSuffixOf` fun
358+ , (obj:idx:[]) <- args =
359+ concat [object obj, field, index idx]
361+ | "." `isPrefixOf` fun, "=" `isSuffixOf` fun
362+ , (obj:val:[]) <- args =
363+ concat [object obj, field, assign val]
365+ | "." `isPrefixOf` fun
366+ , (obj:[]) <- args =
369+ | "." `isPrefixOf` fun
370+ , (obj:[(FUnit, _)]) <- args =
371+ concat [object obj, method, "()"]
373+ | "." `isPrefixOf` fun
374+ , (obj:as) <- args =
375+ concat [object obj, method, arguments as]
378+ , (idx:val:[]) <- args =
379+ concat [array, index idx, assign val]
382+ , (idx:[]) <- args =
385+ | otherwise = fun ++ arguments args
387+ name = filter (`notElem` "[]=") fun
391+ object o = translateVariableName (snd o)
392+ index i = "[" ++ translateVariableName (snd i) ++ "]"
393+ assign v = '=' : translateVariableName (snd v)
333− ++ intercalate "," (map (translateVariableName . snd) args)
395+ '(' : intercalate "," (map (translateVariableName . snd) as) ++ ")"
336translateExpression (SChkCase var cases) =
397translateExpression (SChkCase var cases) =
337 "(function(e){\n"
398 "(function(e){\n"