acid-state-dist-0.1.0.1: examples/HelloWorldMaster.hs

{-# LANGUAGE DeriveDataTypeable #-}
{-# LANGUAGE TemplateHaskell    #-}
{-# LANGUAGE TypeFamilies       #-}

import           Control.Monad.Reader
import           Control.Monad.State
import           Data.Acid
import           Data.Acid.Centered
import           Data.SafeCopy
import           Data.Typeable
import           System.Environment
import           System.Exit (exitSuccess)

------------------------------------------------------
-- The Haskell structure that we want to encapsulate

data HelloWorldState = HelloWorldState String
    deriving (Show, Typeable)

$(deriveSafeCopy 0 'base ''HelloWorldState)

------------------------------------------------------
-- The transaction we will execute over the state.

writeState :: String -> Update HelloWorldState ()
writeState newValue
    = put (HelloWorldState newValue)

queryState :: Query HelloWorldState String
queryState = do HelloWorldState string <- ask
                return string

$(makeAcidic ''HelloWorldState ['writeState, 'queryState])

main :: IO ()
main = do acid <- openMasterState "127.0.0.1" 3333 (HelloWorldState "Hello world")
          putStrLn "Possible commands: x for exit; q for query; uString for update;"
          forever $ do
              input <- getLine
              case input of
                    ('x':_) -> do
                        putStrLn "Bye!"
                        closeAcidState acid
                        exitSuccess
                    ('q':_) -> do
                        string <- query acid QueryState
                        putStrLn $ "The state is: " ++ string
                    ('u':str) -> do
                        update acid (WriteState str)
                        putStrLn "The state has been modified!"
                    _   -> putStrLn $ "Unknown command " ++ input