From bbf80b87ec4c8b37b844e0b3aa634599ad31af7a Mon Sep 17 00:00:00 2001
From: Jonathan Lamothe <jlamothe1980@gmail.com>
Date: Thu, 20 Dec 2018 22:02:24 -0500
Subject: [PATCH] fixed nextPolicy

---
 src/Password.hs | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/src/Password.hs b/src/Password.hs
index a548781..ffd3d84 100644
--- a/src/Password.hs
+++ b/src/Password.hs
@@ -44,7 +44,7 @@ module Password (
   pwCountUpper, pwCountLower, pwCountDigits, pwCountSpecial
   ) where
 
-import Control.Lens (makeLenses, over, (^.))
+import Control.Lens (makeLenses, over, set, (^.))
 import Data.Char (isUpper, isLower, isDigit, isAlphaNum)
 import qualified Data.ByteString as B
 import qualified Data.Map as M
@@ -220,12 +220,17 @@ mkHash = undefined
 
 nextPolicy :: Char -> PWPolicy -> PWPolicy
 nextPolicy x p = over pwLength pred $
-  over lens pred p where
-    lens
-      | isUpper x = pwUpper
-      | isLower x = pwLower
-      | isDigit x = pwDigits
-      | otherwise = pwSpecial . traverse
+  if isUpper x
+  then dec pwUpper
+  else if isLower x
+  then dec pwLower
+  else if isDigit x
+  then dec pwDigits
+  else case p^.pwSpecial of
+    Nothing -> set pwSpecial (Just (-1)) p
+    Just x  -> set pwSpecial (Just $ pred x) p
+  where
+    dec l = over l (max 0 . pred) p
 
 toUTF8 :: String -> B.ByteString
 toUTF8 = undefined