{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE TypeOperators #-}

module Web.Zwaluw (
    -- * Types
    Router, (:-)(..), (<>),
    
    -- * Running routers
    parse, unparse,
    parse1, unparse1,
    
    -- * Constructing routers
    -- | The @constrN@ functions are helper functions to lift constructors of
    -- datatypes to routers. Their first argument is the constructor; their
    -- second argument is a (partial) destructor.
    constr0, constr1, constr2,
    int, slash, lit
  ) where

import Prelude hiding ((.), id)
import Control.Monad
import Control.Category
import Control.Arrow (first)
import Data.Monoid

infixr 8 <>
infixr 8 :-

-- | Infix operator for 'mappend'.
(<>) :: Monoid m => m -> m -> m
(<>) = mappend

data Router a b = Router
  { ser :: b -> [(a, String)]
  , prs :: String -> [(a -> b, String)] }

data a :- b = a :- b deriving (Eq, Show)

xmap :: (b -> a) -> (a -> b) -> Router r a -> Router r b
xmap f g (Router s p) = Router (s . f) ((fmap . liftM . first . fmap) g p)
  
instance Category (Router) where
  id = lit ""
  Router sf pf . Router sg pg = Router 
    (\a -> do
        (b, s) <- sf a
        (c, s') <- sg b
        return (c, s ++ s'))
    (\s -> do
        (f, s') <- pf s
        (g, s'') <- pg s'
        return (f . g, s''))

instance Monoid (Router a b) where
  mempty = Router (const mzero) (const mzero)
  Router sf pf `mappend` Router sg pg = Router 
    (\s -> sf s `mplus` sg s)
    (\s -> pf s `mplus` pg s)

parse :: Router () a -> String -> [a]
parse p = concatMap (\(a, s) -> if (s == "") then [a ()] else []) . prs p

parse1 :: Router () (a :- ()) -> String -> [a]
parse1 p s = map (\(r :- ()) -> r) (parse p s)

unparse :: Router () a -> a -> [String]
unparse p = map snd . ser p

unparse1 :: Router () (a :- ()) -> a -> [String]
unparse1 p x = unparse p (x :- ())

maph :: (b -> a) -> (a -> b) -> Router i (a :- o) -> Router i (b :- o)
maph f g = xmap (\(h :- t) -> f h :- t) (\(h :- t) -> g h :- t)

opt :: Eq a => a -> Router r (a :- r) -> Router r (a :- r)
opt a p = p <> push a

nil :: Router r ([a] :- r)
nil = constr0 [] $ \x -> do [] <- x; Just ()

cons :: Router (a :- [a] :- r) ([a] :- r)
cons = constr2 (:) $ \x -> do a:as <- x; return (a, as)

-- many :: Eq a => (forall r. Router r (a :- r)) -> Router r ([a] :- r)
-- many p = nil <> many1 p

-- many1 :: Eq a => (forall r. Router r (a :- r)) -> Router r ([a] :- r)
-- many1 p = cons . p . many p

satisfy :: (Char -> Bool) -> Router r (Char :- r)
satisfy p = Router
  (\(c :- a) -> if (p c) then return (a, [c]) else mzero)
  (\s -> case s of 
    []     -> mzero
    (c:cs) -> if (p c) then return ((c :-), cs) else mzero)

char :: Router r (Char :- r)
char = satisfy (const True)

digitChar :: Router r (Char :- r)
digitChar = satisfy (\c -> c >= '0' && c <= '9')

digit :: Router r (Int :- r)
digit = maph (head . show) (read . (:[])) digitChar


-- | Routes a constant string.
lit :: String -> Router r r
lit l = Router
  (\b -> return (b, l))
  (\s -> let (s1, s2) = splitAt (length l) s in if s1 == l then return (id, s2) else mzero)

-- | Routes a slash.
slash :: Router r r
slash = lit "/"

-- | Routes any integer.
int :: Router r (Int :- r)
-- int = maph show read $ many1 digitChar
int = Router
  (\(i :- a) -> return (a, show i))
  (\s -> let l = reads s in map (first (:-)) l)



push :: Eq h => h -> Router r (h :- r)
push h = Router 
  (\(h' :- t) -> do guard (h == h'); return (t, ""))
  (\s -> return ((h :-), s))

left :: Router (a :- r) (Either a b :- r)
left = constr1 Left $ \x -> do Left a <- x; return a

right :: Router (b :- r) (Either a b :- r)
right = constr1 Right $ \x -> do Right b <- x; return b

eitherP :: Router r (a :- r) -> Router r (b :- r) -> Router r (Either a b :- r)
eitherP l r = left . l <> right . r

-- | For example:
-- 
-- > nil :: Router r ([a] :- r)
-- > nil = constr0 [] $ \x -> do [] <- x; Just ()
constr0 :: o -> (Maybe o -> Maybe ()) -> Router r (o :- r)
constr0 c d = Router 
  (\(a :- t) -> maybe mzero (\_ -> return (t, "")) (d (return a)))
  (\s -> return ((c :-), s))

-- | For example:
--
-- > left :: Router (a :- r) (Either a b :- r)
-- > left = constr1 Left $ \x -> do Left a <- x; return a
constr1 :: (a -> o) -> (Maybe o -> Maybe a) -> Router (a :- r) (o :- r)
constr1 c d = Router
  (\(a :- t) -> maybe mzero (\a -> return (a :- t, "")) (d (return a)))
  (\s -> return (\(a :- t) -> c a :- t, s))

-- | For example:
--
-- > cons :: Router (a :- [a] :- r) ([a] :- r)
-- > cons = constr2 (:) $ \x -> do a:as <- x; return (a, as)
constr2 :: (a -> b -> o) -> (Maybe o -> Maybe (a, b)) ->
  Router (a :- b :- r) (o :- r)
constr2 c d = Router
  (\(a :- t) ->
    maybe mzero (\(a, b) -> return (a :- b :- t, "")) (d (return a)))
  (\s -> return (\(a :- b :- t) -> c a b :- t, s))