git.iko.soy/Idris-dev
on master help sign in
add support for FFI_Export in js backend but the command line option does not reach js yet
2dd2fd1Rui Barreiro
src/IRTS/CodegenJavaScript.hs+21−11
compare with master
unifiedsplit
@@ -56,14 +56,24 @@ codegenJavaScript ci =
56codegenNode :: CodeGenerator
56codegenNode :: CodeGenerator
57codegenNode ci =
57codegenNode ci =
58 do
58 do
59+ if outputType ci == Executable then
60+ do
59− codegenJs (CGConf { header = "#!/usr/bin/env node\n"
61+ codegenJs (CGConf { header = "#!/usr/bin/env node\n"
60− , footer = ""
62+ , footer = ""
61− , jsbnPath = "jsbn/jsbn-browser.js"
63+ , jsbnPath = "jsbn/jsbn-browser.js"
62− , extraRunTime = "Runtime-node.js"
64+ , extraRunTime = "Runtime-node.js"
63− }
65+ }
64− )
66+ )
65− ci
67+ ci
66− setPermissions (outputFile ci) (emptyPermissions { readable = True
68+ setPermissions (outputFile ci) (emptyPermissions { readable = True
67− , executable = True
69+ , executable = True
68− , writable = True
70+ , writable = True
69− })
71+ })
72+ else
73+ codegenJs (CGConf { header = ""
74+ , footer = ""
75+ , jsbnPath = "jsbn/jsbn-browser.js"
76+ , extraRunTime = "Runtime-node.js"
77+ }
78+ )
79+ ci
src/IRTS/JavaScript/Codegen.hs+16−2
@@ -21,6 +21,7 @@ import IRTS.JavaScript.PrimOp
21import IRTS.JavaScript.Specialize
21import IRTS.JavaScript.Specialize
22import IRTS.Lang
22import IRTS.Lang
23import IRTS.System
23import IRTS.System
24+import IRTS.Exports
24
25
25import Control.Applicative (pure, (<$>))
26import Control.Applicative (pure, (<$>))
26import Control.Monad
27import Control.Monad
@@ -98,6 +99,14 @@ isYes (Just "Y") = True
98isYes (Just "y") = True
99isYes (Just "y") = True
99isYes _ = False
100isYes _ = False
100
101
102+makeExportDecls :: ExportIFace -> [Text]
103+makeExportDecls (Export _ _ e) =
104+ concatMap makeExport e
105+ where
106+ makeExport (ExportData _) =
107+ []
108+ makeExport (ExportFun name (FStr exportname) _ _) =
109+ [T.concat [T.pack $ exportname, ": ", jsName name]]
101
110
102codegenJs :: CGConf -> CodeGenerator
111codegenJs :: CGConf -> CodeGenerator
103codegenJs conf ci =
112codegenJs conf ci =
@@ -105,7 +114,11 @@ codegenJs conf ci =
105 debug <- isYes <$> lookupEnv "IDRISJS_DEBUG"
114 debug <- isYes <$> lookupEnv "IDRISJS_DEBUG"
106 let defs' = Map.fromList $ liftDecls ci
115 let defs' = Map.fromList $ liftDecls ci
107 let defs = globlToCon defs'
116 let defs = globlToCon defs'
117+ let isExec = outputType ci == Executable
118+ putStrLn $ show $ isExec
119+ let used = if isExec then
108− let used = Map.elems $ removeDeadCode defs [sMN 0 "runMain"]
120+ Map.elems $ removeDeadCode defs [sMN 0 "runMain"]
121+ else Map.elems $ removeDeadCode defs (getExpNames $ exportDecls ci)
109 when debug $ do
122 when debug $ do
110 writeFile (outputFile ci ++ ".LDeclsDebug") $ (unlines $ intersperse "" $ map show used) ++ "\n\n\n"
123 writeFile (outputFile ci ++ ".LDeclsDebug") $ (unlines $ intersperse "" $ map show used) ++ "\n\n\n"
111 putStrLn $ "Finished calculating used"
124 putStrLn $ "Finished calculating used"
@@ -136,7 +149,8 @@ codegenJs conf ci =
136 , doPartials (partialApplications stats), "\n"
149 , doPartials (partialApplications stats), "\n"
137 , doHiddenClasses (hiddenClasses stats), "\n"
150 , doHiddenClasses (hiddenClasses stats), "\n"
138 , out, "\n"
151 , out, "\n"
139− , jsName (sMN 0 "runMain"), "();\n"
152+ , if isExec then jsName (sMN 0 "runMain") `T.append` "();\n"
153+ else T.concat ["module.exports = {\n", T.intercalate ";\n" $ concatMap makeExportDecls (exportDecls ci), "\n};\n"]
140 , "}.call(this))"
154 , "}.call(this))"
141 , footer conf
155 , footer conf
142 ]
156 ]