hlint

[Lucu.git] / examples / HelloWorld.hs

diff --git a/examples/HelloWorld.hs b/examples/HelloWorld.hs
index 7515347cfedfd2fd385f326f0dbf2d9941ab826a..2d240e8bb8bb0b751cdaf1d56a76c24b159dd82f 100644 (file)
--- a/examples/HelloWorld.hs
+++ b/examples/HelloWorld.hs
@@ -1,42 +1,48 @@
 {-# LANGUAGE
     OverloadedStrings
+  , QuasiQuotes
   , UnicodeSyntax
   #-}
 import Control.Applicative
 import Control.Monad.Unicode
 import qualified Data.ByteString.Lazy.Char8 as Lazy
+import qualified Data.Collections as C
 import Data.Monoid.Unicode
+import Network
 import Network.HTTP.Lucu
+import Prelude.Unicode
 
 main ∷ IO ()
 main = let config    = defaultConfig { cnfServerPort = "9999" }
-           resources = mkResTree
-                       [ ([]         , helloWorld               )
-                       , (["urandom"], staticFile "/dev/urandom")
-                       , (["inc"    ], staticDir "/usr/include" )
-                       ]
-           fallbacks = [ \ path → case path of
-                                     ["hello"] → return $ Just helloWorld
-                                     _         → return Nothing
+           mapper    = resourceMap resources ⊕ resourceMap fallbacks
+           resources ∷ ResourceTree
+           resources = C.fromList
+                       [ ([]          , nonGreedy helloWorld)
+                       , (["urandom" ], nonGreedy $ staticFile "/dev/urandom")
+                       , (["inc"     ], greedy    $ staticDir  "/usr/include")
+                       , (["inc", "t"], nonGreedy $ staticFile "/usr/include/time.h")
                        ]
+           fallbacks ∷ Path → Maybe (Path, Resource)
+           fallbacks path
+               | path ≡ ["hello"] = Just (path, helloWorld)
+               | otherwise        = Nothing
        in
          do putStrLn "Access http://localhost:9999/ with your browser."
-            runHttpd config resources fallbacks
-
+            withSocketsDo $ runHttpd config mapper
 
-helloWorld ∷ ResourceDef
-helloWorld
-    = emptyResource {
-        resGet
-          = Just $ do setContentType $ parseMIMEType "text/hello"
-                      outputChunk "Hello, "
-                      outputChunk "World!\n"
-                      outputChunk =≪ Lazy.pack <$> getRemoteAddr'
-                      
-      , resPost
-          = Just $ do str1 ← inputChunk 3
-                      str2 ← inputChunk 3
-                      str3 ← inputChunk 3
-                      setContentType $ parseMIMEType "text/hello"
-                      output ("[" ⊕ str1 ⊕ " - " ⊕ str2 ⊕ "#" ⊕ str3 ⊕ "]")
-      }
\ No newline at end of file
+helloWorld ∷ Resource
+helloWorld = C.fromList
+             [ ( GET
+               , do setContentType [mimeType| text/hello |]
+                    putChunk "Hello, "
+                    putChunk "World!\n"
+                    putChunks =≪ Lazy.pack <$> getRemoteAddr'
+               )
+             , ( POST
+               , do str1 ← getChunk 3
+                    str2 ← getChunk 3
+                    str3 ← getChunk 3
+                    setContentType [mimeType| text/hello |]
+                    putChunks $ Lazy.fromChunks ["[", str1, " - ", str2, "#", str3, "]"]
+               )
+             ]