LambdaDB-0.0.0.5: src/Lib.hs

module Lib
  ( dbInit,
    dbProc,
    dbTerm
  ) where

-- Standard libraries
import Data.Data

import Control.Monad
import Control.Monad.Trans.Class
import Control.Monad.Trans.Maybe

-- Custom libraries
import Database.LambdaDB

dbInit :: IO DB
dbInit = do
  putStrLn "Initiate DB ..."
  return initDB

dbProc :: DB -> IO ()
dbProc d = void . runMaybeT $ do
  lift . putStrLn $ "DB is now running"
  void . g $ d
  return ()
  where g db = do
          com <- lift getLine
          newdb <- case com of
                  "" -> lift . return $ db
                  _ ->
                    case (read com) of
                      ComQuit -> mzero
                      ComStatus -> lift . dbStatus $ db
                      ComInsert k v -> lift . dbInsert k v $ db
                      ComFind k -> lift . dbFind k $ db
                      _ -> do
                        lift . putStrLn $ "Command Error"
                        return db
          g newdb

dbStatus :: DB -> IO DB
dbStatus db = do
  putStrLn "On running"
  return db

dbInsert :: Key -> DBData -> DB -> IO DB
dbInsert k v db = do
  return . insertData k v $ db

dbFind :: Key -> DB -> IO DB
dbFind k db = do
  putStrLn . show . findData k $ db
  return db

dbTerm :: IO ()
dbTerm = putStrLn "Terminate DB"