{-# LANGUAGE FlexibleContexts #-}

module Saturn.Unstable.Type.Day 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 Day
  = Day Field.Field
  deriving (Day -> Day -> Bool
(Day -> Day -> Bool) -> (Day -> Day -> Bool) -> Eq Day
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
$c== :: Day -> Day -> Bool
== :: Day -> Day -> Bool
$c/= :: Day -> Day -> Bool
/= :: Day -> Day -> Bool
Eq, Int -> Day -> ShowS
[Day] -> ShowS
Day -> String
(Int -> Day -> ShowS)
-> (Day -> String) -> ([Day] -> ShowS) -> Show Day
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
$cshowsPrec :: Int -> Day -> ShowS
showsPrec :: Int -> Day -> ShowS
$cshow :: Day -> String
show :: Day -> String
$cshowList :: [Day] -> ShowS
showList :: [Day] -> ShowS
Show)

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

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

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

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

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

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

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