Implemented the outline command

[Rakka.git] / Rakka / Environment.hs

diff --git a/Rakka/Environment.hs b/Rakka/Environment.hs
index 069f9eba3ceb34e8ea2d57c4ccf7551b3c155874..b554df8215e4f7ecf23ba93a073031e79f73c56b 100644 (file)
--- a/Rakka/Environment.hs
+++ b/Rakka/Environment.hs
@@ -1,25 +1,77 @@
 module Rakka.Environment
     ( Environment(..)
+    , InterpTable
     , setupEnv
+    , getInterpTable
+    , getInterpTableA
     )
     where
 
+import           Control.Concurrent.STM
+import           Control.Arrow.ArrowIO
+import           Data.Map (Map)
+import qualified Data.Map as M
 import           Network
 import qualified Network.HTTP.Lucu.Config as LC
+import           Rakka.Storage
+import           Rakka.SystemConfig
+import           Rakka.Wiki.Interpreter
+import           Rakka.Wiki.Interpreter.Base
+import           Subversion.Repository
+import           System.Directory
+import           System.FilePath
 
 
 data Environment = Environment {
-      envLocalStateDir :: FilePath
-    , envLucuConf      :: LC.Config
+      envLocalStateDir :: !FilePath
+    , envLucuConf      :: !LC.Config
+    , envRepository    :: !Repository
+    , envStorage       :: !Storage
+    , envSysConf       :: !SystemConfig
+    , envInterpTable   :: !(TVar InterpTable)
     }
 
 
+type InterpTable = Map String Interpreter
+
+
 setupEnv :: FilePath -> PortNumber -> IO Environment
 setupEnv lsdir portNum
-    = do let lucuConf = LC.defaultConfig {
-                          LC.cnfServerPort = PortNumber portNum
-                        }
+    = do let lucuConf  = LC.defaultConfig {
+                           LC.cnfServerPort = PortNumber portNum
+                         }
+             reposPath = lsdir `combine` "repos"
+             storage   = mkStorage
+         
+         reposExist  <- doesDirectoryExist reposPath
+         repos       <- if reposExist then
+                            openRepository reposPath
+                        else
+                            createRepository reposPath [] []
+         sysConf     <- mkSystemConfig lucuConf repos
+         interpTable <- mkInterpTable
+
          return $ Environment {
                       envLocalStateDir = lsdir
                     , envLucuConf      = lucuConf
-                    }
\ No newline at end of file
+                    , envRepository    = repos
+                    , envStorage       = storage
+                    , envSysConf       = sysConf
+                    , envInterpTable   = interpTable
+                    }
+
+
+mkInterpTable :: IO (TVar InterpTable)
+mkInterpTable = newTVarIO (listToTable baseInterpreters)
+    where
+      listToTable :: [Interpreter] -> InterpTable
+      listToTable xs
+          = M.fromList [ (commandName x, x) | x <- xs ]
+
+
+getInterpTable :: Environment -> IO InterpTable
+getInterpTable = atomically . readTVar . envInterpTable
+
+
+getInterpTableA :: ArrowIO a => Environment -> a b InterpTable
+getInterpTableA = arrIO0 . getInterpTable