@@ -1,10 +1,5 @@
11{-# LANGUAGE PatternGuards #-}
22
3−{-
4− BigInteger Javascript code taken from:
5− https://github.com/peterolson/BigInteger.js
6−-}
7−
83module IRTS.CodegenJavaScript (codegenJavaScript) where
94
105import Idris.AbsSyntax
@@ -19,11 +14,10 @@ import Util.System
1914import Control.Arrow
2015import Data.Char
2116import Data.List
17+import qualified Data.Map as Map
2218import System.IO
2319
24−type NamespaceName = String
25−
26−idrNamespace :: NamespaceName
20+idrNamespace :: String
2721idrNamespace = "__IDR__"
2822
2923codegenJavaScript
@@ -31,10 +25,14 @@ codegenJavaScript
3125 -> FilePath
3226 -> OutputType
3327 -> IO ()
34−codegenJavaScript definitions filename outputType =
35− writeFile filename output
28+codegenJavaScript definitions filename outputType = do
29+ path <- getDataDir
30+ idrRuntime <- readFile (path ++ "/js/Runtime.js")
31+ writeFile filename (idrRuntime ++ modules ++ functions ++ mainLoop)
3632 where
3733 def = map (first translateNamespace) definitions
34+
35+ functions = concatMap translateDeclaration def
3836
3937 mainLoop :: String
4038 mainLoop = intercalate "\n" [ "\nfunction main() {"
@@ -42,80 +40,14 @@ codegenJavaScript definitions filename outputType =
4240 , "}\n\nmain();\n"
4341 ]
4442
45− output :: String
46− output = concat [ idrRuntime
47− , concatMap (translateModule Nothing) def
48− , mainLoop
49− ]
50−
51−idrRuntime :: String
52−idrRuntime =
53− createModule Nothing idrNamespace $ concat
54− [ "__IDR__.Type = function(type) { this.type = type; };"
55− , "__IDR__.Int = new __IDR__.Type('Int');"
56− , "__IDR__.Char = new __IDR__.Type('Char');"
57− , "__IDR__.String = new __IDR__.Type('String');"
58− , "__IDR__.Integer = new __IDR__.Type('Integer');"
59− , "__IDR__.Float = new __IDR__.Type('Float');"
60− , "__IDR__.Pointer = new __IDR__.Type('Pointer');"
61− , "__IDR__.Forgot = new __IDR__.Type('Forgot');"
62−
63− , "__IDR__.bigInt=function(){var e=1e7,t=7,n={positive:!1,negative:!0},r=function(e,t){var n=e.value,r=t.value,i=n.length>r.length?n.length:r.length;for(var s=0;s<i;s++)n[s]=n[s]||0,r[s]=r[s]||0;for(var s=i-1;s>=0;s--){if(n[s]!==0||r[s]!==0)break;n.pop(),r.pop()}n.length||(n=[0],r=[0]),e.value=n,t.value=r},i=function(e,s){if(typeof e=='object')return e;e+='';var u=n.positive,a=[];e[0]==='-'&&(u=n.negative,e=e.slice(1));var e=e.split('e');if(e.length>2)throw new Error('Invalid integer');if(e[1]){var f=e[1];f[0]==='+'&&(f=f.slice(1)),f=i(f);if(f.lesser(0))throw new Error('Cannot include negative exponent part for integers');while(f.notEquals(0))e[0]+='0',f=f.prev()}e=e[0],e==='-0'&&(e='0');var l=/^([1-9][0-9]*)$|^0$/.test(e);if(!l)throw new Error('Invalid integer');while(e.length){var c=e.length>t?e.length-t:0;a.push(+e.slice(c)),e=e.slice(0,c)}var h=o(a,u);return s&&r(s,h),h},s=function(e,t){var e=o(e,n.positive),t=o(t,n.positive);if(e.equals(0))throw new Error('Cannot divide by 0');var r=0;do{var i=1,s=o(e.value,n.positive),u=s.times(10);while(u.lesser(t))s=u,i*=10,u=u.times(10);while(s.lesserOrEquals(t))t=t.minus(s),r+=i}while(e.lesserOrEquals(t));return{remainder:t.value,result:r}},o=function(f,l){var c={value:f,sign:l},h={value:f,sign:l,negate:function(e){var t=e||c;return o(t.value,!t.sign)},abs:function(e){var t=e||c;return o(t.value,n.positive)},add:function(t,s){var u,a=c,f;s?(a=i(t))&&(f=i(s)):f=i(t,a),u=a.sign;if(a.sign!==f.sign)return a=o(a.value,n.positive),f=o(f.value,n.positive),u===n.positive?h.subtract(a,f):h.subtract(f,a);r(a,f);var l=a.value,p=f.value,d=[],v=0;for(var m=0;m<l.length||v>0;m++){var g=l[m]+p[m]+v;v=g>e?1:0,g-=v*e,d.push(g)}return o(d,u)},plus:function(e,t){return h.add(e,t)},subtract:function(t,r){var s=c,u;r?(s=i(t))&&(u=i(r)):u=i(t,s);if(s.sign!==u.sign)return h.add(s,h.negate(u));if(s.sign===n.negative)return h.subtract(h.negate(u),h.negate(s));if(h.compare(s,u)===-1)return h.negate(h.subtract(u,s));var a=s.value,f=u.value,l=[],p=0;for(var d=0;d<a.length;d++){a[d]-=p,p=a[d]<f[d]?1:0;var v=p*e+a[d]-f[d];l.push(v)}return o(l,n.positive)},minus:function(e,t){return h.subtract(e,t)},multiply:function(t,n){var r,s=c,u;n?(s=i(t))&&(u=i(n)):u=i(t,s),r=s.sign!==u.sign;var a=s.value,f=u.value,l=[];for(var h=0;h<a.length;h++){l[h]=[];var p=h;while(p--)l[h].push(0)}var d=0;for(var h=0;h<a.length;h++){var v=a[h];for(var p=0;p<f.length||d>0;p++){var m=f[p],g=m?v*m+d:d;d=g>e?Math.floor(g/e):0,g-=d*e,l[h].push(g)}}var y=-1;for(var h=0;h<l.length;h++){var b=l[h].length;b>y&&(y=b)}var w=[],d=0;for(var h=0;h<y||d>0;h++){var E=d;for(var p=0;p<l.length;p++)E+=l[p][h]||0;d=E>e?Math.floor(E/e):0,E-=d*e,w.push(E)}return o(w,r)},times:function(e,t){return h.multiply(e,t)},divmod:function(e,t){var r,u=c,a;t?(u=i(e))&&(a=i(t)):a=i(e,u),r=u.sign!==a.sign;if(o(u.value,u.sign).equals(0))return{quotient:o([0],n.positive),remainder:o([0],n.positive)};if(a.equals(0))throw new Error('Cannot divide by zero');var f=u.value,l=a.value,h=[],p=[];for(var d=f.length-1;d>=0;d--){var e=[f[d]].concat(p),v=s(l,e);h.push(v.result),p=v.remainder}return h.reverse(),{quotient:o(h,r),remainder:o(p,u.sign)}},divide:function(e,t){return h.divmod(e,t).quotient},over:function(e,t){return h.divide(e,t)},mod:function(e,t){return h.divmod(e,t).remainder},pow:function(e,t){var n=c,r;t?(n=i(e))&&(r=i(t)):r=i(e,n);var s=n,f=r;if(f.lesser(0))return u;if(f.equals(0))return a;var l=o(s.value,s.sign);if(f.mod(2).equals(0)){var h=l.pow(f.over(2));return h.times(h)}return l.times(l.pow(f.minus(1)))},next:function(e){var t=e||c;return h.add(t,1)},prev:function(e){var t=e||c;return h.subtract(t,1)},compare:function(e,t){var s=c,o;t?(s=i(e))&&(o=i(t,s)):o=i(e,s),r(s,o);if(s.value.length===1&&o.value.length===1&&s.value[0]===0&&o.value[0]===0)return 0;if(o.sign!==s.sign)return s.sign===n.positive?1:-1;var u=s.sign===n.positive?1:-1,a=s.value,f=o.value;for(var l=a.length-1;l>=0;l--){if(a[l]>f[l])return 1*u;if(f[l]>a[l])return-1*u}return 0},compareAbs:function(e,t){var r=c,s;return t?(r=i(e))&&(s=i(t,r)):s=i(e,r),r.sign=s.sign=n.positive,h.compare(r,s)},equals:function(e,t){return h.compare(e,t)===0},notEquals:function(e,t){return!h.equals(e,t)},lesser:function(e,t){return h.compare(e,t)<0},greater:function(e,t){return h.compare(e,t)>0},greaterOrEquals:function(e,t){return h.compare(e,t)>=0},lesserOrEquals:function(e,t){return h.compare(e,t)<=0},isPositive:function(e){var t=e||c;return t.sign===n.positive},isNegative:function(e){var t=e||c;return t.sign===n.negative},isEven:function(e){var t=e||c;return t.value[0]%2===0},isOdd:function(e){var t=e||c;return t.value[0]%2===1},toString:function(r){var i=r||c,s='',o=i.value.length;while(o--)s+=(e.toString()+i.value[o]).slice(-t);while(s[0]==='0')s=s.slice(1);s.length||(s='0');var u=i.sign===n.positive?'':'-';return u+s},toJSNumber:function(e){return+h.toString(e)},valueOf:function(e){return h.toJSNumber(e)}};return h},u=o([0],n.positive),a=o([1],n.positive),f=o([1],n.negative),l=function(e){return typeof e=='undefined'?u:i(e)};return l.zero=u,l.one=a,l.minusOne=f,l}();typeof module!='undefined'&&(module.exports=__IDR__.bigInt);"
64−
65− , "__IDR__.Tailcall = function(f) { this.f = f };"
66−
67− , "__IDR__.Con = function(i,name,vars)"
68− , "{this.i = i;this.name = name;this.vars = vars;};\n"
69−
70− , "__IDR__.tailcall = function(f){\n"
71− ++ "var __f = f;\n"
72− ++ "while (__f) {\n"
73− ++ "var f = __f;\n"
74− ++ "__f = null;\n"
75− ++ "var ret = f();\n"
76− ++ "if (ret instanceof __IDR__.Tailcall) {\n"
77− ++ "__f = ret.f;"
78− ++ "\n} else {\n"
79− ++ "return ret;"
80− ++ "\n}"
81− ++ "\n}"
82− ++ "\n};\n"
83−
84− , "var newline_regex =/(.*)\\n$/;\n"
85−
86− , "__IDR__.print = function(s){\n"
87− ++ "var m = s.match(newline_regex);\n"
88− ++ "console.log(m ? m[1] : s);"
89− ++ "\n};\n"
90− ]
91−
92−createModule :: Maybe String -> NamespaceName -> String -> String
93−createModule toplevel modname body =
94− concat [header modname, body, footer modname]
95− where
96− header :: NamespaceName -> String
97− header modname =
98− concatMap (++ "\n")
99− [ "\nvar " ++ modname ++ ";"
100− , "(function(" ++ modname ++ "){"
101− ]
102−
103− footer :: NamespaceName -> String
104− footer modname =
105− let m = maybe "" (++ ".") toplevel ++ modname in
106− "\n})("
107− ++ m
108− ++ " || ("
109− ++ m
110− ++ " = {})"
111− ++ ");\n"
112−
113−translateModule :: Maybe String -> ([String], SDecl) -> String
114−translateModule toplevel ([modname], decl) =
115− let body = translateDeclaration modname decl in
116− createModule toplevel modname body
117−translateModule toplevel (n:ns, decl) =
118− createModule toplevel n $ translateModule (Just n) (ns, decl)
43+ modules =
44+ concatMap allocMod mods
45+ where
46+ allocMod m = intercalate "." m ++ " = {};\n"
47+ sortMods a b = compare (length a) (length b)
48+
49+ mods =
50+ drop 1 $ sortBy sortMods $ nub $ concatMap (inits . fst) def
11951
12052translateIdentifier :: String -> String
12153translateIdentifier =
@@ -163,24 +95,19 @@ translateName (UN name) = translateIdentifier name
16395translateName (NS name _) = translateName name
16496translateName (MN i name) = translateIdentifier name ++ show i
16597
166−translateQualifiedName :: Name -> String
167−translateQualifiedName name =
168− intercalate "." (translateNamespace name) ++ "." ++ translateName name
169−
17098translateConstant :: Const -> String
171−translateConstant (I i) = show i
172−translateConstant (BI i) = "__IDR__.bigInt('" ++ show i ++ "')"
173−translateConstant (Fl f) = show f
174−translateConstant (Ch c) = show c
175−translateConstant (Str s) = show s
176−translateConstant IType = "__IDR__.Int"
177−translateConstant ChType = "__IDR__.Char"
178−translateConstant StrType = "__IDR__.String"
179−translateConstant BIType = "__IDR__.Integer"
180−translateConstant FlType = "__IDR__.Float"
181−translateConstant PtrType = "__IDR__.Pointer"
182−translateConstant Forgot = "__IDR__.Forgot"
183−translateConstant c =
99+translateConstant (I i) = show i
100+translateConstant (BI i) = "__IDRRT__.bigInt('" ++ show i ++ "')"
101+translateConstant (Fl f) = show f
102+translateConstant (Ch c) = show c
103+translateConstant (Str s) = show s
104+translateConstant IType = "__IDRRT__.Int"
105+translateConstant ChType = "__IDRRT__.Char"
106+translateConstant StrType = "__IDRRT__.String"
107+translateConstant BIType = "__IDRRT__.Integer"
108+translateConstant FlType = "__IDRRT__.Float"
109+translateConstant Forgot = "__IDRRT__.Forgot"
110+translateConstant c =
184111 "(function(){throw 'Unimplemented Const: " ++ show c ++ "';})()"
185112
186113translateParameterlist =
@@ -188,18 +115,16 @@ translateParameterlist =
188115 where translateParameter (MN i name) = name ++ show i
189116 translateParameter (UN name) = name
190117
191−translateDeclaration :: NamespaceName -> SDecl -> String
192−translateDeclaration modname (SFun name params stackSize body) =
193− modname
194− ++ "."
195− ++ translateName name
118+translateDeclaration :: ([String], SDecl) -> String
119+translateDeclaration (path, SFun name params stackSize body) =
120+ intercalate "." (path ++ [translateName name])
196121 ++ " = function("
197122 ++ intercalate "," p
198123 ++ "){\n"
199124 ++ concatMap assignVar (zip [0..] p)
200125 ++ concatMap allocVar [numP..(numP+stackSize-1)]
201126 ++ "return "
202− ++ translateExpression modname body
127+ ++ translateExpression body
203128 ++ ";\n};\n"
204129 where
205130 numP :: Int
@@ -218,32 +143,32 @@ translateVariableName :: LVar -> String
218143translateVariableName (Loc i) =
219144 "__var_" ++ show i
220145
221−translateExpression :: NamespaceName -> SExp -> String
222−translateExpression modname (SLet name value body) =
146+translateExpression :: SExp -> String
147+translateExpression (SLet name value body) =
223148 "(function("
224149 ++ translateVariableName name
225150 ++ "){\nreturn "
226− ++ translateExpression modname body
151+ ++ translateExpression body
227152 ++ ";\n})("
228− ++ translateExpression modname value
153+ ++ translateExpression value
229154 ++ ")"
230155
231−translateExpression _ (SConst cst) =
156+translateExpression (SConst cst) =
232157 translateConstant cst
233158
234−translateExpression _ (SV var) =
159+translateExpression (SV var) =
235160 translateVariableName var
236161
237−translateExpression modname (SApp False name vars) =
162+translateExpression (SApp False name vars) =
238163 createTailcall $ translateFunctionCall name vars
239164
240−translateExpression modname (SApp True name vars) =
241− "new __IDR__.Tailcall("
165+translateExpression (SApp True name vars) =
166+ "new __IDRRT__.Tailcall("
242167 ++ "function(){\n"
243168 ++ "return " ++ translateFunctionCall name vars
244169 ++ ";\n});"
245170
246−translateExpression _ (SOp op vars)
171+translateExpression (SOp op vars)
247172 | LPlus <- op
248173 , (lhs:rhs:_) <- vars = translateBinaryOp "+" lhs rhs
249174 | LMinus <- op
@@ -331,13 +256,13 @@ translateExpression _ (SOp op vars)
331256 | LIntStr <- op
332257 , (arg:_) <- vars = "String(" ++ translateVariableName arg ++ ")"
333258 | LIntBig <- op
334− , (arg:_) <- vars = "__IDR__.bigint(" ++ translateVariableName arg ++ ")"
259+ , (arg:_) <- vars = "__IDRRT__.bigInt(" ++ translateVariableName arg ++ ")"
335260 | LBigInt <- op
336261 , (arg:_) <- vars = translateVariableName arg ++ ".valueOf()"
337262 | LBigStr <- op
338263 , (arg:_) <- vars = translateVariableName arg ++ ".toString()"
339264 | LStrBig <- op
340− , (arg:_) <- vars = "__IDR__.bigint(" ++ translateVariableName arg ++ ")"
265+ , (arg:_) <- vars = "__IDRRT__.bigInt(" ++ translateVariableName arg ++ ")"
341266 | LFloatStr <- op
342267 , (arg:_) <- vars = "String(" ++ translateVariableName arg ++ ")"
343268 | LStrFloat <- op
@@ -396,98 +321,58 @@ translateExpression _ (SOp op vars)
396321 ++ f
397322 ++ translateVariableName rhs
398323
399−translateExpression _ (SError msg) =
324+translateExpression (SError msg) =
400325 "(function(){throw \'" ++ msg ++ "\';})();"
401326
402−translateExpression _ (SForeign _ _ "putStr" [(FString, var)]) =
403− "__IDR__.print(" ++ translateVariableName var ++ ");"
404−
405−translateExpression _ (SForeign _ _ fun args)
406− | "." `isPrefixOf` fun, "[]=" `isSuffixOf` fun
407− , (obj:idx:val:[]) <- args =
408− concat [object obj, field, index idx, assign val]
327+translateExpression (SForeign _ _ "putStr" [(FString, var)]) =
328+ "__IDRRT__.print(" ++ translateVariableName var ++ ");"
409329
410− | "." `isPrefixOf` fun, "[]" `isSuffixOf` fun
411− , (obj:idx:[]) <- args =
412− concat [object obj, field, index idx]
413−
414− | "." `isPrefixOf` fun, "=" `isSuffixOf` fun
415− , (obj:val:[]) <- args =
416− concat [object obj, field, assign val]
417−
418− | "." `isPrefixOf` fun
419− , (obj:[]) <- args =
420− object obj ++ field
421−
422− | "." `isPrefixOf` fun
423− , (obj:[(FUnit, _)]) <- args =
424− concat [object obj, method, "()"]
425−
426− | "." `isPrefixOf` fun
427− , (obj:as) <- args =
428− concat [object obj, method, arguments as]
429−
430− | "[]=" == fun
431− , (idx:val:[]) <- args =
432− concat [array, index idx, assign val]
433−
434− | "[]" == fun
435− , (idx:[]) <- args =
436− array ++ index idx
330+translateExpression (SForeign _ _ fun args) =
331+ fun
332+ ++ "("
333+ ++ intercalate "," (map (translateVariableName . snd) args)
334+ ++ ");"
437335
438− | otherwise = fun ++ arguments args
439− where
440− name = filter (`notElem` "[]=") fun
441− method = name
442− field = name
443− array = name
444− object o = translateVariableName (snd o)
445− index i = "[" ++ translateVariableName (snd i) ++ "]"
446− assign v = '=' : translateVariableName (snd v)
447− arguments as =
448− '(' : intercalate "," (map (translateVariableName . snd) as) ++ ")"
449−
450−translateExpression modname (SChkCase var cases) =
336+translateExpression (SChkCase var cases) =
451337 "(function(e){\n"
452− ++ intercalate " else " (map (translateCase modname "e") cases)
338+ ++ intercalate " else " (map (translateCase "e") cases)
453339 ++ "\n})("
454340 ++ translateVariableName var
455341 ++ ")"
456342
457−translateExpression modname (SCase var cases) =
343+translateExpression (SCase var cases) =
458344 "(function(e){\n"
459− ++ intercalate " else " (map (translateCase modname "e") cases)
345+ ++ intercalate " else " (map (translateCase "e") cases)
460346 ++ "\n})("
461347 ++ translateVariableName var
462348 ++ ")"
463349
464−translateExpression _ (SCon i name vars) =
465− concat [ "new __IDR__.Con("
350+translateExpression (SCon i name vars) =
351+ concat [ "new __IDRRT__.Con("
466352 , show i
467− , ","
468− , '\'' : translateQualifiedName name ++ "\',["
353+ , ",["
469354 , intercalate "," $ map translateVariableName vars
470355 , "])"
471356 ]
472357
473−translateExpression modname (SUpdate var e) =
474− translateVariableName var ++ " = " ++ translateExpression modname e
358+translateExpression (SUpdate var e) =
359+ translateVariableName var ++ " = " ++ translateExpression e
475360
476−translateExpression modname (SProj var i) =
361+translateExpression (SProj var i) =
477362 translateVariableName var ++ ".vars[" ++ show i ++"]"
478363
479−translateExpression _ SNothing = "null"
364+translateExpression SNothing = "null"
480365
481−translateExpression _ e =
366+translateExpression e =
482367 "(function(){throw 'Not yet implemented: "
483368 ++ filter (/= '\'') (show e)
484369 ++ "';})()"
485370
486−translateCase :: String -> String -> SAlt -> String
487−translateCase modname _ (SDefaultCase e) =
488− createIfBlock "true" (translateExpression modname e)
371+translateCase :: String -> SAlt -> String
372+translateCase _ (SDefaultCase e) =
373+ createIfBlock "true" (translateExpression e)
489374
490−translateCase modname var (SConstCase ty e)
375+translateCase var (SConstCase ty e)
491376 | ChType <- ty = matchHelper "Char"
492377 | StrType <- ty = matchHelper "String"
493378 | IType <- ty = matchHelper "Int"
@@ -496,18 +381,18 @@ translateCase modname var (SConstCase ty e)
496381 | PtrType <- ty = matchHelper "Pointer"
497382 | Forgot <- ty = matchHelper "Forgot"
498383 where
499− matchHelper tyName = translateTypeMatch modname var tyName e
384+ matchHelper tyName = translateTypeMatch var tyName e
500385
501−translateCase modname var (SConstCase cst@(BI _) e) =
386+translateCase var (SConstCase cst@(BI _) e) =
502387 let cond = var ++ ".equals(" ++ translateConstant cst ++ ")" in
503− createIfBlock cond (translateExpression modname e)
388+ createIfBlock cond (translateExpression e)
504389
505−translateCase modname var (SConstCase cst e) =
390+translateCase var (SConstCase cst e) =
506391 let cond = var ++ " == " ++ translateConstant cst in
507− createIfBlock cond (translateExpression modname e)
392+ createIfBlock cond (translateExpression e)
508393
509−translateCase modname var (SConCase a i name vars e) =
510− let isCon = var ++ " instanceof __IDR__.Con"
394+translateCase var (SConCase a i name vars e) =
395+ let isCon = var ++ " instanceof __IDRRT__.Con"
511396 isI = show i ++ " == " ++ var ++ ".i"
512397 params = intercalate "," $ map (("__var_" ++) . show) [a..(a+length vars)]
513398 args = ".apply(this," ++ var ++ ".vars)"
@@ -516,13 +401,13 @@ translateCase modname var (SConCase a i name vars e) =
516401 ++ params
517402 ++ "){\nreturn " ++ b ++ "\n})" ++ args
518403 cond = intercalate " && " [isCon, isI] in
519− createIfBlock cond $ f (translateExpression modname e)
404+ createIfBlock cond $ f (translateExpression e)
520405
521−translateTypeMatch :: String -> String -> String -> SExp -> String
522−translateTypeMatch modname var ty exp =
523− let e = translateExpression modname exp in
406+translateTypeMatch :: String -> String -> SExp -> String
407+translateTypeMatch var ty exp =
408+ let e = translateExpression exp in
524409 createIfBlock (var
525− ++ " instanceof __IDR__.Type && "
410+ ++ " instanceof __IDRRT__.Type && "
526411 ++ var ++ ".type == '"++ ty ++"'") e
527412
528413
@@ -532,7 +417,7 @@ createIfBlock cond e =
532417 ++ ";\n}"
533418
534419createTailcall call =
535− "__IDR__.tailcall(function(){return " ++ call ++ "})"
420+ "__IDRRT__.tailcall(function(){return " ++ call ++ "})"
536421
537422translateFunctionCall name vars =
538423 concat (intersperse "." $ translateNamespace name)