{-# LANGUAGE FlexibleContexts #-}

module Saturn.Unstable.Type.Month where

import qualified Data.Coerce as Coerce
import qualified Data.Set as Set
import qualified Data.Text.Lazy.Builder as Builder
import qualified Data.Word as Word
import qualified Saturn.Unstable.Type.Field as Field
import qualified Text.Parsec as Parsec

newtype Month
  = Month Field.Field
  deriving (Month -> Month -> Bool
(Month -> Month -> Bool) -> (Month -> Month -> Bool) -> Eq Month
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: Month -> Month -> Bool
== :: Month -> Month -> Bool
$c/= :: Month -> Month -> Bool
/= :: Month -> Month -> Bool
Eq, Int -> Month -> ShowS
[Month] -> ShowS
Month -> String
(Int -> Month -> ShowS)
-> (Month -> String) -> ([Month] -> ShowS) -> Show Month
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> Month -> ShowS
showsPrec :: Int -> Month -> ShowS
$cshow :: Month -> String
show :: Month -> String
$cshowList :: [Month] -> ShowS
showList :: [Month] -> ShowS
Show)

bounds :: (Word.Word8, Word.Word8)
bounds :: (Word8, Word8)
bounds = (Word8
1, Word8
12)

fromField :: Field.Field -> Maybe Month
fromField :: Field -> Maybe Month
fromField Field
field =
  if (Word8, Word8) -> Field -> Bool
Field.isValid (Word8, Word8)
bounds Field
field then Month -> Maybe Month
forall a. a -> Maybe a
Just (Month -> Maybe Month) -> Month -> Maybe Month
forall a b. (a -> b) -> a -> b
$ Field -> Month
Month Field
field else Maybe Month
forall a. Maybe a
Nothing

toField :: Month -> Field.Field
toField :: Month -> Field
toField = Month -> Field
forall a b. Coercible a b => a -> b
Coerce.coerce

parsec :: (Parsec.Stream s m Char) => Parsec.ParsecT s u m Month
parsec :: forall s (m :: * -> *) u. Stream s m Char => ParsecT s u m Month
parsec = do
  Field
field <- ParsecT s u m Field
forall s (m :: * -> *) u. Stream s m Char => ParsecT s u m Field
Field.parsec
  ParsecT s u m Month
-> (Month -> ParsecT s u m Month)
-> Maybe Month
-> ParsecT s u m Month
forall b a. b -> (a -> b) -> Maybe a -> b
maybe (String -> ParsecT s u m Month
forall a. String -> ParsecT s u m a
forall (m :: * -> *) a. MonadFail m => String -> m a
fail String
"invalid Month") Month -> ParsecT s u m Month
forall a. a -> ParsecT s u m a
forall (f :: * -> *) a. Applicative f => a -> f a
pure (Maybe Month -> ParsecT s u m Month)
-> Maybe Month -> ParsecT s u m Month
forall a b. (a -> b) -> a -> b
$ Field -> Maybe Month
fromField Field
field

toBuilder :: Month -> Builder.Builder
toBuilder :: Month -> Builder
toBuilder = Field -> Builder
Field.toBuilder (Field -> Builder) -> (Month -> Field) -> Month -> Builder
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Month -> Field
toField

expand :: Month -> Set.Set Word.Word8
expand :: Month -> Set Word8
expand = (Word8, Word8) -> Field -> Set Word8
Field.expand (Word8, Word8)
bounds (Field -> Set Word8) -> (Month -> Field) -> Month -> Set Word8
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Month -> Field
toField

isMatch :: Word.Word8 -> Month -> Bool
isMatch :: Word8 -> Month -> Bool
isMatch Word8
word8 = Word8 -> Set Word8 -> Bool
forall a. Ord a => a -> Set a -> Bool
Set.member Word8
word8 (Set Word8 -> Bool) -> (Month -> Set Word8) -> Month -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Month -> Set Word8
expand