preparation for javascripts

[Rakka.git] / Main.hs

diff --git a/Main.hs b/Main.hs
index 5ce9eb14a79605ae20db7549f39a37a7051d19a9..8027c7ec0978a4caf2aec2cae6ed90cd2c2e3f59 100644 (file)
--- a/Main.hs
+++ b/Main.hs
@@ -6,8 +6,10 @@ import           Network
 import           Network.HTTP.Lucu
 import           Rakka.Environment
 import           Rakka.Resource.Index
+import           Rakka.Resource.JavaScript
 import           Rakka.Resource.Object
 import           Rakka.Resource.Render
+import           Rakka.Storage
 import           Subversion
 import           System.Console.GetOpt
 import           System.Directory
@@ -30,6 +32,7 @@ data CmdOpt
     | OptGroupName String
     | OptLogLevel  Priority
     | OptDisableStderrLog
+    | OptRebuildIndex
     | OptHelp
     deriving (Eq, Show)
 
@@ -76,6 +79,10 @@ options = [ Option ['p'] ["port"]
                    (NoArg OptDisableStderrLog)
                    ("Disable logging to stderr.")
 
+          , Option [] ["rebuild-index"]
+                   (NoArg OptRebuildIndex)
+                   ("Rebuild the index database.")
+
           , Option ['h'] ["help"]
                    (NoArg OptHelp)
                    "Print this message."
@@ -118,7 +125,9 @@ main = withSubversion $
           setupLogger opts
           env <- setupEnv lsdir portNum
 
-          noticeM logger ("Listening to " ++ show portNum ++ "/tcp...")
+          rebuildIndexIfRequested env opts
+
+          infoM logger ("Listening to " ++ show portNum ++ "/tcp...")
           runHttpd (envLucuConf env) (resTree env) [fallbackRender env]
 
           
@@ -126,6 +135,7 @@ resTree :: Environment -> ResTree
 resTree env
     = mkResTree [ ([]        , resIndex  env)
                 , (["object"], resObject env)
+                , (["js"    ], javaScript   )
                 ]
 
 
@@ -202,3 +212,12 @@ createLocalStateDir :: FilePath -> UserID -> GroupID -> IO ()
 createLocalStateDir path uid gid
     = do createDirectoryIfMissing True path
          setOwnerAndGroup path uid gid
+
+
+rebuildIndexIfRequested :: Environment -> [CmdOpt] -> IO ()
+rebuildIndexIfRequested env opts
+    = do let rebuild = isJust $ find (\ x -> case x of
+                                               OptRebuildIndex -> True
+                                               _               -> False) opts
+         when rebuild
+                  $ rebuildIndex (envStorage env)
\ No newline at end of file