Added lists and mail links + some parser cleanup

[yawbih.git] / Wiki.hs

diff --git a/Wiki.hs b/Wiki.hs
index 83b14b6223ceb7d7e7e59d7c6db4f6592d1bdbf2..1371c453e597351ede57c674fb807afb993ecd5b 100644 (file)
--- a/Wiki.hs
+++ b/Wiki.hs
@@ -2,7 +2,7 @@ module Wiki (
        Backend (getCurrent,getList,get,setCurrent,update,listKeys)
        ,PGB
        ,createPGB
-       ,Markup (Text, Paragraph, Link, Bold, Emph, Heading, Url, Underline, Strike, Pre)
+       ,Markup (Text, Paragraph, Link, Bold, Emph, Heading, Url, Underline, Strike, Pre,List)
        ,Document
        ,wikiParser
        ,htmlOutput
@@ -42,59 +42,36 @@ data Markup = Text String
        | Underline [Markup]
        | Strike [Markup]
        | Pre String
+       | List [[Markup]]
 
 type Document = [Markup]
 
 wikiParser :: GenParser Char st Document       
-wikiParser = do 
-       s <- pMain
-       ss <- (wikiParser <|> return [])
-       return (s:ss)
+wikiParser = many1 pMain
 
-pPara = do
-       pEol
-       pEol
-       return (Paragraph)
-
-pSpace = do
-       space 
-       return (Text " ")
-
-pEol = char '\n' <|> do
-       char '\r'
-       char '\n'
-
-pLinkParser = do try (string "]]" >> return ([],[])) 
-       <|> try (do
-               string " | "
-               d <- pString "]]"
-               return ([],d))
-       <|> (do
-               c <- anyChar
-               (l,d) <- pLinkParser
-               return (c:l,d))
+pPara :: GenParser Char st Markup      
+pPara = count 2 pEol >> return (Paragraph)
 
+pLink :: GenParser Char st Markup      
 pLink = do
        string "[["
-       (l,d) <- pLinkParser
+       (l,d) <- pLinkContent
        case d of
                [] -> return (Link l l)
                _ -> return (Link l d)
 
-pBold = pS "**" (\s -> Bold s)
-pEmph = pS "//" (\s -> Emph s)
-pUnderline = pS "__" (\s -> Underline s)
-pStrike = pS "--" (\s -> Strike s)
-
-pS s f = do
-       string s
-       s <- pStopAt s
-       return (f s)
+pList :: GenParser Char st Markup
+pList = do
+       list <- many1 $ try pListItem 
+       return (List list)
 
-pStopAt xs = pUntil pMain xs
-pString xs = pUntil anyChar xs
-pUntil p xs = manyTill p (try $ string xs)
+pListItem :: GenParser Char st [Markup]
+pListItem = do
+       many1 pEol
+       char '*'
+       many pOneLine
        
+pHeading :: GenParser Char st Markup   
 pHeading = do
        many1 pEol
        level <- many1 $ char '='
@@ -102,38 +79,100 @@ pHeading = do
        s <- pStopAt (' ':level)
        return (Heading (length level) s)
 
+pURL :: GenParser Char st Markup       
 pURL = do
        proto <- many1 letter
        string "://"
        s <- many1 (alphaNum <|> oneOf "?.:&-/")
        return (Url (proto++"://"++s))
 
-pPre = do
-       string "<pre>"
+pMail :: GenParser Char st Markup
+pMail = do
+       user <- many1 (alphaNum <|> oneOf ".-_")
+       char '@'
+       server <- many1 (alphaNum <|> oneOf ".-_")
+       return (Url ("mailto:"++user++"@"++server))
+       
+
+pPre :: GenParser Char st Markup       
+pPre = string "<pre>" >> do
        s <- pString "</pre>"
        return (Pre s)
-       
-pMain = try (pHeading)
-       <|> (try (pPara) 
-       <|> pSpace 
-       <|> try(pPre) 
-       <|> try(pBold) 
-       <|> try(pEmph) 
-       <|> try(pUnderline) 
-       <|> try(pStrike) 
-       <|> try (pLink)
-       <|> try (pURL)
-       <|> pOtherChar
-       <|> pText)      
 
+pBold,pEmph,pUnderline,pStrike :: GenParser Char st Markup     
+pBold = pS "**" (\s -> Bold s)
+pEmph = pS "//" (\s -> Emph s)
+pUnderline = pS "__" (\s -> Underline s)
+pStrike = pS "--" (\s -> Strike s)
+pS s f = string s >> do
+       s <- pStopAt s
+       return (f s)
+
+pOtherChar :: GenParser Char st Markup 
 pOtherChar = do
-       c <- oneOf "*,;.:!?[]()'\"=-%$£<>/\\|"
+       c <- noneOf "\n\r"
        return (Text (c:[]))
 
+pText :: GenParser Char st Markup      
 pText = do
        t <- many1 alphaNum
        return (Text t)
 
+pLinkContent :: GenParser Char st (String, String)     
+pLinkContent = do try (string "]]" >> return ([],[])) 
+       <|> try (do
+               string " | "
+               d <- pString "]]"
+               return ([],d))
+       <|> (do
+               c <- anyChar
+               (l,d) <- pLinkContent
+               return (c:l,d))
+
+pStopAt :: String -> GenParser Char st Document        
+pStopAt xs = pUntil pOneLine xs
+
+pString :: String -> GenParser Char st String  
+pString xs = pUntil anyChar xs
+
+pEol :: GenParser Char st Markup       
+pEol = do
+       (string "\n" <|> string "\r\n")
+       return (Text "\r\n")
+
+pUntil :: GenParser Char st a -> String -> GenParser Char st [a]       
+pUntil p xs = manyTill p (try $ string xs)
+       
+pMain :: GenParser Char st Markup      
+pMain = choice [
+       try (pHeading)
+       ,try pList
+       ,try (pPara) 
+       ,try(pPre) 
+       ,try(pBold) 
+       ,try(pEmph) 
+       ,try(pUnderline) 
+       ,try(pStrike) 
+       ,try (pLink)
+       ,try (pURL)
+       ,try (pMail)
+       ,pText
+       ,pOtherChar
+       ,pEol
+       ]
+
+pOneLine = choice [
+       try pBold
+       ,try pEmph
+       ,try pUnderline
+       ,try pStrike
+       ,try pLink
+       ,try pURL
+       ,try pMail
+       ,pText
+       ,pOtherChar
+       ]
+
 newtype PGB = PGB DBService
 
 createPGB :: String -> String -> String -> String -> IO PGB
@@ -183,12 +222,10 @@ instance Backend PGB where
                                return id
 
        listKeys (PGB db) = do
-               list <- selectReturnTuples db $ "SELECT keyword FROM fulltexts GROUP BY keyword ORDER BY lower(keyword)"
+               list <- selectReturnTuples db $ "SELECT keyword FROM current keyword ORDER BY lower(keyword)"
                case list of
                        [[]] -> return []
                        _ -> mapM (\[key] -> return key) list
-               
-                       
 
 tov :: String -> String
 tov s = '\'':escapeQuery s++"'"
@@ -205,10 +242,8 @@ toHtml ((Strike d):xs) = "<strike>"++toHtml d++"</strike>"++toHtml xs
 toHtml ((Heading n d):xs) = "\n<h"++show n++">"++toHtml d++"</h"++show n++">\n"++toHtml xs
 toHtml ((Url l):xs) = "<link: "++l++">"++toHtml xs
 toHtml ((Pre s):xs) = "<pre>"++s++"</pre>"++toHtml xs
-
+toHtml ((List l):xs) = "<ul>\n"++(unlines $ map (\s -> "<li>"++toHtml s++"</li>\n") l) ++ "</ul>"++toHtml xs
 
 htmlOutput s = case parse wikiParser "" s of
-                                       Right n -> do 
-                                               putStr (toHtml n)
-                                       Left e -> do
-                                               print e
+                                       Right n -> putStr (toHtml n)
+                                       Left e -> print e