module Polysemy.Chronos.Time where

import qualified Chronos as Chronos
import Chronos (Timespan(Timespan), dateToDay, dayToDate, dayToTimeMidnight, timeToDayTruncate)

import Polysemy.Chronos.Orphans ()
import Polysemy.Time.At (interpretTimeAt)
import qualified Polysemy.Time.Data.Time as Core
import Polysemy.Time.Data.Time (Time)
import Polysemy.Time.Sleep (tSleep)

-- |Convenience alias for 'Chronos'.
type ChronosTime =
  Time Chronos.Time Chronos.Date

now ::
  Member (Embed IO) r =>
  Sem r Chronos.Time
now :: Sem r Time
now =
  IO Time -> Sem r Time
forall (m :: * -> *) (r :: [(* -> *) -> * -> *]) a.
Member (Embed m) r =>
m a -> Sem r a
embed IO Time
Chronos.now

timeToDate :: Chronos.Time -> Chronos.Date
timeToDate :: Time -> Date
timeToDate =
  Day -> Date
dayToDate (Day -> Date) -> (Time -> Day) -> Time -> Date
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Time -> Day
timeToDayTruncate

dateToTime :: Chronos.Date -> Chronos.Time
dateToTime :: Date -> Time
dateToTime =
  Day -> Time
dayToTimeMidnight (Day -> Time) -> (Date -> Day) -> Date -> Time
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Date -> Day
dateToDay

-- |Interpret 'Time' with the types from 'Chronos'.
interpretTimeChronos ::
  Member (Embed IO) r =>
  InterpreterFor ChronosTime r
interpretTimeChronos :: InterpreterFor ChronosTime r
interpretTimeChronos =
  (forall x (rInitial :: [(* -> *) -> * -> *]).
 ChronosTime (Sem rInitial) x -> Sem r x)
-> Sem (ChronosTime : r) a -> Sem r a
forall (e :: (* -> *) -> * -> *) (r :: [(* -> *) -> * -> *]) a.
FirstOrder e "interpret" =>
(forall x (rInitial :: [(* -> *) -> * -> *]).
 e (Sem rInitial) x -> Sem r x)
-> Sem (e : r) a -> Sem r a
interpret \case
    Core.Now ->
      Sem r x
forall (r :: [(* -> *) -> * -> *]).
Member (Embed IO) r =>
Sem r Time
now
    Core.Today ->
      Time -> Date
timeToDate (Time -> Date) -> Sem r Time -> Sem r Date
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Sem r Time
forall (r :: [(* -> *) -> * -> *]).
Member (Embed IO) r =>
Sem r Time
now
    Core.Sleep t ->
      u -> Sem r ()
forall (r :: [(* -> *) -> * -> *]) t.
(Member (Embed IO) r, TimeUnit t) =>
t -> Sem r ()
tSleep u
t
    Core.SetTime _ ->
      Sem r x
forall (f :: * -> *). Applicative f => f ()
unit
    Core.SetDate _ ->
      Sem r x
forall (f :: * -> *). Applicative f => f ()
unit
{-# INLINE interpretTimeChronos #-}

-- |Interpret 'Time' with the types from 'Chronos', customizing the current time at the start of interpretation.
interpretTimeChronosAt ::
  Member (Embed IO) r =>
  Chronos.Time ->
  InterpreterFor ChronosTime r
interpretTimeChronosAt :: Time -> InterpreterFor ChronosTime r
interpretTimeChronosAt =
  Sem (ChronosTime : r) a -> Sem r a
forall (r :: [(* -> *) -> * -> *]).
Member (Embed IO) r =>
InterpreterFor ChronosTime r
interpretTimeChronos (Sem (ChronosTime : r) a -> Sem r a)
-> (Time -> Sem (ChronosTime : r) a -> Sem (ChronosTime : r) a)
-> Time
-> Sem (ChronosTime : r) a
-> Sem r a
forall c d a b. (c -> d) -> (a -> b -> c) -> a -> b -> d
.: forall t d (r :: [(* -> *) -> * -> *]) a.
(Torsor t Timespan, HasDate t d,
 Members '[Time t d, Embed IO] r) =>
t -> Sem r a -> Sem r a
forall diff t d (r :: [(* -> *) -> * -> *]) a.
(Torsor t diff, HasDate t d, Members '[Time t d, Embed IO] r) =>
t -> Sem r a -> Sem r a
interpretTimeAt @Timespan
{-# INLINE interpretTimeChronosAt #-}

negateTimespan :: Timespan -> Timespan
negateTimespan :: Timespan -> Timespan
negateTimespan (Timespan t :: Int64
t) =
  Int64 -> Timespan
Timespan (-Int64
t)