]> ruin.nu Git - yawbih.git/blobdiff - Wiki.hs
minor change
[yawbih.git] / Wiki.hs
diff --git a/Wiki.hs b/Wiki.hs
index bcfd23694a025b65363372dbb3f4f1ac0fa48978..5e75da1958e2c494a7ac6e643c939e4c28c4f366 100644 (file)
--- a/Wiki.hs
+++ b/Wiki.hs
@@ -1,21 +1,29 @@
 module Wiki (
-       Backend
+       Backend (getCurrent,getList,get,setCurrent,update,listKeys)
        ,PGB
+       ,createPGB
+       ,Markup (Text, Paragraph, Font, Link, Heading, Url, Pre,List)
+       ,FontOp (Bold, Emph, Mono, Underline, Strike)
+       ,Document
+       ,wikiParser
+       ,toHtml
+       ,toLatex
 
 ) where
 
 import Dbconnect
 import Data.Char
+import Text.ParserCombinators.Parsec
 
 class Backend a where
        --Keyword -> (Full text,date)
        getCurrent :: a -> String -> IO (Maybe (String,String))
 
-       --Keyword -> [(id,date)]
+       --Keyword -> [(id,date,author,comment)]
        getList :: a -> String -> IO [(String, String, String, String)]
 
        --Keyword -> id -> Full text
-       get :: a -> String -> String -> IO String
+       get :: a -> String -> String -> IO (Maybe String)
 
        --Keyword -> id -> ()
        setCurrent :: a -> String -> String -> IO Bool
@@ -23,14 +31,171 @@ class Backend a where
        --Keyword -> Full text -> id
        update :: a -> String -> String -> String -> String -> IO String
 
