@@ -3,37 +3,75 @@
3{-# LANGUAGE OverloadedStrings #-}
3{-# LANGUAGE OverloadedStrings #-}
4{-# LANGUAGE TemplateHaskell #-}
4{-# LANGUAGE TemplateHaskell #-}
5{-# LANGUAGE TypeFamilies #-}
5{-# LANGUAGE TypeFamilies #-}
6+{-# LANGUAGE UndecidableInstances #-}
7−module Control.Monad.Validation where
8+module Control.Monad.Validation
11+ , runValidationTEither
28+import Control.Applicative
9import Control.Lens hiding ((.=))
29import Control.Lens hiding ((.=))
10import Control.Monad.Base
30import Control.Monad.Base
11import Control.Monad.Catch
31import Control.Monad.Catch
12import Control.Monad.Except
32import Control.Monad.Except
33+import Control.Monad.Fail
13import Control.Monad.State.Strict
34import Control.Monad.State.Strict
35+import Control.Monad.Trans.Control
14import Data.Aeson
36import Data.Aeson
15import Data.Foldable as F
37import Data.Foldable as F
16import Data.List as L
39import Data.List as L
17import Data.Map.Strict as M
40import Data.Map.Strict as M
19import Data.Text as T
41import Data.Text as T
20import Data.Vector as V
42import Data.Vector as V
21import Test.QuickCheck
43import Test.QuickCheck
45+#if MIN_VERSION_mtl(2,3,0)
47+import Control.Monad.Fix
23−-- | Collects all throwed "warnings" throwed through StateT and "errors" throwed
50+-- | Collects all thrown warnings in 'StateT' and errors
24−-- through ExceptT to single value using Monoid
51+-- in 'ExceptT' into a single value using 'Monoid'.
25−-- FIXME: give more instances like HReaderT and MonadBaseControl/MonadMask
26newtype ValidationT e m a = ValidationT
52newtype ValidationT e m a = ValidationT
27 { unValidationT :: ExceptT e (StateT e m) a
53 { unValidationT :: ExceptT e (StateT e m) a
28 } deriving ( Functor, Applicative, Monad, MonadThrow, MonadCatch
54 } deriving ( Functor, Applicative, Monad, MonadThrow, MonadCatch
29− , MonadBase b )
55+ , MonadBase b, Alternative, MonadFix, MonadFail, Contravariant
56+ , MonadIO, MonadPlus, MonadBaseControl b, MonadMask )
31instance MonadTrans (ValidationT e) where
58instance MonadTrans (ValidationT e) where
32 lift = ValidationT . lift . lift
59 lift = ValidationT . lift . lift
34-- | Map with 'Monoid' instance which 'mappend' its values
61-- | Map with 'Monoid' instance which 'mappend' its values
63+-- This can be used as the `e` in `ValidationT e m a` to provide different
64+-- sets of errors and warnings for different keys.
68+-- [ MonoidMap $ M.fromList [(1, "foo"), (2, "hello, "), (3, "oh no")]
69+-- , MonoidMap $ M.fromList [(1, "bar"), (2, "world")]
72+-- MonoidMap (fromList [(1,"foobar"),(2,"hello, world"),(3,"oh no")])
35newtype MonoidMap k v = MonoidMap (Map k v)
73newtype MonoidMap k v = MonoidMap (Map k v)
36− deriving (Eq, Ord, Show, Arbitrary)
74+ deriving (Eq, Ord, Show, Arbitrary, Foldable)
38makePrisms ''MonoidMap
76makePrisms ''MonoidMap
@@ -51,7 +89,7 @@ instance (Ord k, Semigroup v) => Semigroup (MonoidMap k v) where
52instance (Ord k, Monoid v) => Monoid (MonoidMap k v) where
90instance (Ord k, Monoid v) => Monoid (MonoidMap k v) where
53 mempty = MonoidMap M.empty
91 mempty = MonoidMap M.empty
54− mappend = mmAppend
92+ mappend = (<>)
56instance (ToJSON k, ToJSON v) => ToJSON (MonoidMap k v) where
94instance (ToJSON k, ToJSON v) => ToJSON (MonoidMap k v) where
57 toJSON (MonoidMap m) = toJSON $ L.map toObj $ M.toList m
95 toJSON (MonoidMap m) = toJSON $ L.map toObj $ M.toList m
@@ -61,7 +99,7 @@ instance (ToJSON k, ToJSON v) => ToJSON (MonoidMap k v) where
61 , "value" .= v ]
99 , "value" .= v ]
63instance (Ord k, FromJSON k, FromJSON v) => FromJSON (MonoidMap k v) where
101instance (Ord k, FromJSON k, FromJSON v) => FromJSON (MonoidMap k v) where
64− parseJSON v = withArray "MonoidMap" go v
102+ parseJSON = withArray "MonoidMap" go
66 go arr = do
104 go arr = do
67 keyvals <- traverse fromObj arr
105 keyvals <- traverse fromObj arr
@@ -78,14 +116,14 @@ mmAppend :: (Ord k, Monoid v) => MonoidMap k v -> MonoidMap k v -> MonoidMap k v
79mmAppend (MonoidMap a) (MonoidMap b) = MonoidMap $ M.unionWith (<>) a b
117mmAppend (MonoidMap a) (MonoidMap b) = MonoidMap $ M.unionWith (<>) a b
81−-- | Convenient for 'vZoom' as first artument. Will prevent generation
119+-- | Convenient for 'vZoom' as first argument. Will prevent generation
82−-- of map with 'mempty' values
120+-- of map with 'mempty' values.
83mmSingleton :: (Eq v, Monoid v, Ord k) => k -> v -> MonoidMap k v
121mmSingleton :: (Eq v, Monoid v, Ord k) => k -> v -> MonoidMap k v
84mmSingleton k v
122mmSingleton k v
85 | v == mempty = mempty
123 | v == mempty = mempty
86 | otherwise = MonoidMap . M.singleton k $ v
124 | otherwise = MonoidMap . M.singleton k $ v
88−-- | Set given value to 'mempty'
126+-- | Sets given value to 'mempty'.
89setMempty :: (Monoid s) => ASetter' s a -> a -> s
127setMempty :: (Monoid s) => ASetter' s a -> a -> s
90setMempty setter a = set setter a mempty
128setMempty setter a = set setter a mempty
@@ -94,12 +132,44 @@ neConcat f a
94 | F.null a = Nothing
132 | F.null a = Nothing
95 | otherwise = Just $ F.foldr1 f a
133 | otherwise = Just $ F.foldr1 f a
135+-- | Returns the strings, concatanated with @", "@ if the list is not empty.
137+-- Returns Nothing if the list is empty
139+-- >>> textErrors ["foo", "bar"]
142+-- >>> textErrors ["foo"]
97textErrors :: [Text] -> Maybe Text
147textErrors :: [Text] -> Maybe Text
98textErrors = neConcat (\a b -> a <> ", " <> b)
148textErrors = neConcat (\a b -> a <> ", " <> b)
100−-- | Returns `mempty` instead of error if no warnings was occured. So, your
150+-- | Returns 'mempty' instead of error if no warnings have occurred.
101−-- error should have `Eq` instance to detect that any error was occured. Returns
102−-- Nothing for second element of tuple if compuration was interruped by 'vError'
151+-- Returns 'Nothing' as the second element of tuple if computation was
152+-- interrupted by 'vError'.
154+-- Returns all concatenated errors and warnings and the result if no
155+-- errors have occurred (warnings could have occurred).
158+-- runValidationT $ do
159+-- vWarning ["warning1"]
161+-- vWarning ["warning2"]
164+-- (["error","warning1"],Nothing)
167+-- runValidationT $ do
168+-- vWarning ["warning1"]
169+-- vWarning ["warning2"]
172+-- (["warning1","warning2"],Just 8)
103runValidationT :: (Monoid e, Monad m) => ValidationT e m a -> m (e, Maybe a)
173runValidationT :: (Monoid e, Monad m) => ValidationT e m a -> m (e, Maybe a)
104runValidationT (ValidationT m) = do
174runValidationT (ValidationT m) = do
105 (res, warnings) <- runStateT (runExceptT m) mempty
175 (res, warnings) <- runStateT (runExceptT m) mempty
@@ -107,6 +177,25 @@ runValidationT (ValidationT m) = do
107 Left err -> (err <> warnings, Nothing)
177 Left err -> (err <> warnings, Nothing)
108 Right a -> (warnings, Just a)
178 Right a -> (warnings, Just a)
180+-- | Like 'runValidationT' but doesn't return the result
181+-- if any warning has occurred.
184+-- runValidationTEither $ do
185+-- vWarning ["warning1"]
187+-- vWarning ["warning2"]
190+-- Left ["error","warning1"]
193+-- runValidationTEither $ do
194+-- vWarning ["warning1"]
195+-- vWarning ["warning2"]
198+-- Left ["warning1","warning2"]
110runValidationTEither
199runValidationTEither
111 :: (Monoid e, Eq e, Monad m)
200 :: (Monoid e, Eq e, Monad m)
112 => ValidationT e m a
201 => ValidationT e m a
@@ -117,28 +206,71 @@ runValidationTEither action = do
117 Just a | err == mempty -> Right a
206 Just a | err == mempty -> Right a
118 _ -> Left err
207 _ -> Left err
209+-- | Like 'runValidationTEither', but takes an error handler instead of
210+-- returning errors and warnings.
213+-- handleValidationT (\_ -> return 11) $ do
214+-- vWarning ["warning1"]
216+-- vWarning ["warning2"]
222+-- handleValidationT (\_ -> return 11) $ do
223+-- vWarning ["warning1"]
224+-- vWarning ["warning2"]
120handleValidationT
228handleValidationT
121 :: (Monoid e, Monad m, Eq e)
229 :: (Monoid e, Monad m, Eq e)
122 => (e -> m a)
230 => (e -> m a)
123 -> ValidationT e m a
231 -> ValidationT e m a
125−handleValidationT handler action = do
233+handleValidationT handler action =
126 runValidationTEither action >>= either handler return
234 runValidationTEither action >>= either handler return
128−-- | Stops further execution of validation
236+-- | Stops further execution and appends the given error.
129vError :: (Monad m) => e -> ValidationT e m a
237vError :: (Monad m) => e -> ValidationT e m a
130vError e = ValidationT $ throwError e
238vError e = ValidationT $ throwError e
132−-- | Does not stop further execution, append warning to
240+-- | Does not stop further execution and appends the given warning.
133vWarning :: (Monad m, Monoid e) => e -> ValidationT e m ()
241vWarning :: (Monad m, Monoid e) => e -> ValidationT e m ()
134vWarning e = ValidationT $ modify' (<> e)
242vWarning e = ValidationT $ modify' (<> e)
244+-- | Like 'vError' but allows you to use a setter to insert an error somewhere
245+-- deeper into an empty ('mempty') "e" from "ValidationT e m x", which is then
246+-- combined with all gathered warnings.
136vErrorL :: (Monad m, Monoid e) => ASetter' e a -> a -> ValidationT e m x
247vErrorL :: (Monad m, Monoid e) => ASetter' e a -> a -> ValidationT e m x
137vErrorL l a = vError $ setMempty l a
248vErrorL l a = vError $ setMempty l a
250+-- | Like 'vWarning' but allows you to use a setter to insert an error somewhere
251+-- deeper into an empty ('mempty') "e" from "ValidationT e m x", which is then
252+-- combined with all gathered warnings.
139vWarningL :: (Monad m, Monoid e) => ASetter' e a -> a -> ValidationT e m ()
253vWarningL :: (Monad m, Monoid e) => ASetter' e a -> a -> ValidationT e m ()
140vWarningL l a = vWarning $ setMempty l a
254vWarningL l a = vWarning $ setMempty l a
256+-- | Allows you apply a transformation to the "e" in "ValidationT e m x".
259+--runValidationT . vZoom (Data.Map.singleton "password errors") $ do
260+-- vWarning ["warning1"]
262+-- vWarning ["warning2"]
265+-- (fromList [("password errors",["error","warning1"])],Nothing)
268+-- runValidationT . vZoom (Data.Map.singleton "password errors") $ do
269+-- vWarning ["warning1"]
270+-- vWarning ["warning2"]
273+-- (fromList [("password errors",["warning1","warning2"])],Just 8)
143 :: (Monad m, Monoid a, Monoid b)
275 :: (Monad m, Monoid a, Monoid b)
144 => (a -> b)
276 => (a -> b)
@@ -148,11 +280,21 @@ vZoom up action = do
148 (err, res) <- lift $ runValidationT action
280 (err, res) <- lift $ runValidationT action
149 case res of
281 case res of
150 Nothing -> vError $ up err
282 Nothing -> vError $ up err
151− Just a -> vWarning (up err) *> return a
283+ Just a -> vWarning (up err) $> a
285+-- | Like 'vZoom' but takes a setter instead of a function.
154 :: (Monad m, Monoid a, Monoid b)
287 :: (Monad m, Monoid a, Monoid b)
155 => ASetter' b a
288 => ASetter' b a
156 -> ValidationT a m x
289 -> ValidationT a m x
157 -> ValidationT b m x
290 -> ValidationT b m x
158−vZoomL l action = vZoom (setMempty l) action
291+vZoomL l = vZoom (setMempty l)
293+-- | Turn any warnings the have occurred into errors.
295+ :: (Monad m, Monoid a, Eq a)
296+ => ValidationT a m x
297+ -> ValidationT a m x
299+ err <- lift $ runValidationTEither m
300+ either vError return err