git.iko.soy/elk
on dc795ca help sign in
elk/compiler/Compiler.hs
67 lines 2.3 kB history raw
1{-# OPTIONS_GHC -W #-}
2module Main where
4import Control.Monad (foldM)
5import qualified Data.Maybe as Maybe
6import Text.Blaze.Html.Renderer.String (renderHtml)
7import qualified Data.ByteString.Lazy.Char8 as BS
8import qualified System.Console.CmdArgs as CmdArgs
9import System.Directory
10import System.FilePath
11import GHC.Conc
13import Build.Dependencies (getSortedDependencies)
14import qualified Generate.Html as Html
15import qualified Metadata.Prelude as Prelude
16import qualified Build.Utils as Utils
17import qualified Build.Flags as Flag
18import qualified Build.File as File
19import qualified Elm.Internal.Paths as Path
21main :: IO ()
22main = do setNumCapabilities =<< getNumProcessors
23 compileArgs =<< CmdArgs.cmdArgs Flag.flags
25compileArgs :: Flag.Flags -> IO ()
26compileArgs flags =
27 case Flag.files flags of
28 [] -> putStrLn "Usage: elm [OPTIONS] [FILES]\nFor more help: elm --help"
29 fs -> mapM_ (build flags) fs
31build :: Flag.Flags -> FilePath -> IO ()
32build flags rootFile =
33 do let noPrelude = Flag.no_prelude flags
34 builtIns <- Prelude.interfaces noPrelude
36 files <- if Flag.make flags
37 then getSortedDependencies (Flag.src_dir flags) builtIns rootFile
38 else return [rootFile]
40 moduleName <- File.build flags builtIns files
42 js <- foldM appendToOutput BS.empty files
44 (extension, code) <-
45 if Flag.only_js flags
46 then do putStr "Generating JavaScript ... "
47 return ("js", js)
48 else do putStr "Generating HTML ... "
49 return (makeHtml js moduleName)
51 let targetFile = Utils.buildPath flags rootFile extension
52 createDirectoryIfMissing True (takeDirectory targetFile)
53 BS.writeFile targetFile code
54 putStrLn "Done"
56 where
57 appendToOutput :: BS.ByteString -> FilePath -> IO BS.ByteString
58 appendToOutput js filePath = do
59 src <- BS.readFile (Utils.elmo flags filePath)
60 return (BS.append src js)
62 sources js = map Html.Link (Flag.scripts flags) ++ [ Html.Source js ]
64 makeHtml js moduleName = ("html", BS.pack $ renderHtml html)
65 where
66 rtsPath = Maybe.fromMaybe Path.runtime (Flag.runtime flags)
67 html = Html.generate rtsPath (takeBaseName rootFile) (sources js) moduleName ""