{-# LANGUAGE NamedFieldPuns #-}
{-# LANGUAGE RecordWildCards #-}

module LiveCoding.Migrate.Monad.Trans where

-- base
import Data.Data

-- essence-of-live-coding
import LiveCoding.Cell.Monad.Trans
import LiveCoding.Migrate.Migration

maybeMigrateToState ::
  (Typeable stateInternal', Typeable stateInternal) =>
  State stateT stateInternal ->
  stateInternal' ->
  Maybe (State stateT stateInternal)
maybeMigrateToState State {stateT} stateInternal' = do
  stateInternal <- cast stateInternal'
  return State {..}

{- | Tries to cast the current state into the joint state of a program
   where a state effect has been absorbed into the internal state with 'runStateL' or 'runStateC'.
-}
migrationToState :: Migration
migrationToState = migrationTo2 maybeMigrateToState

maybeMigrateFromState ::
  (Typeable stateInternal', Typeable stateInternal) =>
  State stateT stateInternal ->
  Maybe stateInternal'
maybeMigrateFromState State {stateInternal} = cast stateInternal

-- | Try to extract a state from the current joint state of a program wrapped with 'runStateL' or 'runStateC'.
migrationFromState :: Migration
migrationFromState = constMigrationFrom2 maybeMigrateFromState

-- | Combines 'migrationToState' and 'migrationFromState'.
migrationState :: Migration
migrationState = migrationToState <> migrationFromState