13 files changed, 522 insertions, 0 deletions

diff --git a/src/PursLoader/ChildProcess.js b/src/PursLoader/ChildProcess.js
new file mode 100644
index 0000000..d62aef6
--- /dev/null
+++ b/src/PursLoader/ChildProcess.js
@@ -0,0 +1,40 @@
+'use strict';
+
+// module PursLoader.ChildProcess
+
+var child_process = require('child_process');
+
+var chalk = require('chalk');
+
+function spawnFn(command, args, errback, callback) {
+  return function(){
+    var process = child_process.spawn(command, args);
+
+    var stdout = new Buffer(0);
+
+    var stderr = new Buffer(0);
+
+    process.stdout.on('data', function(data){
+      stdout = Buffer.concat([stdout, new Buffer(data)]);
+    });
+
+    process.stderr.on('data', function(data){
+      stderr = Buffer.concat([stderr, new Buffer(data)]);
+    });
+
+    process.on('close', function(code){
+      var output = stdout.toString('utf-8');
+
+      var error = stderr.toString('utf-8');
+
+      if (error.length > 0) {
+         console.error('\n' + chalk.red('*') + ' ' + error);
+      }
+
+      if (code !== 0) errback(new Error('Process terminated with code ' + code))();
+      else callback(output)();
+    });
+  };
+}
+
+exports.spawnFn = spawnFn;
diff --git a/src/PursLoader/ChildProcess.purs b/src/PursLoader/ChildProcess.purs
new file mode 100644
index 0000000..3bd960b
--- /dev/null
+++ b/src/PursLoader/ChildProcess.purs
@@ -0,0 +1,23 @@
+module PursLoader.ChildProcess
+  ( ChildProcess()
+  , spawn
+  ) where
+
+import Prelude (Unit(), ($))
+
+import Control.Monad.Aff (Aff(), makeAff)
+import Control.Monad.Eff (Eff())
+import Control.Monad.Eff.Exception (Error())
+
+import Data.Function
+
+foreign import data ChildProcess :: !
+
+spawn :: forall eff. String -> Array String -> Aff (cp :: ChildProcess | eff) String
+spawn command args = makeAff $ runFn4 spawnFn command args
+
+foreign import spawnFn :: forall eff. Fn4 String
+                                          (Array String)
+                                          (Error -> Eff (cp :: ChildProcess | eff) Unit)
+                                          (String -> Eff (cp :: ChildProcess | eff) Unit)
+                                          (Eff (cp :: ChildProcess | eff) Unit)
diff --git a/src/PursLoader/FS.js b/src/PursLoader/FS.js
new file mode 100644
index 0000000..1a7f5b0
--- /dev/null
+++ b/src/PursLoader/FS.js
@@ -0,0 +1,36 @@
+'use strict';
+
+// module PursLoader.FS
+
+var fs = require('fs');
+
+var async = require('async');
+
+function writeFileUtf8Fn(filepath, contents, errback, callback) {
+  return function(){
+    fs.writeFile(filepath, contents, function(error){
+      if (error) errback(error)();
+      else callback()();
+    });
+  };
+}
+
+function findFileUtf8Fn(nothing, just, regex, filepaths, errback, callback) {
+  return function(){
+    function findFile(filepath, callback) {
+      fs.readFile(filepath, {encoding: 'utf-8'}, function(error, result){
+        if (error) callback(false);
+        else callback(regex.test(result));
+      });
+    }
+
+    async.detect(filepaths, findFile, function(result){
+      if (!result) callback(nothing)();
+      else callback(just(result))();
+    });
+  };
+}
+
+exports.writeFileUtf8Fn = writeFileUtf8Fn;
+
+exports.findFileUtf8Fn = findFileUtf8Fn;
diff --git a/src/PursLoader/FS.purs b/src/PursLoader/FS.purs
new file mode 100644
index 0000000..969e3d0
--- /dev/null
+++ b/src/PursLoader/FS.purs
@@ -0,0 +1,38 @@
+module PursLoader.FS
+  ( FS()
+  , writeFileUtf8
+  , findFileUtf8
+  ) where
+
+import Prelude (Unit(), ($))
+
+import Control.Monad.Aff (Aff(), makeAff)
+import Control.Monad.Eff (Eff())
+import Control.Monad.Eff.Exception (Error())
+
+import Data.Maybe (Maybe(..))
+import Data.String.Regex (Regex())
+
+import Data.Function
+
+foreign import data FS :: !
+
+writeFileUtf8 :: forall eff. String -> String -> Aff (fs :: FS | eff) Unit
+writeFileUtf8 filepath contents = makeAff $ runFn4 writeFileUtf8Fn filepath contents
+
+foreign import writeFileUtf8Fn :: forall eff. Fn4 String
+                                                  String
+                                                  (Error -> Eff (fs :: FS | eff) Unit)
+                                                  (Unit -> Eff (fs :: FS | eff) Unit)
+                                                  (Eff (fs :: FS | eff) Unit)
+
+findFileUtf8 :: forall eff. Regex -> Array String -> Aff (fs :: FS | eff) (Maybe String)
+findFileUtf8 regexp filepaths = makeAff $ runFn6 findFileUtf8Fn Nothing Just regexp filepaths
+
+foreign import findFileUtf8Fn :: forall eff. Fn6 (Maybe String)
+                                                 (String -> Maybe String)
+                                                 Regex
+                                                 (Array String)
+                                                 (Error -> Eff (fs :: FS | eff) Unit)
+                                                 (Maybe String -> Eff (fs :: FS | eff) Unit)
+                                                 (Eff (fs :: FS | eff) Unit)
diff --git a/src/PursLoader/Glob.js b/src/PursLoader/Glob.js
new file mode 100644
index 0000000..960ae9a
--- /dev/null
+++ b/src/PursLoader/Glob.js
@@ -0,0 +1,18 @@
+'use strict';
+
+// module PursLoader.Glob
+
+var glob = require('glob');
+
+var async = require('async');
+
+function globAllFn(patterns, errback, callback) {
+  return function(){
+    async.map(patterns, glob, function(error, result){
+      if (error) errback(new Error(error))();
+      else callback(result)();
+    });
+  };
+}
+
+exports.globAllFn = globAllFn;
diff --git a/src/PursLoader/Glob.purs b/src/PursLoader/Glob.purs
new file mode 100644
index 0000000..45eeb56
--- /dev/null
+++ b/src/PursLoader/Glob.purs
@@ -0,0 +1,22 @@
+module PursLoader.Glob
+  ( Glob()
+  , globAll
+  ) where
+
+import Prelude (Unit(), ($))
+
+import Control.Monad.Aff (Aff(), makeAff)
+import Control.Monad.Eff (Eff())
+import Control.Monad.Eff.Exception (Error())
+
+import Data.Function
+
+foreign import data Glob :: !
+
+globAll :: forall eff. Array String -> Aff (glob :: Glob | eff) (Array (Array String))
+globAll patterns = makeAff $ runFn3 globAllFn patterns
+
+foreign import globAllFn :: forall eff. Fn3 (Array String)
+                                            (Error -> Eff (glob :: Glob | eff) Unit)
+                                            ((Array (Array String)) -> Eff (glob :: Glob | eff) Unit)
+                                            (Eff (glob :: Glob | eff) Unit)
diff --git a/src/PursLoader/Loader.js b/src/PursLoader/Loader.js
new file mode 100644
index 0000000..98459b6
--- /dev/null
+++ b/src/PursLoader/Loader.js
@@ -0,0 +1,19 @@
+'use strict'
+
+// module PursLoader.Loader
+
+var path = require('path');
+
+var cwd = process.cwd();
+
+function relative(from) {
+  return function(to){
+    return path.relative(from, to);
+  };
+}
+
+exports.cwd = cwd;
+
+exports.relative = relative;
+
+exports.resolve = path.resolve;
diff --git a/src/PursLoader/Loader.purs b/src/PursLoader/Loader.purs
new file mode 100644
index 0000000..3cb99cd
--- /dev/null
+++ b/src/PursLoader/Loader.purs
@@ -0,0 +1,105 @@
+module PursLoader.Loader
+  ( Effects()
+  , loader
+  , loaderFn
+  ) where
+
+import Prelude (Unit(), ($), (<>), (>>=), (<$>), (++), bind, flip, id, pure, return, unit)
+
+import Control.Monad.Aff (Aff(), runAff)
+import Control.Monad.Eff (Eff())
+import Control.Monad.Eff.Class (liftEff)
+import Control.Monad.Eff.Exception (error)
+
+import Data.Array ((!!), concat)
+import Data.Function (Fn2(), mkFn2)
+import Data.Maybe (Maybe(..), fromMaybe, maybe)
+import Data.String (joinWith)
+import Data.String.Regex (match, noFlags, regex, test)
+import Data.Traversable (sequence)
+
+import PursLoader.ChildProcess (ChildProcess(), spawn)
+import PursLoader.FS (FS(), writeFileUtf8, findFileUtf8)
+import PursLoader.Glob (Glob(), globAll)
+import PursLoader.LoaderRef (LoaderRef(), Loader(), async, cacheable, query, clearDependencies, addDependency, resourcePath)
+import PursLoader.LoaderUtil (parseQuery)
+import PursLoader.Options (loaderFFIOption, loaderSrcOption, pscOptions)
+
+type Effects eff = (cp :: ChildProcess, fs :: FS, glob :: Glob, loader :: Loader | eff)
+
+moduleRegex = regex "(?:^|\\n)module\\s+([\\w\\.]+)" noFlags { ignoreCase = true }
+
+foreignRegex = regex "(?:^|\\n)\\s*foreign import\\s+" noFlags { ignoreCase = true }
+
+pscCommand = "psc"
+
+psciCommand = "psci"
+
+psciFilename = ".psci"
+
+(!!!) = flip (!!)
+
+foreign import cwd :: String
+
+foreign import relative :: String -> String -> String
+
+foreign import resolve :: String -> String
+
+mkPsci :: Array (Array String) -> Array (Array String) -> String
+mkPsci srcs ffis = joinWith "\n" ((loadModule <$> concat srcs) <> (loadForeign <$> concat ffis))
+  where
+    loadModule :: String -> String
+    loadModule a = ":m " ++ relative cwd a
+
+    loadForeign :: String -> String
+    loadForeign a = ":f " ++ relative cwd a
+
+findFFI :: forall eff. Array (Array String) -> String -> Aff (fs :: FS | eff) (Maybe String)
+findFFI ffiss name = findFileUtf8 re (concat ffiss)
+  where
+    re = regex ("(?:^|\\n)//\\s*module\\s*" ++ name ++ "\\s*\\n") noFlags
+
+loader' :: forall eff. LoaderRef -> String -> Aff (Effects eff) (Maybe String)
+loader' ref source = do
+  liftEff $ cacheable ref
+
+  let parsed = parseQuery $ query ref
+      srcs = fromMaybe [] (loaderSrcOption parsed)
+      ffis = fromMaybe [] (loaderFFIOption parsed)
+      opts = pscOptions parsed
+
+  spawn pscCommand (srcs <> opts)
+
+  srcss <- globAll srcs
+  ffiss <- globAll ffis
+
+  let psciFile = mkPsci srcss ffiss
+
+  writeFileUtf8 psciFilename psciFile
+
+  let moduleName = match moduleRegex source >>= (!!!) 1 >>= id
+      hasForeign = test foreignRegex source
+      result = (\a -> "module.exports = require('" ++ a ++ "');") <$> moduleName
+
+  liftEff (clearDependencies ref)
+  liftEff (addDependency ref (resourcePath ref))
+  liftEff (sequence $ (\src -> addDependency ref (resolve src)) <$> concat srcss)
+
+  foreignPath <- if hasForeign
+                    then fromMaybe (pure Nothing) (findFFI ffiss <$> moduleName)
+                    else pure Nothing
+
+  fromMaybe (pure unit) ((\path -> liftEff (addDependency ref path)) <$> foreignPath)
+
+  return result
+
+loader :: forall eff. LoaderRef -> String -> Eff (Effects eff) Unit
+loader ref source = do
+  callback <- async ref
+  runAff (\e -> callback (Just e) "")
+         (maybe (callback (Just (error "Loader has failed to run")) "")
+                (callback Nothing))
+         (loader' ref source)
+
+loaderFn :: forall eff. Fn2 LoaderRef String (Eff (Effects eff) Unit)
+loaderFn = mkFn2 loader
diff --git a/src/PursLoader/LoaderRef.js b/src/PursLoader/LoaderRef.js
new file mode 100644
index 0000000..3ce0970
--- /dev/null
+++ b/src/PursLoader/LoaderRef.js
@@ -0,0 +1,56 @@
+'use strict';
+
+// module PursLoader.LoaderRef
+
+function asyncFn(isJust, fromMaybe, ref){
+  return function(){
+    var callback = ref.async();
+    return function(error){
+      return function(value){
+        return function(){
+          return isJust(error) ? callback(fromMaybe(new Error())(error))
+                               : callback(null, value);
+        };
+      };
+    };
+  };
+}
+function cacheable(ref){
+  return function(){
+    return ref.cacheable && ref.cacheable();
+  };
+}
+
+function query(ref){
+  return ref.query;
+}
+
+function clearDependencies(ref){
+  return function(){
+    return ref.clearDependencies();
+  };
+}
+
+function resourcePath(ref){
+  return ref.resourcePath;
+}
+
+function addDependency(ref){
+  return function(dep){
+    return function(){
+      return ref.addDependency(dep);
+    };
+  };
+}
+
+exports.asyncFn = asyncFn;
+
+exports.cacheable = cacheable;
+
+exports.query = query;
+
+exports.clearDependencies = clearDependencies;
+
+exports.resourcePath = resourcePath;
+
+exports.addDependency = addDependency;
diff --git a/src/PursLoader/LoaderRef.purs b/src/PursLoader/LoaderRef.purs
new file mode 100644
index 0000000..33c4f7e
--- /dev/null
+++ b/src/PursLoader/LoaderRef.purs
@@ -0,0 +1,40 @@
+module PursLoader.LoaderRef
+  ( LoaderRef()
+  , Loader()
+  , async
+  , cacheable
+  , query
+  , clearDependencies
+  , addDependency
+  , resourcePath
+  ) where
+
+import Prelude (Unit())
+
+import Control.Monad.Eff (Eff())
+import Control.Monad.Eff.Exception (Error())
+
+import Data.Function (Fn3(), runFn3)
+import Data.Maybe (Maybe(), fromMaybe, isJust)
+
+data LoaderRef
+
+foreign import data Loader :: !
+
+foreign import asyncFn :: forall eff a. Fn3 (Maybe Error -> Boolean)
+                                            (Error -> Maybe Error -> Error)
+                                            LoaderRef
+                                            (Eff (loader :: Loader | eff) (Maybe Error -> a -> Eff (loader :: Loader | eff) Unit))
+
+async :: forall eff a. LoaderRef -> Eff (loader :: Loader | eff) (Maybe Error -> a -> Eff (loader :: Loader | eff) Unit)
+async ref = runFn3 asyncFn isJust fromMaybe ref
+
+foreign import cacheable :: forall eff. LoaderRef -> Eff (loader :: Loader | eff) Unit
+
+foreign import query :: LoaderRef -> String
+
+foreign import clearDependencies :: forall eff. LoaderRef -> Eff (loader :: Loader | eff) Unit
+
+foreign import resourcePath :: LoaderRef -> String
+
+foreign import addDependency :: forall eff. LoaderRef -> String -> Eff (loader :: Loader | eff) Unit
diff --git a/src/PursLoader/LoaderUtil.js b/src/PursLoader/LoaderUtil.js
new file mode 100644
index 0000000..45f2703
--- /dev/null
+++ b/src/PursLoader/LoaderUtil.js
@@ -0,0 +1,7 @@
+'use strict';
+
+// module PursLoader.LoaderUtil
+
+var loaderUtils = require('loader-utils');
+
+exports.parseQuery = loaderUtils.parseQuery;
diff --git a/src/PursLoader/LoaderUtil.purs b/src/PursLoader/LoaderUtil.purs
new file mode 100644
index 0000000..3e5a7bc
--- /dev/null
+++ b/src/PursLoader/LoaderUtil.purs
@@ -0,0 +1,9 @@
+module PursLoader.LoaderUtil
+  ( parseQuery
+  ) where
+
+import Data.Foreign (Foreign())
+
+import PursLoader.LoaderRef (LoaderRef())
+
+foreign import parseQuery :: String -> Foreign
diff --git a/src/PursLoader/Options.purs b/src/PursLoader/Options.purs
new file mode 100644
index 0000000..e3957eb
--- /dev/null
+++ b/src/PursLoader/Options.purs
@@ -0,0 +1,109 @@
+module PursLoader.Options
+  ( pscOptions
+  , loaderSrcOption
+  , loaderFFIOption
+  ) where
+
+import Prelude (Unit(), (<>), (<$>), (<<<), (++), (<*>), const)
+
+import Data.Array (concat)
+import Data.Either (either)
+
+import Data.Foreign (Foreign(), F())
+import Data.Foreign.Class (IsForeign, read, readProp)
+import Data.Foreign.NullOrUndefined (NullOrUndefined(..), runNullOrUndefined)
+
+import Data.Maybe (Maybe(..), maybe, fromMaybe)
+
+noPreludeOpt = "no-prelude"
+
+noOptsOpt = "no-opts"
+
+noMagicDoOpt = "no-magic-do"
+
+noTcoOpt = "no-tco"
+
+verboseErrorsOpt = "verbose-errors"
+
+outputOpt = "output"
+
+commentsOpt = "comments"
+
+noPrefixOpt = "no-prefix"
+
+requirePathOpt = "require-path"
+
+srcOpt = "src"
+
+ffiOpt = "ffi"
+
+newtype Options
+  = Options { noPrelude :: NullOrUndefined Boolean
+            , noOpts :: NullOrUndefined Boolean
+            , noMagicDo :: NullOrUndefined Boolean
+            , noTco :: NullOrUndefined Boolean
+            , verboseErrors :: NullOrUndefined Boolean
+            , comments :: NullOrUndefined Boolean
+            , output :: NullOrUndefined String
+            , noPrefix :: NullOrUndefined Boolean
+            , requirePath :: NullOrUndefined String
+            , src :: NullOrUndefined (Array String)
+            , ffi :: NullOrUndefined (Array String)
+            }
+
+instance isForeignOptions :: IsForeign Options where
+  read obj = Options <$> ({ noPrelude: _
+                          , noOpts: _
+                          , noMagicDo: _
+                          , noTco: _
+                          , verboseErrors: _
+                          , comments: _
+                          , output: _
+                          , noPrefix: _
+                          , requirePath: _
+                          , src: _
+                          , ffi: _
+                          } <$> readProp noPreludeOpt obj
+                            <*> readProp noOptsOpt obj
+                            <*> readProp noMagicDoOpt obj
+                            <*> readProp noTcoOpt obj
+                            <*> readProp verboseErrorsOpt obj
+                            <*> readProp commentsOpt obj
+                            <*> readProp outputOpt obj
+                            <*> readProp noPrefixOpt obj
+                            <*> readProp requirePathOpt obj
+                            <*> readProp srcOpt obj
+                            <*> readProp ffiOpt obj)
+
+class LoaderOption a where
+  opt :: String -> NullOrUndefined a -> Array String
+
+instance booleanLoaderOption :: LoaderOption Boolean where
+  opt key val = maybe [] (\a -> if a then ["--" ++ key] else []) (runNullOrUndefined val)
+
+instance stringLoaderOption :: LoaderOption String where
+  opt key val = maybe [] (\a -> ["--" ++ key ++ "=" ++ a]) (runNullOrUndefined val)
+
+instance arrayLoaderOption :: (LoaderOption a) => LoaderOption (Array a) where
+  opt key val = concat (opt key <$> (NullOrUndefined <<< Just)
+                                <$> (fromMaybe [] (runNullOrUndefined val)))
+
+pscOptions :: Foreign -> Array String
+pscOptions query = either (const []) fold parsed
+  where parsed = read query :: F Options
+        fold (Options a) = opt noPreludeOpt a.noPrelude <>
+                           opt noOptsOpt a.noOpts <>
+                           opt noMagicDoOpt a.noMagicDo <>
+                           opt noTcoOpt a.noTco <>
+                           opt verboseErrorsOpt a.verboseErrors <>
+                           opt commentsOpt a.comments <>
+                           opt outputOpt a.output <>
+                           opt noPrefixOpt a.noPrefix <>
+                           opt requirePathOpt a.requirePath <>
+                           opt ffiOpt a.ffi
+
+loaderSrcOption :: Foreign -> Maybe (Array String)
+loaderSrcOption query = either (const Nothing) (\(Options a) -> runNullOrUndefined a.src) (read query)
+
+loaderFFIOption :: Foreign -> Maybe (Array String)
+loaderFFIOption query = either (const Nothing) (\(Options a) -> runNullOrUndefined a.ffi) (read query)