{-
pandoc-crossref is a pandoc filter for numbering figures,
equations, tables and cross-references to them.
Copyright (C) 2015  Nikolay Yakimov <root@livid.pp.ru>

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
-}

module Text.Pandoc.CrossRef.References.Monad where

import Control.Monad.State
import Data.Function
import Text.Pandoc.CrossRef.References.Types
import Text.Pandoc.CrossRef.Util.Options
import Lens.Micro.TH
import Lens.Micro.Extras

data WState = WState
  { _wsOptions :: Options
  , _wsReferences :: References
  }

makeLenses ''WState

--state monad
newtype WS a = WS (StateT WState ((->) References) a)
  deriving (Functor, Applicative, Monad, MonadState WState)

unwrapWS :: WS a -> WState -> References -> (a, WState)
unwrapWS (WS a) = runStateT a

embedWS :: WState -> WS a -> WS (a, WState)
embedWS wst a = WS $ StateT \oldst refs -> (unwrapWS a wst refs, oldst)

runWS :: WState -> WS a -> (a, WState)
runWS st a = fix $ unwrapWS a st . view wsReferences . snd

liftF :: (References -> a) -> WS a
liftF x = WS $ StateT \wst refs -> (x refs, wst)