1 files changed, 52 insertions, 25 deletions

diff --git a/src/Options.purs b/src/Options.purs
index b96cddc..c47bebc 100644
--- a/src/Options.purs
+++ b/src/Options.purs
@@ -2,6 +2,7 @@ module PursLoader.Options
   ( pscMakeOptions
   , pscMakeDefaultOutput
   , pscMakeOutputOption
+  , loaderSrcOption
   ) where
 
 import Data.Either (either)
@@ -24,6 +25,12 @@ verboseErrorsOpt = "verbose-errors"
 
 outputOpt = "output"
 
+commentsOpt = "comments"
+
+noPrefixOpt = "no-prefix"
+
+srcOpt = "src"
+
 pscMakeDefaultOutput = "output"
 
 newtype Options
@@ -32,29 +39,42 @@ newtype Options
             , noMagicDo :: NullOrUndefined Boolean
             , noTco :: NullOrUndefined Boolean
             , verboseErrors :: NullOrUndefined Boolean
+            , comments :: NullOrUndefined Boolean
             , output :: NullOrUndefined String
+            , noPrefix :: NullOrUndefined Boolean
+            , src :: NullOrUndefined [String]
             }
 
 instance isForeignOptions :: IsForeign Options where
-  read obj = (\a b c d e f ->
-             Options { noPrelude: a
-                     , noOpts: b
-                     , noMagicDo: c
-                     , noTco: d
-                     , verboseErrors: e
-                     , output: f
-                     }) <$> readProp noPreludeOpt obj
-                        <*> readProp noOptsOpt obj
-                        <*> readProp noMagicDoOpt obj
-                        <*> readProp noTcoOpt obj
-                        <*> readProp verboseErrorsOpt obj
-                        <*> readProp outputOpt obj
-
-booleanOpt :: String -> NullOrUndefined Boolean -> [String]
-booleanOpt key opt = maybe [] (\a -> if a then ["--" ++ key] else []) (runNullOrUndefined opt)
-
-stringOpt :: String -> NullOrUndefined String -> [String]
-stringOpt key opt = maybe [] (\a -> ["--" ++ key ++ "=" ++ a]) (runNullOrUndefined opt)
+  read obj = Options <$> ({ noPrelude: _
+                          , noOpts: _
+                          , noMagicDo: _
+                          , noTco: _
+                          , verboseErrors: _
+                          , comments: _
+                          , output: _
+                          , noPrefix: _
+                          , src: _
+                          } <$> 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 srcOpt obj)
+
+class LoaderOption a where
+  opt :: String -> NullOrUndefined a -> [String]
+
+instance booleanLoaderOption :: LoaderOption Boolean where
+  opt key opt = maybe [] (\a -> if a then ["--" ++ key] else [])
+                         (runNullOrUndefined opt)
+
+instance stringLoaderOption :: LoaderOption String where
+  opt key opt = maybe [] (\a -> ["--" ++ key ++ "=" ++ a])
+                         (runNullOrUndefined opt)
 
 pscMakeOutputOption :: Foreign -> Maybe String
 pscMakeOutputOption query = either (const Nothing)
@@ -64,9 +84,16 @@ pscMakeOutputOption query = either (const Nothing)
 pscMakeOptions :: Foreign -> [String]
 pscMakeOptions query = either (const []) fold parsed
   where parsed = read query :: F Options
-        fold (Options a) = booleanOpt noPreludeOpt a.noPrelude <>
-                           booleanOpt noOptsOpt a.noOpts <>
-                           booleanOpt noMagicDoOpt a.noMagicDo <>
-                           booleanOpt noTcoOpt a.noTco <>
-                           booleanOpt verboseErrorsOpt a.verboseErrors <>
-                           stringOpt outputOpt a.output
+        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
+
+loaderSrcOption :: Foreign -> Maybe [String]
+loaderSrcOption query = either (const Nothing)
+                               (\(Options a) -> runNullOrUndefined a.src)
+                               (read query)