+       --[Keyword]
+       listKeys :: a -> IO [String]
+
+data Markup = Text String
+       | Paragraph 
+       | Link String String
+       | Font FontOp [Markup]
+       | Heading Int [Markup]
+       | Url String
+       | Pre [Markup]
+       | List Bool [[Markup]]
+
+data FontOp = Bold | Emph | Mono | Underline | Strike
+
+type Document = [Markup]
+
+wikiParser :: Parser Document  
+wikiParser = many1 pMain
+
+firstInLineChars = "#*="
+
+pPara :: Parser Markup 
+pPara = do 
+       count 2 pEol 
+       notFollowedBy $ oneOf firstInLineChars 
+       return (Paragraph)
+
+pLink :: Parser Markup 
+pLink = do
+       string "[["
+       (l,d) <- pLinkContent
+       case d of
+               [] -> return (Link l l)
+               _ -> return (Link l d)
+
+pList :: Bool -> Parser Markup
+pList enum = do
+       list <- many1 $ try $ pListItem enum
+       return (List enum list)
+
+pListItem :: Bool -> Parser [Markup]
+pListItem enum = do
+       pEol
+       char $ listToken enum
+       many (pOneLine <|> try (pOneEol))
+
+pOneEol = do 
+       c <- pEol
+       notFollowedBy (oneOf $ firstInLineChars++"\r\n")
+       return c
+
+listToken True = '#'
+listToken False = '*'
+       
+pHeading :: Parser Markup      
+pHeading = do
+       pEol
+       level <- many1 $ char '='
+       s <- pStopAt level
+       return (Heading (length level) s)
+
+pURL :: Parser Markup  
+pURL = do
+       proto <- many1 letter
+       string "://"
+       s <- many1 (alphaNum <|> oneOf "?.:&-/~%=\\_")
+       return (Url (proto++"://"++s))
+
+pMail :: Parser Markup
+pMail = do
+       user <- many1 (alphaNum <|> oneOf ".-_")
+       char '@'
+       server <- many1 (alphaNum <|> oneOf ".-_")
+       return (Url ("mailto:"++user++"@"++server))
+       
+
+pPre :: Parser Markup  
+pPre = string "<pre>" >> do
+       s <- pUntil (pOneLine <|> pEol) "</pre>"
+       return (Pre s)
+
+pBold,pEmph,pUnderline,pStrike :: Parser Markup        
+pBold = pFont "**" Bold
+pEmph = pFont "//" Emph
+pUnderline = pFont "__" Underline
+pStrike = pFont "--" Strike
+pMono = pFont "||" Mono
+
+pFont s o = string s >> do
+       s <- pStopAt s
+       return (Font o s)
+
+pOtherChar :: Parser Markup    
+pOtherChar = do
+       c <- noneOf "\n\r"
+       return (Text (c:[]))
+
+pText :: Parser Markup 
+pText = do
+       t <- many1 alphaNum
+       return (Text t)
+
+pLinkContent :: Parser (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 -> Parser Document   
+pStopAt xs = pUntil pOneLine xs
+
+pString :: String -> Parser String     
+pString xs = pUntil anyChar xs
+
+pEol :: Parser Markup  
+pEol = do
+       (string "\n" <|> string "\r\n")
+       return (Text "\r\n")
+
+pUntil :: Parser a -> String -> Parser [a]     
+pUntil p xs = manyTill p (try $ string xs)
+       
+pMain :: Parser Markup 
+pMain = choice [
+       try (pHeading)
+       ,try $ pList True
+       ,try $ pList False
+       ,try (pPara) 
+       ,try(pPre) 
+       ,try(pBold) 
+       ,try(pEmph) 
+       ,try(pUnderline) 
+       ,try(pStrike) 
+       ,try(pMono) 
+       ,try (pLink)
+       ,try (pURL)
+       ,try (pMail)
+       ,pText
+       ,pOtherChar
+       ,pEol
+       ]
+
+pOneLine = choice [
+       try pBold
+       ,try pEmph
+       ,try pUnderline
+       ,try pStrike
+       ,try pMono
+       ,try pLink
+       ,try pURL
+       ,try pMail
+       ,pText
+       ,pOtherChar
+       ]
+
 newtype PGB = PGB DBService
 
 createPGB :: String -> String -> String -> String -> IO PGB
 createPGB host database user password = let db = createDBService host database "" user password Nothing in return (PGB db)
 
-
-testDB = createPGB "wave" "wiki" "wiki" "12wiki34db"
-
 instance Backend PGB where
 
        getCurrent (PGB db) key = do 
@@ -41,26 +206,28 @@ instance Backend PGB where
 
        getList (PGB db) key = do
                list <- selectReturnTuples db $ "SELECT id, timestamp, author, comment from fulltexts WHERE keyword = "++tov key
-               return $ map (\[id,date,author,comment] -> (id,date,author,comment)) list
+               case list of
+                       [[]] -> return []
+                       _ -> return $ map (\[id,date,author,comment] -> (id,date,author,comment)) list
 
-       get (PGB db) key id = return ""
+       get (PGB db) key id = do
+               list <- selectReturnTuples db $ "SELECT fulltext from fulltexts WHERE id = "++tov id    
+               case list of
+                       [s]:_ -> return (Just s)
+                       _ -> return Nothing
 
        setCurrent (PGB db) key id = do
                full <- selectReturnTuples db $ "SELECT keyword FROM fulltexts WHERE keyword="++tov key++" AND id='"++id++"'"
-               cur <- selectReturnTuples db $ "SELECT keyword FROM curtexts WHERE keyword='"++tov key++"'"
+               cur <- selectReturnTuples db $ "SELECT keyword FROM curtexts WHERE keyword="++tov key
                case full of
                        [[]] -> do
-                               print "No text found with this id and key"
                                return False
                        _  -> do
                                rows <- case cur of
-                                       [] -> do
-                                               print "No link found in current, inserting"
-                                               execute db $ "INSERT INTO current (keyword, id) VALUES ("++tov id++","++tov key++")"
+                                       [[]] -> do
+                                               execute db $ "INSERT INTO current (keyword, id) VALUES ("++tov key++","++tov id++")"
                                        _  -> do
-                                               print "Link found in current, updating"
                                                execute db $ "UPDATE current SET id = "++tov id++" WHERE keyword = "++tov key
-                               print rows
                                if rows == 1 then return True
                                        else return False 
 
@@ -71,9 +238,59 @@ instance Backend PGB where
                                [[id]] <- selectReturnTuples db $ "SELECT currval('fulltexts_id_seq')" 
                                setCurrent (PGB db) key id
                                return id
-                       
+
+       listKeys (PGB db) = do
+               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++"'"
                
+toHtml :: [Markup] -> String
+toHtml [] = []
+toHtml ((Paragraph):xs) = "<p>\n"++toHtml xs
+toHtml ((Text s):xs) = s++toHtml xs
+toHtml ((Link l d):xs) = "<link: "++l++" desc: "++d++">"++toHtml xs
+toHtml ((Font o d):xs) = "<"++htmlFontOp o++">"++toHtml d++"</"++htmlFontOp o++">"++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>"++toHtml s++"</pre>"++toHtml xs
+toHtml ((List o l):xs) = "<"++htmlListType o++">\n"++(unlines $ map (\s -> "<li>"++toHtml s++"</li>\n") l) ++ "</"++htmlListType o++">"++toHtml xs
+
+htmlFontOp Bold = "b"
+htmlFontOp Emph = "em"
+htmlFontOp Mono = "tt"
+htmlFontOp Underline = "u"
+htmlFontOp Strike = "strike"
+
+htmlListType True = "ol"
+htmlListType False = "ul"
+
+toLatex :: [Markup] -> String
+toLatex [] = []
+toLatex ((Paragraph):xs) = "\n\n"++toLatex xs
+toLatex ((Text s):xs) = s++toLatex xs
+toLatex ((Link l d):xs) = "{\\em "++d++"}"++toLatex xs
+toLatex ((Font o d):xs) = "{\\"++latexFontOp o++" "++toLatex d++"}"++toHtml xs
+toLatex ((Heading n d):xs) = "\n\\"++(unwords $ take (n-1) $ repeat "sub")++"section{"++toLatex d++"}"++"\n\n"++toLatex xs
+toLatex ((Url l):xs) = "{\\bf "++l++"}"++toLatex xs
+toLatex ((Pre s):xs) = "<pre>"++toLatex s++"</pre>"++toLatex xs
+toLatex ((List o l):xs) = "\n\\begin{"++latexListType o++"}\n"++(unlines $ map (\s -> "\\item "++toLatex s++"\n") l) ++ "\\end{"++latexListType o++"}"++toLatex xs
+
+latexFontOp Bold = "bf"
+latexFontOp Emph = "em"
+latexFontOp Mono = "tt"
+latexFontOp Underline = "u"
+latexFontOp Strike = "strike"
+
+latexListType True = "enumeration"
+latexListType False = "itemize"
+
+output s f = case parse wikiParser "" s of
+                                       Right n -> putStr (f n)
+                                       Left e -> print e
 
+htmlOutput s = output s toHtml
+latexOutput s = output s toLatex